Made buffer size a constant and I've put it to 4096.

Browse Source
This commit is contained in:
retoor 2024-12-12 01:41:47 +00:00
parent 74bfdf8ed5
commit 7da095f1d2
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/shadowssh/app.py
View File

@ -4,6 +4,8 @@ from app.app import Application as BaseApplication
from shadowssh import log from shadowssh import log
BUFFER_SIZE = 4096
class Application(BaseApplication): class Application(BaseApplication):
@ -29,7 +31,7 @@ class Application(BaseApplication):
async def stream(self, reader, writer, port): async def stream(self, reader, writer, port):
while True: while True:
chunk = await reader.read(64) chunk = await reader.read(BUFFER_SIZE)
if not chunk: if not chunk:
writer.close() writer.close()
log.info("Forward connection closed.") log.info("Forward connection closed.")