rms/tg-ws-proxy
1
0
Fork 0
mirror of https://github.com/Flowseal/tg-ws-proxy.git synced 2026-05-17 00:25:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

possible #626 ref

Browse Source
This commit is contained in:
Flowseal
2026-04-10 00:37:27 +03:00
parent ce83b78bac
commit bc79a5e4c1
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
proxy/tg_ws_proxy.py
View File

@@ -428,6 +428,7 @@ async def _handle_client(reader, writer, secret: bytes):
_server_instance = None
_server_stop_event = None
_client_tasks: Set[asyncio.Task] = set()
async def _run(stop_event: Optional[asyncio.Event] = None):
@@ -437,6 +438,7 @@ async def _run(stop_event: Optional[asyncio.Event] = None):
_ws_pool.reset()
ws_blacklist.clear()
dc_fail_until.clear()
_client_tasks.clear()
if proxy_config.fallback_cfproxy:
user = proxy_config.cfproxy_user_domain
@@ -451,7 +453,9 @@ async def _run(stop_event: Optional[asyncio.Event] = None):
secret_bytes = bytes.fromhex(proxy_config.secret)
def client_cb(r, w):
asyncio.create_task(_handle_client(r, w, secret_bytes))
task = asyncio.create_task(_handle_client(r, w, secret_bytes))
_client_tasks.add(task)
task.add_done_callback(_client_tasks.discard)
server = await asyncio.start_server(client_cb, proxy_config.host, proxy_config.port)
_server_instance = server