Do cleanup on SSLClients as well

13 years ago · e2b7d211e8
parent ca9d12bb9c
commit e2b7d211e8
1 changed files with 2 additions and 1 deletions
--- a/daemon/node/core.py
+++ b/daemon/node/core.py
@ -27,11 +27,12 @@ while True:
 		for sock in readable:
 			try:
 				data = sock.recv(1024)
+				cur_client = client_map[sock.fileno()]
 				
 				if data:
-					cur_client = client_map[sock.fileno()]
 					cur_client.process_data(data)
 				else:
+					cur_client.end()
 					select_inputs = remove_from_list(select_inputs, sock)
 					print "NOTICE: Client disconnected"
 			except ssl.SSLError, err: