e4f2d03d28
Explained in more detail in the following RFC: http://www.open-mpi.org/community/lists/devel/2010/03/7589.php This commit was SVN r22872.
116 строки
3.2 KiB
C
116 строки
3.2 KiB
C
/*
|
|
* Copyright (c) 2010 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
|
|
#include "opal/util/argv.h"
|
|
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/rml/base/rml_contact.h"
|
|
#include "orte/mca/routed/routed.h"
|
|
#include "orte/util/name_fns.h"
|
|
#include "orte/runtime/orte_globals.h"
|
|
|
|
#include "rml_oob.h"
|
|
|
|
char*
|
|
orte_rml_oob_get_uri(void)
|
|
{
|
|
char *proc_name = NULL;
|
|
char *proc_addr = NULL;
|
|
char *contact_info = NULL;
|
|
int rc;
|
|
|
|
proc_addr = orte_rml_oob_module.active_oob->oob_get_addr();
|
|
if (NULL == proc_addr) return NULL;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_util_convert_process_name_to_string(&proc_name,
|
|
ORTE_PROC_MY_NAME))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return NULL;
|
|
}
|
|
if (0 > asprintf(&contact_info, "%s;%s", proc_name, proc_addr)) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
}
|
|
free(proc_name);
|
|
free(proc_addr);
|
|
return contact_info;
|
|
}
|
|
|
|
|
|
int
|
|
orte_rml_oob_set_uri(const char* uri)
|
|
{
|
|
orte_process_name_t name;
|
|
char** uris;
|
|
char** ptr;
|
|
int rc = orte_rml_base_parse_uris(uri, &name, &uris);
|
|
if(rc != ORTE_SUCCESS) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
for(ptr = uris; ptr != NULL && *ptr != NULL; ptr++) {
|
|
orte_rml_oob_module.active_oob->oob_set_addr(&name, *ptr);
|
|
}
|
|
|
|
if(uris != NULL) {
|
|
opal_argv_free(uris);
|
|
}
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
int
|
|
orte_rml_oob_get_new_name(orte_process_name_t *name)
|
|
{
|
|
if (NULL != ORTE_PROC_MY_NAME) {
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
|
}
|
|
|
|
return orte_rml_oob_module.active_oob->oob_get_new_name(name);
|
|
|
|
}
|
|
|
|
int
|
|
orte_rml_oob_purge(orte_process_name_t *peer)
|
|
{
|
|
opal_list_item_t *item, *next;
|
|
orte_rml_oob_queued_msg_t *qmsg;
|
|
orte_rml_oob_msg_header_t *hdr;
|
|
orte_process_name_t step;
|
|
|
|
/* clear the oob contact info and pending messages */
|
|
orte_rml_oob_module.active_oob->oob_set_addr(peer, NULL);
|
|
|
|
/* clear our message queue */
|
|
OPAL_THREAD_LOCK(&orte_rml_oob_module.queued_lock);
|
|
item = opal_list_get_first(&orte_rml_oob_module.queued_routing_messages);
|
|
while (item != opal_list_get_end(&orte_rml_oob_module.queued_routing_messages)) {
|
|
next = opal_list_get_next(item);
|
|
qmsg = (orte_rml_oob_queued_msg_t*)item;
|
|
hdr = (orte_rml_oob_msg_header_t*) qmsg->payload[0].iov_base;
|
|
step = orte_routed.get_route(&hdr->destination);
|
|
if (peer->jobid == hdr->destination.jobid &&
|
|
peer->vpid == hdr->destination.vpid) {
|
|
opal_list_remove_item(&orte_rml_oob_module.queued_routing_messages, item);
|
|
OBJ_RELEASE(item);
|
|
} else if (step.jobid == hdr->destination.jobid &&
|
|
step.vpid == hdr->destination.vpid) {
|
|
opal_list_remove_item(&orte_rml_oob_module.queued_routing_messages, item);
|
|
OBJ_RELEASE(item);
|
|
}
|
|
item = next;
|
|
}
|
|
OPAL_THREAD_UNLOCK(&orte_rml_oob_module.queued_lock);
|
|
return ORTE_SUCCESS;
|
|
}
|