move EmulabTestbed to its own file

3 files changed, 246 insertions, 245 deletions

diff --git a/emulab_support.py b/emulab_support.py
deleted file mode 100644
index 99a903c..0000000
--- a/emulab_support.py
+++ /dev/null
@@ -1,224 +0,0 @@
-#
-# Emulab support for Rhumba
-#
-#    Sander Vrijders  <sander.vrijders@intec.ugent.be>
-#    Wouter Tavernier <wouter.tavernier@intec.ugent.be>
-#
-# This library is free software; you can redistribute it and/or
-# modify it under the terms of the GNU Lesser General Public
-# License as published by the Free Software Foundation; either
-# version 2.1 of the License, or (at your option) any later version.
-#
-# This library is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-# Lesser General Public License for more details.
-#
-# You should have received a copy of the GNU Lesser General Public
-# License along with this library; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
-# MA  02110-1301  USA
-
-import os
-import time
-import re
-from ast import literal_eval
-from ssh_support import *
-
-import warnings
-warnings.filterwarnings("ignore")
-
-def ops_server(testbed):
-    '''
-    Return server name of the ops-server (is testbed specific)
-
-    @param testbed: testbed info
-
-    @return: server name of the ops-server
-    '''
-    return 'ops.' + testbed.url
-
-def full_name(testbed, node_name):
-    '''
-    Return server name of a node
-
-    @param node_name: name of the node
-    @param testbed: testbed info
-
-    @return: server name of the node
-    '''
-    return node_name + '.' + testbed.exp_name + '.' + \
-        testbed.proj_name + '.' + testbed.url
-
-
-def get_experiment_list(testbed, project_name = None):
-    '''
-    Get list of made emulab experiments accessible with your credentials
-
-    @param testbed: testbed info
-    @param project_name: optional filter on project
-
-    @return: list of created experiments (strings)
-    '''
-    cmd = '/usr/testbed/bin/sslxmlrpc_client.py -m experiment getlist'
-    out = execute_command(testbed, ops_server(testbed), cmd)
-
-    try:
-        if project_name != None:
-            return  literal_eval(out)[project_name][project_name]
-        else:
-            return  literal_eval(out)
-    except:
-        return { project_name: { project_name: [] }}
-
-def swap_exp_in(testbed):
-    '''
-    Swaps experiment in
-
-    @param testbed: testbed info
-    '''
-    cmd = '/usr/testbed/bin/sslxmlrpc_client.py swapexp proj=' + \
-          testbed.proj_name + \
-          ' exp=' + \
-          testbed.exp_name + \
-          ' direction=in'
-
-    output = execute_command(testbed, ops_server(testbed), cmd)
-
-    return output
-
-def create_experiment(testbed, nodes, links):
-    '''
-    Creates an emulab experiment
-
-    @param testbed: testbed info
-    @param nodes: holds the nodes in the experiment
-    @param links: holds the links in the experiment
-    '''
-    proj_name = testbed.proj_name
-    exp_name = testbed.exp_name
-
-    exp_list = get_experiment_list(testbed)
-
-    try:
-        if exp_name in exp_list[proj_name][proj_name]:
-            print("Experiment already exists.")
-            return
-    except:
-        print("First experiment to be created for that project.")
-
-    ns = generate_ns_script(testbed, nodes, links)
-    dest_file_name = '/users/'+ testbed.username + \
-                     '/temp_ns_file.%s.ns' % os.getpid()
-    copy_file_to_testbed(testbed, ops_server(testbed), ns, dest_file_name)
-
-    cmd = '/usr/testbed/bin/sslxmlrpc_client.py startexp ' + \
-          'batch=false wait=true proj="' + proj_name + \
-          '" exp="' + exp_name + '" noswapin=true ' + \
-          'nsfilepath="' + dest_file_name + '"'
-
-    execute_command(testbed, ops_server(testbed), cmd, time_out = None)
-    execute_command(testbed, ops_server(testbed),'rm ' + dest_file_name)
-    print("New experiment succesfully created.")
-
-def generate_ns_script(testbed, nodes, p2plinks):
-    '''
-    Generate ns script based on network graph.
-    Enables to customize default node image.
-
-    @param nodes: holds the nodes in the experiment
-    @param links: holds the links in the experiment
-    @param testbed: testbed info
-
-    @return: ns2 script for Emulab experiment
-    '''
-
-    ns2_script = "# ns script generated by Rhumba\n"
-    ns2_script += "set ns [new Simulator]\n"
-    ns2_script += "source tb_compat.tcl\n"
-
-    for node in nodes:
-            ns2_script += "set " + node.name + " [$ns node]\n"
-            ns2_script += "tb-set-node-os $" + node.name + " " + \
-                          testbed.image + "\n"
-
-    for link in p2plinks:
-            ns2_script += "set " + link.name + \
-                          " [$ns duplex-link $" + \
-                          link.node_a.name + " $" + \
-                          link.node_b.name + " 1000Mb 0ms DropTail]\n"
-
-    ns2_script += "$ns run\n"
-
-    return ns2_script
-
-def wait_until_nodes_up(testbed):
-    '''
-    Checks if nodes are up
-
-    @param testbed: testbed info
-    '''
-    print("Waiting until all nodes are up")
-
-    cmd = '/usr/testbed/bin/script_wrapper.py expinfo -e' + \
-          testbed.proj_name + \
-          ',' + \
-          testbed.exp_name + \
-          ' -a | grep State | cut -f2,2 -d " "'
-
-    res = execute_command(testbed, ops_server(testbed), cmd)
-    active = False
-    if res == "active":
-        active = True
-    while active != True:
-        res = execute_command(testbed, ops_server(testbed), cmd)
-        if res == "active":
-            active = True
-        print("Still waiting")
-        time.sleep(5)
-
-def complete_experiment_graph(testbed, nodes, p2plinks):
-    '''
-    Gets the interface (ethx) to link mapping
-
-    @param testbed: testbed info
-    @param nodes: holds the nodes in the experiment
-    @param links: holds the links in the experiment
-    '''
-
-    node_full_name = full_name(testbed, nodes[0].name)
-    cmd = 'cat /var/emulab/boot/topomap'
-    topomap = execute_command(testbed, node_full_name, cmd)
-    # Almost as ugly as yo momma
-    index = topomap.rfind("# lans")
-    topo_array = topomap[:index].split('\\n')[1:-1]
-    # Array contains things like 'r2b1,link7:10.1.6.3 link6:10.1.5.3'
-    for item in topo_array:
-        item_array = re.split(',? ?', item)
-        node_name = item_array[0]
-        for item2 in item_array[1:]:
-            item2 = item2.split(':')
-            link_name = item2[0]
-            link_ip = item2[1]
-            for link in p2plinks:
-                if link.name == link_name:
-                    if link.node_a.name == node_name:
-                        link.int_a.ip = link_ip
-                    elif link.node_b.name == node_name:
-                        link.int_b.ip = link_ip
-
-    for node in nodes:
-        cmd = 'cat /var/emulab/boot/ifmap'
-        node_full_name  = full_name(testbed, node.name)
-        output = execute_command(testbed, node_full_name, cmd)
-        output = re.split('\\\\n', output)
-        for item in output:
-            item = item.split()
-            for link in p2plinks:
-                if link.node_a.name == node.name and \
-                   link.int_a.ip == item[1]:
-                    link.int_a.name = item[0]
-                elif link.node_b.name == node.name and \
-                     link.int_b.ip == item[1]:
-                    link.int_b.name =  item[0]
-        node.full_name = full_name(testbed, node.name)
diff --git a/rhumba.py b/rhumba.py
index 1ffd092..c338bf8 100755
--- a/rhumba.py
+++ b/rhumba.py
@@ -19,7 +19,6 @@
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 # MA  02110-1301  USA
 
