fix: socket#recv may return nil on closed connections

Fixes #74
palkan · Apr 26, 2024 · 0ecc9f0 · 0ecc9f0
1 parent e1f61be
commit 0ecc9f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/lite_cable/server/client_socket/base.rb b/lib/lite_cable/server/client_socket/base.rb
@@ -137,7 +137,7 @@ def each_frame
 
           while socket.wait_readable
             data = socket.respond_to?(:recv) ? socket.recv(2000) : socket.readpartial(2000)
-            break if data.empty?
+            break if data.nil? || data.empty?
 
             framebuffer << data
             while frame = framebuffer.next # rubocop:disable Lint/AssignmentInCondition