def _cipher_filter(cipher, instr): inbuf = StringIO(instr) outbuf = StringIO() while 1: buf = inbuf.read() if not buf: break outbuf.write(cipher.update(buf)) outbuf.write(cipher.final()) rv = outbuf.getvalue() inbuf.close() outbuf.close() return rv
def buildNetgroups(self): """Makes the *-machine files""" header = """################################################################### # This file lists hosts in the '%s' machine netgroup, it is # automatically generated. DO NOT EDIT THIS FILE! # # Number of hosts in '%s' machine netgroup: %i #\n\n""" cursor = connection.cursor() # fetches all the hosts that with valid netgroup entries cursor.execute(""" SELECT h.hostname, n.name, h.netgroup, n.only FROM ((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON p.id = n.ip_id WHERE h.netgroup <> '' AND h.netgroup <> 'none' AND h.status = 'active' ORDER BY h.netgroup, h.hostname """) nameslist = cursor.fetchall() # gets the first host and initializes the hash hostdata = nameslist[0] netgroups = {hostdata[2]: [hostdata[0]]} for row in nameslist: # if new netgroup, create it if row[2] not in netgroups: netgroups.update({row[2]: []}) # if it belongs in the netgroup and has multiple interfaces, put them in if hostdata[0] == row[0] and row[3]: netgroups[row[2]].append(row[1]) hostdata = row # if its a new host, write the old one to the hash elif hostdata[0] != row[0]: netgroups[row[2]].append(row[0]) hostdata = row for netgroup in netgroups: fileoutput = StringIO() fileoutput.write(header % (netgroup, netgroup, len(netgroups[netgroup]))) for each in netgroups[netgroup]: fileoutput.write(each + "\n") self.filedata['%s-machines' % netgroup] = fileoutput.getvalue() fileoutput.close() self.Entries['ConfigFile'][ '/my/adm/hostbase/makenets/machines/%s-machines' % netgroup] = self.FetchFile cursor.execute(""" UPDATE hostbase_host SET dirty=0 """)
def buildNetgroups(self): """Makes the *-machine files""" header = """################################################################### # This file lists hosts in the '%s' machine netgroup, it is # automatically generated. DO NOT EDIT THIS FILE! # # Number of hosts in '%s' machine netgroup: %i #\n\n""" cursor = connection.cursor() # fetches all the hosts that with valid netgroup entries cursor.execute(""" SELECT h.hostname, n.name, h.netgroup, n.only FROM ((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON p.id = n.ip_id WHERE h.netgroup <> '' AND h.netgroup <> 'none' AND h.status = 'active' ORDER BY h.netgroup, h.hostname """) nameslist = cursor.fetchall() # gets the first host and initializes the hash hostdata = nameslist[0] netgroups = {hostdata[2]: [hostdata[0]]} for row in nameslist: # if new netgroup, create it if row[2] not in netgroups: netgroups.update({row[2]: []}) # if it belongs in the netgroup and has multiple interfaces, put them in if hostdata[0] == row[0] and row[3]: netgroups[row[2]].append(row[1]) hostdata = row # if its a new host, write the old one to the hash elif hostdata[0] != row[0]: netgroups[row[2]].append(row[0]) hostdata = row for netgroup in netgroups: fileoutput = StringIO() fileoutput.write(header % (netgroup, netgroup, len(netgroups[netgroup]))) for each in netgroups[netgroup]: fileoutput.write(each + "\n") self.filedata['%s-machines' % netgroup] = fileoutput.getvalue() fileoutput.close() self.Entries['ConfigFile']['/my/adm/hostbase/makenets/machines/%s-machines' % netgroup] = self.FetchFile cursor.execute(""" UPDATE hostbase_host SET dirty=0 """)
def buildZones(self): """Pre-build and stash zone files.""" cursor = connection.cursor() cursor.execute("SELECT id, serial FROM hostbase_zone") zones = cursor.fetchall() for zone in zones: # update the serial number for all zone files todaydate = (strftime('%Y%m%d')) try: if todaydate == str(zone[1])[:8]: serial = zone[1] + 1 else: serial = int(todaydate) * 100 except (KeyError): serial = int(todaydate) * 100 cursor.execute( """UPDATE hostbase_zone SET serial = \'%s\' WHERE id = \'%s\'""" % (str(serial), zone[0])) cursor.execute( "SELECT * FROM hostbase_zone WHERE zone NOT LIKE \'%%.rev\'") zones = cursor.fetchall() iplist = [] hosts = {} for zone in zones: zonefile = StringIO() externalzonefile = StringIO() cursor.execute("""SELECT n.name FROM hostbase_zone_nameservers z INNER JOIN hostbase_nameserver n ON z.nameserver_id = n.id WHERE z.zone_id = \'%s\'""" % zone[0]) nameservers = cursor.fetchall() cursor.execute("""SELECT i.ip_addr FROM hostbase_zone_addresses z INNER JOIN hostbase_zoneaddress i ON z.zoneaddress_id = i.id WHERE z.zone_id = \'%s\'""" % zone[0]) addresses = cursor.fetchall() cursor.execute("""SELECT m.priority, m.mx FROM hostbase_zone_mxs z INNER JOIN hostbase_mx m ON z.mx_id = m.id WHERE z.zone_id = \'%s\'""" % zone[0]) mxs = cursor.fetchall() context = Context({ 'zone': zone, 'nameservers': nameservers, 'addresses': addresses, 'mxs': mxs }) zonefile.write(self.templates['zone'].render(context)) externalzonefile.write(self.templates['zone'].render(context)) querystring = """SELECT h.hostname, p.ip_addr, n.name, c.cname, m.priority, m.mx, n.dns_view FROM (((((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON p.id = n.ip_id) INNER JOIN hostbase_name_mxs x ON n.id = x.name_id) INNER JOIN hostbase_mx m ON m.id = x.mx_id) LEFT JOIN hostbase_cname c ON n.id = c.name_id WHERE n.name LIKE '%%%%%s' AND h.status = 'active' ORDER BY h.hostname, n.name, p.ip_addr """ % zone[1] cursor.execute(querystring) zonehosts = cursor.fetchall() prevhost = (None, None, None, None) cnames = StringIO() cnamesexternal = StringIO() for host in zonehosts: if not host[2].split(".", 1)[1] == zone[1]: zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) cnames = StringIO() cnamesexternal = StringIO() continue if not prevhost[1] == host[1] or not prevhost[2] == host[2]: zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) cnames = StringIO() cnamesexternal = StringIO() zonefile.write("%-32s%-10s%-32s\n" % (host[2].split(".", 1)[0], 'A', host[1])) zonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[6] == 'global': externalzonefile.write( "%-32s%-10s%-32s\n" % (host[2].split(".", 1)[0], 'A', host[1])) externalzonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) elif not prevhost[5] == host[5]: zonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[6] == 'global': externalzonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[3]: try: if host[3].split(".", 1)[1] == zone[1]: cnames.write("%-32s%-10s%-32s\n" % (host[3].split( ".", 1)[0], 'CNAME', host[2].split(".", 1)[0])) if host[6] == 'global': cnamesexternal.write( "%-32s%-10s%-32s\n" % (host[3].split(".", 1)[0], 'CNAME', host[2].split(".", 1)[0])) else: cnames.write("%-32s%-10s%-32s\n" % (host[3] + ".", 'CNAME', host[2].split(".", 1)[0])) if host[6] == 'global': cnamesexternal.write( "%-32s%-10s%-32s\n" % (host[3] + ".", 'CNAME', host[2].split( ".", 1)[0])) except: pass prevhost = host zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) zonefile.write("\n\n%s" % zone[9]) externalzonefile.write("\n\n%s" % zone[9]) self.filedata[zone[1]] = zonefile.getvalue() self.filedata[zone[1] + ".external"] = externalzonefile.getvalue() zonefile.close() externalzonefile.close() self.Entries['ConfigFile']["%s/%s" % (self.filepath, zone[1])] = self.FetchFile self.Entries['ConfigFile']["%s/%s.external" % (self.filepath, zone[1])] = self.FetchFile cursor.execute( "SELECT * FROM hostbase_zone WHERE zone LIKE \'%%.rev\' AND zone <> \'.rev\'" ) reversezones = cursor.fetchall() reversenames = [] for reversezone in reversezones: cursor.execute("""SELECT n.name FROM hostbase_zone_nameservers z INNER JOIN hostbase_nameserver n ON z.nameserver_id = n.id WHERE z.zone_id = \'%s\'""" % reversezone[0]) reverse_nameservers = cursor.fetchall() context = Context({ 'inaddr': reversezone[1].rstrip('.rev'), 'zone': reversezone, 'nameservers': reverse_nameservers, }) self.filedata[ reversezone[1]] = self.templates['reversesoa'].render(context) self.filedata[reversezone[1] + '.external'] = self.templates['reversesoa'].render( context) self.filedata[reversezone[1]] += reversezone[9] self.filedata[reversezone[1] + '.external'] += reversezone[9] subnet = reversezone[1].split(".") subnet.reverse() reversenames.append( (reversezone[1].rstrip('.rev'), ".".join(subnet[1:]))) for filename in reversenames: cursor.execute(""" SELECT DISTINCT h.hostname, p.ip_addr, n.dns_view FROM ((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON n.ip_id = p.id WHERE p.ip_addr LIKE '%s%%%%' AND h.status = 'active' ORDER BY p.ip_addr """ % filename[1]) reversehosts = cursor.fetchall() zonefile = StringIO() externalzonefile = StringIO() if len(filename[0].split(".")) == 2: originlist = [] [ originlist.append( (".".join([ip[1].split(".")[2], filename[0]]), ".".join([filename[1], ip[1].split(".")[2]]))) for ip in reversehosts if (".".join([ip[1].split(".")[2], filename[0]]), ".".join([filename[1], ip[1].split(".")[2]]) ) not in originlist ] for origin in originlist: hosts = [(host[1].split("."), host[0]) for host in reversehosts if host[1].rstrip( '0123456789').rstrip('.') == origin[1]] hosts_external = [ (host[1].split("."), host[0]) for host in reversehosts if (host[1].rstrip('0123456789').rstrip('.') == origin[1] and host[2] == 'global') ] context = Context({ 'hosts': hosts, 'inaddr': origin[0], 'fileorigin': filename[0], }) zonefile.write( self.templates['reverseapp'].render(context)) context = Context({ 'hosts': hosts_external, 'inaddr': origin[0], 'fileorigin': filename[0], }) externalzonefile.write( self.templates['reverseapp'].render(context)) else: originlist = [filename[0]] hosts = [(host[1].split("."), host[0]) for host in reversehosts if (host[1].split("."), host[0]) not in hosts] hosts_external = [ (host[1].split("."), host[0]) for host in reversehosts if ((host[1].split("."), host[0]) not in hosts_external and host[2] == 'global') ] context = Context({ 'hosts': hosts, 'inaddr': filename[0], 'fileorigin': None, }) zonefile.write(self.templates['reverseapp'].render(context)) context = Context({ 'hosts': hosts_external, 'inaddr': filename[0], 'fileorigin': None, }) externalzonefile.write( self.templates['reverseapp'].render(context)) self.filedata['%s.rev' % filename[0]] += zonefile.getvalue() self.filedata['%s.rev.external' % filename[0]] += externalzonefile.getvalue() zonefile.close() externalzonefile.close() self.Entries['ConfigFile']['%s/%s.rev' % (self.filepath, filename[0])] = self.FetchFile self.Entries['ConfigFile']['%s/%s.rev.external' % (self.filepath, filename[0])] = self.FetchFile ## here's where the named.conf file gets written context = Context({ 'zones': zones, 'reverses': reversenames, }) self.filedata['named.conf'] = self.templates['named'].render(context) self.Entries['ConfigFile'][ '/my/adm/hostbase/files/named.conf'] = self.FetchFile self.filedata['named.conf.views'] = self.templates[ 'namedviews'].render(context) self.Entries['ConfigFile'][ '/my/adm/hostbase/files/named.conf.views'] = self.FetchFile
def buildZones(self): """Pre-build and stash zone files.""" cursor = connection.cursor() cursor.execute("SELECT id, serial FROM hostbase_zone") zones = cursor.fetchall() for zone in zones: # update the serial number for all zone files todaydate = (strftime('%Y%m%d')) try: if todaydate == str(zone[1])[:8]: serial = zone[1] + 1 else: serial = int(todaydate) * 100 except (KeyError): serial = int(todaydate) * 100 cursor.execute("""UPDATE hostbase_zone SET serial = \'%s\' WHERE id = \'%s\'""" % (str(serial), zone[0])) cursor.execute("SELECT * FROM hostbase_zone WHERE zone NOT LIKE \'%%.rev\'") zones = cursor.fetchall() iplist = [] hosts = {} for zone in zones: zonefile = StringIO() externalzonefile = StringIO() cursor.execute("""SELECT n.name FROM hostbase_zone_nameservers z INNER JOIN hostbase_nameserver n ON z.nameserver_id = n.id WHERE z.zone_id = \'%s\'""" % zone[0]) nameservers = cursor.fetchall() cursor.execute("""SELECT i.ip_addr FROM hostbase_zone_addresses z INNER JOIN hostbase_zoneaddress i ON z.zoneaddress_id = i.id WHERE z.zone_id = \'%s\'""" % zone[0]) addresses = cursor.fetchall() cursor.execute("""SELECT m.priority, m.mx FROM hostbase_zone_mxs z INNER JOIN hostbase_mx m ON z.mx_id = m.id WHERE z.zone_id = \'%s\'""" % zone[0]) mxs = cursor.fetchall() context = Context({ 'zone': zone, 'nameservers': nameservers, 'addresses': addresses, 'mxs': mxs }) zonefile.write(self.templates['zone'].render(context)) externalzonefile.write(self.templates['zone'].render(context)) querystring = """SELECT h.hostname, p.ip_addr, n.name, c.cname, m.priority, m.mx, n.dns_view FROM (((((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON p.id = n.ip_id) INNER JOIN hostbase_name_mxs x ON n.id = x.name_id) INNER JOIN hostbase_mx m ON m.id = x.mx_id) LEFT JOIN hostbase_cname c ON n.id = c.name_id WHERE n.name LIKE '%%%%%s' AND h.status = 'active' ORDER BY h.hostname, n.name, p.ip_addr """ % zone[1] cursor.execute(querystring) zonehosts = cursor.fetchall() prevhost = (None, None, None, None) cnames = StringIO() cnamesexternal = StringIO() for host in zonehosts: if not host[2].split(".", 1)[1] == zone[1]: zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) cnames = StringIO() cnamesexternal = StringIO() continue if not prevhost[1] == host[1] or not prevhost[2] == host[2]: zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) cnames = StringIO() cnamesexternal = StringIO() zonefile.write("%-32s%-10s%-32s\n" % (host[2].split(".", 1)[0], 'A', host[1])) zonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[6] == 'global': externalzonefile.write("%-32s%-10s%-32s\n" % (host[2].split(".", 1)[0], 'A', host[1])) externalzonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) elif not prevhost[5] == host[5]: zonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[6] == 'global': externalzonefile.write("%-32s%-10s%-3s%s.\n" % ('', 'MX', host[4], host[5])) if host[3]: try: if host[3].split(".", 1)[1] == zone[1]: cnames.write("%-32s%-10s%-32s\n" % (host[3].split(".", 1)[0], 'CNAME', host[2].split(".", 1)[0])) if host[6] == 'global': cnamesexternal.write("%-32s%-10s%-32s\n" % (host[3].split(".", 1)[0], 'CNAME', host[2].split(".", 1)[0])) else: cnames.write("%-32s%-10s%-32s\n" % (host[3] + ".", 'CNAME', host[2].split(".", 1)[0])) if host[6] == 'global': cnamesexternal.write("%-32s%-10s%-32s\n" % (host[3] + ".", 'CNAME', host[2].split(".", 1)[0])) except: pass prevhost = host zonefile.write(cnames.getvalue()) externalzonefile.write(cnamesexternal.getvalue()) zonefile.write("\n\n%s" % zone[9]) externalzonefile.write("\n\n%s" % zone[9]) self.filedata[zone[1]] = zonefile.getvalue() self.filedata[zone[1] + ".external"] = externalzonefile.getvalue() zonefile.close() externalzonefile.close() self.Entries['ConfigFile']["%s/%s" % (self.filepath, zone[1])] = self.FetchFile self.Entries['ConfigFile']["%s/%s.external" % (self.filepath, zone[1])] = self.FetchFile cursor.execute("SELECT * FROM hostbase_zone WHERE zone LIKE \'%%.rev\' AND zone <> \'.rev\'") reversezones = cursor.fetchall() reversenames = [] for reversezone in reversezones: cursor.execute("""SELECT n.name FROM hostbase_zone_nameservers z INNER JOIN hostbase_nameserver n ON z.nameserver_id = n.id WHERE z.zone_id = \'%s\'""" % reversezone[0]) reverse_nameservers = cursor.fetchall() context = Context({ 'inaddr': reversezone[1].rstrip('.rev'), 'zone': reversezone, 'nameservers': reverse_nameservers, }) self.filedata[reversezone[1]] = self.templates['reversesoa'].render(context) self.filedata[reversezone[1] + '.external'] = self.templates['reversesoa'].render(context) self.filedata[reversezone[1]] += reversezone[9] self.filedata[reversezone[1] + '.external'] += reversezone[9] subnet = reversezone[1].split(".") subnet.reverse() reversenames.append((reversezone[1].rstrip('.rev'), ".".join(subnet[1:]))) for filename in reversenames: cursor.execute(""" SELECT DISTINCT h.hostname, p.ip_addr, n.dns_view FROM ((hostbase_host h INNER JOIN hostbase_interface i ON h.id = i.host_id) INNER JOIN hostbase_ip p ON i.id = p.interface_id) INNER JOIN hostbase_name n ON n.ip_id = p.id WHERE p.ip_addr LIKE '%s%%%%' AND h.status = 'active' ORDER BY p.ip_addr """ % filename[1]) reversehosts = cursor.fetchall() zonefile = StringIO() externalzonefile = StringIO() if len(filename[0].split(".")) == 2: originlist = [] [originlist.append((".".join([ip[1].split(".")[2], filename[0]]), ".".join([filename[1], ip[1].split(".")[2]]))) for ip in reversehosts if (".".join([ip[1].split(".")[2], filename[0]]), ".".join([filename[1], ip[1].split(".")[2]])) not in originlist] for origin in originlist: hosts = [(host[1].split("."), host[0]) for host in reversehosts if host[1].rstrip('0123456789').rstrip('.') == origin[1]] hosts_external = [(host[1].split("."), host[0]) for host in reversehosts if (host[1].rstrip('0123456789').rstrip('.') == origin[1] and host[2] == 'global')] context = Context({ 'hosts': hosts, 'inaddr': origin[0], 'fileorigin': filename[0], }) zonefile.write(self.templates['reverseapp'].render(context)) context = Context({ 'hosts': hosts_external, 'inaddr': origin[0], 'fileorigin': filename[0], }) externalzonefile.write(self.templates['reverseapp'].render(context)) else: originlist = [filename[0]] hosts = [(host[1].split("."), host[0]) for host in reversehosts if (host[1].split("."), host[0]) not in hosts] hosts_external = [(host[1].split("."), host[0]) for host in reversehosts if ((host[1].split("."), host[0]) not in hosts_external and host[2] == 'global')] context = Context({ 'hosts': hosts, 'inaddr': filename[0], 'fileorigin': None, }) zonefile.write(self.templates['reverseapp'].render(context)) context = Context({ 'hosts': hosts_external, 'inaddr': filename[0], 'fileorigin': None, }) externalzonefile.write(self.templates['reverseapp'].render(context)) self.filedata['%s.rev' % filename[0]] += zonefile.getvalue() self.filedata['%s.rev.external' % filename[0]] += externalzonefile.getvalue() zonefile.close() externalzonefile.close() self.Entries['ConfigFile']['%s/%s.rev' % (self.filepath, filename[0])] = self.FetchFile self.Entries['ConfigFile']['%s/%s.rev.external' % (self.filepath, filename[0])] = self.FetchFile ## here's where the named.conf file gets written context = Context({ 'zones': zones, 'reverses': reversenames, }) self.filedata['named.conf'] = self.templates['named'].render(context) self.Entries['ConfigFile']['/my/adm/hostbase/files/named.conf'] = self.FetchFile self.filedata['named.conf.views'] = self.templates['namedviews'].render(context) self.Entries['ConfigFile']['/my/adm/hostbase/files/named.conf.views'] = self.FetchFile