-import emulab_support as es
 import jfed_support as js
 import abc
 import getpass
@@ -43,26 +42,6 @@ class Testbed:
         raise Exception('create_experiment() not implemented')
 
 
-# Represents an emulab testbed info
-#
-# @url [string] URL of the testbed
-# @image [string] specific image to use
-#
-class EmulabTestbed:
-    def __init__(self, exp_name, username, password = "",
-                 proj_name  = "ARCFIRE", url = "wall1.ilabt.iminds.be",
-                 image = "UBUNTU14-64-STD"):
-        Testbed.__init__(self, exp_name, username, password, proj_name)
-        self.url = url
-        self.image = image
-
-    def create_experiment(self, nodes, links):
-        es.create_experiment(self, nodes, links)
-        es.swap_exp_in(self)
-        es.wait_until_nodes_up(self)
-        es.complete_experiment_graph(self, nodes, links)
-
-
 class jFedTestbed:
     def __init__(self, exp_name, username, cert_file, jfed_jar, exp_hours = "2",
                  proj_name = "ARCFIRE", authority = "wall2.ilabt.iminds.be"):
diff --git a/rhumba_emulab.py b/rhumba_emulab.py
new file mode 100644
index 0000000..db1c299
--- /dev/null
+++ b/rhumba_emulab.py
@@ -0,0 +1,246 @@
+#
+# Emulab support for Rhumba
+#
+#    Sander Vrijders  <sander.vrijders@intec.ugent.be>
+#    Wouter Tavernier <wouter.tavernier@intec.ugent.be>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+# MA  02110-1301  USA
+
+import os
+import time
+import re
+from ast import literal_eval
+from ssh_support import *
+import rhumba
+
+import warnings
+warnings.filterwarnings("ignore")
+
+
+
+# Represents an emulab testbed info
+#
+# @url [string] URL of the testbed
+# @image [string] specific image to use
+#
+class EmulabTestbed(rhumba.Testbed):
+    def __init__(self, exp_name, username, password = "",
+                 proj_name  = "ARCFIRE", url = "wall1.ilabt.iminds.be",
+                 image = "UBUNTU14-64-STD"):
+        rhumba.Testbed.__init__(self, exp_name, username, password, proj_name)
+        self.url = url
+        self.image = image
+
+    def ops_server(self):
+        '''
+        Return server name of the ops-server (is testbed specific)
+
+        @param self: testbed info
+
+        @return: server name of the ops-server
+        '''
+        return 'ops.' + self.url
+
+    def full_name(self, node_name):
+        '''
+        Return server name of a node
+
+        @param node_name: name of the node
+        @param self: testbed info
+
+        @return: server name of the node
+        '''
+        return node_name + '.' + self.exp_name + '.' + \
+            self.proj_name + '.' + self.url
+
+    def get_experiment_list(self, project_name = None):
+        '''
+        Get list of made emulab experiments accessible with your credentials
+
+        @param self: testbed info
+        @param project_name: optional filter on project
+
+        @return: list of created experiments (strings)
+        '''
+        cmd = '/usr/testbed/bin/sslxmlrpc_client.py -m experiment getlist'
+        out = execute_command(self, self.ops_server(), cmd)
+
+        try:
+            if project_name != None:
+                return  literal_eval(out)[project_name][project_name]
+            else:
+                return  literal_eval(out)
+        except:
+            return { project_name: { project_name: [] }}
+
+    def swap_exp_in(self):
+        '''
+        Swaps experiment in
+
+        @param self: testbed info
+        '''
+        cmd = '/usr/testbed/bin/sslxmlrpc_client.py swapexp proj=' + \
+              self.proj_name + \
+              ' exp=' + \
+              self.exp_name + \
+              ' direction=in'
+
+        output = execute_command(self, self.ops_server(), cmd)
+
+        return output
+
+    def _create_experiment(self, nodes, links):
+        '''
+        Creates an emulab experiment
+
+        @param self: testbed info
+        @param nodes: holds the nodes in the experiment
+        @param links: holds the links in the experiment
+        '''
+        proj_name = self.proj_name
+        exp_name = self.exp_name
+
+        exp_list = self.get_experiment_list()
+
+        try:
+            if exp_name in exp_list[proj_name][proj_name]:
+                print("Experiment already exists.")
+                return
+        except:
+            print("First experiment to be created for that project.")
+
+        ns = self.generate_ns_script(nodes, links)
+        dest_file_name = '/users/'+ self.username + \
+                         '/temp_ns_file.%s.ns' % os.getpid()
+        copy_file_to_testbed(self, self.ops_server(), ns, dest_file_name)
+
+        cmd = '/usr/testbed/bin/sslxmlrpc_client.py startexp ' + \
+              'batch=false wait=true proj="' + proj_name + \
+              '" exp="' + exp_name + '" noswapin=true ' + \
+              'nsfilepath="' + dest_file_name + '"'
+
+        execute_command(self, self.ops_server(), cmd, time_out = None)
+        execute_command(self, self.ops_server(),'rm ' + dest_file_name)
+        print("New experiment succesfully created.")
+
+    def generate_ns_script(self, nodes, p2plinks):
+        '''
+        Generate ns script based on network graph.
+        Enables to customize default node image.
+
+        @param nodes: holds the nodes in the experiment
+        @param links: holds the links in the experiment
+        @param self: testbed info
+
+        @return: ns2 script for Emulab experiment
+        '''
+
+        ns2_script = "# ns script generated by Rhumba\n"
+        ns2_script += "set ns [new Simulator]\n"
+        ns2_script += "source tb_compat.tcl\n"
+
+        for node in nodes:
+                ns2_script += "set " + node.name + " [$ns node]\n"
+                ns2_script += "tb-set-node-os $" + node.name + " " + \
+                              self.image + "\n"
+
+        for link in p2plinks:
+                ns2_script += "set " + link.name + \
+                              " [$ns duplex-link $" + \
+                              link.node_a.name + " $" + \
+                              link.node_b.name + " 1000Mb 0ms DropTail]\n"
+
+        ns2_script += "$ns run\n"
+
+        return ns2_script
+
+    def wait_until_nodes_up(self):
+        '''
+        Checks if nodes are up
+
+        @param self: testbed info
+        '''
+        print("Waiting until all nodes are up")
+
+        cmd = '/usr/testbed/bin/script_wrapper.py expinfo -e' + \
+              self.proj_name + \
+              ',' + \
+              self.exp_name + \
+              ' -a | grep State | cut -f2,2 -d " "'
+
+        res = execute_command(self, self.ops_server(), cmd)
+        active = False
+        if res == "active":
+            active = True
+        while active != True:
+            res = execute_command(self, self.ops_server(), cmd)
+            if res == "active":
+                active = True
+            print("Still waiting")
+            time.sleep(5)
+
+    def complete_experiment_graph(self, nodes, p2plinks):
+        '''
+        Gets the interface (ethx) to link mapping
+
+        @param self: testbed info
+        @param nodes: holds the nodes in the experiment
+        @param links: holds the links in the experiment
+        '''
+
+        node_full_name = full_name(self, nodes[0].name)
+        cmd = 'cat /var/emulab/boot/topomap'
+        topomap = execute_command(self, node_full_name, cmd)
+        # Almost as ugly as yo momma
+        index = topomap.rfind("# lans")
+        topo_array = topomap[:index].split('\\n')[1:-1]
+        # Array contains things like 'r2b1,link7:10.1.6.3 link6:10.1.5.3'
+        for item in topo_array:
+            item_array = re.split(',? ?', item)
+            node_name = item_array[0]
+            for item2 in item_array[1:]:
+                item2 = item2.split(':')
+                link_name = item2[0]
+                link_ip = item2[1]
+                for link in p2plinks:
+                    if link.name == link_name:
+                        if link.node_a.name == node_name:
+                            link.int_a.ip = link_ip
+                        elif link.node_b.name == node_name:
+                            link.int_b.ip = link_ip
+
+        for node in nodes:
+            cmd = 'cat /var/emulab/boot/ifmap'
+            node_full_name  = full_name(self, node.name)
+            output = execute_command(self, node_full_name, cmd)
+            output = re.split('\\\\n', output)
+            for item in output:
+                item = item.split()
+                for link in p2plinks:
+                    if link.node_a.name == node.name and \
+                       link.int_a.ip == item[1]:
+                        link.int_a.name = item[0]
+                    elif link.node_b.name == node.name and \
+                         link.int_b.ip == item[1]:
+                        link.int_b.name =  item[0]
+            node.full_name = self.full_name(node.name)
+
+    def create_experiment(self, nodes, links):
+        self._create_experiment(nodes, links)
+        self.swap_exp_in()
+        self.wait_until_nodes_up()
+        es.complete_experiment_graph(nodes, links)
+