Changeset dd3e38b for fedd/federation/proxy_protogeni_segment.py

Timestamp:

Mar 7, 2010 8:01:52 AM (14 years ago)

Author:

Ted Faber <faber@…>

Branches:

axis_example, compt_changes, info-ops, master, version-3.01, version-3.02

Children:

062b991

Parents:

d101c8c

Message:

Add slice/sliver renewal

File:

• fedd/federation/proxy_protogeni_segment.py (modified) (5 diffs)

fedd/federation/proxy_protogeni_segment.py

@@ -91 +91 @@
     # name) has an entry with the allocated machine in hostname and the
     # interfaces in 'interfaces'.  I love having XML parser code lying around.
-    def manifest_to_dict(self, manifest):
-        if self.debug: 
+    def manifest_to_dict(self, manifest, ignore_debug=False):
+        if self.debug and not ignore_debug: 
             self.log.debug("Returning null manifest dict")
             return { }
@@ -388 +388 @@
                     print >>script, "sudo perl -I%s/lib %s/bin/import_key.pl /users/%s/.ssh/authorized_keys /root/.ssh/authorized_keys" % (fed_dir, fed_dir, user)
                     # XXX: debug
+                    # start routing on nodes
+                    print >>script, "sudo perl %s/bin/protogeni_routing.pl" % \
+                            fed_dir
                     if e.get_attribute('startup'):
                         print >>script, "%s \\$USER '%s'" % \
@@ -541 +544 @@
 
         # With manifest in hand, we can export the portal node names.
-        nodes = self.manifest_to_dict(manifest)
-        print nodes
+        nodes = self.manifest_to_dict(manifest, ignore_debug=True)
         self.export_store_info(export_certfile, nodes, parent.ssh_port,
                 connInfo)
@@ -589 +591 @@
         else:
             parent.state_lock.acquire()
+            parent.allocation[aid]['slice_name'] = slicename
             parent.allocation[aid]['slice_credential'] = slice_cred
             parent.allocation[aid]['sliver_credential'] = sliver_cred
@@ -644 +647 @@
         return rv
 
+class renew_segment(segment_base):
+    def __init__(self, log=None, debug=False, keyfile=None,
+            ch_url=None, sa_url=None, cm_url=None):
+        segment_base.__init__(self, log=log, keyfile=keyfile, debug=debug,
+                ch_url=cm_url, sa_url=sa_url, cm_url=cm_url)
+
+    def __call__(self, name, scred, interval, certfile, certpw):
+        ctxt = fedd_ssl_context(my_cert=certfile, password=certpw)
+        try:
+            expiration = time.strftime("%Y%m%dT%H:%M:%S",
+                    time.gmtime(time.time() + interval))
+            cred = self.pg_call(self.sa_url, 'GetCredential', {}, ctxt)
+
+            param = {
+                    'credential': scred,
+                    'expiration': expiration
+                    }
+            r = self.pg_call(self.sa_url, 'RenewSlice', param, ctxt)
+            param = {
+                    'credential': cred,
+                    'hrn': name,
+                    'type': 'Slice',
+                    }
+            slice = self.pg_call(self.sa_url, 'Resolve', param, ctxt)
+            uuid = slice.get('uuid', None)
+            if uuid == None:
+                sys.exit('No uuid for %s' % slicename)
+
+            print 'Calling GetCredential (uuid)'
+            param = {
+                    'credential': cred,
+                    'uuid': uuid,
+                    'type': 'Slice',
+                    }
+            new_scred = self.pg_call(self.sa_url, 'GetCredential', param, ctxt)
+            f = open('./new_slice_cred', 'w')
+            print >>f, new_scred
+            f.close()
+
+        except self.ProtoGENIError, e:
+            self.log.error("Failed to extend slice %s: %s" % (name, e))
+            return None
+        try:
+            print 'Calling RenewSlice (CM)'
+            param = {
+                    'credential': new_scred,
+                    }
+            r = self.pg_call(self.cm_url, 'RenewSlice', param, ctxt)
+        except self.ProtoGENIError, e:
+            self.log.warn("Failed to renew sliver for %s: %s" % (name, e))
+
+        return new_scred
+