xos/tosca/resources/addresspool.py

import os
import pdb
import socket
import sys
import struct
import tempfile
sys.path.append("/opt/tosca")
from translator.toscalib.tosca_template import ToscaTemplate

from core.models import AddressPool

from xosresource import XOSResource

class XOSAddressPool(XOSResource):
    provides = "tosca.nodes.AddressPool"
    xos_model = AddressPool
    copyin_props = ["addresses"]

    def expand_cidr(self, cidr):
        (network, bits) = cidr.split("/")
        network=network.strip()
        bits=int(bits.strip())

        dest = []

        netmask = (~(pow(2,32-bits)-1) & 0xFFFFFFFF)

        count = pow(2, 32-bits)
        for i in range(2, count-1):
            ip = struct.unpack("!L", socket.inet_aton(network))[0]
            ip = ip & netmask | i
            dest.append( socket.inet_ntoa(struct.pack("!L", ip)) )

        return dest

    def get_xos_args(self):
        args = super(XOSAddressPool, self).get_xos_args()

        if "addresses" in args:
            dest = []
            for addr in args["addresses"].split():
                addr=addr.strip()
                if "/" in addr:
                    dest.extend(self.expand_cidr(addr))
                else:
                    dest.append(addr)
            args["addresses"] = " ".join(dest)

        return args