sfa/methods/get_ticket.py

   1 ### $Id$
   2 ### $URL$
   3 import time
   4 from sfa.util.faults import *
   5 from sfa.util.method import Method
   6 from sfa.util.parameter import Parameter, Mixed
   7 from sfa.trust.auth import Auth
   8 from sfa.util.config import Config
   9 from sfa.trust.credential import Credential
  10 from sfa.util.table import SfaTable
  11 from sfa.util.sfaticket import SfaTicket
  12 from sfa.plc.slices import Slices
  13 from sfatables.runtime import SFATablesRules
  14 from sfa.util.rspec import *
  15
  16 class get_ticket(Method):
  17     """
  18     Retrieve a ticket. This operation is currently implemented on PLC
  19     only (see SFA, engineering decisions); it is not implemented on
  20     components.
  21
  22     The ticket is filled in with information from the PLC database. This
  23     information includes resources, and attributes such as user keys and
  24     initscripts.
  25
  26     @param cred credential string
  27     @param name name of the slice to retrieve a ticket for
  28     @param rspec resource specification dictionary
  29
  30     @return the string representation of a ticket object
  31     """
  32
  33     interfaces = ['aggregate', 'slicemgr']
  34
  35     accepts = [
  36         Parameter(str, "Credential string"),
  37         Parameter(str, "Human readable name of slice to retrive a ticket for (hrn)"),
  38         Parameter(str, "Resource specification (rspec)"),
  39         Mixed(Parameter(str, "Human readable name of the original caller"),
  40               Parameter(None, "Origin hrn not specified"))
  41         ]
  42
  43     returns = Parameter(str, "String represeneation of a ticket object")
  44
  45     def call(self, cred, hrn, rspec, origin_hrn=None):
  46         user_cred = Credential(string=cred)
  47
  48         #log the call
  49         if not origin_hrn:
  50             origin_hrn = user_cred.get_gid_caller().get_hrn()
  51         self.api.logger.info("interface: %s\tcaller-hrn: %s\ttarget-hrn: %s\tmethod-name: %s"%(self.api.interface, origin_hrn, hrn, self.name))
  52
  53         # validate the cred
  54         self.api.auth.check(cred, "getticket")
  55
  56         # set the right outgoing rules
  57         manager_base = 'sfa.managers'
  58         if self.api.interface in ['aggregate']:
  59             outgoing_rules = SFATablesRules('OUTGOING')
  60             mgr_type = self.api.config.SFA_AGGREGATE_TYPE
  61             manager_module = manager_base + ".aggregate_manager_%s" % mgr_type
  62             manager = __import__(manager_module, fromlist=[manager_base])
  63         elif self.api.interface in ['slicemgr']:
  64             outgoing_rules = SFATablesRules('FORWARD-OUTGOING')
  65             mgr_type = self.api.config.SFA_SM_TYPE
  66             manager_module = manager_base + ".slice_manager_%s" % mgr_type
  67             manager = __import__(manager_module, fromlist=[manager_base])
  68
  69         # Filter the incoming rspec using sfatables
  70         incoming_rules = SFATablesRules('INCOMING')
  71         #incoming_rules.set_slice(hrn) # This is a temporary kludge. Eventually, we'd like to fetch the context requested by the match/target
  72         contexts = incoming_rules.contexts
  73         caller_hrn = Credential(string=cred).get_gid_caller().get_hrn()
  74         request_context = manager.fetch_context(hrn, caller_hrn, contexts)
  75         incoming_rules.set_context(request_context)
  76         rspec = incoming_rules.apply(rspec)
  77         # remove nodes that are not available at this interface from the rspec
  78         valid_rspec = RSpec(xml=manager.get_rspec(self.api, None, origin_hrn))
  79         valid_nodes = valid_rspec.getDictsByTagName('NodeSpec')
  80         valid_hostnames = [node['name'] for node in valid_nodes]
  81         rspec_object = RSpec(xml=rspec)
  82         rspec_object.filter(tagname='NodeSpec', attribute='name', whitelist=valid_hostnames)
  83         rspec = rspec_object.toxml()
  84         ticket = manager.get_ticket(self.api, hrn, rspec, origin_hrn)
  85
  86         return ticket
  87