1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "ad_ime.h"
#include "ad_ime_common.h"
#include <unistd.h>
#include <sys/types.h>
/* keyval hack to both tell us if we've already initialized im and also
* close it down when mpi exits */
int ADIOI_IME_Initialized = MPI_KEYVAL_INVALID;
void ADIOI_IME_End(int *error_code)
{
int ret;
static char myname[] = "ADIOI_IME_END";
ret = ime_native_finalize();
/* --BEGIN ERROR HANDLING-- */
if (ret != 0) {
*error_code = MPIO_Err_create_code(MPI_SUCCESS,
MPIR_ERR_RECOVERABLE,
myname, __LINE__,
MPI_ERR_FILE, "Error in ime_native_finalize", 0);
return;
}
/* --END ERROR HANDLING-- */
*error_code = MPI_SUCCESS;
}
int ADIOI_IME_End_call(MPI_Comm comm, int keyval, void *attribute_val, void *extra_state)
{
int error_code;
ADIOI_IME_End(&error_code);
MPI_Keyval_free(&keyval);
return error_code;
}
void ADIOI_IME_Init(int rank, int *error_code)
{
/* do nothing if we've already fired up the im interface */
if (ADIOI_IME_Initialized != MPI_KEYVAL_INVALID) {
*error_code = MPI_SUCCESS;
return;
}
ime_native_init();
*error_code = MPI_SUCCESS;
MPI_Keyval_create(MPI_NULL_COPY_FN, ADIOI_IME_End_call, &ADIOI_IME_Initialized, (void *) 0);
/* just like romio does, we make a dummy attribute so we
* get cleaned up */
MPI_Attr_put(MPI_COMM_SELF, ADIOI_IME_Initialized, (void *) 0);
}
/* Return an IME-compatible filename (add 'ime:' prefix).
* New filename must be free'd by the user */
char *ADIOI_IME_Add_prefix(const char *filename)
{
static char myname[] = "ADIOI_IME_ADD_PREFIX";
size_t f_len = strlen(filename) + 1;
char *ime_filename = ADIOI_Malloc(f_len + ADIOI_IME_PREFIX_LEN);
if (!ime_filename) {
MPIO_Err_create_code(MPI_SUCCESS,
MPIR_ERR_FATAL,
myname, __LINE__, MPI_ERR_UNKNOWN, "Error allocating memory", 0);
return NULL;
}
ADIOI_Strncpy(ime_filename, ADIOI_IME_PREFIX, ADIOI_IME_PREFIX_LEN);
ADIOI_Strncpy((ime_filename + ADIOI_IME_PREFIX_LEN), filename, f_len);
return ime_filename;
}
|
