#2064: add handling of offline service to dns, ftp and database

src/primaite/simulator/system/services/database/database_service.py

@@ -173,6 +173,9 @@ class DatabaseService(Service):
         :param session_id: The session identifier.
         :return: True if the Status Code is 200, otherwise False.
         """
+        if not super().receive(payload=payload, session_id=session_id, **kwargs):
+            return False
+
         result = {"status_code": 500, "data": []}
         if isinstance(payload, dict) and payload.get("type"):
             if payload["type"] == "connect_request":

src/primaite/simulator/system/services/dns/dns_server.py

@@ -88,10 +88,14 @@ class DNSServer(Service):
 
         :return: True if DNS request returns a valid IP, otherwise, False
         """
+        if not super().receive(payload=payload, session_id=session_id, **kwargs):
+            return False
+
         # The payload should be a DNS packet
         if not isinstance(payload, DNSPacket):
             _LOGGER.debug(f"{payload} is not a DNSPacket")
             return False
+
         # cast payload into a DNS packet
         payload: DNSPacket = payload
         if payload.dns_request is not None:

src/primaite/simulator/system/services/ftp/ftp_server.py

@@ -86,10 +86,10 @@ class FTPServer(FTPServiceABC):
         prevents an FTP request loop - FTP client and servers can exist on
         the same node.
         """
-        if payload.status_code is not None:
+        if not super().receive(payload=payload, session_id=session_id, **kwargs):
             return False
 
-        if not super().receive(payload=payload, session_id=session_id, **kwargs):
+        if payload.status_code is not None:
             return False
 
         self.send(self._process_ftp_command(payload=payload, session_id=session_id), session_id)

src/primaite/simulator/system/services/web_server/web_server.py

@@ -155,12 +155,12 @@ class WebServer(Service):
         :param: payload: The payload to send.
         :param: session_id: The id of the session. Optional.
         """
+        if not super().receive(payload=payload, session_id=session_id, **kwargs):
+            return False
+
         # check if the payload is an HTTPPacket
         if not isinstance(payload, HttpRequestPacket):
             self.sys_log.error("Payload is not an HTTPPacket")
             return False
 
-        if not super().receive(payload=payload, session_id=session_id, **kwargs):
-            return False
-
         return self._process_http_request(payload=payload, session_id=session_id)