examples/ns3/local_wifi_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 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 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  20
  21 from nepi.execution.ec import ExperimentController
  22
  23 ec = ExperimentController(exp_id = "ns3-local-wifi-ping")
  24
  25 # Simulation will executed in the local machine
  26 node = ec.register_resource("LinuxNode")
  27 ec.set(node, "hostname", "localhost")
  28
  29 # Add a simulation resource
  30 simu = ec.register_resource("LinuxNS3Simulation")
  31 ec.set(simu, "verbose", True)
  32 ec.register_connection(simu, node)
  33
  34 ## Add a ns-3 node with its protocol stack
  35 nsnode1 = ec.register_resource("ns3::Node")
  36 ec.register_connection(nsnode1, simu)
  37
  38 ipv4 = ec.register_resource("ns3::Ipv4L3Protocol")
  39 ec.register_connection(nsnode1, ipv4)
  40 arp = ec.register_resource("ns3::ArpL3Protocol")
  41 ec.register_connection(nsnode1, arp)
  42 icmp = ec.register_resource("ns3::Icmpv4L4Protocol")
  43 ec.register_connection(nsnode1, icmp)
  44
  45 # Adding constant mobility to the ns-3 node
  46 mobility1 = ec.register_resource("ns3::ConstantPositionMobilityModel")
  47 position1 = "%d:%d:%d" % (0, 0, 0)
  48 ec.set(mobility1, "Position", position1)
  49 ec.register_connection(nsnode1, mobility1)
  50
  51 # Add a wifi access point net device to the node
  52 dev1 = ec.register_resource("ns3::WifiNetDevice")
  53 ec.set(dev1, "ip", "10.0.0.1")
  54 ec.set(dev1, "prefix", "30")
  55 ec.register_connection(nsnode1, dev1)
  56
  57 phy1 = ec.register_resource("ns3::YansWifiPhy")
  58 ec.set(phy1, "Standard", "WIFI_PHY_STANDARD_80211a")
  59 ec.register_connection(dev1, phy1)
  60
  61 error1 = ec.register_resource("ns3::NistErrorRateModel")
  62 ec.register_connection(phy1, error1)
  63
  64 manager1 = ec.register_resource("ns3::ArfWifiManager")
  65 ec.register_connection(dev1, manager1)
  66
  67 mac1 = ec.register_resource("ns3::ApWifiMac")
  68 ec.set(mac1, "Standard", "WIFI_PHY_STANDARD_80211a")
  69 ec.register_connection(dev1, mac1)
  70
  71 ## Add another ns-3 node with its protocol stack
  72 nsnode2 = ec.register_resource("ns3::Node")
  73 ec.register_connection(nsnode2, simu)
  74
  75 ipv4 = ec.register_resource("ns3::Ipv4L3Protocol")
  76 ec.register_connection(nsnode2, ipv4)
  77 arp = ec.register_resource("ns3::ArpL3Protocol")
  78 ec.register_connection(nsnode2, arp)
  79 icmp = ec.register_resource("ns3::Icmpv4L4Protocol")
  80 ec.register_connection(nsnode2, icmp)
  81
  82 # Adding constant mobility to the ns-3 node
  83 mobility2 = ec.register_resource("ns3::ConstantPositionMobilityModel")
  84 position2 = "%d:%d:%d" % (50, 50, 0)
  85 ec.set(mobility2, "Position", position1)
  86 ec.register_connection(nsnode2, mobility2)
  87
  88 # Add a wifi station net device to the node
  89 dev2 = ec.register_resource("ns3::WifiNetDevice")
  90 ec.set(dev2, "ip", "10.0.0.2")
  91 ec.set(dev2, "prefix", "30")
  92 ec.register_connection(nsnode2, dev2)
  93
  94 phy2 = ec.register_resource("ns3::YansWifiPhy")
  95 ec.set(phy2, "Standard", "WIFI_PHY_STANDARD_80211a")
  96 ec.register_connection(dev2, phy2)
  97
  98 error2 = ec.register_resource("ns3::NistErrorRateModel")
  99 ec.register_connection(phy2, error2)
 100
 101 manager2 = ec.register_resource("ns3::ArfWifiManager")
 102 ec.register_connection(dev2, manager2)
 103
 104 mac2 = ec.register_resource("ns3::StaWifiMac")
 105 ec.set(mac2, "Standard", "WIFI_PHY_STANDARD_80211a")
 106 ec.register_connection(dev2, mac2)
 107
 108 # Add a wifi channel
 109 chan = ec.register_resource("ns3::YansWifiChannel")
 110 delay = ec.register_resource("ns3::ConstantSpeedPropagationDelayModel")
 111 ec.register_connection(chan, delay)
 112 loss = ec.register_resource("ns3::LogDistancePropagationLossModel")
 113 ec.register_connection(chan, loss)
 114 ec.register_connection(chan, phy1)
 115 ec.register_connection(chan, phy2)
 116
 117 ### create pinger
 118 ping = ec.register_resource("ns3::V4Ping")
 119 ec.set (ping, "Remote", "10.0.0.2")
 120 ec.set (ping, "Interval", "1s")
 121 ec.set (ping, "Verbose", True)
 122 ec.set (ping, "StartTime", "0s")
 123 ec.set (ping, "StopTime", "20s")
 124 ec.register_connection(ping, nsnode1)
 125
 126 ec.deploy()
 127
 128 ec.wait_finished([ping])
 129
 130 stdout = ec.trace(simu, "stdout")
 131
 132 ec.shutdown()
 133
 134 print "PING OUTPUT", stdout