Adds EOFError raise for when a session is dead

rapid7 · Mar 15, 2024 · f9decd3 · f9decd3
1 parent 6347e91
commit f9decd3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/ruby_smb/client.rb b/lib/ruby_smb/client.rb
@@ -575,6 +575,9 @@ def recv_packet(encrypt: false)
       begin
         raw_response = dispatcher.recv_packet
       rescue RubySMB::Error::CommunicationError => e
+        # If raw_response is nil the connection is dead
+        raise EOFError if raw_response.nil?
+
         if encrypt
           raise RubySMB::Error::EncryptionError, "Communication error with the "\
             "remote host: #{e.message}. The server supports encryption but was "\