ft-orchestration introduce cache_size param for all used probes

Author: pascal260303

tools/ft-orchestration/src/probe/cento.py

@@ -112,6 +112,7 @@ def __init__(
         verbose: bool = False,
         mtu: int = 2048,
         sudo: bool = False,
+        cache_size=None,
         **kwargs: dict,
     ):
         interfaces_names = [ifc.name for ifc in interfaces]

tools/ft-orchestration/src/probe/interface.py

@@ -88,6 +88,7 @@ def __init__(
         mtu,
         active_timeout,
         inactive_timeout,
+        cache_size,
         **kwargs,
     ):
         """Initialize the local or remote probe interface as object

tools/ft-orchestration/src/probe/nprobe.py

@@ -121,13 +121,16 @@ def __init__(
         verbose: bool = False,
         mtu: int = 2048,
         sudo: bool = False,
+        cache_size=None,
         **kwargs: dict,
     ):
         interfaces_names = [ifc.name for ifc in interfaces]
         self._interfaces = interfaces_names
         self._zero_copy = any(ifc.startswith("zc:") for ifc in self._interfaces)
         settings: NProbeSettings = NProbeSettings(
-            interface=interfaces_names[0], **kwargs
+            interface=interfaces_names[0],
+            max_num_flows=2**cache_size if cache_size else None,
+            **kwargs,
         )
         self._executor = executor
 

tools/ft-orchestration/src/probe/yaf.py

@@ -183,6 +183,7 @@ def __init__(
         mtu: int = 2048,
         sudo: bool = False,
         inactive_timeout: int = 50,
+        cache_size=None,
         **kwargs: dict,
     ):
         if len(interfaces) > 1:
@@ -206,7 +207,11 @@ def __init__(
         kwargs["pcap"] = YafSettings.PCAPOptions(**(kwargs.get("pcap", {})))
         kwargs["tls"] = YafSettings.TLSOptions(**(kwargs.get("tls", {})))
         kwargs["log"] = YafSettings.LoggingOptions(**(kwargs.get("log", {})))
-        self._settings = YafSettings(idle_timeout=inactive_timeout, **kwargs)
+        self._settings = YafSettings(
+            idle_timeout=inactive_timeout,
+            maxflows=2**cache_size if cache_size else None,
+            **kwargs,
+        )
         self._executor = executor
         if isinstance(executor, RemoteExecutor):
             connection: Connection = executor.get_connection()