aboutsummaryrefslogtreecommitdiff
path: root/src/testbed/gnunet_testbed_mpi_spawn.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/testbed/gnunet_testbed_mpi_spawn.c')
-rw-r--r--src/testbed/gnunet_testbed_mpi_spawn.c327
1 files changed, 0 insertions, 327 deletions
diff --git a/src/testbed/gnunet_testbed_mpi_spawn.c b/src/testbed/gnunet_testbed_mpi_spawn.c
deleted file mode 100644
index 577ffbeba..000000000
--- a/src/testbed/gnunet_testbed_mpi_spawn.c
+++ /dev/null
@@ -1,327 +0,0 @@
1#include "platform.h"
2#include "gnunet_util_lib.h"
3#include "gnunet_testbed_service.h"
4
5
6/**
7 * Generic logging shorthand
8 */
9#define LOG(kind, ...) \
10 GNUNET_log (kind, __VA_ARGS__)
11
12/**
13 * Debug logging shorthand
14 */
15#define LOG_DEBUG(...) \
16 LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
17
18/**
19 * Global result
20 */
21static int ret;
22
23/**
24 * The child process we spawn
25 */
26static struct GNUNET_OS_Process *child;
27
28/**
29 * The arguments including the binary to spawn
30 */
31static char **argv2;
32
33/**
34 * Pipe used to communicate shutdown via signal.
35 */
36static struct GNUNET_DISK_PipeHandle *sigpipe;
37
38/**
39 * Filename of the unique file
40 */
41static char *fn;
42
43/**
44 * Handle to the unique file
45 */
46static int fh;
47
48/**
49 * The return code of the binary
50 */
51static unsigned long child_exit_code;
52
53/**
54 * The process status of the child
55 */
56static enum GNUNET_OS_ProcessStatusType child_status;
57
58/**
59 * Task to kill the child
60 */
61static struct GNUNET_SCHEDULER_Task *terminate_task_id;
62
63/**
64 * Task to kill the child
65 */
66static struct GNUNET_SCHEDULER_Task *child_death_task_id;
67
68/**
69 * The shutdown task
70 */
71static void
72shutdown_task (void *cls)
73{
74 if (0 != child_exit_code)
75 {
76 LOG (GNUNET_ERROR_TYPE_WARNING, "Child exited with error code: %lu\n",
77 child_exit_code);
78 ret = 128 + (int) child_exit_code;
79 }
80 if (0 != fh)
81 {
82 close (fh);
83 }
84 if ((NULL != fn) && (0 != unlink (fn)))
85 {
86 GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "open");
87 ret = GNUNET_SYSERR;
88 }
89}
90
91
92static void
93terminate_task (void *cls)
94{
95 static int hard_kill;
96
97 GNUNET_assert (NULL != child);
98 terminate_task_id =
99 GNUNET_SCHEDULER_add_shutdown (&terminate_task, NULL);
100 if (0 != hard_kill)
101 {
102 switch (hard_kill)
103 {
104 case 1:
105 case 2:
106 LOG (GNUNET_ERROR_TYPE_WARNING,
107 "%d more interrupts needed to send SIGKILL to the child\n",
108 3 - hard_kill);
109 hard_kill++;
110 return;
111
112 case 3:
113 GNUNET_break (0 == GNUNET_OS_process_kill (child, SIGKILL));
114 return;
115 }
116 }
117 hard_kill++;
118 GNUNET_break (0 == GNUNET_OS_process_kill (child, GNUNET_TERM_SIG));
119 LOG (GNUNET_ERROR_TYPE_INFO, _ ("Waiting for child to exit.\n"));
120}
121
122
123/**
124 * Task triggered whenever we receive a SIGCHLD (child
125 * process died).
126 *
127 * @param cls closure, NULL if we need to self-restart
128 */
129static void
130child_death_task (void *cls)
131{
132 const struct GNUNET_DISK_FileHandle *pr;
133 char c[16];
134 const struct GNUNET_SCHEDULER_TaskContext *tc;
135
136
137 pr = GNUNET_DISK_pipe_handle (sigpipe, GNUNET_DISK_PIPE_END_READ);
138 child_death_task_id = NULL;
139 tc = GNUNET_SCHEDULER_get_task_context ();
140 if (0 == (tc->reason & GNUNET_SCHEDULER_REASON_READ_READY))
141 {
142 child_death_task_id =
143 GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
144 pr, &child_death_task, NULL);
145 return;
146 }
147 /* consume the signal */
148 GNUNET_break (0 < GNUNET_DISK_file_read (pr, &c, sizeof(c)));
149 LOG_DEBUG ("Child died\n");
150 GNUNET_SCHEDULER_cancel (terminate_task_id);
151 terminate_task_id = NULL;
152 GNUNET_assert (GNUNET_OK == GNUNET_OS_process_status (child, &child_status,
153 &child_exit_code));
154 GNUNET_OS_process_destroy (child);
155 child = NULL;
156 GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
157}
158
159
160static void
161destroy_hosts (struct GNUNET_TESTBED_Host **hosts, unsigned int nhosts)
162{
163 unsigned int host;
164
165 GNUNET_assert (NULL != hosts);
166 for (host = 0; host < nhosts; host++)
167 if (NULL != hosts[host])
168 GNUNET_TESTBED_host_destroy (hosts[host]);
169 GNUNET_free (hosts);
170 hosts = NULL;
171}
172
173
174/**
175 * The main scheduler run task
176 *
177 * @param cls NULL
178 */
179static void
180run (void *cls)
181{
182 struct GNUNET_TESTBED_Host **hosts;
183 const struct GNUNET_CONFIGURATION_Handle *null_cfg;
184 char *tmpdir;
185 char *hostname;
186 size_t hostname_len;
187 unsigned int nhosts;
188
189 null_cfg = GNUNET_CONFIGURATION_create ();
190 nhosts = GNUNET_TESTBED_hosts_load_from_loadleveler (null_cfg, &hosts);
191 if (0 == nhosts)
192 {
193 GNUNET_break (0);
194 ret = GNUNET_SYSERR;
195 return;
196 }
197 hostname_len = GNUNET_OS_get_hostname_max_length ();
198 hostname = GNUNET_malloc (hostname_len);
199 if (0 != gethostname (hostname, hostname_len))
200 {
201 LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot get hostname. Exiting\n");
202 GNUNET_free (hostname);
203 destroy_hosts (hosts, nhosts);
204 ret = GNUNET_SYSERR;
205 return;
206 }
207 if (NULL == strstr (GNUNET_TESTBED_host_get_hostname (hosts[0]), hostname))
208 {
209 LOG_DEBUG ("Exiting as `%s' is not the lowest host\n", hostname);
210 GNUNET_free (hostname);
211 ret = GNUNET_OK;
212 return;
213 }
214 LOG_DEBUG ("Will be executing `%s' on host `%s'\n", argv2[0], hostname);
215 GNUNET_free (hostname);
216 destroy_hosts (hosts, nhosts);
217 tmpdir = getenv ("TMPDIR");
218 if (NULL == tmpdir)
219 tmpdir = getenv ("TMP");
220 if (NULL == tmpdir)
221 tmpdir = getenv ("TEMP");
222 if (NULL == tmpdir)
223 tmpdir = "/tmp";
224 (void) GNUNET_asprintf (&fn, "%s/gnunet-testbed-spawn.lock", tmpdir);
225 /* Open the unique file; we can create it then we can spawn the child process
226 else we exit */
227 fh = open (fn, O_CREAT | O_EXCL | O_CLOEXEC,
228 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
229 if (-1 == fh)
230 {
231 if (EEXIST == errno)
232 {
233 LOG_DEBUG ("Lock file already created by other process. Exiting\n");
234 ret = GNUNET_OK;
235 return;
236 }
237 GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "open");
238 ret = GNUNET_SYSERR;
239 return;
240 }
241 /* Spawn the new process here */
242 LOG (GNUNET_ERROR_TYPE_INFO, _ ("Spawning process `%s'\n"), argv2[0]);
243 child = GNUNET_OS_start_process_vap (GNUNET_NO, GNUNET_OS_INHERIT_STD_ALL,
244 NULL,
245 NULL, NULL,
246 argv2[0], argv2);
247 if (NULL == child)
248 {
249 GNUNET_break (0);
250 ret = GNUNET_SYSERR;
251 GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
252 return;
253 }
254 ret = GNUNET_OK;
255 terminate_task_id =
256 GNUNET_SCHEDULER_add_shutdown (&terminate_task, NULL);
257 child_death_task_id =
258 GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
259 GNUNET_DISK_pipe_handle (sigpipe,
260 GNUNET_DISK_PIPE_END_READ),
261 &child_death_task, NULL);
262}
263
264
265/**
266 * Signal handler called for SIGCHLD.
267 */
268static void
269sighandler_child_death ()
270{
271 static char c;
272 int old_errno = errno; /* back-up errno */
273
274 GNUNET_break (1 ==
275 GNUNET_DISK_file_write (GNUNET_DISK_pipe_handle
276 (sigpipe, GNUNET_DISK_PIPE_END_WRITE),
277 &c, sizeof(c)));
278 errno = old_errno; /* restore errno */
279}
280
281
282/**
283 * Execution start point
284 */
285int
286main (int argc, char *argv[])
287{
288 struct GNUNET_SIGNAL_Context *shc_chld;
289 unsigned int cnt;
290
291 ret = -1;
292 if (argc < 2)
293 {
294 printf ("Need arguments: gnunet-testbed-mpi-spawn <cmd> <cmd_args>");
295 return 1;
296 }
297 if (GNUNET_OK != GNUNET_log_setup ("gnunet-testbed-spawn", NULL, NULL))
298 {
299 GNUNET_break (0);
300 return 1;
301 }
302 if (NULL == (sigpipe = GNUNET_DISK_pipe (GNUNET_DISK_PF_NONE)))
303 {
304 GNUNET_break (0);
305 ret = GNUNET_SYSERR;
306 return 1;
307 }
308 shc_chld =
309 GNUNET_SIGNAL_handler_install (GNUNET_SIGCHLD, &sighandler_child_death);
310 if (NULL == shc_chld)
311 {
312 LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot install a signal handler\n");
313 return 1;
314 }
315 argv2 = GNUNET_malloc (sizeof(char *) * argc);
316 for (cnt = 1; cnt < argc; cnt++)
317 argv2[cnt - 1] = argv[cnt];
318 GNUNET_SCHEDULER_run (run, NULL);
319 GNUNET_free (argv2);
320 GNUNET_SIGNAL_handler_uninstall (shc_chld);
321 shc_chld = NULL;
322 GNUNET_DISK_pipe_close (sigpipe);
323 GNUNET_free (fn);
324 if (GNUNET_OK != ret)
325 return ret;
326 return 0;
327}