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