sv/tsun-gen3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use weakrefs to solve circular references

Browse Source 
- cleanup logging
This commit is contained in:
Stefan Allius
2023-10-03 01:31:23 +02:00
parent a8449e8417
commit dfe8bcb01e
1 changed files with 15 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
app/src/proxy.py
View File

@@ -1,34 +1,33 @@
import asyncio, logging, traceback import asyncio, logging, traceback, weakref
from async_stream import AsyncStream from async_stream import AsyncStream
class Proxy: class Proxy:
def __init__ (proxy, reader, writer, addr): def __init__ (proxy, reader, writer, addr):
proxy.ServerStream = AsyncStream(proxy, reader, writer, addr) proxy.__ServerStream = AsyncStream(proxy, reader, writer, addr)
proxy.ClientStream = None proxy.__ClientStream = None
async def server_loop(proxy, addr): async def server_loop(proxy, addr):
logging.info(f'Accept connection from {addr}') logging.info(f'Accept connection from {addr}')
await proxy.ServerStream.loop() await proxy.__ServerStream.loop()
logging.info(f'Close server connection {addr}') logging.info(f'Stopped server connection loop {addr}')
if proxy.ClientStream: if proxy.__ClientStream:
logging.debug ("close client connection") logging.debug ("disconnect client connection")
proxy.ClientStream.close() proxy.__ClientStream.disc()
async def client_loop(proxy, addr): async def client_loop(proxy, addr):
await proxy.ClientStream.loop() await proxy.__ClientStream.loop()
logging.info(f'Close client connection {addr}') logging.info(f'Stopped client connection loop {addr}')
proxy.ServerStream.remoteStream = None proxy.__ClientStream = None
proxy.ClientStream = None
async def CreateClientStream (proxy, stream, host, port): async def CreateClientStream (proxy, host, port):
addr = (host, port) addr = (host, port)
try: try:
logging.info(f'Connected to {addr}') logging.info(f'Connected to {addr}')
connect = asyncio.open_connection(host, port) connect = asyncio.open_connection(host, port)
reader, writer = await connect reader, writer = await connect
proxy.ClientStream = AsyncStream(proxy, reader, writer, addr, stream, server_side=False) proxy.__ClientStream = AsyncStream(proxy, reader, writer, addr, weakref.ref(proxy.__ServerStream), server_side=False)
asyncio.create_task(proxy.client_loop(addr)) asyncio.create_task(proxy.client_loop(addr))
except ConnectionRefusedError as error: except ConnectionRefusedError as error:
@@ -37,7 +36,7 @@ class Proxy:
logging.error( logging.error(
f"Proxy: Exception for {addr}:\n" f"Proxy: Exception for {addr}:\n"
f"{traceback.format_exc()}") f"{traceback.format_exc()}")
return proxy.ClientStream return weakref.ref(proxy.__ClientStream)
def __del__ (proxy): def __del__ (proxy):
logging.debug ("Proxy __del__") logging.info ("Proxy __del__")