gitea-mirror/zwift-offline
1
0
Fork 0
mirror of https://github.com/zoffline/zwift-offline.git synced 2026-07-28 14:47:12 -07:00
Code Issues Packages Projects Releases Wiki Activity

Support AAAA record

Browse Source
This commit is contained in:
oldnapalm
2025-10-26 19:57:21 -03:00
parent 288a5a31a9
commit ee1aabf557
1 changed files with 24 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fake_dns.py
+24 -18
View File
@@ -1,5 +1,7 @@
import dns.message import dns.message
import dns.rdata import dns.rdata
import dns.rdataclass
import dns.rdatatype
import dns.resolver import dns.resolver
import socketserver import socketserver
import os import os
@@ -8,26 +10,30 @@ class DNSUDPHandler(socketserver.BaseRequestHandler):
def handle(self): def handle(self):
data = self.request[0] data = self.request[0]
socket = self.request[1] socket = self.request[1]
try: query = dns.message.from_wire(data)
query = dns.message.from_wire(data) response = dns.message.make_response(query)
response = dns.message.make_response(query) for question in query.question:
for question in query.question: rdtype = question.rdtype
name = question.name.to_text() if not rdtype in [dns.rdatatype.A, dns.rdatatype.AAAA]:
if name in DNSServer.namemap and question.rdtype == 1: continue
ip = DNSServer.namemap[name] name = question.name.to_text()
else: if name in DNSServer.namemap:
answer = DNSServer.resolver.cache.data.get((name, 1, 1)) ip = DNSServer.namemap[name]
if (rdtype == dns.rdatatype.A and not '.' in ip) or (rdtype == dns.rdatatype.AAAA and not ':' in ip):
continue
else:
try:
answer = DNSServer.resolver.cache.data.get((name, rdtype, dns.rdataclass.IN))
if not answer: if not answer:
answer = DNSServer.resolver.resolve(name) answer = DNSServer.resolver.resolve(name, rdtype)
DNSServer.resolver.cache.put((name, 1, 1), answer) DNSServer.resolver.cache.put((name, rdtype, dns.rdataclass.IN), answer)
ip = answer[0].to_text() ip = answer[0].to_text()
rdata = dns.rdata.from_text(1, 1, ip) except:
rrset = dns.rrset.RRset(question.name, 1, 1) continue
rrset.add(rdata) rdata = dns.rdata.from_text(dns.rdataclass.IN, rdtype, ip)
response.answer.append(rrset) rrset = dns.rrset.from_rdata(name, 3600, rdata)
socket.sendto(response.to_wire(), self.client_address) response.answer.append(rrset)
except Exception as e: socket.sendto(response.to_wire(), self.client_address)
print('fake_dns: %s' % repr(e))
class DNSServer: class DNSServer:
def __init__(self, port=53): def __init__(self, port=53):