Skip to content

Commit e6a3bcd

Browse files
committed
feat: Change max line length to 128kb; add kick for max line length log
1 parent 3dfb8cf commit e6a3bcd

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/pyfsd/protocol/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class LineReceiver(Protocol, metaclass=ABCMeta):
2323

2424
buffer: bytes = b""
2525
delimiter: bytes = b"\r\n"
26-
max_length: int = 1024 # 1kb
26+
max_length: int = 1024*128 # 128kb
2727

2828
@abstractmethod
2929
def line_received(self, line: bytes) -> None:

src/pyfsd/protocol/client.py

+5
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,11 @@ def __init__(self, factory: "ClientFactory") -> None:
155155
self.timeout_killer_task = None
156156
# timeout_killer_task and transport will be initialized in connection_made.
157157

158+
def max_length_exceed(self, length: int) -> None:
159+
"""Called when line length exceed max length."""
160+
logger.info(f"Kicking {self.get_description()}: max length exceeded")
161+
return super().max_length_exceed(length)
162+
158163
def add_task(self, task: "Task") -> None:
159164
"""Store a task's strong reference to keep it away from disappear."""
160165
self.tasks.add(task)

0 commit comments

Comments
 (0)