/* -*- Mode: C++; c-basic-offset:4 ; -*- */

/*

 *

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

}