From 7eea7d3201f6c9e89f63058ab5fed59ed2ee0f64 Mon Sep 17 00:00:00 2001
From: Alexander Bersenev <bay@hackerdom.ru>
Date: Wed, 27 Jun 2018 11:13:42 +0500
Subject: [PATCH] replace infinite loop with timeout with while loop, when the
 client is bad

---
 mtprotoproxy.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mtprotoproxy.py b/mtprotoproxy.py
index 0363482..39d7afe 100755
--- 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