GNUnet  0.10.x
gnunet_testbed_mpi_spawn.c
Go to the documentation of this file.
1 #include "platform.h"
2 #include "gnunet_util_lib.h"
4 
5 
9 #define LOG(kind,...) \
10  GNUNET_log (kind, __VA_ARGS__)
11 
15 #define LOG_DEBUG(...) \
16  LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
17 
21 static int ret;
22 
26 static struct GNUNET_OS_Process *child;
27 
31 static char **argv2;
32 
37 
41 static char *fn;
42 
46 static int fh;
47 
51 static unsigned long child_exit_code;
52 
57 
62 
67 
71 static void
72 shutdown_task (void *cls)
73 {
74  if (0 != child_exit_code)
75  {
76  LOG (GNUNET_ERROR_TYPE_WARNING, "Child exited with error code: %lu\n",
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  {
88  }
89 }
90 
91 
92 static void
93 terminate_task (void *cls)
94 {
95  static int hard_kill;
96 
97  GNUNET_assert (NULL != child);
98  terminate_task_id =
100  if (0 != hard_kill)
101  {
102  switch (hard_kill)
103  {
104  case 1:
105  case 2:
107  "%d more interrupts needed to send SIGKILL to the child\n",
108  3 - hard_kill);
109  hard_kill++;
110  return;
111  case 3:
112  GNUNET_break (0 == GNUNET_OS_process_kill (child, SIGKILL));
113  return;
114  }
115  }
116  hard_kill++;
118  LOG (GNUNET_ERROR_TYPE_INFO, _("Waiting for child to exit.\n"));
119 }
120 
121 
128 static void
129 child_death_task (void *cls)
130 {
131  const struct GNUNET_DISK_FileHandle *pr;
132  char c[16];
133  const struct GNUNET_SCHEDULER_TaskContext *tc;
134 
135 
137  child_death_task_id = NULL;
140  {
141  child_death_task_id =
143  pr, &child_death_task, NULL);
144  return;
145  }
146  /* consume the signal */
147  GNUNET_break (0 < GNUNET_DISK_file_read (pr, &c, sizeof (c)));
148  LOG_DEBUG ("Child died\n");
149  GNUNET_SCHEDULER_cancel (terminate_task_id);
150  terminate_task_id = NULL;
152  &child_exit_code));
154  child = NULL;
156 }
157 
158 
159 static void
160 destroy_hosts(struct GNUNET_TESTBED_Host **hosts, unsigned int nhosts)
161 {
162  unsigned int host;
163 
164  GNUNET_assert (NULL != hosts);
165  for (host = 0; host < nhosts; host++)
166  if (NULL != hosts[host])
167  GNUNET_TESTBED_host_destroy (hosts[host]);
168  GNUNET_free (hosts);
169  hosts = NULL;
170 }
171 
172 
178 static void
179 run (void *cls)
180 {
181  struct GNUNET_TESTBED_Host **hosts;
182  const struct GNUNET_CONFIGURATION_Handle *null_cfg;
183  char *tmpdir;
184  char *hostname;
185  size_t hostname_len;
186  unsigned int nhosts;
187 
188  null_cfg = GNUNET_CONFIGURATION_create ();
189  nhosts = GNUNET_TESTBED_hosts_load_from_loadleveler (null_cfg, &hosts);
190  if (0 == nhosts)
191  {
192  GNUNET_break (0);
193  ret = GNUNET_SYSERR;
194  return;
195  }
196  hostname_len = GNUNET_OS_get_hostname_max_length ();
197  hostname = GNUNET_malloc (hostname_len);
198  if (0 != gethostname (hostname, hostname_len))
199  {
200  LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot get hostname. Exiting\n");
201  GNUNET_free (hostname);
202  destroy_hosts (hosts, nhosts);
203  ret = GNUNET_SYSERR;
204  return;
205  }
206  if (NULL == strstr (GNUNET_TESTBED_host_get_hostname (hosts[0]), hostname))
207  {
208  LOG_DEBUG ("Exiting as `%s' is not the lowest host\n", hostname);
209  GNUNET_free (hostname);
210  ret = GNUNET_OK;
211  return;
212  }
213  LOG_DEBUG ("Will be executing `%s' on host `%s'\n", argv2[0], hostname);
214  GNUNET_free (hostname);
215  destroy_hosts (hosts, nhosts);
216  tmpdir = getenv ("TMPDIR");
217  if (NULL == tmpdir)
218  tmpdir = getenv ("TMP");
219  if (NULL == tmpdir)
220  tmpdir = getenv ("TEMP");
221  if (NULL == tmpdir)
222  tmpdir = "/tmp";
223  (void) GNUNET_asprintf (&fn, "%s/gnunet-testbed-spawn.lock", tmpdir);
224  /* Open the unique file; we can create it then we can spawn the child process
225  else we exit */
226  fh = open (fn, O_CREAT | O_EXCL | O_CLOEXEC,
227  S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
228  if (-1 == fh)
229  {
230  if (EEXIST == errno)
231  {
232  LOG_DEBUG ("Lock file already created by other process. Exiting\n");
233  ret = GNUNET_OK;
234  return;
235  }
237  ret = GNUNET_SYSERR;
238  return;
239  }
240  /* Spawn the new process here */
241  LOG (GNUNET_ERROR_TYPE_INFO, _("Spawning process `%s'\n"), argv2[0]);
243  NULL, NULL,
244  argv2[0], argv2);
245  if (NULL == child)
246  {
247  GNUNET_break (0);
248  ret = GNUNET_SYSERR;
250  return;
251  }
252  ret = GNUNET_OK;
253  terminate_task_id =
255  child_death_task_id =
257  GNUNET_DISK_pipe_handle (sigpipe,
259  &child_death_task, NULL);
260 }
261 
262 
266 static void
268 {
269  static char c;
270  int old_errno = errno; /* back-up errno */
271 
272  GNUNET_break (1 ==
274  (sigpipe, GNUNET_DISK_PIPE_END_WRITE),
275  &c, sizeof (c)));
276  errno = old_errno; /* restore errno */
277 }
278 
279 
283 int
284 main (int argc, char *argv[])
285 {
286  struct GNUNET_SIGNAL_Context *shc_chld;
287  unsigned int cnt;
288 
289  ret = -1;
290  if (argc < 2)
291  {
292  printf ("Need arguments: gnunet-testbed-mpi-spawn <cmd> <cmd_args>");
293  return 1;
294  }
295  if (GNUNET_OK != GNUNET_log_setup ("gnunet-testbed-spawn", NULL, NULL))
296  {
297  GNUNET_break (0);
298  return 1;
299  }
300  if (NULL == (sigpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO,
301  GNUNET_NO, GNUNET_NO)))
302  {
303  GNUNET_break (0);
304  ret = GNUNET_SYSERR;
305  return 1;
306  }
307  shc_chld =
309  if (NULL == shc_chld)
310  {
311  LOG (GNUNET_ERROR_TYPE_ERROR, "Cannot install a signal handler\n");
312  return 1;
313  }
314  argv2 = GNUNET_malloc (sizeof (char *) * argc);
315  for (cnt = 1; cnt < argc; cnt++)
316  argv2[cnt - 1] = argv[cnt];
317  GNUNET_SCHEDULER_run (run, NULL);
318  GNUNET_free (argv2);
320  shc_chld = NULL;
321  GNUNET_DISK_pipe_close (sigpipe);
323  if (GNUNET_OK != ret)
324  return ret;
325  return 0;
326 }
const struct GNUNET_SCHEDULER_TaskContext * GNUNET_SCHEDULER_get_task_context(void)
Obtain the reasoning why the current task was started.
Definition: scheduler.c:746
static void terminate_task(void *cls)
void GNUNET_OS_process_destroy(struct GNUNET_OS_Process *proc)
Cleans up process structure contents (OS-dependent) and deallocates it.
Definition: os_priority.c:364
#define GNUNET_SIGCHLD
Definition: platform.h:46
ssize_t GNUNET_DISK_file_read(const struct GNUNET_DISK_FileHandle *h, void *result, size_t len)
Read the contents of a binary file into a buffer.
Definition: disk.c:881
int GNUNET_OS_process_status(struct GNUNET_OS_Process *proc, enum GNUNET_OS_ProcessStatusType *type, unsigned long *code)
Retrieve the status of a process, waiting on it if dead.
Definition: os_priority.c:1719
The reading socket is ready.
Context information passed to each scheduler task.
struct GNUNET_SCHEDULER_Task * GNUNET_SCHEDULER_add_shutdown(GNUNET_SCHEDULER_TaskCallback task, void *task_cls)
Schedule a new task to be run on shutdown, that is when a CTRL-C signal is received, or when GNUNET_SCHEDULER_shutdown() is being invoked.
Definition: scheduler.c:1293
#define GNUNET_assert(cond)
Use this for fatal errors that cannot be handled.
static struct GNUNET_SCHEDULER_TaskContext tc
Task context of the current task.
Definition: scheduler.c:417
void GNUNET_SCHEDULER_run(GNUNET_SCHEDULER_TaskCallback task, void *task_cls)
Initialize and run scheduler.
Definition: scheduler.c:716
#define GNUNET_NO
Definition: gnunet_common.h:81
struct GNUNET_SCHEDULER_Task * GNUNET_SCHEDULER_add_read_file(struct GNUNET_TIME_Relative delay, const struct GNUNET_DISK_FileHandle *rfd, GNUNET_SCHEDULER_TaskCallback task, void *task_cls)
Schedule a new task to be run with a specified delay or when the specified file descriptor is ready f...
Definition: scheduler.c:1643
#define GNUNET_OK
Named constants for return values.
Definition: gnunet_common.h:78
void GNUNET_SIGNAL_handler_uninstall(struct GNUNET_SIGNAL_Context *ctx)
Uninstall a previously installed signal hander.
Definition: signal.c:99
#define GNUNET_free_non_null(ptr)
Free the memory pointed to by ptr if ptr is not NULL.
struct GNUNET_CONFIGURATION_Handle * GNUNET_CONFIGURATION_create(void)
Create a new configuration object.
Opaque handle to a host running experiments managed by the testing framework.
enum GNUNET_SCHEDULER_Reason reason
Reason why the task is run now.
#define GNUNET_break(cond)
Use this for internal assertion violations that are not fatal (can be handled) but should not occur...
const struct GNUNET_DISK_FileHandle * GNUNET_DISK_pipe_handle(const struct GNUNET_DISK_PipeHandle *p, enum GNUNET_DISK_PipeEnd n)
Get the handle to a particular pipe end.
Definition: disk.c:2641
static void destroy_hosts(struct GNUNET_TESTBED_Host **hosts, unsigned int nhosts)
#define _(String)
GNU gettext support macro.
Definition: platform.h:208
static unsigned long child_exit_code
The return code of the binary.
#define GNUNET_log_strerror(level, cmd)
Log an error message at log-level &#39;level&#39; that indicates a failure of the command &#39;cmd&#39; with the mess...
int GNUNET_asprintf(char **buf, const char *format,...)
Like asprintf, just portable.
Use this option to have all of the standard streams (stdin, stdout and stderror) be inherited...
#define LOG_DEBUG(...)
Debug logging shorthand.
#define GNUNET_OS_get_hostname_max_length()
Get maximum string length returned by gethostname()
static char * fn
Filename of the unique file.
ssize_t GNUNET_DISK_file_write(const struct GNUNET_DISK_FileHandle *h, const void *buffer, size_t n)
Write a buffer to a file.
Definition: disk.c:1049
struct GNUNET_SCHEDULER_Task * GNUNET_SCHEDULER_add_now(GNUNET_SCHEDULER_TaskCallback task, void *task_cls)
Schedule a new task to be run as soon as possible.
Definition: scheduler.c:1273
GNUNET_OS_ProcessStatusType
Process status types.
static enum GNUNET_OS_ProcessStatusType child_status
The process status of the child.
struct GNUNET_OS_Process * GNUNET_OS_start_process_vap(int pipe_control, enum GNUNET_OS_InheritStdioFlags std_inheritance, struct GNUNET_DISK_PipeHandle *pipe_stdin, struct GNUNET_DISK_PipeHandle *pipe_stdout, struct GNUNET_DISK_PipeHandle *pipe_stderr, const char *filename, char *const argv[])
Start a process.
Definition: os_priority.c:1321
int main(int argc, char *argv[])
Execution start point.
#define GNUNET_TIME_UNIT_FOREVER_REL
Constant used to specify "forever".
The writing-end of a pipe.
The reading-end of a pipe.
static int fh
Handle to the unique file.
static char ** argv2
The arguments including the binary to spawn.
static struct GNUNET_SCHEDULER_Task * terminate_task_id
Task to kill the child.
struct GNUNET_DISK_PipeHandle * GNUNET_DISK_pipe(int blocking_read, int blocking_write, int inherit_read, int inherit_write)
Creates an interprocess channel.
Definition: disk.c:2289
static struct GNUNET_DISK_PipeHandle * sigpipe
Pipe used to communicate shutdown via signal.
struct GNUNET_SIGNAL_Context * GNUNET_SIGNAL_handler_install(int signal, GNUNET_SIGNAL_Handler handler)
Install a signal handler that will be run if the given signal is received.
Definition: signal.c:59
void GNUNET_TESTBED_host_destroy(struct GNUNET_TESTBED_Host *host)
Destroy a host handle.
#define GNUNET_SYSERR
Definition: gnunet_common.h:79
GUID host
static int ret
Global result.
char * getenv()
int GNUNET_DISK_pipe_close(struct GNUNET_DISK_PipeHandle *p)
Closes an interprocess channel.
Definition: disk.c:2603
static struct GNUNET_SCHEDULER_Task * child_death_task_id
Task to kill the child.
const char * GNUNET_TESTBED_host_get_hostname(const struct GNUNET_TESTBED_Host *host)
Obtain the host&#39;s hostname.
static char * hostname
Our hostname; we give this to all the peers we start.
static struct GNUNET_OS_Process * child
The child process we spawn.
static void shutdown_task(void *cls)
The shutdown task.
configuration data
Definition: configuration.c:85
static void run(void *cls)
The main scheduler run task.
unsigned int GNUNET_TESTBED_hosts_load_from_loadleveler(const struct GNUNET_CONFIGURATION_Handle *cfg, struct GNUNET_TESTBED_Host ***hosts)
Loads the set of host allocated by the LoadLeveler Job Scheduler.
Entry in list of pending tasks.
Definition: scheduler.c:134
#define LOG(kind,...)
Generic logging shorthand.
int GNUNET_log_setup(const char *comp, const char *loglevel, const char *logfile)
Setup logging.
static void sighandler_child_death()
Signal handler called for SIGCHLD.
#define GNUNET_TERM_SIG
The termination signal.
Definition: platform.h:282
Handle used to access files (and pipes).
Handle used to manage a pipe.
Definition: disk.c:66
#define GNUNET_malloc(size)
Wrapper around malloc.
#define GNUNET_free(ptr)
Wrapper around free.
int GNUNET_OS_process_kill(struct GNUNET_OS_Process *proc, int sig)
Sends a signal to the process.
Definition: os_priority.c:251
static void child_death_task(void *cls)
Task triggered whenever we receive a SIGCHLD (child process died).
void * GNUNET_SCHEDULER_cancel(struct GNUNET_SCHEDULER_Task *task)
Cancel the task with the specified identifier.
Definition: scheduler.c:965