diff --git a/target_client.py b/target_client.py index 3be5800..aa10c0b 100644 --- a/target_client.py +++ b/target_client.py @@ -95,6 +95,7 @@ def remote_client_handler(tcp_remote:socket): break tcp_remote.close() print("proxy close") + close_all() if _local_server is not None: _local_server.close() @@ -128,7 +129,10 @@ def local_client_handler(tcp_server:socket,addr): if(_remote_client is not None): cmd={'device':'client','option':'disconnect','ip':addr[0],'port':addr[1]} data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default') - _remote_client.send(data) + try: + _remote_client.send(data) + except Exception as e: + print("remote close",str(e)) _local_client.remove(addr_info) diff --git a/target_server.py b/target_server.py index 9015d0f..f6c57bb 100644 --- a/target_server.py +++ b/target_server.py @@ -131,7 +131,7 @@ def remote_client_handler(tcp_client_1:socket): break tcp_client_1.close() print("proxy close.") - + close_all() def main():