Add: check if notus via openvasd is enabled in the openvas config. Th…

…en, do not initialize the mqtt client and daemon
greenbone · Nov 16, 2023 · 2a1a121 · 2a1a121
1 parent fae8d23
commit 2a1a121
Showing 1 changed file with 21 additions and 17 deletions.
diff --git a/ospd_openvas/daemon.py b/ospd_openvas/daemon.py
@@ -495,23 +495,6 @@ def __init__(
         self._mqtt_broker_port = mqtt_broker_port
 
     def init(self, server: BaseServer) -> None:
-        notus_handler = NotusResultHandler(self.report_results)
-
-        if self._mqtt_broker_address:
-            client = MQTTClient(
-                self._mqtt_broker_address, self._mqtt_broker_port, "ospd"
-            )
-            daemon = MQTTDaemon(client)
-            subscriber = MQTTSubscriber(client)
-
-            subscriber.subscribe(ResultMessage, notus_handler.result_handler)
-            daemon.run()
-        else:
-            logger.info(
-                "MQTT Broker Adress empty. MQTT disabled. Unable to get Notus"
-                " results."
-            )
-
         self.scan_collection.init()
 
         server.start(self.handle_client_stream)
@@ -520,6 +503,27 @@ def init(self, server: BaseServer) -> None:
 
         self.set_params_from_openvas_settings()
 
+        # Do not init MQTT daemon if Notus runs via openvasd.
+        if not self.scan_only_params.get("openvasd_server"):
+            notus_handler = NotusResultHandler(self.report_results)
+
+            if self._mqtt_broker_address:
+                client = MQTTClient(
+                    self._mqtt_broker_address, self._mqtt_broker_port, "ospd"
+                )
+                daemon = MQTTDaemon(client)
+                subscriber = MQTTSubscriber(client)
+
+                subscriber.subscribe(
+                    ResultMessage, notus_handler.result_handler
+                )
+                daemon.run()
+            else:
+                logger.info(
+                    "MQTT Broker Address empty. MQTT disabled. "
+                    "Unable to get Notus results."
+                )
+
         with self.feed_lock.wait_for_lock():
             self.update_vts()
             self.set_feed_info()