]>
Commit | Line | Data |
---|---|---|
1da177e4 LT |
1 | /* |
2 | * linux/net/sunrpc/xprt.c | |
3 | * | |
4 | * This is a generic RPC call interface supporting congestion avoidance, | |
5 | * and asynchronous calls. | |
6 | * | |
7 | * The interface works like this: | |
8 | * | |
9 | * - When a process places a call, it allocates a request slot if | |
10 | * one is available. Otherwise, it sleeps on the backlog queue | |
11 | * (xprt_reserve). | |
12 | * - Next, the caller puts together the RPC message, stuffs it into | |
55aa4f58 CL |
13 | * the request struct, and calls xprt_transmit(). |
14 | * - xprt_transmit sends the message and installs the caller on the | |
55ae1aab RL |
15 | * transport's wait list. At the same time, if a reply is expected, |
16 | * it installs a timer that is run after the packet's timeout has | |
17 | * expired. | |
1da177e4 | 18 | * - When a packet arrives, the data_ready handler walks the list of |
55aa4f58 | 19 | * pending requests for that transport. If a matching XID is found, the |
1da177e4 LT |
20 | * caller is woken up, and the timer removed. |
21 | * - When no reply arrives within the timeout interval, the timer is | |
22 | * fired by the kernel and runs xprt_timer(). It either adjusts the | |
23 | * timeout values (minor timeout) or wakes up the caller with a status | |
24 | * of -ETIMEDOUT. | |
25 | * - When the caller receives a notification from RPC that a reply arrived, | |
26 | * it should release the RPC slot, and process the reply. | |
27 | * If the call timed out, it may choose to retry the operation by | |
28 | * adjusting the initial timeout value, and simply calling rpc_call | |
29 | * again. | |
30 | * | |
31 | * Support for async RPC is done through a set of RPC-specific scheduling | |
32 | * primitives that `transparently' work for processes as well as async | |
33 | * tasks that rely on callbacks. | |
34 | * | |
35 | * Copyright (C) 1995-1997, Olaf Kirch <okir@monad.swb.de> | |
55aa4f58 CL |
36 | * |
37 | * Transport switch API copyright (C) 2005, Chuck Lever <cel@netapp.com> | |
1da177e4 LT |
38 | */ |
39 | ||
a246b010 CL |
40 | #include <linux/module.h> |
41 | ||
1da177e4 | 42 | #include <linux/types.h> |
a246b010 | 43 | #include <linux/interrupt.h> |
1da177e4 | 44 | #include <linux/workqueue.h> |
bf3fcf89 | 45 | #include <linux/net.h> |
ff839970 | 46 | #include <linux/ktime.h> |
1da177e4 | 47 | |
a246b010 | 48 | #include <linux/sunrpc/clnt.h> |
11c556b3 | 49 | #include <linux/sunrpc/metrics.h> |
c9acb42e | 50 | #include <linux/sunrpc/bc_xprt.h> |
1da177e4 | 51 | |
55ae1aab RL |
52 | #include "sunrpc.h" |
53 | ||
1da177e4 LT |
54 | /* |
55 | * Local variables | |
56 | */ | |
57 | ||
58 | #ifdef RPC_DEBUG | |
1da177e4 LT |
59 | # define RPCDBG_FACILITY RPCDBG_XPRT |
60 | #endif | |
61 | ||
1da177e4 LT |
62 | /* |
63 | * Local functions | |
64 | */ | |
21de0a95 | 65 | static void xprt_init(struct rpc_xprt *xprt, struct net *net); |
1da177e4 | 66 | static void xprt_request_init(struct rpc_task *, struct rpc_xprt *); |
1da177e4 | 67 | static void xprt_connect_status(struct rpc_task *task); |
1da177e4 | 68 | static int __xprt_get_cong(struct rpc_xprt *, struct rpc_task *); |
4e0038b6 | 69 | static void xprt_destroy(struct rpc_xprt *xprt); |
1da177e4 | 70 | |
5ba03e82 | 71 | static DEFINE_SPINLOCK(xprt_list_lock); |
81c098af TT |
72 | static LIST_HEAD(xprt_list); |
73 | ||
555ee3af CL |
74 | /* |
75 | * The transport code maintains an estimate on the maximum number of out- | |
76 | * standing RPC requests, using a smoothed version of the congestion | |
77 | * avoidance implemented in 44BSD. This is basically the Van Jacobson | |
78 | * congestion algorithm: If a retransmit occurs, the congestion window is | |
79 | * halved; otherwise, it is incremented by 1/cwnd when | |
80 | * | |
81 | * - a reply is received and | |
82 | * - a full number of requests are outstanding and | |
83 | * - the congestion window hasn't been updated recently. | |
84 | */ | |
85 | #define RPC_CWNDSHIFT (8U) | |
86 | #define RPC_CWNDSCALE (1U << RPC_CWNDSHIFT) | |
87 | #define RPC_INITCWND RPC_CWNDSCALE | |
88 | #define RPC_MAXCWND(xprt) ((xprt)->max_reqs << RPC_CWNDSHIFT) | |
89 | ||
90 | #define RPCXPRT_CONGESTED(xprt) ((xprt)->cong >= (xprt)->cwnd) | |
1da177e4 | 91 | |
81c098af TT |
92 | /** |
93 | * xprt_register_transport - register a transport implementation | |
94 | * @transport: transport to register | |
95 | * | |
96 | * If a transport implementation is loaded as a kernel module, it can | |
97 | * call this interface to make itself known to the RPC client. | |
98 | * | |
99 | * Returns: | |
100 | * 0: transport successfully registered | |
101 | * -EEXIST: transport already registered | |
102 | * -EINVAL: transport module being unloaded | |
103 | */ | |
104 | int xprt_register_transport(struct xprt_class *transport) | |
105 | { | |
106 | struct xprt_class *t; | |
107 | int result; | |
108 | ||
109 | result = -EEXIST; | |
110 | spin_lock(&xprt_list_lock); | |
111 | list_for_each_entry(t, &xprt_list, list) { | |
112 | /* don't register the same transport class twice */ | |
4fa016eb | 113 | if (t->ident == transport->ident) |
81c098af TT |
114 | goto out; |
115 | } | |
116 | ||
c9f6cde6 DL |
117 | list_add_tail(&transport->list, &xprt_list); |
118 | printk(KERN_INFO "RPC: Registered %s transport module.\n", | |
119 | transport->name); | |
120 | result = 0; | |
81c098af TT |
121 | |
122 | out: | |
123 | spin_unlock(&xprt_list_lock); | |
124 | return result; | |
125 | } | |
126 | EXPORT_SYMBOL_GPL(xprt_register_transport); | |
127 | ||
128 | /** | |
129 | * xprt_unregister_transport - unregister a transport implementation | |
65b6e42c | 130 | * @transport: transport to unregister |
81c098af TT |
131 | * |
132 | * Returns: | |
133 | * 0: transport successfully unregistered | |
134 | * -ENOENT: transport never registered | |
135 | */ | |
136 | int xprt_unregister_transport(struct xprt_class *transport) | |
137 | { | |
138 | struct xprt_class *t; | |
139 | int result; | |
140 | ||
141 | result = 0; | |
142 | spin_lock(&xprt_list_lock); | |
143 | list_for_each_entry(t, &xprt_list, list) { | |
144 | if (t == transport) { | |
145 | printk(KERN_INFO | |
146 | "RPC: Unregistered %s transport module.\n", | |
147 | transport->name); | |
148 | list_del_init(&transport->list); | |
81c098af TT |
149 | goto out; |
150 | } | |
151 | } | |
152 | result = -ENOENT; | |
153 | ||
154 | out: | |
155 | spin_unlock(&xprt_list_lock); | |
156 | return result; | |
157 | } | |
158 | EXPORT_SYMBOL_GPL(xprt_unregister_transport); | |
159 | ||
441e3e24 TT |
160 | /** |
161 | * xprt_load_transport - load a transport implementation | |
162 | * @transport_name: transport to load | |
163 | * | |
164 | * Returns: | |
165 | * 0: transport successfully loaded | |
166 | * -ENOENT: transport module not available | |
167 | */ | |
168 | int xprt_load_transport(const char *transport_name) | |
169 | { | |
170 | struct xprt_class *t; | |
441e3e24 TT |
171 | int result; |
172 | ||
173 | result = 0; | |
174 | spin_lock(&xprt_list_lock); | |
175 | list_for_each_entry(t, &xprt_list, list) { | |
176 | if (strcmp(t->name, transport_name) == 0) { | |
177 | spin_unlock(&xprt_list_lock); | |
178 | goto out; | |
179 | } | |
180 | } | |
181 | spin_unlock(&xprt_list_lock); | |
ef7ffe8f | 182 | result = request_module("xprt%s", transport_name); |
441e3e24 TT |
183 | out: |
184 | return result; | |
185 | } | |
186 | EXPORT_SYMBOL_GPL(xprt_load_transport); | |
187 | ||
12a80469 CL |
188 | /** |
189 | * xprt_reserve_xprt - serialize write access to transports | |
190 | * @task: task that is requesting access to the transport | |
177c27bf | 191 | * @xprt: pointer to the target transport |
12a80469 CL |
192 | * |
193 | * This prevents mixing the payload of separate requests, and prevents | |
194 | * transport connects from colliding with writes. No congestion control | |
195 | * is provided. | |
196 | */ | |
43cedbf0 | 197 | int xprt_reserve_xprt(struct rpc_xprt *xprt, struct rpc_task *task) |
12a80469 | 198 | { |
12a80469 | 199 | struct rpc_rqst *req = task->tk_rqstp; |
34006cee | 200 | int priority; |
12a80469 CL |
201 | |
202 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) { | |
203 | if (task == xprt->snd_task) | |
204 | return 1; | |
12a80469 CL |
205 | goto out_sleep; |
206 | } | |
207 | xprt->snd_task = task; | |
92551948 | 208 | if (req != NULL) |
43cedbf0 | 209 | req->rq_ntrans++; |
4d4a76f3 | 210 | |
12a80469 CL |
211 | return 1; |
212 | ||
213 | out_sleep: | |
46121cf7 | 214 | dprintk("RPC: %5u failed to lock transport %p\n", |
12a80469 CL |
215 | task->tk_pid, xprt); |
216 | task->tk_timeout = 0; | |
217 | task->tk_status = -EAGAIN; | |
34006cee TM |
218 | if (req == NULL) |
219 | priority = RPC_PRIORITY_LOW; | |
220 | else if (!req->rq_ntrans) | |
221 | priority = RPC_PRIORITY_NORMAL; | |
12a80469 | 222 | else |
34006cee TM |
223 | priority = RPC_PRIORITY_HIGH; |
224 | rpc_sleep_on_priority(&xprt->sending, task, NULL, priority); | |
12a80469 CL |
225 | return 0; |
226 | } | |
12444809 | 227 | EXPORT_SYMBOL_GPL(xprt_reserve_xprt); |
12a80469 | 228 | |
632e3bdc TM |
229 | static void xprt_clear_locked(struct rpc_xprt *xprt) |
230 | { | |
231 | xprt->snd_task = NULL; | |
d19751e7 | 232 | if (!test_bit(XPRT_CLOSE_WAIT, &xprt->state)) { |
632e3bdc TM |
233 | smp_mb__before_clear_bit(); |
234 | clear_bit(XPRT_LOCKED, &xprt->state); | |
235 | smp_mb__after_clear_bit(); | |
236 | } else | |
c1384c9c | 237 | queue_work(rpciod_workqueue, &xprt->task_cleanup); |
632e3bdc TM |
238 | } |
239 | ||
1da177e4 | 240 | /* |
12a80469 CL |
241 | * xprt_reserve_xprt_cong - serialize write access to transports |
242 | * @task: task that is requesting access to the transport | |
243 | * | |
244 | * Same as xprt_reserve_xprt, but Van Jacobson congestion control is | |
245 | * integrated into the decision of whether a request is allowed to be | |
246 | * woken up and given access to the transport. | |
1da177e4 | 247 | */ |
43cedbf0 | 248 | int xprt_reserve_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task) |
1da177e4 LT |
249 | { |
250 | struct rpc_rqst *req = task->tk_rqstp; | |
34006cee | 251 | int priority; |
1da177e4 | 252 | |
2226feb6 | 253 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) { |
1da177e4 LT |
254 | if (task == xprt->snd_task) |
255 | return 1; | |
1da177e4 LT |
256 | goto out_sleep; |
257 | } | |
43cedbf0 TM |
258 | if (req == NULL) { |
259 | xprt->snd_task = task; | |
260 | return 1; | |
261 | } | |
12a80469 | 262 | if (__xprt_get_cong(xprt, task)) { |
1da177e4 | 263 | xprt->snd_task = task; |
43cedbf0 | 264 | req->rq_ntrans++; |
1da177e4 LT |
265 | return 1; |
266 | } | |
632e3bdc | 267 | xprt_clear_locked(xprt); |
1da177e4 | 268 | out_sleep: |
46121cf7 | 269 | dprintk("RPC: %5u failed to lock transport %p\n", task->tk_pid, xprt); |
1da177e4 LT |
270 | task->tk_timeout = 0; |
271 | task->tk_status = -EAGAIN; | |
34006cee TM |
272 | if (req == NULL) |
273 | priority = RPC_PRIORITY_LOW; | |
274 | else if (!req->rq_ntrans) | |
275 | priority = RPC_PRIORITY_NORMAL; | |
1da177e4 | 276 | else |
34006cee TM |
277 | priority = RPC_PRIORITY_HIGH; |
278 | rpc_sleep_on_priority(&xprt->sending, task, NULL, priority); | |
1da177e4 LT |
279 | return 0; |
280 | } | |
12444809 | 281 | EXPORT_SYMBOL_GPL(xprt_reserve_xprt_cong); |
1da177e4 | 282 | |
12a80469 | 283 | static inline int xprt_lock_write(struct rpc_xprt *xprt, struct rpc_task *task) |
1da177e4 LT |
284 | { |
285 | int retval; | |
286 | ||
4a0f8c04 | 287 | spin_lock_bh(&xprt->transport_lock); |
43cedbf0 | 288 | retval = xprt->ops->reserve_xprt(xprt, task); |
4a0f8c04 | 289 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 LT |
290 | return retval; |
291 | } | |
292 | ||
961a828d | 293 | static bool __xprt_lock_write_func(struct rpc_task *task, void *data) |
49e9a890 | 294 | { |
961a828d | 295 | struct rpc_xprt *xprt = data; |
49e9a890 CL |
296 | struct rpc_rqst *req; |
297 | ||
49e9a890 CL |
298 | req = task->tk_rqstp; |
299 | xprt->snd_task = task; | |
92551948 | 300 | if (req) |
49e9a890 | 301 | req->rq_ntrans++; |
961a828d TM |
302 | return true; |
303 | } | |
49e9a890 | 304 | |
961a828d TM |
305 | static void __xprt_lock_write_next(struct rpc_xprt *xprt) |
306 | { | |
307 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) | |
308 | return; | |
309 | ||
310 | if (rpc_wake_up_first(&xprt->sending, __xprt_lock_write_func, xprt)) | |
311 | return; | |
632e3bdc | 312 | xprt_clear_locked(xprt); |
49e9a890 CL |
313 | } |
314 | ||
961a828d | 315 | static bool __xprt_lock_write_cong_func(struct rpc_task *task, void *data) |
1da177e4 | 316 | { |
961a828d | 317 | struct rpc_xprt *xprt = data; |
43cedbf0 | 318 | struct rpc_rqst *req; |
1da177e4 | 319 | |
43cedbf0 TM |
320 | req = task->tk_rqstp; |
321 | if (req == NULL) { | |
322 | xprt->snd_task = task; | |
961a828d | 323 | return true; |
43cedbf0 | 324 | } |
49e9a890 | 325 | if (__xprt_get_cong(xprt, task)) { |
1da177e4 | 326 | xprt->snd_task = task; |
43cedbf0 | 327 | req->rq_ntrans++; |
961a828d | 328 | return true; |
1da177e4 | 329 | } |
961a828d TM |
330 | return false; |
331 | } | |
332 | ||
333 | static void __xprt_lock_write_next_cong(struct rpc_xprt *xprt) | |
334 | { | |
335 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) | |
336 | return; | |
337 | if (RPCXPRT_CONGESTED(xprt)) | |
338 | goto out_unlock; | |
339 | if (rpc_wake_up_first(&xprt->sending, __xprt_lock_write_cong_func, xprt)) | |
340 | return; | |
1da177e4 | 341 | out_unlock: |
632e3bdc | 342 | xprt_clear_locked(xprt); |
1da177e4 LT |
343 | } |
344 | ||
49e9a890 CL |
345 | /** |
346 | * xprt_release_xprt - allow other requests to use a transport | |
347 | * @xprt: transport with other tasks potentially waiting | |
348 | * @task: task that is releasing access to the transport | |
349 | * | |
350 | * Note that "task" can be NULL. No congestion control is provided. | |
1da177e4 | 351 | */ |
49e9a890 | 352 | void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task) |
1da177e4 LT |
353 | { |
354 | if (xprt->snd_task == task) { | |
ee071eff TM |
355 | if (task != NULL) { |
356 | struct rpc_rqst *req = task->tk_rqstp; | |
357 | if (req != NULL) | |
358 | req->rq_bytes_sent = 0; | |
359 | } | |
632e3bdc | 360 | xprt_clear_locked(xprt); |
1da177e4 LT |
361 | __xprt_lock_write_next(xprt); |
362 | } | |
363 | } | |
12444809 | 364 | EXPORT_SYMBOL_GPL(xprt_release_xprt); |
1da177e4 | 365 | |
49e9a890 CL |
366 | /** |
367 | * xprt_release_xprt_cong - allow other requests to use a transport | |
368 | * @xprt: transport with other tasks potentially waiting | |
369 | * @task: task that is releasing access to the transport | |
370 | * | |
371 | * Note that "task" can be NULL. Another task is awoken to use the | |
372 | * transport if the transport's congestion window allows it. | |
373 | */ | |
374 | void xprt_release_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task) | |
375 | { | |
376 | if (xprt->snd_task == task) { | |
ee071eff TM |
377 | if (task != NULL) { |
378 | struct rpc_rqst *req = task->tk_rqstp; | |
379 | if (req != NULL) | |
380 | req->rq_bytes_sent = 0; | |
381 | } | |
632e3bdc | 382 | xprt_clear_locked(xprt); |
49e9a890 CL |
383 | __xprt_lock_write_next_cong(xprt); |
384 | } | |
385 | } | |
12444809 | 386 | EXPORT_SYMBOL_GPL(xprt_release_xprt_cong); |
49e9a890 CL |
387 | |
388 | static inline void xprt_release_write(struct rpc_xprt *xprt, struct rpc_task *task) | |
1da177e4 | 389 | { |
4a0f8c04 | 390 | spin_lock_bh(&xprt->transport_lock); |
49e9a890 | 391 | xprt->ops->release_xprt(xprt, task); |
4a0f8c04 | 392 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 LT |
393 | } |
394 | ||
1da177e4 LT |
395 | /* |
396 | * Van Jacobson congestion avoidance. Check if the congestion window | |
397 | * overflowed. Put the task to sleep if this is the case. | |
398 | */ | |
399 | static int | |
400 | __xprt_get_cong(struct rpc_xprt *xprt, struct rpc_task *task) | |
401 | { | |
402 | struct rpc_rqst *req = task->tk_rqstp; | |
403 | ||
404 | if (req->rq_cong) | |
405 | return 1; | |
46121cf7 | 406 | dprintk("RPC: %5u xprt_cwnd_limited cong = %lu cwnd = %lu\n", |
1da177e4 LT |
407 | task->tk_pid, xprt->cong, xprt->cwnd); |
408 | if (RPCXPRT_CONGESTED(xprt)) | |
409 | return 0; | |
410 | req->rq_cong = 1; | |
411 | xprt->cong += RPC_CWNDSCALE; | |
412 | return 1; | |
413 | } | |
414 | ||
415 | /* | |
416 | * Adjust the congestion window, and wake up the next task | |
417 | * that has been sleeping due to congestion | |
418 | */ | |
419 | static void | |
420 | __xprt_put_cong(struct rpc_xprt *xprt, struct rpc_rqst *req) | |
421 | { | |
422 | if (!req->rq_cong) | |
423 | return; | |
424 | req->rq_cong = 0; | |
425 | xprt->cong -= RPC_CWNDSCALE; | |
49e9a890 | 426 | __xprt_lock_write_next_cong(xprt); |
1da177e4 LT |
427 | } |
428 | ||
a58dd398 CL |
429 | /** |
430 | * xprt_release_rqst_cong - housekeeping when request is complete | |
431 | * @task: RPC request that recently completed | |
432 | * | |
433 | * Useful for transports that require congestion control. | |
434 | */ | |
435 | void xprt_release_rqst_cong(struct rpc_task *task) | |
436 | { | |
a4f0835c TM |
437 | struct rpc_rqst *req = task->tk_rqstp; |
438 | ||
439 | __xprt_put_cong(req->rq_xprt, req); | |
a58dd398 | 440 | } |
12444809 | 441 | EXPORT_SYMBOL_GPL(xprt_release_rqst_cong); |
a58dd398 | 442 | |
46c0ee8b CL |
443 | /** |
444 | * xprt_adjust_cwnd - adjust transport congestion window | |
6a24dfb6 | 445 | * @xprt: pointer to xprt |
46c0ee8b CL |
446 | * @task: recently completed RPC request used to adjust window |
447 | * @result: result code of completed RPC request | |
448 | * | |
1da177e4 LT |
449 | * We use a time-smoothed congestion estimator to avoid heavy oscillation. |
450 | */ | |
6a24dfb6 | 451 | void xprt_adjust_cwnd(struct rpc_xprt *xprt, struct rpc_task *task, int result) |
1da177e4 | 452 | { |
46c0ee8b | 453 | struct rpc_rqst *req = task->tk_rqstp; |
46c0ee8b | 454 | unsigned long cwnd = xprt->cwnd; |
1da177e4 | 455 | |
1da177e4 LT |
456 | if (result >= 0 && cwnd <= xprt->cong) { |
457 | /* The (cwnd >> 1) term makes sure | |
458 | * the result gets rounded properly. */ | |
459 | cwnd += (RPC_CWNDSCALE * RPC_CWNDSCALE + (cwnd >> 1)) / cwnd; | |
460 | if (cwnd > RPC_MAXCWND(xprt)) | |
461 | cwnd = RPC_MAXCWND(xprt); | |
49e9a890 | 462 | __xprt_lock_write_next_cong(xprt); |
1da177e4 LT |
463 | } else if (result == -ETIMEDOUT) { |
464 | cwnd >>= 1; | |
465 | if (cwnd < RPC_CWNDSCALE) | |
466 | cwnd = RPC_CWNDSCALE; | |
467 | } | |
46121cf7 | 468 | dprintk("RPC: cong %ld, cwnd was %ld, now %ld\n", |
1da177e4 LT |
469 | xprt->cong, xprt->cwnd, cwnd); |
470 | xprt->cwnd = cwnd; | |
46c0ee8b | 471 | __xprt_put_cong(xprt, req); |
1da177e4 | 472 | } |
12444809 | 473 | EXPORT_SYMBOL_GPL(xprt_adjust_cwnd); |
1da177e4 | 474 | |
44fbac22 CL |
475 | /** |
476 | * xprt_wake_pending_tasks - wake all tasks on a transport's pending queue | |
477 | * @xprt: transport with waiting tasks | |
478 | * @status: result code to plant in each task before waking it | |
479 | * | |
480 | */ | |
481 | void xprt_wake_pending_tasks(struct rpc_xprt *xprt, int status) | |
482 | { | |
483 | if (status < 0) | |
484 | rpc_wake_up_status(&xprt->pending, status); | |
485 | else | |
486 | rpc_wake_up(&xprt->pending); | |
487 | } | |
12444809 | 488 | EXPORT_SYMBOL_GPL(xprt_wake_pending_tasks); |
44fbac22 | 489 | |
c7b2cae8 CL |
490 | /** |
491 | * xprt_wait_for_buffer_space - wait for transport output buffer to clear | |
492 | * @task: task to be put to sleep | |
0b80ae42 | 493 | * @action: function pointer to be executed after wait |
a9a6b52e TM |
494 | * |
495 | * Note that we only set the timer for the case of RPC_IS_SOFT(), since | |
496 | * we don't in general want to force a socket disconnection due to | |
497 | * an incomplete RPC call transmission. | |
c7b2cae8 | 498 | */ |
b6ddf64f | 499 | void xprt_wait_for_buffer_space(struct rpc_task *task, rpc_action action) |
c7b2cae8 CL |
500 | { |
501 | struct rpc_rqst *req = task->tk_rqstp; | |
502 | struct rpc_xprt *xprt = req->rq_xprt; | |
503 | ||
a9a6b52e | 504 | task->tk_timeout = RPC_IS_SOFT(task) ? req->rq_timeout : 0; |
b6ddf64f | 505 | rpc_sleep_on(&xprt->pending, task, action); |
c7b2cae8 | 506 | } |
12444809 | 507 | EXPORT_SYMBOL_GPL(xprt_wait_for_buffer_space); |
c7b2cae8 CL |
508 | |
509 | /** | |
510 | * xprt_write_space - wake the task waiting for transport output buffer space | |
511 | * @xprt: transport with waiting tasks | |
512 | * | |
513 | * Can be called in a soft IRQ context, so xprt_write_space never sleeps. | |
514 | */ | |
515 | void xprt_write_space(struct rpc_xprt *xprt) | |
516 | { | |
c7b2cae8 CL |
517 | spin_lock_bh(&xprt->transport_lock); |
518 | if (xprt->snd_task) { | |
46121cf7 CL |
519 | dprintk("RPC: write space: waking waiting task on " |
520 | "xprt %p\n", xprt); | |
fda13939 | 521 | rpc_wake_up_queued_task(&xprt->pending, xprt->snd_task); |
c7b2cae8 CL |
522 | } |
523 | spin_unlock_bh(&xprt->transport_lock); | |
524 | } | |
12444809 | 525 | EXPORT_SYMBOL_GPL(xprt_write_space); |
c7b2cae8 | 526 | |
fe3aca29 CL |
527 | /** |
528 | * xprt_set_retrans_timeout_def - set a request's retransmit timeout | |
529 | * @task: task whose timeout is to be set | |
530 | * | |
531 | * Set a request's retransmit timeout based on the transport's | |
532 | * default timeout parameters. Used by transports that don't adjust | |
533 | * the retransmit timeout based on round-trip time estimation. | |
534 | */ | |
535 | void xprt_set_retrans_timeout_def(struct rpc_task *task) | |
536 | { | |
537 | task->tk_timeout = task->tk_rqstp->rq_timeout; | |
538 | } | |
12444809 | 539 | EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_def); |
fe3aca29 | 540 | |
2c53040f | 541 | /** |
fe3aca29 CL |
542 | * xprt_set_retrans_timeout_rtt - set a request's retransmit timeout |
543 | * @task: task whose timeout is to be set | |
cca5172a | 544 | * |
fe3aca29 CL |
545 | * Set a request's retransmit timeout using the RTT estimator. |
546 | */ | |
547 | void xprt_set_retrans_timeout_rtt(struct rpc_task *task) | |
548 | { | |
549 | int timer = task->tk_msg.rpc_proc->p_timer; | |
ba7392bb TM |
550 | struct rpc_clnt *clnt = task->tk_client; |
551 | struct rpc_rtt *rtt = clnt->cl_rtt; | |
fe3aca29 | 552 | struct rpc_rqst *req = task->tk_rqstp; |
ba7392bb | 553 | unsigned long max_timeout = clnt->cl_timeout->to_maxval; |
fe3aca29 CL |
554 | |
555 | task->tk_timeout = rpc_calc_rto(rtt, timer); | |
556 | task->tk_timeout <<= rpc_ntimeo(rtt, timer) + req->rq_retries; | |
557 | if (task->tk_timeout > max_timeout || task->tk_timeout == 0) | |
558 | task->tk_timeout = max_timeout; | |
559 | } | |
12444809 | 560 | EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_rtt); |
fe3aca29 | 561 | |
1da177e4 LT |
562 | static void xprt_reset_majortimeo(struct rpc_rqst *req) |
563 | { | |
ba7392bb | 564 | const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout; |
1da177e4 LT |
565 | |
566 | req->rq_majortimeo = req->rq_timeout; | |
567 | if (to->to_exponential) | |
568 | req->rq_majortimeo <<= to->to_retries; | |
569 | else | |
570 | req->rq_majortimeo += to->to_increment * to->to_retries; | |
571 | if (req->rq_majortimeo > to->to_maxval || req->rq_majortimeo == 0) | |
572 | req->rq_majortimeo = to->to_maxval; | |
573 | req->rq_majortimeo += jiffies; | |
574 | } | |
575 | ||
9903cd1c CL |
576 | /** |
577 | * xprt_adjust_timeout - adjust timeout values for next retransmit | |
578 | * @req: RPC request containing parameters to use for the adjustment | |
579 | * | |
1da177e4 LT |
580 | */ |
581 | int xprt_adjust_timeout(struct rpc_rqst *req) | |
582 | { | |
583 | struct rpc_xprt *xprt = req->rq_xprt; | |
ba7392bb | 584 | const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout; |
1da177e4 LT |
585 | int status = 0; |
586 | ||
587 | if (time_before(jiffies, req->rq_majortimeo)) { | |
588 | if (to->to_exponential) | |
589 | req->rq_timeout <<= 1; | |
590 | else | |
591 | req->rq_timeout += to->to_increment; | |
592 | if (to->to_maxval && req->rq_timeout >= to->to_maxval) | |
593 | req->rq_timeout = to->to_maxval; | |
594 | req->rq_retries++; | |
1da177e4 LT |
595 | } else { |
596 | req->rq_timeout = to->to_initval; | |
597 | req->rq_retries = 0; | |
598 | xprt_reset_majortimeo(req); | |
599 | /* Reset the RTT counters == "slow start" */ | |
4a0f8c04 | 600 | spin_lock_bh(&xprt->transport_lock); |
1da177e4 | 601 | rpc_init_rtt(req->rq_task->tk_client->cl_rtt, to->to_initval); |
4a0f8c04 | 602 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 LT |
603 | status = -ETIMEDOUT; |
604 | } | |
605 | ||
606 | if (req->rq_timeout == 0) { | |
607 | printk(KERN_WARNING "xprt_adjust_timeout: rq_timeout = 0!\n"); | |
608 | req->rq_timeout = 5 * HZ; | |
609 | } | |
610 | return status; | |
611 | } | |
612 | ||
65f27f38 | 613 | static void xprt_autoclose(struct work_struct *work) |
1da177e4 | 614 | { |
65f27f38 DH |
615 | struct rpc_xprt *xprt = |
616 | container_of(work, struct rpc_xprt, task_cleanup); | |
1da177e4 | 617 | |
a246b010 | 618 | xprt->ops->close(xprt); |
66af1e55 | 619 | clear_bit(XPRT_CLOSE_WAIT, &xprt->state); |
1da177e4 LT |
620 | xprt_release_write(xprt, NULL); |
621 | } | |
622 | ||
9903cd1c | 623 | /** |
62da3b24 | 624 | * xprt_disconnect_done - mark a transport as disconnected |
9903cd1c CL |
625 | * @xprt: transport to flag for disconnect |
626 | * | |
1da177e4 | 627 | */ |
62da3b24 | 628 | void xprt_disconnect_done(struct rpc_xprt *xprt) |
1da177e4 | 629 | { |
46121cf7 | 630 | dprintk("RPC: disconnected transport %p\n", xprt); |
4a0f8c04 | 631 | spin_lock_bh(&xprt->transport_lock); |
1da177e4 | 632 | xprt_clear_connected(xprt); |
2a491991 | 633 | xprt_wake_pending_tasks(xprt, -EAGAIN); |
4a0f8c04 | 634 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 | 635 | } |
62da3b24 | 636 | EXPORT_SYMBOL_GPL(xprt_disconnect_done); |
1da177e4 | 637 | |
66af1e55 TM |
638 | /** |
639 | * xprt_force_disconnect - force a transport to disconnect | |
640 | * @xprt: transport to disconnect | |
641 | * | |
642 | */ | |
643 | void xprt_force_disconnect(struct rpc_xprt *xprt) | |
644 | { | |
645 | /* Don't race with the test_bit() in xprt_clear_locked() */ | |
646 | spin_lock_bh(&xprt->transport_lock); | |
647 | set_bit(XPRT_CLOSE_WAIT, &xprt->state); | |
648 | /* Try to schedule an autoclose RPC call */ | |
649 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0) | |
650 | queue_work(rpciod_workqueue, &xprt->task_cleanup); | |
2a491991 | 651 | xprt_wake_pending_tasks(xprt, -EAGAIN); |
66af1e55 TM |
652 | spin_unlock_bh(&xprt->transport_lock); |
653 | } | |
66af1e55 | 654 | |
7c1d71cf TM |
655 | /** |
656 | * xprt_conditional_disconnect - force a transport to disconnect | |
657 | * @xprt: transport to disconnect | |
658 | * @cookie: 'connection cookie' | |
659 | * | |
660 | * This attempts to break the connection if and only if 'cookie' matches | |
661 | * the current transport 'connection cookie'. It ensures that we don't | |
662 | * try to break the connection more than once when we need to retransmit | |
663 | * a batch of RPC requests. | |
664 | * | |
665 | */ | |
666 | void xprt_conditional_disconnect(struct rpc_xprt *xprt, unsigned int cookie) | |
667 | { | |
668 | /* Don't race with the test_bit() in xprt_clear_locked() */ | |
669 | spin_lock_bh(&xprt->transport_lock); | |
670 | if (cookie != xprt->connect_cookie) | |
671 | goto out; | |
672 | if (test_bit(XPRT_CLOSING, &xprt->state) || !xprt_connected(xprt)) | |
673 | goto out; | |
674 | set_bit(XPRT_CLOSE_WAIT, &xprt->state); | |
675 | /* Try to schedule an autoclose RPC call */ | |
676 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0) | |
677 | queue_work(rpciod_workqueue, &xprt->task_cleanup); | |
2a491991 | 678 | xprt_wake_pending_tasks(xprt, -EAGAIN); |
7c1d71cf TM |
679 | out: |
680 | spin_unlock_bh(&xprt->transport_lock); | |
681 | } | |
682 | ||
1da177e4 LT |
683 | static void |
684 | xprt_init_autodisconnect(unsigned long data) | |
685 | { | |
686 | struct rpc_xprt *xprt = (struct rpc_xprt *)data; | |
687 | ||
4a0f8c04 | 688 | spin_lock(&xprt->transport_lock); |
d19751e7 | 689 | if (!list_empty(&xprt->recv)) |
1da177e4 | 690 | goto out_abort; |
2226feb6 | 691 | if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) |
1da177e4 | 692 | goto out_abort; |
4a0f8c04 | 693 | spin_unlock(&xprt->transport_lock); |
f75e6745 TM |
694 | set_bit(XPRT_CONNECTION_CLOSE, &xprt->state); |
695 | queue_work(rpciod_workqueue, &xprt->task_cleanup); | |
1da177e4 LT |
696 | return; |
697 | out_abort: | |
4a0f8c04 | 698 | spin_unlock(&xprt->transport_lock); |
1da177e4 LT |
699 | } |
700 | ||
9903cd1c CL |
701 | /** |
702 | * xprt_connect - schedule a transport connect operation | |
703 | * @task: RPC task that is requesting the connect | |
1da177e4 LT |
704 | * |
705 | */ | |
706 | void xprt_connect(struct rpc_task *task) | |
707 | { | |
ad2368d6 | 708 | struct rpc_xprt *xprt = task->tk_rqstp->rq_xprt; |
1da177e4 | 709 | |
46121cf7 | 710 | dprintk("RPC: %5u xprt_connect xprt %p %s connected\n", task->tk_pid, |
1da177e4 LT |
711 | xprt, (xprt_connected(xprt) ? "is" : "is not")); |
712 | ||
ec739ef0 | 713 | if (!xprt_bound(xprt)) { |
01d37c42 | 714 | task->tk_status = -EAGAIN; |
1da177e4 LT |
715 | return; |
716 | } | |
717 | if (!xprt_lock_write(xprt, task)) | |
718 | return; | |
feb8ca37 TM |
719 | |
720 | if (test_and_clear_bit(XPRT_CLOSE_WAIT, &xprt->state)) | |
721 | xprt->ops->close(xprt); | |
722 | ||
1da177e4 | 723 | if (xprt_connected(xprt)) |
a246b010 CL |
724 | xprt_release_write(xprt, task); |
725 | else { | |
87e3c055 | 726 | task->tk_rqstp->rq_bytes_sent = 0; |
a8ce4a8f | 727 | task->tk_timeout = task->tk_rqstp->rq_timeout; |
5d00837b | 728 | rpc_sleep_on(&xprt->pending, task, xprt_connect_status); |
0b9e7943 TM |
729 | |
730 | if (test_bit(XPRT_CLOSING, &xprt->state)) | |
731 | return; | |
732 | if (xprt_test_and_set_connecting(xprt)) | |
733 | return; | |
262ca07d | 734 | xprt->stat.connect_start = jiffies; |
1b092092 | 735 | xprt->ops->connect(xprt, task); |
1da177e4 | 736 | } |
1da177e4 LT |
737 | } |
738 | ||
9903cd1c | 739 | static void xprt_connect_status(struct rpc_task *task) |
1da177e4 | 740 | { |
ad2368d6 | 741 | struct rpc_xprt *xprt = task->tk_rqstp->rq_xprt; |
1da177e4 | 742 | |
cd983ef8 | 743 | if (task->tk_status == 0) { |
262ca07d CL |
744 | xprt->stat.connect_count++; |
745 | xprt->stat.connect_time += (long)jiffies - xprt->stat.connect_start; | |
46121cf7 | 746 | dprintk("RPC: %5u xprt_connect_status: connection established\n", |
1da177e4 LT |
747 | task->tk_pid); |
748 | return; | |
749 | } | |
750 | ||
1da177e4 | 751 | switch (task->tk_status) { |
2a491991 TM |
752 | case -EAGAIN: |
753 | dprintk("RPC: %5u xprt_connect_status: retrying\n", task->tk_pid); | |
23475d66 | 754 | break; |
1da177e4 | 755 | case -ETIMEDOUT: |
46121cf7 CL |
756 | dprintk("RPC: %5u xprt_connect_status: connect attempt timed " |
757 | "out\n", task->tk_pid); | |
1da177e4 LT |
758 | break; |
759 | default: | |
46121cf7 CL |
760 | dprintk("RPC: %5u xprt_connect_status: error %d connecting to " |
761 | "server %s\n", task->tk_pid, -task->tk_status, | |
4e0038b6 | 762 | xprt->servername); |
23475d66 CL |
763 | xprt_release_write(xprt, task); |
764 | task->tk_status = -EIO; | |
1da177e4 | 765 | } |
1da177e4 LT |
766 | } |
767 | ||
9903cd1c CL |
768 | /** |
769 | * xprt_lookup_rqst - find an RPC request corresponding to an XID | |
770 | * @xprt: transport on which the original request was transmitted | |
771 | * @xid: RPC XID of incoming reply | |
772 | * | |
1da177e4 | 773 | */ |
d8ed029d | 774 | struct rpc_rqst *xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid) |
1da177e4 | 775 | { |
8f3a6de3 | 776 | struct rpc_rqst *entry; |
1da177e4 | 777 | |
8f3a6de3 | 778 | list_for_each_entry(entry, &xprt->recv, rq_list) |
262ca07d CL |
779 | if (entry->rq_xid == xid) |
780 | return entry; | |
46121cf7 CL |
781 | |
782 | dprintk("RPC: xprt_lookup_rqst did not find xid %08x\n", | |
783 | ntohl(xid)); | |
262ca07d CL |
784 | xprt->stat.bad_xids++; |
785 | return NULL; | |
1da177e4 | 786 | } |
12444809 | 787 | EXPORT_SYMBOL_GPL(xprt_lookup_rqst); |
1da177e4 | 788 | |
bbc72cea | 789 | static void xprt_update_rtt(struct rpc_task *task) |
1570c1e4 CL |
790 | { |
791 | struct rpc_rqst *req = task->tk_rqstp; | |
792 | struct rpc_rtt *rtt = task->tk_client->cl_rtt; | |
95c96174 | 793 | unsigned int timer = task->tk_msg.rpc_proc->p_timer; |
d60dbb20 | 794 | long m = usecs_to_jiffies(ktime_to_us(req->rq_rtt)); |
1570c1e4 CL |
795 | |
796 | if (timer) { | |
797 | if (req->rq_ntrans == 1) | |
ff839970 | 798 | rpc_update_rtt(rtt, timer, m); |
1570c1e4 CL |
799 | rpc_set_timeo(rtt, timer, req->rq_ntrans - 1); |
800 | } | |
801 | } | |
802 | ||
9903cd1c CL |
803 | /** |
804 | * xprt_complete_rqst - called when reply processing is complete | |
1570c1e4 | 805 | * @task: RPC request that recently completed |
9903cd1c CL |
806 | * @copied: actual number of bytes received from the transport |
807 | * | |
1570c1e4 | 808 | * Caller holds transport lock. |
1da177e4 | 809 | */ |
1570c1e4 | 810 | void xprt_complete_rqst(struct rpc_task *task, int copied) |
1da177e4 | 811 | { |
1570c1e4 | 812 | struct rpc_rqst *req = task->tk_rqstp; |
fda13939 | 813 | struct rpc_xprt *xprt = req->rq_xprt; |
1da177e4 | 814 | |
1570c1e4 CL |
815 | dprintk("RPC: %5u xid %08x complete (%d bytes received)\n", |
816 | task->tk_pid, ntohl(req->rq_xid), copied); | |
1da177e4 | 817 | |
fda13939 | 818 | xprt->stat.recvs++; |
d60dbb20 | 819 | req->rq_rtt = ktime_sub(ktime_get(), req->rq_xtime); |
bbc72cea CL |
820 | if (xprt->ops->timer != NULL) |
821 | xprt_update_rtt(task); | |
ef759a2e | 822 | |
1da177e4 | 823 | list_del_init(&req->rq_list); |
1e799b67 | 824 | req->rq_private_buf.len = copied; |
dd2b63d0 RL |
825 | /* Ensure all writes are done before we update */ |
826 | /* req->rq_reply_bytes_recvd */ | |
43ac3f29 | 827 | smp_wmb(); |
dd2b63d0 | 828 | req->rq_reply_bytes_recvd = copied; |
fda13939 | 829 | rpc_wake_up_queued_task(&xprt->pending, task); |
1da177e4 | 830 | } |
12444809 | 831 | EXPORT_SYMBOL_GPL(xprt_complete_rqst); |
1da177e4 | 832 | |
46c0ee8b | 833 | static void xprt_timer(struct rpc_task *task) |
1da177e4 | 834 | { |
46c0ee8b | 835 | struct rpc_rqst *req = task->tk_rqstp; |
1da177e4 LT |
836 | struct rpc_xprt *xprt = req->rq_xprt; |
837 | ||
5d00837b TM |
838 | if (task->tk_status != -ETIMEDOUT) |
839 | return; | |
46121cf7 | 840 | dprintk("RPC: %5u xprt_timer\n", task->tk_pid); |
1da177e4 | 841 | |
5d00837b | 842 | spin_lock_bh(&xprt->transport_lock); |
dd2b63d0 | 843 | if (!req->rq_reply_bytes_recvd) { |
46c0ee8b | 844 | if (xprt->ops->timer) |
6a24dfb6 | 845 | xprt->ops->timer(xprt, task); |
5d00837b TM |
846 | } else |
847 | task->tk_status = 0; | |
848 | spin_unlock_bh(&xprt->transport_lock); | |
1da177e4 LT |
849 | } |
850 | ||
4cfc7e60 RI |
851 | static inline int xprt_has_timer(struct rpc_xprt *xprt) |
852 | { | |
853 | return xprt->idle_timeout != 0; | |
854 | } | |
855 | ||
9903cd1c CL |
856 | /** |
857 | * xprt_prepare_transmit - reserve the transport before sending a request | |
858 | * @task: RPC task about to send a request | |
859 | * | |
1da177e4 | 860 | */ |
90051ea7 | 861 | bool xprt_prepare_transmit(struct rpc_task *task) |
1da177e4 LT |
862 | { |
863 | struct rpc_rqst *req = task->tk_rqstp; | |
864 | struct rpc_xprt *xprt = req->rq_xprt; | |
90051ea7 | 865 | bool ret = false; |
1da177e4 | 866 | |
46121cf7 | 867 | dprintk("RPC: %5u xprt_prepare_transmit\n", task->tk_pid); |
1da177e4 | 868 | |
4a0f8c04 | 869 | spin_lock_bh(&xprt->transport_lock); |
8a19a0b6 TM |
870 | if (!req->rq_bytes_sent) { |
871 | if (req->rq_reply_bytes_recvd) { | |
872 | task->tk_status = req->rq_reply_bytes_recvd; | |
873 | goto out_unlock; | |
874 | } | |
875 | if ((task->tk_flags & RPC_TASK_NO_RETRANS_TIMEOUT) | |
876 | && xprt_connected(xprt) | |
877 | && req->rq_connect_cookie == xprt->connect_cookie) { | |
878 | xprt->ops->set_retrans_timeout(task); | |
879 | rpc_sleep_on(&xprt->pending, task, xprt_timer); | |
880 | goto out_unlock; | |
881 | } | |
1da177e4 | 882 | } |
90051ea7 TM |
883 | if (!xprt->ops->reserve_xprt(xprt, task)) { |
884 | task->tk_status = -EAGAIN; | |
885 | goto out_unlock; | |
886 | } | |
887 | ret = true; | |
1da177e4 | 888 | out_unlock: |
4a0f8c04 | 889 | spin_unlock_bh(&xprt->transport_lock); |
90051ea7 | 890 | return ret; |
1da177e4 LT |
891 | } |
892 | ||
e0ab53de | 893 | void xprt_end_transmit(struct rpc_task *task) |
5e5ce5be | 894 | { |
343952fa | 895 | xprt_release_write(task->tk_rqstp->rq_xprt, task); |
5e5ce5be TM |
896 | } |
897 | ||
9903cd1c CL |
898 | /** |
899 | * xprt_transmit - send an RPC request on a transport | |
900 | * @task: controlling RPC task | |
901 | * | |
902 | * We have to copy the iovec because sendmsg fiddles with its contents. | |
903 | */ | |
904 | void xprt_transmit(struct rpc_task *task) | |
1da177e4 | 905 | { |
1da177e4 LT |
906 | struct rpc_rqst *req = task->tk_rqstp; |
907 | struct rpc_xprt *xprt = req->rq_xprt; | |
15a45206 | 908 | int status, numreqs; |
1da177e4 | 909 | |
46121cf7 | 910 | dprintk("RPC: %5u xprt_transmit(%u)\n", task->tk_pid, req->rq_slen); |
1da177e4 | 911 | |
dd2b63d0 | 912 | if (!req->rq_reply_bytes_recvd) { |
55ae1aab RL |
913 | if (list_empty(&req->rq_list) && rpc_reply_expected(task)) { |
914 | /* | |
915 | * Add to the list only if we're expecting a reply | |
916 | */ | |
4a0f8c04 | 917 | spin_lock_bh(&xprt->transport_lock); |
1da177e4 LT |
918 | /* Update the softirq receive buffer */ |
919 | memcpy(&req->rq_private_buf, &req->rq_rcv_buf, | |
920 | sizeof(req->rq_private_buf)); | |
921 | /* Add request to the receive list */ | |
922 | list_add_tail(&req->rq_list, &xprt->recv); | |
4a0f8c04 | 923 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 | 924 | xprt_reset_majortimeo(req); |
0f9dc2b1 TM |
925 | /* Turn off autodisconnect */ |
926 | del_singleshot_timer_sync(&xprt->timer); | |
1da177e4 LT |
927 | } |
928 | } else if (!req->rq_bytes_sent) | |
929 | return; | |
930 | ||
ff839970 | 931 | req->rq_xtime = ktime_get(); |
a246b010 | 932 | status = xprt->ops->send_request(task); |
c8485e4d TM |
933 | if (status != 0) { |
934 | task->tk_status = status; | |
935 | return; | |
936 | } | |
262ca07d | 937 | |
c8485e4d | 938 | dprintk("RPC: %5u xmit complete\n", task->tk_pid); |
468f8613 | 939 | task->tk_flags |= RPC_TASK_SENT; |
c8485e4d | 940 | spin_lock_bh(&xprt->transport_lock); |
262ca07d | 941 | |
c8485e4d | 942 | xprt->ops->set_retrans_timeout(task); |
262ca07d | 943 | |
15a45206 AA |
944 | numreqs = atomic_read(&xprt->num_reqs); |
945 | if (numreqs > xprt->stat.max_slots) | |
946 | xprt->stat.max_slots = numreqs; | |
c8485e4d TM |
947 | xprt->stat.sends++; |
948 | xprt->stat.req_u += xprt->stat.sends - xprt->stat.recvs; | |
949 | xprt->stat.bklog_u += xprt->backlog.qlen; | |
15a45206 AA |
950 | xprt->stat.sending_u += xprt->sending.qlen; |
951 | xprt->stat.pending_u += xprt->pending.qlen; | |
1da177e4 | 952 | |
c8485e4d TM |
953 | /* Don't race with disconnect */ |
954 | if (!xprt_connected(xprt)) | |
955 | task->tk_status = -ENOTCONN; | |
0a660521 | 956 | else { |
55ae1aab RL |
957 | /* |
958 | * Sleep on the pending queue since | |
959 | * we're expecting a reply. | |
960 | */ | |
0a660521 TM |
961 | if (!req->rq_reply_bytes_recvd && rpc_reply_expected(task)) |
962 | rpc_sleep_on(&xprt->pending, task, xprt_timer); | |
963 | req->rq_connect_cookie = xprt->connect_cookie; | |
55ae1aab | 964 | } |
c8485e4d | 965 | spin_unlock_bh(&xprt->transport_lock); |
1da177e4 LT |
966 | } |
967 | ||
ba60eb25 TM |
968 | static void xprt_add_backlog(struct rpc_xprt *xprt, struct rpc_task *task) |
969 | { | |
970 | set_bit(XPRT_CONGESTED, &xprt->state); | |
971 | rpc_sleep_on(&xprt->backlog, task, NULL); | |
972 | } | |
973 | ||
974 | static void xprt_wake_up_backlog(struct rpc_xprt *xprt) | |
975 | { | |
976 | if (rpc_wake_up_next(&xprt->backlog) == NULL) | |
977 | clear_bit(XPRT_CONGESTED, &xprt->state); | |
978 | } | |
979 | ||
980 | static bool xprt_throttle_congested(struct rpc_xprt *xprt, struct rpc_task *task) | |
981 | { | |
982 | bool ret = false; | |
983 | ||
984 | if (!test_bit(XPRT_CONGESTED, &xprt->state)) | |
985 | goto out; | |
986 | spin_lock(&xprt->reserve_lock); | |
987 | if (test_bit(XPRT_CONGESTED, &xprt->state)) { | |
988 | rpc_sleep_on(&xprt->backlog, task, NULL); | |
989 | ret = true; | |
990 | } | |
991 | spin_unlock(&xprt->reserve_lock); | |
992 | out: | |
993 | return ret; | |
994 | } | |
995 | ||
d9ba131d TM |
996 | static struct rpc_rqst *xprt_dynamic_alloc_slot(struct rpc_xprt *xprt, gfp_t gfp_flags) |
997 | { | |
998 | struct rpc_rqst *req = ERR_PTR(-EAGAIN); | |
999 | ||
1000 | if (!atomic_add_unless(&xprt->num_reqs, 1, xprt->max_reqs)) | |
1001 | goto out; | |
1002 | req = kzalloc(sizeof(struct rpc_rqst), gfp_flags); | |
1003 | if (req != NULL) | |
1004 | goto out; | |
1005 | atomic_dec(&xprt->num_reqs); | |
1006 | req = ERR_PTR(-ENOMEM); | |
1007 | out: | |
1008 | return req; | |
1009 | } | |
1010 | ||
1011 | static bool xprt_dynamic_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req) | |
1012 | { | |
1013 | if (atomic_add_unless(&xprt->num_reqs, -1, xprt->min_reqs)) { | |
1014 | kfree(req); | |
1015 | return true; | |
1016 | } | |
1017 | return false; | |
1018 | } | |
1019 | ||
f39c1bfb | 1020 | void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task) |
1da177e4 | 1021 | { |
d9ba131d | 1022 | struct rpc_rqst *req; |
1da177e4 | 1023 | |
f39c1bfb | 1024 | spin_lock(&xprt->reserve_lock); |
1da177e4 | 1025 | if (!list_empty(&xprt->free)) { |
d9ba131d TM |
1026 | req = list_entry(xprt->free.next, struct rpc_rqst, rq_list); |
1027 | list_del(&req->rq_list); | |
1028 | goto out_init_req; | |
1029 | } | |
6b343099 | 1030 | req = xprt_dynamic_alloc_slot(xprt, GFP_NOWAIT|__GFP_NOWARN); |
d9ba131d TM |
1031 | if (!IS_ERR(req)) |
1032 | goto out_init_req; | |
1033 | switch (PTR_ERR(req)) { | |
1034 | case -ENOMEM: | |
d9ba131d TM |
1035 | dprintk("RPC: dynamic allocation of request slot " |
1036 | "failed! Retrying\n"); | |
1afeaf5c | 1037 | task->tk_status = -ENOMEM; |
d9ba131d TM |
1038 | break; |
1039 | case -EAGAIN: | |
ba60eb25 | 1040 | xprt_add_backlog(xprt, task); |
d9ba131d | 1041 | dprintk("RPC: waiting for request slot\n"); |
1afeaf5c TM |
1042 | default: |
1043 | task->tk_status = -EAGAIN; | |
1da177e4 | 1044 | } |
f39c1bfb | 1045 | spin_unlock(&xprt->reserve_lock); |
d9ba131d TM |
1046 | return; |
1047 | out_init_req: | |
1048 | task->tk_status = 0; | |
1049 | task->tk_rqstp = req; | |
1050 | xprt_request_init(task, xprt); | |
f39c1bfb TM |
1051 | spin_unlock(&xprt->reserve_lock); |
1052 | } | |
1053 | EXPORT_SYMBOL_GPL(xprt_alloc_slot); | |
1054 | ||
1055 | void xprt_lock_and_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task) | |
1056 | { | |
1057 | /* Note: grabbing the xprt_lock_write() ensures that we throttle | |
1058 | * new slot allocation if the transport is congested (i.e. when | |
1059 | * reconnecting a stream transport or when out of socket write | |
1060 | * buffer space). | |
1061 | */ | |
1062 | if (xprt_lock_write(xprt, task)) { | |
1063 | xprt_alloc_slot(xprt, task); | |
1064 | xprt_release_write(xprt, task); | |
1065 | } | |
1da177e4 | 1066 | } |
f39c1bfb | 1067 | EXPORT_SYMBOL_GPL(xprt_lock_and_alloc_slot); |
1da177e4 | 1068 | |
ee5ebe85 TM |
1069 | static void xprt_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req) |
1070 | { | |
ee5ebe85 | 1071 | spin_lock(&xprt->reserve_lock); |
c25573b5 TM |
1072 | if (!xprt_dynamic_free_slot(xprt, req)) { |
1073 | memset(req, 0, sizeof(*req)); /* mark unused */ | |
1074 | list_add(&req->rq_list, &xprt->free); | |
1075 | } | |
ba60eb25 | 1076 | xprt_wake_up_backlog(xprt); |
ee5ebe85 TM |
1077 | spin_unlock(&xprt->reserve_lock); |
1078 | } | |
1079 | ||
21de0a95 TM |
1080 | static void xprt_free_all_slots(struct rpc_xprt *xprt) |
1081 | { | |
1082 | struct rpc_rqst *req; | |
1083 | while (!list_empty(&xprt->free)) { | |
1084 | req = list_first_entry(&xprt->free, struct rpc_rqst, rq_list); | |
1085 | list_del(&req->rq_list); | |
1086 | kfree(req); | |
1087 | } | |
1088 | } | |
1089 | ||
d9ba131d TM |
1090 | struct rpc_xprt *xprt_alloc(struct net *net, size_t size, |
1091 | unsigned int num_prealloc, | |
1092 | unsigned int max_alloc) | |
bd1722d4 PE |
1093 | { |
1094 | struct rpc_xprt *xprt; | |
21de0a95 TM |
1095 | struct rpc_rqst *req; |
1096 | int i; | |
bd1722d4 PE |
1097 | |
1098 | xprt = kzalloc(size, GFP_KERNEL); | |
1099 | if (xprt == NULL) | |
1100 | goto out; | |
1101 | ||
21de0a95 TM |
1102 | xprt_init(xprt, net); |
1103 | ||
1104 | for (i = 0; i < num_prealloc; i++) { | |
1105 | req = kzalloc(sizeof(struct rpc_rqst), GFP_KERNEL); | |
1106 | if (!req) | |
8313164c | 1107 | goto out_free; |
21de0a95 TM |
1108 | list_add(&req->rq_list, &xprt->free); |
1109 | } | |
d9ba131d TM |
1110 | if (max_alloc > num_prealloc) |
1111 | xprt->max_reqs = max_alloc; | |
1112 | else | |
1113 | xprt->max_reqs = num_prealloc; | |
1114 | xprt->min_reqs = num_prealloc; | |
1115 | atomic_set(&xprt->num_reqs, num_prealloc); | |
bd1722d4 PE |
1116 | |
1117 | return xprt; | |
1118 | ||
1119 | out_free: | |
21de0a95 | 1120 | xprt_free(xprt); |
bd1722d4 PE |
1121 | out: |
1122 | return NULL; | |
1123 | } | |
1124 | EXPORT_SYMBOL_GPL(xprt_alloc); | |
1125 | ||
e204e621 PE |
1126 | void xprt_free(struct rpc_xprt *xprt) |
1127 | { | |
37aa2133 | 1128 | put_net(xprt->xprt_net); |
21de0a95 | 1129 | xprt_free_all_slots(xprt); |
e204e621 PE |
1130 | kfree(xprt); |
1131 | } | |
1132 | EXPORT_SYMBOL_GPL(xprt_free); | |
1133 | ||
9903cd1c CL |
1134 | /** |
1135 | * xprt_reserve - allocate an RPC request slot | |
1136 | * @task: RPC task requesting a slot allocation | |
1137 | * | |
ba60eb25 TM |
1138 | * If the transport is marked as being congested, or if no more |
1139 | * slots are available, place the task on the transport's | |
9903cd1c CL |
1140 | * backlog queue. |
1141 | */ | |
1142 | void xprt_reserve(struct rpc_task *task) | |
1da177e4 | 1143 | { |
45bc0dce | 1144 | struct rpc_xprt *xprt; |
1da177e4 | 1145 | |
43cedbf0 TM |
1146 | task->tk_status = 0; |
1147 | if (task->tk_rqstp != NULL) | |
1148 | return; | |
1149 | ||
43cedbf0 TM |
1150 | task->tk_timeout = 0; |
1151 | task->tk_status = -EAGAIN; | |
45bc0dce TM |
1152 | rcu_read_lock(); |
1153 | xprt = rcu_dereference(task->tk_client->cl_xprt); | |
ba60eb25 TM |
1154 | if (!xprt_throttle_congested(xprt, task)) |
1155 | xprt->ops->alloc_slot(xprt, task); | |
1156 | rcu_read_unlock(); | |
1157 | } | |
1158 | ||
1159 | /** | |
1160 | * xprt_retry_reserve - allocate an RPC request slot | |
1161 | * @task: RPC task requesting a slot allocation | |
1162 | * | |
1163 | * If no more slots are available, place the task on the transport's | |
1164 | * backlog queue. | |
1165 | * Note that the only difference with xprt_reserve is that we now | |
1166 | * ignore the value of the XPRT_CONGESTED flag. | |
1167 | */ | |
1168 | void xprt_retry_reserve(struct rpc_task *task) | |
1169 | { | |
1170 | struct rpc_xprt *xprt; | |
1171 | ||
1172 | task->tk_status = 0; | |
1173 | if (task->tk_rqstp != NULL) | |
1174 | return; | |
1175 | ||
1176 | task->tk_timeout = 0; | |
1177 | task->tk_status = -EAGAIN; | |
1178 | rcu_read_lock(); | |
1179 | xprt = rcu_dereference(task->tk_client->cl_xprt); | |
f39c1bfb | 1180 | xprt->ops->alloc_slot(xprt, task); |
45bc0dce | 1181 | rcu_read_unlock(); |
1da177e4 LT |
1182 | } |
1183 | ||
d8ed029d | 1184 | static inline __be32 xprt_alloc_xid(struct rpc_xprt *xprt) |
1da177e4 | 1185 | { |
0eae88f3 | 1186 | return (__force __be32)xprt->xid++; |
1da177e4 LT |
1187 | } |
1188 | ||
1189 | static inline void xprt_init_xid(struct rpc_xprt *xprt) | |
1190 | { | |
63862b5b | 1191 | xprt->xid = prandom_u32(); |
1da177e4 LT |
1192 | } |
1193 | ||
9903cd1c | 1194 | static void xprt_request_init(struct rpc_task *task, struct rpc_xprt *xprt) |
1da177e4 LT |
1195 | { |
1196 | struct rpc_rqst *req = task->tk_rqstp; | |
1197 | ||
d9ba131d | 1198 | INIT_LIST_HEAD(&req->rq_list); |
ba7392bb | 1199 | req->rq_timeout = task->tk_client->cl_timeout->to_initval; |
1da177e4 LT |
1200 | req->rq_task = task; |
1201 | req->rq_xprt = xprt; | |
02107148 | 1202 | req->rq_buffer = NULL; |
1da177e4 | 1203 | req->rq_xid = xprt_alloc_xid(xprt); |
0a660521 | 1204 | req->rq_connect_cookie = xprt->connect_cookie - 1; |
92551948 TM |
1205 | req->rq_bytes_sent = 0; |
1206 | req->rq_snd_buf.len = 0; | |
1207 | req->rq_snd_buf.buflen = 0; | |
1208 | req->rq_rcv_buf.len = 0; | |
1209 | req->rq_rcv_buf.buflen = 0; | |
ead5e1c2 | 1210 | req->rq_release_snd_buf = NULL; |
da45828e | 1211 | xprt_reset_majortimeo(req); |
46121cf7 | 1212 | dprintk("RPC: %5u reserved req %p xid %08x\n", task->tk_pid, |
1da177e4 LT |
1213 | req, ntohl(req->rq_xid)); |
1214 | } | |
1215 | ||
9903cd1c CL |
1216 | /** |
1217 | * xprt_release - release an RPC request slot | |
1218 | * @task: task which is finished with the slot | |
1219 | * | |
1da177e4 | 1220 | */ |
9903cd1c | 1221 | void xprt_release(struct rpc_task *task) |
1da177e4 | 1222 | { |
55ae1aab | 1223 | struct rpc_xprt *xprt; |
87ed5003 | 1224 | struct rpc_rqst *req = task->tk_rqstp; |
1da177e4 | 1225 | |
87ed5003 TM |
1226 | if (req == NULL) { |
1227 | if (task->tk_client) { | |
1228 | rcu_read_lock(); | |
1229 | xprt = rcu_dereference(task->tk_client->cl_xprt); | |
1230 | if (xprt->snd_task == task) | |
1231 | xprt_release_write(xprt, task); | |
1232 | rcu_read_unlock(); | |
1233 | } | |
1da177e4 | 1234 | return; |
87ed5003 | 1235 | } |
55ae1aab | 1236 | |
55ae1aab | 1237 | xprt = req->rq_xprt; |
0a702195 WAA |
1238 | if (task->tk_ops->rpc_count_stats != NULL) |
1239 | task->tk_ops->rpc_count_stats(task, task->tk_calldata); | |
1240 | else if (task->tk_client) | |
1241 | rpc_count_iostats(task, task->tk_client->cl_metrics); | |
4a0f8c04 | 1242 | spin_lock_bh(&xprt->transport_lock); |
49e9a890 | 1243 | xprt->ops->release_xprt(xprt, task); |
a58dd398 CL |
1244 | if (xprt->ops->release_request) |
1245 | xprt->ops->release_request(task); | |
1da177e4 LT |
1246 | if (!list_empty(&req->rq_list)) |
1247 | list_del(&req->rq_list); | |
1248 | xprt->last_used = jiffies; | |
4cfc7e60 | 1249 | if (list_empty(&xprt->recv) && xprt_has_timer(xprt)) |
a246b010 | 1250 | mod_timer(&xprt->timer, |
03bf4b70 | 1251 | xprt->last_used + xprt->idle_timeout); |
4a0f8c04 | 1252 | spin_unlock_bh(&xprt->transport_lock); |
ee5ebe85 | 1253 | if (req->rq_buffer) |
55ae1aab | 1254 | xprt->ops->buf_free(req->rq_buffer); |
a17c2153 TM |
1255 | if (req->rq_cred != NULL) |
1256 | put_rpccred(req->rq_cred); | |
1da177e4 | 1257 | task->tk_rqstp = NULL; |
ead5e1c2 BF |
1258 | if (req->rq_release_snd_buf) |
1259 | req->rq_release_snd_buf(req); | |
55ae1aab | 1260 | |
46121cf7 | 1261 | dprintk("RPC: %5u release request %p\n", task->tk_pid, req); |
ee5ebe85 TM |
1262 | if (likely(!bc_prealloc(req))) |
1263 | xprt_free_slot(xprt, req); | |
1264 | else | |
c9acb42e | 1265 | xprt_free_bc_request(req); |
1da177e4 LT |
1266 | } |
1267 | ||
21de0a95 | 1268 | static void xprt_init(struct rpc_xprt *xprt, struct net *net) |
c2866763 | 1269 | { |
21de0a95 | 1270 | atomic_set(&xprt->count, 1); |
c2866763 CL |
1271 | |
1272 | spin_lock_init(&xprt->transport_lock); | |
1273 | spin_lock_init(&xprt->reserve_lock); | |
1274 | ||
1275 | INIT_LIST_HEAD(&xprt->free); | |
1276 | INIT_LIST_HEAD(&xprt->recv); | |
9e00abc3 | 1277 | #if defined(CONFIG_SUNRPC_BACKCHANNEL) |
f9acac1a RL |
1278 | spin_lock_init(&xprt->bc_pa_lock); |
1279 | INIT_LIST_HEAD(&xprt->bc_pa_list); | |
9e00abc3 | 1280 | #endif /* CONFIG_SUNRPC_BACKCHANNEL */ |
f9acac1a | 1281 | |
c2866763 CL |
1282 | xprt->last_used = jiffies; |
1283 | xprt->cwnd = RPC_INITCWND; | |
a509050b | 1284 | xprt->bind_index = 0; |
c2866763 CL |
1285 | |
1286 | rpc_init_wait_queue(&xprt->binding, "xprt_binding"); | |
1287 | rpc_init_wait_queue(&xprt->pending, "xprt_pending"); | |
34006cee | 1288 | rpc_init_priority_wait_queue(&xprt->sending, "xprt_sending"); |
c2866763 CL |
1289 | rpc_init_priority_wait_queue(&xprt->backlog, "xprt_backlog"); |
1290 | ||
c2866763 CL |
1291 | xprt_init_xid(xprt); |
1292 | ||
21de0a95 | 1293 | xprt->xprt_net = get_net(net); |
8d9266ff TM |
1294 | } |
1295 | ||
1296 | /** | |
1297 | * xprt_create_transport - create an RPC transport | |
1298 | * @args: rpc transport creation arguments | |
1299 | * | |
1300 | */ | |
1301 | struct rpc_xprt *xprt_create_transport(struct xprt_create *args) | |
1302 | { | |
1303 | struct rpc_xprt *xprt; | |
1304 | struct xprt_class *t; | |
1305 | ||
1306 | spin_lock(&xprt_list_lock); | |
1307 | list_for_each_entry(t, &xprt_list, list) { | |
1308 | if (t->ident == args->ident) { | |
1309 | spin_unlock(&xprt_list_lock); | |
1310 | goto found; | |
1311 | } | |
1312 | } | |
1313 | spin_unlock(&xprt_list_lock); | |
1314 | printk(KERN_ERR "RPC: transport (%d) not supported\n", args->ident); | |
1315 | return ERR_PTR(-EIO); | |
1316 | ||
1317 | found: | |
1318 | xprt = t->setup(args); | |
1319 | if (IS_ERR(xprt)) { | |
1320 | dprintk("RPC: xprt_create_transport: failed, %ld\n", | |
1321 | -PTR_ERR(xprt)); | |
21de0a95 | 1322 | goto out; |
8d9266ff | 1323 | } |
33d90ac0 BF |
1324 | if (args->flags & XPRT_CREATE_NO_IDLE_TIMEOUT) |
1325 | xprt->idle_timeout = 0; | |
21de0a95 TM |
1326 | INIT_WORK(&xprt->task_cleanup, xprt_autoclose); |
1327 | if (xprt_has_timer(xprt)) | |
1328 | setup_timer(&xprt->timer, xprt_init_autodisconnect, | |
1329 | (unsigned long)xprt); | |
1330 | else | |
1331 | init_timer(&xprt->timer); | |
4e0038b6 TM |
1332 | |
1333 | if (strlen(args->servername) > RPC_MAXNETNAMELEN) { | |
1334 | xprt_destroy(xprt); | |
1335 | return ERR_PTR(-EINVAL); | |
1336 | } | |
1337 | xprt->servername = kstrdup(args->servername, GFP_KERNEL); | |
1338 | if (xprt->servername == NULL) { | |
1339 | xprt_destroy(xprt); | |
1340 | return ERR_PTR(-ENOMEM); | |
1341 | } | |
1342 | ||
46121cf7 | 1343 | dprintk("RPC: created transport %p with %u slots\n", xprt, |
c2866763 | 1344 | xprt->max_reqs); |
21de0a95 | 1345 | out: |
c2866763 CL |
1346 | return xprt; |
1347 | } | |
1348 | ||
9903cd1c CL |
1349 | /** |
1350 | * xprt_destroy - destroy an RPC transport, killing off all requests. | |
a8de240a | 1351 | * @xprt: transport to destroy |
9903cd1c | 1352 | * |
1da177e4 | 1353 | */ |
a8de240a | 1354 | static void xprt_destroy(struct rpc_xprt *xprt) |
1da177e4 | 1355 | { |
46121cf7 | 1356 | dprintk("RPC: destroying transport %p\n", xprt); |
0065db32 | 1357 | del_timer_sync(&xprt->timer); |
c8541ecd | 1358 | |
f6a1cc89 TM |
1359 | rpc_destroy_wait_queue(&xprt->binding); |
1360 | rpc_destroy_wait_queue(&xprt->pending); | |
1361 | rpc_destroy_wait_queue(&xprt->sending); | |
f6a1cc89 | 1362 | rpc_destroy_wait_queue(&xprt->backlog); |
c3ae62ae | 1363 | cancel_work_sync(&xprt->task_cleanup); |
4e0038b6 | 1364 | kfree(xprt->servername); |
c8541ecd CL |
1365 | /* |
1366 | * Tear down transport state and free the rpc_xprt | |
1367 | */ | |
a246b010 | 1368 | xprt->ops->destroy(xprt); |
6b6ca86b | 1369 | } |
1da177e4 | 1370 | |
6b6ca86b TM |
1371 | /** |
1372 | * xprt_put - release a reference to an RPC transport. | |
1373 | * @xprt: pointer to the transport | |
1374 | * | |
1375 | */ | |
1376 | void xprt_put(struct rpc_xprt *xprt) | |
1377 | { | |
a8de240a TM |
1378 | if (atomic_dec_and_test(&xprt->count)) |
1379 | xprt_destroy(xprt); | |
6b6ca86b TM |
1380 | } |
1381 | ||
1382 | /** | |
1383 | * xprt_get - return a reference to an RPC transport. | |
1384 | * @xprt: pointer to the transport | |
1385 | * | |
1386 | */ | |
1387 | struct rpc_xprt *xprt_get(struct rpc_xprt *xprt) | |
1388 | { | |
a8de240a TM |
1389 | if (atomic_inc_not_zero(&xprt->count)) |
1390 | return xprt; | |
1391 | return NULL; | |
1da177e4 | 1392 | } |