blob: b974571126fe90fce45ff6edc522366c463d5d8d [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * linux/net/sunrpc/svc.c
3 *
4 * High-level RPC service routines
5 *
6 * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
Greg Banksbfd24162006-10-02 02:18:01 -07007 *
8 * Multiple threads pools and NUMAisation
9 * Copyright (c) 2006 Silicon Graphics, Inc.
10 * by Greg Banks <gnb@melbourne.sgi.com>
Linus Torvalds1da177e2005-04-16 15:20:36 -070011 */
12
13#include <linux/linkage.h>
14#include <linux/sched.h>
15#include <linux/errno.h>
16#include <linux/net.h>
17#include <linux/in.h>
18#include <linux/mm.h>
Greg Banksa7455442006-10-02 02:17:59 -070019#include <linux/interrupt.h>
20#include <linux/module.h>
Jeff Layton9867d762008-06-10 08:40:38 -040021#include <linux/kthread.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090022#include <linux/slab.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070023
24#include <linux/sunrpc/types.h>
25#include <linux/sunrpc/xdr.h>
26#include <linux/sunrpc/stats.h>
27#include <linux/sunrpc/svcsock.h>
28#include <linux/sunrpc/clnt.h>
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -040029#include <linux/sunrpc/bc_xprt.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070030
31#define RPCDBG_FACILITY RPCDBG_SVCDSP
Linus Torvalds1da177e2005-04-16 15:20:36 -070032
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +040033static void svc_unregister(const struct svc_serv *serv, struct net *net);
Chuck Lever7252d572008-08-18 19:34:08 -040034
Greg Banks42a7fc42007-03-06 01:42:23 -080035#define svc_serv_is_pooled(serv) ((serv)->sv_function)
36
Linus Torvalds1da177e2005-04-16 15:20:36 -070037/*
Greg Banksbfd24162006-10-02 02:18:01 -070038 * Mode for mapping cpus to pools.
39 */
40enum {
Greg Banks42a7fc42007-03-06 01:42:23 -080041 SVC_POOL_AUTO = -1, /* choose one of the others */
Greg Banksbfd24162006-10-02 02:18:01 -070042 SVC_POOL_GLOBAL, /* no mapping, just a single global pool
43 * (legacy & UP mode) */
44 SVC_POOL_PERCPU, /* one pool per cpu */
45 SVC_POOL_PERNODE /* one pool per numa node */
46};
Greg Banks42a7fc42007-03-06 01:42:23 -080047#define SVC_POOL_DEFAULT SVC_POOL_GLOBAL
Greg Banksbfd24162006-10-02 02:18:01 -070048
49/*
50 * Structure for mapping cpus to pools and vice versa.
51 * Setup once during sunrpc initialisation.
52 */
53static struct svc_pool_map {
Greg Banks42a7fc42007-03-06 01:42:23 -080054 int count; /* How many svc_servs use us */
Greg Banksbfd24162006-10-02 02:18:01 -070055 int mode; /* Note: int not enum to avoid
56 * warnings about "enumeration value
57 * not handled in switch" */
58 unsigned int npools;
59 unsigned int *pool_to; /* maps pool id to cpu or node */
60 unsigned int *to_pool; /* maps cpu or node to pool id */
61} svc_pool_map = {
Greg Banks42a7fc42007-03-06 01:42:23 -080062 .count = 0,
63 .mode = SVC_POOL_DEFAULT
Greg Banksbfd24162006-10-02 02:18:01 -070064};
Greg Banks42a7fc42007-03-06 01:42:23 -080065static DEFINE_MUTEX(svc_pool_map_mutex);/* protects svc_pool_map.count only */
Greg Banksbfd24162006-10-02 02:18:01 -070066
Greg Banks42a7fc42007-03-06 01:42:23 -080067static int
68param_set_pool_mode(const char *val, struct kernel_param *kp)
69{
70 int *ip = (int *)kp->arg;
71 struct svc_pool_map *m = &svc_pool_map;
72 int err;
73
74 mutex_lock(&svc_pool_map_mutex);
75
76 err = -EBUSY;
77 if (m->count)
78 goto out;
79
80 err = 0;
81 if (!strncmp(val, "auto", 4))
82 *ip = SVC_POOL_AUTO;
83 else if (!strncmp(val, "global", 6))
84 *ip = SVC_POOL_GLOBAL;
85 else if (!strncmp(val, "percpu", 6))
86 *ip = SVC_POOL_PERCPU;
87 else if (!strncmp(val, "pernode", 7))
88 *ip = SVC_POOL_PERNODE;
89 else
90 err = -EINVAL;
91
92out:
93 mutex_unlock(&svc_pool_map_mutex);
94 return err;
95}
96
97static int
98param_get_pool_mode(char *buf, struct kernel_param *kp)
99{
100 int *ip = (int *)kp->arg;
101
102 switch (*ip)
103 {
104 case SVC_POOL_AUTO:
105 return strlcpy(buf, "auto", 20);
106 case SVC_POOL_GLOBAL:
107 return strlcpy(buf, "global", 20);
108 case SVC_POOL_PERCPU:
109 return strlcpy(buf, "percpu", 20);
110 case SVC_POOL_PERNODE:
111 return strlcpy(buf, "pernode", 20);
112 default:
113 return sprintf(buf, "%d", *ip);
114 }
115}
116
117module_param_call(pool_mode, param_set_pool_mode, param_get_pool_mode,
118 &svc_pool_map.mode, 0644);
Greg Banksbfd24162006-10-02 02:18:01 -0700119
120/*
121 * Detect best pool mapping mode heuristically,
122 * according to the machine's topology.
123 */
124static int
125svc_pool_map_choose_mode(void)
126{
127 unsigned int node;
128
Christoph Lameter62bc62a2009-06-16 15:32:15 -0700129 if (nr_online_nodes > 1) {
Greg Banksbfd24162006-10-02 02:18:01 -0700130 /*
131 * Actually have multiple NUMA nodes,
132 * so split pools on NUMA node boundaries
133 */
134 return SVC_POOL_PERNODE;
135 }
136
H Hartley Sweeten72c33682010-03-05 13:42:43 -0800137 node = first_online_node;
Greg Banksbfd24162006-10-02 02:18:01 -0700138 if (nr_cpus_node(node) > 2) {
139 /*
140 * Non-trivial SMP, or CONFIG_NUMA on
141 * non-NUMA hardware, e.g. with a generic
142 * x86_64 kernel on Xeons. In this case we
143 * want to divide the pools on cpu boundaries.
144 */
145 return SVC_POOL_PERCPU;
146 }
147
148 /* default: one global pool */
149 return SVC_POOL_GLOBAL;
150}
151
152/*
153 * Allocate the to_pool[] and pool_to[] arrays.
154 * Returns 0 on success or an errno.
155 */
156static int
157svc_pool_map_alloc_arrays(struct svc_pool_map *m, unsigned int maxpools)
158{
159 m->to_pool = kcalloc(maxpools, sizeof(unsigned int), GFP_KERNEL);
160 if (!m->to_pool)
161 goto fail;
162 m->pool_to = kcalloc(maxpools, sizeof(unsigned int), GFP_KERNEL);
163 if (!m->pool_to)
164 goto fail_free;
165
166 return 0;
167
168fail_free:
169 kfree(m->to_pool);
J. Bruce Fields61c85042011-12-22 18:22:49 -0700170 m->to_pool = NULL;
Greg Banksbfd24162006-10-02 02:18:01 -0700171fail:
172 return -ENOMEM;
173}
174
175/*
176 * Initialise the pool map for SVC_POOL_PERCPU mode.
177 * Returns number of pools or <0 on error.
178 */
179static int
180svc_pool_map_init_percpu(struct svc_pool_map *m)
181{
Christoph Lameter53b8a312007-02-20 13:57:51 -0800182 unsigned int maxpools = nr_cpu_ids;
Greg Banksbfd24162006-10-02 02:18:01 -0700183 unsigned int pidx = 0;
184 unsigned int cpu;
185 int err;
186
187 err = svc_pool_map_alloc_arrays(m, maxpools);
188 if (err)
189 return err;
190
191 for_each_online_cpu(cpu) {
192 BUG_ON(pidx > maxpools);
193 m->to_pool[cpu] = pidx;
194 m->pool_to[pidx] = cpu;
195 pidx++;
196 }
197 /* cpus brought online later all get mapped to pool0, sorry */
198
199 return pidx;
200};
201
202
203/*
204 * Initialise the pool map for SVC_POOL_PERNODE mode.
205 * Returns number of pools or <0 on error.
206 */
207static int
208svc_pool_map_init_pernode(struct svc_pool_map *m)
209{
Christoph Lameter74c7aa82007-02-20 13:57:51 -0800210 unsigned int maxpools = nr_node_ids;
Greg Banksbfd24162006-10-02 02:18:01 -0700211 unsigned int pidx = 0;
212 unsigned int node;
213 int err;
214
215 err = svc_pool_map_alloc_arrays(m, maxpools);
216 if (err)
217 return err;
218
219 for_each_node_with_cpus(node) {
220 /* some architectures (e.g. SN2) have cpuless nodes */
221 BUG_ON(pidx > maxpools);
222 m->to_pool[node] = pidx;
223 m->pool_to[pidx] = node;
224 pidx++;
225 }
226 /* nodes brought online later all get mapped to pool0, sorry */
227
228 return pidx;
229}
230
231
232/*
Greg Banks42a7fc42007-03-06 01:42:23 -0800233 * Add a reference to the global map of cpus to pools (and
234 * vice versa). Initialise the map if we're the first user.
235 * Returns the number of pools.
Greg Banksbfd24162006-10-02 02:18:01 -0700236 */
237static unsigned int
Greg Banks42a7fc42007-03-06 01:42:23 -0800238svc_pool_map_get(void)
Greg Banksbfd24162006-10-02 02:18:01 -0700239{
240 struct svc_pool_map *m = &svc_pool_map;
241 int npools = -1;
242
Greg Banks42a7fc42007-03-06 01:42:23 -0800243 mutex_lock(&svc_pool_map_mutex);
Greg Banksbfd24162006-10-02 02:18:01 -0700244
Greg Banks42a7fc42007-03-06 01:42:23 -0800245 if (m->count++) {
246 mutex_unlock(&svc_pool_map_mutex);
247 return m->npools;
248 }
249
250 if (m->mode == SVC_POOL_AUTO)
251 m->mode = svc_pool_map_choose_mode();
Greg Banksbfd24162006-10-02 02:18:01 -0700252
253 switch (m->mode) {
254 case SVC_POOL_PERCPU:
255 npools = svc_pool_map_init_percpu(m);
256 break;
257 case SVC_POOL_PERNODE:
258 npools = svc_pool_map_init_pernode(m);
259 break;
260 }
261
262 if (npools < 0) {
263 /* default, or memory allocation failure */
264 npools = 1;
265 m->mode = SVC_POOL_GLOBAL;
266 }
267 m->npools = npools;
268
Greg Banks42a7fc42007-03-06 01:42:23 -0800269 mutex_unlock(&svc_pool_map_mutex);
Greg Banksbfd24162006-10-02 02:18:01 -0700270 return m->npools;
271}
272
Greg Banks42a7fc42007-03-06 01:42:23 -0800273
274/*
275 * Drop a reference to the global map of cpus to pools.
276 * When the last reference is dropped, the map data is
277 * freed; this allows the sysadmin to change the pool
278 * mode using the pool_mode module option without
279 * rebooting or re-loading sunrpc.ko.
280 */
281static void
282svc_pool_map_put(void)
283{
284 struct svc_pool_map *m = &svc_pool_map;
285
286 mutex_lock(&svc_pool_map_mutex);
287
288 if (!--m->count) {
Greg Banks42a7fc42007-03-06 01:42:23 -0800289 kfree(m->to_pool);
J. Bruce Fields61c85042011-12-22 18:22:49 -0700290 m->to_pool = NULL;
Greg Banks42a7fc42007-03-06 01:42:23 -0800291 kfree(m->pool_to);
J. Bruce Fields61c85042011-12-22 18:22:49 -0700292 m->pool_to = NULL;
Greg Banks42a7fc42007-03-06 01:42:23 -0800293 m->npools = 0;
294 }
295
296 mutex_unlock(&svc_pool_map_mutex);
297}
298
299
Eric Dumazet11fd1652011-07-28 20:04:09 +0200300static int svc_pool_map_get_node(unsigned int pidx)
301{
302 const struct svc_pool_map *m = &svc_pool_map;
303
304 if (m->count) {
305 if (m->mode == SVC_POOL_PERCPU)
306 return cpu_to_node(m->pool_to[pidx]);
307 if (m->mode == SVC_POOL_PERNODE)
308 return m->pool_to[pidx];
309 }
310 return NUMA_NO_NODE;
311}
Greg Banksbfd24162006-10-02 02:18:01 -0700312/*
Jeff Layton9867d762008-06-10 08:40:38 -0400313 * Set the given thread's cpus_allowed mask so that it
Greg Banksbfd24162006-10-02 02:18:01 -0700314 * will only run on cpus in the given pool.
Greg Banksbfd24162006-10-02 02:18:01 -0700315 */
Jeff Layton9867d762008-06-10 08:40:38 -0400316static inline void
317svc_pool_map_set_cpumask(struct task_struct *task, unsigned int pidx)
Greg Banksbfd24162006-10-02 02:18:01 -0700318{
319 struct svc_pool_map *m = &svc_pool_map;
Jeff Layton9867d762008-06-10 08:40:38 -0400320 unsigned int node = m->pool_to[pidx];
Greg Banksbfd24162006-10-02 02:18:01 -0700321
322 /*
323 * The caller checks for sv_nrpools > 1, which
Greg Banks42a7fc42007-03-06 01:42:23 -0800324 * implies that we've been initialized.
Greg Banksbfd24162006-10-02 02:18:01 -0700325 */
Weston Andros Adamson1bd58aa2012-10-23 10:43:38 -0400326 WARN_ON_ONCE(m->count == 0);
327 if (m->count == 0)
328 return;
Greg Banksbfd24162006-10-02 02:18:01 -0700329
Jeff Layton9867d762008-06-10 08:40:38 -0400330 switch (m->mode) {
Greg Banksbfd24162006-10-02 02:18:01 -0700331 case SVC_POOL_PERCPU:
Mike Travisc5f59f02008-04-04 18:11:10 -0700332 {
Rusty Russellaa85ea52009-03-30 22:05:15 -0600333 set_cpus_allowed_ptr(task, cpumask_of(node));
Jeff Layton9867d762008-06-10 08:40:38 -0400334 break;
Mike Travisc5f59f02008-04-04 18:11:10 -0700335 }
Greg Banksbfd24162006-10-02 02:18:01 -0700336 case SVC_POOL_PERNODE:
Mike Travisc5f59f02008-04-04 18:11:10 -0700337 {
Rusty Russella70f7302009-03-13 14:49:46 +1030338 set_cpus_allowed_ptr(task, cpumask_of_node(node));
Jeff Layton9867d762008-06-10 08:40:38 -0400339 break;
Greg Banksbfd24162006-10-02 02:18:01 -0700340 }
Mike Travisc5f59f02008-04-04 18:11:10 -0700341 }
Greg Banksbfd24162006-10-02 02:18:01 -0700342}
343
344/*
345 * Use the mapping mode to choose a pool for a given CPU.
346 * Used when enqueueing an incoming RPC. Always returns
347 * a non-NULL pool pointer.
348 */
349struct svc_pool *
350svc_pool_for_cpu(struct svc_serv *serv, int cpu)
351{
352 struct svc_pool_map *m = &svc_pool_map;
353 unsigned int pidx = 0;
354
355 /*
Greg Banks42a7fc42007-03-06 01:42:23 -0800356 * An uninitialised map happens in a pure client when
Greg Banksbfd24162006-10-02 02:18:01 -0700357 * lockd is brought up, so silently treat it the
358 * same as SVC_POOL_GLOBAL.
359 */
Greg Banks42a7fc42007-03-06 01:42:23 -0800360 if (svc_serv_is_pooled(serv)) {
361 switch (m->mode) {
362 case SVC_POOL_PERCPU:
363 pidx = m->to_pool[cpu];
364 break;
365 case SVC_POOL_PERNODE:
366 pidx = m->to_pool[cpu_to_node(cpu)];
367 break;
368 }
Greg Banksbfd24162006-10-02 02:18:01 -0700369 }
370 return &serv->sv_pools[pidx % serv->sv_nrpools];
371}
372
Stanislav Kinsburskybb2224d2012-01-31 15:08:05 +0400373int svc_rpcb_setup(struct svc_serv *serv, struct net *net)
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300374{
375 int err;
376
Stanislav Kinsburskybee42f682012-01-13 14:02:56 +0400377 err = rpcb_create_local(net);
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300378 if (err)
379 return err;
380
381 /* Remove any stale portmap registrations */
Stanislav Kinsburskybee42f682012-01-13 14:02:56 +0400382 svc_unregister(serv, net);
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300383 return 0;
384}
Stanislav Kinsburskybb2224d2012-01-31 15:08:05 +0400385EXPORT_SYMBOL_GPL(svc_rpcb_setup);
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300386
Stanislav Kinsbursky5ecebb72012-01-13 14:03:04 +0400387void svc_rpcb_cleanup(struct svc_serv *serv, struct net *net)
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300388{
Stanislav Kinsbursky5ecebb72012-01-13 14:03:04 +0400389 svc_unregister(serv, net);
390 rpcb_put_local(net);
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300391}
Stanislav Kinsbursky16d05872011-10-25 14:17:28 +0300392EXPORT_SYMBOL_GPL(svc_rpcb_cleanup);
Stanislav Kinsburskyd9908562011-10-25 14:16:58 +0300393
394static int svc_uses_rpcbind(struct svc_serv *serv)
395{
396 struct svc_program *progp;
397 unsigned int i;
398
399 for (progp = serv->sv_program; progp; progp = progp->pg_next) {
400 for (i = 0; i < progp->pg_nvers; i++) {
401 if (progp->pg_vers[i] == NULL)
402 continue;
403 if (progp->pg_vers[i]->vs_hidden == 0)
404 return 1;
405 }
406 }
407
408 return 0;
409}
Greg Banksbfd24162006-10-02 02:18:01 -0700410
Stanislav Kinsbursky9793f7c2012-05-02 16:08:38 +0400411int svc_bind(struct svc_serv *serv, struct net *net)
412{
413 if (!svc_uses_rpcbind(serv))
414 return 0;
415 return svc_rpcb_setup(serv, net);
416}
417EXPORT_SYMBOL_GPL(svc_bind);
418
Greg Banksbfd24162006-10-02 02:18:01 -0700419/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700420 * Create an RPC service
421 */
Greg Banksa7455442006-10-02 02:17:59 -0700422static struct svc_serv *
423__svc_create(struct svc_program *prog, unsigned int bufsize, int npools,
Stanislav Kinsbursky5ecebb72012-01-13 14:03:04 +0400424 void (*shutdown)(struct svc_serv *serv, struct net *net))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700425{
426 struct svc_serv *serv;
Chuck Leverea339d42007-10-26 13:32:56 -0400427 unsigned int vers;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700428 unsigned int xdrsize;
Greg Banks3262c812006-10-02 02:17:58 -0700429 unsigned int i;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700430
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700431 if (!(serv = kzalloc(sizeof(*serv), GFP_KERNEL)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700432 return NULL;
Andreas Gruenbacher9ba02632005-06-22 17:16:24 +0000433 serv->sv_name = prog->pg_name;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700434 serv->sv_program = prog;
435 serv->sv_nrthreads = 1;
436 serv->sv_stats = prog->pg_stats;
NeilBrownc6b0a9f2006-10-06 00:44:05 -0700437 if (bufsize > RPCSVC_MAXPAYLOAD)
438 bufsize = RPCSVC_MAXPAYLOAD;
439 serv->sv_max_payload = bufsize? bufsize : 4096;
440 serv->sv_max_mesg = roundup(serv->sv_max_payload + PAGE_SIZE, PAGE_SIZE);
NeilBrownbc591cc2006-10-02 02:17:44 -0700441 serv->sv_shutdown = shutdown;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700442 xdrsize = 0;
Andreas Gruenbacher9ba02632005-06-22 17:16:24 +0000443 while (prog) {
444 prog->pg_lovers = prog->pg_nvers-1;
445 for (vers=0; vers<prog->pg_nvers ; vers++)
446 if (prog->pg_vers[vers]) {
447 prog->pg_hivers = vers;
448 if (prog->pg_lovers > vers)
449 prog->pg_lovers = vers;
450 if (prog->pg_vers[vers]->vs_xdrsize > xdrsize)
451 xdrsize = prog->pg_vers[vers]->vs_xdrsize;
452 }
453 prog = prog->pg_next;
454 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700455 serv->sv_xdrsize = xdrsize;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700456 INIT_LIST_HEAD(&serv->sv_tempsocks);
457 INIT_LIST_HEAD(&serv->sv_permsocks);
Greg Banks36bdfc82006-10-02 02:17:54 -0700458 init_timer(&serv->sv_temptimer);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700459 spin_lock_init(&serv->sv_lock);
460
Greg Banksa7455442006-10-02 02:17:59 -0700461 serv->sv_nrpools = npools;
Greg Banks3262c812006-10-02 02:17:58 -0700462 serv->sv_pools =
Robert P. J. Daycd861282006-12-13 00:34:52 -0800463 kcalloc(serv->sv_nrpools, sizeof(struct svc_pool),
Greg Banks3262c812006-10-02 02:17:58 -0700464 GFP_KERNEL);
465 if (!serv->sv_pools) {
466 kfree(serv);
467 return NULL;
468 }
469
470 for (i = 0; i < serv->sv_nrpools; i++) {
471 struct svc_pool *pool = &serv->sv_pools[i];
472
Chuck Lever46121cf2007-01-31 12:14:08 -0500473 dprintk("svc: initialising pool %u for %s\n",
Greg Banks3262c812006-10-02 02:17:58 -0700474 i, serv->sv_name);
475
476 pool->sp_id = i;
477 INIT_LIST_HEAD(&pool->sp_threads);
478 INIT_LIST_HEAD(&pool->sp_sockets);
Greg Banksa7455442006-10-02 02:17:59 -0700479 INIT_LIST_HEAD(&pool->sp_all_threads);
Greg Banks3262c812006-10-02 02:17:58 -0700480 spin_lock_init(&pool->sp_lock);
481 }
482
Stanislav Kinsbursky9793f7c2012-05-02 16:08:38 +0400483 if (svc_uses_rpcbind(serv) && (!serv->sv_shutdown))
484 serv->sv_shutdown = svc_rpcb_cleanup;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700485
486 return serv;
487}
488
Greg Banksa7455442006-10-02 02:17:59 -0700489struct svc_serv *
490svc_create(struct svc_program *prog, unsigned int bufsize,
Stanislav Kinsbursky5ecebb72012-01-13 14:03:04 +0400491 void (*shutdown)(struct svc_serv *serv, struct net *net))
Greg Banksa7455442006-10-02 02:17:59 -0700492{
Chuck Lever49a90722009-03-18 20:46:29 -0400493 return __svc_create(prog, bufsize, /*npools*/1, shutdown);
Greg Banksa7455442006-10-02 02:17:59 -0700494}
Trond Myklebust24c37672008-12-23 16:30:12 -0500495EXPORT_SYMBOL_GPL(svc_create);
Greg Banksa7455442006-10-02 02:17:59 -0700496
497struct svc_serv *
498svc_create_pooled(struct svc_program *prog, unsigned int bufsize,
Stanislav Kinsbursky5ecebb72012-01-13 14:03:04 +0400499 void (*shutdown)(struct svc_serv *serv, struct net *net),
Jeff Laytona75c5d02008-06-10 08:40:39 -0400500 svc_thread_fn func, struct module *mod)
Greg Banksa7455442006-10-02 02:17:59 -0700501{
502 struct svc_serv *serv;
Greg Banks42a7fc42007-03-06 01:42:23 -0800503 unsigned int npools = svc_pool_map_get();
Greg Banksa7455442006-10-02 02:17:59 -0700504
Chuck Lever49a90722009-03-18 20:46:29 -0400505 serv = __svc_create(prog, bufsize, npools, shutdown);
Greg Banksa7455442006-10-02 02:17:59 -0700506
507 if (serv != NULL) {
508 serv->sv_function = func;
Greg Banksa7455442006-10-02 02:17:59 -0700509 serv->sv_module = mod;
510 }
511
512 return serv;
513}
Trond Myklebust24c37672008-12-23 16:30:12 -0500514EXPORT_SYMBOL_GPL(svc_create_pooled);
Greg Banksa7455442006-10-02 02:17:59 -0700515
Stanislav Kinsbursky074d0f62012-01-31 14:09:25 +0400516void svc_shutdown_net(struct svc_serv *serv, struct net *net)
517{
Stanislav Kinsbursky074d0f62012-01-31 14:09:25 +0400518 svc_close_net(serv, net);
519
520 if (serv->sv_shutdown)
521 serv->sv_shutdown(serv, net);
522}
523EXPORT_SYMBOL_GPL(svc_shutdown_net);
524
Linus Torvalds1da177e2005-04-16 15:20:36 -0700525/*
Neil Brownbedbdd82008-06-10 08:40:35 -0400526 * Destroy an RPC service. Should be called with appropriate locking to
527 * protect the sv_nrthreads, sv_permsocks and sv_tempsocks.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700528 */
529void
530svc_destroy(struct svc_serv *serv)
531{
Chuck Lever46121cf2007-01-31 12:14:08 -0500532 dprintk("svc: svc_destroy(%s, %d)\n",
Linus Torvalds1da177e2005-04-16 15:20:36 -0700533 serv->sv_program->pg_name,
534 serv->sv_nrthreads);
535
536 if (serv->sv_nrthreads) {
537 if (--(serv->sv_nrthreads) != 0) {
538 svc_sock_update_bufs(serv);
539 return;
540 }
541 } else
542 printk("svc_destroy: no threads for serv=%p!\n", serv);
543
Greg Banks36bdfc82006-10-02 02:17:54 -0700544 del_timer_sync(&serv->sv_temptimer);
Stanislav Kinsbursky074d0f62012-01-31 14:09:25 +0400545
Stanislav Kinsbursky7b147f12012-01-31 14:09:17 +0400546 /*
547 * The last user is gone and thus all sockets have to be destroyed to
548 * the point. Check this.
549 */
550 BUG_ON(!list_empty(&serv->sv_permsocks));
551 BUG_ON(!list_empty(&serv->sv_tempsocks));
NeilBrowncda1fd42007-03-06 01:42:22 -0800552
Linus Torvalds1da177e2005-04-16 15:20:36 -0700553 cache_clean_deferred(serv);
554
Greg Banks42a7fc42007-03-06 01:42:23 -0800555 if (svc_serv_is_pooled(serv))
556 svc_pool_map_put();
557
Greg Banks3262c812006-10-02 02:17:58 -0700558 kfree(serv->sv_pools);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700559 kfree(serv);
560}
Trond Myklebust24c37672008-12-23 16:30:12 -0500561EXPORT_SYMBOL_GPL(svc_destroy);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700562
563/*
564 * Allocate an RPC server's buffer space.
565 * We allocate pages and place them in rq_argpages.
566 */
567static int
Eric Dumazet11fd1652011-07-28 20:04:09 +0200568svc_init_buffer(struct svc_rqst *rqstp, unsigned int size, int node)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700569{
Chuck Lever0dc220f2008-04-14 12:27:52 -0400570 unsigned int pages, arghi;
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800571
Andy Adamsonba176862010-01-26 21:24:04 -0500572 /* bc_xprt uses fore channel allocated buffers */
573 if (svc_is_backchannel(rqstp))
574 return 1;
575
NeilBrownc6b0a9f2006-10-06 00:44:05 -0700576 pages = size / PAGE_SIZE + 1; /* extra page as we hold both request and reply.
577 * We assume one is at most one page
578 */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700579 arghi = 0;
Weston Andros Adamsonb25cd052012-10-23 10:43:41 -0400580 WARN_ON_ONCE(pages > RPCSVC_MAXPAGES);
581 if (pages > RPCSVC_MAXPAGES)
582 pages = RPCSVC_MAXPAGES;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700583 while (pages) {
Eric Dumazet11fd1652011-07-28 20:04:09 +0200584 struct page *p = alloc_pages_node(node, GFP_KERNEL, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700585 if (!p)
586 break;
NeilBrown44524352006-10-04 02:15:46 -0700587 rqstp->rq_pages[arghi++] = p;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700588 pages--;
589 }
Chuck Lever0dc220f2008-04-14 12:27:52 -0400590 return pages == 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700591}
592
593/*
594 * Release an RPC server buffer
595 */
596static void
597svc_release_buffer(struct svc_rqst *rqstp)
598{
Chuck Lever50c8bb12008-04-14 12:27:45 -0400599 unsigned int i;
600
601 for (i = 0; i < ARRAY_SIZE(rqstp->rq_pages); i++)
NeilBrown44524352006-10-04 02:15:46 -0700602 if (rqstp->rq_pages[i])
603 put_page(rqstp->rq_pages[i]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700604}
605
Jeff Layton0113ab32008-01-29 10:30:54 -0500606struct svc_rqst *
Eric Dumazet11fd1652011-07-28 20:04:09 +0200607svc_prepare_thread(struct svc_serv *serv, struct svc_pool *pool, int node)
Jeff Layton0113ab32008-01-29 10:30:54 -0500608{
609 struct svc_rqst *rqstp;
610
Eric Dumazet11fd1652011-07-28 20:04:09 +0200611 rqstp = kzalloc_node(sizeof(*rqstp), GFP_KERNEL, node);
Jeff Layton0113ab32008-01-29 10:30:54 -0500612 if (!rqstp)
613 goto out_enomem;
614
615 init_waitqueue_head(&rqstp->rq_wait);
616
617 serv->sv_nrthreads++;
618 spin_lock_bh(&pool->sp_lock);
619 pool->sp_nrthreads++;
620 list_add(&rqstp->rq_all, &pool->sp_all_threads);
621 spin_unlock_bh(&pool->sp_lock);
622 rqstp->rq_server = serv;
623 rqstp->rq_pool = pool;
624
Eric Dumazet11fd1652011-07-28 20:04:09 +0200625 rqstp->rq_argp = kmalloc_node(serv->sv_xdrsize, GFP_KERNEL, node);
Jeff Layton0113ab32008-01-29 10:30:54 -0500626 if (!rqstp->rq_argp)
627 goto out_thread;
628
Eric Dumazet11fd1652011-07-28 20:04:09 +0200629 rqstp->rq_resp = kmalloc_node(serv->sv_xdrsize, GFP_KERNEL, node);
Jeff Layton0113ab32008-01-29 10:30:54 -0500630 if (!rqstp->rq_resp)
631 goto out_thread;
632
Eric Dumazet11fd1652011-07-28 20:04:09 +0200633 if (!svc_init_buffer(rqstp, serv->sv_max_mesg, node))
Jeff Layton0113ab32008-01-29 10:30:54 -0500634 goto out_thread;
635
636 return rqstp;
637out_thread:
638 svc_exit_thread(rqstp);
639out_enomem:
640 return ERR_PTR(-ENOMEM);
641}
Trond Myklebust24c37672008-12-23 16:30:12 -0500642EXPORT_SYMBOL_GPL(svc_prepare_thread);
Jeff Layton0113ab32008-01-29 10:30:54 -0500643
Linus Torvalds1da177e2005-04-16 15:20:36 -0700644/*
Greg Banksa7455442006-10-02 02:17:59 -0700645 * Choose a pool in which to create a new thread, for svc_set_num_threads
646 */
647static inline struct svc_pool *
648choose_pool(struct svc_serv *serv, struct svc_pool *pool, unsigned int *state)
649{
650 if (pool != NULL)
651 return pool;
652
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800653 return &serv->sv_pools[(*state)++ % serv->sv_nrpools];
Greg Banksa7455442006-10-02 02:17:59 -0700654}
655
656/*
657 * Choose a thread to kill, for svc_set_num_threads
658 */
659static inline struct task_struct *
660choose_victim(struct svc_serv *serv, struct svc_pool *pool, unsigned int *state)
661{
662 unsigned int i;
663 struct task_struct *task = NULL;
664
665 if (pool != NULL) {
666 spin_lock_bh(&pool->sp_lock);
667 } else {
668 /* choose a pool in round-robin fashion */
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800669 for (i = 0; i < serv->sv_nrpools; i++) {
670 pool = &serv->sv_pools[--(*state) % serv->sv_nrpools];
Greg Banksa7455442006-10-02 02:17:59 -0700671 spin_lock_bh(&pool->sp_lock);
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800672 if (!list_empty(&pool->sp_all_threads))
673 goto found_pool;
Greg Banksa7455442006-10-02 02:17:59 -0700674 spin_unlock_bh(&pool->sp_lock);
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800675 }
Greg Banksa7455442006-10-02 02:17:59 -0700676 return NULL;
677 }
678
679found_pool:
680 if (!list_empty(&pool->sp_all_threads)) {
681 struct svc_rqst *rqstp;
682
683 /*
684 * Remove from the pool->sp_all_threads list
685 * so we don't try to kill it again.
686 */
687 rqstp = list_entry(pool->sp_all_threads.next, struct svc_rqst, rq_all);
688 list_del_init(&rqstp->rq_all);
689 task = rqstp->rq_task;
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -0800690 }
Greg Banksa7455442006-10-02 02:17:59 -0700691 spin_unlock_bh(&pool->sp_lock);
692
693 return task;
694}
695
696/*
697 * Create or destroy enough new threads to make the number
698 * of threads the given number. If `pool' is non-NULL, applies
699 * only to threads in that pool, otherwise round-robins between
J. Bruce Fields94cf3172011-12-01 17:51:21 -0500700 * all pools. Caller must ensure that mutual exclusion between this and
701 * server startup or shutdown.
Greg Banksa7455442006-10-02 02:17:59 -0700702 *
703 * Destroying threads relies on the service threads filling in
704 * rqstp->rq_task, which only the nfs ones do. Assumes the serv
705 * has been created using svc_create_pooled().
706 *
707 * Based on code that used to be in nfsd_svc() but tweaked
708 * to be pool-aware.
709 */
710int
711svc_set_num_threads(struct svc_serv *serv, struct svc_pool *pool, int nrservs)
712{
Jeff Layton9867d762008-06-10 08:40:38 -0400713 struct svc_rqst *rqstp;
714 struct task_struct *task;
715 struct svc_pool *chosen_pool;
Greg Banksa7455442006-10-02 02:17:59 -0700716 int error = 0;
717 unsigned int state = serv->sv_nrthreads-1;
Eric Dumazet11fd1652011-07-28 20:04:09 +0200718 int node;
Greg Banksa7455442006-10-02 02:17:59 -0700719
720 if (pool == NULL) {
721 /* The -1 assumes caller has done a svc_get() */
722 nrservs -= (serv->sv_nrthreads-1);
723 } else {
724 spin_lock_bh(&pool->sp_lock);
725 nrservs -= pool->sp_nrthreads;
726 spin_unlock_bh(&pool->sp_lock);
727 }
728
729 /* create new threads */
730 while (nrservs > 0) {
731 nrservs--;
Jeff Layton9867d762008-06-10 08:40:38 -0400732 chosen_pool = choose_pool(serv, pool, &state);
733
Eric Dumazet11fd1652011-07-28 20:04:09 +0200734 node = svc_pool_map_get_node(chosen_pool->sp_id);
735 rqstp = svc_prepare_thread(serv, chosen_pool, node);
Jeff Layton9867d762008-06-10 08:40:38 -0400736 if (IS_ERR(rqstp)) {
737 error = PTR_ERR(rqstp);
Greg Banksa7455442006-10-02 02:17:59 -0700738 break;
739 }
Jeff Layton9867d762008-06-10 08:40:38 -0400740
741 __module_get(serv->sv_module);
Eric Dumazet11fd1652011-07-28 20:04:09 +0200742 task = kthread_create_on_node(serv->sv_function, rqstp,
Kees Cookf1701682013-07-03 15:04:58 -0700743 node, "%s", serv->sv_name);
Jeff Layton9867d762008-06-10 08:40:38 -0400744 if (IS_ERR(task)) {
745 error = PTR_ERR(task);
746 module_put(serv->sv_module);
747 svc_exit_thread(rqstp);
748 break;
749 }
750
751 rqstp->rq_task = task;
752 if (serv->sv_nrpools > 1)
753 svc_pool_map_set_cpumask(task, chosen_pool->sp_id);
754
755 svc_sock_update_bufs(serv);
756 wake_up_process(task);
Greg Banksa7455442006-10-02 02:17:59 -0700757 }
758 /* destroy old threads */
759 while (nrservs < 0 &&
Jeff Layton9867d762008-06-10 08:40:38 -0400760 (task = choose_victim(serv, pool, &state)) != NULL) {
Jeff Laytona75c5d02008-06-10 08:40:39 -0400761 send_sig(SIGINT, task, 1);
Greg Banksa7455442006-10-02 02:17:59 -0700762 nrservs++;
763 }
764
765 return error;
766}
Trond Myklebust24c37672008-12-23 16:30:12 -0500767EXPORT_SYMBOL_GPL(svc_set_num_threads);
Greg Banksa7455442006-10-02 02:17:59 -0700768
769/*
Neil Brownbedbdd82008-06-10 08:40:35 -0400770 * Called from a server thread as it's exiting. Caller must hold the BKL or
771 * the "service mutex", whichever is appropriate for the service.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700772 */
773void
774svc_exit_thread(struct svc_rqst *rqstp)
775{
776 struct svc_serv *serv = rqstp->rq_server;
Greg Banks3262c812006-10-02 02:17:58 -0700777 struct svc_pool *pool = rqstp->rq_pool;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700778
779 svc_release_buffer(rqstp);
Jesper Juhla51482b2005-11-08 09:41:34 -0800780 kfree(rqstp->rq_resp);
781 kfree(rqstp->rq_argp);
782 kfree(rqstp->rq_auth_data);
Greg Banks3262c812006-10-02 02:17:58 -0700783
784 spin_lock_bh(&pool->sp_lock);
785 pool->sp_nrthreads--;
Greg Banksa7455442006-10-02 02:17:59 -0700786 list_del(&rqstp->rq_all);
Greg Banks3262c812006-10-02 02:17:58 -0700787 spin_unlock_bh(&pool->sp_lock);
788
Linus Torvalds1da177e2005-04-16 15:20:36 -0700789 kfree(rqstp);
790
791 /* Release the server */
792 if (serv)
793 svc_destroy(serv);
794}
Trond Myklebust24c37672008-12-23 16:30:12 -0500795EXPORT_SYMBOL_GPL(svc_exit_thread);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700796
797/*
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500798 * Register an "inet" protocol family netid with the local
799 * rpcbind daemon via an rpcbind v4 SET request.
Chuck Levera26cfad2008-08-18 19:34:16 -0400800 *
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500801 * No netconfig infrastructure is available in the kernel, so
802 * we map IP_ protocol numbers to netids by hand.
Chuck Levera26cfad2008-08-18 19:34:16 -0400803 *
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500804 * Returns zero on success; a negative errno value is returned
805 * if any error occurs.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700806 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400807static int __svc_rpcb_register4(struct net *net, const u32 program,
808 const u32 version,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500809 const unsigned short protocol,
810 const unsigned short port)
Chuck Levera26cfad2008-08-18 19:34:16 -0400811{
Chuck Levercadc0fa2009-03-18 20:47:36 -0400812 const struct sockaddr_in sin = {
Chuck Levera26cfad2008-08-18 19:34:16 -0400813 .sin_family = AF_INET,
814 .sin_addr.s_addr = htonl(INADDR_ANY),
815 .sin_port = htons(port),
816 };
Chuck Levercadc0fa2009-03-18 20:47:36 -0400817 const char *netid;
818 int error;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500819
820 switch (protocol) {
821 case IPPROTO_UDP:
822 netid = RPCBIND_NETID_UDP;
823 break;
824 case IPPROTO_TCP:
825 netid = RPCBIND_NETID_TCP;
826 break;
827 default:
Chuck Leverba5c35e2009-03-18 20:46:59 -0400828 return -ENOPROTOOPT;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500829 }
830
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400831 error = rpcb_v4_register(net, program, version,
Chuck Levercadc0fa2009-03-18 20:47:36 -0400832 (const struct sockaddr *)&sin, netid);
833
834 /*
835 * User space didn't support rpcbind v4, so retry this
836 * registration request with the legacy rpcbind v2 protocol.
837 */
838 if (error == -EPROTONOSUPPORT)
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400839 error = rpcb_register(net, program, version, protocol, port);
Chuck Levercadc0fa2009-03-18 20:47:36 -0400840
841 return error;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500842}
843
Eric Dumazetdfd56b82011-12-10 09:48:31 +0000844#if IS_ENABLED(CONFIG_IPV6)
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500845/*
846 * Register an "inet6" protocol family netid with the local
847 * rpcbind daemon via an rpcbind v4 SET request.
848 *
849 * No netconfig infrastructure is available in the kernel, so
850 * we map IP_ protocol numbers to netids by hand.
851 *
852 * Returns zero on success; a negative errno value is returned
853 * if any error occurs.
854 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400855static int __svc_rpcb_register6(struct net *net, const u32 program,
856 const u32 version,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500857 const unsigned short protocol,
858 const unsigned short port)
859{
Chuck Levercadc0fa2009-03-18 20:47:36 -0400860 const struct sockaddr_in6 sin6 = {
Chuck Levera26cfad2008-08-18 19:34:16 -0400861 .sin6_family = AF_INET6,
862 .sin6_addr = IN6ADDR_ANY_INIT,
863 .sin6_port = htons(port),
864 };
Chuck Levercadc0fa2009-03-18 20:47:36 -0400865 const char *netid;
866 int error;
Chuck Levera26cfad2008-08-18 19:34:16 -0400867
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500868 switch (protocol) {
869 case IPPROTO_UDP:
870 netid = RPCBIND_NETID_UDP6;
871 break;
872 case IPPROTO_TCP:
873 netid = RPCBIND_NETID_TCP6;
874 break;
875 default:
Chuck Leverba5c35e2009-03-18 20:46:59 -0400876 return -ENOPROTOOPT;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500877 }
878
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400879 error = rpcb_v4_register(net, program, version,
Chuck Levercadc0fa2009-03-18 20:47:36 -0400880 (const struct sockaddr *)&sin6, netid);
881
882 /*
883 * User space didn't support rpcbind version 4, so we won't
884 * use a PF_INET6 listener.
885 */
886 if (error == -EPROTONOSUPPORT)
887 error = -EAFNOSUPPORT;
888
889 return error;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500890}
Eric Dumazetdfd56b82011-12-10 09:48:31 +0000891#endif /* IS_ENABLED(CONFIG_IPV6) */
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500892
893/*
894 * Register a kernel RPC service via rpcbind version 4.
895 *
896 * Returns zero on success; a negative errno value is returned
897 * if any error occurs.
898 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400899static int __svc_register(struct net *net, const char *progname,
Chuck Lever363f7242009-03-18 20:47:44 -0400900 const u32 program, const u32 version,
Chuck Lever4b62e582009-03-18 20:46:06 -0400901 const int family,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500902 const unsigned short protocol,
903 const unsigned short port)
904{
Chuck Lever363f7242009-03-18 20:47:44 -0400905 int error = -EAFNOSUPPORT;
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500906
Chuck Levera26cfad2008-08-18 19:34:16 -0400907 switch (family) {
Chuck Lever4b62e582009-03-18 20:46:06 -0400908 case PF_INET:
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400909 error = __svc_rpcb_register4(net, program, version,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500910 protocol, port);
Chuck Levercadc0fa2009-03-18 20:47:36 -0400911 break;
Eric Dumazetdfd56b82011-12-10 09:48:31 +0000912#if IS_ENABLED(CONFIG_IPV6)
Chuck Lever4b62e582009-03-18 20:46:06 -0400913 case PF_INET6:
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400914 error = __svc_rpcb_register6(net, program, version,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500915 protocol, port);
Eric Dumazetdfd56b82011-12-10 09:48:31 +0000916#endif
Chuck Levera26cfad2008-08-18 19:34:16 -0400917 }
918
Chuck Lever363f7242009-03-18 20:47:44 -0400919 if (error < 0)
920 printk(KERN_WARNING "svc: failed to register %sv%u RPC "
921 "service (errno %d).\n", progname, version, -error);
922 return error;
Chuck Levera26cfad2008-08-18 19:34:16 -0400923}
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500924
Chuck Levera26cfad2008-08-18 19:34:16 -0400925/**
926 * svc_register - register an RPC service with the local portmapper
927 * @serv: svc_serv struct for the service to register
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400928 * @net: net namespace for the service to register
Chuck Lever4b62e582009-03-18 20:46:06 -0400929 * @family: protocol family of service's listener socket
Chuck Levera26cfad2008-08-18 19:34:16 -0400930 * @proto: transport protocol number to advertise
931 * @port: port to advertise
932 *
Chuck Lever4b62e582009-03-18 20:46:06 -0400933 * Service is registered for any address in the passed-in protocol family
Chuck Levera26cfad2008-08-18 19:34:16 -0400934 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400935int svc_register(const struct svc_serv *serv, struct net *net,
936 const int family, const unsigned short proto,
937 const unsigned short port)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700938{
939 struct svc_program *progp;
Chuck Leverea339d42007-10-26 13:32:56 -0400940 unsigned int i;
Chuck Lever14aeb212008-08-18 19:34:00 -0400941 int error = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700942
Weston Andros Adamson0af39502012-10-23 10:43:45 -0400943 WARN_ON_ONCE(proto == 0 && port == 0);
944 if (proto == 0 && port == 0)
945 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700946
Olaf Kirchbc5fea42006-10-04 02:16:05 -0700947 for (progp = serv->sv_program; progp; progp = progp->pg_next) {
948 for (i = 0; i < progp->pg_nvers; i++) {
949 if (progp->pg_vers[i] == NULL)
950 continue;
951
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500952 dprintk("svc: svc_register(%sv%d, %s, %u, %u)%s\n",
Olaf Kirchbc5fea42006-10-04 02:16:05 -0700953 progp->pg_name,
Chuck Lever2c7eb0b2008-09-15 16:27:23 -0500954 i,
Olaf Kirchbc5fea42006-10-04 02:16:05 -0700955 proto == IPPROTO_UDP? "udp" : "tcp",
956 port,
Chuck Lever4b62e582009-03-18 20:46:06 -0400957 family,
Olaf Kirchbc5fea42006-10-04 02:16:05 -0700958 progp->pg_vers[i]->vs_hidden?
959 " (but not telling portmap)" : "");
960
961 if (progp->pg_vers[i]->vs_hidden)
962 continue;
963
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400964 error = __svc_register(net, progp->pg_name, progp->pg_prog,
Chuck Lever363f7242009-03-18 20:47:44 -0400965 i, family, proto, port);
Olaf Kirchbc5fea42006-10-04 02:16:05 -0700966 if (error < 0)
967 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700968 }
969 }
970
Chuck Lever7252d572008-08-18 19:34:08 -0400971 return error;
972}
973
Chuck Leverd5a86202009-03-18 20:47:29 -0400974/*
975 * If user space is running rpcbind, it should take the v4 UNSET
976 * and clear everything for this [program, version]. If user space
977 * is running portmap, it will reject the v4 UNSET, but won't have
978 * any "inet6" entries anyway. So a PMAP_UNSET should be sufficient
979 * in this case to clear all existing entries for [program, version].
980 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400981static void __svc_unregister(struct net *net, const u32 program, const u32 version,
Chuck Leverf6fb3f62008-09-25 11:56:57 -0400982 const char *progname)
983{
984 int error;
985
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400986 error = rpcb_v4_register(net, program, version, NULL, "");
Chuck Leverd5a86202009-03-18 20:47:29 -0400987
988 /*
989 * User space didn't support rpcbind v4, so retry this
990 * request with the legacy rpcbind v2 protocol.
991 */
992 if (error == -EPROTONOSUPPORT)
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +0400993 error = rpcb_register(net, program, version, 0, 0);
Chuck Leverd5a86202009-03-18 20:47:29 -0400994
Chuck Leverf6fb3f62008-09-25 11:56:57 -0400995 dprintk("svc: %s(%sv%u), error %d\n",
996 __func__, progname, version, error);
997}
998
Chuck Lever7252d572008-08-18 19:34:08 -0400999/*
Chuck Leverf6fb3f62008-09-25 11:56:57 -04001000 * All netids, bind addresses and ports registered for [program, version]
1001 * are removed from the local rpcbind database (if the service is not
1002 * hidden) to make way for a new instance of the service.
Chuck Lever7252d572008-08-18 19:34:08 -04001003 *
Chuck Leverf6fb3f62008-09-25 11:56:57 -04001004 * The result of unregistration is reported via dprintk for those who want
1005 * verification of the result, but is otherwise not important.
Chuck Lever7252d572008-08-18 19:34:08 -04001006 */
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +04001007static void svc_unregister(const struct svc_serv *serv, struct net *net)
Chuck Lever7252d572008-08-18 19:34:08 -04001008{
1009 struct svc_program *progp;
1010 unsigned long flags;
1011 unsigned int i;
Chuck Lever7252d572008-08-18 19:34:08 -04001012
1013 clear_thread_flag(TIF_SIGPENDING);
1014
1015 for (progp = serv->sv_program; progp; progp = progp->pg_next) {
1016 for (i = 0; i < progp->pg_nvers; i++) {
1017 if (progp->pg_vers[i] == NULL)
1018 continue;
1019 if (progp->pg_vers[i]->vs_hidden)
1020 continue;
1021
Chuck Lever7402ab192011-05-09 15:22:55 -04001022 dprintk("svc: attempting to unregister %sv%u\n",
1023 progp->pg_name, i);
Stanislav Kinsbursky5247fab2012-01-13 14:02:48 +04001024 __svc_unregister(net, progp->pg_prog, i, progp->pg_name);
Chuck Lever7252d572008-08-18 19:34:08 -04001025 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001026 }
1027
Chuck Lever7252d572008-08-18 19:34:08 -04001028 spin_lock_irqsave(&current->sighand->siglock, flags);
1029 recalc_sigpending();
1030 spin_unlock_irqrestore(&current->sighand->siglock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001031}
1032
1033/*
J. Bruce Fields7032a3d2012-10-09 18:33:38 -04001034 * dprintk the given error with the address of the client that caused it.
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001035 */
J. Bruce Fields624ab462013-01-08 16:22:15 -05001036#ifdef RPC_DEBUG
Joe Perchesb9075fa2011-10-31 17:11:33 -07001037static __printf(2, 3)
Joe Perchese87cc472012-05-13 21:56:26 +00001038void svc_printk(struct svc_rqst *rqstp, const char *fmt, ...)
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001039{
Joe Perchese87cc472012-05-13 21:56:26 +00001040 struct va_format vaf;
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001041 va_list args;
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001042 char buf[RPC_MAX_ADDRBUFLEN];
1043
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001044 va_start(args, fmt);
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001045
Joe Perchese87cc472012-05-13 21:56:26 +00001046 vaf.fmt = fmt;
1047 vaf.va = &args;
1048
J. Bruce Fields7032a3d2012-10-09 18:33:38 -04001049 dprintk("svc: %s: %pV", svc_print_addr(rqstp, buf, sizeof(buf)), &vaf);
Joe Perchese87cc472012-05-13 21:56:26 +00001050
1051 va_end(args);
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001052}
J. Bruce Fields624ab462013-01-08 16:22:15 -05001053#else
1054static __printf(2,3) void svc_printk(struct svc_rqst *rqstp, const char *fmt, ...) {}
1055#endif
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001056
1057/*
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001058 * Common routine for processing the RPC request.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001059 */
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001060static int
1061svc_process_common(struct svc_rqst *rqstp, struct kvec *argv, struct kvec *resv)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001062{
1063 struct svc_program *progp;
1064 struct svc_version *versp = NULL; /* compiler food */
1065 struct svc_procedure *procp = NULL;
NeilBrown6fb2b472006-10-02 02:17:50 -07001066 struct svc_serv *serv = rqstp->rq_server;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001067 kxdrproc_t xdr;
Alexey Dobriyand8ed0292006-09-26 22:29:38 -07001068 __be32 *statp;
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001069 u32 prog, vers, proc;
Alexey Dobriyand8ed0292006-09-26 22:29:38 -07001070 __be32 auth_stat, rpc_stat;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001071 int auth_res;
J.Bruce Fields8f8e05c2006-10-04 02:16:08 -07001072 __be32 *reply_statp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001073
1074 rpc_stat = rpc_success;
1075
1076 if (argv->iov_len < 6*4)
1077 goto err_short_len;
1078
J. Bruce Fields5c04c462006-06-30 01:56:19 -07001079 /* Will be turned off only in gss privacy case: */
Jens Axboecf8208d2007-06-12 21:22:14 +02001080 rqstp->rq_splice_ok = 1;
Andy Adamson2f425872009-04-03 08:27:32 +03001081 /* Will be turned off only when NFSv4 Sessions are used */
1082 rqstp->rq_usedeferral = 1;
J. Bruce Fields9e701c62011-01-02 21:56:36 -05001083 rqstp->rq_dropme = false;
Tom Tuckere831fe62007-12-30 21:07:29 -06001084
1085 /* Setup reply header */
1086 rqstp->rq_xprt->xpt_ops->xpo_prep_reply_hdr(rqstp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001087
Linus Torvalds1da177e2005-04-16 15:20:36 -07001088 svc_putu32(resv, rqstp->rq_xid);
1089
Alexey Dobriyan76994312006-09-26 22:28:46 -07001090 vers = svc_getnl(argv);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001091
1092 /* First words of reply: */
Alexey Dobriyan76994312006-09-26 22:28:46 -07001093 svc_putnl(resv, 1); /* REPLY */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001094
Linus Torvalds1da177e2005-04-16 15:20:36 -07001095 if (vers != 2) /* RPC version number */
1096 goto err_bad_rpc;
1097
1098 /* Save position in case we later decide to reject: */
J.Bruce Fields8f8e05c2006-10-04 02:16:08 -07001099 reply_statp = resv->iov_base + resv->iov_len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001100
Alexey Dobriyan76994312006-09-26 22:28:46 -07001101 svc_putnl(resv, 0); /* ACCEPT */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001102
Alexey Dobriyan76994312006-09-26 22:28:46 -07001103 rqstp->rq_prog = prog = svc_getnl(argv); /* program number */
1104 rqstp->rq_vers = vers = svc_getnl(argv); /* version number */
1105 rqstp->rq_proc = proc = svc_getnl(argv); /* procedure number */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001106
1107 progp = serv->sv_program;
NeilBrown80d188a2005-11-07 01:00:27 -08001108
1109 for (progp = serv->sv_program; progp; progp = progp->pg_next)
1110 if (prog == progp->pg_prog)
1111 break;
1112
Linus Torvalds1da177e2005-04-16 15:20:36 -07001113 /*
1114 * Decode auth data, and add verifier to reply buffer.
1115 * We do this before anything else in order to get a decent
1116 * auth verifier.
1117 */
1118 auth_res = svc_authenticate(rqstp, &auth_stat);
1119 /* Also give the program a chance to reject this call: */
NeilBrown80d188a2005-11-07 01:00:27 -08001120 if (auth_res == SVC_OK && progp) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001121 auth_stat = rpc_autherr_badcred;
1122 auth_res = progp->pg_authenticate(rqstp);
1123 }
1124 switch (auth_res) {
1125 case SVC_OK:
1126 break;
1127 case SVC_GARBAGE:
Harshula Jayasuriyadd352102008-02-20 10:56:56 +11001128 goto err_garbage;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001129 case SVC_SYSERR:
1130 rpc_stat = rpc_system_err;
1131 goto err_bad;
1132 case SVC_DENIED:
1133 goto err_bad_auth;
NeilBrown1ebede82010-08-12 17:04:07 +10001134 case SVC_CLOSE:
1135 if (test_bit(XPT_TEMP, &rqstp->rq_xprt->xpt_flags))
1136 svc_close_xprt(rqstp->rq_xprt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001137 case SVC_DROP:
1138 goto dropit;
1139 case SVC_COMPLETE:
1140 goto sendit;
1141 }
NeilBrown80d188a2005-11-07 01:00:27 -08001142
Andreas Gruenbacher9ba02632005-06-22 17:16:24 +00001143 if (progp == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001144 goto err_bad_prog;
1145
1146 if (vers >= progp->pg_nvers ||
1147 !(versp = progp->pg_vers[vers]))
1148 goto err_bad_vers;
1149
1150 procp = versp->vs_proc + proc;
1151 if (proc >= versp->vs_nproc || !procp->pc_func)
1152 goto err_bad_proc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001153 rqstp->rq_procinfo = procp;
1154
1155 /* Syntactic check complete */
1156 serv->sv_stats->rpccnt++;
1157
1158 /* Build the reply header. */
1159 statp = resv->iov_base +resv->iov_len;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001160 svc_putnl(resv, RPC_SUCCESS);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001161
1162 /* Bump per-procedure stats counter */
1163 procp->pc_count++;
1164
1165 /* Initialize storage for argp and resp */
1166 memset(rqstp->rq_argp, 0, procp->pc_argsize);
1167 memset(rqstp->rq_resp, 0, procp->pc_ressize);
1168
YOSHIFUJI Hideakicca51722007-02-09 15:38:13 -08001169 /* un-reserve some of the out-queue now that we have a
Linus Torvalds1da177e2005-04-16 15:20:36 -07001170 * better idea of reply size
1171 */
1172 if (procp->pc_xdrressize)
Jeff Laytoncd123012007-05-09 02:34:50 -07001173 svc_reserve_auth(rqstp, procp->pc_xdrressize<<2);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001174
1175 /* Call the function that processes the request. */
1176 if (!versp->vs_dispatch) {
1177 /* Decode arguments */
1178 xdr = procp->pc_decode;
1179 if (xdr && !xdr(rqstp, argv->iov_base, rqstp->rq_argp))
1180 goto err_garbage;
1181
1182 *statp = procp->pc_func(rqstp, rqstp->rq_argp, rqstp->rq_resp);
1183
1184 /* Encode reply */
J. Bruce Fields9e701c62011-01-02 21:56:36 -05001185 if (rqstp->rq_dropme) {
NeilBrownd343fce2006-10-17 00:10:18 -07001186 if (procp->pc_release)
1187 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
1188 goto dropit;
1189 }
Joe Perchesf64f9e72009-11-29 16:55:45 -08001190 if (*statp == rpc_success &&
1191 (xdr = procp->pc_encode) &&
1192 !xdr(rqstp, resv->iov_base+resv->iov_len, rqstp->rq_resp)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001193 dprintk("svc: failed to encode reply\n");
1194 /* serv->sv_stats->rpcsystemerr++; */
1195 *statp = rpc_system_err;
1196 }
1197 } else {
1198 dprintk("svc: calling dispatcher\n");
1199 if (!versp->vs_dispatch(rqstp, statp)) {
1200 /* Release reply info */
1201 if (procp->pc_release)
1202 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
1203 goto dropit;
1204 }
1205 }
1206
1207 /* Check RPC status result */
1208 if (*statp != rpc_success)
1209 resv->iov_len = ((void*)statp) - resv->iov_base + 4;
1210
1211 /* Release reply info */
1212 if (procp->pc_release)
1213 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
1214
1215 if (procp->pc_encode == NULL)
1216 goto dropit;
1217
1218 sendit:
1219 if (svc_authorise(rqstp))
1220 goto dropit;
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001221 return 1; /* Caller can now send it */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001222
1223 dropit:
1224 svc_authorise(rqstp); /* doesn't hurt to call this twice */
1225 dprintk("svc: svc_process dropit\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07001226 return 0;
1227
1228err_short_len:
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001229 svc_printk(rqstp, "short len %Zd, dropping request\n",
1230 argv->iov_len);
NeilBrown34e9a632007-01-29 13:19:52 -08001231
Linus Torvalds1da177e2005-04-16 15:20:36 -07001232 goto dropit; /* drop request */
1233
Linus Torvalds1da177e2005-04-16 15:20:36 -07001234err_bad_rpc:
1235 serv->sv_stats->rpcbadfmt++;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001236 svc_putnl(resv, 1); /* REJECT */
1237 svc_putnl(resv, 0); /* RPC_MISMATCH */
1238 svc_putnl(resv, 2); /* Only RPCv2 supported */
1239 svc_putnl(resv, 2);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001240 goto sendit;
1241
1242err_bad_auth:
1243 dprintk("svc: authentication failed (%d)\n", ntohl(auth_stat));
1244 serv->sv_stats->rpcbadauth++;
1245 /* Restore write pointer to location of accept status: */
J.Bruce Fields8f8e05c2006-10-04 02:16:08 -07001246 xdr_ressize_check(rqstp, reply_statp);
Alexey Dobriyan76994312006-09-26 22:28:46 -07001247 svc_putnl(resv, 1); /* REJECT */
1248 svc_putnl(resv, 1); /* AUTH_ERROR */
1249 svc_putnl(resv, ntohl(auth_stat)); /* status */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001250 goto sendit;
1251
1252err_bad_prog:
Andreas Gruenbacher9ba02632005-06-22 17:16:24 +00001253 dprintk("svc: unknown program %d\n", prog);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001254 serv->sv_stats->rpcbadfmt++;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001255 svc_putnl(resv, RPC_PROG_UNAVAIL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001256 goto sendit;
1257
1258err_bad_vers:
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001259 svc_printk(rqstp, "unknown version (%d for prog %d, %s)\n",
NeilBrown34e9a632007-01-29 13:19:52 -08001260 vers, prog, progp->pg_name);
1261
Linus Torvalds1da177e2005-04-16 15:20:36 -07001262 serv->sv_stats->rpcbadfmt++;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001263 svc_putnl(resv, RPC_PROG_MISMATCH);
1264 svc_putnl(resv, progp->pg_lovers);
1265 svc_putnl(resv, progp->pg_hivers);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001266 goto sendit;
1267
1268err_bad_proc:
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001269 svc_printk(rqstp, "unknown procedure (%d)\n", proc);
NeilBrown34e9a632007-01-29 13:19:52 -08001270
Linus Torvalds1da177e2005-04-16 15:20:36 -07001271 serv->sv_stats->rpcbadfmt++;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001272 svc_putnl(resv, RPC_PROC_UNAVAIL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001273 goto sendit;
1274
1275err_garbage:
Dr. David Alan Gilbert354ecbb2007-08-25 16:09:27 +01001276 svc_printk(rqstp, "failed to decode args\n");
NeilBrown34e9a632007-01-29 13:19:52 -08001277
Linus Torvalds1da177e2005-04-16 15:20:36 -07001278 rpc_stat = rpc_garbage_args;
1279err_bad:
1280 serv->sv_stats->rpcbadfmt++;
Alexey Dobriyan76994312006-09-26 22:28:46 -07001281 svc_putnl(resv, ntohl(rpc_stat));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001282 goto sendit;
1283}
Trond Myklebust24c37672008-12-23 16:30:12 -05001284EXPORT_SYMBOL_GPL(svc_process);
Greg Banks7adae482006-10-04 02:15:47 -07001285
1286/*
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001287 * Process the RPC request.
1288 */
1289int
1290svc_process(struct svc_rqst *rqstp)
1291{
1292 struct kvec *argv = &rqstp->rq_arg.head[0];
1293 struct kvec *resv = &rqstp->rq_res.head[0];
1294 struct svc_serv *serv = rqstp->rq_server;
1295 u32 dir;
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001296
1297 /*
1298 * Setup response xdr_buf.
1299 * Initially it has just one page
1300 */
J. Bruce Fieldsafc59402012-12-10 18:01:37 -05001301 rqstp->rq_next_page = &rqstp->rq_respages[1];
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001302 resv->iov_base = page_address(rqstp->rq_respages[0]);
1303 resv->iov_len = 0;
1304 rqstp->rq_res.pages = rqstp->rq_respages + 1;
1305 rqstp->rq_res.len = 0;
1306 rqstp->rq_res.page_base = 0;
1307 rqstp->rq_res.page_len = 0;
1308 rqstp->rq_res.buflen = PAGE_SIZE;
1309 rqstp->rq_res.tail[0].iov_base = NULL;
1310 rqstp->rq_res.tail[0].iov_len = 0;
1311
1312 rqstp->rq_xid = svc_getu32(argv);
1313
1314 dir = svc_getnl(argv);
1315 if (dir != 0) {
1316 /* direction != CALL */
1317 svc_printk(rqstp, "bad direction %d, dropping request\n", dir);
1318 serv->sv_stats->rpcbadfmt++;
1319 svc_drop(rqstp);
1320 return 0;
1321 }
1322
Andy Adamson4b5b3ba2011-01-06 02:04:24 +00001323 /* Returns 1 for send, 0 for drop */
1324 if (svc_process_common(rqstp, argv, resv))
1325 return svc_send(rqstp);
1326 else {
1327 svc_drop(rqstp);
1328 return 0;
1329 }
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001330}
1331
Trond Myklebust9e00abc2011-07-13 19:20:49 -04001332#if defined(CONFIG_SUNRPC_BACKCHANNEL)
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -04001333/*
1334 * Process a backchannel RPC request that arrived over an existing
1335 * outbound connection
1336 */
1337int
1338bc_svc_process(struct svc_serv *serv, struct rpc_rqst *req,
1339 struct svc_rqst *rqstp)
1340{
1341 struct kvec *argv = &rqstp->rq_arg.head[0];
1342 struct kvec *resv = &rqstp->rq_res.head[0];
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -04001343
1344 /* Build the svc_rqst used by the common processing routine */
Andy Adamson4a19de02011-01-06 02:04:35 +00001345 rqstp->rq_xprt = serv->sv_bc_xprt;
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -04001346 rqstp->rq_xid = req->rq_xid;
1347 rqstp->rq_prot = req->rq_xprt->prot;
1348 rqstp->rq_server = serv;
1349
1350 rqstp->rq_addrlen = sizeof(req->rq_xprt->addr);
1351 memcpy(&rqstp->rq_addr, &req->rq_xprt->addr, rqstp->rq_addrlen);
1352 memcpy(&rqstp->rq_arg, &req->rq_rcv_buf, sizeof(rqstp->rq_arg));
1353 memcpy(&rqstp->rq_res, &req->rq_snd_buf, sizeof(rqstp->rq_res));
1354
1355 /* reset result send buffer "put" position */
1356 resv->iov_len = 0;
1357
1358 if (rqstp->rq_prot != IPPROTO_TCP) {
1359 printk(KERN_ERR "No support for Non-TCP transports!\n");
1360 BUG();
1361 }
1362
1363 /*
1364 * Skip the next two words because they've already been
1365 * processed in the trasport
1366 */
1367 svc_getu32(argv); /* XID */
1368 svc_getnl(argv); /* CALLDIR */
1369
Andy Adamson4b5b3ba2011-01-06 02:04:24 +00001370 /* Returns 1 for send, 0 for drop */
1371 if (svc_process_common(rqstp, argv, resv)) {
1372 memcpy(&req->rq_snd_buf, &rqstp->rq_res,
1373 sizeof(req->rq_snd_buf));
1374 return bc_send(req);
1375 } else {
Trond Myklebustb3b02ae2012-05-31 15:26:38 -04001376 /* drop request */
1377 xprt_free_bc_request(req);
Andy Adamson4b5b3ba2011-01-06 02:04:24 +00001378 return 0;
1379 }
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -04001380}
Trond Myklebust0d961aa2011-07-13 19:24:15 -04001381EXPORT_SYMBOL_GPL(bc_svc_process);
Trond Myklebust9e00abc2011-07-13 19:20:49 -04001382#endif /* CONFIG_SUNRPC_BACKCHANNEL */
Ricardo Labiaga4d6bbb62009-04-01 09:23:07 -04001383
Ricardo Labiaga1cad7ea2009-04-01 09:23:06 -04001384/*
Greg Banks7adae482006-10-04 02:15:47 -07001385 * Return (transport-specific) limit on the rpc payload.
1386 */
1387u32 svc_max_payload(const struct svc_rqst *rqstp)
1388{
Tom Tucker49023152007-12-30 21:07:21 -06001389 u32 max = rqstp->rq_xprt->xpt_class->xcl_max_payload;
Greg Banks7adae482006-10-04 02:15:47 -07001390
NeilBrownc6b0a9f2006-10-06 00:44:05 -07001391 if (rqstp->rq_server->sv_max_payload < max)
1392 max = rqstp->rq_server->sv_max_payload;
Greg Banks7adae482006-10-04 02:15:47 -07001393 return max;
1394}
1395EXPORT_SYMBOL_GPL(svc_max_payload);