在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时关闭套接字。