128 строки
2.8 KiB
C
128 строки
2.8 KiB
C
#ifndef _IOF_BASE_ENDPOINT_
|
|
#define _IOF_BASE_ENDPOINT_
|
|
|
|
#include "ompi_config.h"
|
|
#include "opal/class/opal_list.h"
|
|
#include "opal/event/event.h"
|
|
#include "mca/iof/iof.h"
|
|
#include "mca/iof/base/iof_base_header.h"
|
|
|
|
enum {
|
|
ORTE_IOF_EP_OPEN,
|
|
ORTE_IOF_EP_CLOSING,
|
|
ORTE_IOF_EP_CLOSED
|
|
};
|
|
|
|
/**
|
|
* Structure that represents a published endpoint.
|
|
*/
|
|
|
|
struct orte_iof_base_endpoint_t {
|
|
opal_list_item_t super;
|
|
orte_iof_base_mode_t ep_mode;
|
|
orte_process_name_t ep_name;
|
|
int ep_tag;
|
|
int ep_fd;
|
|
int ep_state;
|
|
uint32_t ep_seq;
|
|
uint32_t ep_ack;
|
|
opal_event_t ep_event;
|
|
opal_list_t ep_frags;
|
|
};
|
|
typedef struct orte_iof_base_endpoint_t orte_iof_base_endpoint_t;
|
|
|
|
OBJ_CLASS_DECLARATION(orte_iof_base_endpoint_t);
|
|
|
|
/*
|
|
* Diff between two sequence numbers allowing for rollover
|
|
*/
|
|
|
|
#define ORTE_IOF_BASE_SEQDIFF(s1,s2) \
|
|
((s1 > s2) ? (s1 - s2) : (s1 + (ULONG_MAX - s2)))
|
|
|
|
|
|
/**
|
|
* Create a local endpoint.
|
|
*
|
|
* @param name Process name corresponding to endpoint.
|
|
* @param mode Source or sink of data (exclusive).
|
|
* @param tag Logical tag for matching.
|
|
* @aram fd Local file descriptor corresponding to endpoint.
|
|
*/
|
|
|
|
int orte_iof_base_endpoint_create(
|
|
const orte_process_name_t* name,
|
|
orte_iof_base_mode_t mode,
|
|
int tag,
|
|
int fd);
|
|
|
|
/**
|
|
* Delete all local endpoints matching the specified
|
|
* name/mask/tag parameters.
|
|
*
|
|
* @paran name Process name corresponding to one or more endpoint(s).
|
|
* @param mask Mask used for name comparisons.
|
|
* @param tag Tag for matching endpoints.
|
|
*/
|
|
|
|
int orte_iof_base_endpoint_delete(
|
|
const orte_process_name_t* name,
|
|
orte_ns_cmp_bitmask_t mask,
|
|
int tag);
|
|
|
|
/**
|
|
* Disable forwarding through the specified endpoint.
|
|
*/
|
|
|
|
int orte_iof_base_endpoint_close(
|
|
orte_iof_base_endpoint_t* endpoint);
|
|
|
|
/**
|
|
* Attempt to match an endpoint based on the destination
|
|
* process name/mask/tag.
|
|
*/
|
|
|
|
orte_iof_base_endpoint_t* orte_iof_base_endpoint_match(
|
|
const orte_process_name_t* dst_name,
|
|
orte_ns_cmp_bitmask_t dst_mask,
|
|
int dst_tag);
|
|
|
|
/**
|
|
* Forward the specified message out the endpoint.
|
|
*/
|
|
|
|
int orte_iof_base_endpoint_forward(
|
|
orte_iof_base_endpoint_t* endpoint,
|
|
const orte_process_name_t* src,
|
|
orte_iof_base_msg_header_t* hdr,
|
|
const unsigned char* data);
|
|
|
|
/*
|
|
* Callback when peer has closed endpoint.
|
|
*/
|
|
|
|
void orte_iof_base_endpoint_closed(
|
|
orte_iof_base_endpoint_t* endpoint);
|
|
|
|
/**
|
|
* Callback when the specified sequence has been
|
|
* acknowledged.
|
|
*/
|
|
|
|
int orte_iof_base_endpoint_ack(
|
|
orte_iof_base_endpoint_t* endpoint,
|
|
uint32_t seq);
|
|
|
|
/**
|
|
* Check for pending I/O
|
|
*/
|
|
|
|
static inline bool orte_iof_base_endpoint_pending(
|
|
orte_iof_base_endpoint_t* endpoint)
|
|
{
|
|
return opal_list_get_size(&endpoint->ep_frags) || (endpoint->ep_seq != endpoint->ep_ack);
|
|
}
|
|
|
|
#endif
|
|
|