Update.
This commit is contained in:
parent
7da095f1d2
commit
cda52d8bd1
@ -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,
|
||||
|
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user