在gets()中从Ruby中损坏的TCP套接字中恢复

时间:2020-03-05 18:53:05  来源:igfitidea点击:

我正在读取TCP套接字上的输入行,类似于以下内容:

class Bla  
  def getcmd
    @sock.gets unless @sock.closed?
  end

  def start     
    srv = TCPServer.new(5000)
    @sock = srv.accept
    while ! @sock.closed?
      ans = getcmd
    end
  end
end

如果端点在getline()运行时终止了连接,则get()会挂起。

我该如何解决?是否需要进行非阻塞或者定时I / O?

解决方案

回答

如果我们相信用于ruby套接字的rdoc,则它们不会实现`gets'。这使我相信,gets是由更高级别的抽象提供的(也许是IO库?),并且可能不知道特定于套接字的事物,例如"连接已关闭"。

尝试使用recvfrom而不是gets

回答

我们可以使用select来查看是否可以安全地从套接字获取,请参见使用此技术的TCPServer的以下实现。

require 'socket'

host, port = 'localhost', 7000

TCPServer.open(host, port) do |server|
  while client = server.accept
    readfds = true
    got = nil
    begin
      readfds, writefds, exceptfds = select([client], nil, nil, 0.1)
      p :r => readfds, :w => writefds, :e => exceptfds

      if readfds
        got = client.gets 
        p got
      end
    end while got
  end
end

这里是一个试图破坏服务器的客户端:

require 'socket'

host, port = 'localhost', 7000

TCPSocket.open(host, port) do |socket|
  socket.puts "Hey there"
  socket.write 'he'
  socket.flush
  socket.close
end

回答

IO#关闭?当阅读器和编写器都关闭时,返回true。
在情况下,@ sock.gets返回nil,然后再次调用getcmd,这将在永无止境的循环中运行。我们可以使用select,也可以在gets返回nil时关闭套接字。