Fix issues

anchal00 · Aug 21, 2024 · 6f11259 · 6f11259
1 parent 1e81558
commit 6f11259
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/optimus/dns/resolver.py b/optimus/dns/resolver.py
@@ -63,7 +63,7 @@ def resolve(qpacket: DNSPacket) -> DNSPacket:
         else:
             # Pick a random NS record and perform lookup for that
             ns_record: NS = random.choice(response_packet.nameserver_records)
-            packet: DNSPacket = __resolve(
+            packet: DNSPacket = resolve(
                 DNSPacket(
                     dns_header=DNSHeader(
                         id=random.randint(0, int(math.pow(2, 16)) - 1),

diff --git a/optimus/optimus_server/router.py b/optimus/optimus_server/router.py
@@ -1,4 +1,5 @@
 import socket
+from optimus.dns.models.packet import DNSPacket
 from optimus.dns.parser.parse import DNSParser
 from optimus.logging.logger import log
 from optimus.dns.resolver import resolve

diff --git a/optimus/optimus_server/udp.py b/optimus/optimus_server/udp.py
@@ -1,7 +1,7 @@
 import socket
 from concurrent import futures
 
-from optimus.dns.handlers import handle_request
+from optimus.optimus_server.router import handle_request
 from optimus.logging.logger import log