replace infinite loop with timeout with while loop, when the client is bad

2026-03-13 23:03:09 +00:00 · 2018-06-27 11:13:42 +05:00
parent 2e86308e90
commit 7eea7d3201
1 changed files with 2 additions and 2 deletions
--- a/mtprotoproxy.py
+++ b/mtprotoproxy.py
@@ -496,9 +496,9 @@ async def handle_handshake(reader, writer):
        writer = CryptoWrappedStreamWriter(writer, encryptor)
        return reader, writer, proto_tag, user, dc_idx, enc_key + enc_iv

-    while True:
+    while await reader.read(READ_BUF_SIZE):
        # just consume all the data
-        await reader.read(READ_BUF_SIZE)
+        pass

    return False