Sven Luethi
f5791e94d5
modified: __init__.py modified: catalog_zone_manager.py modified: dns_server.py modified: notify_handler.py Adding serial to notify messages to allow client to determine state.
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import socketserver
|
|
import socket
|
|
from .logger import get_logger
|
|
|
|
LOGGER = get_logger(__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
|