changeset:   100257:651a6d47bc78
branch:      3.5
parent:      100255:69717fdd9d5a
user:        Martin Panter <vadmium+py@gmail.com>
date:        Thu Feb 18 10:43:55 2016 +0000
files:       Lib/socketserver.py Lib/test/test_socketserver.py Misc/NEWS
description:
Issue #26309: Shut down socketserver request if verify_request() is false

Patch by Aviv Palivoda.


diff -r 69717fdd9d5a -r 651a6d47bc78 Lib/socketserver.py
--- a/Lib/socketserver.py	Wed Feb 17 23:42:46 2016 -0800
+++ b/Lib/socketserver.py	Thu Feb 18 10:43:55 2016 +0000
@@ -319,6 +319,8 @@
             except:
                 self.handle_error(request, client_address)
                 self.shutdown_request(request)
+        else:
+            self.shutdown_request(request)
 
     def handle_timeout(self):
         """Called if no new request arrives within self.timeout.
diff -r 69717fdd9d5a -r 651a6d47bc78 Lib/test/test_socketserver.py
--- a/Lib/test/test_socketserver.py	Wed Feb 17 23:42:46 2016 -0800
+++ b/Lib/test/test_socketserver.py	Thu Feb 18 10:43:55 2016 +0000
@@ -280,6 +280,30 @@
                 socketserver.TCPServer((HOST, -1),
                                        socketserver.StreamRequestHandler)
 
+    def test_shutdown_request_called_if_verify_request_false(self):
+        # Issue #26309: BaseServer should call shutdown_request even if
+        # verify_request is False
+        shutdown_called = False
+
+        class MyServer(socketserver.TCPServer):
+            def verify_request(self, request, client_address):
+                return False
+
+            def shutdown_request(self, request):
+                nonlocal shutdown_called
+                shutdown_called = True
+                super().shutdown_request(request)
+
+        def connect_to_server(proto, addr):
+            s = socket.socket(proto, socket.SOCK_STREAM)
+            s.connect(addr)
+            s.close()
+
+        self.run_server(MyServer,
+                        socketserver.StreamRequestHandler,
+                        connect_to_server)
+        self.assertEqual(shutdown_called, True)
+
 
 class MiscTestCase(unittest.TestCase):
 
diff -r 69717fdd9d5a -r 651a6d47bc78 Misc/NEWS
--- a/Misc/NEWS	Wed Feb 17 23:42:46 2016 -0800
+++ b/Misc/NEWS	Thu Feb 18 10:43:55 2016 +0000
@@ -76,6 +76,10 @@
 Library
 -------
 
+- Issue #26309: In the "socketserver" module, shut down the request (closing
+  the connected socket) when verify_request() returns false.  Patch by Aviv
+  Palivoda.
+
 - Issue #25939: On Windows open the cert store readonly in ssl.enum_certificates.
 
 - Issue #25995: os.walk() no longer uses FDs proportional to the tree depth.