3 ###############################################################################
5 # NEPI, a framework to manage network experiments
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
22 ###############################################################################
24 from nepi.execution.ec import ExperimentController
25 from nepi.execution.runner import ExperimentRunner
26 from nepi.util.netgraph import NetGraph, TopologyType
27 import nepi.data.processing.ccn.parser as ccn_parser
33 content_name = "ccnx:/test/bunny.ts"
37 repofile = os.path.join(
38 os.path.dirname(os.path.realpath(__file__)), "repoFile1.0.8.2")
40 def get_simulator(ec):
41 simulator = ec.filter_resources("LinuxNS3Simulation")
44 node = ec.register_resource("LinuxNode")
45 ec.set(node, "hostname", "localhost")
47 simu = ec.register_resource("LinuxNS3Simulation")
48 ec.register_connection(simu, node)
53 def add_collector(ec, trace_name, subdir, newname = None):
54 collector = ec.register_resource("Collector")
55 ec.set(collector, "traceName", trace_name)
56 ec.set(collector, "subDir", subdir)
58 ec.set(collector, "rename", newname)
62 def add_dce_host(ec, nid):
63 simu = get_simulator(ec)
65 host = ec.register_resource("ns3::Node")
66 ec.set(host, "enableStack", True)
67 ec.register_connection(host, simu)
70 ec.netgraph.annotate_node(nid, "host", host)
72 def add_dce_ccnd(ec, nid):
73 # Retrieve annotation from netgraph
74 host = ec.netgraph.node_annotation(nid, "host")
76 # Add dce ccnd to the dce node
77 ccnd = ec.register_resource("ns3::LinuxDceCCND")
78 ec.set (ccnd, "stackSize", 1<<20)
79 ec.set (ccnd, "debug", 7)
80 ec.set (ccnd, "capacity", 50000)
81 ec.set (ccnd, "StartTime", "1s")
82 ec.set (ccnd, "StopTime", STOP_TIME)
83 ec.register_connection(ccnd, host)
85 # Collector to retrieve ccnd log
86 collector = add_collector(ec, "stderr", nid, "log")
87 ec.register_connection(collector, ccnd)
90 ec.netgraph.annotate_node(nid, "ccnd", ccnd)
92 def add_dce_ccnr(ec, nid):
93 # Retrieve annotation from netgraph
94 host = ec.netgraph.node_annotation(nid, "host")
96 # Add a CCN content repository to the dce node
97 ccnr = ec.register_resource("ns3::LinuxDceCCNR")
98 ec.set (ccnr, "repoFile1", repofile)
99 ec.set (ccnr, "stackSize", 1<<20)
100 ec.set (ccnr, "StartTime", "2s")
101 ec.set (ccnr, "StopTime", STOP_TIME)
102 ec.register_connection(ccnr, host)
104 def add_dce_ccncat(ec, nid):
105 # Retrieve annotation from netgraph
106 host = ec.netgraph.node_annotation(nid, "host")
108 # Add a ccncat application to the dce host
109 ccncat = ec.register_resource("ns3::LinuxDceCCNCat")
110 ec.set (ccncat, "contentName", content_name)
111 ec.set (ccncat, "stackSize", 1<<20)
112 ec.set (ccncat, "StartTime", "8s")
113 ec.set (ccncat, "StopTime", STOP_TIME)
114 ec.register_connection(ccncat, host)
116 def add_dce_fib_entry(ec, nid1, nid2):
117 # Retrieve annotations from netgraph
118 host1 = ec.netgraph.node_annotation(nid1, "host")
119 net = ec.netgraph.edge_net_annotation(nid1, nid2)
122 # Add FIB entry between peer hosts
123 ccndc = ec.register_resource("ns3::LinuxDceFIBEntry")
124 ec.set (ccndc, "protocol", "udp")
125 ec.set (ccndc, "uri", "ccnx:/")
126 ec.set (ccndc, "host", ip2)
127 ec.set (ccndc, "stackSize", 1<<20)
128 ec.set (ccndc, "StartTime", "2s")
129 ec.set (ccndc, "StopTime", STOP_TIME)
130 ec.register_connection(ccndc, host1)
132 def add_dce_net_iface(ec, nid1, nid2):
133 # Retrieve annotations from netgraph
134 host = ec.netgraph.node_annotation(nid1, "host")
135 net = ec.netgraph.edge_net_annotation(nid1, nid2)
137 prefix = net["prefix"]
139 dev = ec.register_resource("ns3::PointToPointNetDevice")
140 ec.set(dev,"DataRate", "5Mbps")
141 ec.set(dev, "ip", ip1)
142 ec.set(dev, "prefix", prefix)
143 ec.register_connection(host, dev)
145 queue = ec.register_resource("ns3::DropTailQueue")
146 ec.register_connection(dev, queue)
150 def avg_interests(ec, run):
152 logs_dir = ec.run_dir
156 interest_expiry_count,
157 interest_dupnonce_count,
159 content_count) = ccn_parser.process_content_history_logs(
161 ec.netgraph.topology)
163 shortest_path = networkx.shortest_path(graph,
164 source = ec.netgraph.sources()[0],
165 target = ec.netgraph.targets()[0])
167 ### Compute metric: Avg number of Interests seen per content name
168 ### normalized by the number of nodes in the shortest path
169 content_name_count = len(content_names.values())
170 nodes_in_shortest_path = len(shortest_path) - 1
171 metric = interest_count / (float(content_name_count) * float(nodes_in_shortest_path))
173 # TODO: DUMP RESULTS TO FILE
174 # TODO: DUMP GRAPH DELAYS!
175 f = open("/tmp/metric", "w+")
176 f.write("%.2f\n" % metric)
178 print " METRIC", metric
182 def add_dce_edge(ec, nid1, nid2):
183 ### Add network interfaces to hosts
184 p2p1 = add_dce_net_iface(ec, nid1, nid2)
185 p2p2 = add_dce_net_iface(ec, nid2, nid1)
187 # Create point to point link between interfaces
188 chan = ec.register_resource("ns3::PointToPointChannel")
189 ec.set(chan, "Delay", "0ms")
191 ec.register_connection(chan, p2p1)
192 ec.register_connection(chan, p2p2)
194 #### Add routing between CCN nodes
195 add_dce_fib_entry(ec, nid1, nid2)
196 add_dce_fib_entry(ec, nid2, nid1)
198 def add_dce_node(ec, nid):
199 ### Add CCN nodes (ec.netgraph holds the topology graph)
200 add_dce_host(ec, nid)
201 add_dce_ccnd(ec, nid)
203 if nid == ec.netgraph.targets()[0]:
204 add_dce_ccnr(ec, nid)
206 if nid == ec.netgraph.sources()[0]:
207 add_dce_ccncat(ec, nid)
209 if __name__ == '__main__':
211 #### Create NEPI Experiment Description with LINEAR topology
212 ec = ExperimentController("dce_ccn",
213 topo_type = TopologyType.LINEAR,
217 add_node_callback = add_dce_node,
218 add_edge_callback = add_dce_edge)
220 print "Results stored at", ec.exp_dir
222 #### Retrieve the consumer to wait for ot to finish
223 ccncat = ec.filter_resources("ns3::LinuxDceCCNCat")
225 #### Run experiment until metric convergences
226 rnr = ExperimentRunner()
227 runs = rnr.run(ec, min_runs = 1, max_runs = 1,
228 compute_metric_callback = avg_interests,