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#ifndef R2CLUSTER_TCP_H
28#define R2CLUSTER_TCP_H
29
30#include <linux/socket.h>
31#ifdef __KERNEL__
32#include <net/sock.h>
33#include <linux/tcp.h>
34#else
35#include <sys/socket.h>
36#endif
37#include <linux/inet.h>
38#include <linux/in.h>
39
40struct r2net_msg {
41 __be16 magic;
42 __be16 data_len;
43 __be16 msg_type;
44 __be16 pad1;
45 __be32 sys_status;
46 __be32 status;
47 __be32 key;
48 __be32 msg_num;
49 __u8 buf[0];
50};
51
52typedef int (r2net_msg_handler_func)(struct r2net_msg *msg, u32 len, void *data,
53 void **ret_data);
54typedef void (r2net_post_msg_handler_func)(int status, void *data,
55 void *ret_data);
56
57#define R2NET_MAX_PAYLOAD_BYTES (4096 - sizeof(struct r2net_msg))
58
59
60#define R2NET_RECONNECT_DELAY_MS_DEFAULT 2000
61
62#define R2NET_KEEPALIVE_DELAY_MS_DEFAULT 2000
63#define R2NET_IDLE_TIMEOUT_MS_DEFAULT 30000
64
65
66
67static inline int r2net_link_down(int err, struct socket *sock)
68{
69 if (sock) {
70 if (sock->sk->sk_state != TCP_ESTABLISHED &&
71 sock->sk->sk_state != TCP_CLOSE_WAIT)
72 return 1;
73 }
74
75 if (err >= 0)
76 return 0;
77 switch (err) {
78
79
80 case -ERESTARTSYS:
81 case -EBADF:
82
83
84 case -ECONNREFUSED:
85 case -ENOTCONN:
86 case -ECONNRESET:
87 case -EPIPE:
88 return 1;
89
90 }
91 return 0;
92}
93
94enum {
95 R2NET_DRIVER_UNINITED,
96 R2NET_DRIVER_READY,
97};
98
99int r2net_send_message(u32 msg_type, u32 key, void *data, u32 len,
100 u8 target_node, int *status);
101int r2net_send_message_vec(u32 msg_type, u32 key, struct kvec *vec,
102 size_t veclen, u8 target_node, int *status);
103
104int r2net_register_handler(u32 msg_type, u32 key, u32 max_len,
105 r2net_msg_handler_func *func, void *data,
106 r2net_post_msg_handler_func *post_func,
107 struct list_head *unreg_list);
108void r2net_unregister_handler_list(struct list_head *list);
109
110void r2net_fill_node_map(unsigned long *map, unsigned bytes);
111
112void r2net_force_data_magic(struct r2net_msg *, u16, u32);
113void r2net_hb_node_up_manual(int);
114struct r2net_node *r2net_nn_from_num(u8);
115
116struct r2nm_node;
117int r2net_register_hb_callbacks(void);
118void r2net_unregister_hb_callbacks(void);
119int r2net_start_listening(struct r2nm_node *node);
120void r2net_stop_listening(struct r2nm_node *node);
121void r2net_disconnect_node(struct r2nm_node *node);
122int r2net_num_connected_peers(void);
123
124int r2net_init(void);
125void r2net_exit(void);
126
127struct r2net_send_tracking;
128struct r2net_sock_container;
129
130#if 0
131int r2net_debugfs_init(void);
132void r2net_debugfs_exit(void);
133void r2net_debug_add_nst(struct r2net_send_tracking *nst);
134void r2net_debug_del_nst(struct r2net_send_tracking *nst);
135void r2net_debug_add_sc(struct r2net_sock_container *sc);
136void r2net_debug_del_sc(struct r2net_sock_container *sc);
137#else
138static inline int r2net_debugfs_init(void)
139{
140 return 0;
141}
142static inline void r2net_debugfs_exit(void)
143{
144}
145static inline void r2net_debug_add_nst(struct r2net_send_tracking *nst)
146{
147}
148static inline void r2net_debug_del_nst(struct r2net_send_tracking *nst)
149{
150}
151static inline void r2net_debug_add_sc(struct r2net_sock_container *sc)
152{
153}
154static inline void r2net_debug_del_sc(struct r2net_sock_container *sc)
155{
156}
157#endif
158
159#endif
160