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.resources.ns3.ns3base import NS3Base
27 class NS3BaseNetDevice(NS3Base):
28 _rtype = "abstract::ns3::NetDevice"
31 def _register_attributes(cls):
32 mac = Attribute("mac", "MAC address for device",
35 ip = Attribute("ip", "IP address for device",
38 prefix = Attribute("prefix", "Network prefix for device",
41 cls._register_attribute(mac)
42 cls._register_attribute(ip)
43 cls._register_attribute(prefix)
47 from nepi.resources.ns3.ns3node import NS3BaseNode
48 nodes = self.get_connected(NS3BaseNode.get_rtype())
51 msg = "Device not connected to node"
53 raise RuntimeError, msg
59 from nepi.resources.ns3.ns3channel import NS3BaseChannel
60 channels = self.get_connected(NS3BaseChannel.get_rtype())
63 msg = "Device not connected to channel"
65 raise RuntimeError, msg
70 def _rms_to_wait(self):
80 others.add(self.channel)
83 def _configure_object(self):
87 mac_uuid = self.simulation.create("Mac48Address", mac)
89 mac_uuid = self.simulation.invoke("singleton::Mac48Address", "Allocate")
90 self.simulation.invoke(self.uuid, "SetAddress", mac_uuid)
94 prefix = self.get("prefix")
96 i = ipaddr.IPAddress(ip)
100 ifindex_uuid = self.simulation.invoke(ipv4.uuid, "AddInterface",
102 ipv4_addr_uuid = self.simulation.create("Ipv4Address", ip)
103 ipv4_mask_uuid = self.simulation.create("Ipv4Mask", "/%s" % str(prefix))
104 inaddr_uuid = self.simulation.create("Ipv4InterfaceAddress",
105 ipv4_addr_uuid, ipv4_mask_uuid)
106 self.simulation.invoke(ipv4.uuid, "AddAddress", ifindex_uuid,
108 self.simulation.invoke(ipv4.uuid, "SetMetric", ifindex_uuid, 1)
109 self.simulation.invoke(ipv4.uuid, "SetUp", ifindex_uuid)
115 def _connect_object(self):
117 if node and node.uuid not in self.connected:
118 self.simulation.invoke(node.uuid, "AddDevice", self.uuid)
119 self._connected.add(node.uuid)
121 channel = self.channel
122 if channel and channel.uuid not in self.connected:
123 self.simulation.invoke(self.uuid, "Attach", channel.uuid)
124 self._connected.add(channel.uuid)