Ruby - 查看端口是否打开

Question

我需要一种快速的方法来确定给定端口是否已使用 Ruby 打开。我目前正在摆弄这个：

require 'socket'

def is_port_open?(ip, port)
  begin
    TCPSocket.new(ip, port)
  rescue Errno::ECONNREFUSED
    return false
  end
  return true
end

如果端口打开它会很好，但它的缺点是偶尔它会等待 10-20 秒然后最终超时，抛出 ETIMEOUT 异常（如果端口关闭） .我的问题是：

是否可以将此代码修改为只等待一秒钟（如果到那时我们什么也得不到，则返回 false）还是有更好的方法来检查给定端口是否在给定主机上打开？

编辑：调用 bash 代码也是可以接受的，只要它可以跨平台工作（例如，Mac OS X、*nix 和 Cygwin），尽管我更喜欢 Ruby 代码。

Answer 1

类似以下的方法可能会起作用：

require 'socket'
require 'timeout'

def is_port_open?(ip, port)
  begin
    Timeout::timeout(1) do
      begin
        s = TCPSocket.new(ip, port)
        s.close
        return true
      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
        return false
      end
    end
  rescue Timeout::Error
  end

  return false
end

Answer 2

为了完整起见，Bash 应该是这样的：

$ netcat $HOST $PORT -w 1 -q 0 </dev/null && do_something

-w 1 指定 1 秒的超时时间，-q 0 表示，连接后，一旦stdin 给出EOF（/dev/null 将立即执行）关闭连接。

Bash 也有自己的内置 TCP/UDP 服务，但它们是编译时选项，我没有使用它们编译的 Bash：P

Answer 3

更多 Ruby 惯用语法：

require 'socket'
require 'timeout'

def port_open?(ip, port, seconds=1)
  Timeout::timeout(seconds) do
    begin
      TCPSocket.new(ip, port).close
      true
    rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
      false
    end
  end
rescue Timeout::Error
  false
end

Answer 4

我对克里斯·赖斯的回答略有不同。仍然可以处理一次尝试超时，但也允许多次重试，直到您放弃。

    def is_port_open?(host, port, timeout, sleep_period)
      begin
        Timeout::timeout(timeout) do
          begin
            s = TCPSocket.new(host, port)
            s.close
            return true
          rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
            sleep(sleep_period)
            retry
          end
        end
      rescue Timeout::Error
        return false
      end
    end

Answer 5

我最近想出了这个解决方案，利用 unix lsof 命令：

def port_open?(port)
  !system("lsof -i:#{port}", out: '/dev/null')
end

Answer 6

所有其他现有答案都是不可取的。使用Timeout 是discouraged。也许事情取决于红宝石版本。至少从 2.0 开始，人们可以简单地使用：

Socket.tcp("www.ruby-lang.org", 10567, connect_timeout: 5) {}

对于较旧的 ruby​​，我能找到的最佳方法是使用非阻塞模式，然后使用 select。此处描述：

• https://spin.atomicobject.com/2013/09/30/socket-connection-timeout-ruby/

Answer 7

所有 *nix 平台：

尝试如下 nc / netcat 命令。

`nc -z -w #{timeout_in_seconds} -G #{timeout_in_seconds} #{host} #{port}`
if $?.exitstatus == 0
  #port is open
else
  #refused, port is closed
end

-z 标志可用于告诉 nc 报告打开的端口，而不是启动连接。

-w 标志表示连接和最终网络读取超时

-G 标志是以秒为单位的连接超时

使用 -n 标志来处理 IP 地址而不是主机名。

例子：

# `nc -z -w 1 -G 1 google.com 80`
# `nc -z -w 1 -G 1 -n 123.234.1.18 80`

Answer 8

我的解决方案来源于发布的解决方案。

require 'socket'
def is_port_open?(ip, port)
  begin
    s = Socket.tcp(ip, port, connect_timeout: 5)
    s.close
    return true
  rescue => e
    # possible exceptions:
    # - Errno::ECONNREFUSED
    # - Errno::EHOSTUNREACH
    # - Errno::ETIMEDOUT
    puts "#{e.class}: #{e.message}"
    return false
  end
end