]>
Commit | Line | Data |
---|---|---|
39de8281 AG |
1 | #ifndef _RDS_RDS_H |
2 | #define _RDS_RDS_H | |
3 | ||
4 | #include <net/sock.h> | |
5 | #include <linux/scatterlist.h> | |
6 | #include <linux/highmem.h> | |
7 | #include <rdma/rdma_cm.h> | |
8 | #include <linux/mutex.h> | |
9 | #include <linux/rds.h> | |
10 | ||
11 | #include "info.h" | |
12 | ||
13 | /* | |
14 | * RDS Network protocol version | |
15 | */ | |
16 | #define RDS_PROTOCOL_3_0 0x0300 | |
17 | #define RDS_PROTOCOL_3_1 0x0301 | |
18 | #define RDS_PROTOCOL_VERSION RDS_PROTOCOL_3_1 | |
19 | #define RDS_PROTOCOL_MAJOR(v) ((v) >> 8) | |
20 | #define RDS_PROTOCOL_MINOR(v) ((v) & 255) | |
21 | #define RDS_PROTOCOL(maj, min) (((maj) << 8) | min) | |
22 | ||
23 | /* | |
24 | * XXX randomly chosen, but at least seems to be unused: | |
25 | * # 18464-18768 Unassigned | |
26 | * We should do better. We want a reserved port to discourage unpriv'ed | |
27 | * userspace from listening. | |
28 | */ | |
29 | #define RDS_PORT 18634 | |
30 | ||
8cbd9606 AG |
31 | #ifdef ATOMIC64_INIT |
32 | #define KERNEL_HAS_ATOMIC64 | |
33 | #endif | |
34 | ||
39de8281 AG |
35 | #ifdef DEBUG |
36 | #define rdsdebug(fmt, args...) pr_debug("%s(): " fmt, __func__ , ##args) | |
37 | #else | |
38 | /* sigh, pr_debug() causes unused variable warnings */ | |
39 | static inline void __attribute__ ((format (printf, 1, 2))) | |
40 | rdsdebug(char *fmt, ...) | |
41 | { | |
42 | } | |
43 | #endif | |
44 | ||
45 | /* XXX is there one of these somewhere? */ | |
46 | #define ceil(x, y) \ | |
47 | ({ unsigned long __x = (x), __y = (y); (__x + __y - 1) / __y; }) | |
48 | ||
49 | #define RDS_FRAG_SHIFT 12 | |
50 | #define RDS_FRAG_SIZE ((unsigned int)(1 << RDS_FRAG_SHIFT)) | |
51 | ||
52 | #define RDS_CONG_MAP_BYTES (65536 / 8) | |
53 | #define RDS_CONG_MAP_LONGS (RDS_CONG_MAP_BYTES / sizeof(unsigned long)) | |
54 | #define RDS_CONG_MAP_PAGES (PAGE_ALIGN(RDS_CONG_MAP_BYTES) / PAGE_SIZE) | |
55 | #define RDS_CONG_MAP_PAGE_BITS (PAGE_SIZE * 8) | |
56 | ||
57 | struct rds_cong_map { | |
58 | struct rb_node m_rb_node; | |
59 | __be32 m_addr; | |
60 | wait_queue_head_t m_waitq; | |
61 | struct list_head m_conn_list; | |
62 | unsigned long m_page_addrs[RDS_CONG_MAP_PAGES]; | |
63 | }; | |
64 | ||
65 | ||
66 | /* | |
67 | * This is how we will track the connection state: | |
68 | * A connection is always in one of the following | |
69 | * states. Updates to the state are atomic and imply | |
70 | * a memory barrier. | |
71 | */ | |
72 | enum { | |
73 | RDS_CONN_DOWN = 0, | |
74 | RDS_CONN_CONNECTING, | |
75 | RDS_CONN_DISCONNECTING, | |
76 | RDS_CONN_UP, | |
77 | RDS_CONN_ERROR, | |
78 | }; | |
79 | ||
80 | /* Bits for c_flags */ | |
81 | #define RDS_LL_SEND_FULL 0 | |
82 | #define RDS_RECONNECT_PENDING 1 | |
83 | ||
84 | struct rds_connection { | |
85 | struct hlist_node c_hash_node; | |
86 | __be32 c_laddr; | |
87 | __be32 c_faddr; | |
88 | unsigned int c_loopback:1; | |
89 | struct rds_connection *c_passive; | |
90 | ||
91 | struct rds_cong_map *c_lcong; | |
92 | struct rds_cong_map *c_fcong; | |
93 | ||
94 | struct mutex c_send_lock; /* protect send ring */ | |
95 | struct rds_message *c_xmit_rm; | |
96 | unsigned long c_xmit_sg; | |
97 | unsigned int c_xmit_hdr_off; | |
98 | unsigned int c_xmit_data_off; | |
99 | unsigned int c_xmit_rdma_sent; | |
15133f6e | 100 | unsigned int c_xmit_atomic_sent; |
39de8281 AG |
101 | |
102 | spinlock_t c_lock; /* protect msg queues */ | |
103 | u64 c_next_tx_seq; | |
104 | struct list_head c_send_queue; | |
105 | struct list_head c_retrans; | |
106 | ||
107 | u64 c_next_rx_seq; | |
108 | ||
109 | struct rds_transport *c_trans; | |
110 | void *c_transport_data; | |
111 | ||
112 | atomic_t c_state; | |
113 | unsigned long c_flags; | |
114 | unsigned long c_reconnect_jiffies; | |
115 | struct delayed_work c_send_w; | |
116 | struct delayed_work c_recv_w; | |
117 | struct delayed_work c_conn_w; | |
118 | struct work_struct c_down_w; | |
119 | struct mutex c_cm_lock; /* protect conn state & cm */ | |
120 | ||
121 | struct list_head c_map_item; | |
122 | unsigned long c_map_queued; | |
123 | unsigned long c_map_offset; | |
124 | unsigned long c_map_bytes; | |
125 | ||
126 | unsigned int c_unacked_packets; | |
127 | unsigned int c_unacked_bytes; | |
128 | ||
129 | /* Protocol version */ | |
130 | unsigned int c_version; | |
131 | }; | |
132 | ||
133 | #define RDS_FLAG_CONG_BITMAP 0x01 | |
134 | #define RDS_FLAG_ACK_REQUIRED 0x02 | |
135 | #define RDS_FLAG_RETRANSMITTED 0x04 | |
7b70d033 | 136 | #define RDS_MAX_ADV_CREDIT 255 |
39de8281 AG |
137 | |
138 | /* | |
139 | * Maximum space available for extension headers. | |
140 | */ | |
141 | #define RDS_HEADER_EXT_SPACE 16 | |
142 | ||
143 | struct rds_header { | |
144 | __be64 h_sequence; | |
145 | __be64 h_ack; | |
146 | __be32 h_len; | |
147 | __be16 h_sport; | |
148 | __be16 h_dport; | |
149 | u8 h_flags; | |
150 | u8 h_credit; | |
151 | u8 h_padding[4]; | |
152 | __sum16 h_csum; | |
153 | ||
154 | u8 h_exthdr[RDS_HEADER_EXT_SPACE]; | |
155 | }; | |
156 | ||
157 | /* | |
158 | * Reserved - indicates end of extensions | |
159 | */ | |
160 | #define RDS_EXTHDR_NONE 0 | |
161 | ||
162 | /* | |
163 | * This extension header is included in the very | |
164 | * first message that is sent on a new connection, | |
165 | * and identifies the protocol level. This will help | |
166 | * rolling updates if a future change requires breaking | |
167 | * the protocol. | |
168 | * NB: This is no longer true for IB, where we do a version | |
169 | * negotiation during the connection setup phase (protocol | |
170 | * version information is included in the RDMA CM private data). | |
171 | */ | |
172 | #define RDS_EXTHDR_VERSION 1 | |
173 | struct rds_ext_header_version { | |
174 | __be32 h_version; | |
175 | }; | |
176 | ||
177 | /* | |
178 | * This extension header is included in the RDS message | |
179 | * chasing an RDMA operation. | |
180 | */ | |
181 | #define RDS_EXTHDR_RDMA 2 | |
182 | struct rds_ext_header_rdma { | |
183 | __be32 h_rdma_rkey; | |
184 | }; | |
185 | ||
186 | /* | |
187 | * This extension header tells the peer about the | |
188 | * destination <R_Key,offset> of the requested RDMA | |
189 | * operation. | |
190 | */ | |
191 | #define RDS_EXTHDR_RDMA_DEST 3 | |
192 | struct rds_ext_header_rdma_dest { | |
193 | __be32 h_rdma_rkey; | |
194 | __be32 h_rdma_offset; | |
195 | }; | |
196 | ||
197 | #define __RDS_EXTHDR_MAX 16 /* for now */ | |
198 | ||
199 | struct rds_incoming { | |
200 | atomic_t i_refcount; | |
201 | struct list_head i_item; | |
202 | struct rds_connection *i_conn; | |
203 | struct rds_header i_hdr; | |
204 | unsigned long i_rx_jiffies; | |
205 | __be32 i_saddr; | |
206 | ||
207 | rds_rdma_cookie_t i_rdma_cookie; | |
208 | }; | |
209 | ||
21f79afa AG |
210 | struct rds_mr { |
211 | struct rb_node r_rb_node; | |
212 | atomic_t r_refcount; | |
213 | u32 r_key; | |
214 | ||
215 | /* A copy of the creation flags */ | |
216 | unsigned int r_use_once:1; | |
217 | unsigned int r_invalidate:1; | |
218 | unsigned int r_write:1; | |
219 | ||
220 | /* This is for RDS_MR_DEAD. | |
221 | * It would be nice & consistent to make this part of the above | |
222 | * bit field here, but we need to use test_and_set_bit. | |
223 | */ | |
224 | unsigned long r_state; | |
225 | struct rds_sock *r_sock; /* back pointer to the socket that owns us */ | |
226 | struct rds_transport *r_trans; | |
227 | void *r_trans_private; | |
228 | }; | |
229 | ||
230 | /* Flags for mr->r_state */ | |
231 | #define RDS_MR_DEAD 0 | |
232 | ||
233 | struct rds_rdma_op { | |
234 | u32 r_key; | |
235 | u64 r_remote_addr; | |
236 | unsigned int r_write:1; | |
237 | unsigned int r_fence:1; | |
238 | unsigned int r_notify:1; | |
239 | unsigned int r_recverr:1; | |
240 | unsigned int r_mapped:1; | |
241 | unsigned int r_active:1; | |
242 | struct rds_notifier *r_notifier; | |
243 | unsigned int r_bytes; | |
244 | unsigned int r_nents; | |
245 | unsigned int r_count; | |
246 | struct scatterlist *r_sg; | |
247 | }; | |
248 | ||
249 | static inline rds_rdma_cookie_t rds_rdma_make_cookie(u32 r_key, u32 offset) | |
250 | { | |
251 | return r_key | (((u64) offset) << 32); | |
252 | } | |
253 | ||
254 | static inline u32 rds_rdma_cookie_key(rds_rdma_cookie_t cookie) | |
255 | { | |
256 | return cookie; | |
257 | } | |
258 | ||
259 | static inline u32 rds_rdma_cookie_offset(rds_rdma_cookie_t cookie) | |
260 | { | |
261 | return cookie >> 32; | |
262 | } | |
263 | ||
15133f6e AG |
264 | /* atomic operation types */ |
265 | #define RDS_ATOMIC_TYPE_CSWP 0 | |
266 | #define RDS_ATOMIC_TYPE_FADD 1 | |
267 | ||
39de8281 AG |
268 | /* |
269 | * m_sock_item and m_conn_item are on lists that are serialized under | |
270 | * conn->c_lock. m_sock_item has additional meaning in that once it is empty | |
271 | * the message will not be put back on the retransmit list after being sent. | |
272 | * messages that are canceled while being sent rely on this. | |
273 | * | |
274 | * m_inc is used by loopback so that it can pass an incoming message straight | |
275 | * back up into the rx path. It embeds a wire header which is also used by | |
276 | * the send path, which is kind of awkward. | |
277 | * | |
278 | * m_sock_item indicates the message's presence on a socket's send or receive | |
279 | * queue. m_rs will point to that socket. | |
280 | * | |
281 | * m_daddr is used by cancellation to prune messages to a given destination. | |
282 | * | |
283 | * The RDS_MSG_ON_SOCK and RDS_MSG_ON_CONN flags are used to avoid lock | |
284 | * nesting. As paths iterate over messages on a sock, or conn, they must | |
285 | * also lock the conn, or sock, to remove the message from those lists too. | |
286 | * Testing the flag to determine if the message is still on the lists lets | |
287 | * us avoid testing the list_head directly. That means each path can use | |
288 | * the message's list_head to keep it on a local list while juggling locks | |
289 | * without confusing the other path. | |
290 | * | |
291 | * m_ack_seq is an optional field set by transports who need a different | |
292 | * sequence number range to invalidate. They can use this in a callback | |
293 | * that they pass to rds_send_drop_acked() to see if each message has been | |
294 | * acked. The HAS_ACK_SEQ flag can be used to detect messages which haven't | |
295 | * had ack_seq set yet. | |
296 | */ | |
297 | #define RDS_MSG_ON_SOCK 1 | |
298 | #define RDS_MSG_ON_CONN 2 | |
299 | #define RDS_MSG_HAS_ACK_SEQ 3 | |
300 | #define RDS_MSG_ACK_REQUIRED 4 | |
301 | #define RDS_MSG_RETRANSMITTED 5 | |
302 | #define RDS_MSG_MAPPED 6 | |
303 | #define RDS_MSG_PAGEVEC 7 | |
304 | ||
305 | struct rds_message { | |
306 | atomic_t m_refcount; | |
307 | struct list_head m_sock_item; | |
308 | struct list_head m_conn_item; | |
309 | struct rds_incoming m_inc; | |
310 | u64 m_ack_seq; | |
311 | __be32 m_daddr; | |
312 | unsigned long m_flags; | |
313 | ||
314 | /* Never access m_rs without holding m_rs_lock. | |
315 | * Lock nesting is | |
316 | * rm->m_rs_lock | |
317 | * -> rs->rs_lock | |
318 | */ | |
319 | spinlock_t m_rs_lock; | |
320 | struct rds_sock *m_rs; | |
39de8281 | 321 | rds_rdma_cookie_t m_rdma_cookie; |
e779137a | 322 | struct { |
15133f6e AG |
323 | struct rm_atomic_op { |
324 | int op_type; | |
325 | uint64_t op_swap_add; | |
326 | uint64_t op_compare; | |
327 | ||
328 | u32 op_rkey; | |
329 | u64 op_remote_addr; | |
330 | unsigned int op_notify:1; | |
331 | unsigned int op_recverr:1; | |
332 | unsigned int op_mapped:1; | |
333 | unsigned int op_active:1; | |
334 | struct rds_notifier *op_notifier; | |
335 | struct scatterlist *op_sg; | |
336 | ||
337 | struct rds_mr *op_rdma_mr; | |
338 | } atomic; | |
339 | struct rm_rdma_op { | |
ff87e97a | 340 | struct rds_rdma_op m_rdma_op; |
e779137a AG |
341 | struct rds_mr *m_rdma_mr; |
342 | } rdma; | |
15133f6e | 343 | struct rm_data_op { |
241eef3e | 344 | unsigned int op_active:1; |
e779137a AG |
345 | unsigned int m_nents; |
346 | unsigned int m_count; | |
fc445084 | 347 | struct scatterlist *m_sg; |
e779137a AG |
348 | } data; |
349 | }; | |
fc445084 AG |
350 | unsigned int m_used_sgs; |
351 | unsigned int m_total_sgs; | |
39de8281 AG |
352 | }; |
353 | ||
354 | /* | |
355 | * The RDS notifier is used (optionally) to tell the application about | |
356 | * completed RDMA operations. Rather than keeping the whole rds message | |
357 | * around on the queue, we allocate a small notifier that is put on the | |
358 | * socket's notifier_list. Notifications are delivered to the application | |
359 | * through control messages. | |
360 | */ | |
361 | struct rds_notifier { | |
362 | struct list_head n_list; | |
363 | uint64_t n_user_token; | |
364 | int n_status; | |
365 | }; | |
366 | ||
367 | /** | |
368 | * struct rds_transport - transport specific behavioural hooks | |
369 | * | |
370 | * @xmit: .xmit is called by rds_send_xmit() to tell the transport to send | |
371 | * part of a message. The caller serializes on the send_sem so this | |
372 | * doesn't need to be reentrant for a given conn. The header must be | |
373 | * sent before the data payload. .xmit must be prepared to send a | |
374 | * message with no data payload. .xmit should return the number of | |
375 | * bytes that were sent down the connection, including header bytes. | |
376 | * Returning 0 tells the caller that it doesn't need to perform any | |
377 | * additional work now. This is usually the case when the transport has | |
378 | * filled the sending queue for its connection and will handle | |
379 | * triggering the rds thread to continue the send when space becomes | |
380 | * available. Returning -EAGAIN tells the caller to retry the send | |
381 | * immediately. Returning -ENOMEM tells the caller to retry the send at | |
382 | * some point in the future. | |
383 | * | |
384 | * @conn_shutdown: conn_shutdown stops traffic on the given connection. Once | |
385 | * it returns the connection can not call rds_recv_incoming(). | |
386 | * This will only be called once after conn_connect returns | |
387 | * non-zero success and will The caller serializes this with | |
388 | * the send and connecting paths (xmit_* and conn_*). The | |
389 | * transport is responsible for other serialization, including | |
390 | * rds_recv_incoming(). This is called in process context but | |
391 | * should try hard not to block. | |
392 | * | |
393 | * @xmit_cong_map: This asks the transport to send the local bitmap down the | |
394 | * given connection. XXX get a better story about the bitmap | |
395 | * flag and header. | |
396 | */ | |
397 | ||
335776bd AG |
398 | #define RDS_TRANS_IB 0 |
399 | #define RDS_TRANS_IWARP 1 | |
400 | #define RDS_TRANS_TCP 2 | |
401 | #define RDS_TRANS_COUNT 3 | |
402 | ||
39de8281 AG |
403 | struct rds_transport { |
404 | char t_name[TRANSNAMSIZ]; | |
405 | struct list_head t_item; | |
406 | struct module *t_owner; | |
407 | unsigned int t_prefer_loopback:1; | |
335776bd | 408 | unsigned int t_type; |
39de8281 AG |
409 | |
410 | int (*laddr_check)(__be32 addr); | |
411 | int (*conn_alloc)(struct rds_connection *conn, gfp_t gfp); | |
412 | void (*conn_free)(void *data); | |
413 | int (*conn_connect)(struct rds_connection *conn); | |
414 | void (*conn_shutdown)(struct rds_connection *conn); | |
415 | void (*xmit_prepare)(struct rds_connection *conn); | |
416 | void (*xmit_complete)(struct rds_connection *conn); | |
417 | int (*xmit)(struct rds_connection *conn, struct rds_message *rm, | |
418 | unsigned int hdr_off, unsigned int sg, unsigned int off); | |
419 | int (*xmit_cong_map)(struct rds_connection *conn, | |
420 | struct rds_cong_map *map, unsigned long offset); | |
421 | int (*xmit_rdma)(struct rds_connection *conn, struct rds_rdma_op *op); | |
241eef3e | 422 | int (*xmit_atomic)(struct rds_connection *conn, struct rds_message *rm); |
39de8281 AG |
423 | int (*recv)(struct rds_connection *conn); |
424 | int (*inc_copy_to_user)(struct rds_incoming *inc, struct iovec *iov, | |
425 | size_t size); | |
39de8281 AG |
426 | void (*inc_free)(struct rds_incoming *inc); |
427 | ||
428 | int (*cm_handle_connect)(struct rdma_cm_id *cm_id, | |
429 | struct rdma_cm_event *event); | |
430 | int (*cm_initiate_connect)(struct rdma_cm_id *cm_id); | |
431 | void (*cm_connect_complete)(struct rds_connection *conn, | |
432 | struct rdma_cm_event *event); | |
433 | ||
434 | unsigned int (*stats_info_copy)(struct rds_info_iterator *iter, | |
435 | unsigned int avail); | |
436 | void (*exit)(void); | |
437 | void *(*get_mr)(struct scatterlist *sg, unsigned long nr_sg, | |
438 | struct rds_sock *rs, u32 *key_ret); | |
439 | void (*sync_mr)(void *trans_private, int direction); | |
440 | void (*free_mr)(void *trans_private, int invalidate); | |
441 | void (*flush_mrs)(void); | |
442 | }; | |
443 | ||
444 | struct rds_sock { | |
445 | struct sock rs_sk; | |
446 | ||
447 | u64 rs_user_addr; | |
448 | u64 rs_user_bytes; | |
449 | ||
450 | /* | |
451 | * bound_addr used for both incoming and outgoing, no INADDR_ANY | |
452 | * support. | |
453 | */ | |
454 | struct rb_node rs_bound_node; | |
455 | __be32 rs_bound_addr; | |
456 | __be32 rs_conn_addr; | |
457 | __be16 rs_bound_port; | |
458 | __be16 rs_conn_port; | |
459 | ||
460 | /* | |
461 | * This is only used to communicate the transport between bind and | |
462 | * initiating connections. All other trans use is referenced through | |
463 | * the connection. | |
464 | */ | |
465 | struct rds_transport *rs_transport; | |
466 | ||
467 | /* | |
468 | * rds_sendmsg caches the conn it used the last time around. | |
469 | * This helps avoid costly lookups. | |
470 | */ | |
471 | struct rds_connection *rs_conn; | |
472 | ||
473 | /* flag indicating we were congested or not */ | |
474 | int rs_congested; | |
b98ba52f AG |
475 | /* seen congestion (ENOBUFS) when sending? */ |
476 | int rs_seen_congestion; | |
39de8281 AG |
477 | |
478 | /* rs_lock protects all these adjacent members before the newline */ | |
479 | spinlock_t rs_lock; | |
480 | struct list_head rs_send_queue; | |
481 | u32 rs_snd_bytes; | |
482 | int rs_rcv_bytes; | |
483 | struct list_head rs_notify_queue; /* currently used for failed RDMAs */ | |
484 | ||
485 | /* Congestion wake_up. If rs_cong_monitor is set, we use cong_mask | |
486 | * to decide whether the application should be woken up. | |
487 | * If not set, we use rs_cong_track to find out whether a cong map | |
488 | * update arrived. | |
489 | */ | |
490 | uint64_t rs_cong_mask; | |
491 | uint64_t rs_cong_notify; | |
492 | struct list_head rs_cong_list; | |
493 | unsigned long rs_cong_track; | |
494 | ||
495 | /* | |
496 | * rs_recv_lock protects the receive queue, and is | |
497 | * used to serialize with rds_release. | |
498 | */ | |
499 | rwlock_t rs_recv_lock; | |
500 | struct list_head rs_recv_queue; | |
501 | ||
502 | /* just for stats reporting */ | |
503 | struct list_head rs_item; | |
504 | ||
505 | /* these have their own lock */ | |
506 | spinlock_t rs_rdma_lock; | |
507 | struct rb_root rs_rdma_keys; | |
508 | ||
509 | /* Socket options - in case there will be more */ | |
510 | unsigned char rs_recverr, | |
511 | rs_cong_monitor; | |
512 | }; | |
513 | ||
514 | static inline struct rds_sock *rds_sk_to_rs(const struct sock *sk) | |
515 | { | |
516 | return container_of(sk, struct rds_sock, rs_sk); | |
517 | } | |
518 | static inline struct sock *rds_rs_to_sk(struct rds_sock *rs) | |
519 | { | |
520 | return &rs->rs_sk; | |
521 | } | |
522 | ||
523 | /* | |
524 | * The stack assigns sk_sndbuf and sk_rcvbuf to twice the specified value | |
525 | * to account for overhead. We don't account for overhead, we just apply | |
526 | * the number of payload bytes to the specified value. | |
527 | */ | |
528 | static inline int rds_sk_sndbuf(struct rds_sock *rs) | |
529 | { | |
530 | return rds_rs_to_sk(rs)->sk_sndbuf / 2; | |
531 | } | |
532 | static inline int rds_sk_rcvbuf(struct rds_sock *rs) | |
533 | { | |
534 | return rds_rs_to_sk(rs)->sk_rcvbuf / 2; | |
535 | } | |
536 | ||
537 | struct rds_statistics { | |
538 | uint64_t s_conn_reset; | |
539 | uint64_t s_recv_drop_bad_checksum; | |
540 | uint64_t s_recv_drop_old_seq; | |
541 | uint64_t s_recv_drop_no_sock; | |
542 | uint64_t s_recv_drop_dead_sock; | |
543 | uint64_t s_recv_deliver_raced; | |
544 | uint64_t s_recv_delivered; | |
545 | uint64_t s_recv_queued; | |
546 | uint64_t s_recv_immediate_retry; | |
547 | uint64_t s_recv_delayed_retry; | |
548 | uint64_t s_recv_ack_required; | |
549 | uint64_t s_recv_rdma_bytes; | |
550 | uint64_t s_recv_ping; | |
551 | uint64_t s_send_queue_empty; | |
552 | uint64_t s_send_queue_full; | |
553 | uint64_t s_send_sem_contention; | |
554 | uint64_t s_send_sem_queue_raced; | |
555 | uint64_t s_send_immediate_retry; | |
556 | uint64_t s_send_delayed_retry; | |
557 | uint64_t s_send_drop_acked; | |
558 | uint64_t s_send_ack_required; | |
559 | uint64_t s_send_queued; | |
560 | uint64_t s_send_rdma; | |
561 | uint64_t s_send_rdma_bytes; | |
562 | uint64_t s_send_pong; | |
563 | uint64_t s_page_remainder_hit; | |
564 | uint64_t s_page_remainder_miss; | |
565 | uint64_t s_copy_to_user; | |
566 | uint64_t s_copy_from_user; | |
567 | uint64_t s_cong_update_queued; | |
568 | uint64_t s_cong_update_received; | |
569 | uint64_t s_cong_send_error; | |
570 | uint64_t s_cong_send_blocked; | |
15133f6e AG |
571 | uint64_t s_atomic_cswp; |
572 | uint64_t s_atomic_fadd; | |
39de8281 AG |
573 | }; |
574 | ||
575 | /* af_rds.c */ | |
576 | void rds_sock_addref(struct rds_sock *rs); | |
577 | void rds_sock_put(struct rds_sock *rs); | |
578 | void rds_wake_sk_sleep(struct rds_sock *rs); | |
579 | static inline void __rds_wake_sk_sleep(struct sock *sk) | |
580 | { | |
aa395145 | 581 | wait_queue_head_t *waitq = sk_sleep(sk); |
39de8281 AG |
582 | |
583 | if (!sock_flag(sk, SOCK_DEAD) && waitq) | |
584 | wake_up(waitq); | |
585 | } | |
586 | extern wait_queue_head_t rds_poll_waitq; | |
587 | ||
588 | ||
589 | /* bind.c */ | |
590 | int rds_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len); | |
591 | void rds_remove_bound(struct rds_sock *rs); | |
592 | struct rds_sock *rds_find_bound(__be32 addr, __be16 port); | |
593 | ||
594 | /* cong.c */ | |
595 | int rds_cong_get_maps(struct rds_connection *conn); | |
596 | void rds_cong_add_conn(struct rds_connection *conn); | |
597 | void rds_cong_remove_conn(struct rds_connection *conn); | |
598 | void rds_cong_set_bit(struct rds_cong_map *map, __be16 port); | |
599 | void rds_cong_clear_bit(struct rds_cong_map *map, __be16 port); | |
600 | int rds_cong_wait(struct rds_cong_map *map, __be16 port, int nonblock, struct rds_sock *rs); | |
601 | void rds_cong_queue_updates(struct rds_cong_map *map); | |
602 | void rds_cong_map_updated(struct rds_cong_map *map, uint64_t); | |
603 | int rds_cong_updated_since(unsigned long *recent); | |
604 | void rds_cong_add_socket(struct rds_sock *); | |
605 | void rds_cong_remove_socket(struct rds_sock *); | |
606 | void rds_cong_exit(void); | |
607 | struct rds_message *rds_cong_update_alloc(struct rds_connection *conn); | |
608 | ||
609 | /* conn.c */ | |
610 | int __init rds_conn_init(void); | |
611 | void rds_conn_exit(void); | |
612 | struct rds_connection *rds_conn_create(__be32 laddr, __be32 faddr, | |
613 | struct rds_transport *trans, gfp_t gfp); | |
614 | struct rds_connection *rds_conn_create_outgoing(__be32 laddr, __be32 faddr, | |
615 | struct rds_transport *trans, gfp_t gfp); | |
2dc39357 | 616 | void rds_conn_shutdown(struct rds_connection *conn); |
39de8281 AG |
617 | void rds_conn_destroy(struct rds_connection *conn); |
618 | void rds_conn_reset(struct rds_connection *conn); | |
619 | void rds_conn_drop(struct rds_connection *conn); | |
620 | void rds_for_each_conn_info(struct socket *sock, unsigned int len, | |
621 | struct rds_info_iterator *iter, | |
622 | struct rds_info_lengths *lens, | |
623 | int (*visitor)(struct rds_connection *, void *), | |
624 | size_t item_len); | |
625 | void __rds_conn_error(struct rds_connection *conn, const char *, ...) | |
626 | __attribute__ ((format (printf, 2, 3))); | |
627 | #define rds_conn_error(conn, fmt...) \ | |
628 | __rds_conn_error(conn, KERN_WARNING "RDS: " fmt) | |
629 | ||
630 | static inline int | |
631 | rds_conn_transition(struct rds_connection *conn, int old, int new) | |
632 | { | |
633 | return atomic_cmpxchg(&conn->c_state, old, new) == old; | |
634 | } | |
635 | ||
636 | static inline int | |
637 | rds_conn_state(struct rds_connection *conn) | |
638 | { | |
639 | return atomic_read(&conn->c_state); | |
640 | } | |
641 | ||
642 | static inline int | |
643 | rds_conn_up(struct rds_connection *conn) | |
644 | { | |
645 | return atomic_read(&conn->c_state) == RDS_CONN_UP; | |
646 | } | |
647 | ||
648 | static inline int | |
649 | rds_conn_connecting(struct rds_connection *conn) | |
650 | { | |
651 | return atomic_read(&conn->c_state) == RDS_CONN_CONNECTING; | |
652 | } | |
653 | ||
654 | /* message.c */ | |
655 | struct rds_message *rds_message_alloc(unsigned int nents, gfp_t gfp); | |
fc445084 AG |
656 | struct scatterlist *rds_message_alloc_sgs(struct rds_message *rm, int nents); |
657 | int rds_message_copy_from_user(struct rds_message *rm, struct iovec *first_iov, | |
39de8281 AG |
658 | size_t total_len); |
659 | struct rds_message *rds_message_map_pages(unsigned long *page_addrs, unsigned int total_len); | |
660 | void rds_message_populate_header(struct rds_header *hdr, __be16 sport, | |
661 | __be16 dport, u64 seq); | |
662 | int rds_message_add_extension(struct rds_header *hdr, | |
663 | unsigned int type, const void *data, unsigned int len); | |
664 | int rds_message_next_extension(struct rds_header *hdr, | |
665 | unsigned int *pos, void *buf, unsigned int *buflen); | |
666 | int rds_message_add_version_extension(struct rds_header *hdr, unsigned int version); | |
667 | int rds_message_get_version_extension(struct rds_header *hdr, unsigned int *version); | |
668 | int rds_message_add_rdma_dest_extension(struct rds_header *hdr, u32 r_key, u32 offset); | |
669 | int rds_message_inc_copy_to_user(struct rds_incoming *inc, | |
670 | struct iovec *first_iov, size_t size); | |
39de8281 AG |
671 | void rds_message_inc_free(struct rds_incoming *inc); |
672 | void rds_message_addref(struct rds_message *rm); | |
673 | void rds_message_put(struct rds_message *rm); | |
674 | void rds_message_wait(struct rds_message *rm); | |
675 | void rds_message_unmapped(struct rds_message *rm); | |
676 | ||
677 | static inline void rds_message_make_checksum(struct rds_header *hdr) | |
678 | { | |
679 | hdr->h_csum = 0; | |
680 | hdr->h_csum = ip_fast_csum((void *) hdr, sizeof(*hdr) >> 2); | |
681 | } | |
682 | ||
683 | static inline int rds_message_verify_checksum(const struct rds_header *hdr) | |
684 | { | |
685 | return !hdr->h_csum || ip_fast_csum((void *) hdr, sizeof(*hdr) >> 2) == 0; | |
686 | } | |
687 | ||
688 | ||
689 | /* page.c */ | |
690 | int rds_page_remainder_alloc(struct scatterlist *scat, unsigned long bytes, | |
691 | gfp_t gfp); | |
692 | int rds_page_copy_user(struct page *page, unsigned long offset, | |
693 | void __user *ptr, unsigned long bytes, | |
694 | int to_user); | |
695 | #define rds_page_copy_to_user(page, offset, ptr, bytes) \ | |
696 | rds_page_copy_user(page, offset, ptr, bytes, 1) | |
697 | #define rds_page_copy_from_user(page, offset, ptr, bytes) \ | |
698 | rds_page_copy_user(page, offset, ptr, bytes, 0) | |
699 | void rds_page_exit(void); | |
700 | ||
701 | /* recv.c */ | |
702 | void rds_inc_init(struct rds_incoming *inc, struct rds_connection *conn, | |
703 | __be32 saddr); | |
704 | void rds_inc_addref(struct rds_incoming *inc); | |
705 | void rds_inc_put(struct rds_incoming *inc); | |
706 | void rds_recv_incoming(struct rds_connection *conn, __be32 saddr, __be32 daddr, | |
707 | struct rds_incoming *inc, gfp_t gfp, enum km_type km); | |
708 | int rds_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, | |
709 | size_t size, int msg_flags); | |
710 | void rds_clear_recv_queue(struct rds_sock *rs); | |
711 | int rds_notify_queue_get(struct rds_sock *rs, struct msghdr *msg); | |
712 | void rds_inc_info_copy(struct rds_incoming *inc, | |
713 | struct rds_info_iterator *iter, | |
714 | __be32 saddr, __be32 daddr, int flip); | |
715 | ||
716 | /* send.c */ | |
717 | int rds_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, | |
718 | size_t payload_len); | |
719 | void rds_send_reset(struct rds_connection *conn); | |
720 | int rds_send_xmit(struct rds_connection *conn); | |
721 | struct sockaddr_in; | |
722 | void rds_send_drop_to(struct rds_sock *rs, struct sockaddr_in *dest); | |
723 | typedef int (*is_acked_func)(struct rds_message *rm, uint64_t ack); | |
724 | void rds_send_drop_acked(struct rds_connection *conn, u64 ack, | |
725 | is_acked_func is_acked); | |
726 | int rds_send_acked_before(struct rds_connection *conn, u64 seq); | |
727 | void rds_send_remove_from_sock(struct list_head *messages, int status); | |
728 | int rds_send_pong(struct rds_connection *conn, __be16 dport); | |
729 | struct rds_message *rds_send_get_message(struct rds_connection *, | |
730 | struct rds_rdma_op *); | |
731 | ||
732 | /* rdma.c */ | |
733 | void rds_rdma_unuse(struct rds_sock *rs, u32 r_key, int force); | |
21f79afa AG |
734 | int rds_get_mr(struct rds_sock *rs, char __user *optval, int optlen); |
735 | int rds_get_mr_for_dest(struct rds_sock *rs, char __user *optval, int optlen); | |
736 | int rds_free_mr(struct rds_sock *rs, char __user *optval, int optlen); | |
737 | void rds_rdma_drop_keys(struct rds_sock *rs); | |
738 | int rds_rdma_extra_size(struct rds_rdma_args *args); | |
739 | int rds_cmsg_rdma_args(struct rds_sock *rs, struct rds_message *rm, | |
740 | struct cmsghdr *cmsg); | |
741 | int rds_cmsg_rdma_dest(struct rds_sock *rs, struct rds_message *rm, | |
742 | struct cmsghdr *cmsg); | |
743 | int rds_cmsg_rdma_args(struct rds_sock *rs, struct rds_message *rm, | |
744 | struct cmsghdr *cmsg); | |
745 | int rds_cmsg_rdma_map(struct rds_sock *rs, struct rds_message *rm, | |
746 | struct cmsghdr *cmsg); | |
747 | void rds_rdma_free_op(struct rds_rdma_op *ro); | |
15133f6e AG |
748 | void rds_rdma_send_complete(struct rds_message *rm, int wc_status); |
749 | void rds_atomic_send_complete(struct rds_message *rm, int wc_status); | |
750 | int rds_cmsg_atomic(struct rds_sock *rs, struct rds_message *rm, | |
751 | struct cmsghdr *cmsg); | |
21f79afa AG |
752 | |
753 | extern void __rds_put_mr_final(struct rds_mr *mr); | |
754 | static inline void rds_mr_put(struct rds_mr *mr) | |
755 | { | |
756 | if (atomic_dec_and_test(&mr->r_refcount)) | |
757 | __rds_put_mr_final(mr); | |
758 | } | |
39de8281 AG |
759 | |
760 | /* stats.c */ | |
9b8de747 | 761 | DECLARE_PER_CPU_SHARED_ALIGNED(struct rds_statistics, rds_stats); |
39de8281 AG |
762 | #define rds_stats_inc_which(which, member) do { \ |
763 | per_cpu(which, get_cpu()).member++; \ | |
764 | put_cpu(); \ | |
765 | } while (0) | |
766 | #define rds_stats_inc(member) rds_stats_inc_which(rds_stats, member) | |
767 | #define rds_stats_add_which(which, member, count) do { \ | |
768 | per_cpu(which, get_cpu()).member += count; \ | |
769 | put_cpu(); \ | |
770 | } while (0) | |
771 | #define rds_stats_add(member, count) rds_stats_add_which(rds_stats, member, count) | |
772 | int __init rds_stats_init(void); | |
773 | void rds_stats_exit(void); | |
774 | void rds_stats_info_copy(struct rds_info_iterator *iter, | |
36cbd3dc JE |
775 | uint64_t *values, const char *const *names, |
776 | size_t nr); | |
39de8281 AG |
777 | |
778 | /* sysctl.c */ | |
779 | int __init rds_sysctl_init(void); | |
780 | void rds_sysctl_exit(void); | |
781 | extern unsigned long rds_sysctl_sndbuf_min; | |
782 | extern unsigned long rds_sysctl_sndbuf_default; | |
783 | extern unsigned long rds_sysctl_sndbuf_max; | |
784 | extern unsigned long rds_sysctl_reconnect_min_jiffies; | |
785 | extern unsigned long rds_sysctl_reconnect_max_jiffies; | |
786 | extern unsigned int rds_sysctl_max_unacked_packets; | |
787 | extern unsigned int rds_sysctl_max_unacked_bytes; | |
788 | extern unsigned int rds_sysctl_ping_enable; | |
789 | extern unsigned long rds_sysctl_trace_flags; | |
790 | extern unsigned int rds_sysctl_trace_level; | |
791 | ||
792 | /* threads.c */ | |
793 | int __init rds_threads_init(void); | |
794 | void rds_threads_exit(void); | |
795 | extern struct workqueue_struct *rds_wq; | |
2dc39357 | 796 | void rds_queue_reconnect(struct rds_connection *conn); |
39de8281 AG |
797 | void rds_connect_worker(struct work_struct *); |
798 | void rds_shutdown_worker(struct work_struct *); | |
799 | void rds_send_worker(struct work_struct *); | |
800 | void rds_recv_worker(struct work_struct *); | |
801 | void rds_connect_complete(struct rds_connection *conn); | |
802 | ||
803 | /* transport.c */ | |
804 | int rds_trans_register(struct rds_transport *trans); | |
805 | void rds_trans_unregister(struct rds_transport *trans); | |
806 | struct rds_transport *rds_trans_get_preferred(__be32 addr); | |
807 | unsigned int rds_trans_stats_info_copy(struct rds_info_iterator *iter, | |
808 | unsigned int avail); | |
809 | int __init rds_trans_init(void); | |
810 | void rds_trans_exit(void); | |
811 | ||
812 | #endif |