内网域名纠错系统

import socket
import dns.resolver

class DNSProxy:
    def __init__(self, upstream_dns):
        self.upstream_dns = upstream_dns

    def resolve(self, domain):
        try:
            answers = dns.resolver.resolve(domain, 'A', raise_on_no_answer=False)
            return [r.address for r in answers]
        except dns.resolver.NXDOMAIN:
            return ["127.0.0.1"]  # 返回本地地址作为纠错
        except dns.resolver.NoAnswer:
            return ["127.0.0.1"]
        except dns.resolver.Timeout:
            return ["127.0.0.1"]

def handle_request(data):
    request = data.decode('utf-8')
    domain = request.split()[1]
    proxy = DNSProxy(['8.8.8.8', '8.8.4.4'])
    ip = proxy.resolve(domain)[0]
    response = f"HTTP/1.1 200 OK\r\nContent-Length: 12\r\n\r\nIP: {ip}"
    return response.encode('utf-8')

def start_proxy():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind(('0.0.0.0', 80))
    server.listen(5)
    print("Proxy server started on port 80")
    while True:
        client, addr = server.accept()
        data = client.recv(1024)
        response = handle_request(data)
        client.sendall(response)
        client.close()

if __name__ == "__main__":
    start_proxy()