diff options
author | Sree Harsha Totakura <totakura@in.tum.de> | 2012-12-07 17:57:48 +0000 |
---|---|---|
committer | Sree Harsha Totakura <totakura@in.tum.de> | 2012-12-07 17:57:48 +0000 |
commit | 658bb6d4f307649b3403fa560816ba985fececc8 (patch) | |
tree | ebae0015970e7852b71e6205bffd8fb1206b8cef /src | |
parent | b90c571bc3811e034d3319c17b1f4fb2496e2d76 (diff) | |
download | gnunet-658bb6d4f307649b3403fa560816ba985fececc8.tar.gz gnunet-658bb6d4f307649b3403fa560816ba985fececc8.zip |
- fork within MPI
Diffstat (limited to 'src')
-rw-r--r-- | src/testbed/gnunet_mpi_test.c | 62 |
1 files changed, 61 insertions, 1 deletions
diff --git a/src/testbed/gnunet_mpi_test.c b/src/testbed/gnunet_mpi_test.c index 26e6fdac1..1e1608228 100644 --- a/src/testbed/gnunet_mpi_test.c +++ b/src/testbed/gnunet_mpi_test.c | |||
@@ -2,11 +2,18 @@ | |||
2 | #include "gnunet_util_lib.h" | 2 | #include "gnunet_util_lib.h" |
3 | #include <mpi.h> | 3 | #include <mpi.h> |
4 | 4 | ||
5 | /** | ||
6 | * Generic logging shorthand | ||
7 | */ | ||
8 | #define LOG(kind,...) \ | ||
9 | GNUNET_log_from (kind, "gnunet-mpi-test", __VA_ARGS__) | ||
10 | |||
5 | int main (int argc, char *argv[]) | 11 | int main (int argc, char *argv[]) |
6 | { | 12 | { |
7 | char *msg; | 13 | char *msg; |
8 | char *filename; | 14 | char *filename; |
9 | pid_t pid; | 15 | pid_t pid; |
16 | pid_t ppid; | ||
10 | int ntasks; | 17 | int ntasks; |
11 | int rank; | 18 | int rank; |
12 | int msg_size; | 19 | int msg_size; |
@@ -19,8 +26,10 @@ int main (int argc, char *argv[]) | |||
19 | goto finalize; | 26 | goto finalize; |
20 | if (MPI_SUCCESS != MPI_Comm_rank (MPI_COMM_WORLD, &rank)) | 27 | if (MPI_SUCCESS != MPI_Comm_rank (MPI_COMM_WORLD, &rank)) |
21 | goto finalize; | 28 | goto finalize; |
22 | (void) GNUNET_asprintf (&filename, "%d.mpiout", rank); | 29 | pid = getpid(); |
30 | (void) GNUNET_asprintf (&filename, "%d-%d.mpiout", (int) pid, rank); | ||
23 | msg_size = GNUNET_asprintf (&msg, "My rank is: %d\n", rank); | 31 | msg_size = GNUNET_asprintf (&msg, "My rank is: %d\n", rank); |
32 | printf ("%s", msg); | ||
24 | if (msg_size == GNUNET_DISK_fn_write (filename, | 33 | if (msg_size == GNUNET_DISK_fn_write (filename, |
25 | msg, msg_size, | 34 | msg, msg_size, |
26 | GNUNET_DISK_PERM_USER_READ | 35 | GNUNET_DISK_PERM_USER_READ |
@@ -30,6 +39,57 @@ int main (int argc, char *argv[]) | |||
30 | ret = GNUNET_OK; | 39 | ret = GNUNET_OK; |
31 | GNUNET_free (filename); | 40 | GNUNET_free (filename); |
32 | GNUNET_free (msg); | 41 | GNUNET_free (msg); |
42 | if (GNUNET_OK != ret) | ||
43 | goto finalize; | ||
44 | |||
45 | ret = GNUNET_SYSERR; | ||
46 | ppid = pid; | ||
47 | pid = fork (); | ||
48 | if (-1 == pid) | ||
49 | { | ||
50 | GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "fork"); | ||
51 | goto finalize; | ||
52 | } | ||
53 | if (0 == pid) | ||
54 | { | ||
55 | /* child code */ | ||
56 | pid = getpid (); | ||
57 | (void) GNUNET_asprintf (&filename, "%d-%d.mpiout", (int) pid, rank); | ||
58 | msg_size = GNUNET_asprintf (&msg, "Child of %d\n", (int) ppid); | ||
59 | printf ("%s", msg); | ||
60 | if (msg_size == GNUNET_DISK_fn_write (filename, | ||
61 | msg, msg_size, | ||
62 | GNUNET_DISK_PERM_USER_READ | ||
63 | | GNUNET_DISK_PERM_GROUP_READ | ||
64 | | GNUNET_DISK_PERM_USER_WRITE | ||
65 | | GNUNET_DISK_PERM_GROUP_WRITE)) | ||
66 | ret = GNUNET_OK; | ||
67 | GNUNET_free (filename); | ||
68 | GNUNET_free (msg); | ||
69 | return (GNUNET_OK == ret) ? 0 : 1; | ||
70 | } | ||
71 | else { | ||
72 | int status; | ||
73 | int childpid; | ||
74 | |||
75 | childpid = waitpid (pid, &status, 0); | ||
76 | if (childpid != pid) | ||
77 | { | ||
78 | GNUNET_break (0); | ||
79 | goto finalize; | ||
80 | } | ||
81 | if (!WIFEXITED (status)) | ||
82 | { | ||
83 | GNUNET_break (0); | ||
84 | goto finalize; | ||
85 | } | ||
86 | if (0 != WEXITSTATUS (status)) | ||
87 | { | ||
88 | GNUNET_break (0); | ||
89 | } | ||
90 | } | ||
91 | ret = GNUNET_OK; | ||
92 | |||
33 | finalize: | 93 | finalize: |
34 | (void) MPI_Finalize (); | 94 | (void) MPI_Finalize (); |
35 | return (GNUNET_OK == ret) ? 0 : 1; | 95 | return (GNUNET_OK == ret) ? 0 : 1; |