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