/*
 * Copyright (C) by Argonne National Laboratory
 *     See COPYRIGHT in top-level directory
 */

#include "mpi.h"
#include "mpitestconf.h"
#ifdef HAVE_IOSTREAM
// Not all C++ compilers have iostream instead of iostream.h
#include <iostream>
#ifdef HAVE_NAMESPACE_STD
// Those that do often need the std namespace; otherwise, a bare "cout"
// is likely to fail to compile
using namespace std;
#endif
#else
#include <iostream.h>
#endif
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#include "mpitestcxx.h"

int main(int argc, char *argv[])
{
    int errs = 0;
    char *port_name, *port_name_out;
    char serv_name[256];
    int merr, mclass;
    const char *errmsg;
    int msglen;
    int rank;

    MTest_Init();

    port_name = new char[MPI::MAX_PORT_NAME];
    port_name_out = new char[MPI::MAX_PORT_NAME];

    rank = MPI::COMM_WORLD.Get_rank();

    /* Note that according to the MPI standard, port_name must
     * have been created by MPI_Open_port.  For current testing
     * purposes, we'll use a fake name.  This test should eventually use
     * a valid name from Open_port */

    strcpy(port_name, "otherhost:122");
    strcpy(serv_name, "MyTest");

    MPI::COMM_WORLD.Set_errhandler(MPI::ERRORS_THROW_EXCEPTIONS);

    if (rank == 0) {
        try {
            MPI::Publish_name(serv_name, MPI::INFO_NULL, port_name);
        }
        catch(MPI::Exception e) {
            errs++;
            errmsg = e.Get_error_string();
            cout << "Error in Publish_name " << errmsg << "\n";
        }

        MPI::COMM_WORLD.Barrier();
        MPI::COMM_WORLD.Barrier();

        try {
            MPI::Unpublish_name(serv_name, MPI::INFO_NULL, port_name);
        }
        catch(MPI::Exception e) {
            errs++;
            errmsg = e.Get_error_string();
            cout << "Error in Unpublish name " << errmsg << "\n";
        }
    } else {
        MPI::COMM_WORLD.Barrier();

        merr = MPI::SUCCESS;
        try {
            MPI::Lookup_name(serv_name, MPI::INFO_NULL, port_name_out);
        }
        catch(MPI::Exception e) {
            errs++;
            merr = e.Get_error_code();
            errmsg = e.Get_error_string();
            cout << "Error in Lookup name " << errmsg << "\n";
        }
        if (merr == MPI::SUCCESS) {
            if (strcmp(port_name, port_name_out)) {
                errs++;
                cout << "Lookup name returned the wrong value (" << port_name_out << ")\n";
            }
        }

        MPI::COMM_WORLD.Barrier();
    }


    MPI::COMM_WORLD.Barrier();

    merr = MPI::SUCCESS;
    port_name_out[0] = 0;
    try {
        MPI::Lookup_name(serv_name, MPI::INFO_NULL, port_name_out);
    }
    catch(MPI::Exception e) {
        merr = e.Get_error_code();
    }
    if (!merr) {
        errs++;
        cout << "Lookup name returned name after it was unpublished\n";
        cout << "The name returned was " << port_name_out << "\n";
    } else {
        /* Must be class MPI::ERR_NAME */
        mclass = MPI::Get_error_class(merr);
        if (mclass != MPI::ERR_NAME) {
            char *lerrmsg;
            lerrmsg = new char[MPI::MAX_ERROR_STRING];
            errs++;
            MPI::Get_error_string(merr, lerrmsg, msglen);
            cout << "Lookup name returned the wrong error class (" << mclass <<
                "), msg " << lerrmsg << "\n";
            delete[]lerrmsg;
        }
    }

    delete[]port_name;
    delete[]port_name_out;

    MTest_Finalize(errs);
    return 0;
}
