]> git.proxmox.com Git - mirror_ubuntu-artful-kernel.git/blame - net/sunrpc/svc.c
SUNRPC: Move exported symbol definitions after function declaration part 2
[mirror_ubuntu-artful-kernel.git] / net / sunrpc / svc.c
CommitLineData
1da177e4
LT
1/*
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>
bfd24160
GB
7 *
8 * Multiple threads pools and NUMAisation
9 * Copyright (c) 2006 Silicon Graphics, Inc.
10 * by Greg Banks <gnb@melbourne.sgi.com>
1da177e4
LT
11 */
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>
a7455442
GB
19#include <linux/interrupt.h>
20#include <linux/module.h>
df3825c5 21#include <linux/sched.h>
1da177e4
LT
22
23#include <linux/sunrpc/types.h>
24#include <linux/sunrpc/xdr.h>
25#include <linux/sunrpc/stats.h>
26#include <linux/sunrpc/svcsock.h>
27#include <linux/sunrpc/clnt.h>
28
29#define RPCDBG_FACILITY RPCDBG_SVCDSP
1da177e4 30
42a7fc4a
GB
31#define svc_serv_is_pooled(serv) ((serv)->sv_function)
32
bfd24160
GB
33/*
34 * Mode for mapping cpus to pools.
35 */
36enum {
42a7fc4a 37 SVC_POOL_AUTO = -1, /* choose one of the others */
bfd24160
GB
38 SVC_POOL_GLOBAL, /* no mapping, just a single global pool
39 * (legacy & UP mode) */
40 SVC_POOL_PERCPU, /* one pool per cpu */
41 SVC_POOL_PERNODE /* one pool per numa node */
42};
42a7fc4a 43#define SVC_POOL_DEFAULT SVC_POOL_GLOBAL
bfd24160
GB
44
45/*
46 * Structure for mapping cpus to pools and vice versa.
47 * Setup once during sunrpc initialisation.
48 */
49static struct svc_pool_map {
42a7fc4a 50 int count; /* How many svc_servs use us */
bfd24160
GB
51 int mode; /* Note: int not enum to avoid
52 * warnings about "enumeration value
53 * not handled in switch" */
54 unsigned int npools;
55 unsigned int *pool_to; /* maps pool id to cpu or node */
56 unsigned int *to_pool; /* maps cpu or node to pool id */
57} svc_pool_map = {
42a7fc4a
GB
58 .count = 0,
59 .mode = SVC_POOL_DEFAULT
bfd24160 60};
42a7fc4a
GB
61static DEFINE_MUTEX(svc_pool_map_mutex);/* protects svc_pool_map.count only */
62
63static int
64param_set_pool_mode(const char *val, struct kernel_param *kp)
65{
66 int *ip = (int *)kp->arg;
67 struct svc_pool_map *m = &svc_pool_map;
68 int err;
69
70 mutex_lock(&svc_pool_map_mutex);
71
72 err = -EBUSY;
73 if (m->count)
74 goto out;
75
76 err = 0;
77 if (!strncmp(val, "auto", 4))
78 *ip = SVC_POOL_AUTO;
79 else if (!strncmp(val, "global", 6))
80 *ip = SVC_POOL_GLOBAL;
81 else if (!strncmp(val, "percpu", 6))
82 *ip = SVC_POOL_PERCPU;
83 else if (!strncmp(val, "pernode", 7))
84 *ip = SVC_POOL_PERNODE;
85 else
86 err = -EINVAL;
87
88out:
89 mutex_unlock(&svc_pool_map_mutex);
90 return err;
91}
92
93static int
94param_get_pool_mode(char *buf, struct kernel_param *kp)
95{
96 int *ip = (int *)kp->arg;
97
98 switch (*ip)
99 {
100 case SVC_POOL_AUTO:
101 return strlcpy(buf, "auto", 20);
102 case SVC_POOL_GLOBAL:
103 return strlcpy(buf, "global", 20);
104 case SVC_POOL_PERCPU:
105 return strlcpy(buf, "percpu", 20);
106 case SVC_POOL_PERNODE:
107 return strlcpy(buf, "pernode", 20);
108 default:
109 return sprintf(buf, "%d", *ip);
110 }
111}
bfd24160 112
42a7fc4a
GB
113module_param_call(pool_mode, param_set_pool_mode, param_get_pool_mode,
114 &svc_pool_map.mode, 0644);
bfd24160
GB
115
116/*
117 * Detect best pool mapping mode heuristically,
118 * according to the machine's topology.
119 */
120static int
121svc_pool_map_choose_mode(void)
122{
123 unsigned int node;
124
125 if (num_online_nodes() > 1) {
126 /*
127 * Actually have multiple NUMA nodes,
128 * so split pools on NUMA node boundaries
129 */
130 return SVC_POOL_PERNODE;
131 }
132
133 node = any_online_node(node_online_map);
134 if (nr_cpus_node(node) > 2) {
135 /*
136 * Non-trivial SMP, or CONFIG_NUMA on
137 * non-NUMA hardware, e.g. with a generic
138 * x86_64 kernel on Xeons. In this case we
139 * want to divide the pools on cpu boundaries.
140 */
141 return SVC_POOL_PERCPU;
142 }
143
144 /* default: one global pool */
145 return SVC_POOL_GLOBAL;
146}
147
148/*
149 * Allocate the to_pool[] and pool_to[] arrays.
150 * Returns 0 on success or an errno.
151 */
152static int
153svc_pool_map_alloc_arrays(struct svc_pool_map *m, unsigned int maxpools)
154{
155 m->to_pool = kcalloc(maxpools, sizeof(unsigned int), GFP_KERNEL);
156 if (!m->to_pool)
157 goto fail;
158 m->pool_to = kcalloc(maxpools, sizeof(unsigned int), GFP_KERNEL);
159 if (!m->pool_to)
160 goto fail_free;
161
162 return 0;
163
164fail_free:
165 kfree(m->to_pool);
166fail:
167 return -ENOMEM;
168}
169
170/*
171 * Initialise the pool map for SVC_POOL_PERCPU mode.
172 * Returns number of pools or <0 on error.
173 */
174static int
175svc_pool_map_init_percpu(struct svc_pool_map *m)
176{
53b8a315 177 unsigned int maxpools = nr_cpu_ids;
bfd24160
GB
178 unsigned int pidx = 0;
179 unsigned int cpu;
180 int err;
181
182 err = svc_pool_map_alloc_arrays(m, maxpools);
183 if (err)
184 return err;
185
186 for_each_online_cpu(cpu) {
187 BUG_ON(pidx > maxpools);
188 m->to_pool[cpu] = pidx;
189 m->pool_to[pidx] = cpu;
190 pidx++;
191 }
192 /* cpus brought online later all get mapped to pool0, sorry */
193
194 return pidx;
195};
196
197
198/*
199 * Initialise the pool map for SVC_POOL_PERNODE mode.
200 * Returns number of pools or <0 on error.
201 */
202static int
203svc_pool_map_init_pernode(struct svc_pool_map *m)
204{
74c7aa8b 205 unsigned int maxpools = nr_node_ids;
bfd24160
GB
206 unsigned int pidx = 0;
207 unsigned int node;
208 int err;
209
210 err = svc_pool_map_alloc_arrays(m, maxpools);
211 if (err)
212 return err;
213
214 for_each_node_with_cpus(node) {
215 /* some architectures (e.g. SN2) have cpuless nodes */
216 BUG_ON(pidx > maxpools);
217 m->to_pool[node] = pidx;
218 m->pool_to[pidx] = node;
219 pidx++;
220 }
221 /* nodes brought online later all get mapped to pool0, sorry */
222
223 return pidx;
224}
225
226
227/*
42a7fc4a
GB
228 * Add a reference to the global map of cpus to pools (and
229 * vice versa). Initialise the map if we're the first user.
230 * Returns the number of pools.
bfd24160
GB
231 */
232static unsigned int
42a7fc4a 233svc_pool_map_get(void)
bfd24160
GB
234{
235 struct svc_pool_map *m = &svc_pool_map;
236 int npools = -1;
237
42a7fc4a
GB
238 mutex_lock(&svc_pool_map_mutex);
239
240 if (m->count++) {
241 mutex_unlock(&svc_pool_map_mutex);
bfd24160 242 return m->npools;
42a7fc4a 243 }
bfd24160 244
42a7fc4a
GB
245 if (m->mode == SVC_POOL_AUTO)
246 m->mode = svc_pool_map_choose_mode();
bfd24160
GB
247
248 switch (m->mode) {
249 case SVC_POOL_PERCPU:
250 npools = svc_pool_map_init_percpu(m);
251 break;
252 case SVC_POOL_PERNODE:
253 npools = svc_pool_map_init_pernode(m);
254 break;
255 }
256
257 if (npools < 0) {
258 /* default, or memory allocation failure */
259 npools = 1;
260 m->mode = SVC_POOL_GLOBAL;
261 }
262 m->npools = npools;
263
42a7fc4a 264 mutex_unlock(&svc_pool_map_mutex);
bfd24160
GB
265 return m->npools;
266}
267
42a7fc4a
GB
268
269/*
270 * Drop a reference to the global map of cpus to pools.
271 * When the last reference is dropped, the map data is
272 * freed; this allows the sysadmin to change the pool
273 * mode using the pool_mode module option without
274 * rebooting or re-loading sunrpc.ko.
275 */
276static void
277svc_pool_map_put(void)
278{
279 struct svc_pool_map *m = &svc_pool_map;
280
281 mutex_lock(&svc_pool_map_mutex);
282
283 if (!--m->count) {
284 m->mode = SVC_POOL_DEFAULT;
285 kfree(m->to_pool);
286 kfree(m->pool_to);
287 m->npools = 0;
288 }
289
290 mutex_unlock(&svc_pool_map_mutex);
291}
292
293
bfd24160
GB
294/*
295 * Set the current thread's cpus_allowed mask so that it
296 * will only run on cpus in the given pool.
297 *
298 * Returns 1 and fills in oldmask iff a cpumask was applied.
299 */
300static inline int
301svc_pool_map_set_cpumask(unsigned int pidx, cpumask_t *oldmask)
302{
303 struct svc_pool_map *m = &svc_pool_map;
304 unsigned int node; /* or cpu */
305
306 /*
307 * The caller checks for sv_nrpools > 1, which
42a7fc4a 308 * implies that we've been initialized.
bfd24160 309 */
42a7fc4a 310 BUG_ON(m->count == 0);
bfd24160
GB
311
312 switch (m->mode)
313 {
314 default:
315 return 0;
316 case SVC_POOL_PERCPU:
317 node = m->pool_to[pidx];
318 *oldmask = current->cpus_allowed;
319 set_cpus_allowed(current, cpumask_of_cpu(node));
320 return 1;
321 case SVC_POOL_PERNODE:
322 node = m->pool_to[pidx];
323 *oldmask = current->cpus_allowed;
324 set_cpus_allowed(current, node_to_cpumask(node));
325 return 1;
326 }
327}
328
329/*
330 * Use the mapping mode to choose a pool for a given CPU.
331 * Used when enqueueing an incoming RPC. Always returns
332 * a non-NULL pool pointer.
333 */
334struct svc_pool *
335svc_pool_for_cpu(struct svc_serv *serv, int cpu)
336{
337 struct svc_pool_map *m = &svc_pool_map;
338 unsigned int pidx = 0;
339
340 /*
42a7fc4a 341 * An uninitialised map happens in a pure client when
bfd24160
GB
342 * lockd is brought up, so silently treat it the
343 * same as SVC_POOL_GLOBAL.
344 */
42a7fc4a
GB
345 if (svc_serv_is_pooled(serv)) {
346 switch (m->mode) {
347 case SVC_POOL_PERCPU:
348 pidx = m->to_pool[cpu];
349 break;
350 case SVC_POOL_PERNODE:
351 pidx = m->to_pool[cpu_to_node(cpu)];
352 break;
353 }
bfd24160
GB
354 }
355 return &serv->sv_pools[pidx % serv->sv_nrpools];
356}
357
358
1da177e4
LT
359/*
360 * Create an RPC service
361 */
a7455442
GB
362static struct svc_serv *
363__svc_create(struct svc_program *prog, unsigned int bufsize, int npools,
bc591ccf 364 void (*shutdown)(struct svc_serv *serv))
1da177e4
LT
365{
366 struct svc_serv *serv;
367 int vers;
368 unsigned int xdrsize;
3262c816 369 unsigned int i;
1da177e4 370
0da974f4 371 if (!(serv = kzalloc(sizeof(*serv), GFP_KERNEL)))
1da177e4 372 return NULL;
9ba02638 373 serv->sv_name = prog->pg_name;
1da177e4
LT
374 serv->sv_program = prog;
375 serv->sv_nrthreads = 1;
376 serv->sv_stats = prog->pg_stats;
c6b0a9f8
N
377 if (bufsize > RPCSVC_MAXPAYLOAD)
378 bufsize = RPCSVC_MAXPAYLOAD;
379 serv->sv_max_payload = bufsize? bufsize : 4096;
380 serv->sv_max_mesg = roundup(serv->sv_max_payload + PAGE_SIZE, PAGE_SIZE);
bc591ccf 381 serv->sv_shutdown = shutdown;
1da177e4 382 xdrsize = 0;
9ba02638
AG
383 while (prog) {
384 prog->pg_lovers = prog->pg_nvers-1;
385 for (vers=0; vers<prog->pg_nvers ; vers++)
386 if (prog->pg_vers[vers]) {
387 prog->pg_hivers = vers;
388 if (prog->pg_lovers > vers)
389 prog->pg_lovers = vers;
390 if (prog->pg_vers[vers]->vs_xdrsize > xdrsize)
391 xdrsize = prog->pg_vers[vers]->vs_xdrsize;
392 }
393 prog = prog->pg_next;
394 }
1da177e4 395 serv->sv_xdrsize = xdrsize;
1da177e4
LT
396 INIT_LIST_HEAD(&serv->sv_tempsocks);
397 INIT_LIST_HEAD(&serv->sv_permsocks);
36bdfc8b 398 init_timer(&serv->sv_temptimer);
1da177e4
LT
399 spin_lock_init(&serv->sv_lock);
400
a7455442 401 serv->sv_nrpools = npools;
3262c816 402 serv->sv_pools =
cd861280 403 kcalloc(serv->sv_nrpools, sizeof(struct svc_pool),
3262c816
GB
404 GFP_KERNEL);
405 if (!serv->sv_pools) {
406 kfree(serv);
407 return NULL;
408 }
409
410 for (i = 0; i < serv->sv_nrpools; i++) {
411 struct svc_pool *pool = &serv->sv_pools[i];
412
46121cf7 413 dprintk("svc: initialising pool %u for %s\n",
3262c816
GB
414 i, serv->sv_name);
415
416 pool->sp_id = i;
417 INIT_LIST_HEAD(&pool->sp_threads);
418 INIT_LIST_HEAD(&pool->sp_sockets);
a7455442 419 INIT_LIST_HEAD(&pool->sp_all_threads);
3262c816
GB
420 spin_lock_init(&pool->sp_lock);
421 }
422
423
1da177e4
LT
424 /* Remove any stale portmap registrations */
425 svc_register(serv, 0, 0);
426
427 return serv;
428}
429
a7455442
GB
430struct svc_serv *
431svc_create(struct svc_program *prog, unsigned int bufsize,
432 void (*shutdown)(struct svc_serv *serv))
433{
434 return __svc_create(prog, bufsize, /*npools*/1, shutdown);
435}
d2f7e79e 436EXPORT_SYMBOL(svc_create);
a7455442
GB
437
438struct svc_serv *
439svc_create_pooled(struct svc_program *prog, unsigned int bufsize,
440 void (*shutdown)(struct svc_serv *serv),
441 svc_thread_fn func, int sig, struct module *mod)
442{
443 struct svc_serv *serv;
42a7fc4a 444 unsigned int npools = svc_pool_map_get();
a7455442 445
bfd24160 446 serv = __svc_create(prog, bufsize, npools, shutdown);
a7455442
GB
447
448 if (serv != NULL) {
449 serv->sv_function = func;
450 serv->sv_kill_signal = sig;
451 serv->sv_module = mod;
452 }
453
454 return serv;
455}
d2f7e79e 456EXPORT_SYMBOL(svc_create_pooled);
a7455442 457
1da177e4 458/*
3262c816 459 * Destroy an RPC service. Should be called with the BKL held
1da177e4
LT
460 */
461void
462svc_destroy(struct svc_serv *serv)
463{
46121cf7 464 dprintk("svc: svc_destroy(%s, %d)\n",
1da177e4
LT
465 serv->sv_program->pg_name,
466 serv->sv_nrthreads);
467
468 if (serv->sv_nrthreads) {
469 if (--(serv->sv_nrthreads) != 0) {
470 svc_sock_update_bufs(serv);
471 return;
472 }
473 } else
474 printk("svc_destroy: no threads for serv=%p!\n", serv);
475
36bdfc8b
GB
476 del_timer_sync(&serv->sv_temptimer);
477
7a182083 478 svc_close_all(&serv->sv_tempsocks);
cda1fd4a 479
bc591ccf
N
480 if (serv->sv_shutdown)
481 serv->sv_shutdown(serv);
482
7a182083 483 svc_close_all(&serv->sv_permsocks);
cda1fd4a
N
484
485 BUG_ON(!list_empty(&serv->sv_permsocks));
486 BUG_ON(!list_empty(&serv->sv_tempsocks));
cca5172a 487
1da177e4
LT
488 cache_clean_deferred(serv);
489
42a7fc4a
GB
490 if (svc_serv_is_pooled(serv))
491 svc_pool_map_put();
492
1da177e4
LT
493 /* Unregister service with the portmapper */
494 svc_register(serv, 0, 0);
3262c816 495 kfree(serv->sv_pools);
1da177e4
LT
496 kfree(serv);
497}
d2f7e79e 498EXPORT_SYMBOL(svc_destroy);
1da177e4
LT
499
500/*
501 * Allocate an RPC server's buffer space.
502 * We allocate pages and place them in rq_argpages.
503 */
504static int
505svc_init_buffer(struct svc_rqst *rqstp, unsigned int size)
506{
507 int pages;
508 int arghi;
cca5172a 509
c6b0a9f8
N
510 pages = size / PAGE_SIZE + 1; /* extra page as we hold both request and reply.
511 * We assume one is at most one page
512 */
1da177e4 513 arghi = 0;
09a62660 514 BUG_ON(pages > RPCSVC_MAXPAGES);
1da177e4
LT
515 while (pages) {
516 struct page *p = alloc_page(GFP_KERNEL);
517 if (!p)
518 break;
44524359 519 rqstp->rq_pages[arghi++] = p;
1da177e4
LT
520 pages--;
521 }
1da177e4
LT
522 return ! pages;
523}
524
525/*
526 * Release an RPC server buffer
527 */
528static void
529svc_release_buffer(struct svc_rqst *rqstp)
530{
44524359
N
531 int i;
532 for (i=0; i<ARRAY_SIZE(rqstp->rq_pages); i++)
533 if (rqstp->rq_pages[i])
534 put_page(rqstp->rq_pages[i]);
1da177e4
LT
535}
536
0113ab34
JL
537struct svc_rqst *
538svc_prepare_thread(struct svc_serv *serv, struct svc_pool *pool)
1da177e4
LT
539{
540 struct svc_rqst *rqstp;
1da177e4 541
0da974f4 542 rqstp = kzalloc(sizeof(*rqstp), GFP_KERNEL);
1da177e4 543 if (!rqstp)
0113ab34 544 goto out_enomem;
1da177e4 545
1da177e4
LT
546 init_waitqueue_head(&rqstp->rq_wait);
547
1da177e4 548 serv->sv_nrthreads++;
3262c816
GB
549 spin_lock_bh(&pool->sp_lock);
550 pool->sp_nrthreads++;
a7455442 551 list_add(&rqstp->rq_all, &pool->sp_all_threads);
3262c816 552 spin_unlock_bh(&pool->sp_lock);
1da177e4 553 rqstp->rq_server = serv;
3262c816 554 rqstp->rq_pool = pool;
bfd24160 555
0113ab34
JL
556 rqstp->rq_argp = kmalloc(serv->sv_xdrsize, GFP_KERNEL);
557 if (!rqstp->rq_argp)
558 goto out_thread;
559
560 rqstp->rq_resp = kmalloc(serv->sv_xdrsize, GFP_KERNEL);
561 if (!rqstp->rq_resp)
562 goto out_thread;
563
564 if (!svc_init_buffer(rqstp, serv->sv_max_mesg))
565 goto out_thread;
566
567 return rqstp;
568out_thread:
569 svc_exit_thread(rqstp);
570out_enomem:
571 return ERR_PTR(-ENOMEM);
572}
573EXPORT_SYMBOL(svc_prepare_thread);
574
575/*
576 * Create a thread in the given pool. Caller must hold BKL.
577 * On a NUMA or SMP machine, with a multi-pool serv, the thread
578 * will be restricted to run on the cpus belonging to the pool.
579 */
580static int
581__svc_create_thread(svc_thread_fn func, struct svc_serv *serv,
582 struct svc_pool *pool)
583{
584 struct svc_rqst *rqstp;
585 int error = -ENOMEM;
586 int have_oldmask = 0;
587 cpumask_t oldmask;
588
589 rqstp = svc_prepare_thread(serv, pool);
590 if (IS_ERR(rqstp)) {
591 error = PTR_ERR(rqstp);
592 goto out;
593 }
594
bfd24160
GB
595 if (serv->sv_nrpools > 1)
596 have_oldmask = svc_pool_map_set_cpumask(pool->sp_id, &oldmask);
597
1da177e4 598 error = kernel_thread((int (*)(void *)) func, rqstp, 0);
bfd24160
GB
599
600 if (have_oldmask)
601 set_cpus_allowed(current, oldmask);
602
1da177e4
LT
603 if (error < 0)
604 goto out_thread;
605 svc_sock_update_bufs(serv);
606 error = 0;
607out:
608 return error;
609
610out_thread:
611 svc_exit_thread(rqstp);
612 goto out;
613}
614
615/*
3262c816
GB
616 * Create a thread in the default pool. Caller must hold BKL.
617 */
618int
619svc_create_thread(svc_thread_fn func, struct svc_serv *serv)
620{
621 return __svc_create_thread(func, serv, &serv->sv_pools[0]);
622}
d2f7e79e 623EXPORT_SYMBOL(svc_create_thread);
3262c816 624
a7455442
GB
625/*
626 * Choose a pool in which to create a new thread, for svc_set_num_threads
627 */
628static inline struct svc_pool *
629choose_pool(struct svc_serv *serv, struct svc_pool *pool, unsigned int *state)
630{
631 if (pool != NULL)
632 return pool;
633
cca5172a 634 return &serv->sv_pools[(*state)++ % serv->sv_nrpools];
a7455442
GB
635}
636
637/*
638 * Choose a thread to kill, for svc_set_num_threads
639 */
640static inline struct task_struct *
641choose_victim(struct svc_serv *serv, struct svc_pool *pool, unsigned int *state)
642{
643 unsigned int i;
644 struct task_struct *task = NULL;
645
646 if (pool != NULL) {
647 spin_lock_bh(&pool->sp_lock);
648 } else {
649 /* choose a pool in round-robin fashion */
cca5172a
YH
650 for (i = 0; i < serv->sv_nrpools; i++) {
651 pool = &serv->sv_pools[--(*state) % serv->sv_nrpools];
a7455442 652 spin_lock_bh(&pool->sp_lock);
cca5172a
YH
653 if (!list_empty(&pool->sp_all_threads))
654 goto found_pool;
a7455442 655 spin_unlock_bh(&pool->sp_lock);
cca5172a 656 }
a7455442
GB
657 return NULL;
658 }
659
660found_pool:
661 if (!list_empty(&pool->sp_all_threads)) {
662 struct svc_rqst *rqstp;
663
664 /*
665 * Remove from the pool->sp_all_threads list
666 * so we don't try to kill it again.
667 */
668 rqstp = list_entry(pool->sp_all_threads.next, struct svc_rqst, rq_all);
669 list_del_init(&rqstp->rq_all);
670 task = rqstp->rq_task;
cca5172a 671 }
a7455442
GB
672 spin_unlock_bh(&pool->sp_lock);
673
674 return task;
675}
676
677/*
678 * Create or destroy enough new threads to make the number
679 * of threads the given number. If `pool' is non-NULL, applies
680 * only to threads in that pool, otherwise round-robins between
681 * all pools. Must be called with a svc_get() reference and
682 * the BKL held.
683 *
684 * Destroying threads relies on the service threads filling in
685 * rqstp->rq_task, which only the nfs ones do. Assumes the serv
686 * has been created using svc_create_pooled().
687 *
688 * Based on code that used to be in nfsd_svc() but tweaked
689 * to be pool-aware.
690 */
691int
692svc_set_num_threads(struct svc_serv *serv, struct svc_pool *pool, int nrservs)
693{
694 struct task_struct *victim;
695 int error = 0;
696 unsigned int state = serv->sv_nrthreads-1;
697
698 if (pool == NULL) {
699 /* The -1 assumes caller has done a svc_get() */
700 nrservs -= (serv->sv_nrthreads-1);
701 } else {
702 spin_lock_bh(&pool->sp_lock);
703 nrservs -= pool->sp_nrthreads;
704 spin_unlock_bh(&pool->sp_lock);
705 }
706
707 /* create new threads */
708 while (nrservs > 0) {
709 nrservs--;
710 __module_get(serv->sv_module);
711 error = __svc_create_thread(serv->sv_function, serv,
712 choose_pool(serv, pool, &state));
713 if (error < 0) {
714 module_put(serv->sv_module);
715 break;
716 }
717 }
718 /* destroy old threads */
719 while (nrservs < 0 &&
720 (victim = choose_victim(serv, pool, &state)) != NULL) {
721 send_sig(serv->sv_kill_signal, victim, 1);
722 nrservs++;
723 }
724
725 return error;
726}
d2f7e79e 727EXPORT_SYMBOL(svc_set_num_threads);
a7455442 728
3262c816
GB
729/*
730 * Called from a server thread as it's exiting. Caller must hold BKL.
1da177e4
LT
731 */
732void
733svc_exit_thread(struct svc_rqst *rqstp)
734{
735 struct svc_serv *serv = rqstp->rq_server;
3262c816 736 struct svc_pool *pool = rqstp->rq_pool;
1da177e4
LT
737
738 svc_release_buffer(rqstp);
a51482bd
JJ
739 kfree(rqstp->rq_resp);
740 kfree(rqstp->rq_argp);
741 kfree(rqstp->rq_auth_data);
3262c816
GB
742
743 spin_lock_bh(&pool->sp_lock);
744 pool->sp_nrthreads--;
a7455442 745 list_del(&rqstp->rq_all);
3262c816
GB
746 spin_unlock_bh(&pool->sp_lock);
747
1da177e4
LT
748 kfree(rqstp);
749
750 /* Release the server */
751 if (serv)
752 svc_destroy(serv);
753}
d2f7e79e 754EXPORT_SYMBOL(svc_exit_thread);
1da177e4
LT
755
756/*
757 * Register an RPC service with the local portmapper.
cca5172a 758 * To unregister a service, call this routine with
1da177e4
LT
759 * proto and port == 0.
760 */
761int
762svc_register(struct svc_serv *serv, int proto, unsigned short port)
763{
764 struct svc_program *progp;
765 unsigned long flags;
766 int i, error = 0, dummy;
767
1da177e4
LT
768 if (!port)
769 clear_thread_flag(TIF_SIGPENDING);
770
bc5fea42
OK
771 for (progp = serv->sv_program; progp; progp = progp->pg_next) {
772 for (i = 0; i < progp->pg_nvers; i++) {
773 if (progp->pg_vers[i] == NULL)
774 continue;
775
46121cf7 776 dprintk("svc: svc_register(%s, %s, %d, %d)%s\n",
bc5fea42
OK
777 progp->pg_name,
778 proto == IPPROTO_UDP? "udp" : "tcp",
779 port,
780 i,
781 progp->pg_vers[i]->vs_hidden?
782 " (but not telling portmap)" : "");
783
784 if (progp->pg_vers[i]->vs_hidden)
785 continue;
786
26080014 787 error = rpcb_register(progp->pg_prog, i, proto, port, &dummy);
bc5fea42
OK
788 if (error < 0)
789 break;
790 if (port && !dummy) {
791 error = -EACCES;
792 break;
793 }
1da177e4
LT
794 }
795 }
796
797 if (!port) {
798 spin_lock_irqsave(&current->sighand->siglock, flags);
799 recalc_sigpending();
800 spin_unlock_irqrestore(&current->sighand->siglock, flags);
801 }
802
803 return error;
804}
805
354ecbb9
DDAG
806/*
807 * Printk the given error with the address of the client that caused it.
808 */
809static int
810__attribute__ ((format (printf, 2, 3)))
811svc_printk(struct svc_rqst *rqstp, const char *fmt, ...)
812{
813 va_list args;
814 int r;
815 char buf[RPC_MAX_ADDRBUFLEN];
816
817 if (!net_ratelimit())
818 return 0;
819
820 printk(KERN_WARNING "svc: %s: ",
821 svc_print_addr(rqstp, buf, sizeof(buf)));
822
823 va_start(args, fmt);
824 r = vprintk(fmt, args);
825 va_end(args);
826
827 return r;
828}
829
1da177e4
LT
830/*
831 * Process the RPC request.
832 */
833int
6fb2b47f 834svc_process(struct svc_rqst *rqstp)
1da177e4
LT
835{
836 struct svc_program *progp;
837 struct svc_version *versp = NULL; /* compiler food */
838 struct svc_procedure *procp = NULL;
839 struct kvec * argv = &rqstp->rq_arg.head[0];
840 struct kvec * resv = &rqstp->rq_res.head[0];
6fb2b47f 841 struct svc_serv *serv = rqstp->rq_server;
1da177e4 842 kxdrproc_t xdr;
d8ed029d
AD
843 __be32 *statp;
844 u32 dir, prog, vers, proc;
845 __be32 auth_stat, rpc_stat;
1da177e4 846 int auth_res;
8f8e05c5 847 __be32 *reply_statp;
1da177e4
LT
848
849 rpc_stat = rpc_success;
850
851 if (argv->iov_len < 6*4)
852 goto err_short_len;
853
854 /* setup response xdr_buf.
cca5172a 855 * Initially it has just one page
1da177e4 856 */
44524359 857 rqstp->rq_resused = 1;
1da177e4
LT
858 resv->iov_base = page_address(rqstp->rq_respages[0]);
859 resv->iov_len = 0;
44524359 860 rqstp->rq_res.pages = rqstp->rq_respages + 1;
1da177e4
LT
861 rqstp->rq_res.len = 0;
862 rqstp->rq_res.page_base = 0;
863 rqstp->rq_res.page_len = 0;
334ccfd5 864 rqstp->rq_res.buflen = PAGE_SIZE;
7c9fdcfb 865 rqstp->rq_res.tail[0].iov_base = NULL;
1da177e4 866 rqstp->rq_res.tail[0].iov_len = 0;
5c04c46a 867 /* Will be turned off only in gss privacy case: */
cf8208d0 868 rqstp->rq_splice_ok = 1;
e831fe65
TT
869
870 /* Setup reply header */
871 rqstp->rq_xprt->xpt_ops->xpo_prep_reply_hdr(rqstp);
1da177e4
LT
872
873 rqstp->rq_xid = svc_getu32(argv);
874 svc_putu32(resv, rqstp->rq_xid);
875
76994313
AD
876 dir = svc_getnl(argv);
877 vers = svc_getnl(argv);
1da177e4
LT
878
879 /* First words of reply: */
76994313 880 svc_putnl(resv, 1); /* REPLY */
1da177e4
LT
881
882 if (dir != 0) /* direction != CALL */
883 goto err_bad_dir;
884 if (vers != 2) /* RPC version number */
885 goto err_bad_rpc;
886
887 /* Save position in case we later decide to reject: */
8f8e05c5 888 reply_statp = resv->iov_base + resv->iov_len;
1da177e4 889
76994313 890 svc_putnl(resv, 0); /* ACCEPT */
1da177e4 891
76994313
AD
892 rqstp->rq_prog = prog = svc_getnl(argv); /* program number */
893 rqstp->rq_vers = vers = svc_getnl(argv); /* version number */
894 rqstp->rq_proc = proc = svc_getnl(argv); /* procedure number */
1da177e4
LT
895
896 progp = serv->sv_program;
80d188a6
N
897
898 for (progp = serv->sv_program; progp; progp = progp->pg_next)
899 if (prog == progp->pg_prog)
900 break;
901
1da177e4
LT
902 /*
903 * Decode auth data, and add verifier to reply buffer.
904 * We do this before anything else in order to get a decent
905 * auth verifier.
906 */
907 auth_res = svc_authenticate(rqstp, &auth_stat);
908 /* Also give the program a chance to reject this call: */
80d188a6 909 if (auth_res == SVC_OK && progp) {
1da177e4
LT
910 auth_stat = rpc_autherr_badcred;
911 auth_res = progp->pg_authenticate(rqstp);
912 }
913 switch (auth_res) {
914 case SVC_OK:
915 break;
916 case SVC_GARBAGE:
917 rpc_stat = rpc_garbage_args;
918 goto err_bad;
919 case SVC_SYSERR:
920 rpc_stat = rpc_system_err;
921 goto err_bad;
922 case SVC_DENIED:
923 goto err_bad_auth;
924 case SVC_DROP:
925 goto dropit;
926 case SVC_COMPLETE:
927 goto sendit;
928 }
80d188a6 929
9ba02638 930 if (progp == NULL)
1da177e4
LT
931 goto err_bad_prog;
932
933 if (vers >= progp->pg_nvers ||
934 !(versp = progp->pg_vers[vers]))
935 goto err_bad_vers;
936
937 procp = versp->vs_proc + proc;
938 if (proc >= versp->vs_nproc || !procp->pc_func)
939 goto err_bad_proc;
940 rqstp->rq_server = serv;
941 rqstp->rq_procinfo = procp;
942
943 /* Syntactic check complete */
944 serv->sv_stats->rpccnt++;
945
946 /* Build the reply header. */
947 statp = resv->iov_base +resv->iov_len;
76994313 948 svc_putnl(resv, RPC_SUCCESS);
1da177e4
LT
949
950 /* Bump per-procedure stats counter */
951 procp->pc_count++;
952
953 /* Initialize storage for argp and resp */
954 memset(rqstp->rq_argp, 0, procp->pc_argsize);
955 memset(rqstp->rq_resp, 0, procp->pc_ressize);
956
cca5172a 957 /* un-reserve some of the out-queue now that we have a
1da177e4
LT
958 * better idea of reply size
959 */
960 if (procp->pc_xdrressize)
cd123012 961 svc_reserve_auth(rqstp, procp->pc_xdrressize<<2);
1da177e4
LT
962
963 /* Call the function that processes the request. */
964 if (!versp->vs_dispatch) {
965 /* Decode arguments */
966 xdr = procp->pc_decode;
967 if (xdr && !xdr(rqstp, argv->iov_base, rqstp->rq_argp))
968 goto err_garbage;
969
970 *statp = procp->pc_func(rqstp, rqstp->rq_argp, rqstp->rq_resp);
971
972 /* Encode reply */
d343fce1
N
973 if (*statp == rpc_drop_reply) {
974 if (procp->pc_release)
975 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
976 goto dropit;
977 }
1da177e4
LT
978 if (*statp == rpc_success && (xdr = procp->pc_encode)
979 && !xdr(rqstp, resv->iov_base+resv->iov_len, rqstp->rq_resp)) {
980 dprintk("svc: failed to encode reply\n");
981 /* serv->sv_stats->rpcsystemerr++; */
982 *statp = rpc_system_err;
983 }
984 } else {
985 dprintk("svc: calling dispatcher\n");
986 if (!versp->vs_dispatch(rqstp, statp)) {
987 /* Release reply info */
988 if (procp->pc_release)
989 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
990 goto dropit;
991 }
992 }
993
994 /* Check RPC status result */
995 if (*statp != rpc_success)
996 resv->iov_len = ((void*)statp) - resv->iov_base + 4;
997
998 /* Release reply info */
999 if (procp->pc_release)
1000 procp->pc_release(rqstp, NULL, rqstp->rq_resp);
1001
1002 if (procp->pc_encode == NULL)
1003 goto dropit;
1004
1005 sendit:
1006 if (svc_authorise(rqstp))
1007 goto dropit;
1008 return svc_send(rqstp);
1009
1010 dropit:
1011 svc_authorise(rqstp); /* doesn't hurt to call this twice */
1012 dprintk("svc: svc_process dropit\n");
1013 svc_drop(rqstp);
1014 return 0;
1015
1016err_short_len:
354ecbb9
DDAG
1017 svc_printk(rqstp, "short len %Zd, dropping request\n",
1018 argv->iov_len);
34e9a63b 1019
1da177e4
LT
1020 goto dropit; /* drop request */
1021
1022err_bad_dir:
354ecbb9 1023 svc_printk(rqstp, "bad direction %d, dropping request\n", dir);
34e9a63b 1024
1da177e4
LT
1025 serv->sv_stats->rpcbadfmt++;
1026 goto dropit; /* drop request */
1027
1028err_bad_rpc:
1029 serv->sv_stats->rpcbadfmt++;
76994313
AD
1030 svc_putnl(resv, 1); /* REJECT */
1031 svc_putnl(resv, 0); /* RPC_MISMATCH */
1032 svc_putnl(resv, 2); /* Only RPCv2 supported */
1033 svc_putnl(resv, 2);
1da177e4
LT
1034 goto sendit;
1035
1036err_bad_auth:
1037 dprintk("svc: authentication failed (%d)\n", ntohl(auth_stat));
1038 serv->sv_stats->rpcbadauth++;
1039 /* Restore write pointer to location of accept status: */
8f8e05c5 1040 xdr_ressize_check(rqstp, reply_statp);
76994313
AD
1041 svc_putnl(resv, 1); /* REJECT */
1042 svc_putnl(resv, 1); /* AUTH_ERROR */
1043 svc_putnl(resv, ntohl(auth_stat)); /* status */
1da177e4
LT
1044 goto sendit;
1045
1046err_bad_prog:
9ba02638 1047 dprintk("svc: unknown program %d\n", prog);
1da177e4 1048 serv->sv_stats->rpcbadfmt++;
76994313 1049 svc_putnl(resv, RPC_PROG_UNAVAIL);
1da177e4
LT
1050 goto sendit;
1051
1052err_bad_vers:
354ecbb9 1053 svc_printk(rqstp, "unknown version (%d for prog %d, %s)\n",
34e9a63b
N
1054 vers, prog, progp->pg_name);
1055
1da177e4 1056 serv->sv_stats->rpcbadfmt++;
76994313
AD
1057 svc_putnl(resv, RPC_PROG_MISMATCH);
1058 svc_putnl(resv, progp->pg_lovers);
1059 svc_putnl(resv, progp->pg_hivers);
1da177e4
LT
1060 goto sendit;
1061
1062err_bad_proc:
354ecbb9 1063 svc_printk(rqstp, "unknown procedure (%d)\n", proc);
34e9a63b 1064
1da177e4 1065 serv->sv_stats->rpcbadfmt++;
76994313 1066 svc_putnl(resv, RPC_PROC_UNAVAIL);
1da177e4
LT
1067 goto sendit;
1068
1069err_garbage:
354ecbb9 1070 svc_printk(rqstp, "failed to decode args\n");
34e9a63b 1071
1da177e4
LT
1072 rpc_stat = rpc_garbage_args;
1073err_bad:
1074 serv->sv_stats->rpcbadfmt++;
76994313 1075 svc_putnl(resv, ntohl(rpc_stat));
1da177e4
LT
1076 goto sendit;
1077}
d2f7e79e 1078EXPORT_SYMBOL(svc_process);
7adae489
GB
1079
1080/*
1081 * Return (transport-specific) limit on the rpc payload.
1082 */
1083u32 svc_max_payload(const struct svc_rqst *rqstp)
1084{
49023155 1085 u32 max = rqstp->rq_xprt->xpt_class->xcl_max_payload;
7adae489 1086
c6b0a9f8
N
1087 if (rqstp->rq_server->sv_max_payload < max)
1088 max = rqstp->rq_server->sv_max_payload;
7adae489
GB
1089 return max;
1090}
1091EXPORT_SYMBOL_GPL(svc_max_payload);