update

wangzhengquan

2020-10-19 af85260254bacac40a68d4f5f61950523beb3a27

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#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;
    int mutex;
    shm_connection_status_t status;
    SHMQueue<shm_msg_t> *queue;
    SHMQueue<shm_msg_t> *remoteQueue;
    LockFreeQueue<shm_msg_t, DM_Allocator> *messageQueue;
    LockFreeQueue<shm_msg_t, DM_Allocator> *acceptQueue;
    std::map<int, shm_socket_t* > *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);
 
/**
 * 功能同shm_sendandrecv, 但是不是线程安全的
 */
int shm_sendandrecv_unsafe(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