nopasaran 0.2.110__py3-none-any.whl → 0.2.111__py3-none-any.whl

nopasaran/tools/udp_dns_socket_server.py

@@ -2,6 +2,7 @@ import socket
 from dnslib import DNSRecord, RR, QTYPE, A, CNAME, MX, TXT, NS, SOA, PTR, AAAA, SRV, DS, RRSIG, NSEC, DNSKEY
 from nopasaran.definitions.events import EventNames
 import logging
+import time
 
 class UDPDNSSocketServer:
     def __init__(self):
@@ -18,31 +19,46 @@ class UDPDNSSocketServer:
         self.sock.settimeout(timeout)
         logging.info(f"Waiting for UDP packets on port {self.sock.getsockname()[1]} with timeout {timeout} seconds")
 
-        try:
-            data, client_addr = self.sock.recvfrom(512)  # max size of a DNS UDP packet
-            logging.debug(f"Received {len(data)} bytes from {client_addr}")
+        start_time = time.time()
 
-            logging.debug("Parsing DNS query...")
-            parsed_query = DNSRecord.parse(data)
-            logging.info(f"Parsed query:\n{parsed_query.toZone()}")
+        while True:
+            try:
+                remaining_time = timeout - (time.time() - start_time)
+                if remaining_time <= 0:
+                    logging.warning("Timeout reached while waiting for a valid DNS query.")
+                    return {"received": None}, EventNames.TIMEOUT.name
 
-            logging.debug("Building DNS response...")
-            response = self.build_response(parsed_query, response_spec)
+                self.sock.settimeout(remaining_time)
+                data, client_addr = self.sock.recvfrom(512)  # max size of a DNS UDP packet
+                logging.debug(f"Received {len(data)} bytes from {client_addr}")
 
-            logging.debug("Sending DNS response...")
-            self.send_dns_response(client_addr, response)
-            logging.info("DNS response sent successfully.")
+                logging.debug("Parsing DNS query...")
+                parsed_query = DNSRecord.parse(data)
 
-            return {
-                "received": str(parsed_query.q)
-            }, EventNames.REQUEST_RECEIVED.name
+                # Check if query section is empty
+                if not parsed_query.questions:
+                    logging.debug("Received DNS packet with no query section. Continuing to wait...")
+                    continue
 
-        except socket.timeout:
-            logging.warning("Timeout reached with no packet received.")
-            return {"received": None}, EventNames.TIMEOUT.name
-        except Exception as e:
-            logging.error(f"Error while handling DNS query: {e}", exc_info=True)
-            return {"received": None}, EventNames.ERROR.name
+                logging.info(f"Parsed query:\n{parsed_query.toZone()}")
+
+                logging.debug("Building DNS response...")
+                response = self.build_response(parsed_query, response_spec)
+
+                logging.debug("Sending DNS response...")
+                self.send_dns_response(client_addr, response)
+                logging.info("DNS response sent successfully.")
+
+                return {
+                    "received": str(parsed_query.q)
+                }, EventNames.REQUEST_RECEIVED.name
+
+            except socket.timeout:
+                logging.warning("Timeout reached while waiting for a valid DNS query.")
+                return {"received": None}, EventNames.TIMEOUT.name
+            except Exception as e:
+                logging.error(f"Error while handling DNS query: {e}", exc_info=True)
+                return {"received": None}, EventNames.ERROR.name
 
     def build_response(self, query_record, response_spec=None):
         qname = str(query_record.q.qname)

{nopasaran-0.2.110.dist-info → nopasaran-0.2.111.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nopasaran
-Version: 0.2.110
+Version: 0.2.111
 Summary: NoPASARAN is an advanced network tool designed to detect, fingerprint, and locate network middleboxes in a unified framework.
 Home-page: https://github.com/BenIlies/NoPASARAN
 Author: Ilies Benhabbour

{nopasaran-0.2.110.dist-info → nopasaran-0.2.111.dist-info}/RECORD

@@ -77,11 +77,11 @@ nopasaran/tools/http_2_socket_client.py,sha256=5mmc8FuT2dNVAihbMdvyytVwME_xSc7me
 nopasaran/tools/http_2_socket_server.py,sha256=8W-bxmdoGyK5moxpC87V1EGuZ9jJpIKhokvZrDV84kk,5551
 nopasaran/tools/https_1_socket_server.py,sha256=ytLVYdH5je4L6ElCKKhhsbU_53gT6QF9l5ym7q5TMmo,8162
 nopasaran/tools/tcp_dns_socket_server.py,sha256=fZl2ujE3C_1iqyz8qquWnePPAj8n2uzWfk3DUEDlgho,7924
-nopasaran/tools/udp_dns_socket_server.py,sha256=N6_27OTo3aKlihu1YqWfgYeCpdR3Xfo2o7ZE2db9Vuw,5568
+nopasaran/tools/udp_dns_socket_server.py,sha256=BrtuddDgfeCZEt11IZ9wb1uJO4cW0KtlQahfyVrY9sg,6283
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nopasaran-0.2.110.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-nopasaran-0.2.110.dist-info/METADATA,sha256=B6RitCFp4VAB_AUhU8_oIVRb2mqrFZIquCBmb_ofg04,5497
-nopasaran-0.2.110.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-nopasaran-0.2.110.dist-info/entry_points.txt,sha256=LaOz5GlWuMLjzg4KOEB5OVTattCXVW6a4nSW-WQajCw,55
-nopasaran-0.2.110.dist-info/top_level.txt,sha256=60R1FzpprzU8iiJ1cBMNOA0F083_lYoctFo7pzOpMwY,16
-nopasaran-0.2.110.dist-info/RECORD,,
+nopasaran-0.2.111.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+nopasaran-0.2.111.dist-info/METADATA,sha256=w6Rc4eHoaI8z17mrzBX4R-zdesq02iY5XFK1lJ1kQy4,5497
+nopasaran-0.2.111.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+nopasaran-0.2.111.dist-info/entry_points.txt,sha256=LaOz5GlWuMLjzg4KOEB5OVTattCXVW6a4nSW-WQajCw,55
+nopasaran-0.2.111.dist-info/top_level.txt,sha256=60R1FzpprzU8iiJ1cBMNOA0F083_lYoctFo7pzOpMwY,16
+nopasaran-0.2.111.dist-info/RECORD,,