plugins/ipv6.py

   1 """
   2 IPv6 test! version: 0.3
   3 """
   4
   5 import logger
   6 import os
   7 import socket
   8 import re
   9
  10 import tools
  11 import libvirt
  12 import uuid
  13 from sliver_libvirt import Sliver_Libvirt
  14 from xml.dom.minidom import parseString
  15
  16 priority=4
  17
  18 radvdConfFile = '/etc/radvd.conf'
  19 ipv6addrtag = 'ipv6_address'
  20
  21 def start():
  22     logger.log("ipv6: plugin starting up...")
  23
  24 def buildLibvirtDefaultNetConfig(dom):
  25
  26         # create the <network> element
  27         networkElem = dom.createElement("network")
  28         # create <name> element
  29         nameElem = dom.createElement("name")
  30         textName = dom.createTextNode("default")
  31         nameElem.appendChild(textName)
  32         # create <uuid> element
  33         uuidElem = dom.createElement("uuid")
  34         textUUID = dom.createTextNode(str(uuid.uuid1()))
  35         uuidElem.appendChild(textUUID)
  36         # create <forward> element
  37         forwardElem = dom.createElement("forward")
  38         forwardElem.setAttribute("mode", "nat")
  39         # create <nat> element
  40         natElem = dom.createElement("nat")
  41         # create <port> element
  42         portElem = dom.createElement("port")
  43         portElem.setAttribute("end", "65535")
  44         portElem.setAttribute("start", "1024")
  45         # create the ipv4 <ip> element
  46         ipElem0 = dom.createElement("ip")
  47         ipElem0.setAttribute("address", "192.168.122.1")
  48         ipElem0.setAttribute("netmask", "255.255.255.0")
  49         # create the <dhcp> element
  50         dhcpElem = dom.createElement("dhcp")
  51         # create the <range> element
  52         rangeElem = dom.createElement("range")
  53         rangeElem.setAttribute("end", "192.168.122.254")
  54         rangeElem.setAttribute("start", "192.168.122.2")
  55         # create the <bridge> element
  56         bridgeElem = dom.createElement("bridge")
  57         bridgeElem.setAttribute("delay", "0")
  58         bridgeElem.setAttribute("name", "virbr0")
  59         bridgeElem.setAttribute("stp", "on")
  60
  61         # build the whole thing
  62         natElem.appendChild(portElem)
  63         forwardElem.appendChild(natElem)
  64
  65         dhcpElem.appendChild(rangeElem)
  66         ipElem0.appendChild(dhcpElem)
  67         networkElem.appendChild(nameElem)
  68         networkElem.appendChild(uuidElem)
  69         networkElem.appendChild(forwardElem)
  70         networkElem.appendChild(bridgeElem)
  71         networkElem.appendChild(ipElem0)
  72         return networkElem
  73
  74 def checkForIPv6(defaultNetworkConfig):
  75         netnodes = defaultNetworkConfig.getElementsByTagName('network')
  76         hasIPv6 = False
  77         for netnode in netnodes:
  78                 ips = netnode.getElementsByTagName('ip')
  79                 for ip in ips:
  80                         if ip.getAttribute('family')=='ipv6':
  81                                 logger.log("ipv6: the configuration already have an IPv6 address/prefix set for the slivers! %s/%s" % (ip.getAttribute('address'), ip.getAttribute('prefix')) )
  82                                 hasIPv6 = True
  83         return hasIPv6
  84
  85
  86
  87 def addIPv6(defaultNetworkConfig, ipv6addr, prefix):
  88
  89         netnodes = defaultNetworkConfig.getElementsByTagName('network')
  90         for netnode in netnodes:
  91                 # create the ipv6 <ip> element 1
  92                 ipElem1 = defaultNetworkConfig.createElement("ip")
  93                 ipElem1.setAttribute("family", "ipv6")
  94                 ipElem1.setAttribute("address", ipv6addr)
  95                 ipElem1.setAttribute("prefix", prefix)
  96                 # create the ipv6 <ip> element 2
  97                 # it's ugly, I know, but we need a link-local address on the interface!
  98                 ipElem2 = defaultNetworkConfig.createElement("ip")
  99                 ipElem2.setAttribute("family", "ipv6")
 100                 ipElem2.setAttribute("address", "fe80:1234::1")
 101                 ipElem2.setAttribute("prefix", "64")
 102                 # adding to the 'defaultNetworkConfig'
 103                 netnode.appendChild(ipElem1)
 104                 netnode.appendChild(ipElem2)
 105         return defaultNetworkConfig
 106
 107 def changeIPv6(dom, ipv6addr, prefix):
 108         ips = dom.getElementsByTagName('ip')
 109         for ip in ips:
 110                 if ip.getAttribute("family")=='ipv6' and not(re.match(r'fe80(.*)', ip.getAttribute("address"), re.I)):
 111                         ip.setAttribute("address", ipv6addr)
 112                         ip.setAttribute("prefix", prefix)
 113         return dom
 114
 115
 116 def removeIPv6(dom):
 117         networks = dom.getElementsByTagName('network')
 118         for network in networks:
 119                 ips = network.getElementsByTagName('ip')
 120                 for ip in ips:
 121                         if ip.getAttribute("family")=='ipv6':
 122                                 network.removeChild(ip)
 123         return dom
 124
 125
 126 def checkIfIPv6IsDifferent(dom, ipv6addr, prefix):
 127         netnodes = dom.getElementsByTagName('network')
 128         for netnode in netnodes:
 129                 ips = netnode.getElementsByTagName('ip')
 130                 for ip in ips:
 131                         if ip.getAttribute('family')=='ipv6' and not( re.match(r'fe80(.*)', ip.getAttribute("address"), re.I) ) and (ip.getAttribute('address')!=ipv6addr or ip.getAttribute('prefix')!=prefix):
 132                                 logger.log("ipv6: the IPv6 address or prefix are different. Change detected!")
 133                                 return True
 134
 135         return False
 136
 137
 138 def setAutostart(network):
 139         try:
 140                 network.setAutostart(1)
 141         except:
 142                 logger.log("ipv6: network could not set to autostart")
 143
 144
 145 def setUp(networkLibvirt, connLibvirt, networkElem, ipv6addr, prefix):
 146         newXml = networkElem.toxml()
 147         #logger.log(networkElem.toxml())
 148         #ret = dir(conn)
 149         #for method in ret:
 150         #       logger.log(repr(method))
 151         networkLibvirt.undefine()
 152         networkLibvirt.destroy()
 153         connLibvirt.networkCreateXML(newXml)
 154         networkDefault = connLibvirt.networkDefineXML(newXml)
 155         setAutostart(networkDefault)
 156         commandForwarding = ['sysctl', '-w', 'net.ipv6.conf.all.forwarding=1']
 157         logger.log_call(commandForwarding, timeout=15*60)
 158         configRadvd = """
 159 interface virbr0
 160 {
 161         AdvSendAdvert on;
 162         MinRtrAdvInterval 30;
 163         MaxRtrAdvInterval 100;
 164         prefix %s
 165         {
 166                 AdvOnLink on;
 167                 AdvAutonomous on;
 168                 AdvRouterAddr off;
 169         };
 170
 171 };
 172 """ % (ipv6addr+"/"+prefix)
 173         f=open(radvdConfFile,'w')
 174         f.write(configRadvd)
 175         f.close()
 176         killRadvd()
 177         startRadvd()
 178         logger.log("ipv6: set up process finalized. Enabled IPv6 address to the slivers!")
 179
 180 def cleanUp(networkLibvirt, connLibvirt, networkElem):
 181         dom = removeIPv6(networkElem)
 182         newXml = dom.toxml()
 183         networkLibvirt.undefine()
 184         networkLibvirt.destroy()
 185         # TODO: set autostart for the network
 186         connLibvirt.networkCreateXML(newXml)
 187         networkDefault = connLibvirt.networkDefineXML(newXml)
 188         setAutostart(networkDefault)
 189         killRadvd()
 190         logger.log("ipv6: cleanup process finalized. The IPv6 support on the slivers was removed.")
 191
 192 def killRadvd():
 193         commandKillRadvd = ['killall', 'radvd']
 194         logger.log_call(commandKillRadvd, timeout=15*60)
 195
 196 def startRadvd():
 197         commandRadvd = ['radvd']
 198         logger.log_call(commandRadvd, timeout=15*60)
 199
 200 def getSliverTagId(slivertags):
 201         for slivertag in slivertags:
 202                 if slivertag['tagname']==ipv6addrtag:
 203                         return slivertag['slice_tag_id']
 204
 205
 206 def SetSliverTag(plc, data, tagname):
 207
 208     for sliver in data['slivers']:
 209         # TODO: what about the prefixlen? should we add on it as well?
 210         # here, I'm just taking the ipv6addr (value)
 211         value,prefixlen = tools.get_sliver_ipv6(sliver['name'])
 212
 213         node_id = tools.node_id()
 214         logger.log("ipv6: slice %s" % (slice) )
 215         logger.log("ipv6: nodeid %s" % (node_id) )
 216         slivertags = plc.GetSliceTags({"name":slice['name'],"node_id":node_id,"tagname":tagname})
 217         logger.log(repr(str(slivertags)))
 218         for tag in slivertags:
 219                 logger.log(repr(str(tag)))
 220
 221         ipv6addr = plc.GetSliceIPv6Address(slice['name'])
 222         # if the value to set is null...
 223         if value is None:
 224                 if ipv6addr is not None or len(ipv6addr)==0:
 225                         # then, let's remove the slice tag
 226                         slivertag_id = getSliverTagId(slivertags)
 227                         plc.DeleteSliceTag(slivertag_id)
 228         else:
 229                 # if the ipv6 addr set on the slice does not exist yet, so, let's add it
 230                 if (len(ipv6addr)==0 or ipv6addr is None) and len(value)>0:
 231                         try:
 232                                 slivertag_id=plc.AddSliceTag(slice['name'],tagname,value,node_id)
 233                                 logger.log("ipv6: slice tag added to slice %s" % (slice['name']) )
 234                         except:
 235                                 logger.log_exc ("ipv6: could not set ipv6 addr tag to the slive. slice=%(slice['name'])s tag=%(tagname)s node_id=%(node_id)d" % locals() )
 236                 # if the ipv6 addr set on the slice is different on the value provided, let's update it
 237                 if len(value)>0 and ipv6addr!=value:
 238                         #slivertag_id=slivertags[0]['slice_tag_id']
 239                         slivertag_id = getSliverTagId(slivertags)
 240                         plc.UpdateSliceTag(slivertag_id,value)
 241
 242
 243 def GetSlivers(data, config, plc):
 244
 245     #return
 246     #for sliver in data['slivers']:
 247         #ipv6addr,prefixlen = tools.get_sliver_ipv6(sliver['name'])
 248         #tools.add_ipv6addr_hosts_line(sliver['name'], data['hostname'], ipv6addr)
 249         #result = tools.search_ipv6addr_hosts(sliver['name'], ipv6addr)
 250         #logger.log("tools: result=%s" % (str(result)) )
 251         #tools.remove_all_ipv6addr_hosts(sliver['name'], data['hostname'])
 252     #return
 253
 254
 255     type = 'sliver.LXC'
 256
 257     interfaces = data['interfaces']
 258     logger.log(repr(interfaces))
 259     for interface in interfaces:
 260         logger.log('ipv6: get interface 1: %r'%(interface))
 261         if 'interface_tag_ids' in interface:
 262                 interface_tag_ids = "interface_tag_ids"
 263                 interface_tag_id = "interface_tag_id"
 264                 settings = plc.GetInterfaceTags({interface_tag_id:interface[interface_tag_ids]})
 265                 isSliversIPv6PrefixSet = False
 266                 for setting in settings:
 267                         #logger.log(repr(setting))
 268                         if setting['tagname']=='sliversipv6prefix':
 269                                 ipv6addrprefix = setting['value'].split('/', 1)
 270                                 ipv6addr = ipv6addrprefix[0]
 271                                 prefix = ipv6addrprefix[1]
 272                                 logger.log("ipv6: %s" % (ipv6addr) )
 273                                 validIPv6 = tools.isValidIPv6(ipv6addr)
 274                                 if not(validIPv6):
 275                                         logger.log("ipv6: the 'sliversipv6prefix' tag presented a non-valid IPv6 address!")
 276                                 else:
 277                                         # connecting to the libvirtd
 278                                         connLibvirt = Sliver_Libvirt.getConnection(type)
 279                                         list = connLibvirt.listAllNetworks()
 280                                         for networkLibvirt in list:
 281                                                 xmldesc = networkLibvirt.XMLDesc()
 282                                                 dom = parseString(xmldesc)
 283                                                 hasIPv6 = checkForIPv6(dom)
 284                                                 if hasIPv6:
 285                                                         # let's first check if the IPv6 is different or is it the same...
 286                                                         isDifferent = checkIfIPv6IsDifferent(dom, ipv6addr, prefix)
 287                                                         if isDifferent:
 288                                                                 logger.log("ipv6: the tag 'sliversipv6prefix' was modified! Updating the configuration with the new one...")
 289                                                                 networkElem = changeIPv6(dom, ipv6addr, prefix)
 290                                                                 setUp(networkLibvirt, connLibvirt, networkElem, ipv6addr, prefix)
 291                                                                 logger.log("ipv6: trying to reboot the slivers...")
 292                                                                 tools.reboot_sliver('blah')
 293                                                 else:
 294                                                         logger.log("ipv6: starting to redefine the virtual network...")
 295                                                         #networkElem = buildLibvirtDefaultNetConfig(dom,ipv6addr,prefix)
 296                                                         networkElem = addIPv6(dom, ipv6addr, prefix)
 297                                                         setUp(networkLibvirt, connLibvirt, networkElem, ipv6addr, prefix)
 298                                                         logger.log("ipv6: trying to reboot the slivers...")
 299                                                         tools.reboot_sliver('blah')
 300                                 isSliversIPv6PrefixSet = True
 301                 if not(isSliversIPv6PrefixSet):
 302                         # connecting to the libvirtd
 303                         connLibvirt = Sliver_Libvirt.getConnection(type)
 304                         list = connLibvirt.listAllNetworks()
 305                         for networkLibvirt in list:
 306                                 xmldesc = networkLibvirt.XMLDesc()
 307                                 dom = parseString(xmldesc)
 308                                 if checkForIPv6(dom):
 309                                         cleanUp(networkLibvirt, connLibvirt, dom)
 310                                         logger.log("ipv6: trying to reboot the slivers...")
 311                                         tools.reboot_sliver('blah')
 312
 313     logger.log("ipv6: all done!")