1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#!/usr/bin/env python
from rumba.model import *
from rumba.utils import ExperimentManager
# import testbed plugins
import rumba.testbeds.emulab as emulab
import rumba.testbeds.jfed as jfed
import rumba.testbeds.qemu as qemu
# import prototype plugins
import rumba.prototypes.ouroboros as our
import rumba.prototypes.rlite as rl
import rumba.prototypes.irati as irati
import argparse
import sys
description = "Script to create the demo from the Rumba paper"
argparser = argparse.ArgumentParser(description = description)
argparser.add_argument('-m', '--metro', type = int, default = '2',
help = "Number of metro networks")
argparser.add_argument('-n', '--metro-nodes', type = int, default = '5',
help = "Number of nodes per metro")
argparser.add_argument('-e', '--end-nodes', type = int, default = '2',
help = "Number of nodes connected to each metro")
args = argparser.parse_args()
log.set_logging_level('DEBUG')
pi = NormalDIF("pi")
pi.dt_strategy = 'minimal'
c = NormalDIF("core")
nodes = []
core_nodes = []
for i in range(0, args.metro):
m = NormalDIF("m-" + str(i))
e_prev = None
m1 = None
mn = None
for j in range(0, args.metro_nodes):
if j is not (args.metro_nodes - 1):
e = ShimEthDIF("e-" + str(i) + "-" + str(j))
else:
e = None
if e_prev == None and e != None:
node = Node("metro-" + str(i) + "-" + str(j),
difs = [m, pi, e],
dif_registrations = {pi : [m], m : [e]})
m1 = node
elif e_prev != None and e != None:
node = Node("metro-" + str(i) + "-" + str(j),
difs = [m, pi, e, e_prev],
dif_registrations = {pi : [m], m : [e, e_prev]})
else:
node = Node("metro-" + str(i) + "-" + str(j),
difs = [m, pi, e_prev],
dif_registrations = {pi : [m], m : [e_prev]})
mn = node
nodes.append(node)
e_prev = e
# Create nodes attached to metro
for k in range(0, args.end_nodes):
ee = ShimEthDIF("e-" + str(i) + "-" + str(j) + "-" + str(k))
end_node = Node("end-" + str(i) + "-" + str(j) + "-" + str(k),
difs = [m, pi, ee],
dif_registrations = {pi : [m], m : [ee]})
node.add_dif(ee)
node.add_dif_registration(m, ee)
nodes.append(end_node)
e = ShimEthDIF("e-" + str(i) + "-" + str(args.metro_nodes))
m1.add_dif(e)
m1.add_dif_registration(m, e)
mn.add_dif(e)
mn.add_dif_registration(m, e)
# Add 2 links from metro to core
e0 = ShimEthDIF("c-e-" + str(i) + "-0")
e1 = ShimEthDIF("c-e-" + str(i) + "-1")
e2 = ShimEthDIF("c-e-" + str(i) + "-2")
c1 = Node("c-" + str(i) + "-0",
difs = [c, m, pi, e0, e2],
dif_registrations = {pi : [m, c], m : [e0], c : [e2]})
c2 = Node("c-" + str(i) + "-1",
difs = [c, m, pi, e1, e2],
dif_registrations = {pi : [m, c], m : [e1], c : [e2]})
nodes.append(c1)
nodes.append(c2)
m1.add_dif(e0)
m1.add_dif_registration(m, e0)
mn.add_dif(e1)
mn.add_dif_registration(m, e1)
for x in range(0, len(core_nodes), 2):
ce0 = ShimEthDIF("c-c-e-" + str(i) + "-" + str(x))
ce1 = ShimEthDIF("c-c-e-" + str(i) + "-" + str(x + 1))
core_nodes[x].add_dif(ce0)
core_nodes[x].add_dif_registration(c, ce0)
core_nodes[x + 1].add_dif(ce1)
core_nodes[x + 1].add_dif_registration(c, ce1)
c1.add_dif(ce0)
c1.add_dif_registration(c, ce0)
c2.add_dif(ce1)
c2.add_dif_registration(c, ce1)
core_nodes.append(c1)
core_nodes.append(c2)
tb = jfed.Testbed(exp_name = "metro2",
cert_file = '/home/sander/cert.pem',
authority = 'exogeni.net:umassvmsite',
username = 'sander')
exp = our.Experiment(tb, nodes = nodes)
with ExperimentManager(exp, do_swap_out = ExperimentManager.PROMPT):
exp.swap_in()
exp.install_prototype()
exp.bootstrap_prototype()
|
