adding option to log every dns queries or only the wanted ones

SuperFola · Jul 20, 2021 · c7a14cc · c7a14cc
1 parent 4884a6d
commit c7a14cc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/config.ini.example b/config.ini.example
@@ -3,6 +3,7 @@ interface=
 root=example.com
 domain=dns.example.com
 host_ip=
+log=* # or the DNS request type, eg TXT
 
 [packets]
 ttl=1

diff --git a/src/server.py b/src/server.py
@@ -131,13 +131,17 @@ def _dns_responder(self, pkt: IP):
         packet = Packet(pkt, self.domain)
         answer = None
 
-        self.logger.info(
-            "[DNS %s] Source %s:%i - on %s",
-            dnstypes[packet.question.qtype],
-            packet.src,
-            packet.sport,
-            packet.qname,
-        )
+        request_name = dnstypes[packet.question.qtype]
+
+        if self.config["server"]["log"] == "*" or \
+                self.config["server"]["log"] == request_name:
+            self.logger.info(
+                "[DNS %s] Source %s:%i - on %s",
+                request_name,
+                packet.src,
+                packet.sport,
+                packet.qname,
+            )
 
         # reject every packet which isn't a DNS A/TXT query
         if packet.is_valid_dnsquery(