- reverting to include MPI code

1 files changed, 10 insertions, 2 deletions

diff --git a/src/testbed/gnunet_mpi_test.c b/src/testbed/gnunet_mpi_test.c
index 9229675ce..1e1608228 100644
--- a/src/testbed/gnunet_mpi_test.c
+++ b/src/testbed/gnunet_mpi_test.c
@@ -1,5 +1,6 @@
 #include "platform.h"
 #include "gnunet_util_lib.h"
+#include <mpi.h>
 
 /**
  * Generic logging shorthand
@@ -13,15 +14,21 @@ int main (int argc, char *argv[])
   char *filename;
   pid_t pid;
   pid_t ppid;
+  int ntasks;
   int rank;
   int msg_size;
   int ret;
 
   ret = GNUNET_SYSERR;
-  rank = 0;
+  if (MPI_SUCCESS != MPI_Init (&argc, &argv))
+    return 1;
+  if (MPI_SUCCESS != MPI_Comm_size (MPI_COMM_WORLD, &ntasks))
+    goto finalize;
+  if (MPI_SUCCESS != MPI_Comm_rank (MPI_COMM_WORLD, &rank))
+    goto finalize;
   pid = getpid();
   (void) GNUNET_asprintf (&filename, "%d-%d.mpiout", (int) pid, rank);
-  msg_size = GNUNET_asprintf (&msg, "My pid is: %d\n", pid);
+  msg_size = GNUNET_asprintf (&msg, "My rank is: %d\n", rank);
   printf ("%s", msg);
   if (msg_size == GNUNET_DISK_fn_write (filename,
                                         msg, msg_size,
@@ -84,5 +91,6 @@ int main (int argc, char *argv[])
   ret = GNUNET_OK;
 
  finalize:
+  (void) MPI_Finalize ();
   return (GNUNET_OK == ret) ? 0 : 1;
 }