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

This commit is contained in:
Alexander Bersenev
2018-06-27 11:13:42 +05:00
parent 2e86308e90
commit 7eea7d3201

View File

@@ -496,9 +496,9 @@ async def handle_handshake(reader, writer):
writer = CryptoWrappedStreamWriter(writer, encryptor) writer = CryptoWrappedStreamWriter(writer, encryptor)
return reader, writer, proto_tag, user, dc_idx, enc_key + enc_iv 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 # just consume all the data
await reader.read(READ_BUF_SIZE) pass
return False return False