aboutsummaryrefslogtreecommitdiff
path: root/tools
diff options
context:
space:
mode:
Diffstat (limited to 'tools')
-rwxr-xr-xtools/democonf2rumba.py195
1 files changed, 133 insertions, 62 deletions
diff --git a/tools/democonf2rumba.py b/tools/democonf2rumba.py
index 9a30174..549c540 100755
--- a/tools/democonf2rumba.py
+++ b/tools/democonf2rumba.py
@@ -3,23 +3,26 @@
import argparse
import re
+import rumba.model as mod
-def make_experiment(filename, bzImage, initramfs):
+
+def make_experiment(filename, experiment_class, experiment_kwargs,
+ testbed_class, testbed_kwargs):
"""
:type filename str
- :param filename: path to the .conf file
- :param bzImage: path to the bzImage file
- :param initramfs: path to the rootfs.cpio file
- :return:
+ :param filename: path to the .conf file
+ :param experiment_class: subclass of mod.Experiment
+ :param experiment_kwargs: args dict for experiment constructor
+ (nodes will be overwritten)
+ :param testbed_class: subclass of mod.Testbed
+ :param testbed_kwargs: args dict for experiment constructor
+ (nodes will be overwritten)
"""
-
shims = {}
nodes = {}
difs = {}
- ex_name = filename.split('/')[-1].split('.')[0]
-
- print('Reading file %s, under project name %s.' % (filename, ex_name))
+ print('Reading file %s.' % (filename,))
with open(filename, 'r') as conf:
@@ -53,9 +56,9 @@ def make_experiment(filename, bzImage, initramfs):
if speed_unit == 'G':
speed = speed * 1000
- shims['shim' + shim] = {'name': 'shim' + shim,
- 'speed': speed,
- 'type': 'eth'}
+ shims[shim] = {'name': shim,
+ 'speed': speed,
+ 'type': 'eth'}
for vm in vm_list:
nodes.setdefault(vm, {'name': vm, 'difs': [],
@@ -98,60 +101,30 @@ def make_experiment(filename, bzImage, initramfs):
# File parsed
- output = '#!/usr/bin/env/python\n\n'
- output += (
- "from rumba.model import *\n\n"
-
- "# import testbed plugins\n"
- "import rumba.testbeds.emulab as emulab\n"
- "import rumba.testbeds.jfed as jfed\n"
- "import rumba.testbeds.faketestbed as fake\n"
- "import rumba.testbeds.qemu as qemu\n\n"
-
- "# import prototype plugins\n"
- "import rumba.prototypes.ouroboros as our\n"
- "import rumba.prototypes.rlite as rl\n"
- "import rumba.prototypes.irati as irati\n\n"
- )
+ parsed_difs = {}
for shim_name, shim in shims.items():
- output += '%(n)s = ShimEthDIF("%(n)s", link_speed=%(speed)s))\n' \
- % {'n': shim_name, 'speed': shim["speed"]}
-
- output += '\n'
+ parsed_difs[shim_name] = (mod.ShimEthDIF(shim_name,
+ link_speed=shim['speed']))
for dif_name, dif in difs.items():
- output += '%(n)s = NormalDIF("%(n)s")\n' % {'n': dif_name}
+ parsed_difs[dif_name] = (mod.NormalDIF(dif_name))
- output += '\n'
+ parsed_nodes = []
+ for node, node_data in nodes.items():
+ name = node_data['name']
+ difs = [parsed_difs[x] for x in node_data['difs']]
+ dif_registrations = {parsed_difs[x]: [parsed_difs[y] for y in l]
+ for x, l in node_data['dif_registrations']
+ .items()}
+ parsed_nodes.append(mod.Node(name, difs, dif_registrations))
- def str_of_list(i):
- return '[' + ', '.join(i) + ']'
+ testbed = testbed_class(**testbed_kwargs)
- def str_of_dict(d):
- return '{' + ', '.join(['%s: %s' % (k, str_of_list(v))
- for (k, v) in d.items()]) + '}'
+ experiment_kwargs['testbed'] = testbed
+ experiment_kwargs['nodes'] = parsed_nodes
- for node, node_data in nodes.items():
- name = node_data['name']
- output += '%(n)s = Node("%(n)s",\n' % {'n': name}
- output += ' difs=%s,\n' \
- % (str_of_list(node_data['difs']))
- output += ' dif_registrations=%s)\n\n' \
- % (str_of_dict(node_data['dif_registrations']))
-
- output += ('tb = qemu.Testbed(exp_name = "%(ex_name)s",\n'
- ' bzimage = "%(bzImage)s",\n'
- ' initramfs = "%(initramfs)s")\n\n'
- 'exp = rl.Experiment(tb, nodes = [a, b, c, d])\n\n'
- 'print(exp)\n\n'
- 'exp.run()\n'
- % {'ex_name': ex_name,
- 'bzImage': bzImage,
- 'initramfs': initramfs})
-
- with open(ex_name + '.py', 'w') as out_file:
- out_file.write(output)
+ return experiment_class(**experiment_kwargs).run()
if __name__ == '__main__':
@@ -160,17 +133,115 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser(description=description,
epilog=epilog)
- parser.add_argument('config', metavar='CONFIG', type=str,
+
+ prototype = parser.add_mutually_exclusive_group(required=True)
+ prototype.add_argument('--irati', action='store_true', default=False)
+ prototype.add_argument('--ouroboros', action='store_true', default=False)
+ prototype.add_argument('--rlite', action='store_true', default=False)
+
+ parser.add_argument('--config', metavar='CONFIG', type=str,
+ required=True,
help='Path to the config file to parse')
- parser.add_argument('bz_image', metavar='BZIMAGE', type=str,
+
+ subparsers = parser.add_subparsers(dest='testbed')
+ emulab_p = subparsers.add_parser('emulab', help='Use emulab testbed')
+ jfed_p = subparsers.add_parser('jfed', help='Use jfed testbed')
+ qemu_p = subparsers.add_parser('qemu', help='Use qemu testbed')
+ fake_p = subparsers.add_parser('fake', help='Use fake testbed')
+
+ def setup_testbed_common_args(t_p):
+
+ t_p.add_argument('-E', '--exp_name', metavar='EXP_NAME', type=str,
+ required=True,
+ help='Experiment name')
+ t_p.add_argument('-U', '--username', metavar='USERNAME', type=str,
+ required=True,
+ help='Testbed user name')
+ t_p.add_argument('-P', '--proj_name', metavar='PROJECT_NAME', type=str,
+ help='Project name')
+ t_p.add_argument('-W', '--password', metavar='PASSWORD', type=str,
+ help='Testbed password')
+
+ for t in [emulab_p, jfed_p, qemu_p, fake_p]:
+ setup_testbed_common_args(t)
+
+ qemu_p.add_argument('-B', '--bzimage', metavar='BZIMAGE', type=str,
+ required=True,
help='path to the bzImage file to use')
- parser.add_argument('initramfs', metavar='INITRAMFS', type=str,
+ qemu_p.add_argument('-I', '--initramfs', metavar='INITRAMFS', type=str,
+ required=True,
help='path to the initramfs file to use')
+ qemu_p.add_argument('-V', '--use_vhost', action='store_true',
+ default=False, help='Use vhost')
+ qemu_p.add_argument('-Q', '--qemu_logs_dir', metavar='QEMU_LOGS', type=str,
+ default=None, help='path to the folder for qemu logs')
+
+ emulab_p.add_argument('-R', '--url', metavar='URL', type=str,
+ default="wall2.ilabt.iminds.be",
+ help='Url')
+ emulab_p.add_argument('-I', '--image', metavar='IMG', type=str,
+ default="UBUNTU14-64-STD",
+ help='Ubuntu image')
+
+ jfed_p.add_argument('-C', '--cert_file', metavar='CERT', type=str,
+ required=True,
+ help='Certificate file')
+ jfed_p.add_argument('-J', '--jar', metavar='JAR', type=str,
+ required=True,
+ help='Jfed jar')
+ jfed_p.add_argument('-H', '--exp_hours', metavar='HOURS', type=int,
+ default=2, help='Experiment hours')
+ jfed_p.add_argument('-A', '--authority', metavar='AUTH', type=str,
+ default="wall2.ilabt.iminds.be",
+ help='Authority')
args = parser.parse_args()
+ if args.testbed == 'emulab':
+ import rumba.testbeds.emulab as emulab
+ test_class = emulab.Testbed
+ testbed_args = {a.dest: getattr(args, a.dest)
+ for a in emulab_p._actions if a.dest != 'help'}
+ elif args.testbed == 'jfed':
+ import rumba.testbeds.jfed as jfed
+ test_class = jfed.Testbed
+ testbed_args = {a.dest: getattr(args, a.dest)
+ for a in jfed_p._actions if a.dest != 'help'}
+ elif args.testbed == 'qemu':
+ import rumba.testbeds.qemu as qemu
+ test_class = qemu.Testbed
+ testbed_args = {a.dest: getattr(args, a.dest)
+ for a in qemu_p._actions if a.dest != 'help'}
+ elif args.testbed == 'fake':
+ import rumba.testbeds.faketestbed as fake
+ test_class = fake.Testbed
+ testbed_args = {a.dest: getattr(args, a.dest)
+ for a in fake_p._actions if a.dest != 'help'}
+ else:
+ if args.testbed is None:
+ print('Testbed type must be specified!')
+ print(parser.format_help())
+ exit(1)
+ raise ValueError('Unexpected testbed: %s.' % args.testbed)
+
+ if args.irati:
+ import rumba.prototypes.irati as irati
+ exp_class = irati.Experiment
+ elif args.ouroboros:
+ import rumba.prototypes.ouroboros as ouroboros
+ exp_class = ouroboros.Experiment
+ elif args.rlite:
+ import rumba.prototypes.rlite as rlite
+ exp_class = rlite.Experiment
+ else:
+ raise ValueError('Unexpected prototype: %s.' % args.testbed)
+
try:
- make_experiment(args.config, args.bz_image, args.initramfs)
+ make_experiment(args.config,
+ experiment_class=exp_class,
+ experiment_kwargs={},
+ testbed_class=test_class,
+ testbed_kwargs=testbed_args)
except KeyboardInterrupt:
print("Interrupted. Closing down.")