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 82 83
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpi.h"
#include <stdio.h>
#include "mpitest.h"
#include "mpitestconf.h"
int verbose = 0;
int handlerCalled = 0;
/* Prototype to suppress compiler warnings */
void user_handler(MPI_File * fh, int *errcode, ...);
void user_handler(MPI_File * fh, int *errcode, ...)
{
if (*errcode != MPI_SUCCESS) {
handlerCalled++;
if (verbose) {
printf("In user_handler with code %d\n", *errcode);
}
}
}
int main(int argc, char *argv[])
{
int rank, errs = 0, rc;
MPI_Errhandler ioerr_handler;
MPI_Status status;
MPI_File fh;
char inbuf[80];
MTest_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
/* Create a file to which to attach the handler */
rc = MPI_File_open(MPI_COMM_WORLD, (char *) "test.txt",
MPI_MODE_CREATE | MPI_MODE_WRONLY | MPI_MODE_DELETE_ON_CLOSE,
MPI_INFO_NULL, &fh);
if (rc) {
errs++;
printf("Unable to open test.txt for writing\n");
}
rc = MPI_File_create_errhandler(user_handler, &ioerr_handler);
if (rc) {
errs++;
printf("MPI_File_create_Errhandler returned an error code: %d\n", rc);
}
rc = MPI_File_set_errhandler(fh, ioerr_handler);
if (rc) {
errs++;
printf("MPI_File_set_errhandler returned an error code: %d\n", rc);
}
/* avoid leaking the errhandler, safe because they have refcount semantics */
rc = MPI_Errhandler_free(&ioerr_handler);
if (rc) {
errs++;
printf("MPI_Errhandler_free returned an error code: %d\n", rc);
}
/* This should generate an error because the file mode is WRONLY */
rc = MPI_File_read_at(fh, 0, inbuf, 80, MPI_BYTE, &status);
if (handlerCalled != 1) {
errs++;
printf("User-defined error handler was not called\n");
}
rc = MPI_File_close(&fh);
if (rc) {
errs++;
printf("MPI_File_close returned an error code: %d\n", rc);
}
MTest_Finalize(errs);
return MTestReturnValue(errs);
}
|