Issue #26309: Shut down socketserver request if verify_request() is false

Patch by Aviv Palivoda.
This commit is contained in:
Martin Panter 2016-02-18 10:43:55 +00:00
parent 8f0432ffbb
commit c12fef9aa3
3 changed files with 30 additions and 0 deletions

View File

@ -319,6 +319,8 @@ class BaseServer:
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.

View File

@ -280,6 +280,30 @@ class SocketServerTest(unittest.TestCase):
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):

View File

@ -76,6 +76,10 @@ Core and Builtins
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.