ui,prefs: allow configuring server's max workers and clients

Related: c356c82

Author: gustavo-iniguez-goya

ui/bin/opensnitch-ui

@@ -221,9 +221,15 @@ Examples:
             ('grpc.max_receive_message_length', maxmsglen),
         )
         print("[server] addr:", args.socket)
+        max_clients = ('grpc.max_allowed_incoming_connections', args.max_clients)
+        cfg_max_clients = cfg.getInt(Config.DEFAULT_SERVER_MAX_CLIENTS, 0)
+        if cfg_max_clients > 0:
+            print(f"[server] limiting the number of allowed nodes to {cfg_max_clients}")
+            max_clients = (max_clients[0], cfg_max_clients)
         if args.max_clients is not None:
-            print("[server] limiting the number of allowed nodes to", args.max_clients)
-            server_options += (('grpc.max_allowed_incoming_connections', args.max_clients),)
+            print(f"[server] limiting the number of allowed nodes to {args.max_clients}")
+            max_clients = (max_clients[0], args.max_clients)
+            server_options += (max_clients,)
 
         service = UIService(app, on_exit, start_in_bg=args.background)
         check_environ()
@@ -232,10 +238,14 @@ Examples:
         # by default, the number of workers are calculated based on the number
         # of cpus: min(32, os.cpu_count() + 4)
         # https://docs.python.org/3.12/library/concurrent.futures.html#concurrent.futures.ThreadPoolExecutor
-        executor = futures.ThreadPoolExecutor()
+
+        cfg_max_workers = cfg.getInt(Config.DEFAULT_SERVER_MAX_WORKERS, 20)
+        executor = None
         if args.max_workers is not None:
             print(f"[server] using {args.max_workers} workers")
             executor = futures.ThreadPoolExecutor(max_workers=args.max_workers)
+        else:
+            executor = futures.ThreadPoolExecutor(max_workers=cfg_max_workers)
         #print("workers:", executor._max_workers)
         #print("thread:", len(executor._threads))
 

ui/opensnitch/dialogs/preferences.py

@@ -101,6 +101,10 @@ def __init__(self, parent=None, appicon=None):
         self.cmdRefreshUIDown.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinUIRefresh, self.REST))
         self.cmdUIDensityUp.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinUIDensity, self.SUM))
         self.cmdUIDensityDown.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinUIDensity, self.REST))
+        self.cmdGrpcWorkersUp.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinGrpcMaxWorkers, self.REST))
+        self.cmdGrpcWorkersDown.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinGrpcMaxWorkers, self.SUM))
+        self.cmdGrpcClientsUp.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinGrpcMaxClients, self.REST))
+        self.cmdGrpcClientsDown.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinGrpcMaxClients, self.SUM))
         self.cmdNodeGcUp.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinNodeGC, self.SUM))
         self.cmdNodeGcDown.clicked.connect(lambda: self._cb_cmd_spin_clicked(self.spinNodeGC, self.REST))
         self.cmdNodeRulesPath.clicked.connect(self._cb_cmd_node_rulespath_clicked)
@@ -208,6 +212,8 @@ def showEvent(self, event):
         self.checkNodeLogMicro.clicked.connect(self._cb_node_needs_update)
         self.comboNodeAddress.currentTextChanged.connect(self._cb_node_needs_update)
         self.comboServerAddr.currentTextChanged.connect(self._cb_node_needs_update)
+        self.spinGrpcMaxWorkers.valueChanged.connect(self._cb_node_needs_update)
+        self.spinGrpcMaxClients.valueChanged.connect(self._cb_node_needs_update)
         self.checkInterceptUnknown.clicked.connect(self._cb_node_needs_update)
         self.checkApplyToNodes.clicked.connect(self._cb_node_needs_update)
         self.comboNodeAction.currentIndexChanged.connect(self._cb_node_needs_update)
@@ -414,6 +420,11 @@ def _load_ui_settings(self):
             server_addr = self.comboServerAddr.itemText(0)
         self.comboServerAddr.setCurrentText(server_addr)
 
+        max_workers = self._cfg.getInt(Config.DEFAULT_SERVER_MAX_WORKERS, 20)
+        self.spinGrpcMaxWorkers.setValue(max_workers)
+        max_clients = self._cfg.getInt(Config.DEFAULT_SERVER_MAX_CLIENTS, 0)
+        self.spinGrpcMaxClients.setValue(max_clients)
+
         self.lineCACertFile.setText(self._cfg.getSettings(Config.AUTH_CA_CERT))
         self.lineCertFile.setText(self._cfg.getSettings(Config.AUTH_CERT))
         self.lineCertKeyFile.setText(self._cfg.getSettings(Config.AUTH_CERTKEY))
@@ -820,6 +831,18 @@ def _save_ui_config(self):
                 self._cfg.setSettings(Config.DEFAULT_SERVER_ADDR, self.comboServerAddr.currentText())
                 self._changes_needs_restart = QC.translate("preferences", "Server address changed")
 
+            old_workers = self._cfg.getInt(Config.DEFAULT_SERVER_MAX_WORKERS, 20)
+            max_workers = self.spinGrpcMaxWorkers.value()
+            if old_workers != max_workers:
+                self._cfg.setSettings(Config.DEFAULT_SERVER_MAX_WORKERS, int(self.spinGrpcMaxWorkers.value()))
+                self._changes_needs_restart = QC.translate("preferences", "Server max workers changed")
+
+            old_clients = self._cfg.getInt(Config.DEFAULT_SERVER_MAX_CLIENTS, 0)
+            max_clients = self.spinGrpcMaxClients.value()
+            if old_clients != max_clients:
+                self._cfg.setSettings(Config.DEFAULT_SERVER_MAX_CLIENTS, int(self.spinGrpcMaxClients.value()))
+                self._changes_needs_restart = QC.translate("preferences", "Server max clients changed")
+
             if savedauthtype != authtype or self.lineCertFile.text() != cert or \
                     self.lineCertKeyFile.text() != certkey or self.lineCACertFile.text() != cacert:
                 self._changes_needs_restart = QC.translate("preferences", "Certificates changed")