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, Types
21 from nepi.execution.resource import clsinit_copy
22 from nepi.resources.ns3.ns3base import NS3Base
25 class NS3BaseNode(NS3Base):
26 _rtype = "abstract::ns3::Node"
29 def _register_attributes(cls):
30 enable_dce = Attribute("enableDCE",
31 "This node will run in DCE emulation mode ",
36 cls._register_attribute(enable_dce)
40 from nepi.resources.ns3.ns3simulation import NS3Simulation
41 for guid in self.connections:
42 rm = self.ec.get_resource(guid)
43 if isinstance(rm, NS3Simulation):
46 msg = "Node not connected to simulation"
48 raise RuntimeError, msg
52 from nepi.resources.ns3.ns3ipv4l3protocol import NS3BaseIpv4L3Protocol
53 ipv4s = self.get_connected(NS3BaseIpv4L3Protocol.get_rtype())
54 if ipv4s: return ipv4s[0]
59 from nepi.resources.ns3.ns3mobilitymodel import NS3BaseMobilityModel
60 mobility = self.get_connected(NS3BaseMobilityModel.get_rtype())
61 if mobility: return mobility[0]
66 from nepi.resources.ns3.ns3netdevice import NS3BaseNetDevice
67 devices = self.get_connected(NS3BaseNetDevice.get_rtype())
70 msg = "Node not connected to devices"
72 raise RuntimeError, msg
77 def _rms_to_wait(self):
79 rms.add(self.simulation)
85 mobility = self.mobility
91 def _configure_object(self):
92 ### node.AggregateObject(PacketSocketFactory())
93 uuid_packet_socket_factory = self.simulation.create("PacketSocketFactory")
94 self.simulation.invoke(self.uuid, "AggregateObject", uuid_packet_socket_factory)
96 if self.get("enableDCE") == True:
99 def _connect_object(self):
102 self.simulation.invoke(self.uuid, "AggregateObject", ipv4.uuid)
104 mobility = self.mobility
106 self.simulation.invoke(self.uuid, "AggregateObject", mobility.uuid)
109 # TODO: All these component types should be configurable somewhere
111 manager_uuid = self.simulation.create("ns3::TaskManager")
112 m_schedulerFactory.SetTypeId ("ns3::RrTaskScheduler");
113 m_managerFactory.SetTypeId ("ns3::DceManager");
114 m_networkStackFactory.SetTypeId ("ns3::Ns3SocketFdFactory");
115 m_delayFactory.SetTypeId ("ns3::RandomProcessDelayModel");
117 Ptr<TaskManager> taskManager = m_taskManagerFactory.Create<TaskManager> ();
118 Ptr<TaskScheduler> scheduler = m_schedulerFactory.Create<TaskScheduler> ();
119 Ptr<LoaderFactory> loader = m_loaderFactory.Create<LoaderFactory> ();
120 Ptr<SocketFdFactory> networkStack = m_networkStackFactory.Create<SocketFdFactory> ();
121 Ptr<ProcessDelayModel> delay = m_delayFactory.Create<ProcessDelayModel> ();
123 taskManager->SetScheduler (scheduler);
124 taskManager->SetDelayModel (delay);
125 manager->SetAttribute ("FirstPid", UintegerValue (g_firstPid.GetInteger (0, 0xffff)));
127 node->AggregateObject (taskManager);
128 node->AggregateObject (loader);
129 node->AggregateObject (manager);
130 node->AggregateObject (networkStack);
131 node->AggregateObject (CreateObject<LocalSocketFdFactory> ());
132 manager->AggregateObject (CreateObject<DceNodeContext> ());
133 manager->SetVirtualPath (GetVirtualPath ());