mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
a803371fa4
Adds a new source requested in #239. This source allows a user to pull data from a legacy system (Bind9, etc.) that does not have an API/existing provider via AXFR Zone Transfer.
163 lines
4.6 KiB
Python
163 lines
4.6 KiB
Python
#
|
|
#
|
|
#
|
|
|
|
from __future__ import absolute_import, division, print_function, \
|
|
unicode_literals
|
|
|
|
import dns.name
|
|
import dns.query
|
|
import dns.zone
|
|
import dns.rdatatype
|
|
|
|
from dns.exception import DNSException
|
|
|
|
from collections import defaultdict
|
|
import logging
|
|
|
|
from ..record import Record
|
|
from .base import BaseSource
|
|
|
|
|
|
class AxfrBaseSource(BaseSource):
|
|
|
|
SUPPORTS_GEO = False
|
|
SUPPORTS = set(('A', 'AAAA', 'CNAME', 'MX', 'NS', 'PTR', 'SPF',
|
|
'SRV', 'TXT'))
|
|
|
|
def __init__(self, id):
|
|
super(AxfrBaseSource, self).__init__(id)
|
|
|
|
def _data_for_multiple(self, _type, records):
|
|
return {
|
|
'ttl': records[0]['ttl'],
|
|
'type': _type,
|
|
'values': [r['value'] for r in records]
|
|
}
|
|
|
|
_data_for_A = _data_for_multiple
|
|
_data_for_AAAA = _data_for_multiple
|
|
_data_for_NS = _data_for_multiple
|
|
|
|
def _data_for_MX(self, _type, records):
|
|
values = []
|
|
for record in records:
|
|
preference, exchange = record['value'].split(' ', 1)
|
|
values.append({
|
|
'preference': preference,
|
|
'exchange': exchange,
|
|
})
|
|
return {
|
|
'ttl': records[0]['ttl'],
|
|
'type': _type,
|
|
'values': values
|
|
}
|
|
|
|
def _data_for_TXT(self, _type, records):
|
|
values = [value['value'].replace(';', '\\;') for value in records]
|
|
return {
|
|
'ttl': records[0]['ttl'],
|
|
'type': _type,
|
|
'values': values
|
|
}
|
|
|
|
_data_for_SPF = _data_for_TXT
|
|
|
|
def _data_for_single(self, _type, records):
|
|
record = records[0]
|
|
return {
|
|
'ttl': record['ttl'],
|
|
'type': _type,
|
|
'value': record['value']
|
|
}
|
|
|
|
_data_for_CNAME = _data_for_single
|
|
_data_for_PTR = _data_for_single
|
|
|
|
def _data_for_SRV(self, _type, records):
|
|
values = []
|
|
for record in records:
|
|
priority, weight, port, target = record['value'].split(' ', 3)
|
|
values.append({
|
|
'priority': priority,
|
|
'weight': weight,
|
|
'port': port,
|
|
'target': target,
|
|
})
|
|
return {
|
|
'type': _type,
|
|
'ttl': records[0]['ttl'],
|
|
'values': values
|
|
}
|
|
|
|
def populate(self, zone, target=False, lenient=False):
|
|
self.log.debug('populate: name=%s, target=%s, lenient=%s', zone.name,
|
|
target, lenient)
|
|
|
|
values = defaultdict(lambda: defaultdict(list))
|
|
for record in self.zone_records(zone):
|
|
_type = record['type']
|
|
if _type not in self.SUPPORTS:
|
|
continue
|
|
name = zone.hostname_from_fqdn(record['name'])
|
|
values[name][record['type']].append(record)
|
|
|
|
before = len(zone.records)
|
|
for name, types in values.items():
|
|
for _type, records in types.items():
|
|
data_for = getattr(self, '_data_for_{}'.format(_type))
|
|
record = Record.new(zone, name, data_for(_type, records),
|
|
source=self, lenient=lenient)
|
|
zone.add_record(record, lenient=lenient)
|
|
|
|
self.log.info('populate: found %s records',
|
|
len(zone.records) - before)
|
|
|
|
|
|
class AxfrSourceException(Exception):
|
|
pass
|
|
|
|
|
|
class AxfrSourceZoneTransferFailed(AxfrSourceException):
|
|
|
|
def __init__(self):
|
|
super(AxfrSourceZoneTransferFailed, self).__init__(
|
|
'Unable to Perform Zone Transfer')
|
|
|
|
|
|
class AxfrSource(AxfrBaseSource):
|
|
'''
|
|
Axfr zonefile importer to import data
|
|
|
|
axfr:
|
|
class: octodns.source.axfr.AxfrSource
|
|
# The address of nameserver to perform zone transfer against
|
|
master: ns1.example.com
|
|
'''
|
|
def __init__(self, id, master):
|
|
self.log = logging.getLogger('AxfrSource[{}]'.format(id))
|
|
self.log.debug('__init__: id=%s, master=%s', id, master)
|
|
super(AxfrSource, self).__init__(id)
|
|
self.master = master
|
|
|
|
def zone_records(self, zone):
|
|
try:
|
|
z = dns.zone.from_xfr(dns.query.xfr(self.master, zone.name,
|
|
relativize=False),
|
|
relativize=False)
|
|
except DNSException:
|
|
raise AxfrSourceZoneTransferFailed()
|
|
|
|
records = []
|
|
|
|
for (name, ttl, rdata) in z.iterate_rdatas():
|
|
rdtype = dns.rdatatype.to_text(rdata.rdtype)
|
|
records.append({
|
|
"name": name.to_text(),
|
|
"ttl": ttl,
|
|
"type": rdtype,
|
|
"value": rdata.to_text()
|
|
})
|
|
|
|
return records
|