123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333 |
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_testbed_service.h"
- /**
- * Generic logging shorthand
- */
- #define LOG(kind,...) \
- GNUNET_log (kind, __VA_ARGS__)
- /**
- * Debug logging shorthand
- */
- #define LOG_DEBUG(...) \
- LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
- /**
- * Global result
- */
- static int ret;
- /**
- * The child process we spawn
- */
- static struct GNUNET_OS_Process *child;
- /**
- * The arguments including the binary to spawn
- */
- static char **argv2;
- /**
- * Pipe used to communicate shutdown via signal.
- */
- static struct GNUNET_DISK_PipeHandle *sigpipe;
- /**
- * Filename of the unique file
- */
- static char *fn;
- /**
- * Handle to the unique file
- */
- static int fh;
- /**
- * The return code of the binary
- */
- static unsigned long child_exit_code;
- /**
- * The process status of the child
- */
- static enum GNUNET_OS_ProcessStatusType child_status;
- /**
- * The shutdown task
- */
- static GNUNET_SCHEDULER_TaskIdentifier shutdown_task_id;
- /**
- * Task to kill the child
- */
- static GNUNET_SCHEDULER_TaskIdentifier terminate_task_id;
- /**
- * Task to kill the child
- */
- static GNUNET_SCHEDULER_TaskIdentifier child_death_task_id;
- /**
- * The shutdown task
- */
- static void
- shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- shutdown_task_id = GNUNET_SCHEDULER_NO_TASK;
- if (0 != child_exit_code)
- {
- LOG (GNUNET_ERROR_TYPE_WARNING, "Child exited with error code: %lu\n",
- child_exit_code);
- ret = 128 + (int) child_exit_code;
- }
- if (0 != fh)
- {
- close (fh);
- }
- if ((NULL != fn) && (0 != unlink (fn)))
- {
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "open");
- ret = GNUNET_SYSERR;
- }
- }
- static void
- terminate_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- static int hard_kill;
- GNUNET_assert (NULL != child);
- terminate_task_id =
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
- &terminate_task, NULL);
- if (0 != hard_kill)
- {
- switch (hard_kill)
- {
- case 1:
- case 2:
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "%d more interrupts needed to send SIGKILL to the child\n",
- 3 - hard_kill);
- hard_kill++;
- return;
- case 3:
- GNUNET_break (0 == GNUNET_OS_process_kill (child, SIGKILL));
- return;
- }
- }
- hard_kill++;
- GNUNET_break (0 == GNUNET_OS_process_kill (child, GNUNET_TERM_SIG));
- LOG (GNUNET_ERROR_TYPE_INFO, _("Waiting for child to exit.\n"));
- }
- /**
- * Task triggered whenever we receive a SIGCHLD (child
- * process died).
- *
- * @param cls closure, NULL if we need to self-restart
- * @param tc context
- */
- static void
- child_death_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- const struct GNUNET_DISK_FileHandle *pr;
- char c[16];
- pr = GNUNET_DISK_pipe_handle (sigpipe, GNUNET_DISK_PIPE_END_READ);
- child_death_task_id = GNUNET_SCHEDULER_NO_TASK;
- if (0 == (tc->reason & GNUNET_SCHEDULER_REASON_READ_READY))
- {
- child_death_task_id =
- GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
- pr, &child_death_task, NULL);
- return;
- }
- /* consume the signal */
- GNUNET_break (0 < GNUNET_DISK_file_read (pr, &c, sizeof (c)));
- LOG_DEBUG ("Child died\n");
- GNUNET_SCHEDULER_cancel (terminate_task_id);
- terminate_task_id = GNUNET_SCHEDULER_NO_TASK;
- GNUNET_assert (GNUNET_OK == GNUNET_OS_process_status (child, &child_status,
- &child_exit_code));
- GNUNET_OS_process_destroy (child);
- child = NULL;
- shutdown_task_id = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
- }
- static void
- destroy_hosts(struct GNUNET_TESTBED_Host **hosts, unsigned int nhosts)
- {
- unsigned int host;
- GNUNET_assert (NULL != hosts);
- for (host = 0; host < nhosts; host++)
- if (NULL != hosts[host])
- GNUNET_TESTBED_host_destroy (hosts[host]);
- GNUNET_free (hosts);
- hosts = NULL;
- }
- /**
- * The main scheduler run task
- *
- * @param cls NULL
- * @param tc scheduler task context
- */
- static void
- run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- struct GNUNET_TESTBED_Host **hosts;
- const struct GNUNET_CONFIGURATION_Handle *null_cfg;
- char *tmpdir;
- char *hostname;
- size_t hostname_len;
- unsigned int nhosts;
- null_cfg = GNUNET_CONFIGURATION_create ();
- nhosts = GNUNET_TESTBED_hosts_load_from_loadleveler (null_cfg, &hosts);
- if (0 == nhosts)
- {
- GNUNET_break (0);
- ret = GNUNET_SYSERR;
- return;
- }
- hostname_len = GNUNET_OS_get_hostname_max_length ();
- hostname = GNUNET_malloc (hostname_len);
- if (0 != gethostname (hostname, hostname_len))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot get hostname. Exiting\n");
- GNUNET_free (hostname);
- destroy_hosts (hosts, nhosts);
- ret = GNUNET_SYSERR;
- return;
- }
- if (NULL == strstr (GNUNET_TESTBED_host_get_hostname (hosts[0]), hostname))
- {
- LOG_DEBUG ("Exiting as `%s' is not the lowest host\n", hostname);
- GNUNET_free (hostname);
- ret = GNUNET_OK;
- return;
- }
- LOG_DEBUG ("Will be executing `%s' on host `%s'\n", argv2[0], hostname);
- GNUNET_free (hostname);
- destroy_hosts (hosts, nhosts);
- tmpdir = getenv ("TMPDIR");
- if (NULL == tmpdir)
- tmpdir = getenv ("TMP");
- if (NULL == tmpdir)
- tmpdir = getenv ("TEMP");
- if (NULL == tmpdir)
- tmpdir = "/tmp";
- (void) GNUNET_asprintf (&fn, "%s/gnunet-testbed-spawn.lock", tmpdir);
- /* Open the unique file; we can create it then we can spawn the child process
- else we exit */
- fh = open (fn, O_CREAT | O_EXCL | O_CLOEXEC,
- S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
- if (-1 == fh)
- {
- if (EEXIST == errno)
- {
- LOG_DEBUG ("Lock file already created by other process. Exiting\n");
- ret = GNUNET_OK;
- return;
- }
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "open");
- ret = GNUNET_SYSERR;
- return;
- }
- /* Spawn the new process here */
- LOG (GNUNET_ERROR_TYPE_INFO, _("Spawning process `%s'\n"), argv2[0]);
- child = GNUNET_OS_start_process_vap (GNUNET_NO, GNUNET_OS_INHERIT_STD_ALL, NULL,
- NULL, NULL,
- argv2[0], argv2);
- if (NULL == child)
- {
- GNUNET_break (0);
- ret = GNUNET_SYSERR;
- shutdown_task_id = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
- return;
- }
- ret = GNUNET_OK;
- terminate_task_id =
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
- &terminate_task, NULL);
- child_death_task_id =
- GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
- GNUNET_DISK_pipe_handle (sigpipe,
- GNUNET_DISK_PIPE_END_READ),
- &child_death_task, NULL);
- }
- /**
- * Signal handler called for SIGCHLD.
- */
- static void
- sighandler_child_death ()
- {
- static char c;
- int old_errno = errno; /* back-up errno */
- GNUNET_break (1 ==
- GNUNET_DISK_file_write (GNUNET_DISK_pipe_handle
- (sigpipe, GNUNET_DISK_PIPE_END_WRITE),
- &c, sizeof (c)));
- errno = old_errno; /* restore errno */
- }
- /**
- * Execution start point
- */
- int
- main (int argc, char *argv[])
- {
- struct GNUNET_SIGNAL_Context *shc_chld;
- unsigned int cnt;
- ret = -1;
- if (argc < 2)
- {
- printf ("Need arguments: gnunet-testbed-mpi-spawn <cmd> <cmd_args>");
- return 1;
- }
- if (GNUNET_OK != GNUNET_log_setup ("gnunet-testbed-spawn", NULL, NULL))
- {
- GNUNET_break (0);
- return 1;
- }
- if (NULL == (sigpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO,
- GNUNET_NO, GNUNET_NO)))
- {
- GNUNET_break (0);
- ret = GNUNET_SYSERR;
- return 1;
- }
- shc_chld =
- GNUNET_SIGNAL_handler_install (GNUNET_SIGCHLD, &sighandler_child_death);
- if (NULL == shc_chld)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot install a signal handler\n");
- return 1;
- }
- argv2 = GNUNET_malloc (sizeof (char *) * argc);
- for (cnt = 1; cnt < argc; cnt++)
- argv2[cnt - 1] = argv[cnt];
- GNUNET_SCHEDULER_run (run, NULL);
- GNUNET_free (argv2);
- GNUNET_SIGNAL_handler_uninstall (shc_chld);
- shc_chld = NULL;
- GNUNET_DISK_pipe_close (sigpipe);
- GNUNET_free_non_null (fn);
- if (GNUNET_OK != ret)
- return ret;
- return 0;
- }
|