ca1ebeed5a68c6d9717a1b6eb4ef3bf338542a29
[nepi.git] / examples / openvswitch / ovs_ping_3switches_line.py
1 #!/usr/bin/env python
2 #
3 #    NEPI, a framework to manage network experiments
4 #    Copyright (C) 2013 INRIA
5 #
6 #    This program is free software: you can redistribute it and/or modify
7 #    it under the terms of the GNU General Public License as published by
8 #    the Free Software Foundation, either version 3 of the License, or
9 #    (at your option) any later version.
10 #
11 #    This program is distributed in the hope that it will be useful,
12 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
13 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 #    GNU General Public License for more details.
15 #
16 #    You should have received a copy of the GNU General Public License
17 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 #
19 # Authors :  Julien Tribino <julien.tribino@inria.fr>
20 #          Alina Quereilhac <alina.quereilhac@inria.fr>
21 #
22 # Topology :
23 #
24 #         Switch1 ------ Switch2 ------- Switch2         
25 #            /              |                \           
26 #           /               |                 \          
27 #          /                |                  \         
28 #       Host1             Host3               Host2   
29
30
31
32 from nepi.execution.ec import ExperimentController
33 import os, time
34
35 def add_node(ec, host, user, pl_user, pl_password):
36     node = ec.register_resource("PlanetlabNode")
37     ec.set(node, "hostname", host)
38     ec.set(node, "username", user)
39     if pl_user:
40         ec.set(node, "pluser", pl_user)
41     if pl_password:
42         ec.set(node, "plpassword", pl_password)
43     ec.set(node, "cleanHome", True)
44     ec.set(node, "cleanProcesses", True)
45
46     return node
47
48 def add_ovs(ec, bridge_name, virtual_ip_pref, controller_ip, controller_port, node):
49     ovs = ec.register_resource("OVSSwitch")
50     ec.set(ovs, "bridge_name", bridge_name)
51     ec.set(ovs, "virtual_ip_pref", virtual_ip_pref)
52     ec.set(ovs, "controller_ip", controller_ip)
53     ec.set(ovs, "controller_port", controller_port)
54     ec.register_connection(ovs, node)
55     return ovs
56
57 def add_port(ec, port_name, ovs):
58     port = ec.register_resource("OVSPort")
59     ec.set(port, "port_name", port_name)
60     ec.register_connection(port, ovs)
61     return port
62
63 def add_tap(ec, ip4, prefix4, pointopoint, node):
64     tap = ec.register_resource("PlanetlabTap")
65     ec.set(tap, "ip4", ip4)
66     ec.set(tap, "prefix4", prefix4)
67     ec.set(tap, "pointopoint", pointopoint)
68     ec.set(tap, "up", True)
69     ec.register_connection(tap, node)
70     return tap
71
72 def add_tunnel(ec, network, port0, tap):
73     tunnel = ec.register_resource("OVSTunnel")
74     ec.set(tunnel, "network", network)
75     ec.register_connection(port0, tunnel)
76     ec.register_connection(tunnel, tap)
77     return tunnel
78
79 def add_app(ec, command, node):
80     app = ec.register_resource("LinuxApplication")
81     ec.set(app, "command", command)
82     ec.register_connection(app, node)
83     return app
84
85 # Create the EC
86 ec = ExperimentController(exp_id = "test-tr")
87
88 switch1 = "planetlab2.virtues.fi"
89 switch2 = "planetlab2.upc.es"
90 switch3 = "planetlab2.cs.aueb.gr"
91 host1 = "planetlab2.ionio.gr"
92 host2 = "iraplab2.iralab.uni-karlsruhe.de"
93 host3 = "planetlab2.diku.dk"
94
95 ip_controller = "194.254.215.12"
96 network = "192.168.3.0"
97
98 slicename = "inria_nepi"
99
100 pl_user = os.environ.get("PL_USER")
101 pl_password = os.environ.get("PL_PASS")
102
103 s1_node = add_node(ec, switch1, slicename, pl_user, pl_password)
104 s2_node = add_node(ec, switch2, slicename, pl_user, pl_password)
105 s3_node = add_node(ec, switch3, slicename, pl_user, pl_password)
106
107 # Add switches 
108 ovs1 = add_ovs(ec, "nepi_bridge_1", "192.168.3.2/24", ip_controller, "6633", s1_node)
109 ovs2 = add_ovs(ec, "nepi_bridge_2", "192.168.3.4/24", ip_controller, "6633", s2_node)
110 ovs3 = add_ovs(ec, "nepi_bridge_3", "192.168.3.6/24", ip_controller, "6633", s3_node)
111
112 # Add ports on ovs
113 port1 = add_port(ec, "nepi_port1", ovs1)
114 port4 = add_port(ec, "nepi_port4", ovs1)
115 port7 = add_port(ec, "nepi_port7", ovs1)
116 port2 = add_port(ec, "nepi_port2", ovs2)
117 port5 = add_port(ec, "nepi_port5", ovs2)
118 #port8 = add_port(ec, "nepi_port8", ovs2)
119 port3 = add_port(ec, "nepi_port3", ovs3)
120 port6 = add_port(ec, "nepi_port6", ovs3)
121 #port9 = add_port(ec, "nepi_port9", ovs3)
122
123 h1_node = add_node(ec, host1, slicename, pl_user, pl_password)
124 h2_node = add_node(ec, host2, slicename, pl_user, pl_password)
125 h3_node = add_node(ec, host3, slicename, pl_user, pl_password)
126
127 # Add tap devices
128 tap1 = add_tap(ec, "192.168.3.1", 24, "192.168.3.2", h1_node)
129 tap2 = add_tap(ec, "192.168.3.3", 24, "192.168.3.4", h2_node)
130 tap3 = add_tap(ec, "192.168.3.5", 24, "192.168.3.6", h3_node)
131
132 # Connect the nodes
133 tunnel1 = add_tunnel(ec, network, port1, tap1)
134 tunnel2 = add_tunnel(ec, network, port2, tap2)
135 tunnel3 = add_tunnel(ec, network, port3, tap3)
136 tunnel4 = add_tunnel(ec, network, port4, port5)
137 tunnel5 = add_tunnel(ec, network, port7, port6)
138 #tunnel6 = add_tunnel(ec, network, port8, port9)
139
140 # Add ping commands
141 app1 = add_app(ec, "ping -c5 192.168.3.4", s1_node)
142 app2 = add_app(ec, "ping -c5 192.168.3.6", s1_node)
143 app3 = add_app(ec, "ping -c5 192.168.3.2", s2_node)
144 app4 = add_app(ec, "ping -c5 192.168.3.6", s2_node)
145 app5 = add_app(ec, "ping -c5 192.168.3.2", s3_node)
146 app6 = add_app(ec, "ping -c5 192.168.3.4", s3_node)
147
148 app7 = add_app(ec, "ping -c5 192.168.3.3", h1_node)
149 app8 = add_app(ec, "ping -c5 192.168.3.5", h1_node)
150 app9 = add_app(ec, "ping -c5 192.168.3.1", h2_node)
151 app10 = add_app(ec, "ping -c5 192.168.3.5", h2_node)
152 app11 = add_app(ec, "ping -c5 192.168.3.1", h3_node)
153 app12 = add_app(ec, "ping -c5 192.168.3.3", h3_node)
154
155 ec.deploy()
156
157 ec.wait_finished([app1, app2, app3, app4, app5, app6, app7, app8, app9, app10, app11, app12])
158
159 # Retreive ping results and save
160 # them in a file
161 ping1 = ec.trace(app1, 'stdout')
162 ping2 = ec.trace(app2, 'stdout')
163 ping3 = ec.trace(app3, 'stdout')
164 ping4 = ec.trace(app4, 'stdout')
165 ping5 = ec.trace(app5, 'stdout')
166 ping6 = ec.trace(app6, 'stdout')
167 ping7 = ec.trace(app7, 'stdout')
168 ping8 = ec.trace(app8, 'stdout')
169 ping9 = ec.trace(app9, 'stdout')
170 ping10 = ec.trace(app10, 'stdout')
171 ping11 = ec.trace(app11, 'stdout')
172 ping12 = ec.trace(app12, 'stdout')
173
174
175 f = open("examples/openvswitch/ovs_ping_3switches_line.txt", 'w')
176
177 if not ping12:
178   ec.shutdown()
179
180 f.write("************ Ping From Switch 1 : 192.168.3.2 ********************\n\n")
181 f.write(ping1)
182 f.write("--------------------------------------\n")
183 f.write(ping2)
184 f.write("************ Ping From Switch 2 : 192.168.3.4 ********************\n\n")
185 f.write(ping3)
186 f.write("--------------------------------------\n")
187 f.write(ping4)
188 f.write("************ Ping From Switch 3 : 192.168.3.6 ********************\n\n")
189 f.write(ping5)
190 f.write("--------------------------------------\n")
191 f.write(ping6)
192 f.write("************ Ping From Host 1 : 192.168.3.1 ********************\n\n")
193 f.write(ping7)
194 f.write("--------------------------------------\n")
195 f.write(ping8)
196 f.write("************ Ping From Host 2 : 192.168.3.3 ********************\n\n")
197 f.write(ping9)
198 f.write("--------------------------------------\n")
199 f.write(ping10)
200 f.write("************ Ping From Host 3 : 192.168.3.5 ********************\n\n")
201 f.write(ping11)
202 f.write("--------------------------------------\n")
203 f.write(ping12)
204 f.close()
205
206 # Delete the overlay network
207 ec.shutdown()
208
209
210
211
212