Changeset b4b19c7 for fedd/federation/proxy_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/proxy_emulab_segment.py (modified) (2 diffs)

fedd/federation/proxy_emulab_segment.py

@@ -25 +25 @@
 $ns run 
 """
+        self.node = { }
 
     def get_state(self, user, host, pid, eid):
@@ -78 +79 @@
 
 
+    def get_mapping(self, pid, eid):
+        # command to test experiment state
+        expinfo_exec = "/usr/testbed/bin/expinfo"  
+        # The expinfo ssh command.  Note the identity restriction to use
+        # only the identity provided in the pubkey given.
+        cmd = [self.ssh_exec, '-o', 'IdentitiesOnly yes', '-o', 
+                'StrictHostKeyChecking no', '-i', 
+                self.ssh_privkey_file, "%s@%s" % (user, host), 
+                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
+
+
     def __call__(self, parent, eid, pid, user, tclfile, tmpdir, timeout=0):
         """