src/testbed/gnunet_mpi_test.c

   1 #include "platform.h"
   2 #include "gnunet_util_lib.h"
   3
   4 /**
   5  * Generic logging shorthand
   6  */
   7 #define LOG(kind,...)                                           \
   8   GNUNET_log_from (kind, "gnunet-mpi-test", __VA_ARGS__)
   9
  10 int main (int argc, char *argv[])
  11 {
  12   char *msg;
  13   char *filename;
  14   pid_t pid;
  15   pid_t ppid;
  16   int rank;
  17   int msg_size;
  18   int ret;
  19
  20   ret = GNUNET_SYSERR;
  21   rank = 0;
  22   pid = getpid();
  23   (void) GNUNET_asprintf (&filename, "%d-%d.mpiout", (int) pid, rank);
  24   msg_size = GNUNET_asprintf (&msg, "My pid is: %d\n", pid);
  25   printf ("%s", msg);
  26   if (msg_size == GNUNET_DISK_fn_write (filename,
  27                                         msg, msg_size,
  28                                         GNUNET_DISK_PERM_USER_READ
  29                                         | GNUNET_DISK_PERM_GROUP_READ
  30                                         | GNUNET_DISK_PERM_USER_WRITE
  31                                         | GNUNET_DISK_PERM_GROUP_WRITE))
  32     ret = GNUNET_OK;
  33   GNUNET_free (filename);
  34   GNUNET_free (msg);
  35   if (GNUNET_OK != ret)
  36     goto finalize;
  37
  38   ret = GNUNET_SYSERR;
  39   ppid = pid;
  40   pid = fork ();
  41   if (-1 == pid)
  42   {
  43     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "fork");
  44     goto finalize;
  45   }
  46   if (0 == pid)
  47   {
  48     /* child code */
  49     pid = getpid ();
  50     (void) GNUNET_asprintf (&filename, "%d-%d.mpiout", (int) pid, rank);
  51     msg_size = GNUNET_asprintf (&msg, "Child of %d\n", (int) ppid);
  52     printf ("%s", msg);
  53     if (msg_size == GNUNET_DISK_fn_write (filename,
  54                                         msg, msg_size,
  55                                         GNUNET_DISK_PERM_USER_READ
  56                                         | GNUNET_DISK_PERM_GROUP_READ
  57                                         | GNUNET_DISK_PERM_USER_WRITE
  58                                         | GNUNET_DISK_PERM_GROUP_WRITE))
  59     ret = GNUNET_OK;
  60     GNUNET_free (filename);
  61     GNUNET_free (msg);
  62     return (GNUNET_OK == ret) ? 0 : 1;
  63   }
  64   else {
  65     int status;
  66     int childpid;
  67
  68     childpid = waitpid (pid, &status, 0);
  69     if (childpid != pid)
  70     {
  71       GNUNET_break (0);
  72       goto finalize;
  73     }
  74     if (!WIFEXITED (status))
  75     {
  76       GNUNET_break (0);
  77       goto finalize;
  78     }
  79     if (0 != WEXITSTATUS (status))
  80     {
  81       GNUNET_break (0);
  82     }
  83   }
  84   ret = GNUNET_OK;
  85
  86  finalize:
  87   return (GNUNET_OK == ret) ? 0 : 1;
  88 }