gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-21 01:55:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

move setting instant rst into a function

Browse Source
This commit is contained in:
Alexander Bersenev
2019-07-25 02:09:10 +05:00
parent f5d41e9aa7
commit c1fdc4c0a3
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mtprotoproxy.py
View File

@@ -765,6 +765,12 @@ def set_bufsizes(sock, recv_buf, send_buf):
try_setsockopt(sock, socket.SOL_SOCKET, socket.SO_SNDBUF, send_buf) try_setsockopt(sock, socket.SOL_SOCKET, socket.SO_SNDBUF, send_buf)
def set_instant_rst(sock):
INSTANT_RST = b"\x01\x00\x00\x00\x00\x00\x00\x00"
if hasattr(socket, "SO_LINGER"):
try_setsockopt(sock, socket.SOL_SOCKET, socket.SO_LINGER, INSTANT_RST)
async def handle_pseudo_tls_handshake(handshake, reader, writer): async def handle_pseudo_tls_handshake(handshake, reader, writer):
global used_handshakes global used_handshakes
@@ -842,10 +848,7 @@ async def handle_handshake(reader, writer):
tls_handshake_result = await handle_pseudo_tls_handshake(handshake, reader, writer) tls_handshake_result = await handle_pseudo_tls_handshake(handshake, reader, writer)
if not tls_handshake_result: if not tls_handshake_result:
if hasattr(socket, "SO_LINGER"): set_instant_rst(writer.get_extra_info("socket"))
INSTANT_RST = b"\x01\x00\x00\x00\x00\x00\x00\x00"
try_setsockopt(writer.get_extra_info("socket"),
socket.SOL_SOCKET, socket.SO_LINGER, INSTANT_RST)
return False return False
reader, writer = tls_handshake_result reader, writer = tls_handshake_result
handshake = await reader.readexactly(HANDSHAKE_LEN) handshake = await reader.readexactly(HANDSHAKE_LEN)