2 # NEPI, a framework to manage network experiments
3 # Copyright (C) 2014 INRIA
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
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.
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/>.
18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
20 from nepi.execution.attribute import Attribute, Flags
21 from nepi.execution.resource import clsinit_copy
22 from nepi.execution.trace import Trace
23 from nepi.resources.ns3.ns3base import NS3Base
28 class NS3BaseNetDevice(NS3Base):
29 _rtype = "abstract::ns3::NetDevice"
32 def _register_attributes(cls):
33 mac = Attribute("mac", "MAC address for device",
36 ip = Attribute("ip", "IP address for device",
39 prefix = Attribute("prefix", "Network prefix for device",
42 cls._register_attribute(mac)
43 cls._register_attribute(ip)
44 cls._register_attribute(prefix)
47 def _register_traces(cls):
48 pcap = Trace("pcap", "Dump traffic sniffed on the network device in Pcap format")
49 promisc_pcap = Trace("promiscPcap", "Dump traffic sniffed in promiscuous mode on the network device in Pcap format")
50 ascii = Trace("ascii", "Dump traffic sniffed on the network device in Ascii format")
52 cls._register_trace(pcap)
53 cls._register_trace(promisc_pcap)
54 cls._register_trace(ascii)
56 def __init__(self, ec, guid):
57 super(NS3BaseNetDevice, self).__init__(ec, guid)
58 self._ascii_helper_uuid = None
59 self._device_helper_uuid = None
63 from nepi.resources.ns3.ns3node import NS3BaseNode
64 nodes = self.get_connected(NS3BaseNode.get_rtype())
67 msg = "Device not connected to node"
69 raise RuntimeError, msg
75 from nepi.resources.ns3.ns3channel import NS3BaseChannel
76 channels = self.get_connected(NS3BaseChannel.get_rtype())
79 msg = "Device not connected to channel"
81 raise RuntimeError, msg
87 from nepi.resources.ns3.ns3queue import NS3BaseQueue
88 queue = self.get_connected(NS3BaseQueue.get_rtype())
91 msg = "Device not connected to queue"
93 raise RuntimeError, msg
98 def ascii_helper_uuid(self):
99 if not self._ascii_helper_uuid:
100 self._ascii_helper_uuid = self.simulation.create("AsciiTraceHelper")
101 return self._ascii_helper_uuid
104 def device_helper_uuid(self):
105 if not self._device_helper_uuid:
106 rtype = self.get_rtype()
107 if rtype == "ns3::PointToPointNetDevice":
108 classname = "PointToPointHelper"
109 elif rtype == "ns3::CsmaNetDevice":
110 classname = "CsmaHelper"
111 elif rtype == "ns3::EmuNetDevice":
112 classname = "EmuHelper"
113 elif rtype == "ns3::FdNetDevice":
114 classname = "FdNetDeviceHelper"
115 elif rtype in [ "ns3::BaseStationNetDevice", "SubscriberStationNetDevice" ]:
116 classname = "WimaxHelper"
117 elif rtype == "ns3::WifiNetDevice":
118 classname = "YansWifiPhyHelper"
120 self._device_helper_uuid = self.simulation.create(classname)
122 return self._device_helper_uuid
125 def _rms_to_wait(self):
135 rms.add(self.channel)
138 def _configure_object(self):
140 self._configure_mac_address()
143 self._configure_ip_address()
146 self._configure_traces()
148 def _configure_mac_address(self):
149 mac = self.get("mac")
151 mac_uuid = self.simulation.create("Mac48Address", mac)
153 mac_uuid = self.simulation.invoke("singleton::Mac48Address", "Allocate")
155 self.simulation.invoke(self.uuid, "SetAddress", mac_uuid)
157 def _configure_ip_address(self):
159 prefix = self.get("prefix")
161 i = ipaddr.IPAddress(ip)
164 ipv4 = self.node.ipv4
165 ifindex_uuid = self.simulation.invoke(ipv4.uuid, "AddInterface",
167 ipv4_addr_uuid = self.simulation.create("Ipv4Address", ip)
168 ipv4_mask_uuid = self.simulation.create("Ipv4Mask", "/%s" % str(prefix))
169 inaddr_uuid = self.simulation.create("Ipv4InterfaceAddress",
170 ipv4_addr_uuid, ipv4_mask_uuid)
171 self.simulation.invoke(ipv4.uuid, "AddAddress", ifindex_uuid,
173 self.simulation.invoke(ipv4.uuid, "SetMetric", ifindex_uuid, 1)
174 self.simulation.invoke(ipv4.uuid, "SetUp", ifindex_uuid)
180 def _configure_traces(self):
181 if self.trace_enabled("pcap"):
182 helper_uuid = self.device_helper_uuid
184 filename = "trace-pcap-netdev-%d.pcap" % self.guid
185 self._trace_filename["pcap"] = filename
187 filepath = self.simulation.trace_filepath(filename)
189 self.simulation.invoke(helper_uuid, "EnablePcap", filepath,
190 self.uuid, promiscuous = False, explicitFilename = True)
192 if self.trace_enabled("promiscPcap"):
193 helper_uuid = self.device_helper_uuid
195 filename = "trace-promisc-pcap-netdev-%d.pcap" % self.guid
196 self._trace_filename["promiscPcap"] = filename
198 filepath = self.simulation.trace_filepath(filename)
200 self.simulation.invoke(helper_uuid, "EnablePcap", filepath,
201 self.uuid, promiscuous = True, explicitFilename = True)
203 if self.trace_enabled("ascii"):
204 helper_uuid = self.device_helper_uuid
205 ascii_helper_uuid = self.ascii_helper_uuid
207 filename = "trace-ascii-netdev-%d.tr" % self.guid
208 self._trace_filename["ascii"] = filename
210 filepath = self.simulation.trace_filepath(filename)
211 stream_uuid = self.simulation.invoke(ascii_helper_uuid,
212 "CreateFileStream", filepath)
213 self.simulation.invoke(helper_uuid, "EnableAscii", stream_uuid,
216 def _connect_object(self):
218 if node and node.uuid not in self.connected:
219 self.simulation.invoke(node.uuid, "AddDevice", self.uuid)
220 self._connected.add(node.uuid)
222 channel = self.channel
223 if channel and channel.uuid not in self.connected:
224 self.simulation.invoke(self.uuid, "Attach", channel.uuid)
225 self._connected.add(channel.uuid)
227 # Verify that the device has a queue. If no queue is added a segfault