5b9fa7e998
the RML/OOB merge in slightly easier This commit was SVN r15530. The following SVN revision numbers were found above: r15517 --> open-mpi/ompi@41977fcc95 r15520 --> open-mpi/ompi@9cbc9df1b8 r15527 --> open-mpi/ompi@2d17dd9516
325 строки
9.1 KiB
C
325 строки
9.1 KiB
C
/* -*- C -*-
|
|
*
|
|
* Copyright (c) 2004-2007 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 Name Server
|
|
*
|
|
* The Open MPI Name Server provides unique name ranges for processes
|
|
* within the universe. Each universe will have one name server
|
|
* running within the seed daemon. This is done to prevent the
|
|
* inadvertent duplication of names.
|
|
*/
|
|
|
|
/*
|
|
* includes
|
|
*/
|
|
#include "orte_config.h"
|
|
|
|
#include "orte/orte_constants.h"
|
|
#include "orte/util/proc_info.h"
|
|
#include "opal/util/output.h"
|
|
#include "opal/mca/mca.h"
|
|
#include "opal/mca/base/mca_base_param.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/rml/rml.h"
|
|
#include "orte/mca/rml/base/rml_contact.h"
|
|
|
|
#include "ns_proxy.h"
|
|
|
|
|
|
/*
|
|
* Struct of function pointers that need to be initialized
|
|
*/
|
|
mca_ns_base_component_t mca_ns_proxy_component = {
|
|
{
|
|
MCA_NS_BASE_VERSION_2_0_0,
|
|
|
|
"proxy", /* MCA module name */
|
|
ORTE_MAJOR_VERSION, /* MCA module major version */
|
|
ORTE_MINOR_VERSION, /* MCA module minor version */
|
|
ORTE_RELEASE_VERSION, /* MCA module release version */
|
|
orte_ns_proxy_open, /* module open */
|
|
orte_ns_proxy_close /* module close */
|
|
},
|
|
{
|
|
/* The component is checkpoint ready */
|
|
MCA_BASE_METADATA_PARAM_CHECKPOINT
|
|
},
|
|
orte_ns_proxy_init, /* module init */
|
|
orte_ns_proxy_finalize /* module shutdown */
|
|
};
|
|
|
|
/*
|
|
* setup the function pointers for the module
|
|
*/
|
|
static mca_ns_base_module_t orte_ns_proxy_module = {
|
|
/* init */
|
|
orte_ns_proxy_module_init,
|
|
/** node functions */
|
|
orte_ns_proxy_create_nodeids,
|
|
orte_ns_proxy_get_node_info,
|
|
orte_ns_base_convert_nodeid_to_string,
|
|
orte_ns_base_convert_string_to_nodeid,
|
|
/* jobid functions */
|
|
orte_ns_proxy_create_jobid,
|
|
orte_ns_proxy_get_job_descendants,
|
|
orte_ns_proxy_get_job_children,
|
|
orte_ns_proxy_get_root_job,
|
|
orte_ns_proxy_get_parent_job,
|
|
orte_ns_proxy_get_job_family,
|
|
orte_ns_base_get_jobid_string,
|
|
orte_ns_base_convert_jobid_to_string,
|
|
orte_ns_base_convert_string_to_jobid,
|
|
orte_ns_proxy_reserve_range,
|
|
orte_ns_proxy_get_vpid_range,
|
|
/* vpid functions */
|
|
orte_ns_base_get_vpid_string,
|
|
orte_ns_base_convert_vpid_to_string,
|
|
orte_ns_base_convert_string_to_vpid,
|
|
/* name functions */
|
|
orte_ns_base_create_process_name,
|
|
orte_ns_proxy_create_my_name,
|
|
orte_ns_base_convert_string_to_process_name,
|
|
orte_ns_base_get_proc_name_string,
|
|
orte_ns_base_compare_fields,
|
|
/* peer functions */
|
|
orte_ns_proxy_get_peers,
|
|
/* tag server functions */
|
|
orte_ns_proxy_assign_rml_tag,
|
|
/* data type functions */
|
|
orte_ns_proxy_define_data_type,
|
|
/* diagnostic functions */
|
|
orte_ns_proxy_dump_jobs,
|
|
orte_ns_proxy_dump_tags,
|
|
orte_ns_proxy_dump_datatypes,
|
|
orte_ns_proxy_ft_event
|
|
};
|
|
|
|
/*
|
|
* Whether or not we allowed this component to be selected
|
|
*/
|
|
static bool initialized = false;
|
|
|
|
/* constructor - used to initialize state of taglist instance */
|
|
static void orte_ns_proxy_tagitem_construct(orte_ns_proxy_tagitem_t* tagitem)
|
|
{
|
|
tagitem->tag = ORTE_RML_TAG_MAX;
|
|
tagitem->name = NULL;
|
|
}
|
|
|
|
/* destructor - used to free any resources held by instance */
|
|
static void orte_ns_proxy_tagitem_destructor(orte_ns_proxy_tagitem_t* tagitem)
|
|
{
|
|
if (NULL != tagitem->name) {
|
|
free(tagitem->name);
|
|
}
|
|
}
|
|
|
|
/* define instance of opal_class_t */
|
|
OBJ_CLASS_INSTANCE(
|
|
orte_ns_proxy_tagitem_t, /* type name */
|
|
opal_object_t, /* parent "class" name */
|
|
orte_ns_proxy_tagitem_construct, /* constructor */
|
|
orte_ns_proxy_tagitem_destructor); /* destructor */
|
|
|
|
/* constructor - used to initialize state of dtilist instance */
|
|
static void orte_ns_proxy_dti_construct(orte_ns_proxy_dti_t* dti)
|
|
{
|
|
dti->id = ORTE_DSS_ID_MAX;
|
|
dti->name = NULL;
|
|
}
|
|
|
|
/* destructor - used to free any resources held by instance */
|
|
static void orte_ns_proxy_dti_destructor(orte_ns_proxy_dti_t* dti)
|
|
{
|
|
if (NULL != dti->name) {
|
|
free(dti->name);
|
|
}
|
|
}
|
|
|
|
/* define instance of opal_class_t */
|
|
OBJ_CLASS_INSTANCE(
|
|
orte_ns_proxy_dti_t, /* type name */
|
|
opal_object_t, /* parent "class" name */
|
|
orte_ns_proxy_dti_construct, /* constructor */
|
|
orte_ns_proxy_dti_destructor); /* destructor */
|
|
|
|
/*
|
|
* globals needed within proxy component
|
|
*/
|
|
|
|
orte_ns_proxy_globals_t orte_ns_proxy;
|
|
|
|
|
|
/*
|
|
* Open the proxy component and obtain the name of my proxy.
|
|
*/
|
|
int orte_ns_proxy_open(void)
|
|
{
|
|
int id, param;
|
|
|
|
id = mca_base_param_register_int("ns", "proxy", "debug", NULL, 0);
|
|
mca_base_param_lookup_int(id, &orte_ns_proxy.debug);
|
|
|
|
id = mca_base_param_register_int("ns", "proxy", "maxsize", NULL,
|
|
ORTE_NS_ARRAY_MAX_SIZE);
|
|
mca_base_param_lookup_int(id, ¶m);
|
|
orte_ns_proxy.max_size = (size_t)param;
|
|
|
|
id = mca_base_param_register_int("ns", "proxy", "blocksize", NULL,
|
|
ORTE_NS_ARRAY_BLOCK_SIZE);
|
|
mca_base_param_lookup_int(id, ¶m);
|
|
orte_ns_proxy.block_size = (size_t)param;
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
/*
|
|
* ditto for this one
|
|
*/
|
|
int orte_ns_proxy_close(void)
|
|
{
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
mca_ns_base_module_t* orte_ns_proxy_init(int *priority)
|
|
{
|
|
orte_process_name_t name;
|
|
int ret, rc;
|
|
|
|
/* If we are NOT to host a proxy, then we want to be selected, so do all
|
|
the setup and return the module */
|
|
/* opal_output(mca_ns_base_output, "ns_proxy: entered init\n"); */
|
|
if (NULL != orte_process_info.ns_replica_uri) {
|
|
|
|
/* Return a module (choose an arbitrary, positive priority --
|
|
it's only relevant compared to other ns components). If
|
|
we're not the seed, then we don't want to be selected, so
|
|
return NULL. */
|
|
|
|
*priority = 10;
|
|
|
|
/* define the proxy for us to use */
|
|
if(ORTE_SUCCESS != (ret = orte_rml_base_parse_uris(orte_process_info.ns_replica_uri, &name, NULL))) {
|
|
ORTE_ERROR_LOG(ret);
|
|
return NULL;
|
|
}
|
|
if(ORTE_SUCCESS != (ret = orte_dss.copy((void**)&orte_process_info.ns_replica, &name, ORTE_NAME))) {
|
|
ORTE_ERROR_LOG(ret);
|
|
return NULL;
|
|
}
|
|
|
|
/* initialize the taglist */
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_pointer_array_init(&(orte_ns_proxy.tags),
|
|
(orte_std_cntr_t)orte_ns_proxy.block_size,
|
|
(orte_std_cntr_t)orte_ns_proxy.max_size,
|
|
(orte_std_cntr_t)orte_ns_proxy.block_size))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return NULL;
|
|
}
|
|
orte_ns_proxy.num_tags = 0;
|
|
|
|
/* initialize the dtlist */
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_pointer_array_init(&(orte_ns_proxy.dts),
|
|
(orte_std_cntr_t)orte_ns_proxy.block_size,
|
|
(orte_std_cntr_t)orte_ns_proxy.max_size,
|
|
(orte_std_cntr_t)orte_ns_proxy.block_size))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return NULL;
|
|
}
|
|
orte_ns_proxy.num_dts = 0;
|
|
|
|
/* setup the thread lock */
|
|
OBJ_CONSTRUCT(&orte_ns_proxy.mutex, opal_mutex_t);
|
|
|
|
/* Return the module */
|
|
|
|
initialized = true;
|
|
return &orte_ns_proxy_module;
|
|
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
* module init function
|
|
*/
|
|
int orte_ns_proxy_module_init(void)
|
|
{
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
/*
|
|
* finalize routine
|
|
*/
|
|
int orte_ns_proxy_finalize(void)
|
|
{
|
|
orte_ns_proxy_tagitem_t **tag;
|
|
orte_ns_proxy_dti_t **dti;
|
|
orte_std_cntr_t i;
|
|
|
|
/* free all tracking storage, but only if this component was initialized */
|
|
|
|
if (initialized) {
|
|
tag = (orte_ns_proxy_tagitem_t**)(orte_ns_proxy.tags)->addr;
|
|
for (i=0; i < (orte_ns_proxy.tags)->size; i++) {
|
|
if (NULL != tag[i]) OBJ_RELEASE(tag[i]);
|
|
}
|
|
OBJ_RELEASE(orte_ns_proxy.tags);
|
|
|
|
dti = (orte_ns_proxy_dti_t**)(orte_ns_proxy.dts)->addr;
|
|
for (i=0; i < (orte_ns_proxy.dts)->size; i++) {
|
|
if (NULL != dti[i]) OBJ_RELEASE(dti[i]);
|
|
}
|
|
OBJ_RELEASE(orte_ns_proxy.dts);
|
|
|
|
initialized = false;
|
|
}
|
|
|
|
/* All done */
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
int orte_ns_proxy_ft_event(int state) {
|
|
if(OPAL_CRS_CHECKPOINT == state) {
|
|
;
|
|
}
|
|
else if(OPAL_CRS_CONTINUE == state) {
|
|
;
|
|
}
|
|
else if(OPAL_CRS_RESTART == state) {
|
|
;
|
|
}
|
|
else if(OPAL_CRS_TERM == state ) {
|
|
;
|
|
}
|
|
else {
|
|
;
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|