test OOM by slow read attack

framework/deproxy/deproxy_base.py

@@ -59,6 +59,12 @@ def set_rst_tcp_to_closing_connection(self) -> None:
     def set_lock(self, polling_lock: threading.Lock) -> None:
         self.__polling_lock = polling_lock
 
+    def set_size_of_receiving_buffer(self, new_buffer_size: int) -> None:
+        """Set the size of the receiving buffer."""
+        self.__acquire()
+        self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, new_buffer_size)
+        self.__release()
+
     def _bind(self, address: tuple) -> None:
         """
         Wrapper for `bind` method to add some log details.

framework/test_suite/tester.py

@@ -14,21 +14,21 @@
 import run_config
 from framework.deproxy import deproxy_client, deproxy_manager
 from framework.deproxy.deproxy_auto_parser import DeproxyAutoParser
-from framework.deproxy.deproxy_server import deproxy_srv_factory
+from framework.deproxy.deproxy_server import StaticDeproxyServer, deproxy_srv_factory
 from framework.helpers import clickhouse, dmesg, error, remote, tf_cfg, util
 from framework.helpers.memworker import MemoryChecker
 from framework.helpers.networker import NetWorker
 from framework.helpers.tf_cfg import test_logger
 from framework.helpers.util import fill_template
-from framework.services import base_server, curl_client, external_client
+from framework.services import curl_client, external_client
 from framework.services import tempesta as tfw
 from framework.services import wrk_client
-from framework.services.docker_server import docker_srv_factory
-from framework.services.nginx_server import nginx_srv_factory
+from framework.services.docker_server import DockerServer, docker_srv_factory
+from framework.services.nginx_server import Nginx, nginx_srv_factory
 from framework.services.stateful import Stateful
 
 __author__ = "Tempesta Technologies, Inc."
-__copyright__ = "Copyright (C) 2018-2025 Tempesta Technologies, Inc."
+__copyright__ = "Copyright (C) 2018-2026 Tempesta Technologies, Inc."
 __license__ = "GPL2"
 
 
@@ -282,7 +282,7 @@ def __create_servers(self):
             # Copy description to keep it clean between several tests.
             self.__create_backend(server.copy())
 
-    def get_server(self, sid: str | int) -> base_server.BaseServer:
+    def get_server(self, sid: str | int) -> StaticDeproxyServer | DockerServer | Nginx:
         """Return client with specified id"""
         server = self.__servers.get(sid, None)
         if server is None: