gpr_replica_segment_tl.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$
 */
/** @file:
 *
 * The Open MPI general purpose registry - support functions.
 *
 */

/*
 * includes
 */

#include "orte_config.h"

#include "opal/util/trace.h"

#include "orte/mca/gpr/replica/transition_layer/gpr_replica_tl.h"

int orte_gpr_replica_find_seg(orte_gpr_replica_segment_t **seg,
                              bool create, char *segment)
{
    int rc=ORTE_SUCCESS;
    orte_std_cntr_t i, cntri;
    orte_gpr_replica_segment_t **ptr;

    OPAL_TRACE(3);

    /* initialize to nothing */
    *seg = NULL;
    
    if (NULL == segment) {
        /* this is an allowed value - the index function, for example,
         * will pass this to us if we want the index of the global level
         * of the registry (i.e., the index of segment names). Just return
         * NULL and we'll be okay
         */
        return ORTE_SUCCESS;
    }
    
    /* search the registry segments to find which one is being referenced */
    ptr = (orte_gpr_replica_segment_t**)(orte_gpr_replica.segments->addr);
    cntri = 0;
    for (i=0; cntri < orte_gpr_replica.num_segs &&
              i < (orte_gpr_replica.segments)->size; i++) {
        if (NULL != ptr[i]) {
            cntri++;
            if (0 == strcmp(segment, ptr[i]->name)) {
                *seg = ptr[i];
                return ORTE_SUCCESS;
            }
        }
    }
    
    if (!create) {
        /* couldn't find it and don't want it created - just return NULL */
        return ORTE_ERR_BAD_PARAM;
    }
    
    /* add the segment to the registry */
    *seg = OBJ_NEW(orte_gpr_replica_segment_t);
    (*seg)->name = strdup(segment);
    if (0 > orte_pointer_array_add(&i, orte_gpr_replica.segments, (void*)(*seg))) {
        OBJ_RELEASE(*seg);
        return rc;
    }
    (*seg)->itag = i;
    (orte_gpr_replica.num_segs)++;
    
    return ORTE_SUCCESS;
}