2011-04-23 15:19:01 +00:00
|
|
|
import urllib2
|
|
|
|
|
from BeautifulSoup import BeautifulStoneSoup as BS
|
|
|
|
|
import re
|
|
|
|
|
|
2011-04-25 00:52:56 +00:00
|
|
|
from bcommon.models import Key
|
|
|
|
|
import dns.query
|
|
|
|
|
|
2011-04-23 15:19:01 +00:00
|
|
|
def list_server_zones(dns_hostname):
|
2011-04-25 00:52:56 +00:00
|
|
|
""" Take a DNS server, and list the DNS zones it provides resolution for. """
|
2011-04-23 15:19:01 +00:00
|
|
|
# I should take the dns_hostname here, get the object from the DB,
|
|
|
|
|
# and use the status port attribute for the urllib2 query.
|
|
|
|
|
myreq = urllib2.Request("http://%s:853" % dns_hostname)
|
|
|
|
|
try:
|
|
|
|
|
http_request = urllib2.urlopen(myreq)
|
|
|
|
|
except urllib2.URLError, err_reason: # Error retrieving zone list.
|
|
|
|
|
return { 'errors' : err_reason, 'error_context' : "Trying to retrieve zone list from %s" % dns_hostname }
|
|
|
|
|
|
|
|
|
|
return_array = []
|
|
|
|
|
xmloutput = http_request.read()
|
|
|
|
|
mysoup = BS(xmloutput)
|
|
|
|
|
zones = mysoup.findAll('zone')
|
|
|
|
|
for current_zone in zones: # Interate over found zones
|
|
|
|
|
zone_name = current_zone.find('name').contents[0]
|
|
|
|
|
try: # Is this zone of 'IN' type?
|
|
|
|
|
in_zone = re.search(r"(.*)\/IN", zone_name).group(1)
|
|
|
|
|
return_array.append(in_zone)
|
|
|
|
|
except:
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
return return_array
|
2011-04-25 00:52:56 +00:00
|
|
|
|
|
|
|
|
def list_zone_records(dns_hostname, zone_name):
|
|
|
|
|
""" Take a DNS zone, and list all the records it contains. """
|
|
|
|
|
# Need to move most of this logic into a helper method.
|
|
|
|
|
try:
|
|
|
|
|
zone = dns.zone.from_xfr(dns.query.xfr(dns_hostname, zone_name))
|
|
|
|
|
except dns.exception.FormError:
|
|
|
|
|
# There was an error querying the server for the specific zone.
|
|
|
|
|
# Example: a zone that does not exist on the server.
|
|
|
|
|
return { 'errors' : 'Encountered a FormError when querying %s on %s' % (zone_name, dns_hostname) }
|
|
|
|
|
except socket.gaierror, e:
|
|
|
|
|
# TODO: Need to better handle errors here.
|
|
|
|
|
return { 'errors' : "Problems querying DNS server %s: %s" % (dns_hostname, e) }
|
|
|
|
|
|
|
|
|
|
names = zone.nodes.keys()
|
|
|
|
|
names.sort() # Sort the array alphabetically
|
|
|
|
|
record_array = []
|
|
|
|
|
for current_name in names:
|
|
|
|
|
current_record = zone[current_name].to_text(current_name)
|
|
|
|
|
for split_record in current_record.split("\n"): # Split the records on the newline
|
|
|
|
|
record_array.append({'rr_name' : split_record.split(" ")[0],
|
|
|
|
|
'rr_ttl' : split_record.split(" ")[1],
|
|
|
|
|
'rr_class' : split_record.split(" ")[2],
|
|
|
|
|
'rr_type' : split_record.split(" ")[3],
|
|
|
|
|
'rr_data' : split_record.split(" ")[4]})
|
|
|
|
|
return record_array
|
|
|
|
|
|
|
|
|
|
def add_record(clean_data):
|
|
|
|
|
key_name = Key.objects.get(name=(clean_data['tsig_key'])).name
|
|
|
|
|
key_data = Key.objects.get(name=(clean_data['tsig_key'])).data
|
|
|
|
|
key_algorithm = Key.objects.get(name=(clean_data['tsig_key'])).algorithm
|
|
|
|
|
keyring = dns.tsigkeyring.from_text({ key_name : key_data })
|
|
|
|
|
dns_update = dns.update.Update(clean_data['rr_domain'], keyring = keyring, keyalgorithm=key_algorithm)
|
|
|
|
|
dns_update.replace(str(clean_data['rr_name']), 10, str(clean_data['rr_type']), str(clean_data['rr_data']))
|
|
|
|
|
response = dns.query.tcp(dns_update, clean_data['dns_hostname'])
|
|
|
|
|
return response
|
