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>
20 from nepi.execution.attribute import Attribute, Flags, Types
21 from nepi.execution.resource import clsinit_copy, ResourceState, \
22 ResourceAction, reschedule_delay
23 from nepi.resources.linux.application import LinuxApplication
24 from nepi.resources.linux.ccn.ccnr import LinuxCCNR
25 from nepi.util.timefuncs import tnow
30 class LinuxCCNContent(LinuxApplication):
31 _rtype = "LinuxCCNContent"
34 def _register_attributes(cls):
35 content_name = Attribute("contentName",
36 "The name of the content to publish (e.g. ccn:/VIDEO) ",
37 flags = Flags.ExecReadOnly)
39 content = Attribute("content",
40 "The content to publish. It can be a path to a file or plain text ",
41 flags = Flags.ExecReadOnly)
43 cls._register_attribute(content_name)
44 cls._register_attribute(content)
46 def __init__(self, ec, guid):
47 super(LinuxCCNContent, self).__init__(ec, guid)
48 self._home = "content-%s" % self.guid
52 ccnr = self.get_connected(LinuxCCNR.rtype())
53 if ccnr: return ccnr[0]
58 if self.ccnr: return self.ccnr.ccnd
63 if self.ccnr: return self.ccnr.node
67 if not self.ccnr or self.ccnr.state < ResourceState.READY:
68 self.debug("---- RESCHEDULING DEPLOY ---- node state %s " % self.node.state )
70 # ccnr needs to wait until ccnd is deployed and running
71 self.ec.schedule(reschedule_delay, self.deploy)
74 if not self.get("command"):
75 self.set("command", self._start_command)
77 if not self.get("env"):
78 self.set("env", self._environment)
80 # set content to stdin, so the content will be
81 # uploaded during provision
82 self.set("stdin", self.get("content"))
84 command = self.get("command")
86 self.info("Deploying command '%s' " % command)
94 self.debug("----- READY ---- ")
95 self._ready_time = tnow()
96 self._state = ResourceState.READY
98 def upload_start_command(self):
99 command = self.get("command")
100 env = self.get("env")
103 # We want to make sure the content is published
104 # before the experiment starts.
105 # Run the command as a bash script in the background,
106 # in the host ( but wait until the command has
107 # finished to continue )
108 env = self.replace_paths(env)
109 command = self.replace_paths(command)
111 (out, err), proc = self.execute_command(command, env)
115 msg = "Failed to execute command"
116 self.error(msg, out, err)
117 raise RuntimeError, msg
120 if self._state == ResourceState.READY:
121 command = self.get("command")
122 self.info("Starting command '%s'" % command)
124 self._start_time = tnow()
125 self._state = ResourceState.STARTED
127 msg = " Failed to execute command '%s'" % command
128 self.error(msg, out, err)
129 self._state = ResourceState.FAILED
130 raise RuntimeError, msg
137 def _start_command(self):
138 return "ccnseqwriter -r %s < %s" % (self.get("contentName"),
139 os.path.join(self.app_home, 'stdin'))
142 def _environment(self):
143 return self.ccnd.path
145 def execute_command(self, command, env):
146 environ = self.node.format_environment(env, inline = True)
147 command = environ + command
148 command = self.replace_paths(command)
150 return self.node.execute(command)
152 def valid_connection(self, guid):