1
2
3
4
5
6
7
8
9
10#ifndef _LINUX_SUNRPC_XPRT_H
11#define _LINUX_SUNRPC_XPRT_H
12
13#include <linux/uio.h>
14#include <linux/socket.h>
15#include <linux/in.h>
16#include <linux/ktime.h>
17#include <linux/kref.h>
18#include <linux/sunrpc/sched.h>
19#include <linux/sunrpc/xdr.h>
20#include <linux/sunrpc/msg_prot.h>
21
22#ifdef __KERNEL__
23
24#define RPC_MIN_SLOT_TABLE (2U)
25#define RPC_DEF_SLOT_TABLE (16U)
26#define RPC_MAX_SLOT_TABLE_LIMIT (65536U)
27#define RPC_MAX_SLOT_TABLE RPC_MAX_SLOT_TABLE_LIMIT
28
29#define RPC_CWNDSHIFT (8U)
30#define RPC_CWNDSCALE (1U << RPC_CWNDSHIFT)
31#define RPC_INITCWND RPC_CWNDSCALE
32#define RPC_MAXCWND(xprt) ((xprt)->max_reqs << RPC_CWNDSHIFT)
33#define RPCXPRT_CONGESTED(xprt) ((xprt)->cong >= (xprt)->cwnd)
34
35
36
37
38struct rpc_timeout {
39 unsigned long to_initval,
40 to_maxval,
41 to_increment;
42 unsigned int to_retries;
43 unsigned char to_exponential;
44};
45
46enum rpc_display_format_t {
47 RPC_DISPLAY_ADDR = 0,
48 RPC_DISPLAY_PORT,
49 RPC_DISPLAY_PROTO,
50 RPC_DISPLAY_HEX_ADDR,
51 RPC_DISPLAY_HEX_PORT,
52 RPC_DISPLAY_NETID,
53 RPC_DISPLAY_MAX,
54};
55
56struct rpc_task;
57struct rpc_xprt;
58struct seq_file;
59struct svc_serv;
60struct net;
61
62
63
64
65struct rpc_rqst {
66
67
68
69 struct rpc_xprt * rq_xprt;
70 struct xdr_buf rq_snd_buf;
71 struct xdr_buf rq_rcv_buf;
72
73
74
75
76 struct rpc_task * rq_task;
77 struct rpc_cred * rq_cred;
78 __be32 rq_xid;
79 int rq_cong;
80 u32 rq_seqno;
81 int rq_enc_pages_num;
82 struct page **rq_enc_pages;
83
84 void (*rq_release_snd_buf)(struct rpc_rqst *);
85
86 union {
87 struct list_head rq_list;
88 struct rb_node rq_recv;
89 };
90
91 struct list_head rq_xmit;
92 struct list_head rq_xmit2;
93
94 void *rq_buffer;
95 size_t rq_callsize;
96 void *rq_rbuffer;
97 size_t rq_rcvsize;
98 size_t rq_xmit_bytes_sent;
99 size_t rq_reply_bytes_recvd;
100
101
102 struct xdr_buf rq_private_buf;
103
104
105 unsigned long rq_majortimeo;
106 unsigned long rq_timeout;
107 ktime_t rq_rtt;
108 unsigned int rq_retries;
109 unsigned int rq_connect_cookie;
110
111
112
113 atomic_t rq_pin;
114
115
116
117
118 u32 rq_bytes_sent;
119
120 ktime_t rq_xtime;
121 int rq_ntrans;
122
123#if defined(CONFIG_SUNRPC_BACKCHANNEL)
124 struct list_head rq_bc_list;
125 unsigned long rq_bc_pa_state;
126 struct list_head rq_bc_pa_list;
127#endif
128};
129#define rq_svec rq_snd_buf.head
130#define rq_slen rq_snd_buf.len
131
132struct rpc_xprt_ops {
133 void (*set_buffer_size)(struct rpc_xprt *xprt, size_t sndsize, size_t rcvsize);
134 int (*reserve_xprt)(struct rpc_xprt *xprt, struct rpc_task *task);
135 void (*release_xprt)(struct rpc_xprt *xprt, struct rpc_task *task);
136 void (*alloc_slot)(struct rpc_xprt *xprt, struct rpc_task *task);
137 void (*free_slot)(struct rpc_xprt *xprt,
138 struct rpc_rqst *req);
139 void (*rpcbind)(struct rpc_task *task);
140 void (*set_port)(struct rpc_xprt *xprt, unsigned short port);
141 void (*connect)(struct rpc_xprt *xprt, struct rpc_task *task);
142 int (*buf_alloc)(struct rpc_task *task);
143 void (*buf_free)(struct rpc_task *task);
144 void (*prepare_request)(struct rpc_rqst *req);
145 int (*send_request)(struct rpc_rqst *req);
146 void (*set_retrans_timeout)(struct rpc_task *task);
147 void (*timer)(struct rpc_xprt *xprt, struct rpc_task *task);
148 void (*release_request)(struct rpc_task *task);
149 void (*close)(struct rpc_xprt *xprt);
150 void (*destroy)(struct rpc_xprt *xprt);
151 void (*set_connect_timeout)(struct rpc_xprt *xprt,
152 unsigned long connect_timeout,
153 unsigned long reconnect_timeout);
154 void (*print_stats)(struct rpc_xprt *xprt, struct seq_file *seq);
155 int (*enable_swap)(struct rpc_xprt *xprt);
156 void (*disable_swap)(struct rpc_xprt *xprt);
157 void (*inject_disconnect)(struct rpc_xprt *xprt);
158 int (*bc_setup)(struct rpc_xprt *xprt,
159 unsigned int min_reqs);
160 size_t (*bc_maxpayload)(struct rpc_xprt *xprt);
161 void (*bc_free_rqst)(struct rpc_rqst *rqst);
162 void (*bc_destroy)(struct rpc_xprt *xprt,
163 unsigned int max_reqs);
164};
165
166
167
168
169
170
171
172
173
174
175#define XPRT_TRANSPORT_BC (1 << 31)
176enum xprt_transports {
177 XPRT_TRANSPORT_UDP = IPPROTO_UDP,
178 XPRT_TRANSPORT_TCP = IPPROTO_TCP,
179 XPRT_TRANSPORT_BC_TCP = IPPROTO_TCP | XPRT_TRANSPORT_BC,
180 XPRT_TRANSPORT_RDMA = 256,
181 XPRT_TRANSPORT_BC_RDMA = XPRT_TRANSPORT_RDMA | XPRT_TRANSPORT_BC,
182 XPRT_TRANSPORT_LOCAL = 257,
183};
184
185struct rpc_xprt {
186 struct kref kref;
187 const struct rpc_xprt_ops *ops;
188
189 const struct rpc_timeout *timeout;
190 struct sockaddr_storage addr;
191 size_t addrlen;
192 int prot;
193
194 unsigned long cong;
195 unsigned long cwnd;
196
197 size_t max_payload;
198
199
200 struct rpc_wait_queue binding;
201 struct rpc_wait_queue sending;
202 struct rpc_wait_queue pending;
203 struct rpc_wait_queue backlog;
204 struct list_head free;
205 unsigned int max_reqs;
206 unsigned int min_reqs;
207 unsigned int num_reqs;
208 unsigned long state;
209 unsigned char resvport : 1;
210 atomic_t swapper;
211
212 unsigned int bind_index;
213
214
215
216
217 struct list_head xprt_switch;
218
219
220
221
222 unsigned long bind_timeout,
223 reestablish_timeout;
224 unsigned int connect_cookie;
225
226
227
228
229
230
231 struct work_struct task_cleanup;
232 struct timer_list timer;
233 unsigned long last_used,
234 idle_timeout,
235 connect_timeout,
236 max_reconnect_timeout;
237
238
239
240
241 spinlock_t transport_lock;
242 spinlock_t reserve_lock;
243 spinlock_t queue_lock;
244 u32 xid;
245 struct rpc_task * snd_task;
246
247 struct list_head xmit_queue;
248
249 struct svc_xprt *bc_xprt;
250#if defined(CONFIG_SUNRPC_BACKCHANNEL)
251 struct svc_serv *bc_serv;
252
253 int bc_alloc_count;
254 atomic_t bc_free_slots;
255 spinlock_t bc_pa_lock;
256
257 struct list_head bc_pa_list;
258
259#endif
260
261 struct rb_root recv_queue;
262
263 struct {
264 unsigned long bind_count,
265 connect_count,
266 connect_start,
267 connect_time,
268 sends,
269 recvs,
270 bad_xids,
271 max_slots;
272
273 unsigned long long req_u,
274 bklog_u,
275 sending_u,
276 pending_u;
277 } stat;
278
279 struct net *xprt_net;
280 const char *servername;
281 const char *address_strings[RPC_DISPLAY_MAX];
282#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
283 struct dentry *debugfs;
284 atomic_t inject_disconnect;
285#endif
286 struct rcu_head rcu;
287};
288
289#if defined(CONFIG_SUNRPC_BACKCHANNEL)
290
291
292
293#define RPC_BC_PA_IN_USE 0x0001
294
295#endif
296
297#if defined(CONFIG_SUNRPC_BACKCHANNEL)
298static inline int bc_prealloc(struct rpc_rqst *req)
299{
300 return test_bit(RPC_BC_PA_IN_USE, &req->rq_bc_pa_state);
301}
302#else
303static inline int bc_prealloc(struct rpc_rqst *req)
304{
305 return 0;
306}
307#endif
308
309#define XPRT_CREATE_INFINITE_SLOTS (1U)
310#define XPRT_CREATE_NO_IDLE_TIMEOUT (1U << 1)
311
312struct xprt_create {
313 int ident;
314 struct net * net;
315 struct sockaddr * srcaddr;
316 struct sockaddr * dstaddr;
317 size_t addrlen;
318 const char *servername;
319 struct svc_xprt *bc_xprt;
320 struct rpc_xprt_switch *bc_xps;
321 unsigned int flags;
322};
323
324struct xprt_class {
325 struct list_head list;
326 int ident;
327 struct rpc_xprt * (*setup)(struct xprt_create *);
328 struct module *owner;
329 char name[32];
330};
331
332
333
334
335struct rpc_xprt *xprt_create_transport(struct xprt_create *args);
336void xprt_connect(struct rpc_task *task);
337unsigned long xprt_reconnect_delay(const struct rpc_xprt *xprt);
338void xprt_reconnect_backoff(struct rpc_xprt *xprt,
339 unsigned long init_to);
340void xprt_reserve(struct rpc_task *task);
341void xprt_retry_reserve(struct rpc_task *task);
342int xprt_reserve_xprt(struct rpc_xprt *xprt, struct rpc_task *task);
343int xprt_reserve_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task);
344void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task);
345void xprt_free_slot(struct rpc_xprt *xprt,
346 struct rpc_rqst *req);
347void xprt_request_prepare(struct rpc_rqst *req);
348bool xprt_prepare_transmit(struct rpc_task *task);
349void xprt_request_enqueue_transmit(struct rpc_task *task);
350void xprt_request_enqueue_receive(struct rpc_task *task);
351void xprt_request_wait_receive(struct rpc_task *task);
352void xprt_request_dequeue_xprt(struct rpc_task *task);
353bool xprt_request_need_retransmit(struct rpc_task *task);
354void xprt_transmit(struct rpc_task *task);
355void xprt_end_transmit(struct rpc_task *task);
356int xprt_adjust_timeout(struct rpc_rqst *req);
357void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task);
358void xprt_release_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task);
359void xprt_release(struct rpc_task *task);
360struct rpc_xprt * xprt_get(struct rpc_xprt *xprt);
361void xprt_put(struct rpc_xprt *xprt);
362struct rpc_xprt * xprt_alloc(struct net *net, size_t size,
363 unsigned int num_prealloc,
364 unsigned int max_req);
365void xprt_free(struct rpc_xprt *);
366
367static inline int
368xprt_enable_swap(struct rpc_xprt *xprt)
369{
370 return xprt->ops->enable_swap(xprt);
371}
372
373static inline void
374xprt_disable_swap(struct rpc_xprt *xprt)
375{
376 xprt->ops->disable_swap(xprt);
377}
378
379
380
381
382int xprt_register_transport(struct xprt_class *type);
383int xprt_unregister_transport(struct xprt_class *type);
384int xprt_load_transport(const char *);
385void xprt_set_retrans_timeout_def(struct rpc_task *task);
386void xprt_set_retrans_timeout_rtt(struct rpc_task *task);
387void xprt_wake_pending_tasks(struct rpc_xprt *xprt, int status);
388void xprt_wait_for_buffer_space(struct rpc_xprt *xprt);
389bool xprt_write_space(struct rpc_xprt *xprt);
390void xprt_adjust_cwnd(struct rpc_xprt *xprt, struct rpc_task *task, int result);
391struct rpc_rqst * xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid);
392void xprt_update_rtt(struct rpc_task *task);
393void xprt_complete_rqst(struct rpc_task *task, int copied);
394void xprt_pin_rqst(struct rpc_rqst *req);
395void xprt_unpin_rqst(struct rpc_rqst *req);
396void xprt_release_rqst_cong(struct rpc_task *task);
397bool xprt_request_get_cong(struct rpc_xprt *xprt, struct rpc_rqst *req);
398void xprt_disconnect_done(struct rpc_xprt *xprt);
399void xprt_force_disconnect(struct rpc_xprt *xprt);
400void xprt_conditional_disconnect(struct rpc_xprt *xprt, unsigned int cookie);
401
402bool xprt_lock_connect(struct rpc_xprt *, struct rpc_task *, void *);
403void xprt_unlock_connect(struct rpc_xprt *, void *);
404
405
406
407
408#define XPRT_LOCKED (0)
409#define XPRT_CONNECTED (1)
410#define XPRT_CONNECTING (2)
411#define XPRT_CLOSE_WAIT (3)
412#define XPRT_BOUND (4)
413#define XPRT_BINDING (5)
414#define XPRT_CLOSING (6)
415#define XPRT_CONGESTED (9)
416#define XPRT_CWND_WAIT (10)
417#define XPRT_WRITE_SPACE (11)
418
419static inline void xprt_set_connected(struct rpc_xprt *xprt)
420{
421 set_bit(XPRT_CONNECTED, &xprt->state);
422}
423
424static inline void xprt_clear_connected(struct rpc_xprt *xprt)
425{
426 clear_bit(XPRT_CONNECTED, &xprt->state);
427}
428
429static inline int xprt_connected(struct rpc_xprt *xprt)
430{
431 return test_bit(XPRT_CONNECTED, &xprt->state);
432}
433
434static inline int xprt_test_and_set_connected(struct rpc_xprt *xprt)
435{
436 return test_and_set_bit(XPRT_CONNECTED, &xprt->state);
437}
438
439static inline int xprt_test_and_clear_connected(struct rpc_xprt *xprt)
440{
441 return test_and_clear_bit(XPRT_CONNECTED, &xprt->state);
442}
443
444static inline void xprt_clear_connecting(struct rpc_xprt *xprt)
445{
446 smp_mb__before_atomic();
447 clear_bit(XPRT_CONNECTING, &xprt->state);
448 smp_mb__after_atomic();
449}
450
451static inline int xprt_connecting(struct rpc_xprt *xprt)
452{
453 return test_bit(XPRT_CONNECTING, &xprt->state);
454}
455
456static inline int xprt_test_and_set_connecting(struct rpc_xprt *xprt)
457{
458 return test_and_set_bit(XPRT_CONNECTING, &xprt->state);
459}
460
461static inline void xprt_set_bound(struct rpc_xprt *xprt)
462{
463 test_and_set_bit(XPRT_BOUND, &xprt->state);
464}
465
466static inline int xprt_bound(struct rpc_xprt *xprt)
467{
468 return test_bit(XPRT_BOUND, &xprt->state);
469}
470
471static inline void xprt_clear_bound(struct rpc_xprt *xprt)
472{
473 clear_bit(XPRT_BOUND, &xprt->state);
474}
475
476static inline void xprt_clear_binding(struct rpc_xprt *xprt)
477{
478 smp_mb__before_atomic();
479 clear_bit(XPRT_BINDING, &xprt->state);
480 smp_mb__after_atomic();
481}
482
483static inline int xprt_test_and_set_binding(struct rpc_xprt *xprt)
484{
485 return test_and_set_bit(XPRT_BINDING, &xprt->state);
486}
487
488#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
489extern unsigned int rpc_inject_disconnect;
490static inline void xprt_inject_disconnect(struct rpc_xprt *xprt)
491{
492 if (!rpc_inject_disconnect)
493 return;
494 if (atomic_dec_return(&xprt->inject_disconnect))
495 return;
496 atomic_set(&xprt->inject_disconnect, rpc_inject_disconnect);
497 xprt->ops->inject_disconnect(xprt);
498}
499#else
500static inline void xprt_inject_disconnect(struct rpc_xprt *xprt)
501{
502}
503#endif
504
505#endif
506
507#endif
508