1 files changed, 67 insertions, 0 deletions
diff --git a/rumba/prototypes/ouroboros.py b/rumba/prototypes/ouroboros.py
index a7a7fc9..653f564 100644
--- a/rumba/prototypes/ouroboros.py
+++ b/rumba/prototypes/ouroboros.py
@@ -26,6 +26,7 @@
 
 import time
 import subprocess
+import re
 
 import rumba.ssh_support as ssh
 import rumba.model as mod
@@ -250,3 +251,69 @@ class Experiment(mod.Experiment):
     def destroy_dif(self, dif):
         for ipcp in dif.ipcps:
             ipcp.node.execute_command('irm i d n ' + ipcp.name)
+
+    def parse_stats(self, lines, spaces=0):
+        d = {}
+
+        while len(lines):
+            line = lines[0]
+
+            if not re.match(" {%i}.*" % spaces, line):
+                return d
+
+            lines.pop(0)
+
+            line = line.strip()
+
+            if re.match(".*:.*", line):
+                head, tail = line.split(":", 1)
+
+                if len(tail) == 0:
+                    d[head] = self.parse_stats(lines, spaces+1)
+                else:
+                    d[head] = tail.strip()
+
+        return d
+
+
+    def export_dif_bandwidth(self, filename, dif):
+        f = open(filename, 'w')
+
+        for node in dif.members:
+            ipcp = node.get_ipcp_by_dif(dif)
+
+            # Get IPCP address
+            if not hasattr(ipcp, 'address'):
+                path = '/tmp/ouroboros/'+ ipcp.name + '/dt*'
+                dt_path = node.execute_command('ls -d %s' % path)
+                dts = dt_path.split('.')
+                ipcp.address = int(dts[-1])
+                logger.info('IPCP %s has dt component '
+                            'with address %d' % (ipcp.name, ipcp.address))
+
+        for node in dif.members:
+            ipcp = node.get_ipcp_by_dif(dif)
+
+            dt_path = '/tmp/ouroboros/' + ipcp.name + '/dt.' + \
+                str(ipcp.address) + '/'
+
+            # Get flows to other endpoints
+            fd = node.execute_command('ls --ignore=[01] %s' % dt_path)
+            fds = fd.split('\n')
+            for fd in fds:
+                fd_path = dt_path + fd
+                fd_file = node.execute_command('cat %s' % fd_path)
+
+                d = self.parse_stats(fd_file.splitlines())
+                remote = d["Endpoint address"]
+                ipcp2_name = ''
+                for ipcp2 in dif.ipcps:
+                    if ipcp2.address == int(remote):
+                        ipcp2_name = ipcp2.name
+
+                nr = d["Qos cube   0"]["sent (bytes)"]
+
+                f.write('%s;%s;%s\n' % (ipcp.name, ipcp2_name, nr))
+
+        f.close()
+        logger.info('Wrote stats to %s', filename)