use print() - import print_function - should be fine for both py2 and py3
[nepi.git] / examples / dce / wrapped_local_p2p_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 __future__ import print_function
21
22 from nepi.execution.ec import ExperimentController 
23
24 def add_ns3_node(ec, simu):
25     node = ec.register_resource("ns3::Node")
26     ec.register_connection(node, simu)
27
28     ipv4 = ec.register_resource("ns3::Ipv4L3Protocol")
29     ec.register_connection(node, ipv4)
30
31     arp = ec.register_resource("ns3::ArpL3Protocol")
32     ec.register_connection(node, arp)
33     
34     icmp = ec.register_resource("ns3::Icmpv4L4Protocol")
35     ec.register_connection(node, icmp)
36
37     udp = ec.register_resource("ns3::UdpL4Protocol")
38     ec.register_connection(node, udp)
39
40     tcp = ec.register_resource("ns3::TcpL4Protocol")
41     ec.register_connection(node, tcp)
42
43     return node
44
45 def add_device(ec, node, ip,  prefix):
46     dev = ec.register_resource("ns3::PointToPointNetDevice")
47     ec.set(dev, "ip", ip)
48     ec.set(dev, "prefix", prefix)
49     ec.register_connection(node, dev)
50
51     queue = ec.register_resource("ns3::DropTailQueue")
52     ec.register_connection(dev, queue)
53
54     return dev
55
56 ec = ExperimentController(exp_id = "dce-ping-app")
57
58 node = ec.register_resource("linux::Node")
59 ec.set(node, "hostname", "localhost")
60 ec.set(node, "cleanProcesses", True)
61
62 simu = ec.register_resource("linux::ns3::Simulation")
63 ec.register_connection(simu, node)
64
65 nsnode1 = add_ns3_node(ec, simu)
66 dev1 = add_device(ec, nsnode1, "10.0.0.1", "30")
67 ec.set(dev1, "DataRate", "5Mbps")
68
69 nsnode2 = add_ns3_node(ec, simu)
70 dev2 = add_device(ec, nsnode2, "10.0.0.2", "30")
71 ec.set(dev2, "DataRate", "5Mbps")
72
73 # Create channel
74 chan = ec.register_resource("ns3::PointToPointChannel")
75 ec.set(chan, "Delay", "2ms")
76
77 ec.register_connection(chan, dev1)
78 ec.register_connection(chan, dev2)
79
80 ### create applications
81 ping = ec.register_resource("linux::ns3::dce::Ping")
82 ec.set (ping, "stackSize", 1<<20)
83 ec.set (ping, "target", "10.0.0.2")
84 ec.set (ping, "count", "10")
85 ec.set (ping, "packetSize", "1000")
86 ec.set (ping, "StartTime", "1s")
87 ec.set (ping, "StopTime", "20s")
88 ec.register_connection(ping, nsnode1)
89
90 ec.deploy()
91
92 ec.wait_finished([ping])
93
94 stdout = ec.trace(ping, "stdout") 
95
96 ec.shutdown()
97
98 print("PING OUTPUT", stdout)
99