xref: /openbmc/linux/net/sunrpc/svc_xprt.c (revision e75bafbf)
11d8206b9STom Tucker /*
21d8206b9STom Tucker  * linux/net/sunrpc/svc_xprt.c
31d8206b9STom Tucker  *
41d8206b9STom Tucker  * Author: Tom Tucker <tom@opengridcomputing.com>
51d8206b9STom Tucker  */
61d8206b9STom Tucker 
71d8206b9STom Tucker #include <linux/sched.h>
81d8206b9STom Tucker #include <linux/errno.h>
91d8206b9STom Tucker #include <linux/freezer.h>
107086721fSJeff Layton #include <linux/kthread.h>
115a0e3ad6STejun Heo #include <linux/slab.h>
121d8206b9STom Tucker #include <net/sock.h>
131d8206b9STom Tucker #include <linux/sunrpc/stats.h>
141d8206b9STom Tucker #include <linux/sunrpc/svc_xprt.h>
15dcf1a357SH Hartley Sweeten #include <linux/sunrpc/svcsock.h>
1699de8ea9SJ. Bruce Fields #include <linux/sunrpc/xprt.h>
173a9a231dSPaul Gortmaker #include <linux/module.h>
181d8206b9STom Tucker 
191d8206b9STom Tucker #define RPCDBG_FACILITY	RPCDBG_SVCXPRT
201d8206b9STom Tucker 
210f0257eaSTom Tucker static struct svc_deferred_req *svc_deferred_dequeue(struct svc_xprt *xprt);
220f0257eaSTom Tucker static int svc_deferred_recv(struct svc_rqst *rqstp);
230f0257eaSTom Tucker static struct cache_deferred_req *svc_defer(struct cache_req *req);
240f0257eaSTom Tucker static void svc_age_temp_xprts(unsigned long closure);
257710ec36SJ. Bruce Fields static void svc_delete_xprt(struct svc_xprt *xprt);
260f0257eaSTom Tucker 
270f0257eaSTom Tucker /* apparently the "standard" is that clients close
280f0257eaSTom Tucker  * idle connections after 5 minutes, servers after
290f0257eaSTom Tucker  * 6 minutes
300f0257eaSTom Tucker  *   http://www.connectathon.org/talks96/nfstcp.pdf
310f0257eaSTom Tucker  */
320f0257eaSTom Tucker static int svc_conn_age_period = 6*60;
330f0257eaSTom Tucker 
341d8206b9STom Tucker /* List of registered transport classes */
351d8206b9STom Tucker static DEFINE_SPINLOCK(svc_xprt_class_lock);
361d8206b9STom Tucker static LIST_HEAD(svc_xprt_class_list);
371d8206b9STom Tucker 
380f0257eaSTom Tucker /* SMP locking strategy:
390f0257eaSTom Tucker  *
400f0257eaSTom Tucker  *	svc_pool->sp_lock protects most of the fields of that pool.
410f0257eaSTom Tucker  *	svc_serv->sv_lock protects sv_tempsocks, sv_permsocks, sv_tmpcnt.
420f0257eaSTom Tucker  *	when both need to be taken (rare), svc_serv->sv_lock is first.
430f0257eaSTom Tucker  *	BKL protects svc_serv->sv_nrthread.
440f0257eaSTom Tucker  *	svc_sock->sk_lock protects the svc_sock->sk_deferred list
450f0257eaSTom Tucker  *             and the ->sk_info_authunix cache.
460f0257eaSTom Tucker  *
470f0257eaSTom Tucker  *	The XPT_BUSY bit in xprt->xpt_flags prevents a transport being
480f0257eaSTom Tucker  *	enqueued multiply. During normal transport processing this bit
490f0257eaSTom Tucker  *	is set by svc_xprt_enqueue and cleared by svc_xprt_received.
500f0257eaSTom Tucker  *	Providers should not manipulate this bit directly.
510f0257eaSTom Tucker  *
520f0257eaSTom Tucker  *	Some flags can be set to certain values at any time
530f0257eaSTom Tucker  *	providing that certain rules are followed:
540f0257eaSTom Tucker  *
550f0257eaSTom Tucker  *	XPT_CONN, XPT_DATA:
560f0257eaSTom Tucker  *		- Can be set or cleared at any time.
570f0257eaSTom Tucker  *		- After a set, svc_xprt_enqueue must be called to enqueue
580f0257eaSTom Tucker  *		  the transport for processing.
590f0257eaSTom Tucker  *		- After a clear, the transport must be read/accepted.
600f0257eaSTom Tucker  *		  If this succeeds, it must be set again.
610f0257eaSTom Tucker  *	XPT_CLOSE:
620f0257eaSTom Tucker  *		- Can set at any time. It is never cleared.
630f0257eaSTom Tucker  *      XPT_DEAD:
640f0257eaSTom Tucker  *		- Can only be set while XPT_BUSY is held which ensures
650f0257eaSTom Tucker  *		  that no other thread will be using the transport or will
660f0257eaSTom Tucker  *		  try to set XPT_DEAD.
670f0257eaSTom Tucker  */
680f0257eaSTom Tucker 
691d8206b9STom Tucker int svc_reg_xprt_class(struct svc_xprt_class *xcl)
701d8206b9STom Tucker {
711d8206b9STom Tucker 	struct svc_xprt_class *cl;
721d8206b9STom Tucker 	int res = -EEXIST;
731d8206b9STom Tucker 
741d8206b9STom Tucker 	dprintk("svc: Adding svc transport class '%s'\n", xcl->xcl_name);
751d8206b9STom Tucker 
761d8206b9STom Tucker 	INIT_LIST_HEAD(&xcl->xcl_list);
771d8206b9STom Tucker 	spin_lock(&svc_xprt_class_lock);
781d8206b9STom Tucker 	/* Make sure there isn't already a class with the same name */
791d8206b9STom Tucker 	list_for_each_entry(cl, &svc_xprt_class_list, xcl_list) {
801d8206b9STom Tucker 		if (strcmp(xcl->xcl_name, cl->xcl_name) == 0)
811d8206b9STom Tucker 			goto out;
821d8206b9STom Tucker 	}
831d8206b9STom Tucker 	list_add_tail(&xcl->xcl_list, &svc_xprt_class_list);
841d8206b9STom Tucker 	res = 0;
851d8206b9STom Tucker out:
861d8206b9STom Tucker 	spin_unlock(&svc_xprt_class_lock);
871d8206b9STom Tucker 	return res;
881d8206b9STom Tucker }
891d8206b9STom Tucker EXPORT_SYMBOL_GPL(svc_reg_xprt_class);
901d8206b9STom Tucker 
911d8206b9STom Tucker void svc_unreg_xprt_class(struct svc_xprt_class *xcl)
921d8206b9STom Tucker {
931d8206b9STom Tucker 	dprintk("svc: Removing svc transport class '%s'\n", xcl->xcl_name);
941d8206b9STom Tucker 	spin_lock(&svc_xprt_class_lock);
951d8206b9STom Tucker 	list_del_init(&xcl->xcl_list);
961d8206b9STom Tucker 	spin_unlock(&svc_xprt_class_lock);
971d8206b9STom Tucker }
981d8206b9STom Tucker EXPORT_SYMBOL_GPL(svc_unreg_xprt_class);
991d8206b9STom Tucker 
100dc9a16e4STom Tucker /*
101dc9a16e4STom Tucker  * Format the transport list for printing
102dc9a16e4STom Tucker  */
103dc9a16e4STom Tucker int svc_print_xprts(char *buf, int maxlen)
104dc9a16e4STom Tucker {
1058f3a6de3SPavel Emelyanov 	struct svc_xprt_class *xcl;
106dc9a16e4STom Tucker 	char tmpstr[80];
107dc9a16e4STom Tucker 	int len = 0;
108dc9a16e4STom Tucker 	buf[0] = '\0';
109dc9a16e4STom Tucker 
110dc9a16e4STom Tucker 	spin_lock(&svc_xprt_class_lock);
1118f3a6de3SPavel Emelyanov 	list_for_each_entry(xcl, &svc_xprt_class_list, xcl_list) {
112dc9a16e4STom Tucker 		int slen;
113dc9a16e4STom Tucker 
114dc9a16e4STom Tucker 		sprintf(tmpstr, "%s %d\n", xcl->xcl_name, xcl->xcl_max_payload);
115dc9a16e4STom Tucker 		slen = strlen(tmpstr);
116dc9a16e4STom Tucker 		if (len + slen > maxlen)
117dc9a16e4STom Tucker 			break;
118dc9a16e4STom Tucker 		len += slen;
119dc9a16e4STom Tucker 		strcat(buf, tmpstr);
120dc9a16e4STom Tucker 	}
121dc9a16e4STom Tucker 	spin_unlock(&svc_xprt_class_lock);
122dc9a16e4STom Tucker 
123dc9a16e4STom Tucker 	return len;
124dc9a16e4STom Tucker }
125dc9a16e4STom Tucker 
126e1b3157fSTom Tucker static void svc_xprt_free(struct kref *kref)
127e1b3157fSTom Tucker {
128e1b3157fSTom Tucker 	struct svc_xprt *xprt =
129e1b3157fSTom Tucker 		container_of(kref, struct svc_xprt, xpt_ref);
130e1b3157fSTom Tucker 	struct module *owner = xprt->xpt_class->xcl_owner;
131e3bfca01SPavel Emelyanov 	if (test_bit(XPT_CACHE_AUTH, &xprt->xpt_flags))
132e3bfca01SPavel Emelyanov 		svcauth_unix_info_release(xprt);
1334fb8518bSPavel Emelyanov 	put_net(xprt->xpt_net);
13499de8ea9SJ. Bruce Fields 	/* See comment on corresponding get in xs_setup_bc_tcp(): */
13599de8ea9SJ. Bruce Fields 	if (xprt->xpt_bc_xprt)
13699de8ea9SJ. Bruce Fields 		xprt_put(xprt->xpt_bc_xprt);
137e1b3157fSTom Tucker 	xprt->xpt_ops->xpo_free(xprt);
138e1b3157fSTom Tucker 	module_put(owner);
139e1b3157fSTom Tucker }
140e1b3157fSTom Tucker 
141e1b3157fSTom Tucker void svc_xprt_put(struct svc_xprt *xprt)
142e1b3157fSTom Tucker {
143e1b3157fSTom Tucker 	kref_put(&xprt->xpt_ref, svc_xprt_free);
144e1b3157fSTom Tucker }
145e1b3157fSTom Tucker EXPORT_SYMBOL_GPL(svc_xprt_put);
146e1b3157fSTom Tucker 
1471d8206b9STom Tucker /*
1481d8206b9STom Tucker  * Called by transport drivers to initialize the transport independent
1491d8206b9STom Tucker  * portion of the transport instance.
1501d8206b9STom Tucker  */
151bd4620ddSStanislav Kinsbursky void svc_xprt_init(struct net *net, struct svc_xprt_class *xcl,
152bd4620ddSStanislav Kinsbursky 		   struct svc_xprt *xprt, struct svc_serv *serv)
1531d8206b9STom Tucker {
1541d8206b9STom Tucker 	memset(xprt, 0, sizeof(*xprt));
1551d8206b9STom Tucker 	xprt->xpt_class = xcl;
1561d8206b9STom Tucker 	xprt->xpt_ops = xcl->xcl_ops;
157e1b3157fSTom Tucker 	kref_init(&xprt->xpt_ref);
158bb5cf160STom Tucker 	xprt->xpt_server = serv;
1597a182083STom Tucker 	INIT_LIST_HEAD(&xprt->xpt_list);
1607a182083STom Tucker 	INIT_LIST_HEAD(&xprt->xpt_ready);
1618c7b0172STom Tucker 	INIT_LIST_HEAD(&xprt->xpt_deferred);
162edc7a894SJ. Bruce Fields 	INIT_LIST_HEAD(&xprt->xpt_users);
163a50fea26STom Tucker 	mutex_init(&xprt->xpt_mutex);
164def13d74STom Tucker 	spin_lock_init(&xprt->xpt_lock);
1654e5caaa5STom Tucker 	set_bit(XPT_BUSY, &xprt->xpt_flags);
1664cfc7e60SRahul Iyer 	rpc_init_wait_queue(&xprt->xpt_bc_pending, "xpt_bc_pending");
167bd4620ddSStanislav Kinsbursky 	xprt->xpt_net = get_net(net);
1681d8206b9STom Tucker }
1691d8206b9STom Tucker EXPORT_SYMBOL_GPL(svc_xprt_init);
170b700cbb1STom Tucker 
1715dd248f6SChuck Lever static struct svc_xprt *__svc_xpo_create(struct svc_xprt_class *xcl,
1725dd248f6SChuck Lever 					 struct svc_serv *serv,
17362832c03SPavel Emelyanov 					 struct net *net,
1749652ada3SChuck Lever 					 const int family,
1759652ada3SChuck Lever 					 const unsigned short port,
1769652ada3SChuck Lever 					 int flags)
177b700cbb1STom Tucker {
178b700cbb1STom Tucker 	struct sockaddr_in sin = {
179b700cbb1STom Tucker 		.sin_family		= AF_INET,
180e6f1cebfSAl Viro 		.sin_addr.s_addr	= htonl(INADDR_ANY),
181b700cbb1STom Tucker 		.sin_port		= htons(port),
182b700cbb1STom Tucker 	};
183dfd56b8bSEric Dumazet #if IS_ENABLED(CONFIG_IPV6)
1845dd248f6SChuck Lever 	struct sockaddr_in6 sin6 = {
1855dd248f6SChuck Lever 		.sin6_family		= AF_INET6,
1865dd248f6SChuck Lever 		.sin6_addr		= IN6ADDR_ANY_INIT,
1875dd248f6SChuck Lever 		.sin6_port		= htons(port),
1885dd248f6SChuck Lever 	};
189dfd56b8bSEric Dumazet #endif
1905dd248f6SChuck Lever 	struct sockaddr *sap;
1915dd248f6SChuck Lever 	size_t len;
1925dd248f6SChuck Lever 
1939652ada3SChuck Lever 	switch (family) {
1949652ada3SChuck Lever 	case PF_INET:
1955dd248f6SChuck Lever 		sap = (struct sockaddr *)&sin;
1965dd248f6SChuck Lever 		len = sizeof(sin);
1975dd248f6SChuck Lever 		break;
198dfd56b8bSEric Dumazet #if IS_ENABLED(CONFIG_IPV6)
1999652ada3SChuck Lever 	case PF_INET6:
2005dd248f6SChuck Lever 		sap = (struct sockaddr *)&sin6;
2015dd248f6SChuck Lever 		len = sizeof(sin6);
2025dd248f6SChuck Lever 		break;
203dfd56b8bSEric Dumazet #endif
2045dd248f6SChuck Lever 	default:
2055dd248f6SChuck Lever 		return ERR_PTR(-EAFNOSUPPORT);
2065dd248f6SChuck Lever 	}
2075dd248f6SChuck Lever 
20862832c03SPavel Emelyanov 	return xcl->xcl_ops->xpo_create(serv, net, sap, len, flags);
2095dd248f6SChuck Lever }
2105dd248f6SChuck Lever 
2116741019cSJ. Bruce Fields /*
2126741019cSJ. Bruce Fields  * svc_xprt_received conditionally queues the transport for processing
2136741019cSJ. Bruce Fields  * by another thread. The caller must hold the XPT_BUSY bit and must
2146741019cSJ. Bruce Fields  * not thereafter touch transport data.
2156741019cSJ. Bruce Fields  *
2166741019cSJ. Bruce Fields  * Note: XPT_DATA only gets cleared when a read-attempt finds no (or
2176741019cSJ. Bruce Fields  * insufficient) data.
2186741019cSJ. Bruce Fields  */
2196741019cSJ. Bruce Fields static void svc_xprt_received(struct svc_xprt *xprt)
2206741019cSJ. Bruce Fields {
221ff1fdb9bSWeston Andros Adamson 	WARN_ON_ONCE(!test_bit(XPT_BUSY, &xprt->xpt_flags));
222ff1fdb9bSWeston Andros Adamson 	if (!test_bit(XPT_BUSY, &xprt->xpt_flags))
223ff1fdb9bSWeston Andros Adamson 		return;
2246741019cSJ. Bruce Fields 	/* As soon as we clear busy, the xprt could be closed and
2256741019cSJ. Bruce Fields 	 * 'put', so we need a reference to call svc_xprt_enqueue with:
2266741019cSJ. Bruce Fields 	 */
2276741019cSJ. Bruce Fields 	svc_xprt_get(xprt);
2286741019cSJ. Bruce Fields 	clear_bit(XPT_BUSY, &xprt->xpt_flags);
2296741019cSJ. Bruce Fields 	svc_xprt_enqueue(xprt);
2306741019cSJ. Bruce Fields 	svc_xprt_put(xprt);
2316741019cSJ. Bruce Fields }
2326741019cSJ. Bruce Fields 
23339b55301SJ. Bruce Fields void svc_add_new_perm_xprt(struct svc_serv *serv, struct svc_xprt *new)
23439b55301SJ. Bruce Fields {
23539b55301SJ. Bruce Fields 	clear_bit(XPT_TEMP, &new->xpt_flags);
23639b55301SJ. Bruce Fields 	spin_lock_bh(&serv->sv_lock);
23739b55301SJ. Bruce Fields 	list_add(&new->xpt_list, &serv->sv_permsocks);
23839b55301SJ. Bruce Fields 	spin_unlock_bh(&serv->sv_lock);
23939b55301SJ. Bruce Fields 	svc_xprt_received(new);
24039b55301SJ. Bruce Fields }
24139b55301SJ. Bruce Fields 
2429652ada3SChuck Lever int svc_create_xprt(struct svc_serv *serv, const char *xprt_name,
243fc5d00b0SPavel Emelyanov 		    struct net *net, const int family,
244fc5d00b0SPavel Emelyanov 		    const unsigned short port, int flags)
2455dd248f6SChuck Lever {
2465dd248f6SChuck Lever 	struct svc_xprt_class *xcl;
2475dd248f6SChuck Lever 
248b700cbb1STom Tucker 	dprintk("svc: creating transport %s[%d]\n", xprt_name, port);
249b700cbb1STom Tucker 	spin_lock(&svc_xprt_class_lock);
250b700cbb1STom Tucker 	list_for_each_entry(xcl, &svc_xprt_class_list, xcl_list) {
251b700cbb1STom Tucker 		struct svc_xprt *newxprt;
252ed2849d3SNeilBrown 		unsigned short newport;
2534e5caaa5STom Tucker 
2544e5caaa5STom Tucker 		if (strcmp(xprt_name, xcl->xcl_name))
2554e5caaa5STom Tucker 			continue;
2564e5caaa5STom Tucker 
2574e5caaa5STom Tucker 		if (!try_module_get(xcl->xcl_owner))
2584e5caaa5STom Tucker 			goto err;
2594e5caaa5STom Tucker 
2604e5caaa5STom Tucker 		spin_unlock(&svc_xprt_class_lock);
26162832c03SPavel Emelyanov 		newxprt = __svc_xpo_create(xcl, serv, net, family, port, flags);
262b700cbb1STom Tucker 		if (IS_ERR(newxprt)) {
263b700cbb1STom Tucker 			module_put(xcl->xcl_owner);
2644e5caaa5STom Tucker 			return PTR_ERR(newxprt);
265b700cbb1STom Tucker 		}
26639b55301SJ. Bruce Fields 		svc_add_new_perm_xprt(serv, newxprt);
267ed2849d3SNeilBrown 		newport = svc_xprt_local_port(newxprt);
268ed2849d3SNeilBrown 		return newport;
269b700cbb1STom Tucker 	}
2704e5caaa5STom Tucker  err:
271b700cbb1STom Tucker 	spin_unlock(&svc_xprt_class_lock);
272b700cbb1STom Tucker 	dprintk("svc: transport %s not found\n", xprt_name);
27368717908SChuck Lever 
27468717908SChuck Lever 	/* This errno is exposed to user space.  Provide a reasonable
27568717908SChuck Lever 	 * perror msg for a bad transport. */
27668717908SChuck Lever 	return -EPROTONOSUPPORT;
277b700cbb1STom Tucker }
278b700cbb1STom Tucker EXPORT_SYMBOL_GPL(svc_create_xprt);
2799dbc240fSTom Tucker 
2809dbc240fSTom Tucker /*
2819dbc240fSTom Tucker  * Copy the local and remote xprt addresses to the rqstp structure
2829dbc240fSTom Tucker  */
2839dbc240fSTom Tucker void svc_xprt_copy_addrs(struct svc_rqst *rqstp, struct svc_xprt *xprt)
2849dbc240fSTom Tucker {
2859dbc240fSTom Tucker 	memcpy(&rqstp->rq_addr, &xprt->xpt_remote, xprt->xpt_remotelen);
2869dbc240fSTom Tucker 	rqstp->rq_addrlen = xprt->xpt_remotelen;
2879dbc240fSTom Tucker 
2889dbc240fSTom Tucker 	/*
2899dbc240fSTom Tucker 	 * Destination address in request is needed for binding the
2909dbc240fSTom Tucker 	 * source address in RPC replies/callbacks later.
2919dbc240fSTom Tucker 	 */
292849a1cf1SMi Jinlong 	memcpy(&rqstp->rq_daddr, &xprt->xpt_local, xprt->xpt_locallen);
293849a1cf1SMi Jinlong 	rqstp->rq_daddrlen = xprt->xpt_locallen;
2949dbc240fSTom Tucker }
2959dbc240fSTom Tucker EXPORT_SYMBOL_GPL(svc_xprt_copy_addrs);
2969dbc240fSTom Tucker 
2970f0257eaSTom Tucker /**
2980f0257eaSTom Tucker  * svc_print_addr - Format rq_addr field for printing
2990f0257eaSTom Tucker  * @rqstp: svc_rqst struct containing address to print
3000f0257eaSTom Tucker  * @buf: target buffer for formatted address
3010f0257eaSTom Tucker  * @len: length of target buffer
3020f0257eaSTom Tucker  *
3030f0257eaSTom Tucker  */
3040f0257eaSTom Tucker char *svc_print_addr(struct svc_rqst *rqstp, char *buf, size_t len)
3050f0257eaSTom Tucker {
3060f0257eaSTom Tucker 	return __svc_print_addr(svc_addr(rqstp), buf, len);
3070f0257eaSTom Tucker }
3080f0257eaSTom Tucker EXPORT_SYMBOL_GPL(svc_print_addr);
3090f0257eaSTom Tucker 
3100f0257eaSTom Tucker /*
3110f0257eaSTom Tucker  * Queue up an idle server thread.  Must have pool->sp_lock held.
3120f0257eaSTom Tucker  * Note: this is really a stack rather than a queue, so that we only
3130f0257eaSTom Tucker  * use as many different threads as we need, and the rest don't pollute
3140f0257eaSTom Tucker  * the cache.
3150f0257eaSTom Tucker  */
3160f0257eaSTom Tucker static void svc_thread_enqueue(struct svc_pool *pool, struct svc_rqst *rqstp)
3170f0257eaSTom Tucker {
3180f0257eaSTom Tucker 	list_add(&rqstp->rq_list, &pool->sp_threads);
3190f0257eaSTom Tucker }
3200f0257eaSTom Tucker 
3210f0257eaSTom Tucker /*
3220f0257eaSTom Tucker  * Dequeue an nfsd thread.  Must have pool->sp_lock held.
3230f0257eaSTom Tucker  */
3240f0257eaSTom Tucker static void svc_thread_dequeue(struct svc_pool *pool, struct svc_rqst *rqstp)
3250f0257eaSTom Tucker {
3260f0257eaSTom Tucker 	list_del(&rqstp->rq_list);
3270f0257eaSTom Tucker }
3280f0257eaSTom Tucker 
3299c335c0bSJ. Bruce Fields static bool svc_xprt_has_something_to_do(struct svc_xprt *xprt)
3309c335c0bSJ. Bruce Fields {
3319c335c0bSJ. Bruce Fields 	if (xprt->xpt_flags & ((1<<XPT_CONN)|(1<<XPT_CLOSE)))
3329c335c0bSJ. Bruce Fields 		return true;
3339c335c0bSJ. Bruce Fields 	if (xprt->xpt_flags & ((1<<XPT_DATA)|(1<<XPT_DEFERRED)))
3349c335c0bSJ. Bruce Fields 		return xprt->xpt_ops->xpo_has_wspace(xprt);
3359c335c0bSJ. Bruce Fields 	return false;
3369c335c0bSJ. Bruce Fields }
3379c335c0bSJ. Bruce Fields 
3380f0257eaSTom Tucker /*
3390f0257eaSTom Tucker  * Queue up a transport with data pending. If there are idle nfsd
3400f0257eaSTom Tucker  * processes, wake 'em up.
3410f0257eaSTom Tucker  *
3420f0257eaSTom Tucker  */
3430f0257eaSTom Tucker void svc_xprt_enqueue(struct svc_xprt *xprt)
3440f0257eaSTom Tucker {
3450f0257eaSTom Tucker 	struct svc_pool *pool;
3460f0257eaSTom Tucker 	struct svc_rqst	*rqstp;
3470f0257eaSTom Tucker 	int cpu;
3480f0257eaSTom Tucker 
3499c335c0bSJ. Bruce Fields 	if (!svc_xprt_has_something_to_do(xprt))
3500f0257eaSTom Tucker 		return;
3510f0257eaSTom Tucker 
3520f0257eaSTom Tucker 	cpu = get_cpu();
3530f0257eaSTom Tucker 	pool = svc_pool_for_cpu(xprt->xpt_server, cpu);
3540f0257eaSTom Tucker 	put_cpu();
3550f0257eaSTom Tucker 
3560f0257eaSTom Tucker 	spin_lock_bh(&pool->sp_lock);
3570f0257eaSTom Tucker 
35878c210efSJ. Bruce Fields 	if (!list_empty(&pool->sp_threads) &&
35978c210efSJ. Bruce Fields 	    !list_empty(&pool->sp_sockets))
36078c210efSJ. Bruce Fields 		printk(KERN_ERR
36178c210efSJ. Bruce Fields 		       "svc_xprt_enqueue: "
36278c210efSJ. Bruce Fields 		       "threads and transports both waiting??\n");
36378c210efSJ. Bruce Fields 
36403cf6c9fSGreg Banks 	pool->sp_stats.packets++;
36503cf6c9fSGreg Banks 
3660f0257eaSTom Tucker 	/* Mark transport as busy. It will remain in this state until
3670f0257eaSTom Tucker 	 * the provider calls svc_xprt_received. We update XPT_BUSY
3680f0257eaSTom Tucker 	 * atomically because it also guards against trying to enqueue
3690f0257eaSTom Tucker 	 * the transport twice.
3700f0257eaSTom Tucker 	 */
3710f0257eaSTom Tucker 	if (test_and_set_bit(XPT_BUSY, &xprt->xpt_flags)) {
3720f0257eaSTom Tucker 		/* Don't enqueue transport while already enqueued */
3730f0257eaSTom Tucker 		dprintk("svc: transport %p busy, not enqueued\n", xprt);
3740f0257eaSTom Tucker 		goto out_unlock;
3750f0257eaSTom Tucker 	}
3760f0257eaSTom Tucker 
37778c210efSJ. Bruce Fields 	if (!list_empty(&pool->sp_threads)) {
3780f0257eaSTom Tucker 		rqstp = list_entry(pool->sp_threads.next,
3790f0257eaSTom Tucker 				   struct svc_rqst,
3800f0257eaSTom Tucker 				   rq_list);
3810f0257eaSTom Tucker 		dprintk("svc: transport %p served by daemon %p\n",
3820f0257eaSTom Tucker 			xprt, rqstp);
3830f0257eaSTom Tucker 		svc_thread_dequeue(pool, rqstp);
3840f0257eaSTom Tucker 		if (rqstp->rq_xprt)
3850f0257eaSTom Tucker 			printk(KERN_ERR
3860f0257eaSTom Tucker 				"svc_xprt_enqueue: server %p, rq_xprt=%p!\n",
3870f0257eaSTom Tucker 				rqstp, rqstp->rq_xprt);
3880f0257eaSTom Tucker 		rqstp->rq_xprt = xprt;
3890f0257eaSTom Tucker 		svc_xprt_get(xprt);
39003cf6c9fSGreg Banks 		pool->sp_stats.threads_woken++;
3910f0257eaSTom Tucker 		wake_up(&rqstp->rq_wait);
3920f0257eaSTom Tucker 	} else {
3930f0257eaSTom Tucker 		dprintk("svc: transport %p put into queue\n", xprt);
3940f0257eaSTom Tucker 		list_add_tail(&xprt->xpt_ready, &pool->sp_sockets);
39503cf6c9fSGreg Banks 		pool->sp_stats.sockets_queued++;
3960f0257eaSTom Tucker 	}
3970f0257eaSTom Tucker 
3980f0257eaSTom Tucker out_unlock:
3990f0257eaSTom Tucker 	spin_unlock_bh(&pool->sp_lock);
4000f0257eaSTom Tucker }
4010f0257eaSTom Tucker EXPORT_SYMBOL_GPL(svc_xprt_enqueue);
4020f0257eaSTom Tucker 
4030f0257eaSTom Tucker /*
4040f0257eaSTom Tucker  * Dequeue the first transport.  Must be called with the pool->sp_lock held.
4050f0257eaSTom Tucker  */
4060f0257eaSTom Tucker static struct svc_xprt *svc_xprt_dequeue(struct svc_pool *pool)
4070f0257eaSTom Tucker {
4080f0257eaSTom Tucker 	struct svc_xprt	*xprt;
4090f0257eaSTom Tucker 
4100f0257eaSTom Tucker 	if (list_empty(&pool->sp_sockets))
4110f0257eaSTom Tucker 		return NULL;
4120f0257eaSTom Tucker 
4130f0257eaSTom Tucker 	xprt = list_entry(pool->sp_sockets.next,
4140f0257eaSTom Tucker 			  struct svc_xprt, xpt_ready);
4150f0257eaSTom Tucker 	list_del_init(&xprt->xpt_ready);
4160f0257eaSTom Tucker 
4170f0257eaSTom Tucker 	dprintk("svc: transport %p dequeued, inuse=%d\n",
4180f0257eaSTom Tucker 		xprt, atomic_read(&xprt->xpt_ref.refcount));
4190f0257eaSTom Tucker 
4200f0257eaSTom Tucker 	return xprt;
4210f0257eaSTom Tucker }
4220f0257eaSTom Tucker 
4230f0257eaSTom Tucker /**
4240f0257eaSTom Tucker  * svc_reserve - change the space reserved for the reply to a request.
4250f0257eaSTom Tucker  * @rqstp:  The request in question
4260f0257eaSTom Tucker  * @space: new max space to reserve
4270f0257eaSTom Tucker  *
4280f0257eaSTom Tucker  * Each request reserves some space on the output queue of the transport
4290f0257eaSTom Tucker  * to make sure the reply fits.  This function reduces that reserved
4300f0257eaSTom Tucker  * space to be the amount of space used already, plus @space.
4310f0257eaSTom Tucker  *
4320f0257eaSTom Tucker  */
4330f0257eaSTom Tucker void svc_reserve(struct svc_rqst *rqstp, int space)
4340f0257eaSTom Tucker {
4350f0257eaSTom Tucker 	space += rqstp->rq_res.head[0].iov_len;
4360f0257eaSTom Tucker 
4370f0257eaSTom Tucker 	if (space < rqstp->rq_reserved) {
4380f0257eaSTom Tucker 		struct svc_xprt *xprt = rqstp->rq_xprt;
4390f0257eaSTom Tucker 		atomic_sub((rqstp->rq_reserved - space), &xprt->xpt_reserved);
4400f0257eaSTom Tucker 		rqstp->rq_reserved = space;
4410f0257eaSTom Tucker 
4420f0257eaSTom Tucker 		svc_xprt_enqueue(xprt);
4430f0257eaSTom Tucker 	}
4440f0257eaSTom Tucker }
44524c3767eSTrond Myklebust EXPORT_SYMBOL_GPL(svc_reserve);
4460f0257eaSTom Tucker 
4470f0257eaSTom Tucker static void svc_xprt_release(struct svc_rqst *rqstp)
4480f0257eaSTom Tucker {
4490f0257eaSTom Tucker 	struct svc_xprt	*xprt = rqstp->rq_xprt;
4500f0257eaSTom Tucker 
4510f0257eaSTom Tucker 	rqstp->rq_xprt->xpt_ops->xpo_release_rqst(rqstp);
4520f0257eaSTom Tucker 
4532779e3aeSTom Tucker 	kfree(rqstp->rq_deferred);
4542779e3aeSTom Tucker 	rqstp->rq_deferred = NULL;
4552779e3aeSTom Tucker 
4560f0257eaSTom Tucker 	svc_free_res_pages(rqstp);
4570f0257eaSTom Tucker 	rqstp->rq_res.page_len = 0;
4580f0257eaSTom Tucker 	rqstp->rq_res.page_base = 0;
4590f0257eaSTom Tucker 
4600f0257eaSTom Tucker 	/* Reset response buffer and release
4610f0257eaSTom Tucker 	 * the reservation.
4620f0257eaSTom Tucker 	 * But first, check that enough space was reserved
4630f0257eaSTom Tucker 	 * for the reply, otherwise we have a bug!
4640f0257eaSTom Tucker 	 */
4650f0257eaSTom Tucker 	if ((rqstp->rq_res.len) >  rqstp->rq_reserved)
4660f0257eaSTom Tucker 		printk(KERN_ERR "RPC request reserved %d but used %d\n",
4670f0257eaSTom Tucker 		       rqstp->rq_reserved,
4680f0257eaSTom Tucker 		       rqstp->rq_res.len);
4690f0257eaSTom Tucker 
4700f0257eaSTom Tucker 	rqstp->rq_res.head[0].iov_len = 0;
4710f0257eaSTom Tucker 	svc_reserve(rqstp, 0);
4720f0257eaSTom Tucker 	rqstp->rq_xprt = NULL;
4730f0257eaSTom Tucker 
4740f0257eaSTom Tucker 	svc_xprt_put(xprt);
4750f0257eaSTom Tucker }
4760f0257eaSTom Tucker 
4770f0257eaSTom Tucker /*
4780f0257eaSTom Tucker  * External function to wake up a server waiting for data
4790f0257eaSTom Tucker  * This really only makes sense for services like lockd
4800f0257eaSTom Tucker  * which have exactly one thread anyway.
4810f0257eaSTom Tucker  */
4820f0257eaSTom Tucker void svc_wake_up(struct svc_serv *serv)
4830f0257eaSTom Tucker {
4840f0257eaSTom Tucker 	struct svc_rqst	*rqstp;
4850f0257eaSTom Tucker 	unsigned int i;
4860f0257eaSTom Tucker 	struct svc_pool *pool;
4870f0257eaSTom Tucker 
4880f0257eaSTom Tucker 	for (i = 0; i < serv->sv_nrpools; i++) {
4890f0257eaSTom Tucker 		pool = &serv->sv_pools[i];
4900f0257eaSTom Tucker 
4910f0257eaSTom Tucker 		spin_lock_bh(&pool->sp_lock);
4920f0257eaSTom Tucker 		if (!list_empty(&pool->sp_threads)) {
4930f0257eaSTom Tucker 			rqstp = list_entry(pool->sp_threads.next,
4940f0257eaSTom Tucker 					   struct svc_rqst,
4950f0257eaSTom Tucker 					   rq_list);
4960f0257eaSTom Tucker 			dprintk("svc: daemon %p woken up.\n", rqstp);
4970f0257eaSTom Tucker 			/*
4980f0257eaSTom Tucker 			svc_thread_dequeue(pool, rqstp);
4990f0257eaSTom Tucker 			rqstp->rq_xprt = NULL;
5000f0257eaSTom Tucker 			 */
5010f0257eaSTom Tucker 			wake_up(&rqstp->rq_wait);
50235525b79SAndriy Skulysh 		} else
50335525b79SAndriy Skulysh 			pool->sp_task_pending = 1;
5040f0257eaSTom Tucker 		spin_unlock_bh(&pool->sp_lock);
5050f0257eaSTom Tucker 	}
5060f0257eaSTom Tucker }
50724c3767eSTrond Myklebust EXPORT_SYMBOL_GPL(svc_wake_up);
5080f0257eaSTom Tucker 
5090f0257eaSTom Tucker int svc_port_is_privileged(struct sockaddr *sin)
5100f0257eaSTom Tucker {
5110f0257eaSTom Tucker 	switch (sin->sa_family) {
5120f0257eaSTom Tucker 	case AF_INET:
5130f0257eaSTom Tucker 		return ntohs(((struct sockaddr_in *)sin)->sin_port)
5140f0257eaSTom Tucker 			< PROT_SOCK;
5150f0257eaSTom Tucker 	case AF_INET6:
5160f0257eaSTom Tucker 		return ntohs(((struct sockaddr_in6 *)sin)->sin6_port)
5170f0257eaSTom Tucker 			< PROT_SOCK;
5180f0257eaSTom Tucker 	default:
5190f0257eaSTom Tucker 		return 0;
5200f0257eaSTom Tucker 	}
5210f0257eaSTom Tucker }
5220f0257eaSTom Tucker 
5230f0257eaSTom Tucker /*
524c9233eb7SJeff Layton  * Make sure that we don't have too many active connections. If we have,
525c9233eb7SJeff Layton  * something must be dropped. It's not clear what will happen if we allow
526c9233eb7SJeff Layton  * "too many" connections, but when dealing with network-facing software,
527c9233eb7SJeff Layton  * we have to code defensively. Here we do that by imposing hard limits.
5280f0257eaSTom Tucker  *
5290f0257eaSTom Tucker  * There's no point in trying to do random drop here for DoS
5300f0257eaSTom Tucker  * prevention. The NFS clients does 1 reconnect in 15 seconds. An
5310f0257eaSTom Tucker  * attacker can easily beat that.
5320f0257eaSTom Tucker  *
5330f0257eaSTom Tucker  * The only somewhat efficient mechanism would be if drop old
5340f0257eaSTom Tucker  * connections from the same IP first. But right now we don't even
5350f0257eaSTom Tucker  * record the client IP in svc_sock.
536c9233eb7SJeff Layton  *
537c9233eb7SJeff Layton  * single-threaded services that expect a lot of clients will probably
538c9233eb7SJeff Layton  * need to set sv_maxconn to override the default value which is based
539c9233eb7SJeff Layton  * on the number of threads
5400f0257eaSTom Tucker  */
5410f0257eaSTom Tucker static void svc_check_conn_limits(struct svc_serv *serv)
5420f0257eaSTom Tucker {
543c9233eb7SJeff Layton 	unsigned int limit = serv->sv_maxconn ? serv->sv_maxconn :
544c9233eb7SJeff Layton 				(serv->sv_nrthreads+3) * 20;
545c9233eb7SJeff Layton 
546c9233eb7SJeff Layton 	if (serv->sv_tmpcnt > limit) {
5470f0257eaSTom Tucker 		struct svc_xprt *xprt = NULL;
5480f0257eaSTom Tucker 		spin_lock_bh(&serv->sv_lock);
5490f0257eaSTom Tucker 		if (!list_empty(&serv->sv_tempsocks)) {
5500f0257eaSTom Tucker 			/* Try to help the admin */
551e87cc472SJoe Perches 			net_notice_ratelimited("%s: too many open connections, consider increasing the %s\n",
552c9233eb7SJeff Layton 					       serv->sv_name, serv->sv_maxconn ?
553e87cc472SJoe Perches 					       "max number of connections" :
554e87cc472SJoe Perches 					       "number of threads");
5550f0257eaSTom Tucker 			/*
5560f0257eaSTom Tucker 			 * Always select the oldest connection. It's not fair,
5570f0257eaSTom Tucker 			 * but so is life
5580f0257eaSTom Tucker 			 */
5590f0257eaSTom Tucker 			xprt = list_entry(serv->sv_tempsocks.prev,
5600f0257eaSTom Tucker 					  struct svc_xprt,
5610f0257eaSTom Tucker 					  xpt_list);
5620f0257eaSTom Tucker 			set_bit(XPT_CLOSE, &xprt->xpt_flags);
5630f0257eaSTom Tucker 			svc_xprt_get(xprt);
5640f0257eaSTom Tucker 		}
5650f0257eaSTom Tucker 		spin_unlock_bh(&serv->sv_lock);
5660f0257eaSTom Tucker 
5670f0257eaSTom Tucker 		if (xprt) {
5680f0257eaSTom Tucker 			svc_xprt_enqueue(xprt);
5690f0257eaSTom Tucker 			svc_xprt_put(xprt);
5700f0257eaSTom Tucker 		}
5710f0257eaSTom Tucker 	}
5720f0257eaSTom Tucker }
5730f0257eaSTom Tucker 
5746797fa5aSJ. Bruce Fields int svc_alloc_arg(struct svc_rqst *rqstp)
5750f0257eaSTom Tucker {
5760f0257eaSTom Tucker 	struct svc_serv *serv = rqstp->rq_server;
5770f0257eaSTom Tucker 	struct xdr_buf *arg;
5786797fa5aSJ. Bruce Fields 	int pages;
5796797fa5aSJ. Bruce Fields 	int i;
5800f0257eaSTom Tucker 
5810f0257eaSTom Tucker 	/* now allocate needed pages.  If we get a failure, sleep briefly */
5820f0257eaSTom Tucker 	pages = (serv->sv_max_mesg + PAGE_SIZE) / PAGE_SIZE;
583b25cd058SWeston Andros Adamson 	WARN_ON_ONCE(pages >= RPCSVC_MAXPAGES);
584b25cd058SWeston Andros Adamson 	if (pages >= RPCSVC_MAXPAGES)
585b25cd058SWeston Andros Adamson 		/* use as many pages as possible */
586b25cd058SWeston Andros Adamson 		pages = RPCSVC_MAXPAGES - 1;
5870f0257eaSTom Tucker 	for (i = 0; i < pages ; i++)
5880f0257eaSTom Tucker 		while (rqstp->rq_pages[i] == NULL) {
5890f0257eaSTom Tucker 			struct page *p = alloc_page(GFP_KERNEL);
5900f0257eaSTom Tucker 			if (!p) {
5917b54fe61SJeff Layton 				set_current_state(TASK_INTERRUPTIBLE);
5927b54fe61SJeff Layton 				if (signalled() || kthread_should_stop()) {
5937b54fe61SJeff Layton 					set_current_state(TASK_RUNNING);
5947086721fSJeff Layton 					return -EINTR;
5957b54fe61SJeff Layton 				}
5967b54fe61SJeff Layton 				schedule_timeout(msecs_to_jiffies(500));
5970f0257eaSTom Tucker 			}
5980f0257eaSTom Tucker 			rqstp->rq_pages[i] = p;
5990f0257eaSTom Tucker 		}
6000f0257eaSTom Tucker 	rqstp->rq_pages[i++] = NULL; /* this might be seen in nfs_read_actor */
6010f0257eaSTom Tucker 
6020f0257eaSTom Tucker 	/* Make arg->head point to first page and arg->pages point to rest */
6030f0257eaSTom Tucker 	arg = &rqstp->rq_arg;
6040f0257eaSTom Tucker 	arg->head[0].iov_base = page_address(rqstp->rq_pages[0]);
6050f0257eaSTom Tucker 	arg->head[0].iov_len = PAGE_SIZE;
6060f0257eaSTom Tucker 	arg->pages = rqstp->rq_pages + 1;
6070f0257eaSTom Tucker 	arg->page_base = 0;
6080f0257eaSTom Tucker 	/* save at least one page for response */
6090f0257eaSTom Tucker 	arg->page_len = (pages-2)*PAGE_SIZE;
6100f0257eaSTom Tucker 	arg->len = (pages-1)*PAGE_SIZE;
6110f0257eaSTom Tucker 	arg->tail[0].iov_len = 0;
6126797fa5aSJ. Bruce Fields 	return 0;
6136797fa5aSJ. Bruce Fields }
6140f0257eaSTom Tucker 
6156797fa5aSJ. Bruce Fields struct svc_xprt *svc_get_next_xprt(struct svc_rqst *rqstp, long timeout)
6166797fa5aSJ. Bruce Fields {
6176797fa5aSJ. Bruce Fields 	struct svc_xprt *xprt;
6186797fa5aSJ. Bruce Fields 	struct svc_pool		*pool = rqstp->rq_pool;
6196797fa5aSJ. Bruce Fields 	DECLARE_WAITQUEUE(wait, current);
6206797fa5aSJ. Bruce Fields 	long			time_left;
6210f0257eaSTom Tucker 
622f16b6e8dSNeilBrown 	/* Normally we will wait up to 5 seconds for any required
623f16b6e8dSNeilBrown 	 * cache information to be provided.
624f16b6e8dSNeilBrown 	 */
625f16b6e8dSNeilBrown 	rqstp->rq_chandle.thread_wait = 5*HZ;
626f16b6e8dSNeilBrown 
6270f0257eaSTom Tucker 	spin_lock_bh(&pool->sp_lock);
6280f0257eaSTom Tucker 	xprt = svc_xprt_dequeue(pool);
6290f0257eaSTom Tucker 	if (xprt) {
6300f0257eaSTom Tucker 		rqstp->rq_xprt = xprt;
6310f0257eaSTom Tucker 		svc_xprt_get(xprt);
632f16b6e8dSNeilBrown 
633f16b6e8dSNeilBrown 		/* As there is a shortage of threads and this request
6346610f720SJ. Bruce Fields 		 * had to be queued, don't allow the thread to wait so
635f16b6e8dSNeilBrown 		 * long for cache updates.
636f16b6e8dSNeilBrown 		 */
637f16b6e8dSNeilBrown 		rqstp->rq_chandle.thread_wait = 1*HZ;
63835525b79SAndriy Skulysh 		pool->sp_task_pending = 0;
6390f0257eaSTom Tucker 	} else {
64035525b79SAndriy Skulysh 		if (pool->sp_task_pending) {
64135525b79SAndriy Skulysh 			pool->sp_task_pending = 0;
64235525b79SAndriy Skulysh 			spin_unlock_bh(&pool->sp_lock);
64335525b79SAndriy Skulysh 			return ERR_PTR(-EAGAIN);
64435525b79SAndriy Skulysh 		}
6450f0257eaSTom Tucker 		/* No data pending. Go to sleep */
6460f0257eaSTom Tucker 		svc_thread_enqueue(pool, rqstp);
6470f0257eaSTom Tucker 
6480f0257eaSTom Tucker 		/*
6490f0257eaSTom Tucker 		 * We have to be able to interrupt this wait
6500f0257eaSTom Tucker 		 * to bring down the daemons ...
6510f0257eaSTom Tucker 		 */
6520f0257eaSTom Tucker 		set_current_state(TASK_INTERRUPTIBLE);
6537086721fSJeff Layton 
6547086721fSJeff Layton 		/*
6557086721fSJeff Layton 		 * checking kthread_should_stop() here allows us to avoid
6567086721fSJeff Layton 		 * locking and signalling when stopping kthreads that call
6577086721fSJeff Layton 		 * svc_recv. If the thread has already been woken up, then
6587086721fSJeff Layton 		 * we can exit here without sleeping. If not, then it
6597086721fSJeff Layton 		 * it'll be woken up quickly during the schedule_timeout
6607086721fSJeff Layton 		 */
6617086721fSJeff Layton 		if (kthread_should_stop()) {
6627086721fSJeff Layton 			set_current_state(TASK_RUNNING);
6637086721fSJeff Layton 			spin_unlock_bh(&pool->sp_lock);
6646797fa5aSJ. Bruce Fields 			return ERR_PTR(-EINTR);
6657086721fSJeff Layton 		}
6667086721fSJeff Layton 
6670f0257eaSTom Tucker 		add_wait_queue(&rqstp->rq_wait, &wait);
6680f0257eaSTom Tucker 		spin_unlock_bh(&pool->sp_lock);
6690f0257eaSTom Tucker 
67003cf6c9fSGreg Banks 		time_left = schedule_timeout(timeout);
6710f0257eaSTom Tucker 
6720f0257eaSTom Tucker 		try_to_freeze();
6730f0257eaSTom Tucker 
6740f0257eaSTom Tucker 		spin_lock_bh(&pool->sp_lock);
6750f0257eaSTom Tucker 		remove_wait_queue(&rqstp->rq_wait, &wait);
67603cf6c9fSGreg Banks 		if (!time_left)
67703cf6c9fSGreg Banks 			pool->sp_stats.threads_timedout++;
6780f0257eaSTom Tucker 
6790f0257eaSTom Tucker 		xprt = rqstp->rq_xprt;
6800f0257eaSTom Tucker 		if (!xprt) {
6810f0257eaSTom Tucker 			svc_thread_dequeue(pool, rqstp);
6820f0257eaSTom Tucker 			spin_unlock_bh(&pool->sp_lock);
6830f0257eaSTom Tucker 			dprintk("svc: server %p, no data yet\n", rqstp);
6847086721fSJeff Layton 			if (signalled() || kthread_should_stop())
6856797fa5aSJ. Bruce Fields 				return ERR_PTR(-EINTR);
6867086721fSJeff Layton 			else
6876797fa5aSJ. Bruce Fields 				return ERR_PTR(-EAGAIN);
6880f0257eaSTom Tucker 		}
6890f0257eaSTom Tucker 	}
6900f0257eaSTom Tucker 	spin_unlock_bh(&pool->sp_lock);
6916797fa5aSJ. Bruce Fields 	return xprt;
6926797fa5aSJ. Bruce Fields }
6930f0257eaSTom Tucker 
69465b2e665SJ. Bruce Fields void svc_add_new_temp_xprt(struct svc_serv *serv, struct svc_xprt *newxpt)
69565b2e665SJ. Bruce Fields {
69665b2e665SJ. Bruce Fields 	spin_lock_bh(&serv->sv_lock);
69765b2e665SJ. Bruce Fields 	set_bit(XPT_TEMP, &newxpt->xpt_flags);
69865b2e665SJ. Bruce Fields 	list_add(&newxpt->xpt_list, &serv->sv_tempsocks);
69965b2e665SJ. Bruce Fields 	serv->sv_tmpcnt++;
70065b2e665SJ. Bruce Fields 	if (serv->sv_temptimer.function == NULL) {
70165b2e665SJ. Bruce Fields 		/* setup timer to age temp transports */
70265b2e665SJ. Bruce Fields 		setup_timer(&serv->sv_temptimer, svc_age_temp_xprts,
70365b2e665SJ. Bruce Fields 			    (unsigned long)serv);
70465b2e665SJ. Bruce Fields 		mod_timer(&serv->sv_temptimer,
70565b2e665SJ. Bruce Fields 			  jiffies + svc_conn_age_period * HZ);
70665b2e665SJ. Bruce Fields 	}
70765b2e665SJ. Bruce Fields 	spin_unlock_bh(&serv->sv_lock);
70865b2e665SJ. Bruce Fields 	svc_xprt_received(newxpt);
70965b2e665SJ. Bruce Fields }
71065b2e665SJ. Bruce Fields 
7116797fa5aSJ. Bruce Fields static int svc_handle_xprt(struct svc_rqst *rqstp, struct svc_xprt *xprt)
7126797fa5aSJ. Bruce Fields {
7136797fa5aSJ. Bruce Fields 	struct svc_serv *serv = rqstp->rq_server;
7146797fa5aSJ. Bruce Fields 	int len = 0;
7156797fa5aSJ. Bruce Fields 
7161b644b6eSJ. Bruce Fields 	if (test_bit(XPT_CLOSE, &xprt->xpt_flags)) {
7171b644b6eSJ. Bruce Fields 		dprintk("svc_recv: found XPT_CLOSE\n");
7181b644b6eSJ. Bruce Fields 		svc_delete_xprt(xprt);
719ca7896cdSJ. Bruce Fields 		/* Leave XPT_BUSY set on the dead xprt: */
7206797fa5aSJ. Bruce Fields 		return 0;
721ca7896cdSJ. Bruce Fields 	}
722ca7896cdSJ. Bruce Fields 	if (test_bit(XPT_LISTENER, &xprt->xpt_flags)) {
7230f0257eaSTom Tucker 		struct svc_xprt *newxpt;
7240f0257eaSTom Tucker 		/*
7250f0257eaSTom Tucker 		 * We know this module_get will succeed because the
7260f0257eaSTom Tucker 		 * listener holds a reference too
7270f0257eaSTom Tucker 		 */
72865b2e665SJ. Bruce Fields 		__module_get(xprt->xpt_class->xcl_owner);
7290f0257eaSTom Tucker 		svc_check_conn_limits(xprt->xpt_server);
73065b2e665SJ. Bruce Fields 		newxpt = xprt->xpt_ops->xpo_accept(xprt);
73165b2e665SJ. Bruce Fields 		if (newxpt)
73265b2e665SJ. Bruce Fields 			svc_add_new_temp_xprt(serv, newxpt);
7339c335c0bSJ. Bruce Fields 	} else if (xprt->xpt_ops->xpo_has_wspace(xprt)) {
7346797fa5aSJ. Bruce Fields 		/* XPT_DATA|XPT_DEFERRED case: */
7350f0257eaSTom Tucker 		dprintk("svc: server %p, pool %u, transport %p, inuse=%d\n",
7366797fa5aSJ. Bruce Fields 			rqstp, rqstp->rq_pool->sp_id, xprt,
7370f0257eaSTom Tucker 			atomic_read(&xprt->xpt_ref.refcount));
7380f0257eaSTom Tucker 		rqstp->rq_deferred = svc_deferred_dequeue(xprt);
739ca7896cdSJ. Bruce Fields 		if (rqstp->rq_deferred)
7400f0257eaSTom Tucker 			len = svc_deferred_recv(rqstp);
741ca7896cdSJ. Bruce Fields 		else
7420f0257eaSTom Tucker 			len = xprt->xpt_ops->xpo_recvfrom(rqstp);
7430f0257eaSTom Tucker 		dprintk("svc: got len=%d\n", len);
744d10f27a7SJ. Bruce Fields 		rqstp->rq_reserved = serv->sv_max_mesg;
745d10f27a7SJ. Bruce Fields 		atomic_add(rqstp->rq_reserved, &xprt->xpt_reserved);
7460f0257eaSTom Tucker 	}
7476797fa5aSJ. Bruce Fields 	/* clear XPT_BUSY: */
748ca7896cdSJ. Bruce Fields 	svc_xprt_received(xprt);
7496797fa5aSJ. Bruce Fields 	return len;
7506797fa5aSJ. Bruce Fields }
7516797fa5aSJ. Bruce Fields 
7526797fa5aSJ. Bruce Fields /*
7536797fa5aSJ. Bruce Fields  * Receive the next request on any transport.  This code is carefully
7546797fa5aSJ. Bruce Fields  * organised not to touch any cachelines in the shared svc_serv
7556797fa5aSJ. Bruce Fields  * structure, only cachelines in the local svc_pool.
7566797fa5aSJ. Bruce Fields  */
7576797fa5aSJ. Bruce Fields int svc_recv(struct svc_rqst *rqstp, long timeout)
7586797fa5aSJ. Bruce Fields {
7596797fa5aSJ. Bruce Fields 	struct svc_xprt		*xprt = NULL;
7606797fa5aSJ. Bruce Fields 	struct svc_serv		*serv = rqstp->rq_server;
7616797fa5aSJ. Bruce Fields 	int			len, err;
7626797fa5aSJ. Bruce Fields 
7636797fa5aSJ. Bruce Fields 	dprintk("svc: server %p waiting for data (to = %ld)\n",
7646797fa5aSJ. Bruce Fields 		rqstp, timeout);
7656797fa5aSJ. Bruce Fields 
7666797fa5aSJ. Bruce Fields 	if (rqstp->rq_xprt)
7676797fa5aSJ. Bruce Fields 		printk(KERN_ERR
7686797fa5aSJ. Bruce Fields 			"svc_recv: service %p, transport not NULL!\n",
7696797fa5aSJ. Bruce Fields 			 rqstp);
7706797fa5aSJ. Bruce Fields 	if (waitqueue_active(&rqstp->rq_wait))
7716797fa5aSJ. Bruce Fields 		printk(KERN_ERR
7726797fa5aSJ. Bruce Fields 			"svc_recv: service %p, wait queue active!\n",
7736797fa5aSJ. Bruce Fields 			 rqstp);
7746797fa5aSJ. Bruce Fields 
7756797fa5aSJ. Bruce Fields 	err = svc_alloc_arg(rqstp);
7766797fa5aSJ. Bruce Fields 	if (err)
7776797fa5aSJ. Bruce Fields 		return err;
7786797fa5aSJ. Bruce Fields 
7796797fa5aSJ. Bruce Fields 	try_to_freeze();
7806797fa5aSJ. Bruce Fields 	cond_resched();
7816797fa5aSJ. Bruce Fields 	if (signalled() || kthread_should_stop())
7826797fa5aSJ. Bruce Fields 		return -EINTR;
7836797fa5aSJ. Bruce Fields 
7846797fa5aSJ. Bruce Fields 	xprt = svc_get_next_xprt(rqstp, timeout);
7856797fa5aSJ. Bruce Fields 	if (IS_ERR(xprt))
7866797fa5aSJ. Bruce Fields 		return PTR_ERR(xprt);
7876797fa5aSJ. Bruce Fields 
7886797fa5aSJ. Bruce Fields 	len = svc_handle_xprt(rqstp, xprt);
7890f0257eaSTom Tucker 
7900f0257eaSTom Tucker 	/* No data, incomplete (TCP) read, or accept() */
7919f9d2ebeSJ. Bruce Fields 	if (len <= 0)
792ca7896cdSJ. Bruce Fields 		goto out;
793ca7896cdSJ. Bruce Fields 
7940f0257eaSTom Tucker 	clear_bit(XPT_OLD, &xprt->xpt_flags);
7950f0257eaSTom Tucker 
7960f0257eaSTom Tucker 	rqstp->rq_secure = svc_port_is_privileged(svc_addr(rqstp));
7970f0257eaSTom Tucker 	rqstp->rq_chandle.defer = svc_defer;
7980f0257eaSTom Tucker 
7990f0257eaSTom Tucker 	if (serv->sv_stats)
8000f0257eaSTom Tucker 		serv->sv_stats->netcnt++;
8010f0257eaSTom Tucker 	return len;
802ca7896cdSJ. Bruce Fields out:
803ca7896cdSJ. Bruce Fields 	rqstp->rq_res.len = 0;
804ca7896cdSJ. Bruce Fields 	svc_xprt_release(rqstp);
805ca7896cdSJ. Bruce Fields 	return -EAGAIN;
8060f0257eaSTom Tucker }
80724c3767eSTrond Myklebust EXPORT_SYMBOL_GPL(svc_recv);
8080f0257eaSTom Tucker 
8090f0257eaSTom Tucker /*
8100f0257eaSTom Tucker  * Drop request
8110f0257eaSTom Tucker  */
8120f0257eaSTom Tucker void svc_drop(struct svc_rqst *rqstp)
8130f0257eaSTom Tucker {
8140f0257eaSTom Tucker 	dprintk("svc: xprt %p dropped request\n", rqstp->rq_xprt);
8150f0257eaSTom Tucker 	svc_xprt_release(rqstp);
8160f0257eaSTom Tucker }
81724c3767eSTrond Myklebust EXPORT_SYMBOL_GPL(svc_drop);
8180f0257eaSTom Tucker 
8190f0257eaSTom Tucker /*
8200f0257eaSTom Tucker  * Return reply to client.
8210f0257eaSTom Tucker  */
8220f0257eaSTom Tucker int svc_send(struct svc_rqst *rqstp)
8230f0257eaSTom Tucker {
8240f0257eaSTom Tucker 	struct svc_xprt	*xprt;
8250f0257eaSTom Tucker 	int		len;
8260f0257eaSTom Tucker 	struct xdr_buf	*xb;
8270f0257eaSTom Tucker 
8280f0257eaSTom Tucker 	xprt = rqstp->rq_xprt;
8290f0257eaSTom Tucker 	if (!xprt)
8300f0257eaSTom Tucker 		return -EFAULT;
8310f0257eaSTom Tucker 
8320f0257eaSTom Tucker 	/* release the receive skb before sending the reply */
8330f0257eaSTom Tucker 	rqstp->rq_xprt->xpt_ops->xpo_release_rqst(rqstp);
8340f0257eaSTom Tucker 
8350f0257eaSTom Tucker 	/* calculate over-all length */
8360f0257eaSTom Tucker 	xb = &rqstp->rq_res;
8370f0257eaSTom Tucker 	xb->len = xb->head[0].iov_len +
8380f0257eaSTom Tucker 		xb->page_len +
8390f0257eaSTom Tucker 		xb->tail[0].iov_len;
8400f0257eaSTom Tucker 
8410f0257eaSTom Tucker 	/* Grab mutex to serialize outgoing data. */
8420f0257eaSTom Tucker 	mutex_lock(&xprt->xpt_mutex);
843f06f00a2SJ. Bruce Fields 	if (test_bit(XPT_DEAD, &xprt->xpt_flags)
844f06f00a2SJ. Bruce Fields 			|| test_bit(XPT_CLOSE, &xprt->xpt_flags))
8450f0257eaSTom Tucker 		len = -ENOTCONN;
8460f0257eaSTom Tucker 	else
8470f0257eaSTom Tucker 		len = xprt->xpt_ops->xpo_sendto(rqstp);
8480f0257eaSTom Tucker 	mutex_unlock(&xprt->xpt_mutex);
8494cfc7e60SRahul Iyer 	rpc_wake_up(&xprt->xpt_bc_pending);
8500f0257eaSTom Tucker 	svc_xprt_release(rqstp);
8510f0257eaSTom Tucker 
8520f0257eaSTom Tucker 	if (len == -ECONNREFUSED || len == -ENOTCONN || len == -EAGAIN)
8530f0257eaSTom Tucker 		return 0;
8540f0257eaSTom Tucker 	return len;
8550f0257eaSTom Tucker }
8560f0257eaSTom Tucker 
8570f0257eaSTom Tucker /*
8580f0257eaSTom Tucker  * Timer function to close old temporary transports, using
8590f0257eaSTom Tucker  * a mark-and-sweep algorithm.
8600f0257eaSTom Tucker  */
8610f0257eaSTom Tucker static void svc_age_temp_xprts(unsigned long closure)
8620f0257eaSTom Tucker {
8630f0257eaSTom Tucker 	struct svc_serv *serv = (struct svc_serv *)closure;
8640f0257eaSTom Tucker 	struct svc_xprt *xprt;
8650f0257eaSTom Tucker 	struct list_head *le, *next;
8660f0257eaSTom Tucker 
8670f0257eaSTom Tucker 	dprintk("svc_age_temp_xprts\n");
8680f0257eaSTom Tucker 
8690f0257eaSTom Tucker 	if (!spin_trylock_bh(&serv->sv_lock)) {
8700f0257eaSTom Tucker 		/* busy, try again 1 sec later */
8710f0257eaSTom Tucker 		dprintk("svc_age_temp_xprts: busy\n");
8720f0257eaSTom Tucker 		mod_timer(&serv->sv_temptimer, jiffies + HZ);
8730f0257eaSTom Tucker 		return;
8740f0257eaSTom Tucker 	}
8750f0257eaSTom Tucker 
8760f0257eaSTom Tucker 	list_for_each_safe(le, next, &serv->sv_tempsocks) {
8770f0257eaSTom Tucker 		xprt = list_entry(le, struct svc_xprt, xpt_list);
8780f0257eaSTom Tucker 
8790f0257eaSTom Tucker 		/* First time through, just mark it OLD. Second time
8800f0257eaSTom Tucker 		 * through, close it. */
8810f0257eaSTom Tucker 		if (!test_and_set_bit(XPT_OLD, &xprt->xpt_flags))
8820f0257eaSTom Tucker 			continue;
883f64f9e71SJoe Perches 		if (atomic_read(&xprt->xpt_ref.refcount) > 1 ||
884f64f9e71SJoe Perches 		    test_bit(XPT_BUSY, &xprt->xpt_flags))
8850f0257eaSTom Tucker 			continue;
886e75bafbfSJ. Bruce Fields 		list_del_init(le);
8870f0257eaSTom Tucker 		set_bit(XPT_CLOSE, &xprt->xpt_flags);
8880f0257eaSTom Tucker 		set_bit(XPT_DETACHED, &xprt->xpt_flags);
8890f0257eaSTom Tucker 		dprintk("queuing xprt %p for closing\n", xprt);
8900f0257eaSTom Tucker 
8910f0257eaSTom Tucker 		/* a thread will dequeue and close it soon */
8920f0257eaSTom Tucker 		svc_xprt_enqueue(xprt);
8930f0257eaSTom Tucker 	}
894e75bafbfSJ. Bruce Fields 	spin_unlock_bh(&serv->sv_lock);
8950f0257eaSTom Tucker 
8960f0257eaSTom Tucker 	mod_timer(&serv->sv_temptimer, jiffies + svc_conn_age_period * HZ);
8970f0257eaSTom Tucker }
8980f0257eaSTom Tucker 
899edc7a894SJ. Bruce Fields static void call_xpt_users(struct svc_xprt *xprt)
900edc7a894SJ. Bruce Fields {
901edc7a894SJ. Bruce Fields 	struct svc_xpt_user *u;
902edc7a894SJ. Bruce Fields 
903edc7a894SJ. Bruce Fields 	spin_lock(&xprt->xpt_lock);
904edc7a894SJ. Bruce Fields 	while (!list_empty(&xprt->xpt_users)) {
905edc7a894SJ. Bruce Fields 		u = list_first_entry(&xprt->xpt_users, struct svc_xpt_user, list);
906edc7a894SJ. Bruce Fields 		list_del(&u->list);
907edc7a894SJ. Bruce Fields 		u->callback(u);
908edc7a894SJ. Bruce Fields 	}
909edc7a894SJ. Bruce Fields 	spin_unlock(&xprt->xpt_lock);
910edc7a894SJ. Bruce Fields }
911edc7a894SJ. Bruce Fields 
9120f0257eaSTom Tucker /*
9130f0257eaSTom Tucker  * Remove a dead transport
9140f0257eaSTom Tucker  */
9157710ec36SJ. Bruce Fields static void svc_delete_xprt(struct svc_xprt *xprt)
9160f0257eaSTom Tucker {
9170f0257eaSTom Tucker 	struct svc_serv	*serv = xprt->xpt_server;
91822945e4aSTom Tucker 	struct svc_deferred_req *dr;
91922945e4aSTom Tucker 
92022945e4aSTom Tucker 	/* Only do this once */
92122945e4aSTom Tucker 	if (test_and_set_bit(XPT_DEAD, &xprt->xpt_flags))
922ac9303ebSJ. Bruce Fields 		BUG();
9230f0257eaSTom Tucker 
9240f0257eaSTom Tucker 	dprintk("svc: svc_delete_xprt(%p)\n", xprt);
9250f0257eaSTom Tucker 	xprt->xpt_ops->xpo_detach(xprt);
9260f0257eaSTom Tucker 
9270f0257eaSTom Tucker 	spin_lock_bh(&serv->sv_lock);
9280f0257eaSTom Tucker 	if (!test_and_set_bit(XPT_DETACHED, &xprt->xpt_flags))
9290f0257eaSTom Tucker 		list_del_init(&xprt->xpt_list);
93001047298SWeston Andros Adamson 	WARN_ON_ONCE(!list_empty(&xprt->xpt_ready));
9310f0257eaSTom Tucker 	if (test_bit(XPT_TEMP, &xprt->xpt_flags))
9320f0257eaSTom Tucker 		serv->sv_tmpcnt--;
933788e69e5SJ. Bruce Fields 	spin_unlock_bh(&serv->sv_lock);
93422945e4aSTom Tucker 
935ab1b18f7SNeil Brown 	while ((dr = svc_deferred_dequeue(xprt)) != NULL)
93622945e4aSTom Tucker 		kfree(dr);
93722945e4aSTom Tucker 
938edc7a894SJ. Bruce Fields 	call_xpt_users(xprt);
93922945e4aSTom Tucker 	svc_xprt_put(xprt);
9400f0257eaSTom Tucker }
9410f0257eaSTom Tucker 
9420f0257eaSTom Tucker void svc_close_xprt(struct svc_xprt *xprt)
9430f0257eaSTom Tucker {
9440f0257eaSTom Tucker 	set_bit(XPT_CLOSE, &xprt->xpt_flags);
9450f0257eaSTom Tucker 	if (test_and_set_bit(XPT_BUSY, &xprt->xpt_flags))
9460f0257eaSTom Tucker 		/* someone else will have to effect the close */
9470f0257eaSTom Tucker 		return;
948b1763316SJ. Bruce Fields 	/*
949b1763316SJ. Bruce Fields 	 * We expect svc_close_xprt() to work even when no threads are
950b1763316SJ. Bruce Fields 	 * running (e.g., while configuring the server before starting
951b1763316SJ. Bruce Fields 	 * any threads), so if the transport isn't busy, we delete
952b1763316SJ. Bruce Fields 	 * it ourself:
953b1763316SJ. Bruce Fields 	 */
9540f0257eaSTom Tucker 	svc_delete_xprt(xprt);
9550f0257eaSTom Tucker }
956a217813fSTom Tucker EXPORT_SYMBOL_GPL(svc_close_xprt);
9570f0257eaSTom Tucker 
958719f8bccSJ. Bruce Fields static void svc_close_list(struct svc_serv *serv, struct list_head *xprt_list, struct net *net)
9590f0257eaSTom Tucker {
9600f0257eaSTom Tucker 	struct svc_xprt *xprt;
9610f0257eaSTom Tucker 
962719f8bccSJ. Bruce Fields 	spin_lock(&serv->sv_lock);
963b4f36f88SJ. Bruce Fields 	list_for_each_entry(xprt, xprt_list, xpt_list) {
9647b147f1fSStanislav Kinsbursky 		if (xprt->xpt_net != net)
9657b147f1fSStanislav Kinsbursky 			continue;
9660f0257eaSTom Tucker 		set_bit(XPT_CLOSE, &xprt->xpt_flags);
967b4f36f88SJ. Bruce Fields 		set_bit(XPT_BUSY, &xprt->xpt_flags);
9680f0257eaSTom Tucker 	}
969719f8bccSJ. Bruce Fields 	spin_unlock(&serv->sv_lock);
9700f0257eaSTom Tucker }
9710f0257eaSTom Tucker 
9727b147f1fSStanislav Kinsbursky static void svc_clear_pools(struct svc_serv *serv, struct net *net)
9732fefb8a0SJ. Bruce Fields {
974b4f36f88SJ. Bruce Fields 	struct svc_pool *pool;
975b4f36f88SJ. Bruce Fields 	struct svc_xprt *xprt;
976b4f36f88SJ. Bruce Fields 	struct svc_xprt *tmp;
977b4f36f88SJ. Bruce Fields 	int i;
978b4f36f88SJ. Bruce Fields 
979b4f36f88SJ. Bruce Fields 	for (i = 0; i < serv->sv_nrpools; i++) {
980b4f36f88SJ. Bruce Fields 		pool = &serv->sv_pools[i];
981b4f36f88SJ. Bruce Fields 
982b4f36f88SJ. Bruce Fields 		spin_lock_bh(&pool->sp_lock);
9836f513365SStanislav Kinsbursky 		list_for_each_entry_safe(xprt, tmp, &pool->sp_sockets, xpt_ready) {
9847b147f1fSStanislav Kinsbursky 			if (xprt->xpt_net != net)
9857b147f1fSStanislav Kinsbursky 				continue;
986b4f36f88SJ. Bruce Fields 			list_del_init(&xprt->xpt_ready);
987b4f36f88SJ. Bruce Fields 		}
988b4f36f88SJ. Bruce Fields 		spin_unlock_bh(&pool->sp_lock);
989b4f36f88SJ. Bruce Fields 	}
9906f513365SStanislav Kinsbursky }
9916f513365SStanislav Kinsbursky 
992719f8bccSJ. Bruce Fields static void svc_clear_list(struct svc_serv *serv, struct list_head *xprt_list, struct net *net)
9936f513365SStanislav Kinsbursky {
9946f513365SStanislav Kinsbursky 	struct svc_xprt *xprt;
9956f513365SStanislav Kinsbursky 	struct svc_xprt *tmp;
996719f8bccSJ. Bruce Fields 	LIST_HEAD(victims);
9976f513365SStanislav Kinsbursky 
998719f8bccSJ. Bruce Fields 	spin_lock(&serv->sv_lock);
9993a22bf50SStanislav Kinsbursky 	list_for_each_entry_safe(xprt, tmp, xprt_list, xpt_list) {
10007b147f1fSStanislav Kinsbursky 		if (xprt->xpt_net != net)
10017b147f1fSStanislav Kinsbursky 			continue;
1002719f8bccSJ. Bruce Fields 		list_move(&xprt->xpt_list, &victims);
10033a22bf50SStanislav Kinsbursky 	}
1004719f8bccSJ. Bruce Fields 	spin_unlock(&serv->sv_lock);
1005719f8bccSJ. Bruce Fields 
1006719f8bccSJ. Bruce Fields 	list_for_each_entry_safe(xprt, tmp, &victims, xpt_list)
1007719f8bccSJ. Bruce Fields 		svc_delete_xprt(xprt);
10083a22bf50SStanislav Kinsbursky }
10093a22bf50SStanislav Kinsbursky 
10107b147f1fSStanislav Kinsbursky void svc_close_net(struct svc_serv *serv, struct net *net)
10113a22bf50SStanislav Kinsbursky {
1012719f8bccSJ. Bruce Fields 	svc_close_list(serv, &serv->sv_tempsocks, net);
1013719f8bccSJ. Bruce Fields 	svc_close_list(serv, &serv->sv_permsocks, net);
10146f513365SStanislav Kinsbursky 
10157b147f1fSStanislav Kinsbursky 	svc_clear_pools(serv, net);
1016b4f36f88SJ. Bruce Fields 	/*
1017b4f36f88SJ. Bruce Fields 	 * At this point the sp_sockets lists will stay empty, since
10183ddbe879SJ. Bruce Fields 	 * svc_xprt_enqueue will not add new entries without taking the
1019b4f36f88SJ. Bruce Fields 	 * sp_lock and checking XPT_BUSY.
1020b4f36f88SJ. Bruce Fields 	 */
1021719f8bccSJ. Bruce Fields 	svc_clear_list(serv, &serv->sv_tempsocks, net);
1022719f8bccSJ. Bruce Fields 	svc_clear_list(serv, &serv->sv_permsocks, net);
10232fefb8a0SJ. Bruce Fields }
10242fefb8a0SJ. Bruce Fields 
10250f0257eaSTom Tucker /*
10260f0257eaSTom Tucker  * Handle defer and revisit of requests
10270f0257eaSTom Tucker  */
10280f0257eaSTom Tucker 
10290f0257eaSTom Tucker static void svc_revisit(struct cache_deferred_req *dreq, int too_many)
10300f0257eaSTom Tucker {
10310f0257eaSTom Tucker 	struct svc_deferred_req *dr =
10320f0257eaSTom Tucker 		container_of(dreq, struct svc_deferred_req, handle);
10330f0257eaSTom Tucker 	struct svc_xprt *xprt = dr->xprt;
10340f0257eaSTom Tucker 
103522945e4aSTom Tucker 	spin_lock(&xprt->xpt_lock);
103622945e4aSTom Tucker 	set_bit(XPT_DEFERRED, &xprt->xpt_flags);
103722945e4aSTom Tucker 	if (too_many || test_bit(XPT_DEAD, &xprt->xpt_flags)) {
103822945e4aSTom Tucker 		spin_unlock(&xprt->xpt_lock);
103922945e4aSTom Tucker 		dprintk("revisit canceled\n");
10400f0257eaSTom Tucker 		svc_xprt_put(xprt);
10410f0257eaSTom Tucker 		kfree(dr);
10420f0257eaSTom Tucker 		return;
10430f0257eaSTom Tucker 	}
10440f0257eaSTom Tucker 	dprintk("revisit queued\n");
10450f0257eaSTom Tucker 	dr->xprt = NULL;
10460f0257eaSTom Tucker 	list_add(&dr->handle.recent, &xprt->xpt_deferred);
10470f0257eaSTom Tucker 	spin_unlock(&xprt->xpt_lock);
10480f0257eaSTom Tucker 	svc_xprt_enqueue(xprt);
10490f0257eaSTom Tucker 	svc_xprt_put(xprt);
10500f0257eaSTom Tucker }
10510f0257eaSTom Tucker 
1052260c1d12STom Tucker /*
1053260c1d12STom Tucker  * Save the request off for later processing. The request buffer looks
1054260c1d12STom Tucker  * like this:
1055260c1d12STom Tucker  *
1056260c1d12STom Tucker  * <xprt-header><rpc-header><rpc-pagelist><rpc-tail>
1057260c1d12STom Tucker  *
1058260c1d12STom Tucker  * This code can only handle requests that consist of an xprt-header
1059260c1d12STom Tucker  * and rpc-header.
1060260c1d12STom Tucker  */
10610f0257eaSTom Tucker static struct cache_deferred_req *svc_defer(struct cache_req *req)
10620f0257eaSTom Tucker {
10630f0257eaSTom Tucker 	struct svc_rqst *rqstp = container_of(req, struct svc_rqst, rq_chandle);
10640f0257eaSTom Tucker 	struct svc_deferred_req *dr;
10650f0257eaSTom Tucker 
10662f425878SAndy Adamson 	if (rqstp->rq_arg.page_len || !rqstp->rq_usedeferral)
10670f0257eaSTom Tucker 		return NULL; /* if more than a page, give up FIXME */
10680f0257eaSTom Tucker 	if (rqstp->rq_deferred) {
10690f0257eaSTom Tucker 		dr = rqstp->rq_deferred;
10700f0257eaSTom Tucker 		rqstp->rq_deferred = NULL;
10710f0257eaSTom Tucker 	} else {
1072260c1d12STom Tucker 		size_t skip;
1073260c1d12STom Tucker 		size_t size;
10740f0257eaSTom Tucker 		/* FIXME maybe discard if size too large */
1075260c1d12STom Tucker 		size = sizeof(struct svc_deferred_req) + rqstp->rq_arg.len;
10760f0257eaSTom Tucker 		dr = kmalloc(size, GFP_KERNEL);
10770f0257eaSTom Tucker 		if (dr == NULL)
10780f0257eaSTom Tucker 			return NULL;
10790f0257eaSTom Tucker 
10800f0257eaSTom Tucker 		dr->handle.owner = rqstp->rq_server;
10810f0257eaSTom Tucker 		dr->prot = rqstp->rq_prot;
10820f0257eaSTom Tucker 		memcpy(&dr->addr, &rqstp->rq_addr, rqstp->rq_addrlen);
10830f0257eaSTom Tucker 		dr->addrlen = rqstp->rq_addrlen;
10840f0257eaSTom Tucker 		dr->daddr = rqstp->rq_daddr;
10850f0257eaSTom Tucker 		dr->argslen = rqstp->rq_arg.len >> 2;
1086260c1d12STom Tucker 		dr->xprt_hlen = rqstp->rq_xprt_hlen;
1087260c1d12STom Tucker 
1088260c1d12STom Tucker 		/* back up head to the start of the buffer and copy */
1089260c1d12STom Tucker 		skip = rqstp->rq_arg.len - rqstp->rq_arg.head[0].iov_len;
10900f0257eaSTom Tucker 		memcpy(dr->args, rqstp->rq_arg.head[0].iov_base - skip,
10910f0257eaSTom Tucker 		       dr->argslen << 2);
10920f0257eaSTom Tucker 	}
10930f0257eaSTom Tucker 	svc_xprt_get(rqstp->rq_xprt);
10940f0257eaSTom Tucker 	dr->xprt = rqstp->rq_xprt;
10959e701c61SJ. Bruce Fields 	rqstp->rq_dropme = true;
10960f0257eaSTom Tucker 
10970f0257eaSTom Tucker 	dr->handle.revisit = svc_revisit;
10980f0257eaSTom Tucker 	return &dr->handle;
10990f0257eaSTom Tucker }
11000f0257eaSTom Tucker 
11010f0257eaSTom Tucker /*
11020f0257eaSTom Tucker  * recv data from a deferred request into an active one
11030f0257eaSTom Tucker  */
11040f0257eaSTom Tucker static int svc_deferred_recv(struct svc_rqst *rqstp)
11050f0257eaSTom Tucker {
11060f0257eaSTom Tucker 	struct svc_deferred_req *dr = rqstp->rq_deferred;
11070f0257eaSTom Tucker 
1108260c1d12STom Tucker 	/* setup iov_base past transport header */
1109260c1d12STom Tucker 	rqstp->rq_arg.head[0].iov_base = dr->args + (dr->xprt_hlen>>2);
1110260c1d12STom Tucker 	/* The iov_len does not include the transport header bytes */
1111260c1d12STom Tucker 	rqstp->rq_arg.head[0].iov_len = (dr->argslen<<2) - dr->xprt_hlen;
11120f0257eaSTom Tucker 	rqstp->rq_arg.page_len = 0;
1113260c1d12STom Tucker 	/* The rq_arg.len includes the transport header bytes */
11140f0257eaSTom Tucker 	rqstp->rq_arg.len     = dr->argslen<<2;
11150f0257eaSTom Tucker 	rqstp->rq_prot        = dr->prot;
11160f0257eaSTom Tucker 	memcpy(&rqstp->rq_addr, &dr->addr, dr->addrlen);
11170f0257eaSTom Tucker 	rqstp->rq_addrlen     = dr->addrlen;
1118260c1d12STom Tucker 	/* Save off transport header len in case we get deferred again */
1119260c1d12STom Tucker 	rqstp->rq_xprt_hlen   = dr->xprt_hlen;
11200f0257eaSTom Tucker 	rqstp->rq_daddr       = dr->daddr;
11210f0257eaSTom Tucker 	rqstp->rq_respages    = rqstp->rq_pages;
1122260c1d12STom Tucker 	return (dr->argslen<<2) - dr->xprt_hlen;
11230f0257eaSTom Tucker }
11240f0257eaSTom Tucker 
11250f0257eaSTom Tucker 
11260f0257eaSTom Tucker static struct svc_deferred_req *svc_deferred_dequeue(struct svc_xprt *xprt)
11270f0257eaSTom Tucker {
11280f0257eaSTom Tucker 	struct svc_deferred_req *dr = NULL;
11290f0257eaSTom Tucker 
11300f0257eaSTom Tucker 	if (!test_bit(XPT_DEFERRED, &xprt->xpt_flags))
11310f0257eaSTom Tucker 		return NULL;
11320f0257eaSTom Tucker 	spin_lock(&xprt->xpt_lock);
11330f0257eaSTom Tucker 	if (!list_empty(&xprt->xpt_deferred)) {
11340f0257eaSTom Tucker 		dr = list_entry(xprt->xpt_deferred.next,
11350f0257eaSTom Tucker 				struct svc_deferred_req,
11360f0257eaSTom Tucker 				handle.recent);
11370f0257eaSTom Tucker 		list_del_init(&dr->handle.recent);
113862bac4afSJ. Bruce Fields 	} else
113962bac4afSJ. Bruce Fields 		clear_bit(XPT_DEFERRED, &xprt->xpt_flags);
11400f0257eaSTom Tucker 	spin_unlock(&xprt->xpt_lock);
11410f0257eaSTom Tucker 	return dr;
11420f0257eaSTom Tucker }
11437fcb98d5STom Tucker 
1144156e6209SChuck Lever /**
1145156e6209SChuck Lever  * svc_find_xprt - find an RPC transport instance
1146156e6209SChuck Lever  * @serv: pointer to svc_serv to search
1147156e6209SChuck Lever  * @xcl_name: C string containing transport's class name
11484cb54ca2SStanislav Kinsbursky  * @net: owner net pointer
1149156e6209SChuck Lever  * @af: Address family of transport's local address
1150156e6209SChuck Lever  * @port: transport's IP port number
1151156e6209SChuck Lever  *
11527fcb98d5STom Tucker  * Return the transport instance pointer for the endpoint accepting
11537fcb98d5STom Tucker  * connections/peer traffic from the specified transport class,
11547fcb98d5STom Tucker  * address family and port.
11557fcb98d5STom Tucker  *
11567fcb98d5STom Tucker  * Specifying 0 for the address family or port is effectively a
11577fcb98d5STom Tucker  * wild-card, and will result in matching the first transport in the
11587fcb98d5STom Tucker  * service's list that has a matching class name.
11597fcb98d5STom Tucker  */
1160156e6209SChuck Lever struct svc_xprt *svc_find_xprt(struct svc_serv *serv, const char *xcl_name,
11614cb54ca2SStanislav Kinsbursky 			       struct net *net, const sa_family_t af,
11624cb54ca2SStanislav Kinsbursky 			       const unsigned short port)
11637fcb98d5STom Tucker {
11647fcb98d5STom Tucker 	struct svc_xprt *xprt;
11657fcb98d5STom Tucker 	struct svc_xprt *found = NULL;
11667fcb98d5STom Tucker 
11677fcb98d5STom Tucker 	/* Sanity check the args */
1168156e6209SChuck Lever 	if (serv == NULL || xcl_name == NULL)
11697fcb98d5STom Tucker 		return found;
11707fcb98d5STom Tucker 
11717fcb98d5STom Tucker 	spin_lock_bh(&serv->sv_lock);
11727fcb98d5STom Tucker 	list_for_each_entry(xprt, &serv->sv_permsocks, xpt_list) {
11734cb54ca2SStanislav Kinsbursky 		if (xprt->xpt_net != net)
11744cb54ca2SStanislav Kinsbursky 			continue;
11757fcb98d5STom Tucker 		if (strcmp(xprt->xpt_class->xcl_name, xcl_name))
11767fcb98d5STom Tucker 			continue;
11777fcb98d5STom Tucker 		if (af != AF_UNSPEC && af != xprt->xpt_local.ss_family)
11787fcb98d5STom Tucker 			continue;
1179156e6209SChuck Lever 		if (port != 0 && port != svc_xprt_local_port(xprt))
11807fcb98d5STom Tucker 			continue;
11817fcb98d5STom Tucker 		found = xprt;
1182a217813fSTom Tucker 		svc_xprt_get(xprt);
11837fcb98d5STom Tucker 		break;
11847fcb98d5STom Tucker 	}
11857fcb98d5STom Tucker 	spin_unlock_bh(&serv->sv_lock);
11867fcb98d5STom Tucker 	return found;
11877fcb98d5STom Tucker }
11887fcb98d5STom Tucker EXPORT_SYMBOL_GPL(svc_find_xprt);
11899571af18STom Tucker 
1190335c54bdSChuck Lever static int svc_one_xprt_name(const struct svc_xprt *xprt,
1191335c54bdSChuck Lever 			     char *pos, int remaining)
1192335c54bdSChuck Lever {
1193335c54bdSChuck Lever 	int len;
1194335c54bdSChuck Lever 
1195335c54bdSChuck Lever 	len = snprintf(pos, remaining, "%s %u\n",
1196335c54bdSChuck Lever 			xprt->xpt_class->xcl_name,
1197335c54bdSChuck Lever 			svc_xprt_local_port(xprt));
1198335c54bdSChuck Lever 	if (len >= remaining)
1199335c54bdSChuck Lever 		return -ENAMETOOLONG;
1200335c54bdSChuck Lever 	return len;
1201335c54bdSChuck Lever }
1202335c54bdSChuck Lever 
1203335c54bdSChuck Lever /**
1204335c54bdSChuck Lever  * svc_xprt_names - format a buffer with a list of transport names
1205335c54bdSChuck Lever  * @serv: pointer to an RPC service
1206335c54bdSChuck Lever  * @buf: pointer to a buffer to be filled in
1207335c54bdSChuck Lever  * @buflen: length of buffer to be filled in
1208335c54bdSChuck Lever  *
1209335c54bdSChuck Lever  * Fills in @buf with a string containing a list of transport names,
1210335c54bdSChuck Lever  * each name terminated with '\n'.
1211335c54bdSChuck Lever  *
1212335c54bdSChuck Lever  * Returns positive length of the filled-in string on success; otherwise
1213335c54bdSChuck Lever  * a negative errno value is returned if an error occurs.
12149571af18STom Tucker  */
1215335c54bdSChuck Lever int svc_xprt_names(struct svc_serv *serv, char *buf, const int buflen)
12169571af18STom Tucker {
12179571af18STom Tucker 	struct svc_xprt *xprt;
1218335c54bdSChuck Lever 	int len, totlen;
1219335c54bdSChuck Lever 	char *pos;
12209571af18STom Tucker 
12219571af18STom Tucker 	/* Sanity check args */
12229571af18STom Tucker 	if (!serv)
12239571af18STom Tucker 		return 0;
12249571af18STom Tucker 
12259571af18STom Tucker 	spin_lock_bh(&serv->sv_lock);
1226335c54bdSChuck Lever 
1227335c54bdSChuck Lever 	pos = buf;
1228335c54bdSChuck Lever 	totlen = 0;
12299571af18STom Tucker 	list_for_each_entry(xprt, &serv->sv_permsocks, xpt_list) {
1230335c54bdSChuck Lever 		len = svc_one_xprt_name(xprt, pos, buflen - totlen);
1231335c54bdSChuck Lever 		if (len < 0) {
1232335c54bdSChuck Lever 			*buf = '\0';
1233335c54bdSChuck Lever 			totlen = len;
1234335c54bdSChuck Lever 		}
1235335c54bdSChuck Lever 		if (len <= 0)
12369571af18STom Tucker 			break;
1237335c54bdSChuck Lever 
1238335c54bdSChuck Lever 		pos += len;
12399571af18STom Tucker 		totlen += len;
12409571af18STom Tucker 	}
1241335c54bdSChuck Lever 
12429571af18STom Tucker 	spin_unlock_bh(&serv->sv_lock);
12439571af18STom Tucker 	return totlen;
12449571af18STom Tucker }
12459571af18STom Tucker EXPORT_SYMBOL_GPL(svc_xprt_names);
124603cf6c9fSGreg Banks 
124703cf6c9fSGreg Banks 
124803cf6c9fSGreg Banks /*----------------------------------------------------------------------------*/
124903cf6c9fSGreg Banks 
125003cf6c9fSGreg Banks static void *svc_pool_stats_start(struct seq_file *m, loff_t *pos)
125103cf6c9fSGreg Banks {
125203cf6c9fSGreg Banks 	unsigned int pidx = (unsigned int)*pos;
125303cf6c9fSGreg Banks 	struct svc_serv *serv = m->private;
125403cf6c9fSGreg Banks 
125503cf6c9fSGreg Banks 	dprintk("svc_pool_stats_start, *pidx=%u\n", pidx);
125603cf6c9fSGreg Banks 
125703cf6c9fSGreg Banks 	if (!pidx)
125803cf6c9fSGreg Banks 		return SEQ_START_TOKEN;
125903cf6c9fSGreg Banks 	return (pidx > serv->sv_nrpools ? NULL : &serv->sv_pools[pidx-1]);
126003cf6c9fSGreg Banks }
126103cf6c9fSGreg Banks 
126203cf6c9fSGreg Banks static void *svc_pool_stats_next(struct seq_file *m, void *p, loff_t *pos)
126303cf6c9fSGreg Banks {
126403cf6c9fSGreg Banks 	struct svc_pool *pool = p;
126503cf6c9fSGreg Banks 	struct svc_serv *serv = m->private;
126603cf6c9fSGreg Banks 
126703cf6c9fSGreg Banks 	dprintk("svc_pool_stats_next, *pos=%llu\n", *pos);
126803cf6c9fSGreg Banks 
126903cf6c9fSGreg Banks 	if (p == SEQ_START_TOKEN) {
127003cf6c9fSGreg Banks 		pool = &serv->sv_pools[0];
127103cf6c9fSGreg Banks 	} else {
127203cf6c9fSGreg Banks 		unsigned int pidx = (pool - &serv->sv_pools[0]);
127303cf6c9fSGreg Banks 		if (pidx < serv->sv_nrpools-1)
127403cf6c9fSGreg Banks 			pool = &serv->sv_pools[pidx+1];
127503cf6c9fSGreg Banks 		else
127603cf6c9fSGreg Banks 			pool = NULL;
127703cf6c9fSGreg Banks 	}
127803cf6c9fSGreg Banks 	++*pos;
127903cf6c9fSGreg Banks 	return pool;
128003cf6c9fSGreg Banks }
128103cf6c9fSGreg Banks 
128203cf6c9fSGreg Banks static void svc_pool_stats_stop(struct seq_file *m, void *p)
128303cf6c9fSGreg Banks {
128403cf6c9fSGreg Banks }
128503cf6c9fSGreg Banks 
128603cf6c9fSGreg Banks static int svc_pool_stats_show(struct seq_file *m, void *p)
128703cf6c9fSGreg Banks {
128803cf6c9fSGreg Banks 	struct svc_pool *pool = p;
128903cf6c9fSGreg Banks 
129003cf6c9fSGreg Banks 	if (p == SEQ_START_TOKEN) {
129178c210efSJ. Bruce Fields 		seq_puts(m, "# pool packets-arrived sockets-enqueued threads-woken threads-timedout\n");
129203cf6c9fSGreg Banks 		return 0;
129303cf6c9fSGreg Banks 	}
129403cf6c9fSGreg Banks 
129578c210efSJ. Bruce Fields 	seq_printf(m, "%u %lu %lu %lu %lu\n",
129603cf6c9fSGreg Banks 		pool->sp_id,
129703cf6c9fSGreg Banks 		pool->sp_stats.packets,
129803cf6c9fSGreg Banks 		pool->sp_stats.sockets_queued,
129903cf6c9fSGreg Banks 		pool->sp_stats.threads_woken,
130003cf6c9fSGreg Banks 		pool->sp_stats.threads_timedout);
130103cf6c9fSGreg Banks 
130203cf6c9fSGreg Banks 	return 0;
130303cf6c9fSGreg Banks }
130403cf6c9fSGreg Banks 
130503cf6c9fSGreg Banks static const struct seq_operations svc_pool_stats_seq_ops = {
130603cf6c9fSGreg Banks 	.start	= svc_pool_stats_start,
130703cf6c9fSGreg Banks 	.next	= svc_pool_stats_next,
130803cf6c9fSGreg Banks 	.stop	= svc_pool_stats_stop,
130903cf6c9fSGreg Banks 	.show	= svc_pool_stats_show,
131003cf6c9fSGreg Banks };
131103cf6c9fSGreg Banks 
131203cf6c9fSGreg Banks int svc_pool_stats_open(struct svc_serv *serv, struct file *file)
131303cf6c9fSGreg Banks {
131403cf6c9fSGreg Banks 	int err;
131503cf6c9fSGreg Banks 
131603cf6c9fSGreg Banks 	err = seq_open(file, &svc_pool_stats_seq_ops);
131703cf6c9fSGreg Banks 	if (!err)
131803cf6c9fSGreg Banks 		((struct seq_file *) file->private_data)->private = serv;
131903cf6c9fSGreg Banks 	return err;
132003cf6c9fSGreg Banks }
132103cf6c9fSGreg Banks EXPORT_SYMBOL(svc_pool_stats_open);
132203cf6c9fSGreg Banks 
132303cf6c9fSGreg Banks /*----------------------------------------------------------------------------*/
1324