Server#closed_socket? - parameter may be a MiniSSL::Socket (#2596)

puma · Apr 18, 2021 · 28a4cb2 · 28a4cb2
1 parent 7ca15c1
commit 28a4cb2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/puma/server.rb b/lib/puma/server.rb
@@ -169,11 +169,11 @@ def uncork_socket(socket)
       UNPACK_TCP_STATE_FROM_TCP_INFO = "C".freeze
 
       def closed_socket?(socket)
-        return false unless socket.kind_of? TCPSocket
-        return false unless @precheck_closing
+        skt = socket.to_io
+        return false unless skt.kind_of?(TCPSocket) && @precheck_closing
 
         begin
-          tcp_info = socket.getsockopt(Socket::IPPROTO_TCP, Socket::TCP_INFO)
+          tcp_info = skt.getsockopt(Socket::IPPROTO_TCP, Socket::TCP_INFO)
         rescue IOError, SystemCallError
           Thread.current.purge_interrupt_queue if Thread.current.respond_to? :purge_interrupt_queue
           @precheck_closing = false