From 0054637ace1bfc48f365e346b382fde5cd5a28b4 Mon Sep 17 00:00:00 2001
From: Jeremy Evans <code@jeremyevans.net>
Date: Thu, 8 Aug 2019 15:44:26 -0700
Subject: [PATCH] Do not raise an exception on a closed DRb socket

This rescues some exceptions that could happen with a closed or
shutdown DRb socket.  This can prevent the server from
exiting if an client socket is closed directly after it is
accepted.

Fixes [Bug #8039]
---
 lib/drb/drb.rb       |  2 ++
 test/drb/test_drb.rb | 16 ++++++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/lib/drb/drb.rb b/lib/drb/drb.rb
index caffd5df25..88f066d3ac 100644
--- a/lib/drb/drb.rb
+++ b/lib/drb/drb.rb
@@ -1011,6 +1011,8 @@ def alive?
 
     def set_sockopt(soc) # :nodoc:
       soc.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+    rescue IOError, Errno::ECONNRESET, Errno::EINVAL
+      # closed/shutdown socket, ignore error
     end
   end
 
diff --git a/test/drb/test_drb.rb b/test/drb/test_drb.rb
index 4e8f5cb0ec..f318ba7f22 100644
--- a/test/drb/test_drb.rb
+++ b/test/drb/test_drb.rb
@@ -327,4 +327,20 @@ def test_bug4409
   end
 end
 
+class TestDRbTCP < Test::Unit::TestCase
+  def test_set_sockopt
+    server = DRb::DRbServer.new('druby://:0')
+    host, port, = DRb::DRbTCPSocket.send(:parse_uri, server.uri)
+    socket = TCPSocket.open host, port
+    socket.shutdown
+    socket.close
+    client = DRb::DRbTCPSocket.new(server.uri, socket)
+    assert client
+    client.close
+    server.stop_service
+    server.thread.join
+    (server.thread.group.list - [Thread.current]).each(&:join)
+  end
+end
+
 end
-- 
2.21.0