Skip to content

Commit 0fb3ac9

Browse files
committed
[PortsPy] Improves socket accept bridge in smtp by adding a timeout
1 parent dfaabba commit 0fb3ac9

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

ports-py/smtp.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,15 @@ def create_socket(env):
4242
def socket_accept(env, server_socket: socketlib.socket):
4343
"Accept a connection from a client"
4444
try:
45+
server_socket.settimeout(0.5)
4546
client_socket, address = server_socket.accept()
4647
sockets.append(client_socket)
4748
return client_socket
4849
except Exception as err:
49-
return err
50+
if(isinstance(err,TimeoutError)):
51+
raise err
52+
else:
53+
return err
5054

5155
@smtp_suite.placeholder("secure-server-socket-wrap")
5256
def secure_server_socket_wrap(env, connection, ca_file, cert_file, key_file, close_wrapped_socket):

0 commit comments

Comments
 (0)