firewall: fix dns errors

circle/firewall/fw.py

 import re
+import logging
+from netaddr import IPAddress, AddrFormatError
 from datetime import datetime, timedelta
 from itertools import product
 
@@ -10,6 +12,7 @@ from django.template import loader, Context
 
 
 settings = django.conf.settings.FIREWALL_SETTINGS
+logger = logging.getLogger(__name__)
 
 
 class BuildFirewall:
@@ -132,15 +135,11 @@ def ipset():
 
 
 def ipv6_to_octal(ipv6):
-    while len(ipv6.split(':')) < 8:
-        ipv6 = ipv6.replace('::', ':::')
+    ipv6 = IPAddress(ipv6, version=6)
     octets = []
-    for part in ipv6.split(':'):
-        if not part:
-            octets.extend([0, 0])
-        else:
+    for part in ipv6.words:
         # Pad hex part to 4 digits.
-            part = '%04x' % int(part, 16)
+        part = '%04x' % part
         octets.append(int(part[:2], 16))
         octets.append(int(part[2:], 16))
     return '\\' + '\\'.join(['%03o' % x for x in octets])
@@ -173,6 +172,7 @@ def generate_ptr_records():
         if host.ipv6:
             DNS.append("^%s:%s:%s" % (host.ipv6.reverse_dns,
                                       reverse, settings['dns_ttl']))
+
     return DNS
 
 
@@ -196,7 +196,12 @@ def generate_records():
         if r.type == 'MX':
             params['address'], params['dist'] = r.address.split(':', 2)
         if r.type == 'AAAA':
+            try:
                 params['octal'] = ipv6_to_octal(r.address)
+            except AddrFormatError:
+                logger.error('Invalid ipv6 address: %s, record: %s',
+                             r.address, r)
+                continue
         if r.type == 'TXT':
             params['octal'] = txt_to_octal(r.address)
         retval.append(types[r.type] % params)
@@ -249,14 +254,14 @@ def dhcp():
                     'net': str(i_vlan.network4.network),
                     'netmask': str(i_vlan.network4.netmask),
                     'domain': i_vlan.domain,
-                    'router': i_vlan.ipv4,
-                    'ntp': i_vlan.ipv4,
+                    'router': i_vlan.network4.ip,
+                    'ntp': i_vlan.network4.ip,
                     'dnsserver': settings['rdns_ip'],
                     'extra': ("range %s" % i_vlan.dhcp_pool
                               if m else "deny unknown-clients"),
                     'interface': i_vlan.name,
                     'name': i_vlan.name,
-                    'tftp': i_vlan.ipv4
+                    'tftp': i_vlan.network4.ip,
                 })
 
                 for i_host in i_vlan.host_set.all():

circle/firewall/tests/test_firewall.py

-from netaddr import IPSet
+from netaddr import IPSet, AddrFormatError
 
 from django.test import TestCase
 from django.contrib.auth.models import User
 from ..admin import HostAdmin
 from firewall.models import Vlan, Domain, Record, Host
+from firewall.fw import dns, ipv6_to_octal
 from django.forms import ValidationError
 from ..iptables import IptRule, IptChain, InvalidRuleExcepion
 
@@ -157,3 +158,39 @@ class IptablesTestCase(TestCase):
         ch.add(*self.r)
         compiled = ch.compile()
         self.assertEqual(len(compiled.splitlines()), len(ch))
+
+
+class DnsTestCase(TestCase):
+    def setUp(self):
+        self.u1 = User.objects.create(username='user1')
+        self.u1.save()
+        d = Domain(name='example.org', owner=self.u1)
+        d.save()
+        self.vlan = Vlan(vid=1, name='test', network4='10.0.0.0/29',
+                         network6='2001:738:2001:4031::/80', domain=d,
+                         owner=self.u1)
+        self.vlan.save()
+        for i in range(1, 6):
+            Host(hostname='h-%d' % i, mac='01:02:03:04:05:%02d' % i,
+                 ipv4='10.0.0.%d' % i, vlan=self.vlan,
+                 owner=self.u1).save()
+        self.r1 = Record(name='tst', type='A', address='127.0.0.1',
+                         domain=d, owner=self.u1)
+        self.rb = Record(name='tst', type='AAAA', address='1.0.0.1',
+                         domain=d, owner=self.u1)
+        self.r2 = Record(name='ts', type='AAAA', address='2001:123:45::6',
+                         domain=d, owner=self.u1)
+        self.r1.save()
+        self.r2.save()
+
+    def test_bad_aaaa_record(self):
+        self.assertRaises(AddrFormatError, ipv6_to_octal, self.rb.address)
+
+    def test_good_aaaa_record(self):
+        ipv6_to_octal(self.r2.address)
+
+    def test_dns_func(self):
+        records = dns()
+        self.assertEqual(Host.objects.count() * 2 +         # soa
+                         len((self.r1, self.r2)) + 1,
+                         len(records))