gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-13 23:03:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

implement read timeout, if server sends nothing for 1 min, connection drops

Browse Source
This commit is contained in:
Alexander Bersenev
2026-02-18 02:37:36 +05:00
parent 368669a546
commit 0614c35020
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mtprotoproxy.py
View File

@@ -264,6 +264,9 @@ def init_config():
# telegram servers connect timeout in seconds # telegram servers connect timeout in seconds
conf_dict.setdefault("TG_CONNECT_TIMEOUT", 10) conf_dict.setdefault("TG_CONNECT_TIMEOUT", 10)
# drop connection if no data from telegram server for this many seconds
conf_dict.setdefault("TG_READ_TIMEOUT", 60)
# listen address for IPv4 # listen address for IPv4
conf_dict.setdefault("LISTEN_ADDR_IPV4", "0.0.0.0") conf_dict.setdefault("LISTEN_ADDR_IPV4", "0.0.0.0")
@@ -1584,6 +1587,10 @@ async def do_middleproxy_handshake(proto_tag, dc_idx, cl_ip, cl_port):
async def tg_connect_reader_to_writer(rd, wr, user, rd_buf_size, is_upstream): async def tg_connect_reader_to_writer(rd, wr, user, rd_buf_size, is_upstream):
try: try:
while True: while True:
if not is_upstream:
data = await asyncio.wait_for(rd.read(rd_buf_size),
timeout=config.TG_READ_TIMEOUT)
else:
data = await rd.read(rd_buf_size) data = await rd.read(rd_buf_size)
if isinstance(data, tuple): if isinstance(data, tuple):
data, extra = data data, extra = data
@@ -1605,7 +1612,7 @@ async def tg_connect_reader_to_writer(rd, wr, user, rd_buf_size, is_upstream):
wr.write(data, extra) wr.write(data, extra)
await wr.drain() await wr.drain()
except (OSError, asyncio.IncompleteReadError) as e: except (OSError, asyncio.IncompleteReadError, asyncio.TimeoutError) as e:
# print_err(e) # print_err(e)
pass pass