Fix port data type assignments

mozilla · Jun 8, 2017 · 2f62966 · 2f62966
1 parent 2470c6a
commit 2f62966
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/ssh_scan/client.rb b/lib/ssh_scan/client.rb
@@ -10,7 +10,7 @@ def initialize(ip, port, timeout = 3)
       @ip = ip
       @timeout = timeout
 
-      @port = port
+      @port = port.to_i
       @client_banner = SSHScan::Constants::DEFAULT_CLIENT_BANNER
       @server_banner = nil
       @kex_init_raw = SSHScan::Constants::DEFAULT_KEY_INIT.to_binary_s

diff --git a/lib/ssh_scan/scan_engine.rb b/lib/ssh_scan/scan_engine.rb
@@ -22,7 +22,7 @@ def scan_target(socket, opts)
       timeout = opts["timeout"]
 
       result = SSHScan::Result.new()
-      result.port = port
+      result.port = port.to_i
 
       # Start the scan timer
       result.set_start_time