diff --git a/python/opendht.pyx b/python/opendht.pyx index d859a04b..985f4dd4 100644 --- a/python/opendht.pyx +++ b/python/opendht.pyx @@ -866,6 +866,9 @@ cdef class DhtConfig(object): def setRateLimit(self, ssize_t max_req_per_sec, ssize_t max_peer_req_per_sec): self._config.dht_config.node_config.max_req_per_sec = max_req_per_sec self._config.dht_config.node_config.max_peer_req_per_sec = max_peer_req_per_sec + def setCryptoRateLimit(self, ssize_t max_crypto_per_sec, ssize_t max_peer_crypto_per_sec): + self._config.dht_config.node_config.max_crypto_per_sec = max_crypto_per_sec + self._config.dht_config.node_config.max_peer_crypto_per_sec = max_peer_crypto_per_sec def setProxyInfo(self, str proxy_server, str push_node_id="", str push_token="", str push_topic="", str push_platform="", str user_agent="", Certificate server_ca = Certificate()): self._config.proxy_server = proxy_server.encode() self._config.push_node_id = push_node_id.encode() diff --git a/python/opendht_cpp.pxd b/python/opendht_cpp.pxd index 24768b4d..1d5a1a1a 100644 --- a/python/opendht_cpp.pxd +++ b/python/opendht_cpp.pxd @@ -283,6 +283,8 @@ cdef extern from "opendht/callbacks.h" namespace "dht": string persist_path size_t max_req_per_sec size_t max_peer_req_per_sec + ssize_t max_crypto_per_sec + ssize_t max_peer_crypto_per_sec cppclass SecureDhtConfig: Config node_config Identity id diff --git a/python/tools/pingpong.py b/python/tools/pingpong.py index f708d8d6..a5ca19e6 100755 --- a/python/tools/pingpong.py +++ b/python/tools/pingpong.py @@ -11,6 +11,7 @@ from opendht import aio as dht async def main(): config = dht.DhtConfig() config.setRateLimit(-1, -1) + config.setCryptoRateLimit(-1, -1) ping_node = dht.DhtRunner() pong_node = dht.DhtRunner() diff --git a/tools/perftest.cpp b/tools/perftest.cpp index 92319d4a..484769b0 100644 --- a/tools/perftest.cpp +++ b/tools/perftest.cpp @@ -35,6 +35,8 @@ benchPingPong(unsigned netSize, unsigned n_parallel) DhtRunner::Config config {}; config.dht_config.node_config.max_peer_req_per_sec = -1; config.dht_config.node_config.max_req_per_sec = -1; + config.dht_config.node_config.max_crypto_per_sec = -1; + config.dht_config.node_config.max_peer_crypto_per_sec = -1; DhtRunner ping_node, pong_node; diff --git a/tools/tools_common.h b/tools/tools_common.h index 0fb36517..76f524df 100644 --- a/tools/tools_common.h +++ b/tools/tools_common.h @@ -277,6 +277,8 @@ getDhtConfig(dht_params& params) if (params.no_rate_limit) { config.dht_config.node_config.max_req_per_sec = -1; config.dht_config.node_config.max_peer_req_per_sec = -1; + config.dht_config.node_config.max_crypto_per_sec = -1; + config.dht_config.node_config.max_peer_crypto_per_sec = -1; config.dht_config.node_config.max_searches = -1; config.dht_config.node_config.max_store_size = -1; }