From 2cdc96f4b136792580fb64afd0085557a6563941 Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Thu, 30 May 2024 00:09:34 +1000
Subject: [PATCH] feat: Add tcp proxy v1

---
 .gitea/workflows/build.yml |  2 +-
 proxy.py                   | 49 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 proxy.py

diff --git a/.gitea/workflows/build.yml b/.gitea/workflows/build.yml
index 6be6b03..e7d1236 100644
--- a/.gitea/workflows/build.yml
+++ b/.gitea/workflows/build.yml
@@ -4,7 +4,7 @@ on:
   push:
 
 jobs:
-  Build Docker:
+  Build_Docker:
     runs-on: [ubuntu-latest, amd] # Add amd to require amd64
     steps:
       - name: Checkout
diff --git a/proxy.py b/proxy.py
new file mode 100644
index 0000000..fe2163c
--- /dev/null
+++ b/proxy.py
@@ -0,0 +1,49 @@
+import socket
+from dnslib.server import DNSHandler, BaseResolver, DNSServer, DNSLogger
+from dnslib import DNSRecord
+
+UDP_IP = '127.0.0.1'
+UDP_PORT = 5350
+TCP_IP = '0.0.0.0'
+TCP_PORT = 5351
+
+logger = DNSLogger(log="none")
+
+class TCPHandler(DNSHandler):
+    def handle(self):
+        try:
+            data = self.request.recv(1024).strip()
+            dns_request = DNSRecord.parse(data)
+            with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as udp_socket:
+                udp_socket.sendto(data, (UDP_IP, UDP_PORT))
+                response_data, _ = udp_socket.recvfrom(1024)
+            dns_response = DNSRecord.parse(response_data)
+            self.request.sendall(response_data)
+        except Exception as e:
+            print(e)
+
+class SimpleResolver(BaseResolver):
+    def resolve(self, request, handler):
+        try:
+            with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as udp_socket:
+                udp_socket.sendto(request.pack(), (UDP_IP, UDP_PORT))
+                response_data, _ = udp_socket.recvfrom(1024)
+            dns_response = DNSRecord.parse(response_data)
+            return dns_response
+        except Exception as e:
+            return request.reply()
+
+def start_tcp_server():
+    server = DNSServer(SimpleResolver(), address=TCP_IP, port=TCP_PORT, tcp=True, logger=logger)
+    print(f"TCP server listening on {TCP_IP}:{TCP_PORT}")
+    server.start_thread()
+
+def start_udp_server():
+    server = DNSServer(SimpleResolver(), address=TCP_IP, port=TCP_PORT, logger=logger)
+    print(f"UDP server listening on {TCP_IP}:{TCP_PORT}")
+    server.start()
+
+if __name__ == "__main__":
+    start_tcp_server()
+    start_udp_server()
+    print("Started")