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

read initial tls header byte by byte

Browse Source
This commit is contained in:
Alexander Bersenev
2019-08-16 15:07:27 +05:00
parent e061cd81c4
commit ccc8c3fb14
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mtprotoproxy.py
View File

@@ -974,9 +974,14 @@ async def handle_handshake(reader, writer):
await handle_bad_client(reader, writer, None) await handle_bad_client(reader, writer, None)
return False return False
handshake = await reader.readexactly(TLS_START_LEN) is_tls_handshake = True
handshake = b""
for byte_num in range(TLS_START_LEN):
handshake += await reader.readexactly(1)
if handshake[-1] != TLS_START_BYTES[byte_num]:
is_tls_handshake = False
if handshake == TLS_START_BYTES: if is_tls_handshake:
handshake += await reader.readexactly(TLS_HANDSHAKE_LEN - TLS_START_LEN) handshake += await reader.readexactly(TLS_HANDSHAKE_LEN - TLS_START_LEN)
tls_handshake_result = await handle_pseudo_tls_handshake(handshake, reader, writer, peer) tls_handshake_result = await handle_pseudo_tls_handshake(handshake, reader, writer, peer)