From 554191bc005a7a9aef6324021e02aff7f7e38afc Mon Sep 17 00:00:00 2001 From: ranchuan Date: Tue, 8 Oct 2024 18:46:04 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=AD=E5=BC=80=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- target_client.py | 6 +++++- target_server.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) 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():