3 # vsys_vnet_admin.py - basic command-line client to show assigned network
4 # segments to the vsys_vnet tag in PlanetLab, and to assign new vsys_vnet tags
5 # using available segments
7 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
11 from optparse import OptionParser
18 """ Returns a XMLRPC proxy to the PLAPI service on the given url """
19 return xmlrpclib.ServerProxy(url, allow_none = True)
21 def get_assigned(plapi, auth):
22 """ Returns a dictionary with the assigned network segments as key
23 and the slice name as value"""
24 info = plapi.GetSliceTags(auth, {'tagname': 'vsys_vnet'}, ('name', 'value'))
25 return dict((d['value'], d['name']) for d in info)
27 def get_next_available(assigned, base, prefix):
28 """ Returns the next available network segment to be assigned to a
30 net = ipaddr.IPNetwork(base)
31 iter_sub = net.iter_subnets(new_prefix=prefix)
36 if not [ s for s in map(ipaddr.IPNetwork, assigned.keys()) \
37 if sub == s or s in sub or sub in s]:
44 def assign_vsys_vent(plapi, auth, vsys_vnet, pl_slice):
45 return plapi.AddSliceTag(auth, pl_slice, 'vsys_vnet', vsys_vnet)
47 if __name__ == "__main__":
48 pl_user = os.environ.get('PL_USER')
49 pl_pwd = os.environ.get('PL_PASS')
50 pl_url = "https://www.planet-lab.eu:443/PLCAPI/"
51 pl_slice = os.environ.get('PL_SLICE')
53 usage = "usage: %prog -u <user> -p <password> -l <url> -S <slice> --assign -b <base_address> -n <prefix>"
54 parser = OptionParser(usage=usage)
55 parser.add_option("-u", "--user", dest="pl_user", help="PlanetLab account user name", default=pl_user)
56 parser.add_option("-p", "--password", dest="pl_pwd", help="PlanetLab account password", default=pl_pwd)
57 parser.add_option("-S", "--slice", dest="pl_slice", help="PlanetLab slice name", default=pl_slice)
58 parser.add_option("-l", "--url", dest="pl_url", help="PlanetLab XMLRPC url", default=pl_url)
59 parser.add_option("-b", "--base", dest="basenet", help="Base network address to perform assignment (defaults to 10.0.0.0/8)", default="10.0.0.0/8")
60 parser.add_option("-n", "--prefix", dest="prefix", help="Network prefix for segments (defaults to 21)", default = "21")
61 parser.add_option("-a", "--assign", action="store_true", dest="assign", default=False,
62 help="If specified, assign next available network segment as a vsys_vnet tag to the user slice")
64 (options, args) = parser.parse_args()
66 if not options.pl_user or not options.pl_pwd:
67 print "Error: PlanetLab user and password are required to proceed."
70 api = plapi(options.pl_url)
72 auth = dict(AuthMethod='password', Username=options.pl_user, AuthString=options.pl_pwd)
73 assigned = get_assigned(api, auth)
76 vsys_vnet = get_next_available(assigned, options.basenet, int(options.prefix))
78 print "Error: There are no available network segments at the moment to assign"
81 print "The next available network segment is %s." % vsys_vnet
82 proceed = raw_input ("Do you wish to proceed with the assignment of the vsys_vnet tag? [y/N] ")
83 if proceed.lower() in ['yes', 'y']:
84 pl_slice = options.pl_slice
86 pl_slice = raw_input ("Please, enter your slice name \n")
88 if pl_slice in assigned.values():
89 print "Error: This slice already has a tag vsys_vnet assigned!!!"
92 slice_tag_id = assign_vsys_vent(api, auth, vsys_vnet, pl_slice)
94 if slice_tag_id and slice_tag_id > 0:
95 print "Success: the new slice tag id is %d" % slice_tag_id
97 print "Error: Something wrong happened!"
100 pp = pprint.PrettyPrinter(indent=4)