source: genesis/tools/getrange.py@ 10032

#!/usr/bin/env python
# zoek de eerste vrije (== niet in een wleiden.conf gedefinieerde) interlink
# range
# lodewijk@wirelessleiden.nl
import re
import sys
import glob

def parseaddr(s):
        f = s.split('.')
        return (long(f[0]) << 24L) + \
                (long(f[1]) << 16L) + \
                (long(f[2]) << 8L) + \
                long(f[3])

def showaddr(a):
        return "%d.%d.%d.%d" % ((a >> 24) & 0xff, (a >> 16) & 0xff, (a >> 8) & 0xff, a & 0xff)

def get_ranges(interlink, size, aantal):
        numaddrs = 1 << (32 - size)
        
        substre = re.compile("^(\$[a-zA-Z][^ =]*) ?= ?'([^']*)';")
        
        
        bezet = {}
        
        for fname in glob.glob("*/wleiden.conf"):
                substs = {}
                lineNr = 0
                for l in open(fname):
                        lineNr += 1
                        l = l[:-1]
                        match = substre.match(l)
                        if match != None:
                                substs[match.group(1)] = match.group(2)
                        if l.startswith("IP=") or l.startswith("$master_ip="):
                                if l.startswith("$master_ip="):
                                        right = l.split("'")[1]
                                else:
                                        right = l[3:]
                                if '/' not in right:
                                        addr = right
                                        mask = 32
                                else:
                                        addr, mask = right.split('/')
                                        if addr in substs:
                                                addr = substs[addr]
                                try:
                                        addr = parseaddr(addr)
                                        mask = int(mask)
                                        addr = addr & ~((1 << (32 - mask)) - 1)
                                        for i in range(0, (1 << (32 - mask))):
                                                bezet[addr + i] = 1
                                except ValueError, e:
                                        print "[FOUT] in bestand '%s' regel '%i': %s" % (fname, lineNr, l)

        # Fetching smallest in use IP 
        # XXX: Currently not used for anything
        i = parseaddr("255.255.255.255")
        for k in bezet.keys():
                if k < i:
                        i = k
        
        # Interlinks are living in special ranges usually
        if interlink:
                i = parseaddr("172.16.3.0")
        else:
                i = parseaddr("172.17.0.0")
        
        ranges = []
        for n in xrange(aantal):
                okay = False
                while not okay: 
                        while bezet.has_key(i):
                                i = i + numaddrs
                        
                        # Assume Ok, till we find a already used IP
                        okay = True
                        for j in range(numaddrs):
                                if bezet.has_key(i + j):
                                        i = i + numaddrs
                                        okay = False
                                        break
                ranges.append(i)
                bezet[i] = 1

        return ranges

def main():
        if len(sys.argv) < 2:
                print "Gebruik: %s <interlink|subnet> <grootte> [aantal]" % sys.argv[0]
                print "Voorbeelden:"
                print "\tVoor drie /29 interlinks - '%s interlink 29 3'" % sys.argv[0]
                print "\tVoor een (1) /24 subnet  - '%s subnet 24'" % sys.argv[0]
                exit(1)
        
        # Argument parsing
        interlink = sys.argv[1] == 'interlink'
        size = int(sys.argv[2])
        try:
                aantal = int(sys.argv[3])
        except (KeyError,IndexError):
            aantal = 1
        
        ranges = get_ranges(interlink, size, aantal)
        numaddrs = 1 << (32 - size)

        for i in ranges:
                print "%s/%d:" % (showaddr(i), size),
                if size > 28:
                        print " en ".join([showaddr(i) for i in range(i + 1, i + numaddrs - 1)])
                else:
                        print " en ".join([showaddr(i) for i in range(i + 1, i + 4)]),
                        print "...",
                        print " en ".join([showaddr(i) for i in range(i + numaddrs - 7, i + numaddrs - 4)])

if __name__ == "__main__":
    main()