sds_portals_utcp_module.c

MPI stands for the Message Passing Interface. Written by the MPI Forum (a large committee comprising

/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 *
 */

#include "orte_config.h"

#include <string.h>

#include "orte/orte_constants.h"
#include "orte/mca/sds/sds.h"
#include "orte/mca/sds/base/base.h"
#include "orte/mca/sds/portals_utcp/sds_portals_utcp.h"
#include "orte/util/proc_info.h"
#include "orte/mca/ns/base/base.h"
#include "orte/mca/errmgr/base/base.h"

orte_sds_base_module_t orte_sds_portals_utcp_module = {
    orte_sds_portals_utcp_contact_universe,
    orte_sds_portals_utcp_set_name,
    orte_sds_portals_utcp_finalize,
};

int
orte_sds_portals_utcp_contact_universe(void)
{
    orte_process_info.seed = false;

    /* since we are seed, ensure that all replica info is NULL'd */
    if (NULL != orte_process_info.ns_replica_uri) {
	free(orte_process_info.ns_replica_uri);
	orte_process_info.ns_replica_uri = NULL;
    }
    if (NULL != orte_process_info.ns_replica) {
	free(orte_process_info.ns_replica);
	orte_process_info.ns_replica = NULL;
    }

    if (NULL != orte_process_info.gpr_replica_uri) {
	free(orte_process_info.gpr_replica_uri);
	orte_process_info.gpr_replica_uri = NULL;
    }
    if (NULL != orte_process_info.gpr_replica) {
	free(orte_process_info.gpr_replica);
	orte_process_info.gpr_replica = NULL;
    }

    return ORTE_SUCCESS;
}

int
orte_sds_portals_utcp_set_name(void)
{
    int rc, i, len, num_procs;
    orte_cellid_t cellid;
    orte_jobid_t jobid;
    orte_vpid_t vpid;
    char* vpid_string;
    char *nidmap_string;

    vpid_string = getenv("PTL_MY_RID");
    nidmap_string = getenv("PTL_NIDMAP");
    if (NULL == vpid_string || NULL == nidmap_string ||
        NULL == getenv("PTL_PIDMAP") || NULL == getenv("PTL_IFACE")) {
        return ORTE_ERR_NOT_FOUND;
    }

    /* Get our process information
     *
     * we're going to make up the cellid and jobid.  find our vpid,
     * assuming range starts at 0 
     */
    cellid = 0;
    jobid = 1; /* not 0, since it has special meaning */
    if (ORTE_SUCCESS != (rc = orte_ns.convert_string_to_vpid(&vpid, vpid_string))) {
        ORTE_ERROR_LOG(rc);
        return(rc);
    }
    if (ORTE_SUCCESS != (rc = orte_ns.create_process_name(
                                                          &(orte_process_info.my_name),
                                                          cellid,
                                                          jobid,
                                                          vpid))) {
        ORTE_ERROR_LOG(rc);
        return rc;
    }

    /*
     * Get our peer information.  We assume vpids start at 0.  We
     * assume that there are <num : + 1> procs, since the nidmap is a
     * : seperated list of nids, and the utcp reference implementation
     * assumes all will be present
     */
    len = strlen(nidmap_string);
    num_procs = 1;
    for (i = 0 ; i < len ; ++i) {
        if (nidmap_string[i] == ':') num_procs++;
    }
    
    orte_process_info.vpid_start = (orte_vpid_t) 0;
    orte_process_info.num_procs = (size_t) num_procs;

    return ORTE_SUCCESS;
}


int 
orte_sds_portals_utcp_finalize(void)
{
    return ORTE_SUCCESS;
}