da38d240d0269bc334e70ea619b86b85d7ab284f
[nepi.git] / src / nepi / resources / netns / netnsnodeinterface.py
1 #
2 #    NEPI, a framework to manage network experiments
3 #    Copyright (C) 2014 INRIA
4 #
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.
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 nepi.execution.attribute import Attribute, Flags, Types
21 from nepi.execution.resource import clsinit_copy
22 from nepi.resources.netns.netnsinterface import NetNSInterface
23
24 @clsinit_copy
25 class NetNSNodeInterface(NetNSInterface):
26     _rtype = "netns::NodeInterface"
27
28     @classmethod
29     def _register_attributes(cls):
30         up = Attribute("up", "Interface up", 
31                 default = True,
32                 type = Types.Bool)
33
34         cls._register_attribute(up)
35
36     @property
37     def emulation(self):
38         return self.node.emulation
39
40     @property
41     def node(self):
42         from nepi.resources.netns.netnsnode import NetNSNode
43         node = self.get_connected(NetNSNode.get_rtype())
44
45         if not node: 
46             msg = "Route not connected to Node!!"
47             self.error(msg)
48             raise RuntimeError, msg
49
50         return node[0]
51
52     @property
53     def switch(self):
54         from nepi.resources.netns.netnsswitch import NetNSSwitch
55         switch = self.get_connected(NetNSSwitch.get_rtype())
56         if switch: return switch[0]
57         return None
58
59     @property
60     def _rms_to_wait(self):
61         return [self.node, self.switch]
62
63     def _instantiate_object(self):
64         self._uuid = self.emulation.invoke(self.node.uuid, "add_if")
65         self.emulation.invoke(self.switch.uuid, "connect", self.uuid)
66         self.emulation.emu_set(self.uuid, "up", self.get("up"))
67
68