def __init__ (self, config): self.aggregate = VMAggregate() self.shell = VTShell()
class VTSfaDriver:
def __init__ (self, config):
self.aggregate = VMAggregate()
self.shell = VTShell()
def list_resources (self,options):
version_manager = VersionManager()
rspec_version = 'OcfVt'#version_manager.get_version(options.get('geni_rspec_version'))
version_string = "rspec_%s" % (rspec_version)
slice_leaf = None
if options.get("slice"):
slice_leaf = options['slice']
rspec = self.aggregate.get_rspec(version=rspec_version,options=options, slice_leaf=slice_leaf)
return rspec
def crud_slice(self,slice_leaf,authority, action=None):
slicename = slice_leaf
try:
slice = self.shell.GetSlice(slicename,authority)
except Exception as e:
raise RecordNotFound(slice_leaf)
for vm in slice['vms']:
if action == 'start_slice':
self.shell.StartSlice(vm['node-id'],vm['vm-id'])
elif action == 'stop_slice':
self.shell.StopSlice(vm['node-id'],vm['vm-id'])
elif action == 'delete_slice':
self.shell.DeleteSlice(vm['node-id'],vm['vm-id'])
try:
logging.info("LDAP: deleting project in LDAP of SSH gateway")
session = ldapManager()
con = session.bind()
if con:
ldapprj = "%s.%s" % (authority, slicename)
ldapprj = ldapprj.replace("\\" ,"")
#logging.error("ldapprj: "+ ldapprj)
status = session.delProject(con,ldapprj)
if not status:
logging.error("Cannot delete entry in LDAP of the SSH gateway")
else:
logging.error("Cannot contact LDAP of the SSH gateway")
except:
logging.error("Cannot contact LDAP of the SSH gateway")
elif action == 'reset_slice':
self.shell.RebootSlice(vm['node-id'],vm['vm-id'])
return 1
def create_sliver (self,slice_leaf,authority,rspec_string, users, options, expiration_date):
rspec = RSpec(rspec_string,'OcfVt')
requested_attributes = rspec.version.get_slice_attributes()
requested_attributes = self.shell.convert_to_uuid(requested_attributes) #Converts component_id URNs to UUIDs
projectName = authority#users[0]['slice_record']['authority']
sliceName = slice_leaf
self.shell.CreateSliver(requested_attributes,projectName,sliceName,expiration_date)
created_vms = list()
nodes = list()
for slivers in requested_attributes:
node = self.shell.GetNodes(uuid=slivers['component_id'])
#print "\n\n\n\n\n\nslivers: %s\n\n\n\n\n\n" % slivers['slivers']
for vm in slivers['slivers']:
#node = self.shell.GetNodes(uuid=vm['server-id'])
if not node in nodes:
nodes.append(node)
try:
#print "\n\n\n\n\n\nusers: %s\n\n\n\n\n\n" % str(users)
for user in users:
xrn = Xrn(user['urn'], 'user')
user_name = xrn.get_leaf()
# Store user SSH key for future use
for user_key in user['keys']:
if not VirtualMachineKeys.objects.filter(project_uuid=vm["project-id"], slice_uuid=vm["slice-id"], vm_uuid=vm['uuid'],user_name=user_name ,ssh_key=user_key):
key_entry = VirtualMachineKeys(project_uuid=vm["project-id"], slice_uuid=vm["slice-id"], vm_uuid=vm['uuid'],user_name=user_name ,ssh_key=user_key)
key_entry.save()
except Exception as e:
logging.error("create_sliver > Could not store user SSH key. Details: %s" % str(e))
#ip = self.shell.get_ip_from_vm(vm_name=vm['name'],slice_name=vm['slice-name'],project=authority)
#created_vms.append({'vm-name':vm['name'],'vm-ip':ip,'vm-state':'ongoing','slice-name':slice_leaf,'node-name':node.name})
#add ssh keys to ldap of ssh gateway
if len(nodes):
logging.info("create_slice > Connecting to LDAP")
session = ldapManager()
con = session.bind()
logging.info("create_slice > Connected to LDAP. Connection: %s" % str(con))
if con:
logging.info("LDAP: trying to create the following users: %s" % str(users))
for user in users:
logging.info("Sending users to LDAP")
#logging.error("project: "+str(projectName)+" slicename"+str(sliceName))
ldapprj = "%s.%s" % (projectName, sliceName)
ldapprj = ldapprj.replace("\\" ,"")
logging.info("LDAP project: " + ldapprj)
session.add_modify_project_users(con, user["urn"], ldapprj, user["keys"])
logging.info("User added to project: " + str(ldapprj) + ", SSH key: " + str(user['keys']))
#status=session.add_project(con,projectName+"."+sliceName)
#if status != 0:
#logging.error("users"+users)
# for index,user in enumerate(users):
#logging.error(user)
# status=session.add_user(con,"user"+str(index),user['urn'],projectName+"."+sliceName,user['keys'])
# if status ==0:
# logging.error("can't add user to ldap of ssh gateway:"+ str(user))
#else:
# logging.error("can't add project to ldap of ssh gateway: "+projectName+"."+sliceName)
else:
logging.error("Cannot contact LDAP of the SSH gateway")
return self.aggregate.get_rspec(slice_leaf=slice_leaf,projectName=projectName,version=rspec.version,created_vms=created_vms,new_nodes=nodes)
def sliver_status(self,slice_leaf,authority,options):
slice = self.shell.GetSlice(slice_leaf,authority)
result = dict()
List = list()
for vm in slice['vms']:
List.append({'vm-name':vm['vm-name'],'vm-ip':vm['vm-ip'],'vm-state': vm['vm-state'], 'node-name': vm['node-name']})
return List
def get_expiration(self,creds,slice_hrn):
for cred in creds:
credential = Credential(string=cred)
if credential.get_gid_caller().get_hrn() == slice_hrn:
return credential.get_expiration()
return None
