DAOS-16722 client: to intercept PMPI_Init() in libpil4dfs

Features: pil4dfs Allow-unstable-test: true faults-enabled: false Required-githooks: true Skipped-githooks: codespell Signed-off-by: Lei Huang <[email protected]> Required-githooks: true
daos-stack · Oct 23, 2024 · 622e817 · 622e817
1 parent 8668521
commit 622e817
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/client/dfuse/pil4dfs/int_dfs.c b/src/client/dfuse/pil4dfs/int_dfs.c
@@ -470,6 +470,7 @@ static int (*next_tcgetattr)(int fd, void *termios_p);
 /* end NOT supported by DAOS */
 
 static int (*next_mpi_init)(int *argc, char ***argv);
+static int (*next_pmpi_init)(int *argc, char ***argv);
 
 /* to do!! */
 /**
@@ -1040,6 +1041,22 @@ MPI_Init(int *argc, char ***argv)
 	return rc;
 }
 
+int
+PMPI_Init(int *argc, char ***argv)
+{
+	int rc;
+
+	if (next_pmpi_init == NULL) {
+		next_pmpi_init = dlsym(RTLD_NEXT, "PMPI_Init");
+		D_ASSERT(next_pmpi_init != NULL);
+	}
+
+	atomic_fetch_add_relaxed(&mpi_init_count, 1);
+	rc = next_pmpi_init(argc, argv);
+	atomic_fetch_add_relaxed(&mpi_init_count, -1);
+	return rc;
+}
+
 /** determine whether a path (both relative and absolute) is on DAOS or not. If yes,
  *  returns parent object, item name, full path of parent dir, full absolute path, and
  *  the pointer to struct dfs_mt.