#!/usr/bin/python # Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. import json import os import time import logging import cs_ip import cs_guestnetwork import cs_cmdline import cs_vmp import cs_network_acl import cs_vmdata import cs_dhcp import cs_forwardingrules import cs_site2sitevpn from pprint import pprint class DataBag: DPATH = "/etc/cloudstack" def __init__(self): self.bdata = {} def load(self): data = self.bdata if not os.path.exists(self.DPATH): os.makedirs(self.DPATH) self.fpath = self.DPATH + '/' + self.key + '.json' try: handle = open(self.fpath) except IOError: logging.debug("Creating data bag type %s", self.key) data.update({"id": self.key}) else: logging.debug("Loading data bag type %s", self.key) data = json.load(handle) handle.close() self.dbag = data def save(self, dbag): try: handle = open(self.fpath, 'w') except IOError: logging.error("Could not write data bag %s", self.key) else: logging.debug("Writing data bag type %s", self.key) jsono = json.dumps(dbag, indent=4, sort_keys=True) handle.write(jsono) def getDataBag(self): return self.dbag def setKey(self, key): self.key = key class updateDataBag: DPATH = "/etc/cloudstack" def __init__(self, qFile): self.qFile = qFile self.fpath = '' self.bdata = {} self.process() def process(self): self.db = DataBag() if (self.qFile.type == "staticnatrules" or self.qFile.type == "forwardrules"): self.db.setKey("forwardingrules") else: self.db.setKey(self.qFile.type) dbag = self.db.load() logging.info("Command of type %s received", self.qFile.type) if self.qFile.type == 'ips': dbag = self.processIP(self.db.getDataBag()) elif self.qFile.type == 'guestnetwork': dbag = self.processGuestNetwork(self.db.getDataBag()) elif self.qFile.type == 'cmdline': dbag = self.processCL(self.db.getDataBag()) elif self.qFile.type == 'vmpassword': dbag = self.processVMpassword(self.db.getDataBag()) elif self.qFile.type == 'networkacl': dbag = self.process_network_acl(self.db.getDataBag()) elif self.qFile.type == 'vmdata': dbag = self.processVmData(self.db.getDataBag()) elif self.qFile.type == 'dhcpentry': dbag = self.process_dhcp_entry(self.db.getDataBag()) elif self.qFile.type == 'staticnatrules' or self.qFile.type == 'forwardrules': dbag = self.processForwardingRules(self.db.getDataBag()) elif self.qFile.type == 'site2sitevpn': dbag = self.process_site2sitevpn(self.db.getDataBag()) else: logging.error("Error I do not know what to do with file of type %s", self.qFile.type) return self.db.save(dbag) def processGuestNetwork(self, dbag): d = self.qFile.data dp = {} dp['public_ip'] = d['router_guest_ip'] dp['netmask'] = d['router_guest_netmask'] dp['source_nat'] = False dp['add'] = d['add'] dp['one_to_one_nat'] = False dp['gateway'] = d['router_guest_gateway'] dp['nic_dev_id'] = d['device'][3] dp['nw_type'] = 'guest' qf = QueueFile() qf.load({'ip_address': [dp], 'type': 'ips'}) if 'domain_name' not in d.keys() or d['domain_name'] == '': d['domain_name'] = "cloudnine.internal" return cs_guestnetwork.merge(dbag, self.qFile.data) def process_dhcp_entry(self, dbag): return cs_dhcp.merge(dbag, self.qFile.data) def process_site2sitevpn(self, dbag): return cs_site2sitevpn.merge(dbag, self.qFile.data) def process_network_acl(self, dbag): return cs_network_acl.merge(dbag, self.qFile.data) def processVMpassword(self, dbag): return cs_vmp.merge(dbag, self.qFile.data) def processForwardingRules(self, dbag): # to be used by both staticnat and portforwarding return cs_forwardingrules.merge(dbag, self.qFile.data) def processIP(self, dbag): for ip in self.qFile.data["ip_address"]: dbag = cs_ip.merge(dbag, ip) return dbag def processCL(self, dbag): # Convert the ip stuff to an ip object and pass that into cs_ip_merge # "eth0ip": "192.168.56.32", # "eth0mask": "255.255.255.0", self.newData = [] if (self.qFile.data['cmd_line']['type'] == "router"): self.processCLItem('0', "guest") self.processCLItem('1', "control") self.processCLItem('2', "public") elif (self.qFile.data['cmd_line']['type'] == "vpcrouter"): self.processCLItem('0', "control") return cs_cmdline.merge(dbag, self.qFile.data) def processCLItem(self, num, nw_type): key = 'eth' + num + 'ip' dp = {} if(key in self.qFile.data['cmd_line']): dp['public_ip'] = self.qFile.data['cmd_line'][key] dp['netmask'] = self.qFile.data['cmd_line']['eth' + num + 'mask'] dp['source_nat'] = False dp['add'] = True dp['one_to_one_nat'] = False if('localgw' in self.qFile.data['cmd_line']): dp['gateway'] = self.qFile.data['cmd_line']['localgw'] else: dp['gateway'] = 'None' dp['nic_dev_id'] = num dp['nw_type'] = nw_type qf = QueueFile() qf.load({'ip_address': [dp], 'type': 'ips'}) def processVmData(self, dbag): cs_vmdata.merge(dbag, self.qFile.data) return dbag class QueueFile: fileName = '' configCache = "/var/cache/cloud" keep = True data = {} def load(self, data): if data is not None: self.data = data self.type = self.data["type"] proc = updateDataBag(self) return fn = self.configCache + '/' + self.fileName try: handle = open(fn) except IOError: logging.error("Could not open %s", fn) else: self.data = json.load(handle) self.type = self.data["type"] handle.close() if self.keep: self.__moveFile(fn, self.configCache + "/processed") else: os.remove(fn) proc = updateDataBag(self) def setFile(self, name): self.fileName = name def getType(self): return self.type def getData(self): return self.data def setPath(self, path): self.configCache = path def __moveFile(self, origPath, path): if not os.path.exists(path): os.makedirs(path) timestamp = str(int(round(time.time()))) os.rename(origPath, path + "/" + self.fileName + "." + timestamp)