189 lines
6.6 KiB
Python
189 lines
6.6 KiB
Python
import logging
|
|
from typing import List, Optional
|
|
|
|
import dns.query
|
|
import dns.rdatatype
|
|
import dns.resolver
|
|
from django.db import IntegrityError
|
|
from django_rq import job
|
|
from dns import rcode
|
|
from netaddr import ip
|
|
|
|
from netbox_ddns.models import ACTION_CREATE, ACTION_DELETE, DNSStatus, ReverseZone, Zone
|
|
from netbox_ddns.utils import get_soa
|
|
|
|
logger = logging.getLogger('netbox_ddns')
|
|
|
|
|
|
def status_update(output: List[str], operation: str, response) -> None:
|
|
code = response.rcode()
|
|
|
|
if code == dns.rcode.NOERROR:
|
|
message = f"{operation} successful"
|
|
logger.info(message)
|
|
else:
|
|
message = f"{operation} failed: {dns.rcode.to_text(code)}"
|
|
logger.error(message)
|
|
|
|
output.append(message)
|
|
|
|
|
|
def create_forward(dns_name: str, address: ip.IPAddress, status: Optional[DNSStatus], output: List[str]):
|
|
zone = Zone.objects.find_for_dns_name(dns_name)
|
|
if zone:
|
|
logger.debug(f"Found zone {zone.name} for {dns_name}")
|
|
if status:
|
|
status.forward_action = ACTION_CREATE
|
|
|
|
# Check the SOA, we don't want to write to a parent zone if it has delegated authority
|
|
soa = get_soa(dns_name)
|
|
if soa == zone.name:
|
|
record_type = 'A' if address.version == 4 else 'AAAA'
|
|
update = zone.server.create_update(zone.name)
|
|
update.add(
|
|
dns_name,
|
|
zone.ttl,
|
|
record_type,
|
|
str(address)
|
|
)
|
|
response = dns.query.udp(update, zone.server.address)
|
|
status_update(output, f'Adding {dns_name} {record_type} {address}', response)
|
|
if status:
|
|
status.forward_rcode = response.rcode()
|
|
else:
|
|
logger.warning(f"Can't update zone {zone.name} for {dns_name}, "
|
|
f"it has delegated authority for {soa}")
|
|
if status:
|
|
status.forward_rcode = rcode.NOTAUTH
|
|
else:
|
|
logger.debug(f"No zone found for {dns_name}")
|
|
|
|
|
|
def delete_forward(dns_name: str, address: ip.IPAddress, status: Optional[DNSStatus], output: List[str]):
|
|
zone = Zone.objects.find_for_dns_name(dns_name)
|
|
if zone:
|
|
logger.debug(f"Found zone {zone.name} for {dns_name}")
|
|
if status:
|
|
status.forward_action = ACTION_DELETE
|
|
|
|
# Check the SOA, we don't want to write to a parent zone if it has delegated authority
|
|
soa = get_soa(dns_name)
|
|
if soa == zone.name:
|
|
record_type = 'A' if address.version == 4 else 'AAAA'
|
|
update = zone.server.create_update(zone.name)
|
|
update.delete(
|
|
dns_name,
|
|
record_type,
|
|
str(address)
|
|
)
|
|
response = dns.query.udp(update, zone.server.address)
|
|
status_update(output, f'Deleting {dns_name} {record_type} {address}', response)
|
|
if status:
|
|
status.forward_rcode = response.rcode()
|
|
else:
|
|
logger.warning(f"Can't update zone {zone.name} {dns_name}, "
|
|
f"it has delegated authority for {soa}")
|
|
if status:
|
|
status.forward_rcode = rcode.NOTAUTH
|
|
else:
|
|
logger.debug(f"No zone found for {dns_name}")
|
|
|
|
|
|
def create_reverse(dns_name: str, address: ip.IPAddress, status: Optional[DNSStatus], output: List[str]):
|
|
zone = ReverseZone.objects.find_for_address(address)
|
|
if zone and dns_name:
|
|
record_name = zone.record_name(address)
|
|
logger.debug(f"Found zone {zone.name} for {record_name}")
|
|
if status:
|
|
status.reverse_action = ACTION_CREATE
|
|
|
|
# Check the SOA, we don't want to write to a parent zone if it has delegated authority
|
|
soa = get_soa(record_name)
|
|
if soa == zone.name:
|
|
update = zone.server.create_update(zone.name)
|
|
update.add(
|
|
record_name,
|
|
zone.ttl,
|
|
'ptr',
|
|
dns_name
|
|
)
|
|
response = dns.query.udp(update, zone.server.address)
|
|
status_update(output, f'Adding {record_name} PTR {dns_name}', response)
|
|
if status:
|
|
status.reverse_rcode = response.rcode()
|
|
else:
|
|
logger.warning(f"Can't update zone {zone.name} for {record_name}, "
|
|
f"it has delegated authority for {soa}")
|
|
if status:
|
|
status.reverse_rcode = rcode.NOTAUTH
|
|
else:
|
|
logger.debug(f"No zone found for {address}")
|
|
|
|
|
|
def delete_reverse(dns_name: str, address: ip.IPAddress, status: Optional[DNSStatus], output: List[str]):
|
|
zone = ReverseZone.objects.find_for_address(address)
|
|
if zone and dns_name:
|
|
record_name = zone.record_name(address)
|
|
logger.debug(f"Found zone {zone.name} for {record_name}")
|
|
if status:
|
|
status.reverse_action = ACTION_DELETE
|
|
|
|
# Check the SOA, we don't want to write to a parent zone if it has delegated authority
|
|
soa = get_soa(record_name)
|
|
if soa == zone.name:
|
|
update = zone.server.create_update(zone.name)
|
|
update.delete(
|
|
record_name,
|
|
'ptr',
|
|
dns_name
|
|
)
|
|
response = dns.query.udp(update, zone.server.address)
|
|
status_update(output, f'Deleting {record_name} PTR {dns_name}', response)
|
|
if status:
|
|
status.reverse_rcode = response.rcode()
|
|
else:
|
|
logger.warning(f"Can't update zone {zone.name} for {record_name}, "
|
|
f"it has delegated authority for {soa}")
|
|
if status:
|
|
status.reverse_rcode = rcode.NOTAUTH
|
|
else:
|
|
logger.debug(f"No zone found for {address}")
|
|
|
|
|
|
@job
|
|
def dns_create(dns_name: str, address: ip.IPAddress, forward=True, reverse=True, status: DNSStatus = None):
|
|
output = []
|
|
|
|
if forward:
|
|
create_forward(dns_name, address, status, output)
|
|
if reverse:
|
|
create_reverse(dns_name, address, status, output)
|
|
|
|
if status:
|
|
try:
|
|
status.save()
|
|
except IntegrityError:
|
|
# Race condition when creating?
|
|
status.save(force_update=True)
|
|
|
|
return ', '.join(output)
|
|
|
|
|
|
@job
|
|
def dns_delete(dns_name: str, address: ip.IPAddress, forward=True, reverse=True, status: DNSStatus = None):
|
|
output = []
|
|
|
|
if forward:
|
|
delete_forward(dns_name, address, status, output)
|
|
if reverse:
|
|
delete_reverse(dns_name, address, status, output)
|
|
|
|
if status:
|
|
try:
|
|
status.save()
|
|
except IntegrityError:
|
|
# Race condition when creating?
|
|
status.save(force_update=True)
|
|
|
|
return ', '.join(output)
|