noham/trio_http_proxy
1
0
Fork 0
mirror of https://github.com/NohamR/trio_http_proxy.git synced 2025-05-24 14:32:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

allow list

Browse Source
This commit is contained in:
√(noham)² 2024-07-27 22:10:28 +02:00
parent 4811fa459d
commit 21ed85c304
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
trio_http_proxy.py
View File

@ -13,14 +13,15 @@ from itertools import count
from os import getenv from os import getenv
from textwrap import indent from textwrap import indent
from traceback import format_exc from traceback import format_exc
import h11 import h11
import trio import trio
DEFAULT_PORT = 8080 DEFAULT_PORT = 8080
PORT = int(getenv('PORT', DEFAULT_PORT)) PORT = int(getenv('PORT', DEFAULT_PORT))
# List of allowed IP addresses
ALLOWED_IPS = ['127.0.0.1', '82.66.241.83', '136.243.50.102', '178.254.72.180'] + [f'192.168.1.{i}' for i in range(1, 254)]
prn = partial(print, end='') prn = partial(print, end='')
indented = partial(indent, prefix=' ') indented = partial(indent, prefix=' ')
decoded_and_indented = lambda some_bytes: indented(some_bytes.decode()) decoded_and_indented = lambda some_bytes: indented(some_bytes.decode())
@ -30,9 +31,17 @@ CV_DEST_STREAM = ContextVar('dest_stream', default=None)
CV_PIPE_FROM = ContextVar('pipe_from', default=None) CV_PIPE_FROM = ContextVar('pipe_from', default=None)
async def http_proxy(client_stream, _nextid=count(1).__next__): async def http_proxy(client_stream, client_address, _nextid=count(1).__next__):
client_stream.id = _nextid() client_stream.id = _nextid()
CV_CLIENT_STREAM.set(client_stream) CV_CLIENT_STREAM.set(client_stream)
client_ip = client_address[0]
if client_ip not in ALLOWED_IPS:
log(f'Connection from {client_ip} is not allowed.')
await client_stream.aclose()
return
async with client_stream: async with client_stream:
try: try:
dest_stream = await tunnel(client_stream) dest_stream = await tunnel(client_stream)
@ -46,7 +55,7 @@ async def http_proxy(client_stream, _nextid=count(1).__next__):
async def start_server(server=http_proxy, port=PORT): async def start_server(server=http_proxy, port=PORT):
print(f'* Starting {server.__name__} on port {port or "(OS-selected port)"}...') print(f'* Starting {server.__name__} on port {port or "(OS-selected port)"}...')
try: try:
await trio.serve_tcp(server, port) await trio.serve_tcp(lambda stream: server(stream, stream.socket.getpeername()), port)
except KeyboardInterrupt: except KeyboardInterrupt:
print('\nGoodbye for now.') print('\nGoodbye for now.')