2 # NEPI, a framework to manage network experiments
3 # Copyright (C) 2013 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>
19 # Julien Tribino <julien.tribino@inria.fr>
21 from nepi.execution.resource import ResourceManager, clsinit_copy, \
22 ResourceState, reschedule_delay, failtrap
23 from nepi.execution.attribute import Attribute, Flags
24 from nepi.resources.omf.omf_resource import ResourceGateway, OMFResource
25 from nepi.resources.omf.node import OMFNode
26 from nepi.resources.omf.omf_api import OMFAPIFactory
29 class OMFApplication(OMFResource):
31 .. class:: Class Args :
33 :param ec: The Experiment controller
34 :type ec: ExperimentController
35 :param guid: guid of the RM
37 :param creds: Credentials to communicate with the rm (XmppClient)
42 This class is used only by the Experiment Controller through the
46 _rtype = "OMFApplication"
47 _authorized_connections = ["OMFNode"]
50 def _register_attributes(cls):
51 """ Register the attributes of an OMF application
54 appid = Attribute("appid", "Name of the application")
55 path = Attribute("path", "Path of the application")
56 args = Attribute("args", "Argument of the application")
57 env = Attribute("env", "Environnement variable of the application")
58 stdin = Attribute("stdin", "Input of the application", default = "")
59 cls._register_attribute(appid)
60 cls._register_attribute(path)
61 cls._register_attribute(args)
62 cls._register_attribute(env)
63 cls._register_attribute(stdin)
65 def __init__(self, ec, guid):
67 :param ec: The Experiment controller
68 :type ec: ExperimentController
69 :param guid: guid of the RM
71 :param creds: Credentials to communicate with the rm (XmppClient for OMF)
75 super(OMFApplication, self).__init__(ec, guid)
94 rm_list = self.get_connected(OMFNode.rtype())
95 if rm_list: return rm_list[0]
98 def stdin_hook(self, old_value, new_value):
99 self._omf_api.send_stdin(self.node.get('hostname'), new_value, self.get('appid'))
102 def add_set_hook(self):
103 attr = self._attrs["stdin"]
104 attr.set_hook = self.stdin_hook
106 def valid_connection(self, guid):
107 """ Check if the connection with the guid in parameter is possible.
108 Only meaningful connections are allowed.
110 :param guid: Guid of RM it will be connected
115 rm = self.ec.get_resource(guid)
116 if rm.rtype() not in self._authorized_connections:
117 msg = ("Connection between %s %s and %s %s refused: "
118 "An Application can be connected only to a Node" ) % \
119 (self.rtype(), self._guid, rm.rtype(), guid)
124 elif len(self.connections) != 0 :
125 msg = ("Connection between %s %s and %s %s refused: "
126 "This Application is already connected" ) % \
127 (self.rtype(), self._guid, rm.rtype(), guid)
133 msg = "Connection between %s %s and %s %s accepted" % (
134 self.rtype(), self._guid, rm.rtype(), guid)
141 """ Deploy the RM. It means nothing special for an application
142 for now (later it will be upload sources, ...)
143 It becomes DEPLOYED after getting the xmpp client.
146 if not self._omf_api :
147 self._omf_api = OMFAPIFactory.get_api(self.get('xmppSlice'),
148 self.get('xmppHost'), self.get('xmppPort'),
149 self.get('xmppPassword'), exp_id = self.exp_id)
151 if not self._omf_api :
152 msg = "Credentials are not initialzed. XMPP Connections impossible"
154 raise RuntimeError, msg
156 super(OMFApplication, self).deploy()
160 """ Start the RM. It means : Send Xmpp Message Using OMF protocol
161 to execute the application.
162 It becomes STARTED before the messages are sent (for coordination)
165 if not (self.get('appid') and self.get('path')) :
166 msg = "Application's information are not initialized"
168 raise RuntimeError, msg
170 if not self.get('args'):
171 self.set('args', " ")
172 if not self.get('env'):
175 # Some information to check the information in parameter
176 msg = " " + self.rtype() + " ( Guid : " + str(self._guid) +") : " + \
177 self.get('appid') + " : " + self.get('path') + " : " + \
178 self.get('args') + " : " + self.get('env')
182 self._omf_api.execute(self.node.get('hostname'),self.get('appid'), \
183 self.get('args'), self.get('path'), self.get('env'))
184 except AttributeError:
185 msg = "Credentials are not initialzed. XMPP Connections impossible"
189 super(OMFApplication, self).start()
193 """ Stop the RM. It means : Send Xmpp Message Using OMF protocol to
194 kill the application.
195 State is set to STOPPED after the message is sent.
199 self._omf_api.exit(self.node.get('hostname'),self.get('appid'))
200 except AttributeError:
201 msg = "Credentials were not initialzed. XMPP Connections impossible"
205 super(OMFApplication, self).stop()
209 """ Clean the RM at the end of the experiment and release the API.
214 OMFAPIFactory.release_api(self.get('xmppSlice'),
215 self.get('xmppHost'), self.get('xmppPort'),
216 self.get('xmppPassword'), exp_id = self.exp_id)
219 err = traceback.format_exc()
222 super(OMFApplication, self).release()