Skip to content

Commit

Permalink
ssl: context wrapped listener failed to supply _context in accept()
Browse files Browse the repository at this point in the history
  • Loading branch information
temoto committed Oct 18, 2020
1 parent f2ebbb8 commit 20161f4
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 1 deletion.
3 changes: 2 additions & 1 deletion eventlet/green/ssl.py
Expand Up @@ -411,7 +411,8 @@ def accept(self):
ssl_version=self.ssl_version,
ca_certs=self.ca_certs,
do_handshake_on_connect=False,
suppress_ragged_eofs=self.suppress_ragged_eofs)
suppress_ragged_eofs=self.suppress_ragged_eofs,
_context=self._context)
return (new_ssl, addr)

def dup(self):
Expand Down
17 changes: 17 additions & 0 deletions tests/ssl_test.py
Expand Up @@ -325,3 +325,20 @@ def accept(listener):
server_to_client.close()

listener.close()

def test_context_wrapped_accept(self):
context = ssl.SSLContext(ssl.PROTOCOL_SERVER)
context.load_cert_chain(tests.certificate_file, tests.private_key_file)

def client(addr):
with ssl.wrap_socket(eventlet.connect(addr)) as client_sock:
client_sock.send(b"success")

with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_sock:
server_sock.bind(('localhost', 0))
server_sock.listen(5)
eventlet.spawn(client, server_sock.getsockname())
with context.wrap_socket(server_sock, server_side=True) as server_tls:
peer, _ = server_tls.accept()
assert peer.recv(32) == b"success"
peer.close()

0 comments on commit 20161f4

Please sign in to comment.