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.trace import Trace, TraceAttr
22 from nepi.execution.resource import ResourceManager, clsinit_copy, ResourceState, \
24 from nepi.resources.linux.application import LinuxApplication
25 from nepi.resources.linux.node import OSType
26 from nepi.util.timefuncs import tnow, tdiffsec
30 # TODO: use ccndlogging to dynamically change the logging level
34 class LinuxCCND(LinuxApplication):
38 def _register_attributes(cls):
39 debug = Attribute("debug", "Sets the CCND_DEBUG environmental variable. "
40 " Allowed values are : \n"
42 " 1 - basic messages (any non-zero value gets these) \n"
43 " 2 - interest messages \n"
44 " 4 - content messages \n"
45 " 8 - matching details \n"
46 " 16 - interest details \n"
47 " 32 - gory interest details \n"
48 " 64 - log occasional human-readable timestamps \n"
49 " 128 - face registration debugging \n"
50 " -1 - max logging \n"
51 " Or apply bitwise OR to these values to get combinations of them",
53 flags = Flags.ExecReadOnly)
55 port = Attribute("port", "Sets the CCN_LOCAL_PORT environmental variable. "
57 flags = Flags.ExecReadOnly)
59 sockname = Attribute("sockname",
60 "Sets the CCN_LOCAL_SCOKNAME environmental variable. "
61 "Defaults to /tmp/.ccnd.sock",
62 flags = Flags.ExecReadOnly)
64 capacity = Attribute("capacity",
65 "Sets the CCND_CAP environmental variable. "
66 "Capacity limit in terms of ContentObjects",
67 flags = Flags.ExecReadOnly)
69 mtu = Attribute("mtu", "Sets the CCND_MTU environmental variable. ",
70 flags = Flags.ExecReadOnly)
72 data_pause = Attribute("dataPauseMicrosec",
73 "Sets the CCND_DATA_PAUSE_MICROSEC environmental variable. ",
74 flags = Flags.ExecReadOnly)
76 default_stale = Attribute("defaultTimeToStale",
77 "Sets the CCND_DEFAULT_TIME_TO_STALE environmental variable. ",
78 flags = Flags.ExecReadOnly)
80 max_stale = Attribute("maxTimeToStale",
81 "Sets the CCND_MAX_TIME_TO_STALE environmental variable. ",
82 flags = Flags.ExecReadOnly)
84 max_rte = Attribute("maxRteMicrosec",
85 "Sets the CCND_MAX_RTE_MICROSEC environmental variable. ",
86 flags = Flags.ExecReadOnly)
88 keystore = Attribute("keyStoreDirectory",
89 "Sets the CCND_KEYSTORE_DIRECTORY environmental variable. ",
90 flags = Flags.ExecReadOnly)
92 listen_on = Attribute("listenOn",
93 "Sets the CCND_LISTEN_ON environmental variable. ",
94 flags = Flags.ExecReadOnly)
96 autoreg = Attribute("autoreg",
97 "Sets the CCND_AUTOREG environmental variable. ",
98 flags = Flags.ExecReadOnly)
100 prefix = Attribute("prefix",
101 "Sets the CCND_PREFIX environmental variable. ",
102 flags = Flags.ExecReadOnly)
104 cls._register_attribute(debug)
105 cls._register_attribute(port)
106 cls._register_attribute(sockname)
107 cls._register_attribute(capacity)
108 cls._register_attribute(mtu)
109 cls._register_attribute(data_pause)
110 cls._register_attribute(default_stale)
111 cls._register_attribute(max_stale)
112 cls._register_attribute(max_rte)
113 cls._register_attribute(keystore)
114 cls._register_attribute(listen_on)
115 cls._register_attribute(autoreg)
116 cls._register_attribute(prefix)
119 def _register_traces(cls):
120 log = Trace("log", "CCND log output")
121 status = Trace("status", "ccndstatus output")
123 cls._register_trace(log)
124 cls._register_trace(status)
126 def __init__(self, ec, guid):
127 super(LinuxCCND, self).__init__(ec, guid)
128 self._home = "ccnd-%s" % self.guid
129 self._version = "ccnx"
137 return "PATH=$PATH:${BIN}/%s/" % self.version
140 if not self.node or self.node.state < ResourceState.READY:
141 self.debug("---- RESCHEDULING DEPLOY ---- node state %s " % self.node.state )
143 # ccnd needs to wait until node is deployed and running
144 self.ec.schedule(reschedule_delay, self.deploy)
146 if not self.get("command"):
147 self.set("command", self._start_command)
149 if not self.get("depends"):
150 self.set("depends", self._dependencies)
152 if not self.get("sources"):
153 self.set("sources", self._sources)
155 sources = self.get("sources")
156 source = sources.split(" ")[0]
157 basename = os.path.basename(source)
158 self._version = ( basename.strip().replace(".tar.gz", "")
161 .replace(".zip","") )
163 if not self.get("build"):
164 self.set("build", self._build)
166 if not self.get("install"):
167 self.set("install", self._install)
169 if not self.get("env"):
170 self.set("env", self._environment)
172 command = self.get("command")
173 env = self.get("env")
175 self.info("Deploying command '%s' " % command)
177 # create home dir for application
178 self.node.mkdir(self.run_home)
181 self.upload_sources()
189 # install dependencies
190 self.install_dependencies()
198 # We want to make sure the repository is running
199 # before the experiment starts.
200 # Run the command as a bash script in background,
201 # in the host ( but wait until the command has
202 # finished to continue )
203 env = self.replace_paths(env)
204 command = self.replace_paths(command)
206 shfile = os.path.join(self.app_home, "start.sh")
207 self.node.run_and_wait(command, self.run_home,
211 raise_on_error = True)
213 self.debug("----- READY ---- ")
214 self._ready_time = tnow()
215 self._state = ResourceState.READY
218 if self._state == ResourceState.READY:
219 command = self.get("command")
220 self.info("Starting command '%s'" % command)
222 self._start_time = tnow()
223 self._state = ResourceState.STARTED
225 msg = " Failed to execute command '%s'" % command
226 self.error(msg, out, err)
227 self._state = ResourceState.FAILED
228 raise RuntimeError, msg
231 command = self.get('command') or ''
234 if state == ResourceState.STARTED:
235 self.info("Stopping command '%s'" % command)
238 env = self.get("env")
240 # replace application specific paths in the command
241 command = self.replace_paths(command)
242 env = env and self.replace_paths(env)
244 # Upload the command to a file, and execute asynchronously
245 shfile = os.path.join(self.app_home, "stop.sh")
246 self.node.run_and_wait(command, self.run_home,
250 pidfile = "ccndstop_pidfile",
251 ecodefile = "ccndstop_exitcode",
252 stdout = "ccndstop_stdout",
253 stderr = "ccndstop_stderr")
255 self._stop_time = tnow()
256 self._state = ResourceState.STOPPED
260 # First check if the ccnd has failed
261 state_check_delay = 0.5
262 if self._state == ResourceState.STARTED and \
263 tdiffsec(tnow(), self._last_state_check) > state_check_delay:
264 (out, err), proc = self._ccndstatus
266 retcode = proc.poll()
268 if retcode == 1 and err.find("No such file or directory") > -1:
269 # ccnd is not running (socket not found)
270 self._state = ResourceState.FINISHED
273 msg = " Failed to execute command '%s'" % self.get("command")
274 self.error(msg, out, err)
275 self._state = ResourceState.FAILED
277 self._last_state_check = tnow()
282 def _ccndstatus(self):
283 env = self.get('env') or ""
284 environ = self.node.format_environment(env, inline = True)
285 command = environ + " ccndstatus"
286 command = self.replace_paths(command)
288 return self.node.execute(command)
291 def _start_command(self):
295 def _dependencies(self):
296 if self.node.use_rpm:
297 return ( " autoconf openssl-devel expat-devel libpcap-devel "
298 " ecryptfs-utils-devel libxml2-devel automake gawk "
299 " gcc gcc-c++ git pcre-devel make ")
300 elif self.node.use_deb:
301 return ( " autoconf libssl-dev libexpat-dev libpcap-dev "
302 " libecryptfs0 libxml2-utils automake gawk gcc g++ "
303 " git-core pkg-config libpcre3-dev make ")
308 return "http://www.ccnx.org/releases/ccnx-0.7.2.tar.gz"
312 sources = self.get("sources").split(" ")[0]
313 sources = os.path.basename(sources)
316 # Evaluate if ccnx binaries are already installed
318 " test -f ${BIN}/%(version)s/ccnd && "
319 " echo 'binaries found, nothing to do' "
321 # If not, untar and build
323 " mkdir -p ${SRC}/%(version)s && "
324 " tar xf ${SRC}/%(sources)s --strip-components=1 -C ${SRC}/%(version)s "
326 "cd ${SRC}/%(version)s && "
327 # Just execute and silence warnings...
328 " ( ./configure && make ) "
329 " )") % ({ 'sources': sources,
330 'version': self.version
336 # Evaluate if ccnx binaries are already installed
338 " test -f ${BIN}/%(version)s/ccnd && "
339 " echo 'binaries found, nothing to do' "
342 " mkdir -p ${BIN}/%(version)s && "
343 " mv ${SRC}/%(version)s/bin/* ${BIN}/%(version)s/ "
345 ) % ({ 'version': self.version
349 def _environment(self):
351 "debug": "CCND_DEBUG",
352 "port": "CCN_LOCAL_PORT",
353 "sockname" : "CCN_LOCAL_SOCKNAME",
354 "capacity" : "CCND_CAP",
356 "dataPauseMicrosec" : "CCND_DATA_PAUSE_MICROSEC",
357 "defaultTimeToStale" : "CCND_DEFAULT_TIME_TO_STALE",
358 "maxTimeToStale" : "CCND_MAX_TIME_TO_STALE",
359 "maxRteMicrosec" : "CCND_MAX_RTE_MICROSEC",
360 "keyStoreDirectory" : "CCND_KEYSTORE_DIRECTORY",
361 "listenOn" : "CCND_LISTEN_ON",
362 "autoreg" : "CCND_AUTOREG",
363 "prefix" : "CCND_PREFIX",
367 env += " ".join(map(lambda k: "%s=%s" % (envs.get(k), str(self.get(k))) \
368 if self.get(k) else "", envs.keys()))
372 def valid_connection(self, guid):