Sven Luthi
c32161b8a2
modified: netbox_dns_bridge/dns_server.py modified: netbox_dns_bridge/management/commands/dns-export-zone.py modified: netbox_dns_bridge/management/commands/dns-export-zones.py modified: netbox_dns_bridge/management/commands/dns-settings.py modified: netbox_dns_bridge/management/commands/dns-transfer-endpoint.py modified: netbox_dns_bridge/migrations/0001_initial.py modified: netbox_dns_bridge/models.py modified: netbox_dns_bridge/request_handler.py modified: netbox_dns_bridge/signals.py modified: netbox_dns_bridge/utils.py Black code formating corrected
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import logging
|
|
import socketserver
|
|
import socket
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DNSAddressMixin:
|
|
def _resolve_address(self, server_address, socktype, proto):
|
|
host, port = server_address
|
|
|
|
infos = socket.getaddrinfo(
|
|
host, port, socket.AF_UNSPEC, socktype, proto, socket.AI_PASSIVE
|
|
)
|
|
|
|
family, _, _, _, sockaddr = infos[0]
|
|
self.address_family = family
|
|
return sockaddr
|
|
|
|
|
|
class TCPDNSServer(DNSAddressMixin, socketserver.TCPServer):
|
|
allow_reuse_address = True
|
|
|
|
def __init__(self, server_address, handler_class, keyring, tsig_view_map):
|
|
sockaddr = self._resolve_address(
|
|
server_address, socket.SOCK_STREAM, socket.IPPROTO_TCP
|
|
)
|
|
|
|
super().__init__(sockaddr, handler_class)
|
|
|
|
self.keyring = keyring
|
|
self.tsig_view_map = tsig_view_map
|
|
|
|
|
|
class UDPDNSServer(DNSAddressMixin, socketserver.UDPServer):
|
|
allow_reuse_address = True
|
|
|
|
def __init__(self, server_address, handler_class, keyring, tsig_view_map):
|
|
sockaddr = self._resolve_address(
|
|
server_address, socket.SOCK_DGRAM, socket.IPPROTO_UDP
|
|
)
|
|
|
|
super().__init__(sockaddr, handler_class)
|
|
|
|
self.keyring = keyring
|
|
self.tsig_view_map = tsig_view_map
|