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

optimize reading a handshake by detecting tls early

Browse Source
This commit is contained in:
Alexander Bersenev
2019-08-15 18:26:56 +05:00
parent 7527d402d6
commit e061cd81c4
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mtprotoproxy.py
View File

@@ -963,6 +963,7 @@ async def handle_handshake(reader, writer):
global used_handshakes
TLS_START_BYTES = b"\x16\x03\x01\x02\x00\x01\x00\x01\xfc\x03\x03"
TLS_START_LEN = len(TLS_START_BYTES)
EMPTY_READ_BUF_SIZE = 4096
peer = writer.get_extra_info('peername')[:2]
@@ -973,10 +974,10 @@ async def handle_handshake(reader, writer):
await handle_bad_client(reader, writer, None)
return False
handshake = await reader.readexactly(HANDSHAKE_LEN)
handshake = await reader.readexactly(TLS_START_LEN)
if handshake.startswith(TLS_START_BYTES):
handshake += await reader.readexactly(TLS_HANDSHAKE_LEN - HANDSHAKE_LEN)
if handshake == TLS_START_BYTES:
handshake += await reader.readexactly(TLS_HANDSHAKE_LEN - TLS_START_LEN)
tls_handshake_result = await handle_pseudo_tls_handshake(handshake, reader, writer, peer)
if not tls_handshake_result:
@@ -988,6 +989,7 @@ async def handle_handshake(reader, writer):
if config.TLS_ONLY:
await handle_bad_client(reader, writer, handshake)
return False
handshake += await reader.readexactly(HANDSHAKE_LEN - TLS_START_LEN)
dec_prekey_and_iv = handshake[SKIP_LEN:SKIP_LEN+PREKEY_LEN+IV_LEN]
dec_prekey, dec_iv = dec_prekey_and_iv[:PREKEY_LEN], dec_prekey_and_iv[PREKEY_LEN:]