Changeset b4b19c7 for fedd/federation/local_emulab_segment.py

Timestamp:

Apr 21, 2010 5:31:03 AM (14 years ago)

Author:

Ted Faber <faber@…>

Branches:

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

Children:

f7a54c6

Parents:

f54e8e4

Message:

Get topology information into the info operation, as annotations of a topology description. This required adding such information to the start segment replies as well

File:

• fedd/federation/local_emulab_segment.py (modified) (3 diffs)

fedd/federation/local_emulab_segment.py

@@ -90 +90 @@
 $ns run 
 """
+        self.node = { }
 
     def get_state(self, pid, eid):
@@ -138 +139 @@
             self.log.debug("State is %s" % state)
             return state
+
+    def get_mapping(self, pid, eid):
+        # command to test experiment state
+        expinfo_exec = "/usr/testbed/bin/expinfo"  
+        # The expinfo command. 
+        cmd = [ expinfo_exec, '-m', pid, eid]
+
+        dev_null = None
+        try:
+            dev_null = open("/dev/null", "a")
+        except IOError, e:
+            self.log.error("[get_state]: can't open /dev/null: %s" %e)
+
+        if self.debug:
+            rv = 0
+        else:
+            self.log.debug("Getting mapping for %s %s" % (pid, eid))
+            phys_start = re.compile('^Physical\s+Node\s+Mapping')
+            phys_line = re.compile('(\S+)\s+\S+\s+\S+\s+(.*)')
+            phys_end = re.compile('^$')
+            status = subprocess.Popen(cmd, stdout=subprocess.PIPE,
+                    stderr=dev_null, close_fds=True)
+
+            # Parse the info output.  Format:
+            #
+            # stuff
+            # Physical Node Mapping:
+            # ID              Type         OS              Physical    
+            # --------------- ------------ --------------- ------------
+            # virtual         dummy        dummy           physical
+            #
+            foundit = False
+            skip = 0
+            for line in status.stdout:
+                if phys_start.match(line):
+                    skip = 2
+                    foundit = True
+                elif not foundit:
+                    continue
+                elif skip > 0:
+                    skip -= 1
+                elif phys_end.match(line):
+                    break
+                else:
+                    m = phys_line.match(line.strip())
+                    if m: self.node[m.group(1)] = m.group(2)
+                    else: self.log.warn(
+                            "Matching failed while parsing node mapping")
+            rv = status.wait()
+
+        # If the experiment is not present the subcommand returns a
+        # non-zero return value.  If we successfully parsed a "none"
+        # outcome, ignore the return code.
+        if rv != 0 :
+            raise service_error(service_error.internal,
+                    "Cannot get node mapping of segment:%s/%s" % (pid, eid))
+        else:
+            return True
+
 
 
@@ -242 +302 @@
                 state = self.get_state(pid, eid)
                 self.log.debug("[start_segment]: state is %s" % state)
-                return state == 'active'
+                if state != 'active':
+                    return False
         # Everything has gone OK.
+        self.get_mapping(pid,eid)
         return True