2 * linux/net/sunrpc/svc.c
4 * High-level RPC service routines
6 * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
9 #include <linux/linkage.h>
10 #include <linux/sched.h>
11 #include <linux/errno.h>
12 #include <linux/net.h>
16 #include <linux/sunrpc/types.h>
17 #include <linux/sunrpc/xdr.h>
18 #include <linux/sunrpc/stats.h>
19 #include <linux/sunrpc/svcsock.h>
20 #include <linux/sunrpc/clnt.h>
22 #define RPCDBG_FACILITY RPCDBG_SVCDSP
23 #define RPC_PARANOIA 1
26 * Create an RPC service
29 svc_create(struct svc_program
*prog
, unsigned int bufsize
,
30 void (*shutdown
)(struct svc_serv
*serv
))
32 struct svc_serv
*serv
;
36 if (!(serv
= kzalloc(sizeof(*serv
), GFP_KERNEL
)))
38 serv
->sv_name
= prog
->pg_name
;
39 serv
->sv_program
= prog
;
40 serv
->sv_nrthreads
= 1;
41 serv
->sv_stats
= prog
->pg_stats
;
42 serv
->sv_bufsz
= bufsize
? bufsize
: 4096;
43 serv
->sv_shutdown
= shutdown
;
46 prog
->pg_lovers
= prog
->pg_nvers
-1;
47 for (vers
=0; vers
<prog
->pg_nvers
; vers
++)
48 if (prog
->pg_vers
[vers
]) {
49 prog
->pg_hivers
= vers
;
50 if (prog
->pg_lovers
> vers
)
51 prog
->pg_lovers
= vers
;
52 if (prog
->pg_vers
[vers
]->vs_xdrsize
> xdrsize
)
53 xdrsize
= prog
->pg_vers
[vers
]->vs_xdrsize
;
57 serv
->sv_xdrsize
= xdrsize
;
58 INIT_LIST_HEAD(&serv
->sv_threads
);
59 INIT_LIST_HEAD(&serv
->sv_sockets
);
60 INIT_LIST_HEAD(&serv
->sv_tempsocks
);
61 INIT_LIST_HEAD(&serv
->sv_permsocks
);
62 spin_lock_init(&serv
->sv_lock
);
64 /* Remove any stale portmap registrations */
65 svc_register(serv
, 0, 0);
71 * Destroy an RPC service
74 svc_destroy(struct svc_serv
*serv
)
76 struct svc_sock
*svsk
;
78 dprintk("RPC: svc_destroy(%s, %d)\n",
79 serv
->sv_program
->pg_name
,
82 if (serv
->sv_nrthreads
) {
83 if (--(serv
->sv_nrthreads
) != 0) {
84 svc_sock_update_bufs(serv
);
88 printk("svc_destroy: no threads for serv=%p!\n", serv
);
90 while (!list_empty(&serv
->sv_tempsocks
)) {
91 svsk
= list_entry(serv
->sv_tempsocks
.next
,
94 svc_delete_socket(svsk
);
96 if (serv
->sv_shutdown
)
97 serv
->sv_shutdown(serv
);
99 while (!list_empty(&serv
->sv_permsocks
)) {
100 svsk
= list_entry(serv
->sv_permsocks
.next
,
103 svc_delete_socket(svsk
);
106 cache_clean_deferred(serv
);
108 /* Unregister service with the portmapper */
109 svc_register(serv
, 0, 0);
114 * Allocate an RPC server's buffer space.
115 * We allocate pages and place them in rq_argpages.
118 svc_init_buffer(struct svc_rqst
*rqstp
, unsigned int size
)
123 if (size
> RPCSVC_MAXPAYLOAD
)
124 size
= RPCSVC_MAXPAYLOAD
;
125 pages
= 2 + (size
+ PAGE_SIZE
-1) / PAGE_SIZE
;
126 rqstp
->rq_argused
= 0;
127 rqstp
->rq_resused
= 0;
129 BUG_ON(pages
> RPCSVC_MAXPAGES
);
131 struct page
*p
= alloc_page(GFP_KERNEL
);
134 rqstp
->rq_argpages
[arghi
++] = p
;
137 rqstp
->rq_arghi
= arghi
;
142 * Release an RPC server buffer
145 svc_release_buffer(struct svc_rqst
*rqstp
)
147 while (rqstp
->rq_arghi
)
148 put_page(rqstp
->rq_argpages
[--rqstp
->rq_arghi
]);
149 while (rqstp
->rq_resused
) {
150 if (rqstp
->rq_respages
[--rqstp
->rq_resused
] == NULL
)
152 put_page(rqstp
->rq_respages
[rqstp
->rq_resused
]);
154 rqstp
->rq_argused
= 0;
158 * Create a server thread
161 svc_create_thread(svc_thread_fn func
, struct svc_serv
*serv
)
163 struct svc_rqst
*rqstp
;
166 rqstp
= kzalloc(sizeof(*rqstp
), GFP_KERNEL
);
170 init_waitqueue_head(&rqstp
->rq_wait
);
172 if (!(rqstp
->rq_argp
= kmalloc(serv
->sv_xdrsize
, GFP_KERNEL
))
173 || !(rqstp
->rq_resp
= kmalloc(serv
->sv_xdrsize
, GFP_KERNEL
))
174 || !svc_init_buffer(rqstp
, serv
->sv_bufsz
))
177 serv
->sv_nrthreads
++;
178 rqstp
->rq_server
= serv
;
179 error
= kernel_thread((int (*)(void *)) func
, rqstp
, 0);
182 svc_sock_update_bufs(serv
);
188 svc_exit_thread(rqstp
);
193 * Destroy an RPC server thread
196 svc_exit_thread(struct svc_rqst
*rqstp
)
198 struct svc_serv
*serv
= rqstp
->rq_server
;
200 svc_release_buffer(rqstp
);
201 kfree(rqstp
->rq_resp
);
202 kfree(rqstp
->rq_argp
);
203 kfree(rqstp
->rq_auth_data
);
206 /* Release the server */
212 * Register an RPC service with the local portmapper.
213 * To unregister a service, call this routine with
214 * proto and port == 0.
217 svc_register(struct svc_serv
*serv
, int proto
, unsigned short port
)
219 struct svc_program
*progp
;
221 int i
, error
= 0, dummy
;
223 progp
= serv
->sv_program
;
225 dprintk("RPC: svc_register(%s, %s, %d)\n",
226 progp
->pg_name
, proto
== IPPROTO_UDP
? "udp" : "tcp", port
);
229 clear_thread_flag(TIF_SIGPENDING
);
231 for (i
= 0; i
< progp
->pg_nvers
; i
++) {
232 if (progp
->pg_vers
[i
] == NULL
)
234 error
= rpc_register(progp
->pg_prog
, i
, proto
, port
, &dummy
);
237 if (port
&& !dummy
) {
244 spin_lock_irqsave(¤t
->sighand
->siglock
, flags
);
246 spin_unlock_irqrestore(¤t
->sighand
->siglock
, flags
);
253 * Process the RPC request.
256 svc_process(struct svc_serv
*serv
, struct svc_rqst
*rqstp
)
258 struct svc_program
*progp
;
259 struct svc_version
*versp
= NULL
; /* compiler food */
260 struct svc_procedure
*procp
= NULL
;
261 struct kvec
* argv
= &rqstp
->rq_arg
.head
[0];
262 struct kvec
* resv
= &rqstp
->rq_res
.head
[0];
265 u32 dir
, prog
, vers
, proc
;
266 __be32 auth_stat
, rpc_stat
;
268 __be32
*accept_statp
;
270 rpc_stat
= rpc_success
;
272 if (argv
->iov_len
< 6*4)
275 /* setup response xdr_buf.
276 * Initially it has just one page
278 svc_take_page(rqstp
); /* must succeed */
279 resv
->iov_base
= page_address(rqstp
->rq_respages
[0]);
281 rqstp
->rq_res
.pages
= rqstp
->rq_respages
+1;
282 rqstp
->rq_res
.len
= 0;
283 rqstp
->rq_res
.page_base
= 0;
284 rqstp
->rq_res
.page_len
= 0;
285 rqstp
->rq_res
.buflen
= PAGE_SIZE
;
286 rqstp
->rq_res
.tail
[0].iov_base
= NULL
;
287 rqstp
->rq_res
.tail
[0].iov_len
= 0;
288 /* Will be turned off only in gss privacy case: */
289 rqstp
->rq_sendfile_ok
= 1;
290 /* tcp needs a space for the record length... */
291 if (rqstp
->rq_prot
== IPPROTO_TCP
)
294 rqstp
->rq_xid
= svc_getu32(argv
);
295 svc_putu32(resv
, rqstp
->rq_xid
);
297 dir
= svc_getnl(argv
);
298 vers
= svc_getnl(argv
);
300 /* First words of reply: */
301 svc_putnl(resv
, 1); /* REPLY */
303 if (dir
!= 0) /* direction != CALL */
305 if (vers
!= 2) /* RPC version number */
308 /* Save position in case we later decide to reject: */
309 accept_statp
= resv
->iov_base
+ resv
->iov_len
;
311 svc_putnl(resv
, 0); /* ACCEPT */
313 rqstp
->rq_prog
= prog
= svc_getnl(argv
); /* program number */
314 rqstp
->rq_vers
= vers
= svc_getnl(argv
); /* version number */
315 rqstp
->rq_proc
= proc
= svc_getnl(argv
); /* procedure number */
317 progp
= serv
->sv_program
;
319 for (progp
= serv
->sv_program
; progp
; progp
= progp
->pg_next
)
320 if (prog
== progp
->pg_prog
)
324 * Decode auth data, and add verifier to reply buffer.
325 * We do this before anything else in order to get a decent
328 auth_res
= svc_authenticate(rqstp
, &auth_stat
);
329 /* Also give the program a chance to reject this call: */
330 if (auth_res
== SVC_OK
&& progp
) {
331 auth_stat
= rpc_autherr_badcred
;
332 auth_res
= progp
->pg_authenticate(rqstp
);
338 rpc_stat
= rpc_garbage_args
;
341 rpc_stat
= rpc_system_err
;
354 if (vers
>= progp
->pg_nvers
||
355 !(versp
= progp
->pg_vers
[vers
]))
358 procp
= versp
->vs_proc
+ proc
;
359 if (proc
>= versp
->vs_nproc
|| !procp
->pc_func
)
361 rqstp
->rq_server
= serv
;
362 rqstp
->rq_procinfo
= procp
;
364 /* Syntactic check complete */
365 serv
->sv_stats
->rpccnt
++;
367 /* Build the reply header. */
368 statp
= resv
->iov_base
+resv
->iov_len
;
369 svc_putnl(resv
, RPC_SUCCESS
);
371 /* Bump per-procedure stats counter */
374 /* Initialize storage for argp and resp */
375 memset(rqstp
->rq_argp
, 0, procp
->pc_argsize
);
376 memset(rqstp
->rq_resp
, 0, procp
->pc_ressize
);
378 /* un-reserve some of the out-queue now that we have a
379 * better idea of reply size
381 if (procp
->pc_xdrressize
)
382 svc_reserve(rqstp
, procp
->pc_xdrressize
<<2);
384 /* Call the function that processes the request. */
385 if (!versp
->vs_dispatch
) {
386 /* Decode arguments */
387 xdr
= procp
->pc_decode
;
388 if (xdr
&& !xdr(rqstp
, argv
->iov_base
, rqstp
->rq_argp
))
391 *statp
= procp
->pc_func(rqstp
, rqstp
->rq_argp
, rqstp
->rq_resp
);
394 if (*statp
== rpc_success
&& (xdr
= procp
->pc_encode
)
395 && !xdr(rqstp
, resv
->iov_base
+resv
->iov_len
, rqstp
->rq_resp
)) {
396 dprintk("svc: failed to encode reply\n");
397 /* serv->sv_stats->rpcsystemerr++; */
398 *statp
= rpc_system_err
;
401 dprintk("svc: calling dispatcher\n");
402 if (!versp
->vs_dispatch(rqstp
, statp
)) {
403 /* Release reply info */
404 if (procp
->pc_release
)
405 procp
->pc_release(rqstp
, NULL
, rqstp
->rq_resp
);
410 /* Check RPC status result */
411 if (*statp
!= rpc_success
)
412 resv
->iov_len
= ((void*)statp
) - resv
->iov_base
+ 4;
414 /* Release reply info */
415 if (procp
->pc_release
)
416 procp
->pc_release(rqstp
, NULL
, rqstp
->rq_resp
);
418 if (procp
->pc_encode
== NULL
)
422 if (svc_authorise(rqstp
))
424 return svc_send(rqstp
);
427 svc_authorise(rqstp
); /* doesn't hurt to call this twice */
428 dprintk("svc: svc_process dropit\n");
434 printk("svc: short len %Zd, dropping request\n", argv
->iov_len
);
436 goto dropit
; /* drop request */
440 printk("svc: bad direction %d, dropping request\n", dir
);
442 serv
->sv_stats
->rpcbadfmt
++;
443 goto dropit
; /* drop request */
446 serv
->sv_stats
->rpcbadfmt
++;
447 svc_putnl(resv
, 1); /* REJECT */
448 svc_putnl(resv
, 0); /* RPC_MISMATCH */
449 svc_putnl(resv
, 2); /* Only RPCv2 supported */
454 dprintk("svc: authentication failed (%d)\n", ntohl(auth_stat
));
455 serv
->sv_stats
->rpcbadauth
++;
456 /* Restore write pointer to location of accept status: */
457 xdr_ressize_check(rqstp
, accept_statp
);
458 svc_putnl(resv
, 1); /* REJECT */
459 svc_putnl(resv
, 1); /* AUTH_ERROR */
460 svc_putnl(resv
, ntohl(auth_stat
)); /* status */
464 dprintk("svc: unknown program %d\n", prog
);
465 serv
->sv_stats
->rpcbadfmt
++;
466 svc_putnl(resv
, RPC_PROG_UNAVAIL
);
471 printk("svc: unknown version (%d)\n", vers
);
473 serv
->sv_stats
->rpcbadfmt
++;
474 svc_putnl(resv
, RPC_PROG_MISMATCH
);
475 svc_putnl(resv
, progp
->pg_lovers
);
476 svc_putnl(resv
, progp
->pg_hivers
);
481 printk("svc: unknown procedure (%d)\n", proc
);
483 serv
->sv_stats
->rpcbadfmt
++;
484 svc_putnl(resv
, RPC_PROC_UNAVAIL
);
489 printk("svc: failed to decode args\n");
491 rpc_stat
= rpc_garbage_args
;
493 serv
->sv_stats
->rpcbadfmt
++;
494 svc_putnl(resv
, ntohl(rpc_stat
));