/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */

/*

 *

 *  (C) 2006 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);

}