gitea-mirror/zwift-offline
1
0
Fork 0
mirror of https://github.com/zoffline/zwift-offline.git synced 2025-12-12 15:49:40 -08:00
Code Issues Packages Projects Releases Wiki Activity

Add DNS resolver cache

Browse Source
This commit is contained in:
oldnapalm
2024-07-11 18:40:18 -03:00
parent d0707a1aa2
commit cd8024ea5d
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fake_dns.py
View File

@@ -22,7 +22,11 @@ class DNSQuery:
if namemap.__contains__(name):
ip = namemap[name]
else:
ip = DNSServer.resolver.resolve(name)[0].to_text()
answer = DNSServer.resolver.cache.data.get((name, 1, 1))
if not answer:
answer = DNSServer.resolver.resolve(name)
DNSServer.resolver.cache.put((name, 1, 1), answer)
ip = answer[0].to_text()
packet += self.data[:2] + b'\x81\x80'
packet += self.data[4:6] + self.data[4:6] + b'\x00\x00\x00\x00'
packet += self.data[12:]

16
standalone.py
View File

@@ -46,12 +46,16 @@ if not CDN_PROXY:
from urllib3.util import connection
orig_create_connection = connection.create_connection
def patched_create_connection(address, *args, **kwargs):
try:
host, port = address
answer = resolver.resolve(host)[0].to_text()
address = (answer, port)
except Exception as exc:
print('dns.resolver: %s' % repr(exc))
host, port = address
answer = resolver.cache.data.get((host, 1, 1))
if not answer:
try:
answer = resolver.resolve(host)
resolver.cache.put((host, 1, 1), answer)
except Exception as exc:
print('dns.resolver: %s' % repr(exc))
if answer:
address = (answer[0].to_text(), port)
return orig_create_connection(address, *args, **kwargs)
connection.create_connection = patched_create_connection
CDN_PROXY = True