#ifndef __SHM_SOCKET_H__ #define __SHM_SOCKET_H__ #include "usg_common.h" #include "usg_typedef.h" #include "shm_queue.h" enum shm_msg_type_t { SHM_SOCKET_OPEN = 1, SHM_SOCKET_OPEN_REPLY = 2, SHM_SOCKET_CLOSE = 3, SHM_COMMON_MSG = 4 }; enum shm_socket_flag_t { SHM_MSG_TIMEOUT = 1, SHM_MSG_NOWAIT = 2 }; enum shm_socket_type_t { SHM_SOCKET_STREAM = 1, SHM_SOCKET_DGRAM = 2 }; enum shm_socket_error_type_t { SHM_SOCKET_ECONNFAILED = 1, SHM_SOCKET_ETIMEOUT = 2 }; enum shm_connection_status_t { SHM_CONN_CLOSED=1, SHM_CONN_LISTEN=2, SHM_CONN_ESTABLISHED=3 }; typedef struct shm_msg_t { int port; shm_msg_type_t type; size_t size; void * buf; } shm_msg_t; typedef struct shm_socket_t { shm_socket_type_t socket_type; // 本地port int port; bool force_bind; shm_connection_status_t status; SHMQueue *queue; SHMQueue *remoteQueue; LockFreeQueue *messageQueue; LockFreeQueue *acceptQueue; std::map *clientSocketMap; pthread_t dispatch_thread; } shm_socket_t; size_t shm_socket_remove_keys(int keys[], size_t length); shm_socket_t *shm_open_socket(shm_socket_type_t socket_type); int shm_close_socket(shm_socket_t * socket) ; int shm_socket_bind(shm_socket_t * socket, int port) ; int shm_socket_force_bind(shm_socket_t * socket, int port) ; int shm_listen(shm_socket_t * socket) ; shm_socket_t* shm_accept(shm_socket_t* socket); int shm_connect(shm_socket_t * socket, int port); int shm_send(shm_socket_t * socket, const void *buf, const int size) ; int shm_recv(shm_socket_t * socket, void **buf, int *size) ; int shm_sendto(shm_socket_t *socket, const void *buf, const int size, const int port, const struct timespec * timeout = NULL, const int flags=0); int shm_recvfrom(shm_socket_t *socket, void **buf, int *size, int *port, struct timespec * timeout = NULL, int flags=0); int shm_sendandrecv(shm_socket_t *socket, const void *send_buf, const int send_size, const int send_port, void **recv_buf, int *recv_size, struct timespec * timeout = NULL, int flags=0); #endif