From f79cc000f72fe075d32cb7ca012e551edc8bc84d Mon Sep 17 00:00:00 2001
From: Caleb Fontenot <foley2431@gmail.com>
Date: Tue, 11 Jan 2022 13:56:13 -0600
Subject: [PATCH] Implement most functionallity

---
 .gitmodules |  3 +++
 main.py     | 54 ++++++++++++++++++++++++++++++++++++++++++++---------
 pythonDoh   |  1 +
 3 files changed, 49 insertions(+), 9 deletions(-)
 create mode 100644 .gitmodules
 create mode 160000 pythonDoh

diff --git a/.gitmodules b/.gitmodules
new file mode 100644
index 0000000..91a4c28
--- /dev/null
+++ b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "pythonDoh"]
+	path = pythonDoh
+	url = git@github.com:stamparm/python-doh.git
diff --git a/main.py b/main.py
index a84055d..13dfc56 100755
--- a/main.py
+++ b/main.py
@@ -1,18 +1,54 @@
 #!/bin/python
-from dnslib import server
-import dnslib
-from dns_over_https import SecureDNS
-print("DoH Proxy resolver - <Written by Caleb Fontenot>")
+from dnslib import server,RR,QTYPE
+#import dnslib
+import re
+import dns.query
+import dns.message
+print("DoT (DNS over TLS) Proxy resolver - <Written by Caleb Fontenot>")
+ttl=53
+class bcolors:
+    OK = '\033[92m' #GREEN
+    WARNING = '\033[93m' #YELLOW
+    FAIL = '\033[91m' #RED
+    RESET = '\033[0m' #RESET COLOR
 
-class DoHProxy:
+# REGEX Magic, how does it work
+IPV4SEG  = r'(?:25[0-5]|(?:2[0-4]|1{0,1}[0-9]){0,1}[0-9])'
+IPV4ADDR = r'(?:(?:' + IPV4SEG + r'\.){3,3}' + IPV4SEG + r')'
+class DoTProxy:
     def resolve(self,request,handler):
         reply = request.reply()
-        reply.add_answer(dnslib.dns.RR.fromZone("abc.def. 60 A 1.2.3.4"))
+        print(bcolors.WARNING, request.q.qname, str(QTYPE[request.q.qtype]), bcolors.RESET)
+        IP = DoTquery(str(request.q.qname), QTYPE[request.q.qtype])
+        query_response = str(request.q.qname)+" "+str(ttl)+" "+str(QTYPE[request.q.qtype])+" "+IP
+        print(str(query_response))
+        reply.add_answer(*RR.fromZone(query_response))
         return reply
     
 #Setup DNS server, this will listen for incoming DNS packets
-resolver = DoHProxy()
-logger = server.DNSLogger(prefix=True)
-server = server.DNSServer(resolver,port=53,address="localhost",logger=logger, tcp=False)
+resolver = DoTProxy()
+logger = server.DNSLogger(prefix=False)
+server = server.DNSServer(resolver,port=53,address="0.0.0.0",logger=logger, tcp=False)
+#server.start_thread()
+# Query upstream DoH server
+def DoTquery(domain, query_type):
+    #response = pythonDoh.client.query(str(domain), type=str(query_type), server="calebfontenot.com", verbose=True, fallback=False)
+    if str(domain) == "conntest.nintendowifi.net.":
+        print("bruh moment")
+        return "172.104.88.237"
+    else:
+        q = dns.message.make_query(str(domain), str(query_type))
+        response = dns.query.tls(q, "74.80.18.217", server_hostname="calebfontenot.com", port=853)
+        print(response)
+        # There's probably a better way of doing this, but I can't figure it out...
+        if str(query_type) == "A":    
+            regex = re.compile(IPV4ADDR)
+            IP = regex.findall(str(response.answer))
+            print(bcolors.OK+"IP is", str(IP[0])+bcolors.RESET)
+        elif str(query_type) == "AAAA":
+            print("Not implemented yet!")
+        else:
+            return("Queries beyond A or AAAA are not implemented")
+        return IP[0]
 #server.start_thread()
 server.start()
\ No newline at end of file
diff --git a/pythonDoh b/pythonDoh
new file mode 160000
index 0000000..5b918f6
--- /dev/null
+++ b/pythonDoh
@@ -0,0 +1 @@
+Subproject commit 5b918f6fec0aabe2f964c59b7ef92282562148d0