xref: /linux/include/linux/sunrpc/svc.h (revision 84b9b44b)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * linux/include/linux/sunrpc/svc.h
4  *
5  * RPC server declarations.
6  *
7  * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
8  */
9 
10 
11 #ifndef SUNRPC_SVC_H
12 #define SUNRPC_SVC_H
13 
14 #include <linux/in.h>
15 #include <linux/in6.h>
16 #include <linux/sunrpc/types.h>
17 #include <linux/sunrpc/xdr.h>
18 #include <linux/sunrpc/auth.h>
19 #include <linux/sunrpc/svcauth.h>
20 #include <linux/wait.h>
21 #include <linux/mm.h>
22 #include <linux/pagevec.h>
23 
24 /*
25  *
26  * RPC service thread pool.
27  *
28  * Pool of threads and temporary sockets.  Generally there is only
29  * a single one of these per RPC service, but on NUMA machines those
30  * services that can benefit from it (i.e. nfs but not lockd) will
31  * have one pool per NUMA node.  This optimisation reduces cross-
32  * node traffic on multi-node NUMA NFS servers.
33  */
34 struct svc_pool {
35 	unsigned int		sp_id;	    	/* pool id; also node id on NUMA */
36 	spinlock_t		sp_lock;	/* protects all fields */
37 	struct list_head	sp_sockets;	/* pending sockets */
38 	unsigned int		sp_nrthreads;	/* # of threads in pool */
39 	struct list_head	sp_all_threads;	/* all server threads */
40 
41 	/* statistics on pool operation */
42 	struct percpu_counter	sp_sockets_queued;
43 	struct percpu_counter	sp_threads_woken;
44 	struct percpu_counter	sp_threads_timedout;
45 
46 #define	SP_TASK_PENDING		(0)		/* still work to do even if no
47 						 * xprt is queued. */
48 #define SP_CONGESTED		(1)
49 	unsigned long		sp_flags;
50 } ____cacheline_aligned_in_smp;
51 
52 /*
53  * RPC service.
54  *
55  * An RPC service is a ``daemon,'' possibly multithreaded, which
56  * receives and processes incoming RPC messages.
57  * It has one or more transport sockets associated with it, and maintains
58  * a list of idle threads waiting for input.
59  *
60  * We currently do not support more than one RPC program per daemon.
61  */
62 struct svc_serv {
63 	struct svc_program *	sv_program;	/* RPC program */
64 	struct svc_stat *	sv_stats;	/* RPC statistics */
65 	spinlock_t		sv_lock;
66 	struct kref		sv_refcnt;
67 	unsigned int		sv_nrthreads;	/* # of server threads */
68 	unsigned int		sv_maxconn;	/* max connections allowed or
69 						 * '0' causing max to be based
70 						 * on number of threads. */
71 
72 	unsigned int		sv_max_payload;	/* datagram payload size */
73 	unsigned int		sv_max_mesg;	/* max_payload + 1 page for overheads */
74 	unsigned int		sv_xdrsize;	/* XDR buffer size */
75 	struct list_head	sv_permsocks;	/* all permanent sockets */
76 	struct list_head	sv_tempsocks;	/* all temporary sockets */
77 	int			sv_tmpcnt;	/* count of temporary sockets */
78 	struct timer_list	sv_temptimer;	/* timer for aging temporary sockets */
79 
80 	char *			sv_name;	/* service name */
81 
82 	unsigned int		sv_nrpools;	/* number of thread pools */
83 	struct svc_pool *	sv_pools;	/* array of thread pools */
84 	int			(*sv_threadfn)(void *data);
85 
86 #if defined(CONFIG_SUNRPC_BACKCHANNEL)
87 	struct list_head	sv_cb_list;	/* queue for callback requests
88 						 * that arrive over the same
89 						 * connection */
90 	spinlock_t		sv_cb_lock;	/* protects the svc_cb_list */
91 	wait_queue_head_t	sv_cb_waitq;	/* sleep here if there are no
92 						 * entries in the svc_cb_list */
93 	bool			sv_bc_enabled;	/* service uses backchannel */
94 #endif /* CONFIG_SUNRPC_BACKCHANNEL */
95 };
96 
97 /**
98  * svc_get() - increment reference count on a SUNRPC serv
99  * @serv:  the svc_serv to have count incremented
100  *
101  * Returns: the svc_serv that was passed in.
102  */
103 static inline struct svc_serv *svc_get(struct svc_serv *serv)
104 {
105 	kref_get(&serv->sv_refcnt);
106 	return serv;
107 }
108 
109 void svc_destroy(struct kref *);
110 
111 /**
112  * svc_put - decrement reference count on a SUNRPC serv
113  * @serv:  the svc_serv to have count decremented
114  *
115  * When the reference count reaches zero, svc_destroy()
116  * is called to clean up and free the serv.
117  */
118 static inline void svc_put(struct svc_serv *serv)
119 {
120 	kref_put(&serv->sv_refcnt, svc_destroy);
121 }
122 
123 /**
124  * svc_put_not_last - decrement non-final reference count on SUNRPC serv
125  * @serv:  the svc_serv to have count decremented
126  *
127  * Returns: %true is refcount was decremented.
128  *
129  * If the refcount is 1, it is not decremented and instead failure is reported.
130  */
131 static inline bool svc_put_not_last(struct svc_serv *serv)
132 {
133 	return refcount_dec_not_one(&serv->sv_refcnt.refcount);
134 }
135 
136 /*
137  * Maximum payload size supported by a kernel RPC server.
138  * This is use to determine the max number of pages nfsd is
139  * willing to return in a single READ operation.
140  *
141  * These happen to all be powers of 2, which is not strictly
142  * necessary but helps enforce the real limitation, which is
143  * that they should be multiples of PAGE_SIZE.
144  *
145  * For UDP transports, a block plus NFS,RPC, and UDP headers
146  * has to fit into the IP datagram limit of 64K.  The largest
147  * feasible number for all known page sizes is probably 48K,
148  * but we choose 32K here.  This is the same as the historical
149  * Linux limit; someone who cares more about NFS/UDP performance
150  * can test a larger number.
151  *
152  * For TCP transports we have more freedom.  A size of 1MB is
153  * chosen to match the client limit.  Other OSes are known to
154  * have larger limits, but those numbers are probably beyond
155  * the point of diminishing returns.
156  */
157 #define RPCSVC_MAXPAYLOAD	(1*1024*1024u)
158 #define RPCSVC_MAXPAYLOAD_TCP	RPCSVC_MAXPAYLOAD
159 #define RPCSVC_MAXPAYLOAD_UDP	(32*1024u)
160 
161 extern u32 svc_max_payload(const struct svc_rqst *rqstp);
162 
163 /*
164  * RPC Requsts and replies are stored in one or more pages.
165  * We maintain an array of pages for each server thread.
166  * Requests are copied into these pages as they arrive.  Remaining
167  * pages are available to write the reply into.
168  *
169  * Pages are sent using ->sendpage so each server thread needs to
170  * allocate more to replace those used in sending.  To help keep track
171  * of these pages we have a receive list where all pages initialy live,
172  * and a send list where pages are moved to when there are to be part
173  * of a reply.
174  *
175  * We use xdr_buf for holding responses as it fits well with NFS
176  * read responses (that have a header, and some data pages, and possibly
177  * a tail) and means we can share some client side routines.
178  *
179  * The xdr_buf.head kvec always points to the first page in the rq_*pages
180  * list.  The xdr_buf.pages pointer points to the second page on that
181  * list.  xdr_buf.tail points to the end of the first page.
182  * This assumes that the non-page part of an rpc reply will fit
183  * in a page - NFSd ensures this.  lockd also has no trouble.
184  *
185  * Each request/reply pair can have at most one "payload", plus two pages,
186  * one for the request, and one for the reply.
187  * We using ->sendfile to return read data, we might need one extra page
188  * if the request is not page-aligned.  So add another '1'.
189  */
190 #define RPCSVC_MAXPAGES		((RPCSVC_MAXPAYLOAD+PAGE_SIZE-1)/PAGE_SIZE \
191 				+ 2 + 1)
192 
193 /*
194  * The context of a single thread, including the request currently being
195  * processed.
196  */
197 struct svc_rqst {
198 	struct list_head	rq_all;		/* all threads list */
199 	struct rcu_head		rq_rcu_head;	/* for RCU deferred kfree */
200 	struct svc_xprt *	rq_xprt;	/* transport ptr */
201 
202 	struct sockaddr_storage	rq_addr;	/* peer address */
203 	size_t			rq_addrlen;
204 	struct sockaddr_storage	rq_daddr;	/* dest addr of request
205 						 *  - reply from here */
206 	size_t			rq_daddrlen;
207 
208 	struct svc_serv *	rq_server;	/* RPC service definition */
209 	struct svc_pool *	rq_pool;	/* thread pool */
210 	const struct svc_procedure *rq_procinfo;/* procedure info */
211 	struct auth_ops *	rq_authop;	/* authentication flavour */
212 	struct svc_cred		rq_cred;	/* auth info */
213 	void *			rq_xprt_ctxt;	/* transport specific context ptr */
214 	struct svc_deferred_req*rq_deferred;	/* deferred request we are replaying */
215 
216 	struct xdr_buf		rq_arg;
217 	struct xdr_stream	rq_arg_stream;
218 	struct xdr_stream	rq_res_stream;
219 	struct page		*rq_scratch_page;
220 	struct xdr_buf		rq_res;
221 	struct page		*rq_pages[RPCSVC_MAXPAGES + 1];
222 	struct page *		*rq_respages;	/* points into rq_pages */
223 	struct page *		*rq_next_page; /* next reply page to use */
224 	struct page *		*rq_page_end;  /* one past the last page */
225 
226 	struct pagevec		rq_pvec;
227 	struct kvec		rq_vec[RPCSVC_MAXPAGES]; /* generally useful.. */
228 	struct bio_vec		rq_bvec[RPCSVC_MAXPAGES];
229 
230 	__be32			rq_xid;		/* transmission id */
231 	u32			rq_prog;	/* program number */
232 	u32			rq_vers;	/* program version */
233 	u32			rq_proc;	/* procedure number */
234 	u32			rq_prot;	/* IP protocol */
235 	int			rq_cachetype;	/* catering to nfsd */
236 #define	RQ_SECURE	(0)			/* secure port */
237 #define	RQ_LOCAL	(1)			/* local request */
238 #define	RQ_USEDEFERRAL	(2)			/* use deferral */
239 #define	RQ_DROPME	(3)			/* drop current reply */
240 #define	RQ_SPLICE_OK	(4)			/* turned off in gss privacy
241 						 * to prevent encrypting page
242 						 * cache pages */
243 #define	RQ_VICTIM	(5)			/* about to be shut down */
244 #define	RQ_BUSY		(6)			/* request is busy */
245 #define	RQ_DATA		(7)			/* request has data */
246 	unsigned long		rq_flags;	/* flags field */
247 	ktime_t			rq_qtime;	/* enqueue time */
248 
249 	void *			rq_argp;	/* decoded arguments */
250 	void *			rq_resp;	/* xdr'd results */
251 	__be32			*rq_accept_statp;
252 	void *			rq_auth_data;	/* flavor-specific data */
253 	__be32			rq_auth_stat;	/* authentication status */
254 	int			rq_auth_slack;	/* extra space xdr code
255 						 * should leave in head
256 						 * for krb5i, krb5p.
257 						 */
258 	int			rq_reserved;	/* space on socket outq
259 						 * reserved for this request
260 						 */
261 	ktime_t			rq_stime;	/* start time */
262 
263 	struct cache_req	rq_chandle;	/* handle passed to caches for
264 						 * request delaying
265 						 */
266 	/* Catering to nfsd */
267 	struct auth_domain *	rq_client;	/* RPC peer info */
268 	struct auth_domain *	rq_gssclient;	/* "gss/"-style peer info */
269 	struct svc_cacherep *	rq_cacherep;	/* cache info */
270 	struct task_struct	*rq_task;	/* service thread */
271 	struct net		*rq_bc_net;	/* pointer to backchannel's
272 						 * net namespace
273 						 */
274 	void **			rq_lease_breaker; /* The v4 client breaking a lease */
275 };
276 
277 #define SVC_NET(rqst) (rqst->rq_xprt ? rqst->rq_xprt->xpt_net : rqst->rq_bc_net)
278 
279 /*
280  * Rigorous type checking on sockaddr type conversions
281  */
282 static inline struct sockaddr_in *svc_addr_in(const struct svc_rqst *rqst)
283 {
284 	return (struct sockaddr_in *) &rqst->rq_addr;
285 }
286 
287 static inline struct sockaddr_in6 *svc_addr_in6(const struct svc_rqst *rqst)
288 {
289 	return (struct sockaddr_in6 *) &rqst->rq_addr;
290 }
291 
292 static inline struct sockaddr *svc_addr(const struct svc_rqst *rqst)
293 {
294 	return (struct sockaddr *) &rqst->rq_addr;
295 }
296 
297 static inline struct sockaddr_in *svc_daddr_in(const struct svc_rqst *rqst)
298 {
299 	return (struct sockaddr_in *) &rqst->rq_daddr;
300 }
301 
302 static inline struct sockaddr_in6 *svc_daddr_in6(const struct svc_rqst *rqst)
303 {
304 	return (struct sockaddr_in6 *) &rqst->rq_daddr;
305 }
306 
307 static inline struct sockaddr *svc_daddr(const struct svc_rqst *rqst)
308 {
309 	return (struct sockaddr *) &rqst->rq_daddr;
310 }
311 
312 struct svc_deferred_req {
313 	u32			prot;	/* protocol (UDP or TCP) */
314 	struct svc_xprt		*xprt;
315 	struct sockaddr_storage	addr;	/* where reply must go */
316 	size_t			addrlen;
317 	struct sockaddr_storage	daddr;	/* where reply must come from */
318 	size_t			daddrlen;
319 	void			*xprt_ctxt;
320 	struct cache_deferred_req handle;
321 	int			argslen;
322 	__be32			args[];
323 };
324 
325 struct svc_process_info {
326 	union {
327 		int  (*dispatch)(struct svc_rqst *rqstp);
328 		struct {
329 			unsigned int lovers;
330 			unsigned int hivers;
331 		} mismatch;
332 	};
333 };
334 
335 /*
336  * List of RPC programs on the same transport endpoint
337  */
338 struct svc_program {
339 	struct svc_program *	pg_next;	/* other programs (same xprt) */
340 	u32			pg_prog;	/* program number */
341 	unsigned int		pg_lovers;	/* lowest version */
342 	unsigned int		pg_hivers;	/* highest version */
343 	unsigned int		pg_nvers;	/* number of versions */
344 	const struct svc_version **pg_vers;	/* version array */
345 	char *			pg_name;	/* service name */
346 	char *			pg_class;	/* class name: services sharing authentication */
347 	struct svc_stat *	pg_stats;	/* rpc statistics */
348 	int			(*pg_authenticate)(struct svc_rqst *);
349 	__be32			(*pg_init_request)(struct svc_rqst *,
350 						   const struct svc_program *,
351 						   struct svc_process_info *);
352 	int			(*pg_rpcbind_set)(struct net *net,
353 						  const struct svc_program *,
354 						  u32 version, int family,
355 						  unsigned short proto,
356 						  unsigned short port);
357 };
358 
359 /*
360  * RPC program version
361  */
362 struct svc_version {
363 	u32			vs_vers;	/* version number */
364 	u32			vs_nproc;	/* number of procedures */
365 	const struct svc_procedure *vs_proc;	/* per-procedure info */
366 	unsigned long __percpu	*vs_count;	/* call counts */
367 	u32			vs_xdrsize;	/* xdrsize needed for this version */
368 
369 	/* Don't register with rpcbind */
370 	bool			vs_hidden;
371 
372 	/* Don't care if the rpcbind registration fails */
373 	bool			vs_rpcb_optnl;
374 
375 	/* Need xprt with congestion control */
376 	bool			vs_need_cong_ctrl;
377 
378 	/* Dispatch function */
379 	int			(*vs_dispatch)(struct svc_rqst *rqstp);
380 };
381 
382 /*
383  * RPC procedure info
384  */
385 struct svc_procedure {
386 	/* process the request: */
387 	__be32			(*pc_func)(struct svc_rqst *);
388 	/* XDR decode args: */
389 	bool			(*pc_decode)(struct svc_rqst *rqstp,
390 					     struct xdr_stream *xdr);
391 	/* XDR encode result: */
392 	bool			(*pc_encode)(struct svc_rqst *rqstp,
393 					     struct xdr_stream *xdr);
394 	/* XDR free result: */
395 	void			(*pc_release)(struct svc_rqst *);
396 	unsigned int		pc_argsize;	/* argument struct size */
397 	unsigned int		pc_argzero;	/* how much of argument to clear */
398 	unsigned int		pc_ressize;	/* result struct size */
399 	unsigned int		pc_cachetype;	/* cache info (NFS) */
400 	unsigned int		pc_xdrressize;	/* maximum size of XDR reply */
401 	const char *		pc_name;	/* for display */
402 };
403 
404 /*
405  * Function prototypes.
406  */
407 int svc_rpcb_setup(struct svc_serv *serv, struct net *net);
408 void svc_rpcb_cleanup(struct svc_serv *serv, struct net *net);
409 int svc_bind(struct svc_serv *serv, struct net *net);
410 struct svc_serv *svc_create(struct svc_program *, unsigned int,
411 			    int (*threadfn)(void *data));
412 struct svc_rqst *svc_rqst_alloc(struct svc_serv *serv,
413 					struct svc_pool *pool, int node);
414 bool		   svc_rqst_replace_page(struct svc_rqst *rqstp,
415 					 struct page *page);
416 void		   svc_rqst_release_pages(struct svc_rqst *rqstp);
417 void		   svc_rqst_free(struct svc_rqst *);
418 void		   svc_exit_thread(struct svc_rqst *);
419 struct svc_serv *  svc_create_pooled(struct svc_program *, unsigned int,
420 				     int (*threadfn)(void *data));
421 int		   svc_set_num_threads(struct svc_serv *, struct svc_pool *, int);
422 int		   svc_pool_stats_open(struct svc_serv *serv, struct file *file);
423 void		   svc_process(struct svc_rqst *rqstp);
424 int		   bc_svc_process(struct svc_serv *, struct rpc_rqst *,
425 			struct svc_rqst *);
426 int		   svc_register(const struct svc_serv *, struct net *, const int,
427 				const unsigned short, const unsigned short);
428 
429 void		   svc_wake_up(struct svc_serv *);
430 void		   svc_reserve(struct svc_rqst *rqstp, int space);
431 struct svc_pool   *svc_pool_for_cpu(struct svc_serv *serv);
432 char *		   svc_print_addr(struct svc_rqst *, char *, size_t);
433 const char *	   svc_proc_name(const struct svc_rqst *rqstp);
434 int		   svc_encode_result_payload(struct svc_rqst *rqstp,
435 					     unsigned int offset,
436 					     unsigned int length);
437 unsigned int	   svc_fill_write_vector(struct svc_rqst *rqstp,
438 					 struct xdr_buf *payload);
439 char		  *svc_fill_symlink_pathname(struct svc_rqst *rqstp,
440 					     struct kvec *first, void *p,
441 					     size_t total);
442 __be32		   svc_generic_init_request(struct svc_rqst *rqstp,
443 					    const struct svc_program *progp,
444 					    struct svc_process_info *procinfo);
445 int		   svc_generic_rpcbind_set(struct net *net,
446 					   const struct svc_program *progp,
447 					   u32 version, int family,
448 					   unsigned short proto,
449 					   unsigned short port);
450 int		   svc_rpcbind_set_version(struct net *net,
451 					   const struct svc_program *progp,
452 					   u32 version, int family,
453 					   unsigned short proto,
454 					   unsigned short port);
455 
456 #define	RPC_MAX_ADDRBUFLEN	(63U)
457 
458 /*
459  * When we want to reduce the size of the reserved space in the response
460  * buffer, we need to take into account the size of any checksum data that
461  * may be at the end of the packet. This is difficult to determine exactly
462  * for all cases without actually generating the checksum, so we just use a
463  * static value.
464  */
465 static inline void svc_reserve_auth(struct svc_rqst *rqstp, int space)
466 {
467 	svc_reserve(rqstp, space + rqstp->rq_auth_slack);
468 }
469 
470 /**
471  * svcxdr_init_decode - Prepare an xdr_stream for Call decoding
472  * @rqstp: controlling server RPC transaction context
473  *
474  */
475 static inline void svcxdr_init_decode(struct svc_rqst *rqstp)
476 {
477 	struct xdr_stream *xdr = &rqstp->rq_arg_stream;
478 	struct xdr_buf *buf = &rqstp->rq_arg;
479 	struct kvec *argv = buf->head;
480 
481 	WARN_ON(buf->len != buf->head->iov_len + buf->page_len + buf->tail->iov_len);
482 	buf->len = buf->head->iov_len + buf->page_len + buf->tail->iov_len;
483 
484 	xdr_init_decode(xdr, buf, argv->iov_base, NULL);
485 	xdr_set_scratch_page(xdr, rqstp->rq_scratch_page);
486 }
487 
488 /**
489  * svcxdr_init_encode - Prepare an xdr_stream for svc Reply encoding
490  * @rqstp: controlling server RPC transaction context
491  *
492  */
493 static inline void svcxdr_init_encode(struct svc_rqst *rqstp)
494 {
495 	struct xdr_stream *xdr = &rqstp->rq_res_stream;
496 	struct xdr_buf *buf = &rqstp->rq_res;
497 	struct kvec *resv = buf->head;
498 
499 	xdr_reset_scratch_buffer(xdr);
500 
501 	xdr->buf = buf;
502 	xdr->iov = resv;
503 	xdr->p   = resv->iov_base + resv->iov_len;
504 	xdr->end = resv->iov_base + PAGE_SIZE;
505 	buf->len = resv->iov_len;
506 	xdr->page_ptr = buf->pages - 1;
507 	buf->buflen = PAGE_SIZE * (rqstp->rq_page_end - buf->pages);
508 	xdr->rqst = NULL;
509 }
510 
511 /**
512  * svcxdr_set_auth_slack -
513  * @rqstp: RPC transaction
514  * @slack: buffer space to reserve for the transaction's security flavor
515  *
516  * Set the request's slack space requirement, and set aside that much
517  * space in the rqstp's rq_res.head for use when the auth wraps the Reply.
518  */
519 static inline void svcxdr_set_auth_slack(struct svc_rqst *rqstp, int slack)
520 {
521 	struct xdr_stream *xdr = &rqstp->rq_res_stream;
522 	struct xdr_buf *buf = &rqstp->rq_res;
523 	struct kvec *resv = buf->head;
524 
525 	rqstp->rq_auth_slack = slack;
526 
527 	xdr->end -= XDR_QUADLEN(slack);
528 	buf->buflen -= rqstp->rq_auth_slack;
529 
530 	WARN_ON(xdr->iov != resv);
531 	WARN_ON(xdr->p > xdr->end);
532 }
533 
534 /**
535  * svcxdr_set_accept_stat - Reserve space for the accept_stat field
536  * @rqstp: RPC transaction context
537  *
538  * Return values:
539  *   %true: Success
540  *   %false: No response buffer space was available
541  */
542 static inline bool svcxdr_set_accept_stat(struct svc_rqst *rqstp)
543 {
544 	struct xdr_stream *xdr = &rqstp->rq_res_stream;
545 
546 	rqstp->rq_accept_statp = xdr_reserve_space(xdr, XDR_UNIT);
547 	if (unlikely(!rqstp->rq_accept_statp))
548 		return false;
549 	*rqstp->rq_accept_statp = rpc_success;
550 	return true;
551 }
552 
553 #endif /* SUNRPC_SVC_H */
554