11da177e4SLinus Torvalds /* 21da177e4SLinus Torvalds * linux/net/sunrpc/xprt.c 31da177e4SLinus Torvalds * 41da177e4SLinus Torvalds * This is a generic RPC call interface supporting congestion avoidance, 51da177e4SLinus Torvalds * and asynchronous calls. 61da177e4SLinus Torvalds * 71da177e4SLinus Torvalds * The interface works like this: 81da177e4SLinus Torvalds * 91da177e4SLinus Torvalds * - When a process places a call, it allocates a request slot if 101da177e4SLinus Torvalds * one is available. Otherwise, it sleeps on the backlog queue 111da177e4SLinus Torvalds * (xprt_reserve). 121da177e4SLinus Torvalds * - Next, the caller puts together the RPC message, stuffs it into 1355aa4f58SChuck Lever * the request struct, and calls xprt_transmit(). 1455aa4f58SChuck Lever * - xprt_transmit sends the message and installs the caller on the 1555ae1aabSRicardo Labiaga * transport's wait list. At the same time, if a reply is expected, 1655ae1aabSRicardo Labiaga * it installs a timer that is run after the packet's timeout has 1755ae1aabSRicardo Labiaga * expired. 181da177e4SLinus Torvalds * - When a packet arrives, the data_ready handler walks the list of 1955aa4f58SChuck Lever * pending requests for that transport. If a matching XID is found, the 201da177e4SLinus Torvalds * caller is woken up, and the timer removed. 211da177e4SLinus Torvalds * - When no reply arrives within the timeout interval, the timer is 221da177e4SLinus Torvalds * fired by the kernel and runs xprt_timer(). It either adjusts the 231da177e4SLinus Torvalds * timeout values (minor timeout) or wakes up the caller with a status 241da177e4SLinus Torvalds * of -ETIMEDOUT. 251da177e4SLinus Torvalds * - When the caller receives a notification from RPC that a reply arrived, 261da177e4SLinus Torvalds * it should release the RPC slot, and process the reply. 271da177e4SLinus Torvalds * If the call timed out, it may choose to retry the operation by 281da177e4SLinus Torvalds * adjusting the initial timeout value, and simply calling rpc_call 291da177e4SLinus Torvalds * again. 301da177e4SLinus Torvalds * 311da177e4SLinus Torvalds * Support for async RPC is done through a set of RPC-specific scheduling 321da177e4SLinus Torvalds * primitives that `transparently' work for processes as well as async 331da177e4SLinus Torvalds * tasks that rely on callbacks. 341da177e4SLinus Torvalds * 351da177e4SLinus Torvalds * Copyright (C) 1995-1997, Olaf Kirch <okir@monad.swb.de> 3655aa4f58SChuck Lever * 3755aa4f58SChuck Lever * Transport switch API copyright (C) 2005, Chuck Lever <cel@netapp.com> 381da177e4SLinus Torvalds */ 391da177e4SLinus Torvalds 40a246b010SChuck Lever #include <linux/module.h> 41a246b010SChuck Lever 421da177e4SLinus Torvalds #include <linux/types.h> 43a246b010SChuck Lever #include <linux/interrupt.h> 441da177e4SLinus Torvalds #include <linux/workqueue.h> 45bf3fcf89SChuck Lever #include <linux/net.h> 46ff839970SChuck Lever #include <linux/ktime.h> 471da177e4SLinus Torvalds 48a246b010SChuck Lever #include <linux/sunrpc/clnt.h> 4911c556b3SChuck Lever #include <linux/sunrpc/metrics.h> 50c9acb42eSTrond Myklebust #include <linux/sunrpc/bc_xprt.h> 51fda1bfefSTrond Myklebust #include <linux/rcupdate.h> 521da177e4SLinus Torvalds 533705ad64SJeff Layton #include <trace/events/sunrpc.h> 543705ad64SJeff Layton 5555ae1aabSRicardo Labiaga #include "sunrpc.h" 5655ae1aabSRicardo Labiaga 571da177e4SLinus Torvalds /* 581da177e4SLinus Torvalds * Local variables 591da177e4SLinus Torvalds */ 601da177e4SLinus Torvalds 61f895b252SJeff Layton #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) 621da177e4SLinus Torvalds # define RPCDBG_FACILITY RPCDBG_XPRT 631da177e4SLinus Torvalds #endif 641da177e4SLinus Torvalds 651da177e4SLinus Torvalds /* 661da177e4SLinus Torvalds * Local functions 671da177e4SLinus Torvalds */ 6821de0a95STrond Myklebust static void xprt_init(struct rpc_xprt *xprt, struct net *net); 6937ac86c3SChuck Lever static __be32 xprt_alloc_xid(struct rpc_xprt *xprt); 701da177e4SLinus Torvalds static void xprt_connect_status(struct rpc_task *task); 711da177e4SLinus Torvalds static int __xprt_get_cong(struct rpc_xprt *, struct rpc_task *); 7229807318SNeil Brown static void __xprt_put_cong(struct rpc_xprt *, struct rpc_rqst *); 734e0038b6STrond Myklebust static void xprt_destroy(struct rpc_xprt *xprt); 741da177e4SLinus Torvalds 755ba03e82SJiri Slaby static DEFINE_SPINLOCK(xprt_list_lock); 7681c098afS\"Talpey, Thomas\ static LIST_HEAD(xprt_list); 7781c098afS\"Talpey, Thomas\ 7812a80469SChuck Lever /** 7981c098afS\"Talpey, Thomas\ * xprt_register_transport - register a transport implementation 8081c098afS\"Talpey, Thomas\ * @transport: transport to register 8181c098afS\"Talpey, Thomas\ * 8281c098afS\"Talpey, Thomas\ * If a transport implementation is loaded as a kernel module, it can 8381c098afS\"Talpey, Thomas\ * call this interface to make itself known to the RPC client. 8481c098afS\"Talpey, Thomas\ * 8581c098afS\"Talpey, Thomas\ * Returns: 8681c098afS\"Talpey, Thomas\ * 0: transport successfully registered 8781c098afS\"Talpey, Thomas\ * -EEXIST: transport already registered 8881c098afS\"Talpey, Thomas\ * -EINVAL: transport module being unloaded 8981c098afS\"Talpey, Thomas\ */ 9081c098afS\"Talpey, Thomas\ int xprt_register_transport(struct xprt_class *transport) 9181c098afS\"Talpey, Thomas\ { 9281c098afS\"Talpey, Thomas\ struct xprt_class *t; 9381c098afS\"Talpey, Thomas\ int result; 9481c098afS\"Talpey, Thomas\ 9581c098afS\"Talpey, Thomas\ result = -EEXIST; 9681c098afS\"Talpey, Thomas\ spin_lock(&xprt_list_lock); 9781c098afS\"Talpey, Thomas\ list_for_each_entry(t, &xprt_list, list) { 9881c098afS\"Talpey, Thomas\ /* don't register the same transport class twice */ 994fa016ebS\"Talpey, Thomas\ if (t->ident == transport->ident) 10081c098afS\"Talpey, Thomas\ goto out; 10181c098afS\"Talpey, Thomas\ } 10281c098afS\"Talpey, Thomas\ 10381c098afS\"Talpey, Thomas\ list_add_tail(&transport->list, &xprt_list); 10481c098afS\"Talpey, Thomas\ printk(KERN_INFO "RPC: Registered %s transport module.\n", 10581c098afS\"Talpey, Thomas\ transport->name); 10681c098afS\"Talpey, Thomas\ result = 0; 10781c098afS\"Talpey, Thomas\ 10881c098afS\"Talpey, Thomas\ out: 10981c098afS\"Talpey, Thomas\ spin_unlock(&xprt_list_lock); 11081c098afS\"Talpey, Thomas\ return result; 11181c098afS\"Talpey, Thomas\ } 11281c098afS\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_register_transport); 11381c098afS\"Talpey, Thomas\ 11481c098afS\"Talpey, Thomas\ /** 11581c098afS\"Talpey, Thomas\ * xprt_unregister_transport - unregister a transport implementation 11665b6e42cSRandy Dunlap * @transport: transport to unregister 11781c098afS\"Talpey, Thomas\ * 11881c098afS\"Talpey, Thomas\ * Returns: 11981c098afS\"Talpey, Thomas\ * 0: transport successfully unregistered 12081c098afS\"Talpey, Thomas\ * -ENOENT: transport never registered 12181c098afS\"Talpey, Thomas\ */ 12281c098afS\"Talpey, Thomas\ int xprt_unregister_transport(struct xprt_class *transport) 12381c098afS\"Talpey, Thomas\ { 12481c098afS\"Talpey, Thomas\ struct xprt_class *t; 12581c098afS\"Talpey, Thomas\ int result; 12681c098afS\"Talpey, Thomas\ 12781c098afS\"Talpey, Thomas\ result = 0; 12881c098afS\"Talpey, Thomas\ spin_lock(&xprt_list_lock); 12981c098afS\"Talpey, Thomas\ list_for_each_entry(t, &xprt_list, list) { 13081c098afS\"Talpey, Thomas\ if (t == transport) { 13181c098afS\"Talpey, Thomas\ printk(KERN_INFO 13281c098afS\"Talpey, Thomas\ "RPC: Unregistered %s transport module.\n", 13381c098afS\"Talpey, Thomas\ transport->name); 13481c098afS\"Talpey, Thomas\ list_del_init(&transport->list); 13581c098afS\"Talpey, Thomas\ goto out; 13681c098afS\"Talpey, Thomas\ } 13781c098afS\"Talpey, Thomas\ } 13881c098afS\"Talpey, Thomas\ result = -ENOENT; 13981c098afS\"Talpey, Thomas\ 14081c098afS\"Talpey, Thomas\ out: 14181c098afS\"Talpey, Thomas\ spin_unlock(&xprt_list_lock); 14281c098afS\"Talpey, Thomas\ return result; 14381c098afS\"Talpey, Thomas\ } 14481c098afS\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_unregister_transport); 14581c098afS\"Talpey, Thomas\ 14681c098afS\"Talpey, Thomas\ /** 147441e3e24STom Talpey * xprt_load_transport - load a transport implementation 148441e3e24STom Talpey * @transport_name: transport to load 149441e3e24STom Talpey * 150441e3e24STom Talpey * Returns: 151441e3e24STom Talpey * 0: transport successfully loaded 152441e3e24STom Talpey * -ENOENT: transport module not available 153441e3e24STom Talpey */ 154441e3e24STom Talpey int xprt_load_transport(const char *transport_name) 155441e3e24STom Talpey { 156441e3e24STom Talpey struct xprt_class *t; 157441e3e24STom Talpey int result; 158441e3e24STom Talpey 159441e3e24STom Talpey result = 0; 160441e3e24STom Talpey spin_lock(&xprt_list_lock); 161441e3e24STom Talpey list_for_each_entry(t, &xprt_list, list) { 162441e3e24STom Talpey if (strcmp(t->name, transport_name) == 0) { 163441e3e24STom Talpey spin_unlock(&xprt_list_lock); 164441e3e24STom Talpey goto out; 165441e3e24STom Talpey } 166441e3e24STom Talpey } 167441e3e24STom Talpey spin_unlock(&xprt_list_lock); 168ef7ffe8fSAlex Riesen result = request_module("xprt%s", transport_name); 169441e3e24STom Talpey out: 170441e3e24STom Talpey return result; 171441e3e24STom Talpey } 172441e3e24STom Talpey EXPORT_SYMBOL_GPL(xprt_load_transport); 173441e3e24STom Talpey 174441e3e24STom Talpey /** 17512a80469SChuck Lever * xprt_reserve_xprt - serialize write access to transports 17612a80469SChuck Lever * @task: task that is requesting access to the transport 177177c27bfSRandy Dunlap * @xprt: pointer to the target transport 17812a80469SChuck Lever * 17912a80469SChuck Lever * This prevents mixing the payload of separate requests, and prevents 18012a80469SChuck Lever * transport connects from colliding with writes. No congestion control 18112a80469SChuck Lever * is provided. 1821da177e4SLinus Torvalds */ 18343cedbf0STrond Myklebust int xprt_reserve_xprt(struct rpc_xprt *xprt, struct rpc_task *task) 1841da177e4SLinus Torvalds { 18512a80469SChuck Lever struct rpc_rqst *req = task->tk_rqstp; 18634006ceeSTrond Myklebust int priority; 18712a80469SChuck Lever 18812a80469SChuck Lever if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) { 18912a80469SChuck Lever if (task == xprt->snd_task) 19012a80469SChuck Lever return 1; 19112a80469SChuck Lever goto out_sleep; 19212a80469SChuck Lever } 19312a80469SChuck Lever xprt->snd_task = task; 19492551948STrond Myklebust if (req != NULL) 19512a80469SChuck Lever req->rq_ntrans++; 1964d4a76f3Sj223yang@asset.uwaterloo.ca 19712a80469SChuck Lever return 1; 19812a80469SChuck Lever 19912a80469SChuck Lever out_sleep: 20046121cf7SChuck Lever dprintk("RPC: %5u failed to lock transport %p\n", 20112a80469SChuck Lever task->tk_pid, xprt); 20212a80469SChuck Lever task->tk_timeout = 0; 20312a80469SChuck Lever task->tk_status = -EAGAIN; 20434006ceeSTrond Myklebust if (req == NULL) 20534006ceeSTrond Myklebust priority = RPC_PRIORITY_LOW; 20634006ceeSTrond Myklebust else if (!req->rq_ntrans) 20734006ceeSTrond Myklebust priority = RPC_PRIORITY_NORMAL; 20812a80469SChuck Lever else 20934006ceeSTrond Myklebust priority = RPC_PRIORITY_HIGH; 21034006ceeSTrond Myklebust rpc_sleep_on_priority(&xprt->sending, task, NULL, priority); 21112a80469SChuck Lever return 0; 21212a80469SChuck Lever } 21312444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_reserve_xprt); 21412a80469SChuck Lever 215632e3bdcSTrond Myklebust static void xprt_clear_locked(struct rpc_xprt *xprt) 216632e3bdcSTrond Myklebust { 217632e3bdcSTrond Myklebust xprt->snd_task = NULL; 218d19751e7STrond Myklebust if (!test_bit(XPRT_CLOSE_WAIT, &xprt->state)) { 2194e857c58SPeter Zijlstra smp_mb__before_atomic(); 220632e3bdcSTrond Myklebust clear_bit(XPRT_LOCKED, &xprt->state); 2214e857c58SPeter Zijlstra smp_mb__after_atomic(); 222632e3bdcSTrond Myklebust } else 22340a5f1b1STrond Myklebust queue_work(xprtiod_workqueue, &xprt->task_cleanup); 224632e3bdcSTrond Myklebust } 225632e3bdcSTrond Myklebust 22612a80469SChuck Lever /* 22712a80469SChuck Lever * xprt_reserve_xprt_cong - serialize write access to transports 22812a80469SChuck Lever * @task: task that is requesting access to the transport 22912a80469SChuck Lever * 23012a80469SChuck Lever * Same as xprt_reserve_xprt, but Van Jacobson congestion control is 23112a80469SChuck Lever * integrated into the decision of whether a request is allowed to be 23212a80469SChuck Lever * woken up and given access to the transport. 23312a80469SChuck Lever */ 23443cedbf0STrond Myklebust int xprt_reserve_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task) 23512a80469SChuck Lever { 2361da177e4SLinus Torvalds struct rpc_rqst *req = task->tk_rqstp; 23734006ceeSTrond Myklebust int priority; 2381da177e4SLinus Torvalds 2392226feb6SChuck Lever if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) { 2401da177e4SLinus Torvalds if (task == xprt->snd_task) 2411da177e4SLinus Torvalds return 1; 2421da177e4SLinus Torvalds goto out_sleep; 2431da177e4SLinus Torvalds } 24443cedbf0STrond Myklebust if (req == NULL) { 24543cedbf0STrond Myklebust xprt->snd_task = task; 24643cedbf0STrond Myklebust return 1; 24743cedbf0STrond Myklebust } 24812a80469SChuck Lever if (__xprt_get_cong(xprt, task)) { 2491da177e4SLinus Torvalds xprt->snd_task = task; 2501da177e4SLinus Torvalds req->rq_ntrans++; 2511da177e4SLinus Torvalds return 1; 2521da177e4SLinus Torvalds } 253632e3bdcSTrond Myklebust xprt_clear_locked(xprt); 2541da177e4SLinus Torvalds out_sleep: 25529807318SNeil Brown if (req) 25629807318SNeil Brown __xprt_put_cong(xprt, req); 25746121cf7SChuck Lever dprintk("RPC: %5u failed to lock transport %p\n", task->tk_pid, xprt); 2581da177e4SLinus Torvalds task->tk_timeout = 0; 2591da177e4SLinus Torvalds task->tk_status = -EAGAIN; 26034006ceeSTrond Myklebust if (req == NULL) 26134006ceeSTrond Myklebust priority = RPC_PRIORITY_LOW; 26234006ceeSTrond Myklebust else if (!req->rq_ntrans) 26334006ceeSTrond Myklebust priority = RPC_PRIORITY_NORMAL; 2641da177e4SLinus Torvalds else 26534006ceeSTrond Myklebust priority = RPC_PRIORITY_HIGH; 26634006ceeSTrond Myklebust rpc_sleep_on_priority(&xprt->sending, task, NULL, priority); 2671da177e4SLinus Torvalds return 0; 2681da177e4SLinus Torvalds } 26912444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_reserve_xprt_cong); 2701da177e4SLinus Torvalds 27112a80469SChuck Lever static inline int xprt_lock_write(struct rpc_xprt *xprt, struct rpc_task *task) 2721da177e4SLinus Torvalds { 2731da177e4SLinus Torvalds int retval; 2741da177e4SLinus Torvalds 2754a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 27643cedbf0STrond Myklebust retval = xprt->ops->reserve_xprt(xprt, task); 2774a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 2781da177e4SLinus Torvalds return retval; 2791da177e4SLinus Torvalds } 2801da177e4SLinus Torvalds 281961a828dSTrond Myklebust static bool __xprt_lock_write_func(struct rpc_task *task, void *data) 2821da177e4SLinus Torvalds { 283961a828dSTrond Myklebust struct rpc_xprt *xprt = data; 28449e9a890SChuck Lever struct rpc_rqst *req; 28549e9a890SChuck Lever 28649e9a890SChuck Lever req = task->tk_rqstp; 28749e9a890SChuck Lever xprt->snd_task = task; 28892551948STrond Myklebust if (req) 28949e9a890SChuck Lever req->rq_ntrans++; 290961a828dSTrond Myklebust return true; 291961a828dSTrond Myklebust } 292961a828dSTrond Myklebust 293961a828dSTrond Myklebust static void __xprt_lock_write_next(struct rpc_xprt *xprt) 294961a828dSTrond Myklebust { 295961a828dSTrond Myklebust if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) 29649e9a890SChuck Lever return; 29749e9a890SChuck Lever 298f1dc237cSTrond Myklebust if (rpc_wake_up_first_on_wq(xprtiod_workqueue, &xprt->sending, 299f1dc237cSTrond Myklebust __xprt_lock_write_func, xprt)) 300961a828dSTrond Myklebust return; 301632e3bdcSTrond Myklebust xprt_clear_locked(xprt); 30249e9a890SChuck Lever } 30349e9a890SChuck Lever 304961a828dSTrond Myklebust static bool __xprt_lock_write_cong_func(struct rpc_task *task, void *data) 30549e9a890SChuck Lever { 306961a828dSTrond Myklebust struct rpc_xprt *xprt = data; 30743cedbf0STrond Myklebust struct rpc_rqst *req; 3081da177e4SLinus Torvalds 30943cedbf0STrond Myklebust req = task->tk_rqstp; 31043cedbf0STrond Myklebust if (req == NULL) { 3111da177e4SLinus Torvalds xprt->snd_task = task; 312961a828dSTrond Myklebust return true; 31343cedbf0STrond Myklebust } 31443cedbf0STrond Myklebust if (__xprt_get_cong(xprt, task)) { 31543cedbf0STrond Myklebust xprt->snd_task = task; 3161da177e4SLinus Torvalds req->rq_ntrans++; 317961a828dSTrond Myklebust return true; 3181da177e4SLinus Torvalds } 319961a828dSTrond Myklebust return false; 320961a828dSTrond Myklebust } 321961a828dSTrond Myklebust 322961a828dSTrond Myklebust static void __xprt_lock_write_next_cong(struct rpc_xprt *xprt) 323961a828dSTrond Myklebust { 324961a828dSTrond Myklebust if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) 325961a828dSTrond Myklebust return; 326961a828dSTrond Myklebust if (RPCXPRT_CONGESTED(xprt)) 327961a828dSTrond Myklebust goto out_unlock; 328f1dc237cSTrond Myklebust if (rpc_wake_up_first_on_wq(xprtiod_workqueue, &xprt->sending, 329f1dc237cSTrond Myklebust __xprt_lock_write_cong_func, xprt)) 330961a828dSTrond Myklebust return; 3311da177e4SLinus Torvalds out_unlock: 332632e3bdcSTrond Myklebust xprt_clear_locked(xprt); 3331da177e4SLinus Torvalds } 3341da177e4SLinus Torvalds 3350695314eSTrond Myklebust static void xprt_task_clear_bytes_sent(struct rpc_task *task) 3360695314eSTrond Myklebust { 3370695314eSTrond Myklebust if (task != NULL) { 3380695314eSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 3390695314eSTrond Myklebust if (req != NULL) 3400695314eSTrond Myklebust req->rq_bytes_sent = 0; 3410695314eSTrond Myklebust } 3420695314eSTrond Myklebust } 3430695314eSTrond Myklebust 34449e9a890SChuck Lever /** 34549e9a890SChuck Lever * xprt_release_xprt - allow other requests to use a transport 34649e9a890SChuck Lever * @xprt: transport with other tasks potentially waiting 34749e9a890SChuck Lever * @task: task that is releasing access to the transport 34849e9a890SChuck Lever * 34949e9a890SChuck Lever * Note that "task" can be NULL. No congestion control is provided. 3501da177e4SLinus Torvalds */ 35149e9a890SChuck Lever void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task) 3521da177e4SLinus Torvalds { 3531da177e4SLinus Torvalds if (xprt->snd_task == task) { 3540695314eSTrond Myklebust xprt_task_clear_bytes_sent(task); 355632e3bdcSTrond Myklebust xprt_clear_locked(xprt); 3561da177e4SLinus Torvalds __xprt_lock_write_next(xprt); 3571da177e4SLinus Torvalds } 3581da177e4SLinus Torvalds } 35912444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_release_xprt); 3601da177e4SLinus Torvalds 36149e9a890SChuck Lever /** 36249e9a890SChuck Lever * xprt_release_xprt_cong - allow other requests to use a transport 36349e9a890SChuck Lever * @xprt: transport with other tasks potentially waiting 36449e9a890SChuck Lever * @task: task that is releasing access to the transport 36549e9a890SChuck Lever * 36649e9a890SChuck Lever * Note that "task" can be NULL. Another task is awoken to use the 36749e9a890SChuck Lever * transport if the transport's congestion window allows it. 36849e9a890SChuck Lever */ 36949e9a890SChuck Lever void xprt_release_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task) 37049e9a890SChuck Lever { 37149e9a890SChuck Lever if (xprt->snd_task == task) { 3720695314eSTrond Myklebust xprt_task_clear_bytes_sent(task); 373632e3bdcSTrond Myklebust xprt_clear_locked(xprt); 37449e9a890SChuck Lever __xprt_lock_write_next_cong(xprt); 37549e9a890SChuck Lever } 37649e9a890SChuck Lever } 37712444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_release_xprt_cong); 37849e9a890SChuck Lever 37949e9a890SChuck Lever static inline void xprt_release_write(struct rpc_xprt *xprt, struct rpc_task *task) 3801da177e4SLinus Torvalds { 3814a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 38249e9a890SChuck Lever xprt->ops->release_xprt(xprt, task); 3834a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 3841da177e4SLinus Torvalds } 3851da177e4SLinus Torvalds 3861da177e4SLinus Torvalds /* 3871da177e4SLinus Torvalds * Van Jacobson congestion avoidance. Check if the congestion window 3881da177e4SLinus Torvalds * overflowed. Put the task to sleep if this is the case. 3891da177e4SLinus Torvalds */ 3901da177e4SLinus Torvalds static int 3911da177e4SLinus Torvalds __xprt_get_cong(struct rpc_xprt *xprt, struct rpc_task *task) 3921da177e4SLinus Torvalds { 3931da177e4SLinus Torvalds struct rpc_rqst *req = task->tk_rqstp; 3941da177e4SLinus Torvalds 3951da177e4SLinus Torvalds if (req->rq_cong) 3961da177e4SLinus Torvalds return 1; 39746121cf7SChuck Lever dprintk("RPC: %5u xprt_cwnd_limited cong = %lu cwnd = %lu\n", 3981da177e4SLinus Torvalds task->tk_pid, xprt->cong, xprt->cwnd); 3991da177e4SLinus Torvalds if (RPCXPRT_CONGESTED(xprt)) 4001da177e4SLinus Torvalds return 0; 4011da177e4SLinus Torvalds req->rq_cong = 1; 4021da177e4SLinus Torvalds xprt->cong += RPC_CWNDSCALE; 4031da177e4SLinus Torvalds return 1; 4041da177e4SLinus Torvalds } 4051da177e4SLinus Torvalds 4061da177e4SLinus Torvalds /* 4071da177e4SLinus Torvalds * Adjust the congestion window, and wake up the next task 4081da177e4SLinus Torvalds * that has been sleeping due to congestion 4091da177e4SLinus Torvalds */ 4101da177e4SLinus Torvalds static void 4111da177e4SLinus Torvalds __xprt_put_cong(struct rpc_xprt *xprt, struct rpc_rqst *req) 4121da177e4SLinus Torvalds { 4131da177e4SLinus Torvalds if (!req->rq_cong) 4141da177e4SLinus Torvalds return; 4151da177e4SLinus Torvalds req->rq_cong = 0; 4161da177e4SLinus Torvalds xprt->cong -= RPC_CWNDSCALE; 41749e9a890SChuck Lever __xprt_lock_write_next_cong(xprt); 4181da177e4SLinus Torvalds } 4191da177e4SLinus Torvalds 42046c0ee8bSChuck Lever /** 421a58dd398SChuck Lever * xprt_release_rqst_cong - housekeeping when request is complete 422a58dd398SChuck Lever * @task: RPC request that recently completed 423a58dd398SChuck Lever * 424a58dd398SChuck Lever * Useful for transports that require congestion control. 425a58dd398SChuck Lever */ 426a58dd398SChuck Lever void xprt_release_rqst_cong(struct rpc_task *task) 427a58dd398SChuck Lever { 428a4f0835cSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 429a4f0835cSTrond Myklebust 430a4f0835cSTrond Myklebust __xprt_put_cong(req->rq_xprt, req); 431a58dd398SChuck Lever } 43212444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_release_rqst_cong); 433a58dd398SChuck Lever 434a58dd398SChuck Lever /** 43546c0ee8bSChuck Lever * xprt_adjust_cwnd - adjust transport congestion window 4366a24dfb6STrond Myklebust * @xprt: pointer to xprt 43746c0ee8bSChuck Lever * @task: recently completed RPC request used to adjust window 43846c0ee8bSChuck Lever * @result: result code of completed RPC request 43946c0ee8bSChuck Lever * 4404f4cf5adSChuck Lever * The transport code maintains an estimate on the maximum number of out- 4414f4cf5adSChuck Lever * standing RPC requests, using a smoothed version of the congestion 4424f4cf5adSChuck Lever * avoidance implemented in 44BSD. This is basically the Van Jacobson 4434f4cf5adSChuck Lever * congestion algorithm: If a retransmit occurs, the congestion window is 4444f4cf5adSChuck Lever * halved; otherwise, it is incremented by 1/cwnd when 4454f4cf5adSChuck Lever * 4464f4cf5adSChuck Lever * - a reply is received and 4474f4cf5adSChuck Lever * - a full number of requests are outstanding and 4484f4cf5adSChuck Lever * - the congestion window hasn't been updated recently. 4491da177e4SLinus Torvalds */ 4506a24dfb6STrond Myklebust void xprt_adjust_cwnd(struct rpc_xprt *xprt, struct rpc_task *task, int result) 4511da177e4SLinus Torvalds { 45246c0ee8bSChuck Lever struct rpc_rqst *req = task->tk_rqstp; 45346c0ee8bSChuck Lever unsigned long cwnd = xprt->cwnd; 4541da177e4SLinus Torvalds 4551da177e4SLinus Torvalds if (result >= 0 && cwnd <= xprt->cong) { 4561da177e4SLinus Torvalds /* The (cwnd >> 1) term makes sure 4571da177e4SLinus Torvalds * the result gets rounded properly. */ 4581da177e4SLinus Torvalds cwnd += (RPC_CWNDSCALE * RPC_CWNDSCALE + (cwnd >> 1)) / cwnd; 4591da177e4SLinus Torvalds if (cwnd > RPC_MAXCWND(xprt)) 4601da177e4SLinus Torvalds cwnd = RPC_MAXCWND(xprt); 46149e9a890SChuck Lever __xprt_lock_write_next_cong(xprt); 4621da177e4SLinus Torvalds } else if (result == -ETIMEDOUT) { 4631da177e4SLinus Torvalds cwnd >>= 1; 4641da177e4SLinus Torvalds if (cwnd < RPC_CWNDSCALE) 4651da177e4SLinus Torvalds cwnd = RPC_CWNDSCALE; 4661da177e4SLinus Torvalds } 4671da177e4SLinus Torvalds dprintk("RPC: cong %ld, cwnd was %ld, now %ld\n", 4681da177e4SLinus Torvalds xprt->cong, xprt->cwnd, cwnd); 4691da177e4SLinus Torvalds xprt->cwnd = cwnd; 47046c0ee8bSChuck Lever __xprt_put_cong(xprt, req); 4711da177e4SLinus Torvalds } 47212444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_adjust_cwnd); 4731da177e4SLinus Torvalds 47444fbac22SChuck Lever /** 47544fbac22SChuck Lever * xprt_wake_pending_tasks - wake all tasks on a transport's pending queue 47644fbac22SChuck Lever * @xprt: transport with waiting tasks 47744fbac22SChuck Lever * @status: result code to plant in each task before waking it 47844fbac22SChuck Lever * 47944fbac22SChuck Lever */ 48044fbac22SChuck Lever void xprt_wake_pending_tasks(struct rpc_xprt *xprt, int status) 48144fbac22SChuck Lever { 48244fbac22SChuck Lever if (status < 0) 48344fbac22SChuck Lever rpc_wake_up_status(&xprt->pending, status); 48444fbac22SChuck Lever else 48544fbac22SChuck Lever rpc_wake_up(&xprt->pending); 48644fbac22SChuck Lever } 48712444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_wake_pending_tasks); 48844fbac22SChuck Lever 489c7b2cae8SChuck Lever /** 490c7b2cae8SChuck Lever * xprt_wait_for_buffer_space - wait for transport output buffer to clear 491c7b2cae8SChuck Lever * @task: task to be put to sleep 4920b80ae42SRandy Dunlap * @action: function pointer to be executed after wait 493a9a6b52eSTrond Myklebust * 494a9a6b52eSTrond Myklebust * Note that we only set the timer for the case of RPC_IS_SOFT(), since 495a9a6b52eSTrond Myklebust * we don't in general want to force a socket disconnection due to 496a9a6b52eSTrond Myklebust * an incomplete RPC call transmission. 497c7b2cae8SChuck Lever */ 498b6ddf64fSTrond Myklebust void xprt_wait_for_buffer_space(struct rpc_task *task, rpc_action action) 499c7b2cae8SChuck Lever { 500c7b2cae8SChuck Lever struct rpc_rqst *req = task->tk_rqstp; 501c7b2cae8SChuck Lever struct rpc_xprt *xprt = req->rq_xprt; 502c7b2cae8SChuck Lever 503a9a6b52eSTrond Myklebust task->tk_timeout = RPC_IS_SOFT(task) ? req->rq_timeout : 0; 504b6ddf64fSTrond Myklebust rpc_sleep_on(&xprt->pending, task, action); 505c7b2cae8SChuck Lever } 50612444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_wait_for_buffer_space); 507c7b2cae8SChuck Lever 508c7b2cae8SChuck Lever /** 509c7b2cae8SChuck Lever * xprt_write_space - wake the task waiting for transport output buffer space 510c7b2cae8SChuck Lever * @xprt: transport with waiting tasks 511c7b2cae8SChuck Lever * 512c7b2cae8SChuck Lever * Can be called in a soft IRQ context, so xprt_write_space never sleeps. 513c7b2cae8SChuck Lever */ 514c7b2cae8SChuck Lever void xprt_write_space(struct rpc_xprt *xprt) 515c7b2cae8SChuck Lever { 516c7b2cae8SChuck Lever spin_lock_bh(&xprt->transport_lock); 517c7b2cae8SChuck Lever if (xprt->snd_task) { 51846121cf7SChuck Lever dprintk("RPC: write space: waking waiting task on " 51946121cf7SChuck Lever "xprt %p\n", xprt); 5202275cde4STrond Myklebust rpc_wake_up_queued_task_on_wq(xprtiod_workqueue, 5212275cde4STrond Myklebust &xprt->pending, xprt->snd_task); 522c7b2cae8SChuck Lever } 523c7b2cae8SChuck Lever spin_unlock_bh(&xprt->transport_lock); 524c7b2cae8SChuck Lever } 52512444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_write_space); 526c7b2cae8SChuck Lever 527fe3aca29SChuck Lever /** 528fe3aca29SChuck Lever * xprt_set_retrans_timeout_def - set a request's retransmit timeout 529fe3aca29SChuck Lever * @task: task whose timeout is to be set 530fe3aca29SChuck Lever * 531fe3aca29SChuck Lever * Set a request's retransmit timeout based on the transport's 532fe3aca29SChuck Lever * default timeout parameters. Used by transports that don't adjust 533fe3aca29SChuck Lever * the retransmit timeout based on round-trip time estimation. 534fe3aca29SChuck Lever */ 535fe3aca29SChuck Lever void xprt_set_retrans_timeout_def(struct rpc_task *task) 536fe3aca29SChuck Lever { 537fe3aca29SChuck Lever task->tk_timeout = task->tk_rqstp->rq_timeout; 538fe3aca29SChuck Lever } 53912444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_def); 540fe3aca29SChuck Lever 5412c53040fSBen Hutchings /** 542fe3aca29SChuck Lever * xprt_set_retrans_timeout_rtt - set a request's retransmit timeout 543fe3aca29SChuck Lever * @task: task whose timeout is to be set 544fe3aca29SChuck Lever * 545fe3aca29SChuck Lever * Set a request's retransmit timeout using the RTT estimator. 546fe3aca29SChuck Lever */ 547fe3aca29SChuck Lever void xprt_set_retrans_timeout_rtt(struct rpc_task *task) 548fe3aca29SChuck Lever { 549fe3aca29SChuck Lever int timer = task->tk_msg.rpc_proc->p_timer; 550ba7392bbSTrond Myklebust struct rpc_clnt *clnt = task->tk_client; 551ba7392bbSTrond Myklebust struct rpc_rtt *rtt = clnt->cl_rtt; 552fe3aca29SChuck Lever struct rpc_rqst *req = task->tk_rqstp; 553ba7392bbSTrond Myklebust unsigned long max_timeout = clnt->cl_timeout->to_maxval; 554fe3aca29SChuck Lever 555fe3aca29SChuck Lever task->tk_timeout = rpc_calc_rto(rtt, timer); 556fe3aca29SChuck Lever task->tk_timeout <<= rpc_ntimeo(rtt, timer) + req->rq_retries; 557fe3aca29SChuck Lever if (task->tk_timeout > max_timeout || task->tk_timeout == 0) 558fe3aca29SChuck Lever task->tk_timeout = max_timeout; 559fe3aca29SChuck Lever } 56012444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_rtt); 561fe3aca29SChuck Lever 5621da177e4SLinus Torvalds static void xprt_reset_majortimeo(struct rpc_rqst *req) 5631da177e4SLinus Torvalds { 564ba7392bbSTrond Myklebust const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout; 5651da177e4SLinus Torvalds 5661da177e4SLinus Torvalds req->rq_majortimeo = req->rq_timeout; 5671da177e4SLinus Torvalds if (to->to_exponential) 5681da177e4SLinus Torvalds req->rq_majortimeo <<= to->to_retries; 5691da177e4SLinus Torvalds else 5701da177e4SLinus Torvalds req->rq_majortimeo += to->to_increment * to->to_retries; 5711da177e4SLinus Torvalds if (req->rq_majortimeo > to->to_maxval || req->rq_majortimeo == 0) 5721da177e4SLinus Torvalds req->rq_majortimeo = to->to_maxval; 5731da177e4SLinus Torvalds req->rq_majortimeo += jiffies; 5741da177e4SLinus Torvalds } 5751da177e4SLinus Torvalds 5769903cd1cSChuck Lever /** 5779903cd1cSChuck Lever * xprt_adjust_timeout - adjust timeout values for next retransmit 5789903cd1cSChuck Lever * @req: RPC request containing parameters to use for the adjustment 5799903cd1cSChuck Lever * 5801da177e4SLinus Torvalds */ 5811da177e4SLinus Torvalds int xprt_adjust_timeout(struct rpc_rqst *req) 5821da177e4SLinus Torvalds { 5831da177e4SLinus Torvalds struct rpc_xprt *xprt = req->rq_xprt; 584ba7392bbSTrond Myklebust const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout; 5851da177e4SLinus Torvalds int status = 0; 5861da177e4SLinus Torvalds 5871da177e4SLinus Torvalds if (time_before(jiffies, req->rq_majortimeo)) { 5881da177e4SLinus Torvalds if (to->to_exponential) 5891da177e4SLinus Torvalds req->rq_timeout <<= 1; 5901da177e4SLinus Torvalds else 5911da177e4SLinus Torvalds req->rq_timeout += to->to_increment; 5921da177e4SLinus Torvalds if (to->to_maxval && req->rq_timeout >= to->to_maxval) 5931da177e4SLinus Torvalds req->rq_timeout = to->to_maxval; 5941da177e4SLinus Torvalds req->rq_retries++; 5951da177e4SLinus Torvalds } else { 5961da177e4SLinus Torvalds req->rq_timeout = to->to_initval; 5971da177e4SLinus Torvalds req->rq_retries = 0; 5981da177e4SLinus Torvalds xprt_reset_majortimeo(req); 5991da177e4SLinus Torvalds /* Reset the RTT counters == "slow start" */ 6004a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 6011da177e4SLinus Torvalds rpc_init_rtt(req->rq_task->tk_client->cl_rtt, to->to_initval); 6024a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 6031da177e4SLinus Torvalds status = -ETIMEDOUT; 6041da177e4SLinus Torvalds } 6051da177e4SLinus Torvalds 6061da177e4SLinus Torvalds if (req->rq_timeout == 0) { 6071da177e4SLinus Torvalds printk(KERN_WARNING "xprt_adjust_timeout: rq_timeout = 0!\n"); 6081da177e4SLinus Torvalds req->rq_timeout = 5 * HZ; 6091da177e4SLinus Torvalds } 6101da177e4SLinus Torvalds return status; 6111da177e4SLinus Torvalds } 6121da177e4SLinus Torvalds 61365f27f38SDavid Howells static void xprt_autoclose(struct work_struct *work) 6141da177e4SLinus Torvalds { 61565f27f38SDavid Howells struct rpc_xprt *xprt = 61665f27f38SDavid Howells container_of(work, struct rpc_xprt, task_cleanup); 6171da177e4SLinus Torvalds 61866af1e55STrond Myklebust clear_bit(XPRT_CLOSE_WAIT, &xprt->state); 6194876cc77STrond Myklebust xprt->ops->close(xprt); 6201da177e4SLinus Torvalds xprt_release_write(xprt, NULL); 62179234c3dSTrond Myklebust wake_up_bit(&xprt->state, XPRT_LOCKED); 6221da177e4SLinus Torvalds } 6231da177e4SLinus Torvalds 6249903cd1cSChuck Lever /** 62562da3b24STrond Myklebust * xprt_disconnect_done - mark a transport as disconnected 6269903cd1cSChuck Lever * @xprt: transport to flag for disconnect 6279903cd1cSChuck Lever * 6281da177e4SLinus Torvalds */ 62962da3b24STrond Myklebust void xprt_disconnect_done(struct rpc_xprt *xprt) 6301da177e4SLinus Torvalds { 6311da177e4SLinus Torvalds dprintk("RPC: disconnected transport %p\n", xprt); 6324a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 6331da177e4SLinus Torvalds xprt_clear_connected(xprt); 6342a491991STrond Myklebust xprt_wake_pending_tasks(xprt, -EAGAIN); 6354a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 6361da177e4SLinus Torvalds } 63762da3b24STrond Myklebust EXPORT_SYMBOL_GPL(xprt_disconnect_done); 6381da177e4SLinus Torvalds 63966af1e55STrond Myklebust /** 64066af1e55STrond Myklebust * xprt_force_disconnect - force a transport to disconnect 64166af1e55STrond Myklebust * @xprt: transport to disconnect 64266af1e55STrond Myklebust * 64366af1e55STrond Myklebust */ 64466af1e55STrond Myklebust void xprt_force_disconnect(struct rpc_xprt *xprt) 64566af1e55STrond Myklebust { 64666af1e55STrond Myklebust /* Don't race with the test_bit() in xprt_clear_locked() */ 64766af1e55STrond Myklebust spin_lock_bh(&xprt->transport_lock); 64866af1e55STrond Myklebust set_bit(XPRT_CLOSE_WAIT, &xprt->state); 64966af1e55STrond Myklebust /* Try to schedule an autoclose RPC call */ 65066af1e55STrond Myklebust if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0) 65140a5f1b1STrond Myklebust queue_work(xprtiod_workqueue, &xprt->task_cleanup); 6522a491991STrond Myklebust xprt_wake_pending_tasks(xprt, -EAGAIN); 65366af1e55STrond Myklebust spin_unlock_bh(&xprt->transport_lock); 65466af1e55STrond Myklebust } 655e2a4f4fbSChuck Lever EXPORT_SYMBOL_GPL(xprt_force_disconnect); 65666af1e55STrond Myklebust 6577f3a1d1eSTrond Myklebust static unsigned int 6587f3a1d1eSTrond Myklebust xprt_connect_cookie(struct rpc_xprt *xprt) 6597f3a1d1eSTrond Myklebust { 6607f3a1d1eSTrond Myklebust return READ_ONCE(xprt->connect_cookie); 6617f3a1d1eSTrond Myklebust } 6627f3a1d1eSTrond Myklebust 6637f3a1d1eSTrond Myklebust static bool 6647f3a1d1eSTrond Myklebust xprt_request_retransmit_after_disconnect(struct rpc_task *task) 6657f3a1d1eSTrond Myklebust { 6667f3a1d1eSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 6677f3a1d1eSTrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 6687f3a1d1eSTrond Myklebust 6697f3a1d1eSTrond Myklebust return req->rq_connect_cookie != xprt_connect_cookie(xprt) || 6707f3a1d1eSTrond Myklebust !xprt_connected(xprt); 6717f3a1d1eSTrond Myklebust } 6727f3a1d1eSTrond Myklebust 6737c1d71cfSTrond Myklebust /** 6747c1d71cfSTrond Myklebust * xprt_conditional_disconnect - force a transport to disconnect 6757c1d71cfSTrond Myklebust * @xprt: transport to disconnect 6767c1d71cfSTrond Myklebust * @cookie: 'connection cookie' 6777c1d71cfSTrond Myklebust * 6787c1d71cfSTrond Myklebust * This attempts to break the connection if and only if 'cookie' matches 6797c1d71cfSTrond Myklebust * the current transport 'connection cookie'. It ensures that we don't 6807c1d71cfSTrond Myklebust * try to break the connection more than once when we need to retransmit 6817c1d71cfSTrond Myklebust * a batch of RPC requests. 6827c1d71cfSTrond Myklebust * 6837c1d71cfSTrond Myklebust */ 6847c1d71cfSTrond Myklebust void xprt_conditional_disconnect(struct rpc_xprt *xprt, unsigned int cookie) 6857c1d71cfSTrond Myklebust { 6867c1d71cfSTrond Myklebust /* Don't race with the test_bit() in xprt_clear_locked() */ 6877c1d71cfSTrond Myklebust spin_lock_bh(&xprt->transport_lock); 6887c1d71cfSTrond Myklebust if (cookie != xprt->connect_cookie) 6897c1d71cfSTrond Myklebust goto out; 6902c2ee6d2SNeilBrown if (test_bit(XPRT_CLOSING, &xprt->state)) 6917c1d71cfSTrond Myklebust goto out; 6927c1d71cfSTrond Myklebust set_bit(XPRT_CLOSE_WAIT, &xprt->state); 6937c1d71cfSTrond Myklebust /* Try to schedule an autoclose RPC call */ 6947c1d71cfSTrond Myklebust if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0) 69540a5f1b1STrond Myklebust queue_work(xprtiod_workqueue, &xprt->task_cleanup); 6962a491991STrond Myklebust xprt_wake_pending_tasks(xprt, -EAGAIN); 6977c1d71cfSTrond Myklebust out: 6987c1d71cfSTrond Myklebust spin_unlock_bh(&xprt->transport_lock); 6997c1d71cfSTrond Myklebust } 7007c1d71cfSTrond Myklebust 701ad3331acSTrond Myklebust static bool 702ad3331acSTrond Myklebust xprt_has_timer(const struct rpc_xprt *xprt) 703ad3331acSTrond Myklebust { 704ad3331acSTrond Myklebust return xprt->idle_timeout != 0; 705ad3331acSTrond Myklebust } 706ad3331acSTrond Myklebust 707ad3331acSTrond Myklebust static void 708ad3331acSTrond Myklebust xprt_schedule_autodisconnect(struct rpc_xprt *xprt) 709ad3331acSTrond Myklebust __must_hold(&xprt->transport_lock) 710ad3331acSTrond Myklebust { 711ef3f5434STrond Myklebust if (list_empty(&xprt->recv_queue) && xprt_has_timer(xprt)) 712ad3331acSTrond Myklebust mod_timer(&xprt->timer, xprt->last_used + xprt->idle_timeout); 713ad3331acSTrond Myklebust } 714ad3331acSTrond Myklebust 7151da177e4SLinus Torvalds static void 716ff861c4dSKees Cook xprt_init_autodisconnect(struct timer_list *t) 7171da177e4SLinus Torvalds { 718ff861c4dSKees Cook struct rpc_xprt *xprt = from_timer(xprt, t, timer); 7191da177e4SLinus Torvalds 7204a0f8c04SChuck Lever spin_lock(&xprt->transport_lock); 721ef3f5434STrond Myklebust if (!list_empty(&xprt->recv_queue)) 7221da177e4SLinus Torvalds goto out_abort; 723ad3331acSTrond Myklebust /* Reset xprt->last_used to avoid connect/autodisconnect cycling */ 724ad3331acSTrond Myklebust xprt->last_used = jiffies; 7252226feb6SChuck Lever if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) 7261da177e4SLinus Torvalds goto out_abort; 7274a0f8c04SChuck Lever spin_unlock(&xprt->transport_lock); 72840a5f1b1STrond Myklebust queue_work(xprtiod_workqueue, &xprt->task_cleanup); 7291da177e4SLinus Torvalds return; 7301da177e4SLinus Torvalds out_abort: 7314a0f8c04SChuck Lever spin_unlock(&xprt->transport_lock); 7321da177e4SLinus Torvalds } 7331da177e4SLinus Torvalds 734718ba5b8STrond Myklebust bool xprt_lock_connect(struct rpc_xprt *xprt, 735718ba5b8STrond Myklebust struct rpc_task *task, 736718ba5b8STrond Myklebust void *cookie) 737718ba5b8STrond Myklebust { 738718ba5b8STrond Myklebust bool ret = false; 739718ba5b8STrond Myklebust 740718ba5b8STrond Myklebust spin_lock_bh(&xprt->transport_lock); 741718ba5b8STrond Myklebust if (!test_bit(XPRT_LOCKED, &xprt->state)) 742718ba5b8STrond Myklebust goto out; 743718ba5b8STrond Myklebust if (xprt->snd_task != task) 744718ba5b8STrond Myklebust goto out; 7450695314eSTrond Myklebust xprt_task_clear_bytes_sent(task); 746718ba5b8STrond Myklebust xprt->snd_task = cookie; 747718ba5b8STrond Myklebust ret = true; 748718ba5b8STrond Myklebust out: 749718ba5b8STrond Myklebust spin_unlock_bh(&xprt->transport_lock); 750718ba5b8STrond Myklebust return ret; 751718ba5b8STrond Myklebust } 752718ba5b8STrond Myklebust 753718ba5b8STrond Myklebust void xprt_unlock_connect(struct rpc_xprt *xprt, void *cookie) 754718ba5b8STrond Myklebust { 755718ba5b8STrond Myklebust spin_lock_bh(&xprt->transport_lock); 756718ba5b8STrond Myklebust if (xprt->snd_task != cookie) 757718ba5b8STrond Myklebust goto out; 758718ba5b8STrond Myklebust if (!test_bit(XPRT_LOCKED, &xprt->state)) 759718ba5b8STrond Myklebust goto out; 760718ba5b8STrond Myklebust xprt->snd_task =NULL; 761718ba5b8STrond Myklebust xprt->ops->release_xprt(xprt, NULL); 762ad3331acSTrond Myklebust xprt_schedule_autodisconnect(xprt); 763718ba5b8STrond Myklebust out: 764718ba5b8STrond Myklebust spin_unlock_bh(&xprt->transport_lock); 76579234c3dSTrond Myklebust wake_up_bit(&xprt->state, XPRT_LOCKED); 766718ba5b8STrond Myklebust } 767718ba5b8STrond Myklebust 7689903cd1cSChuck Lever /** 7699903cd1cSChuck Lever * xprt_connect - schedule a transport connect operation 7709903cd1cSChuck Lever * @task: RPC task that is requesting the connect 7711da177e4SLinus Torvalds * 7721da177e4SLinus Torvalds */ 7731da177e4SLinus Torvalds void xprt_connect(struct rpc_task *task) 7741da177e4SLinus Torvalds { 775ad2368d6STrond Myklebust struct rpc_xprt *xprt = task->tk_rqstp->rq_xprt; 7761da177e4SLinus Torvalds 77746121cf7SChuck Lever dprintk("RPC: %5u xprt_connect xprt %p %s connected\n", task->tk_pid, 7781da177e4SLinus Torvalds xprt, (xprt_connected(xprt) ? "is" : "is not")); 7791da177e4SLinus Torvalds 780ec739ef0SChuck Lever if (!xprt_bound(xprt)) { 78101d37c42STrond Myklebust task->tk_status = -EAGAIN; 7821da177e4SLinus Torvalds return; 7831da177e4SLinus Torvalds } 7841da177e4SLinus Torvalds if (!xprt_lock_write(xprt, task)) 7851da177e4SLinus Torvalds return; 786feb8ca37STrond Myklebust 787feb8ca37STrond Myklebust if (test_and_clear_bit(XPRT_CLOSE_WAIT, &xprt->state)) 788feb8ca37STrond Myklebust xprt->ops->close(xprt); 789feb8ca37STrond Myklebust 790718ba5b8STrond Myklebust if (!xprt_connected(xprt)) { 7911da177e4SLinus Torvalds task->tk_rqstp->rq_bytes_sent = 0; 792a8ce4a8fSTrond Myklebust task->tk_timeout = task->tk_rqstp->rq_timeout; 7932c2ee6d2SNeilBrown task->tk_rqstp->rq_connect_cookie = xprt->connect_cookie; 7945d00837bSTrond Myklebust rpc_sleep_on(&xprt->pending, task, xprt_connect_status); 7950b9e7943STrond Myklebust 7960b9e7943STrond Myklebust if (test_bit(XPRT_CLOSING, &xprt->state)) 7970b9e7943STrond Myklebust return; 7980b9e7943STrond Myklebust if (xprt_test_and_set_connecting(xprt)) 7990b9e7943STrond Myklebust return; 800262ca07dSChuck Lever xprt->stat.connect_start = jiffies; 8011b092092STrond Myklebust xprt->ops->connect(xprt, task); 8021da177e4SLinus Torvalds } 803718ba5b8STrond Myklebust xprt_release_write(xprt, task); 8041da177e4SLinus Torvalds } 8051da177e4SLinus Torvalds 8069903cd1cSChuck Lever static void xprt_connect_status(struct rpc_task *task) 8071da177e4SLinus Torvalds { 808ad2368d6STrond Myklebust struct rpc_xprt *xprt = task->tk_rqstp->rq_xprt; 8091da177e4SLinus Torvalds 810cd983ef8SChuck Lever if (task->tk_status == 0) { 811262ca07dSChuck Lever xprt->stat.connect_count++; 812262ca07dSChuck Lever xprt->stat.connect_time += (long)jiffies - xprt->stat.connect_start; 81346121cf7SChuck Lever dprintk("RPC: %5u xprt_connect_status: connection established\n", 8141da177e4SLinus Torvalds task->tk_pid); 8151da177e4SLinus Torvalds return; 8161da177e4SLinus Torvalds } 8171da177e4SLinus Torvalds 8181da177e4SLinus Torvalds switch (task->tk_status) { 8190fe8d04eSTrond Myklebust case -ECONNREFUSED: 8200fe8d04eSTrond Myklebust case -ECONNRESET: 8210fe8d04eSTrond Myklebust case -ECONNABORTED: 8220fe8d04eSTrond Myklebust case -ENETUNREACH: 8230fe8d04eSTrond Myklebust case -EHOSTUNREACH: 8242fc193cfSTrond Myklebust case -EPIPE: 8252a491991STrond Myklebust case -EAGAIN: 8262a491991STrond Myklebust dprintk("RPC: %5u xprt_connect_status: retrying\n", task->tk_pid); 82723475d66SChuck Lever break; 8281da177e4SLinus Torvalds case -ETIMEDOUT: 82946121cf7SChuck Lever dprintk("RPC: %5u xprt_connect_status: connect attempt timed " 83046121cf7SChuck Lever "out\n", task->tk_pid); 8311da177e4SLinus Torvalds break; 8321da177e4SLinus Torvalds default: 83346121cf7SChuck Lever dprintk("RPC: %5u xprt_connect_status: error %d connecting to " 83446121cf7SChuck Lever "server %s\n", task->tk_pid, -task->tk_status, 8354e0038b6STrond Myklebust xprt->servername); 83623475d66SChuck Lever task->tk_status = -EIO; 83723475d66SChuck Lever } 8381da177e4SLinus Torvalds } 8391da177e4SLinus Torvalds 8409903cd1cSChuck Lever /** 8419903cd1cSChuck Lever * xprt_lookup_rqst - find an RPC request corresponding to an XID 8429903cd1cSChuck Lever * @xprt: transport on which the original request was transmitted 8439903cd1cSChuck Lever * @xid: RPC XID of incoming reply 8449903cd1cSChuck Lever * 84575c84151STrond Myklebust * Caller holds xprt->queue_lock. 8461da177e4SLinus Torvalds */ 847d8ed029dSAlexey Dobriyan struct rpc_rqst *xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid) 8481da177e4SLinus Torvalds { 8498f3a6de3SPavel Emelyanov struct rpc_rqst *entry; 8501da177e4SLinus Torvalds 851ef3f5434STrond Myklebust list_for_each_entry(entry, &xprt->recv_queue, rq_recv) 8523705ad64SJeff Layton if (entry->rq_xid == xid) { 8533705ad64SJeff Layton trace_xprt_lookup_rqst(xprt, xid, 0); 8540b87a46bSChuck Lever entry->rq_rtt = ktime_sub(ktime_get(), entry->rq_xtime); 855262ca07dSChuck Lever return entry; 8563705ad64SJeff Layton } 85746121cf7SChuck Lever 85846121cf7SChuck Lever dprintk("RPC: xprt_lookup_rqst did not find xid %08x\n", 85946121cf7SChuck Lever ntohl(xid)); 8603705ad64SJeff Layton trace_xprt_lookup_rqst(xprt, xid, -ENOENT); 861262ca07dSChuck Lever xprt->stat.bad_xids++; 862262ca07dSChuck Lever return NULL; 8631da177e4SLinus Torvalds } 86412444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_lookup_rqst); 8651da177e4SLinus Torvalds 866cf9946cdSTrond Myklebust static bool 867cf9946cdSTrond Myklebust xprt_is_pinned_rqst(struct rpc_rqst *req) 868cf9946cdSTrond Myklebust { 869cf9946cdSTrond Myklebust return atomic_read(&req->rq_pin) != 0; 870cf9946cdSTrond Myklebust } 871cf9946cdSTrond Myklebust 872729749bbSTrond Myklebust /** 873729749bbSTrond Myklebust * xprt_pin_rqst - Pin a request on the transport receive list 874729749bbSTrond Myklebust * @req: Request to pin 875729749bbSTrond Myklebust * 876729749bbSTrond Myklebust * Caller must ensure this is atomic with the call to xprt_lookup_rqst() 877cf9946cdSTrond Myklebust * so should be holding the xprt receive lock. 878729749bbSTrond Myklebust */ 879729749bbSTrond Myklebust void xprt_pin_rqst(struct rpc_rqst *req) 880729749bbSTrond Myklebust { 881cf9946cdSTrond Myklebust atomic_inc(&req->rq_pin); 882729749bbSTrond Myklebust } 8839590d083SChuck Lever EXPORT_SYMBOL_GPL(xprt_pin_rqst); 884729749bbSTrond Myklebust 885729749bbSTrond Myklebust /** 886729749bbSTrond Myklebust * xprt_unpin_rqst - Unpin a request on the transport receive list 887729749bbSTrond Myklebust * @req: Request to pin 888729749bbSTrond Myklebust * 889cf9946cdSTrond Myklebust * Caller should be holding the xprt receive lock. 890729749bbSTrond Myklebust */ 891729749bbSTrond Myklebust void xprt_unpin_rqst(struct rpc_rqst *req) 892729749bbSTrond Myklebust { 893cf9946cdSTrond Myklebust if (!test_bit(RPC_TASK_MSG_PIN_WAIT, &req->rq_task->tk_runstate)) { 894cf9946cdSTrond Myklebust atomic_dec(&req->rq_pin); 895cf9946cdSTrond Myklebust return; 896cf9946cdSTrond Myklebust } 897cf9946cdSTrond Myklebust if (atomic_dec_and_test(&req->rq_pin)) 898cf9946cdSTrond Myklebust wake_up_var(&req->rq_pin); 899729749bbSTrond Myklebust } 9009590d083SChuck Lever EXPORT_SYMBOL_GPL(xprt_unpin_rqst); 901729749bbSTrond Myklebust 902729749bbSTrond Myklebust static void xprt_wait_on_pinned_rqst(struct rpc_rqst *req) 903729749bbSTrond Myklebust { 904cf9946cdSTrond Myklebust wait_var_event(&req->rq_pin, !xprt_is_pinned_rqst(req)); 905729749bbSTrond Myklebust } 906729749bbSTrond Myklebust 907edc81dcdSTrond Myklebust static bool 908edc81dcdSTrond Myklebust xprt_request_data_received(struct rpc_task *task) 909edc81dcdSTrond Myklebust { 910edc81dcdSTrond Myklebust return !test_bit(RPC_TASK_NEED_RECV, &task->tk_runstate) && 911edc81dcdSTrond Myklebust READ_ONCE(task->tk_rqstp->rq_reply_bytes_recvd) != 0; 912edc81dcdSTrond Myklebust } 913edc81dcdSTrond Myklebust 914edc81dcdSTrond Myklebust static bool 915edc81dcdSTrond Myklebust xprt_request_need_enqueue_receive(struct rpc_task *task, struct rpc_rqst *req) 916edc81dcdSTrond Myklebust { 917edc81dcdSTrond Myklebust return !test_bit(RPC_TASK_NEED_RECV, &task->tk_runstate) && 918edc81dcdSTrond Myklebust READ_ONCE(task->tk_rqstp->rq_reply_bytes_recvd) == 0; 919edc81dcdSTrond Myklebust } 920edc81dcdSTrond Myklebust 921edc81dcdSTrond Myklebust /** 922edc81dcdSTrond Myklebust * xprt_request_enqueue_receive - Add an request to the receive queue 923edc81dcdSTrond Myklebust * @task: RPC task 924edc81dcdSTrond Myklebust * 925edc81dcdSTrond Myklebust */ 926edc81dcdSTrond Myklebust void 927edc81dcdSTrond Myklebust xprt_request_enqueue_receive(struct rpc_task *task) 928edc81dcdSTrond Myklebust { 929edc81dcdSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 930edc81dcdSTrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 931edc81dcdSTrond Myklebust 932edc81dcdSTrond Myklebust if (!xprt_request_need_enqueue_receive(task, req)) 933edc81dcdSTrond Myklebust return; 934edc81dcdSTrond Myklebust spin_lock(&xprt->queue_lock); 935edc81dcdSTrond Myklebust 936edc81dcdSTrond Myklebust /* Update the softirq receive buffer */ 937edc81dcdSTrond Myklebust memcpy(&req->rq_private_buf, &req->rq_rcv_buf, 938edc81dcdSTrond Myklebust sizeof(req->rq_private_buf)); 939edc81dcdSTrond Myklebust 940edc81dcdSTrond Myklebust /* Add request to the receive list */ 941ef3f5434STrond Myklebust list_add_tail(&req->rq_recv, &xprt->recv_queue); 942edc81dcdSTrond Myklebust set_bit(RPC_TASK_NEED_RECV, &task->tk_runstate); 943edc81dcdSTrond Myklebust spin_unlock(&xprt->queue_lock); 944edc81dcdSTrond Myklebust 945edc81dcdSTrond Myklebust xprt_reset_majortimeo(req); 946edc81dcdSTrond Myklebust /* Turn off autodisconnect */ 947edc81dcdSTrond Myklebust del_singleshot_timer_sync(&xprt->timer); 948edc81dcdSTrond Myklebust } 949edc81dcdSTrond Myklebust 950edc81dcdSTrond Myklebust /** 951edc81dcdSTrond Myklebust * xprt_request_dequeue_receive_locked - Remove a request from the receive queue 952edc81dcdSTrond Myklebust * @task: RPC task 953edc81dcdSTrond Myklebust * 954edc81dcdSTrond Myklebust * Caller must hold xprt->queue_lock. 955edc81dcdSTrond Myklebust */ 956edc81dcdSTrond Myklebust static void 957edc81dcdSTrond Myklebust xprt_request_dequeue_receive_locked(struct rpc_task *task) 958edc81dcdSTrond Myklebust { 959edc81dcdSTrond Myklebust if (test_and_clear_bit(RPC_TASK_NEED_RECV, &task->tk_runstate)) 960ef3f5434STrond Myklebust list_del(&task->tk_rqstp->rq_recv); 961edc81dcdSTrond Myklebust } 962edc81dcdSTrond Myklebust 963ecd465eeSChuck Lever /** 964ecd465eeSChuck Lever * xprt_update_rtt - Update RPC RTT statistics 965ecd465eeSChuck Lever * @task: RPC request that recently completed 966ecd465eeSChuck Lever * 96775c84151STrond Myklebust * Caller holds xprt->queue_lock. 968ecd465eeSChuck Lever */ 969ecd465eeSChuck Lever void xprt_update_rtt(struct rpc_task *task) 9701da177e4SLinus Torvalds { 9711570c1e4SChuck Lever struct rpc_rqst *req = task->tk_rqstp; 9721570c1e4SChuck Lever struct rpc_rtt *rtt = task->tk_client->cl_rtt; 97395c96174SEric Dumazet unsigned int timer = task->tk_msg.rpc_proc->p_timer; 974d60dbb20STrond Myklebust long m = usecs_to_jiffies(ktime_to_us(req->rq_rtt)); 9751570c1e4SChuck Lever 9761da177e4SLinus Torvalds if (timer) { 9771da177e4SLinus Torvalds if (req->rq_ntrans == 1) 978ff839970SChuck Lever rpc_update_rtt(rtt, timer, m); 9791570c1e4SChuck Lever rpc_set_timeo(rtt, timer, req->rq_ntrans - 1); 9801da177e4SLinus Torvalds } 9811da177e4SLinus Torvalds } 982ecd465eeSChuck Lever EXPORT_SYMBOL_GPL(xprt_update_rtt); 9831da177e4SLinus Torvalds 9841570c1e4SChuck Lever /** 9851570c1e4SChuck Lever * xprt_complete_rqst - called when reply processing is complete 9861570c1e4SChuck Lever * @task: RPC request that recently completed 9871570c1e4SChuck Lever * @copied: actual number of bytes received from the transport 9881570c1e4SChuck Lever * 98975c84151STrond Myklebust * Caller holds xprt->queue_lock. 9901570c1e4SChuck Lever */ 9911570c1e4SChuck Lever void xprt_complete_rqst(struct rpc_task *task, int copied) 9921570c1e4SChuck Lever { 9931570c1e4SChuck Lever struct rpc_rqst *req = task->tk_rqstp; 994fda13939STrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 9951da177e4SLinus Torvalds 9961570c1e4SChuck Lever dprintk("RPC: %5u xid %08x complete (%d bytes received)\n", 9971570c1e4SChuck Lever task->tk_pid, ntohl(req->rq_xid), copied); 9983705ad64SJeff Layton trace_xprt_complete_rqst(xprt, req->rq_xid, copied); 9991da177e4SLinus Torvalds 1000fda13939STrond Myklebust xprt->stat.recvs++; 1001ef759a2eSChuck Lever 10021e799b67STrond Myklebust req->rq_private_buf.len = copied; 1003dd2b63d0SRicardo Labiaga /* Ensure all writes are done before we update */ 1004dd2b63d0SRicardo Labiaga /* req->rq_reply_bytes_recvd */ 100543ac3f29STrond Myklebust smp_wmb(); 1006dd2b63d0SRicardo Labiaga req->rq_reply_bytes_recvd = copied; 1007edc81dcdSTrond Myklebust xprt_request_dequeue_receive_locked(task); 1008fda13939STrond Myklebust rpc_wake_up_queued_task(&xprt->pending, task); 10091da177e4SLinus Torvalds } 101012444809S\"Talpey, Thomas\ EXPORT_SYMBOL_GPL(xprt_complete_rqst); 10111da177e4SLinus Torvalds 101246c0ee8bSChuck Lever static void xprt_timer(struct rpc_task *task) 10131da177e4SLinus Torvalds { 10141da177e4SLinus Torvalds struct rpc_rqst *req = task->tk_rqstp; 10151da177e4SLinus Torvalds struct rpc_xprt *xprt = req->rq_xprt; 10161da177e4SLinus Torvalds 10175d00837bSTrond Myklebust if (task->tk_status != -ETIMEDOUT) 10185d00837bSTrond Myklebust return; 101946c0ee8bSChuck Lever 102082476d9fSChuck Lever trace_xprt_timer(xprt, req->rq_xid, task->tk_status); 1021dd2b63d0SRicardo Labiaga if (!req->rq_reply_bytes_recvd) { 102246c0ee8bSChuck Lever if (xprt->ops->timer) 10236a24dfb6STrond Myklebust xprt->ops->timer(xprt, task); 10245d00837bSTrond Myklebust } else 10255d00837bSTrond Myklebust task->tk_status = 0; 10261da177e4SLinus Torvalds } 10271da177e4SLinus Torvalds 10289903cd1cSChuck Lever /** 10297f3a1d1eSTrond Myklebust * xprt_request_wait_receive - wait for the reply to an RPC request 10307f3a1d1eSTrond Myklebust * @task: RPC task about to send a request 10317f3a1d1eSTrond Myklebust * 10327f3a1d1eSTrond Myklebust */ 10337f3a1d1eSTrond Myklebust void xprt_request_wait_receive(struct rpc_task *task) 10347f3a1d1eSTrond Myklebust { 10357f3a1d1eSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 10367f3a1d1eSTrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 10377f3a1d1eSTrond Myklebust 10387f3a1d1eSTrond Myklebust if (!test_bit(RPC_TASK_NEED_RECV, &task->tk_runstate)) 10397f3a1d1eSTrond Myklebust return; 10407f3a1d1eSTrond Myklebust /* 10417f3a1d1eSTrond Myklebust * Sleep on the pending queue if we're expecting a reply. 10427f3a1d1eSTrond Myklebust * The spinlock ensures atomicity between the test of 10437f3a1d1eSTrond Myklebust * req->rq_reply_bytes_recvd, and the call to rpc_sleep_on(). 10447f3a1d1eSTrond Myklebust */ 10457f3a1d1eSTrond Myklebust spin_lock(&xprt->queue_lock); 10467f3a1d1eSTrond Myklebust if (test_bit(RPC_TASK_NEED_RECV, &task->tk_runstate)) { 10477f3a1d1eSTrond Myklebust xprt->ops->set_retrans_timeout(task); 10487f3a1d1eSTrond Myklebust rpc_sleep_on(&xprt->pending, task, xprt_timer); 10497f3a1d1eSTrond Myklebust /* 10507f3a1d1eSTrond Myklebust * Send an extra queue wakeup call if the 10517f3a1d1eSTrond Myklebust * connection was dropped in case the call to 10527f3a1d1eSTrond Myklebust * rpc_sleep_on() raced. 10537f3a1d1eSTrond Myklebust */ 10547f3a1d1eSTrond Myklebust if (xprt_request_retransmit_after_disconnect(task)) 10557f3a1d1eSTrond Myklebust rpc_wake_up_queued_task_set_status(&xprt->pending, 10567f3a1d1eSTrond Myklebust task, -ENOTCONN); 10577f3a1d1eSTrond Myklebust } 10587f3a1d1eSTrond Myklebust spin_unlock(&xprt->queue_lock); 10597f3a1d1eSTrond Myklebust } 10607f3a1d1eSTrond Myklebust 1061944b0429STrond Myklebust static bool 1062944b0429STrond Myklebust xprt_request_need_enqueue_transmit(struct rpc_task *task, struct rpc_rqst *req) 1063944b0429STrond Myklebust { 1064762e4e67STrond Myklebust return !test_bit(RPC_TASK_NEED_XMIT, &task->tk_runstate); 1065944b0429STrond Myklebust } 1066944b0429STrond Myklebust 1067944b0429STrond Myklebust /** 1068944b0429STrond Myklebust * xprt_request_enqueue_transmit - queue a task for transmission 1069944b0429STrond Myklebust * @task: pointer to rpc_task 1070944b0429STrond Myklebust * 1071944b0429STrond Myklebust * Add a task to the transmission queue. 1072944b0429STrond Myklebust */ 1073944b0429STrond Myklebust void 1074944b0429STrond Myklebust xprt_request_enqueue_transmit(struct rpc_task *task) 1075944b0429STrond Myklebust { 1076944b0429STrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 1077944b0429STrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 1078944b0429STrond Myklebust 1079944b0429STrond Myklebust if (xprt_request_need_enqueue_transmit(task, req)) { 1080944b0429STrond Myklebust spin_lock(&xprt->queue_lock); 1081944b0429STrond Myklebust list_add_tail(&req->rq_xmit, &xprt->xmit_queue); 1082944b0429STrond Myklebust set_bit(RPC_TASK_NEED_XMIT, &task->tk_runstate); 1083944b0429STrond Myklebust spin_unlock(&xprt->queue_lock); 1084944b0429STrond Myklebust } 1085944b0429STrond Myklebust } 1086944b0429STrond Myklebust 1087944b0429STrond Myklebust /** 1088944b0429STrond Myklebust * xprt_request_dequeue_transmit_locked - remove a task from the transmission queue 1089944b0429STrond Myklebust * @task: pointer to rpc_task 1090944b0429STrond Myklebust * 1091944b0429STrond Myklebust * Remove a task from the transmission queue 1092944b0429STrond Myklebust * Caller must hold xprt->queue_lock 1093944b0429STrond Myklebust */ 1094944b0429STrond Myklebust static void 1095944b0429STrond Myklebust xprt_request_dequeue_transmit_locked(struct rpc_task *task) 1096944b0429STrond Myklebust { 1097944b0429STrond Myklebust xprt_task_clear_bytes_sent(task); 1098944b0429STrond Myklebust if (test_and_clear_bit(RPC_TASK_NEED_XMIT, &task->tk_runstate)) 1099944b0429STrond Myklebust list_del(&task->tk_rqstp->rq_xmit); 1100944b0429STrond Myklebust } 1101944b0429STrond Myklebust 1102944b0429STrond Myklebust /** 1103944b0429STrond Myklebust * xprt_request_dequeue_transmit - remove a task from the transmission queue 1104944b0429STrond Myklebust * @task: pointer to rpc_task 1105944b0429STrond Myklebust * 1106944b0429STrond Myklebust * Remove a task from the transmission queue 1107944b0429STrond Myklebust */ 1108944b0429STrond Myklebust static void 1109944b0429STrond Myklebust xprt_request_dequeue_transmit(struct rpc_task *task) 1110944b0429STrond Myklebust { 1111944b0429STrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 1112944b0429STrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 1113944b0429STrond Myklebust 1114944b0429STrond Myklebust spin_lock(&xprt->queue_lock); 1115944b0429STrond Myklebust xprt_request_dequeue_transmit_locked(task); 1116944b0429STrond Myklebust spin_unlock(&xprt->queue_lock); 1117944b0429STrond Myklebust } 1118944b0429STrond Myklebust 11197f3a1d1eSTrond Myklebust /** 1120762e4e67STrond Myklebust * xprt_request_need_retransmit - Test if a task needs retransmission 1121762e4e67STrond Myklebust * @task: pointer to rpc_task 1122762e4e67STrond Myklebust * 1123762e4e67STrond Myklebust * Test for whether a connection breakage requires the task to retransmit 1124762e4e67STrond Myklebust */ 1125762e4e67STrond Myklebust bool 1126762e4e67STrond Myklebust xprt_request_need_retransmit(struct rpc_task *task) 1127762e4e67STrond Myklebust { 1128762e4e67STrond Myklebust return xprt_request_retransmit_after_disconnect(task); 1129762e4e67STrond Myklebust } 1130762e4e67STrond Myklebust 1131762e4e67STrond Myklebust /** 11329903cd1cSChuck Lever * xprt_prepare_transmit - reserve the transport before sending a request 11339903cd1cSChuck Lever * @task: RPC task about to send a request 11349903cd1cSChuck Lever * 11351da177e4SLinus Torvalds */ 113690051ea7STrond Myklebust bool xprt_prepare_transmit(struct rpc_task *task) 11371da177e4SLinus Torvalds { 11381da177e4SLinus Torvalds struct rpc_rqst *req = task->tk_rqstp; 11391da177e4SLinus Torvalds struct rpc_xprt *xprt = req->rq_xprt; 114090051ea7STrond Myklebust bool ret = false; 11411da177e4SLinus Torvalds 114246121cf7SChuck Lever dprintk("RPC: %5u xprt_prepare_transmit\n", task->tk_pid); 11431da177e4SLinus Torvalds 11444a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 11458a19a0b6STrond Myklebust if (!req->rq_bytes_sent) { 11468a19a0b6STrond Myklebust if (req->rq_reply_bytes_recvd) { 114790051ea7STrond Myklebust task->tk_status = req->rq_reply_bytes_recvd; 11481da177e4SLinus Torvalds goto out_unlock; 11491da177e4SLinus Torvalds } 1150944b0429STrond Myklebust if (!test_bit(RPC_TASK_NEED_XMIT, &task->tk_runstate)) 11518a19a0b6STrond Myklebust goto out_unlock; 11528a19a0b6STrond Myklebust } 115390051ea7STrond Myklebust if (!xprt->ops->reserve_xprt(xprt, task)) { 115490051ea7STrond Myklebust task->tk_status = -EAGAIN; 115590051ea7STrond Myklebust goto out_unlock; 115690051ea7STrond Myklebust } 115790051ea7STrond Myklebust ret = true; 11581da177e4SLinus Torvalds out_unlock: 11594a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 116090051ea7STrond Myklebust return ret; 11611da177e4SLinus Torvalds } 11621da177e4SLinus Torvalds 1163e0ab53deSTrond Myklebust void xprt_end_transmit(struct rpc_task *task) 11645e5ce5beSTrond Myklebust { 1165343952faSRahul Iyer xprt_release_write(task->tk_rqstp->rq_xprt, task); 11665e5ce5beSTrond Myklebust } 11675e5ce5beSTrond Myklebust 11689903cd1cSChuck Lever /** 11699903cd1cSChuck Lever * xprt_transmit - send an RPC request on a transport 11709903cd1cSChuck Lever * @task: controlling RPC task 11719903cd1cSChuck Lever * 11729903cd1cSChuck Lever * We have to copy the iovec because sendmsg fiddles with its contents. 11739903cd1cSChuck Lever */ 11749903cd1cSChuck Lever void xprt_transmit(struct rpc_task *task) 11751da177e4SLinus Torvalds { 11761da177e4SLinus Torvalds struct rpc_rqst *req = task->tk_rqstp; 11771da177e4SLinus Torvalds struct rpc_xprt *xprt = req->rq_xprt; 117890d91b0cSTrond Myklebust unsigned int connect_cookie; 1179ff699ea8SChuck Lever int status; 11801da177e4SLinus Torvalds 118146121cf7SChuck Lever dprintk("RPC: %5u xprt_transmit(%u)\n", task->tk_pid, req->rq_slen); 11821da177e4SLinus Torvalds 1183edc81dcdSTrond Myklebust if (!req->rq_bytes_sent) { 1184edc81dcdSTrond Myklebust if (xprt_request_data_received(task)) 1185944b0429STrond Myklebust goto out_dequeue; 11863021a5bbSTrond Myklebust /* Verify that our message lies in the RPCSEC_GSS window */ 1187edc81dcdSTrond Myklebust if (rpcauth_xmit_need_reencode(task)) { 11883021a5bbSTrond Myklebust task->tk_status = -EBADMSG; 1189944b0429STrond Myklebust goto out_dequeue; 11903021a5bbSTrond Myklebust } 11911da177e4SLinus Torvalds } 11921da177e4SLinus Torvalds 119390d91b0cSTrond Myklebust connect_cookie = xprt->connect_cookie; 119450f484e2STrond Myklebust status = xprt->ops->send_request(req, task); 11953705ad64SJeff Layton trace_xprt_transmit(xprt, req->rq_xid, status); 1196c8485e4dSTrond Myklebust if (status != 0) { 1197c8485e4dSTrond Myklebust task->tk_status = status; 1198c8485e4dSTrond Myklebust return; 1199c8485e4dSTrond Myklebust } 12007ebbbc6eSTrond Myklebust 12014a068258SChuck Lever xprt_inject_disconnect(xprt); 1202c8485e4dSTrond Myklebust 120346121cf7SChuck Lever dprintk("RPC: %5u xmit complete\n", task->tk_pid); 1204468f8613SBryan Schumaker task->tk_flags |= RPC_TASK_SENT; 1205fe3aca29SChuck Lever spin_lock_bh(&xprt->transport_lock); 1206262ca07dSChuck Lever 1207262ca07dSChuck Lever xprt->stat.sends++; 1208262ca07dSChuck Lever xprt->stat.req_u += xprt->stat.sends - xprt->stat.recvs; 1209262ca07dSChuck Lever xprt->stat.bklog_u += xprt->backlog.qlen; 121015a45206SAndy Adamson xprt->stat.sending_u += xprt->sending.qlen; 121115a45206SAndy Adamson xprt->stat.pending_u += xprt->pending.qlen; 1212fe3aca29SChuck Lever spin_unlock_bh(&xprt->transport_lock); 121390d91b0cSTrond Myklebust 121490d91b0cSTrond Myklebust req->rq_connect_cookie = connect_cookie; 1215944b0429STrond Myklebust out_dequeue: 1216944b0429STrond Myklebust xprt_request_dequeue_transmit(task); 12171da177e4SLinus Torvalds } 12181da177e4SLinus Torvalds 1219ba60eb25STrond Myklebust static void xprt_add_backlog(struct rpc_xprt *xprt, struct rpc_task *task) 1220ba60eb25STrond Myklebust { 1221ba60eb25STrond Myklebust set_bit(XPRT_CONGESTED, &xprt->state); 1222ba60eb25STrond Myklebust rpc_sleep_on(&xprt->backlog, task, NULL); 1223ba60eb25STrond Myklebust } 1224ba60eb25STrond Myklebust 1225ba60eb25STrond Myklebust static void xprt_wake_up_backlog(struct rpc_xprt *xprt) 1226ba60eb25STrond Myklebust { 1227ba60eb25STrond Myklebust if (rpc_wake_up_next(&xprt->backlog) == NULL) 1228ba60eb25STrond Myklebust clear_bit(XPRT_CONGESTED, &xprt->state); 1229ba60eb25STrond Myklebust } 1230ba60eb25STrond Myklebust 1231ba60eb25STrond Myklebust static bool xprt_throttle_congested(struct rpc_xprt *xprt, struct rpc_task *task) 1232ba60eb25STrond Myklebust { 1233ba60eb25STrond Myklebust bool ret = false; 1234ba60eb25STrond Myklebust 1235ba60eb25STrond Myklebust if (!test_bit(XPRT_CONGESTED, &xprt->state)) 1236ba60eb25STrond Myklebust goto out; 1237ba60eb25STrond Myklebust spin_lock(&xprt->reserve_lock); 1238ba60eb25STrond Myklebust if (test_bit(XPRT_CONGESTED, &xprt->state)) { 1239ba60eb25STrond Myklebust rpc_sleep_on(&xprt->backlog, task, NULL); 1240ba60eb25STrond Myklebust ret = true; 1241ba60eb25STrond Myklebust } 1242ba60eb25STrond Myklebust spin_unlock(&xprt->reserve_lock); 1243ba60eb25STrond Myklebust out: 1244ba60eb25STrond Myklebust return ret; 1245ba60eb25STrond Myklebust } 1246ba60eb25STrond Myklebust 124792ea011fSTrond Myklebust static struct rpc_rqst *xprt_dynamic_alloc_slot(struct rpc_xprt *xprt) 1248d9ba131dSTrond Myklebust { 1249d9ba131dSTrond Myklebust struct rpc_rqst *req = ERR_PTR(-EAGAIN); 1250d9ba131dSTrond Myklebust 1251ff699ea8SChuck Lever if (xprt->num_reqs >= xprt->max_reqs) 1252d9ba131dSTrond Myklebust goto out; 1253ff699ea8SChuck Lever ++xprt->num_reqs; 125492ea011fSTrond Myklebust spin_unlock(&xprt->reserve_lock); 125592ea011fSTrond Myklebust req = kzalloc(sizeof(struct rpc_rqst), GFP_NOFS); 125692ea011fSTrond Myklebust spin_lock(&xprt->reserve_lock); 1257d9ba131dSTrond Myklebust if (req != NULL) 1258d9ba131dSTrond Myklebust goto out; 1259ff699ea8SChuck Lever --xprt->num_reqs; 1260d9ba131dSTrond Myklebust req = ERR_PTR(-ENOMEM); 1261d9ba131dSTrond Myklebust out: 1262d9ba131dSTrond Myklebust return req; 1263d9ba131dSTrond Myklebust } 1264d9ba131dSTrond Myklebust 1265d9ba131dSTrond Myklebust static bool xprt_dynamic_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req) 1266d9ba131dSTrond Myklebust { 1267ff699ea8SChuck Lever if (xprt->num_reqs > xprt->min_reqs) { 1268ff699ea8SChuck Lever --xprt->num_reqs; 1269d9ba131dSTrond Myklebust kfree(req); 1270d9ba131dSTrond Myklebust return true; 1271d9ba131dSTrond Myklebust } 1272d9ba131dSTrond Myklebust return false; 1273d9ba131dSTrond Myklebust } 1274d9ba131dSTrond Myklebust 1275f39c1bfbSTrond Myklebust void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task) 12761da177e4SLinus Torvalds { 1277d9ba131dSTrond Myklebust struct rpc_rqst *req; 12781da177e4SLinus Torvalds 1279f39c1bfbSTrond Myklebust spin_lock(&xprt->reserve_lock); 12801da177e4SLinus Torvalds if (!list_empty(&xprt->free)) { 1281d9ba131dSTrond Myklebust req = list_entry(xprt->free.next, struct rpc_rqst, rq_list); 1282d9ba131dSTrond Myklebust list_del(&req->rq_list); 1283d9ba131dSTrond Myklebust goto out_init_req; 1284d9ba131dSTrond Myklebust } 128592ea011fSTrond Myklebust req = xprt_dynamic_alloc_slot(xprt); 1286d9ba131dSTrond Myklebust if (!IS_ERR(req)) 1287d9ba131dSTrond Myklebust goto out_init_req; 1288d9ba131dSTrond Myklebust switch (PTR_ERR(req)) { 1289d9ba131dSTrond Myklebust case -ENOMEM: 1290d9ba131dSTrond Myklebust dprintk("RPC: dynamic allocation of request slot " 1291d9ba131dSTrond Myklebust "failed! Retrying\n"); 12921afeaf5cSTrond Myklebust task->tk_status = -ENOMEM; 1293d9ba131dSTrond Myklebust break; 1294d9ba131dSTrond Myklebust case -EAGAIN: 1295ba60eb25STrond Myklebust xprt_add_backlog(xprt, task); 1296d9ba131dSTrond Myklebust dprintk("RPC: waiting for request slot\n"); 1297e9d47639SGustavo A. R. Silva /* fall through */ 12981afeaf5cSTrond Myklebust default: 1299d9ba131dSTrond Myklebust task->tk_status = -EAGAIN; 13001afeaf5cSTrond Myklebust } 1301f39c1bfbSTrond Myklebust spin_unlock(&xprt->reserve_lock); 1302d9ba131dSTrond Myklebust return; 1303d9ba131dSTrond Myklebust out_init_req: 1304ff699ea8SChuck Lever xprt->stat.max_slots = max_t(unsigned int, xprt->stat.max_slots, 1305ff699ea8SChuck Lever xprt->num_reqs); 130637ac86c3SChuck Lever spin_unlock(&xprt->reserve_lock); 130737ac86c3SChuck Lever 1308d9ba131dSTrond Myklebust task->tk_status = 0; 13091da177e4SLinus Torvalds task->tk_rqstp = req; 13101da177e4SLinus Torvalds } 1311f39c1bfbSTrond Myklebust EXPORT_SYMBOL_GPL(xprt_alloc_slot); 1312f39c1bfbSTrond Myklebust 1313f39c1bfbSTrond Myklebust void xprt_lock_and_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task) 1314f39c1bfbSTrond Myklebust { 1315f39c1bfbSTrond Myklebust /* Note: grabbing the xprt_lock_write() ensures that we throttle 1316f39c1bfbSTrond Myklebust * new slot allocation if the transport is congested (i.e. when 1317f39c1bfbSTrond Myklebust * reconnecting a stream transport or when out of socket write 1318f39c1bfbSTrond Myklebust * buffer space). 1319f39c1bfbSTrond Myklebust */ 1320f39c1bfbSTrond Myklebust if (xprt_lock_write(xprt, task)) { 1321f39c1bfbSTrond Myklebust xprt_alloc_slot(xprt, task); 1322f39c1bfbSTrond Myklebust xprt_release_write(xprt, task); 1323f39c1bfbSTrond Myklebust } 1324f39c1bfbSTrond Myklebust } 1325f39c1bfbSTrond Myklebust EXPORT_SYMBOL_GPL(xprt_lock_and_alloc_slot); 13261da177e4SLinus Torvalds 1327a9cde23aSChuck Lever void xprt_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req) 1328ee5ebe85STrond Myklebust { 1329ee5ebe85STrond Myklebust spin_lock(&xprt->reserve_lock); 1330c25573b5STrond Myklebust if (!xprt_dynamic_free_slot(xprt, req)) { 1331c25573b5STrond Myklebust memset(req, 0, sizeof(*req)); /* mark unused */ 1332ee5ebe85STrond Myklebust list_add(&req->rq_list, &xprt->free); 1333c25573b5STrond Myklebust } 1334ba60eb25STrond Myklebust xprt_wake_up_backlog(xprt); 1335ee5ebe85STrond Myklebust spin_unlock(&xprt->reserve_lock); 1336ee5ebe85STrond Myklebust } 1337a9cde23aSChuck Lever EXPORT_SYMBOL_GPL(xprt_free_slot); 1338ee5ebe85STrond Myklebust 133921de0a95STrond Myklebust static void xprt_free_all_slots(struct rpc_xprt *xprt) 134021de0a95STrond Myklebust { 134121de0a95STrond Myklebust struct rpc_rqst *req; 134221de0a95STrond Myklebust while (!list_empty(&xprt->free)) { 134321de0a95STrond Myklebust req = list_first_entry(&xprt->free, struct rpc_rqst, rq_list); 134421de0a95STrond Myklebust list_del(&req->rq_list); 134521de0a95STrond Myklebust kfree(req); 134621de0a95STrond Myklebust } 134721de0a95STrond Myklebust } 134821de0a95STrond Myklebust 1349d9ba131dSTrond Myklebust struct rpc_xprt *xprt_alloc(struct net *net, size_t size, 1350d9ba131dSTrond Myklebust unsigned int num_prealloc, 1351d9ba131dSTrond Myklebust unsigned int max_alloc) 1352bd1722d4SPavel Emelyanov { 1353bd1722d4SPavel Emelyanov struct rpc_xprt *xprt; 135421de0a95STrond Myklebust struct rpc_rqst *req; 135521de0a95STrond Myklebust int i; 1356bd1722d4SPavel Emelyanov 1357bd1722d4SPavel Emelyanov xprt = kzalloc(size, GFP_KERNEL); 1358bd1722d4SPavel Emelyanov if (xprt == NULL) 1359bd1722d4SPavel Emelyanov goto out; 1360bd1722d4SPavel Emelyanov 136121de0a95STrond Myklebust xprt_init(xprt, net); 136221de0a95STrond Myklebust 136321de0a95STrond Myklebust for (i = 0; i < num_prealloc; i++) { 136421de0a95STrond Myklebust req = kzalloc(sizeof(struct rpc_rqst), GFP_KERNEL); 136521de0a95STrond Myklebust if (!req) 13668313164cSwangweidong goto out_free; 136721de0a95STrond Myklebust list_add(&req->rq_list, &xprt->free); 136821de0a95STrond Myklebust } 1369d9ba131dSTrond Myklebust if (max_alloc > num_prealloc) 1370d9ba131dSTrond Myklebust xprt->max_reqs = max_alloc; 1371d9ba131dSTrond Myklebust else 137221de0a95STrond Myklebust xprt->max_reqs = num_prealloc; 1373d9ba131dSTrond Myklebust xprt->min_reqs = num_prealloc; 1374ff699ea8SChuck Lever xprt->num_reqs = num_prealloc; 1375bd1722d4SPavel Emelyanov 1376bd1722d4SPavel Emelyanov return xprt; 1377bd1722d4SPavel Emelyanov 1378bd1722d4SPavel Emelyanov out_free: 137921de0a95STrond Myklebust xprt_free(xprt); 1380bd1722d4SPavel Emelyanov out: 1381bd1722d4SPavel Emelyanov return NULL; 1382bd1722d4SPavel Emelyanov } 1383bd1722d4SPavel Emelyanov EXPORT_SYMBOL_GPL(xprt_alloc); 1384bd1722d4SPavel Emelyanov 1385e204e621SPavel Emelyanov void xprt_free(struct rpc_xprt *xprt) 1386e204e621SPavel Emelyanov { 138737aa2133SPavel Emelyanov put_net(xprt->xprt_net); 138821de0a95STrond Myklebust xprt_free_all_slots(xprt); 1389fda1bfefSTrond Myklebust kfree_rcu(xprt, rcu); 1390e204e621SPavel Emelyanov } 1391e204e621SPavel Emelyanov EXPORT_SYMBOL_GPL(xprt_free); 1392e204e621SPavel Emelyanov 1393902c5887STrond Myklebust static void 1394902c5887STrond Myklebust xprt_init_connect_cookie(struct rpc_rqst *req, struct rpc_xprt *xprt) 1395902c5887STrond Myklebust { 1396902c5887STrond Myklebust req->rq_connect_cookie = xprt_connect_cookie(xprt) - 1; 1397902c5887STrond Myklebust } 1398902c5887STrond Myklebust 13999dc6edcfSTrond Myklebust static __be32 14009dc6edcfSTrond Myklebust xprt_alloc_xid(struct rpc_xprt *xprt) 14019dc6edcfSTrond Myklebust { 14029dc6edcfSTrond Myklebust __be32 xid; 14039dc6edcfSTrond Myklebust 14049dc6edcfSTrond Myklebust spin_lock(&xprt->reserve_lock); 14059dc6edcfSTrond Myklebust xid = (__force __be32)xprt->xid++; 14069dc6edcfSTrond Myklebust spin_unlock(&xprt->reserve_lock); 14079dc6edcfSTrond Myklebust return xid; 14089dc6edcfSTrond Myklebust } 14099dc6edcfSTrond Myklebust 14109dc6edcfSTrond Myklebust static void 14119dc6edcfSTrond Myklebust xprt_init_xid(struct rpc_xprt *xprt) 14129dc6edcfSTrond Myklebust { 14139dc6edcfSTrond Myklebust xprt->xid = prandom_u32(); 14149dc6edcfSTrond Myklebust } 14159dc6edcfSTrond Myklebust 14169dc6edcfSTrond Myklebust static void 14179dc6edcfSTrond Myklebust xprt_request_init(struct rpc_task *task) 14189dc6edcfSTrond Myklebust { 14199dc6edcfSTrond Myklebust struct rpc_xprt *xprt = task->tk_xprt; 14209dc6edcfSTrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 14219dc6edcfSTrond Myklebust 14229dc6edcfSTrond Myklebust req->rq_timeout = task->tk_client->cl_timeout->to_initval; 14239dc6edcfSTrond Myklebust req->rq_task = task; 14249dc6edcfSTrond Myklebust req->rq_xprt = xprt; 14259dc6edcfSTrond Myklebust req->rq_buffer = NULL; 14269dc6edcfSTrond Myklebust req->rq_xid = xprt_alloc_xid(xprt); 1427902c5887STrond Myklebust xprt_init_connect_cookie(req, xprt); 14289dc6edcfSTrond Myklebust req->rq_bytes_sent = 0; 14299dc6edcfSTrond Myklebust req->rq_snd_buf.len = 0; 14309dc6edcfSTrond Myklebust req->rq_snd_buf.buflen = 0; 14319dc6edcfSTrond Myklebust req->rq_rcv_buf.len = 0; 14329dc6edcfSTrond Myklebust req->rq_rcv_buf.buflen = 0; 14339dc6edcfSTrond Myklebust req->rq_release_snd_buf = NULL; 14349dc6edcfSTrond Myklebust xprt_reset_majortimeo(req); 14359dc6edcfSTrond Myklebust dprintk("RPC: %5u reserved req %p xid %08x\n", task->tk_pid, 14369dc6edcfSTrond Myklebust req, ntohl(req->rq_xid)); 14379dc6edcfSTrond Myklebust } 14389dc6edcfSTrond Myklebust 14399dc6edcfSTrond Myklebust static void 14409dc6edcfSTrond Myklebust xprt_do_reserve(struct rpc_xprt *xprt, struct rpc_task *task) 14419dc6edcfSTrond Myklebust { 14429dc6edcfSTrond Myklebust xprt->ops->alloc_slot(xprt, task); 14439dc6edcfSTrond Myklebust if (task->tk_rqstp != NULL) 14449dc6edcfSTrond Myklebust xprt_request_init(task); 14459dc6edcfSTrond Myklebust } 14469dc6edcfSTrond Myklebust 14479903cd1cSChuck Lever /** 14489903cd1cSChuck Lever * xprt_reserve - allocate an RPC request slot 14499903cd1cSChuck Lever * @task: RPC task requesting a slot allocation 14509903cd1cSChuck Lever * 1451ba60eb25STrond Myklebust * If the transport is marked as being congested, or if no more 1452ba60eb25STrond Myklebust * slots are available, place the task on the transport's 14539903cd1cSChuck Lever * backlog queue. 14549903cd1cSChuck Lever */ 14559903cd1cSChuck Lever void xprt_reserve(struct rpc_task *task) 14561da177e4SLinus Torvalds { 1457fb43d172STrond Myklebust struct rpc_xprt *xprt = task->tk_xprt; 14581da177e4SLinus Torvalds 145943cedbf0STrond Myklebust task->tk_status = 0; 146043cedbf0STrond Myklebust if (task->tk_rqstp != NULL) 146143cedbf0STrond Myklebust return; 146243cedbf0STrond Myklebust 146343cedbf0STrond Myklebust task->tk_timeout = 0; 146443cedbf0STrond Myklebust task->tk_status = -EAGAIN; 1465ba60eb25STrond Myklebust if (!xprt_throttle_congested(xprt, task)) 14669dc6edcfSTrond Myklebust xprt_do_reserve(xprt, task); 1467ba60eb25STrond Myklebust } 1468ba60eb25STrond Myklebust 1469ba60eb25STrond Myklebust /** 1470ba60eb25STrond Myklebust * xprt_retry_reserve - allocate an RPC request slot 1471ba60eb25STrond Myklebust * @task: RPC task requesting a slot allocation 1472ba60eb25STrond Myklebust * 1473ba60eb25STrond Myklebust * If no more slots are available, place the task on the transport's 1474ba60eb25STrond Myklebust * backlog queue. 1475ba60eb25STrond Myklebust * Note that the only difference with xprt_reserve is that we now 1476ba60eb25STrond Myklebust * ignore the value of the XPRT_CONGESTED flag. 1477ba60eb25STrond Myklebust */ 1478ba60eb25STrond Myklebust void xprt_retry_reserve(struct rpc_task *task) 1479ba60eb25STrond Myklebust { 1480fb43d172STrond Myklebust struct rpc_xprt *xprt = task->tk_xprt; 1481ba60eb25STrond Myklebust 1482ba60eb25STrond Myklebust task->tk_status = 0; 1483ba60eb25STrond Myklebust if (task->tk_rqstp != NULL) 1484ba60eb25STrond Myklebust return; 1485ba60eb25STrond Myklebust 1486ba60eb25STrond Myklebust task->tk_timeout = 0; 1487ba60eb25STrond Myklebust task->tk_status = -EAGAIN; 14889dc6edcfSTrond Myklebust xprt_do_reserve(xprt, task); 14891da177e4SLinus Torvalds } 14901da177e4SLinus Torvalds 1491edc81dcdSTrond Myklebust static void 1492edc81dcdSTrond Myklebust xprt_request_dequeue_all(struct rpc_task *task, struct rpc_rqst *req) 1493edc81dcdSTrond Myklebust { 1494edc81dcdSTrond Myklebust struct rpc_xprt *xprt = req->rq_xprt; 1495edc81dcdSTrond Myklebust 1496944b0429STrond Myklebust if (test_bit(RPC_TASK_NEED_XMIT, &task->tk_runstate) || 1497944b0429STrond Myklebust test_bit(RPC_TASK_NEED_RECV, &task->tk_runstate) || 1498edc81dcdSTrond Myklebust xprt_is_pinned_rqst(req)) { 1499edc81dcdSTrond Myklebust spin_lock(&xprt->queue_lock); 1500944b0429STrond Myklebust xprt_request_dequeue_transmit_locked(task); 1501edc81dcdSTrond Myklebust xprt_request_dequeue_receive_locked(task); 1502edc81dcdSTrond Myklebust while (xprt_is_pinned_rqst(req)) { 1503edc81dcdSTrond Myklebust set_bit(RPC_TASK_MSG_PIN_WAIT, &task->tk_runstate); 1504edc81dcdSTrond Myklebust spin_unlock(&xprt->queue_lock); 1505edc81dcdSTrond Myklebust xprt_wait_on_pinned_rqst(req); 1506edc81dcdSTrond Myklebust spin_lock(&xprt->queue_lock); 1507edc81dcdSTrond Myklebust clear_bit(RPC_TASK_MSG_PIN_WAIT, &task->tk_runstate); 1508edc81dcdSTrond Myklebust } 1509edc81dcdSTrond Myklebust spin_unlock(&xprt->queue_lock); 1510edc81dcdSTrond Myklebust } 1511edc81dcdSTrond Myklebust } 1512edc81dcdSTrond Myklebust 15139903cd1cSChuck Lever /** 15149903cd1cSChuck Lever * xprt_release - release an RPC request slot 15159903cd1cSChuck Lever * @task: task which is finished with the slot 15169903cd1cSChuck Lever * 15171da177e4SLinus Torvalds */ 15189903cd1cSChuck Lever void xprt_release(struct rpc_task *task) 15191da177e4SLinus Torvalds { 152055ae1aabSRicardo Labiaga struct rpc_xprt *xprt; 152187ed5003STrond Myklebust struct rpc_rqst *req = task->tk_rqstp; 15221da177e4SLinus Torvalds 152387ed5003STrond Myklebust if (req == NULL) { 152487ed5003STrond Myklebust if (task->tk_client) { 1525fb43d172STrond Myklebust xprt = task->tk_xprt; 152687ed5003STrond Myklebust if (xprt->snd_task == task) 152787ed5003STrond Myklebust xprt_release_write(xprt, task); 152887ed5003STrond Myklebust } 15291da177e4SLinus Torvalds return; 153087ed5003STrond Myklebust } 153155ae1aabSRicardo Labiaga 153255ae1aabSRicardo Labiaga xprt = req->rq_xprt; 15330a702195SWeston Andros Adamson if (task->tk_ops->rpc_count_stats != NULL) 15340a702195SWeston Andros Adamson task->tk_ops->rpc_count_stats(task, task->tk_calldata); 15350a702195SWeston Andros Adamson else if (task->tk_client) 15360a702195SWeston Andros Adamson rpc_count_iostats(task, task->tk_client->cl_metrics); 1537edc81dcdSTrond Myklebust xprt_request_dequeue_all(task, req); 15384a0f8c04SChuck Lever spin_lock_bh(&xprt->transport_lock); 153949e9a890SChuck Lever xprt->ops->release_xprt(xprt, task); 1540a58dd398SChuck Lever if (xprt->ops->release_request) 1541a58dd398SChuck Lever xprt->ops->release_request(task); 15421da177e4SLinus Torvalds xprt->last_used = jiffies; 1543ad3331acSTrond Myklebust xprt_schedule_autodisconnect(xprt); 15444a0f8c04SChuck Lever spin_unlock_bh(&xprt->transport_lock); 1545ee5ebe85STrond Myklebust if (req->rq_buffer) 15463435c74aSChuck Lever xprt->ops->buf_free(task); 15474a068258SChuck Lever xprt_inject_disconnect(xprt); 1548a17c2153STrond Myklebust if (req->rq_cred != NULL) 1549a17c2153STrond Myklebust put_rpccred(req->rq_cred); 15501da177e4SLinus Torvalds task->tk_rqstp = NULL; 1551ead5e1c2SJ. Bruce Fields if (req->rq_release_snd_buf) 1552ead5e1c2SJ. Bruce Fields req->rq_release_snd_buf(req); 155355ae1aabSRicardo Labiaga 155446121cf7SChuck Lever dprintk("RPC: %5u release request %p\n", task->tk_pid, req); 1555ee5ebe85STrond Myklebust if (likely(!bc_prealloc(req))) 1556a9cde23aSChuck Lever xprt->ops->free_slot(xprt, req); 1557ee5ebe85STrond Myklebust else 1558c9acb42eSTrond Myklebust xprt_free_bc_request(req); 15591da177e4SLinus Torvalds } 15601da177e4SLinus Torvalds 1561902c5887STrond Myklebust #ifdef CONFIG_SUNRPC_BACKCHANNEL 1562902c5887STrond Myklebust void 1563902c5887STrond Myklebust xprt_init_bc_request(struct rpc_rqst *req, struct rpc_task *task) 1564902c5887STrond Myklebust { 1565902c5887STrond Myklebust struct xdr_buf *xbufp = &req->rq_snd_buf; 1566902c5887STrond Myklebust 1567902c5887STrond Myklebust task->tk_rqstp = req; 1568902c5887STrond Myklebust req->rq_task = task; 1569902c5887STrond Myklebust xprt_init_connect_cookie(req, req->rq_xprt); 1570902c5887STrond Myklebust /* 1571902c5887STrond Myklebust * Set up the xdr_buf length. 1572902c5887STrond Myklebust * This also indicates that the buffer is XDR encoded already. 1573902c5887STrond Myklebust */ 1574902c5887STrond Myklebust xbufp->len = xbufp->head[0].iov_len + xbufp->page_len + 1575902c5887STrond Myklebust xbufp->tail[0].iov_len; 1576902c5887STrond Myklebust req->rq_bytes_sent = 0; 1577902c5887STrond Myklebust } 1578902c5887STrond Myklebust #endif 1579902c5887STrond Myklebust 158021de0a95STrond Myklebust static void xprt_init(struct rpc_xprt *xprt, struct net *net) 1581c2866763SChuck Lever { 158230c5116bSTrond Myklebust kref_init(&xprt->kref); 1583c2866763SChuck Lever 1584c2866763SChuck Lever spin_lock_init(&xprt->transport_lock); 1585c2866763SChuck Lever spin_lock_init(&xprt->reserve_lock); 158675c84151STrond Myklebust spin_lock_init(&xprt->queue_lock); 1587c2866763SChuck Lever 1588c2866763SChuck Lever INIT_LIST_HEAD(&xprt->free); 1589ef3f5434STrond Myklebust INIT_LIST_HEAD(&xprt->recv_queue); 1590944b0429STrond Myklebust INIT_LIST_HEAD(&xprt->xmit_queue); 15919e00abc3STrond Myklebust #if defined(CONFIG_SUNRPC_BACKCHANNEL) 1592f9acac1aSRicardo Labiaga spin_lock_init(&xprt->bc_pa_lock); 1593f9acac1aSRicardo Labiaga INIT_LIST_HEAD(&xprt->bc_pa_list); 15949e00abc3STrond Myklebust #endif /* CONFIG_SUNRPC_BACKCHANNEL */ 159580b14d5eSTrond Myklebust INIT_LIST_HEAD(&xprt->xprt_switch); 1596f9acac1aSRicardo Labiaga 1597c2866763SChuck Lever xprt->last_used = jiffies; 1598c2866763SChuck Lever xprt->cwnd = RPC_INITCWND; 1599a509050bSChuck Lever xprt->bind_index = 0; 1600c2866763SChuck Lever 1601c2866763SChuck Lever rpc_init_wait_queue(&xprt->binding, "xprt_binding"); 1602c2866763SChuck Lever rpc_init_wait_queue(&xprt->pending, "xprt_pending"); 160334006ceeSTrond Myklebust rpc_init_priority_wait_queue(&xprt->sending, "xprt_sending"); 1604c2866763SChuck Lever rpc_init_priority_wait_queue(&xprt->backlog, "xprt_backlog"); 1605c2866763SChuck Lever 1606c2866763SChuck Lever xprt_init_xid(xprt); 1607c2866763SChuck Lever 160821de0a95STrond Myklebust xprt->xprt_net = get_net(net); 16098d9266ffSTrond Myklebust } 16108d9266ffSTrond Myklebust 16118d9266ffSTrond Myklebust /** 16128d9266ffSTrond Myklebust * xprt_create_transport - create an RPC transport 16138d9266ffSTrond Myklebust * @args: rpc transport creation arguments 16148d9266ffSTrond Myklebust * 16158d9266ffSTrond Myklebust */ 16168d9266ffSTrond Myklebust struct rpc_xprt *xprt_create_transport(struct xprt_create *args) 16178d9266ffSTrond Myklebust { 16188d9266ffSTrond Myklebust struct rpc_xprt *xprt; 16198d9266ffSTrond Myklebust struct xprt_class *t; 16208d9266ffSTrond Myklebust 16218d9266ffSTrond Myklebust spin_lock(&xprt_list_lock); 16228d9266ffSTrond Myklebust list_for_each_entry(t, &xprt_list, list) { 16238d9266ffSTrond Myklebust if (t->ident == args->ident) { 16248d9266ffSTrond Myklebust spin_unlock(&xprt_list_lock); 16258d9266ffSTrond Myklebust goto found; 16268d9266ffSTrond Myklebust } 16278d9266ffSTrond Myklebust } 16288d9266ffSTrond Myklebust spin_unlock(&xprt_list_lock); 16293c45ddf8SChuck Lever dprintk("RPC: transport (%d) not supported\n", args->ident); 16308d9266ffSTrond Myklebust return ERR_PTR(-EIO); 16318d9266ffSTrond Myklebust 16328d9266ffSTrond Myklebust found: 16338d9266ffSTrond Myklebust xprt = t->setup(args); 16348d9266ffSTrond Myklebust if (IS_ERR(xprt)) { 16358d9266ffSTrond Myklebust dprintk("RPC: xprt_create_transport: failed, %ld\n", 16368d9266ffSTrond Myklebust -PTR_ERR(xprt)); 163721de0a95STrond Myklebust goto out; 16388d9266ffSTrond Myklebust } 163933d90ac0SJ. Bruce Fields if (args->flags & XPRT_CREATE_NO_IDLE_TIMEOUT) 164033d90ac0SJ. Bruce Fields xprt->idle_timeout = 0; 164121de0a95STrond Myklebust INIT_WORK(&xprt->task_cleanup, xprt_autoclose); 164221de0a95STrond Myklebust if (xprt_has_timer(xprt)) 1643ff861c4dSKees Cook timer_setup(&xprt->timer, xprt_init_autodisconnect, 0); 164421de0a95STrond Myklebust else 1645ff861c4dSKees Cook timer_setup(&xprt->timer, NULL, 0); 16464e0038b6STrond Myklebust 16474e0038b6STrond Myklebust if (strlen(args->servername) > RPC_MAXNETNAMELEN) { 16484e0038b6STrond Myklebust xprt_destroy(xprt); 16494e0038b6STrond Myklebust return ERR_PTR(-EINVAL); 16504e0038b6STrond Myklebust } 16514e0038b6STrond Myklebust xprt->servername = kstrdup(args->servername, GFP_KERNEL); 16524e0038b6STrond Myklebust if (xprt->servername == NULL) { 16534e0038b6STrond Myklebust xprt_destroy(xprt); 16544e0038b6STrond Myklebust return ERR_PTR(-ENOMEM); 16554e0038b6STrond Myklebust } 16564e0038b6STrond Myklebust 16573f940098SJeff Layton rpc_xprt_debugfs_register(xprt); 1658388f0c77SJeff Layton 1659c2866763SChuck Lever dprintk("RPC: created transport %p with %u slots\n", xprt, 1660c2866763SChuck Lever xprt->max_reqs); 166121de0a95STrond Myklebust out: 1662c2866763SChuck Lever return xprt; 1663c2866763SChuck Lever } 1664c2866763SChuck Lever 1665528fd354STrond Myklebust static void xprt_destroy_cb(struct work_struct *work) 1666528fd354STrond Myklebust { 1667528fd354STrond Myklebust struct rpc_xprt *xprt = 1668528fd354STrond Myklebust container_of(work, struct rpc_xprt, task_cleanup); 1669528fd354STrond Myklebust 1670528fd354STrond Myklebust rpc_xprt_debugfs_unregister(xprt); 1671528fd354STrond Myklebust rpc_destroy_wait_queue(&xprt->binding); 1672528fd354STrond Myklebust rpc_destroy_wait_queue(&xprt->pending); 1673528fd354STrond Myklebust rpc_destroy_wait_queue(&xprt->sending); 1674528fd354STrond Myklebust rpc_destroy_wait_queue(&xprt->backlog); 1675528fd354STrond Myklebust kfree(xprt->servername); 1676528fd354STrond Myklebust /* 1677528fd354STrond Myklebust * Tear down transport state and free the rpc_xprt 1678528fd354STrond Myklebust */ 1679528fd354STrond Myklebust xprt->ops->destroy(xprt); 1680528fd354STrond Myklebust } 1681528fd354STrond Myklebust 16829903cd1cSChuck Lever /** 16839903cd1cSChuck Lever * xprt_destroy - destroy an RPC transport, killing off all requests. 1684a8de240aSTrond Myklebust * @xprt: transport to destroy 16859903cd1cSChuck Lever * 16861da177e4SLinus Torvalds */ 1687a8de240aSTrond Myklebust static void xprt_destroy(struct rpc_xprt *xprt) 16881da177e4SLinus Torvalds { 16891da177e4SLinus Torvalds dprintk("RPC: destroying transport %p\n", xprt); 169079234c3dSTrond Myklebust 1691528fd354STrond Myklebust /* 1692528fd354STrond Myklebust * Exclude transport connect/disconnect handlers and autoclose 1693528fd354STrond Myklebust */ 169479234c3dSTrond Myklebust wait_on_bit_lock(&xprt->state, XPRT_LOCKED, TASK_UNINTERRUPTIBLE); 169579234c3dSTrond Myklebust 16960065db32STrond Myklebust del_timer_sync(&xprt->timer); 1697c8541ecdSChuck Lever 1698c8541ecdSChuck Lever /* 1699528fd354STrond Myklebust * Destroy sockets etc from the system workqueue so they can 1700528fd354STrond Myklebust * safely flush receive work running on rpciod. 1701c8541ecdSChuck Lever */ 1702528fd354STrond Myklebust INIT_WORK(&xprt->task_cleanup, xprt_destroy_cb); 1703528fd354STrond Myklebust schedule_work(&xprt->task_cleanup); 17046b6ca86bSTrond Myklebust } 17051da177e4SLinus Torvalds 170630c5116bSTrond Myklebust static void xprt_destroy_kref(struct kref *kref) 170730c5116bSTrond Myklebust { 170830c5116bSTrond Myklebust xprt_destroy(container_of(kref, struct rpc_xprt, kref)); 170930c5116bSTrond Myklebust } 171030c5116bSTrond Myklebust 171130c5116bSTrond Myklebust /** 171230c5116bSTrond Myklebust * xprt_get - return a reference to an RPC transport. 171330c5116bSTrond Myklebust * @xprt: pointer to the transport 171430c5116bSTrond Myklebust * 171530c5116bSTrond Myklebust */ 171630c5116bSTrond Myklebust struct rpc_xprt *xprt_get(struct rpc_xprt *xprt) 171730c5116bSTrond Myklebust { 171830c5116bSTrond Myklebust if (xprt != NULL && kref_get_unless_zero(&xprt->kref)) 171930c5116bSTrond Myklebust return xprt; 172030c5116bSTrond Myklebust return NULL; 172130c5116bSTrond Myklebust } 172230c5116bSTrond Myklebust EXPORT_SYMBOL_GPL(xprt_get); 172330c5116bSTrond Myklebust 17246b6ca86bSTrond Myklebust /** 17256b6ca86bSTrond Myklebust * xprt_put - release a reference to an RPC transport. 17266b6ca86bSTrond Myklebust * @xprt: pointer to the transport 17276b6ca86bSTrond Myklebust * 17286b6ca86bSTrond Myklebust */ 17296b6ca86bSTrond Myklebust void xprt_put(struct rpc_xprt *xprt) 17306b6ca86bSTrond Myklebust { 173130c5116bSTrond Myklebust if (xprt != NULL) 173230c5116bSTrond Myklebust kref_put(&xprt->kref, xprt_destroy_kref); 17336b6ca86bSTrond Myklebust } 17345d252f90SChuck Lever EXPORT_SYMBOL_GPL(xprt_put); 1735