This commit is contained in:
retoor 2024-12-16 21:05:24 +01:00
parent 7da095f1d2
commit cda52d8bd1
2 changed files with 9 additions and 2 deletions

View File

@ -1,5 +1,6 @@
import argparse
import asyncio
from concurrent.futures import ThreadPoolExecutor as Executor
from shadowssh.app import Application
@ -19,6 +20,12 @@ def parse_args():
def serve():
args = parse_args()
app = Application()
executor = Executor(max_workers=500)
loop = asyncio.get_event_loop()
loop.set_default_executor(executor)
asyncio.run(
app.serve(
args.host,

View File

@ -25,7 +25,7 @@ class Application(BaseApplication):
super().__init__(*args, **kwargs)
async def connect(self, host, port):
log.info(f"Connected to {host}:{port}.")
log.info(f"Forwarding to {host}:{port}.")
reader, writer = await asyncio.open_connection(host, port)
return reader, writer
@ -34,7 +34,7 @@ class Application(BaseApplication):
chunk = await reader.read(BUFFER_SIZE)
if not chunk:
writer.close()
log.info("Forward connection closed.")
log.debug("Forward connection closed.")
return
writer.write(chunk)
await writer.drain()