busyloop / net-ssh-simple Goto Github PK

def with_session(host, opts={}, &block)
    opts[:timeout] ||= 60
    opts[:timeout] = 2**32 if opts[:timeout] == 0
    opts[:operation_timeout] ||= 3600
    opts[:operation_timeout] = 2**32 if opts[:operation_timeout] == 0

irb(main):001:0> require 'socket'; Socket.tcp('127.0.0.1', 1, connect_timeout: 2**32).close
RangeError: bignum too big to convert into `long'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:60:in `wait_writable'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:60:in `connect_internal'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:139:in `connect'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:636:in `block in tcp'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:231:in `each'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:231:in `foreach'
	from /home/promote/.rbenv/versions/2.3.3/lib/ruby/2.3.0/socket.rb:626:in `tcp'
	from (irb):1
	from /home/promote/.rbenv/versions/2.3.3/bin/irb:11:in `<main>'

diff --git a/lib/net/ssh/simple.rb b/lib/net/ssh/simple.rb
index f9263e6..42dca56 100755
--- a/lib/net/ssh/simple.rb
+++ b/lib/net/ssh/simple.rb
@@ -602,7 +602,7 @@ module Net
       is_64bit_platform = true
       begin
         Socket.tcp('127.0.0.1', 1, connect_timeout: 2**32).close
-      rescue Errno::EINVAL
+      rescue Errno::EINVAL, RangeError
         is_64bit_platform = false
       rescue
       end