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

be more clear about proxy protocol usage

Browse Source
This commit is contained in:
Alexander Bersenev
2019-08-19 03:31:10 +05:00
parent 1a0977b10e
commit 01fd1a34c2
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

@@ -125,7 +125,7 @@ def init_config():
# allows to connect in tls mode only # allows to connect in tls mode only
conf_dict.setdefault("TLS_ONLY", False) conf_dict.setdefault("TLS_ONLY", False)
# support proxy protocol v1/v2 for incoming connections # accept incoming connections only with proxy protocol v1/v2, use for nginx/haproxy
conf_dict.setdefault("PROXY_PROTOCOL", False) conf_dict.setdefault("PROXY_PROTOCOL", False)
# set the tls domain for the proxy, has an influence only on starting message # set the tls domain for the proxy, has an influence only on starting message
@@ -1001,14 +1001,16 @@ async def handle_handshake(reader, writer):
TLS_START_BYTES = b"\x16\x03\x01\x02\x00\x01\x00\x01\xfc\x03\x03" TLS_START_BYTES = b"\x16\x03\x01\x02\x00\x01\x00\x01\xfc\x03\x03"
if writer.transport.is_closing() or writer.get_extra_info('peername') is None: if writer.transport.is_closing() or writer.get_extra_info("peername") is None:
return False return False
peer = writer.get_extra_info('peername')[:2] peer = writer.get_extra_info("peername")[:2]
if config.PROXY_PROTOCOL: if config.PROXY_PROTOCOL:
ip = peer[0]
peer = await handle_proxy_protocol(reader, peer) peer = await handle_proxy_protocol(reader, peer)
if not peer: if not peer:
print_err("Client from %s sent bad proxy protocol headers" % ip)
await handle_bad_client(reader, writer, None) await handle_bad_client(reader, writer, None)
return False return False