Changeset dd3e38b for fedd/federation/protogeni_access.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/protogeni_access.py (modified) (5 diffs)

fedd/federation/protogeni_access.py

@@ -4 +4 @@
 import stat # for chmod constants
 import re
+import time
 import string
 import copy
@@ -76 +77 @@
         self.staging_host = config.get("access", "staging_host") \
                 or "ops.emulab.net"
+
+        self.renewal_interval = config.get("access", "renewal") or (3 * 60 * 60)
+        self.renewal_interval = int(self.renewal_interval)
 
         self.ch_url = config.get("access", "ch_url")
@@ -131 +135 @@
         self.start_segment = proxy_protogeni_segment.start_segment
         self.stop_segment = proxy_protogeni_segment.stop_segment
+        self.renew_segment = proxy_protogeni_segment.renew_segment
 
         self.call_SetValue = service_caller('SetValue')
         self.call_GetValue = service_caller('GetValue')
+
+        self.RenewSlices()
 
         self.soap_services = {\
@@ -212 +219 @@
 
             # Access line (t, p, u) -> (a, pw) line
-            # XXX: you are here
             m = access_re.match(line)
             if m != None:
@@ -1114 +1120 @@
         stopper(self, user, staging, slice_cred, cf, cpw)
         return { 'allocID': req['allocID'] }
+
+    def RenewSlices(self):
+        self.log.info("Scanning for slices to renew")
+        self.state_lock.acquire()
+        aids = self.allocation.keys()
+        self.state_lock.release()
+
+        for aid in aids:
+            self.state_lock.acquire()
+            if self.allocation.has_key(aid):
+                name = self.allocation[aid].get('slice_name', None)
+                scred = self.allocation[aid].get('slice_credential', None)
+                cf, user, ssh_key, cpw = self.allocation[aid]['credentials']
+            else:
+                name = None
+                scred = None
+            self.state_lock.release()
+
+            # There's a ProtoGENI slice associated with the segment; renew it.
+            if name and scred:
+                renewer = self.renew_segment(log=self.log, 
+                        debug=self.create_debug, keyfile=ssh_key,
+                        cm_url = self.cm_url, sa_url = self.sa_url,
+                        ch_url = self.ch_url)
+                new_scred = renewer(name, scred, self.renewal_interval, cf, cpw)
+                if new_scred:
+                    self.log.info("Slice %s renewed until %s GMT" % \
+                            (name, time.asctime(time.gmtime(\
+                                time.time()+self.renewal_interval))))
+                    self.state_lock.acquire()
+                    if self.allocation.has_key(aid):
+                        self.allocation[aid]['slice_credential'] = new_scred
+                    self.state_lock.release()
+                else:
+                    self.log.info("Failed to renew slice %s " % name)
+
+        # Let's do this all again soon.  (4 tries before the slices time out)   
+        t = Timer(self.renewal_interval/4, self.RenewSlices)
+        t.start()