From 277383d6914a997900ac2b013eb4e443bac5cf53 Mon Sep 17 00:00:00 2001 From: Andy Bavier Date: Fri, 17 Jul 2009 17:29:19 +0000 Subject: [PATCH] Ported code from create_topo_attributes.py --- sfa/rspecs/aggregates/rspec_manager_vini.py | 379 ++++++++++++++++++-- 1 file changed, 357 insertions(+), 22 deletions(-) diff --git a/sfa/rspecs/aggregates/rspec_manager_vini.py b/sfa/rspecs/aggregates/rspec_manager_vini.py index dd68a6e5..3a936b44 100644 --- a/sfa/rspecs/aggregates/rspec_manager_vini.py +++ b/sfa/rspecs/aggregates/rspec_manager_vini.py @@ -4,10 +4,311 @@ from sfa.util.rspec import Rspec from sfa.server.registry import Registries from sfa.plc.nodes import * import sys -import pdb +import socket SFA_VINI_DEFAULT_RSPEC = '/etc/sfa/vini.rspec' +class Node: + def __init__(self, node): + self.id = node['node_id'] + self.hostname = node['hostname'] + self.shortname = self.hostname.replace('.vini-veritas.net', '') + self.site_id = node['site_id'] + self.ipaddr = socket.gethostbyname(self.hostname) + + def get_link_id(self, remote): + if self.id < remote.id: + link = (self.id<<7) + remote.id + else: + link = (remote.id<<7) + self.id + return link + + def get_iface_id(self, remote): + if self.id < remote.id: + iface = 1 + else: + iface = 2 + return iface + + def get_virt_ip(self, remote): + link = self.get_link_id(remote) + iface = self.get_iface_id(remote) + first = link >> 6 + second = ((link & 0x3f)<<2) + iface + return "192.168.%d.%d" % (first, second) + + def get_virt_net(self, remote): + link = self.get_link_id(remote) + first = link >> 6 + second = (link & 0x3f)<<2 + return "192.168.%d.%d/30" % (first, second) + + def get_site(self, sites): + return sites[self.site_id] + + def adjacent_nodes(self, sites, nodes, node_ids): + mysite = self.get_site(sites) + adj_ids = mysite.adj_node_ids.intersection(node_ids) + adj_nodes = [] + for id in adj_ids: + adj_nodes.append(nodes[id]) + return adj_nodes + + def init_rspecs(self): + self.rspecs = [] + + def add_rspec(self, remote): + my_ip = self.get_virt_ip(remote) + remote_ip = remote.get_virt_ip(self) + net = self.get_virt_net(remote) + rspec = remote.id, remote.ipaddr, "1Mbit", my_ip, remote_ip, net + self.rspecs.append(rspec) + + +class Site: + def __init__(self, site): + self.id = site['site_id'] + self.node_ids = site['node_ids'] + self.adj_site_ids = set() + self.adj_node_ids = set() + + def get_sitenodes(self, nodes): + n = [] + for i in self.node_ids: + n.append(nodes[i]) + return n + + def add_adjacency(self, site): + self.adj_site_ids.add(site.id) + for n in site.node_ids: + self.adj_node_ids.add(n) + + +class Slice: + def __init__(self, slice): + self.id = slice['slice_id'] + self.name = slice['name'] + self.node_ids = set(slice['node_ids']) + self.slice_tag_ids = slice['slice_tag_ids'] + + def get_tag(self, tagname, slicetags, node = None): + for i in self.slice_tag_ids: + tag = slicetags[i] + if tag.tagname == tagname: + if (not node) or (node.id == tag.node_id): + return tag + else: + return None + + def get_nodes(self, nodes): + n = [] + for id in self.node_ids: + n.append(nodes[id]) + return n + + + # Add a new slice tag + def add_tag(self, tagname, value, slicetags, node = None): + record = {'slice_tag_id':None, 'slice_id':self.id, 'tagname':tagname, 'value':value} + if node: + record['node_id'] = node.id + else: + record['node_id'] = None + tag = Slicetag(record) + slicetags[tag.id] = tag + self.slice_tag_ids.append(tag.id) + tag.changed = True + tag.updated = True + return tag + + # Update a slice tag if it exists, else add it + def update_tag(self, tagname, value, slicetags, node = None): + tag = self.get_tag(tagname, slicetags, node) + if tag and tag.value == value: + value = "no change" + elif tag: + tag.value = value + tag.changed = True + else: + tag = self.add_tag(tagname, value, slicetags, node) + tag.updated = True + + def assign_egre_key(self, slicetags): + if not self.get_tag('egre_key', slicetags): + try: + key = free_egre_key(slicetags) + self.update_tag('egre_key', key, slicetags) + except: + # Should handle this case... + pass + return + + def turn_on_netns(self, slicetags): + tag = self.get_tag('netns', slicetags) + if (not tag) or (tag.value != '1'): + self.update_tag('netns', '1', slicetags) + return + + def turn_off_netns(self, slicetags): + tag = self.get_tag('netns', slicetags) + if tag and (tag.value != '0'): + tag.delete() + return + + def add_cap_net_admin(self, slicetags): + tag = self.get_tag('capabilities', slicetags) + if tag: + caps = tag.value.split(',') + for cap in caps: + if cap == "CAP_NET_ADMIN": + return + else: + newcaps = "CAP_NET_ADMIN," + tag.value + self.update_tag('capabilities', newcaps, slicetags) + else: + self.add_tag('capabilities', 'CAP_NET_ADMIN', slicetags) + return + + def remove_cap_net_admin(self, slicetags): + tag = self.get_tag('capabilities', slicetags) + if tag: + caps = tag.value.split(',') + newcaps = [] + for cap in caps: + if cap != "CAP_NET_ADMIN": + newcaps.append(cap) + if newcaps: + value = ','.join(newcaps) + self.update_tag('capabilities', value, slicetags) + else: + tag.delete() + return + + # Update the vsys/setup-link and vsys/setup-nat slice tags. + def add_vsys_tags(self, slicetags): + link = nat = False + for i in self.slice_tag_ids: + tag = slicetags[i] + if tag.tagname == 'vsys': + if tag.value == 'setup-link': + link = True + elif tag.value == 'setup-nat': + nat = True + if not link: + self.add_tag('vsys', 'setup-link', slicetags) + if not nat: + self.add_tag('vsys', 'setup-nat', slicetags) + return + + +class Slicetag: + newid = -1 + def __init__(self, tag): + self.id = tag['slice_tag_id'] + if not self.id: + # Make one up for the time being... + self.id = Slicetag.newid + Slicetag.newid -= 1 + self.slice_id = tag['slice_id'] + self.tagname = tag['tagname'] + self.value = tag['value'] + self.node_id = tag['node_id'] + self.updated = False + self.changed = False + self.deleted = False + + # Mark a tag as deleted + def delete(self): + self.deleted = True + self.updated = True + + def write(self, slices, nodes, dryrun): + if not dryrun: + if self.changed: + if int(self.id) > 0: + UpdateSliceTag(self.id, self.value) + else: + AddSliceTag(self.slice_id, self.tagname, self.value, self.node_id) + elif self.deleted and int(self.id) > 0: + DeleteSliceTag(self.id) + else: + slice = slices[self.slice_id].name + if self.node_id: + node = nodes[tag.node_id].hostname + if self.updated: + if self.deleted: + self.value = "deleted" + elif not self.changed: + self.value = "no change" + if int(self.id) < 0: + self.id = "new" + if self.node_id: + print "[%s] %s: %s (%s, %s)" % (self.id, self.tagname, self.value, slice, node) + else: + print "[%s] %s: %s (%s)" % (self.id, self.tagname, self.value, slice) + + +""" +Create a dictionary of site objects keyed by site ID +""" +def get_sites(): + tmp = [] + for site in GetSites(): + t = site['site_id'], Site(site) + tmp.append(t) + return dict(tmp) + + +""" +Create a dictionary of node objects keyed by node ID +""" +def get_nodes(api): + tmp = [] + for node in api.plshell.GetNodes(api.plauth): + t = node['node_id'], Node(node) + tmp.append(t) + return dict(tmp) + +""" +Create a dictionary of slice objects keyed by slice ID +""" +def get_slice(api, slicename): + slice = api.plshell.GetSlices(api.plauth, [slicename]) + if slice: + return Slice(slice[0]) + else: + return None + +""" +Create a dictionary of slicetag objects keyed by slice tag ID +""" +def get_slice_tags(api): + tmp = [] + for tag in api.plshell.GetSliceTags(api.plauth): + t = tag['slice_tag_id'], Slicetag(tag) + tmp.append(t) + return dict(tmp) + +""" +Find a free EGRE key +""" +def free_egre_key(slicetags): + used = set() + for i in slicetags: + tag = slicetags[i] + if tag.tagname == 'egre_key': + used.add(int(tag.value)) + + for i in range(1, 256): + if i not in used: + key = i + break + else: + raise KeyError("No more EGRE keys available") + + return "%s" % key + + """ Copied from create_slice_aggregate() in sfa.plc.slices """ @@ -118,31 +419,58 @@ def create_slice_vini_aggregate(api, hrn, nodes): return 1 def get_rspec(api, hrn): - rspec = None + # Get default rspec + default = Rspec() + default.parseFile(SFA_VINI_DEFAULT_RSPEC) if (hrn): - # XX rspec is expected to be xml, not None. - # call the default sfa.plc.nodes.get_rspec() methods - # until things are complete here + slicename = hrn_to_pl_slicename(hrn) + defaultrspec = default.toGenDict() + nodedict = get_nodedict(defaultrspec) + + # call the default sfa.plc.nodes.get_rspec() method nodes = Nodes(api) rspec = nodes.get_rspec(hrn) - - # Convert HRN to slice name - # Get SliceTags for the slice - # Construct LinkSpecs from the topo_rspec SliceTags - # The first field is the NodeId of the remote node. - # So the endpoints are the SliceTag node and the remote node. + # Grab all the PLC info we'll need at once + slice = get_slice(api, slicename) + if slice: + nodes = get_nodes(api) + tags = get_slice_tags(api) - # How to: - # - avoid duplicates? - # - verify both ends of the link? - pass + # Add the node tags from the Capacity statement to Node objects + for (k, v) in nodedict.iteritems(): + for id in nodes: + if v == nodes[id].hostname: + nodes[id].tag = k + + endpoints = [] + for node in slice.get_nodes(nodes): + linktag = slice.get_tag('topo_rspec', tags, node) + if linktag: + l = eval(linktag.value) + for (id, realip, bw, lvip, rvip, vnet) in l: + endpoints.append((node.id, id, bw)) + + if endpoints: + linkspecs = [] + for (l, r, bw) in endpoints: + if (r, l, bw) in endpoints: + if l < r: + edict = {} + edict['endpoint'] = [nodes[l].tag, nodes[r].tag] + edict['bw'] = [bw] + linkspecs.append(edict) + + d = default.toDict() + d['Rspec']['Request'][0]['NetSpec'][0]['LinkSpec'] = linkspecs + d['Rspec']['Request'][0]['NetSpec'][0]['name'] = hrn + new = Rspec() + new.parseDict(d) + rspec = new.toxml() else: # Return canned response for now... - r = Rspec() - r.parseFile(SFA_VINI_DEFAULT_RSPEC) - rspec = r.toxml() + rspec = default.toxml() return rspec @@ -162,20 +490,27 @@ def create_slice(api, hrn, xml): return True -def rspec_to_nodeset(rspec): +def get_nodedict(rspec): nodedict = {} - nodes = set() - try: + try: sitespecs = rspec['Rspec'][0]['Capacity'][0]['NetSpec'][0]['SiteSpec'] for s in sitespecs: for node in s['NodeSpec']: nodedict[node['name'][0]] = node['hostname'][0] + except KeyError: + pass + + return nodedict + +def rspec_to_nodeset(rspec): + nodes = set() + try: + nodedict = get_nodedict(rspec) linkspecs = rspec['Rspec'][0]['Request'][0]['NetSpec'][0]['LinkSpec'] for l in linkspecs: for e in l['endpoint']: nodes.add(nodedict[e]) - except KeyError: # Bad Rspec pass -- 2.43.0