98 строки
3.3 KiB
C
98 строки
3.3 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
|
||
|
|
* University Research and Technology
|
||
|
|
* Corporation. All rights reserved.
|
||
|
|
* Copyright (c) 2004-2006 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 (c) 2006-2013 Los Alamos National Security, LLC.
|
||
|
|
* All rights reserved.
|
||
|
|
* Copyright (c) 2010-2011 Cisco Systems, Inc. All rights reserved.
|
||
|
|
* Copyright (c) 2013-2014 Intel, Inc. All rights reserved.
|
||
|
|
* $COPYRIGHT$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef _MCA_OOB_USOCK_H_
|
||
|
|
#define _MCA_OOB_USOCK_H_
|
||
|
|
|
||
|
|
#include "orte_config.h"
|
||
|
|
|
||
|
|
#include "orte/types.h"
|
||
|
|
|
||
|
|
#include "opal/mca/base/base.h"
|
||
|
|
#include "opal/class/opal_free_list.h"
|
||
|
|
#include "opal/class/opal_hash_table.h"
|
||
|
|
#include "opal/mca/event/event.h"
|
||
|
|
|
||
|
|
#include "orte/mca/oob/oob.h"
|
||
|
|
#include "orte/mca/oob/base/base.h"
|
||
|
|
|
||
|
|
|
||
|
|
BEGIN_C_DECLS
|
||
|
|
|
||
|
|
/* define some debug levels */
|
||
|
|
#define OOB_USOCK_DEBUG_FAIL 2
|
||
|
|
#define OOB_USOCK_DEBUG_CONNECT 7
|
||
|
|
|
||
|
|
/* forward declare a couple of structures */
|
||
|
|
struct mca_oob_usock_module_t;
|
||
|
|
struct mca_oob_usock_msg_error_t;
|
||
|
|
|
||
|
|
/* Module definition */
|
||
|
|
typedef void (*mca_oob_usock_module_init_fn_t)(void);
|
||
|
|
typedef void (*mca_oob_usock_module_fini_fn_t)(void);
|
||
|
|
typedef void (*mca_oob_usock_module_accept_connection_fn_t)(const int accepted_fd,
|
||
|
|
const struct sockaddr *addr);
|
||
|
|
typedef void (*mca_oob_usock_module_ping_fn_t)(const orte_process_name_t *proc);
|
||
|
|
typedef void (*mca_oob_usock_module_send_nb_fn_t)(orte_rml_send_t *msg);
|
||
|
|
typedef void (*mca_oob_usock_module_ft_event_fn_t)(int state);
|
||
|
|
|
||
|
|
typedef struct {
|
||
|
|
mca_oob_usock_module_init_fn_t init;
|
||
|
|
mca_oob_usock_module_fini_fn_t finalize;
|
||
|
|
mca_oob_usock_module_accept_connection_fn_t accept_connection;
|
||
|
|
mca_oob_usock_module_ping_fn_t ping;
|
||
|
|
mca_oob_usock_module_send_nb_fn_t send_nb;
|
||
|
|
mca_oob_usock_module_ft_event_fn_t ft_event;
|
||
|
|
} mca_oob_usock_module_api_t;
|
||
|
|
typedef struct {
|
||
|
|
mca_oob_usock_module_api_t api;
|
||
|
|
opal_event_base_t *ev_base; /* event base for the module progress thread */
|
||
|
|
bool ev_active;
|
||
|
|
opal_thread_t progress_thread;
|
||
|
|
opal_hash_table_t peers; // peer connection info
|
||
|
|
} mca_oob_usock_module_t;
|
||
|
|
ORTE_MODULE_DECLSPEC extern mca_oob_usock_module_t mca_oob_usock_module;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* the state of the connection
|
||
|
|
*/
|
||
|
|
typedef enum {
|
||
|
|
MCA_OOB_USOCK_UNCONNECTED,
|
||
|
|
MCA_OOB_USOCK_CLOSED,
|
||
|
|
MCA_OOB_USOCK_RESOLVE,
|
||
|
|
MCA_OOB_USOCK_CONNECTING,
|
||
|
|
MCA_OOB_USOCK_CONNECT_ACK,
|
||
|
|
MCA_OOB_USOCK_CONNECTED,
|
||
|
|
MCA_OOB_USOCK_FAILED,
|
||
|
|
MCA_OOB_USOCK_ACCEPTING
|
||
|
|
} mca_oob_usock_state_t;
|
||
|
|
|
||
|
|
/* module-level shared functions */
|
||
|
|
ORTE_MODULE_DECLSPEC void mca_oob_usock_send_handler(int fd, short args, void *cbdata);
|
||
|
|
ORTE_MODULE_DECLSPEC void mca_oob_usock_recv_handler(int fd, short args, void *cbdata);
|
||
|
|
|
||
|
|
|
||
|
|
END_C_DECLS
|
||
|
|
|
||
|
|
#endif /* MCA_OOB_USOCK_H_ */
|
||
|
|
|