test/resources/planetlab/ns3/cross_ns3_planetlab_fdtunnel_ping.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 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  19
  20 from nepi.execution.ec import ExperimentController
  21 from nepi.execution.resource import ResourceState, ResourceAction
  22 from nepi.execution.trace import TraceAttr
  23
  24 from test_utils import skipIfAnyNotAliveWithIdentity
  25
  26 import os
  27 import time
  28 import unittest
  29
  30 def add_ns3_node(ec, simu):
  31     node = ec.register_resource("ns3::Node")
  32     ec.set(node, "enableStack", True)
  33     ec.register_connection(node, simu)
  34
  35     return node
  36
  37 def add_fd_device(ec, node, ip, prefix):
  38     dev = ec.register_resource("ns3::FdNetDevice")
  39     ec.set(dev, "ip", ip)
  40     ec.set(dev, "prefix", prefix)
  41     ec.register_connection(node, dev)
  42
  43     return dev
  44
  45 def add_tap_device(ec, node, ip, prefix):
  46     dev = ec.register_resource("linux::Tap")
  47     ec.set(dev, "ip", ip)
  48     ec.set(dev, "prefix", prefix)
  49     ec.register_connection(node, dev)
  50
  51     return dev
  52
  53 def add_point2point_device(ec, node, ip, prefix):
  54     dev = ec.register_resource("ns3::PointToPointNetDevice")
  55     ec.set(dev, "ip", ip)
  56     ec.set(dev, "prefix", prefix)
  57     ec.register_connection(node, dev)
  58
  59     queue = ec.register_resource("ns3::DropTailQueue")
  60     ec.register_connection(dev, queue)
  61
  62     return dev
  63
  64 class LinuxNS3FdNetDeviceTest(unittest.TestCase):
  65     def setUp(self):
  66         self.hostname1 = "onelab4.warsaw.rd.tp.pl"
  67         self.hostname2 = "inriarennes1.irisa.fr"
  68         self.slicename = "inria_nepi"
  69         self.identity = "%s/.ssh/id_rsa_planetlab" % (os.environ['HOME'])
  70
  71     @skipIfAnyNotAliveWithIdentity
  72     def t_cross_tunnel_ping(self, username1, hostname1, identity1,
  73             username2, hostname2, identity2):
  74         ec = ExperimentController(exp_id = "test-ns3-pl-fd-tunnel")
  75
  76         node1 = ec.register_resource("planetlab::Node")
  77         ec.set(node1, "hostname", hostname1)
  78         ec.set(node1, "username", username1)
  79         ec.set(node1, "identity", identity1)
  80         ec.set(node1, "cleanProcesses", True)
  81         ec.set(node1, "cleanExperiment", True)
  82
  83         simu1 = ec.register_resource("linux::ns3::Simulation")
  84         ec.set(simu1, "simulatorImplementationType", "ns3::RealtimeSimulatorImpl")
  85         ec.set(simu1, "checksumEnabled", True)
  86         ec.set(simu1, "verbose", True)
  87         #ec.set(simu1, "buildMode", "debug")
  88         #ec.set(simu1, "nsLog", "FdNetDevice")
  89         ec.register_connection(simu1, node1)
  90
  91         nsnode1 = ec.register_resource("ns3::Node")
  92         ec.set(nsnode1, "enableStack", True)
  93         ec.register_connection(nsnode1, simu1)
  94
  95         fddev1 = ec.register_resource("ns3::FdNetDevice")
  96         ec.set(fddev1, "ip", "10.0.0.1")
  97         ec.set(fddev1, "prefix", "30")
  98         ec.register_connection(nsnode1, fddev1)
  99
 100         node2 = ec.register_resource("planetlab::Node")
 101         ec.set(node2, "hostname", hostname2)
 102         ec.set(node2, "username", username2)
 103         ec.set(node2, "identity", identity2)
 104         ec.set(node2, "cleanProcesses", True)
 105         ec.set(node2, "cleanExperiment", True)
 106
 107         simu2 = ec.register_resource("linux::ns3::Simulation")
 108         ec.set(simu2, "simulatorImplementationType", "ns3::RealtimeSimulatorImpl")
 109         ec.set(simu2, "checksumEnabled", True)
 110         ec.set(simu2, "verbose", True)
 111         #ec.set(simu2, "buildMode", "debug")
 112         #ec.set(simu2, "nsLog", "FdNetDevice")
 113         ec.register_connection(simu2, node2)
 114
 115         nsnode2 = ec.register_resource("ns3::Node")
 116         ec.set(nsnode2, "enableStack", True)
 117         ec.register_connection(nsnode2, simu2)
 118
 119         fddev2 = ec.register_resource("ns3::FdNetDevice")
 120         ec.set(fddev2, "ip", "10.0.0.2")
 121         ec.set(fddev2, "prefix", "30")
 122         ec.register_connection(nsnode2, fddev2)
 123
 124         tunnel = ec.register_resource("planetlab::ns3::FdUdpTunnel")
 125         ec.register_connection(tunnel, fddev1)
 126         ec.register_connection(tunnel, fddev2)
 127
 128         ping = ec.register_resource("ns3::V4Ping")
 129         ec.set (ping, "Remote", "10.0.0.2")
 130         ec.set (ping, "Interval", "1s")
 131         ec.set (ping, "Verbose", True)
 132         ec.set (ping, "StartTime", "0s")
 133         ec.set (ping, "StopTime", "20s")
 134         ec.register_connection(ping, nsnode1)
 135
 136         ec.deploy()
 137
 138         ec.wait_finished([ping])
 139
 140         stdout = ec.trace(simu1, "stdout")
 141
 142         expected = "20 packets transmitted, 20 received, 0% packet loss"
 143         self.assertTrue(stdout.find(expected) > -1)
 144
 145         ec.shutdown()
 146
 147     def test_cross_tunnel_ping(self):
 148         self.t_cross_tunnel_ping(self.slicename, self.hostname1, self.identity,
 149                 self.slicename, self.hostname2, self.identity)
 150
 151
 152 if __name__ == '__main__':
 153     unittest.main()
 154