SUNRPC: Kill rpc_clnt->cl_oneshot
[GitHub/moto-9609/android_kernel_motorola_exynos9610.git] / net / sunrpc / rpcb_clnt.c
1 /*
2 * In-kernel rpcbind client supporting versions 2, 3, and 4 of the rpcbind
3 * protocol
4 *
5 * Based on RFC 1833: "Binding Protocols for ONC RPC Version 2" and
6 * RFC 3530: "Network File System (NFS) version 4 Protocol"
7 *
8 * Original: Gilles Quillard, Bull Open Source, 2005 <gilles.quillard@bull.net>
9 * Updated: Chuck Lever, Oracle Corporation, 2007 <chuck.lever@oracle.com>
10 *
11 * Descended from net/sunrpc/pmap_clnt.c,
12 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
13 */
14
15 #include <linux/types.h>
16 #include <linux/socket.h>
17 #include <linux/kernel.h>
18 #include <linux/errno.h>
19
20 #include <linux/sunrpc/clnt.h>
21 #include <linux/sunrpc/sched.h>
22
23 #ifdef RPC_DEBUG
24 # define RPCDBG_FACILITY RPCDBG_BIND
25 #endif
26
27 #define RPCBIND_PROGRAM (100000u)
28 #define RPCBIND_PORT (111u)
29
30 enum {
31 RPCBPROC_NULL,
32 RPCBPROC_SET,
33 RPCBPROC_UNSET,
34 RPCBPROC_GETPORT,
35 RPCBPROC_GETADDR = 3, /* alias for GETPORT */
36 RPCBPROC_DUMP,
37 RPCBPROC_CALLIT,
38 RPCBPROC_BCAST = 5, /* alias for CALLIT */
39 RPCBPROC_GETTIME,
40 RPCBPROC_UADDR2TADDR,
41 RPCBPROC_TADDR2UADDR,
42 RPCBPROC_GETVERSADDR,
43 RPCBPROC_INDIRECT,
44 RPCBPROC_GETADDRLIST,
45 RPCBPROC_GETSTAT,
46 };
47
48 #define RPCB_HIGHPROC_2 RPCBPROC_CALLIT
49 #define RPCB_HIGHPROC_3 RPCBPROC_TADDR2UADDR
50 #define RPCB_HIGHPROC_4 RPCBPROC_GETSTAT
51
52 /*
53 * r_addr
54 *
55 * Quoting RFC 3530, section 2.2:
56 *
57 * For TCP over IPv4 and for UDP over IPv4, the format of r_addr is the
58 * US-ASCII string:
59 *
60 * h1.h2.h3.h4.p1.p2
61 *
62 * The prefix, "h1.h2.h3.h4", is the standard textual form for
63 * representing an IPv4 address, which is always four octets long.
64 * Assuming big-endian ordering, h1, h2, h3, and h4, are respectively,
65 * the first through fourth octets each converted to ASCII-decimal.
66 * Assuming big-endian ordering, p1 and p2 are, respectively, the first
67 * and second octets each converted to ASCII-decimal. For example, if a
68 * host, in big-endian order, has an address of 0x0A010307 and there is
69 * a service listening on, in big endian order, port 0x020F (decimal
70 * 527), then the complete universal address is "10.1.3.7.2.15".
71 *
72 * ...
73 *
74 * For TCP over IPv6 and for UDP over IPv6, the format of r_addr is the
75 * US-ASCII string:
76 *
77 * x1:x2:x3:x4:x5:x6:x7:x8.p1.p2
78 *
79 * The suffix "p1.p2" is the service port, and is computed the same way
80 * as with universal addresses for TCP and UDP over IPv4. The prefix,
81 * "x1:x2:x3:x4:x5:x6:x7:x8", is the standard textual form for
82 * representing an IPv6 address as defined in Section 2.2 of [RFC2373].
83 * Additionally, the two alternative forms specified in Section 2.2 of
84 * [RFC2373] are also acceptable.
85 *
86 * XXX: Currently this implementation does not explicitly convert the
87 * stored address to US-ASCII on non-ASCII systems.
88 */
89 #define RPCB_MAXADDRLEN (128u)
90
91 /*
92 * r_netid
93 *
94 * Quoting RFC 3530, section 2.2:
95 *
96 * For TCP over IPv4 the value of r_netid is the string "tcp". For UDP
97 * over IPv4 the value of r_netid is the string "udp".
98 *
99 * ...
100 *
101 * For TCP over IPv6 the value of r_netid is the string "tcp6". For UDP
102 * over IPv6 the value of r_netid is the string "udp6".
103 */
104 #define RPCB_NETID_UDP "\165\144\160" /* "udp" */
105 #define RPCB_NETID_TCP "\164\143\160" /* "tcp" */
106 #define RPCB_NETID_UDP6 "\165\144\160\066" /* "udp6" */
107 #define RPCB_NETID_TCP6 "\164\143\160\066" /* "tcp6" */
108
109 #define RPCB_MAXNETIDLEN (4u)
110
111 /*
112 * r_owner
113 *
114 * The "owner" is allowed to unset a service in the rpcbind database.
115 * We always use the following (arbitrary) fixed string.
116 */
117 #define RPCB_OWNER_STRING "rpcb"
118 #define RPCB_MAXOWNERLEN sizeof(RPCB_OWNER_STRING)
119
120 static void rpcb_getport_done(struct rpc_task *, void *);
121 extern struct rpc_program rpcb_program;
122
123 struct rpcbind_args {
124 struct rpc_xprt * r_xprt;
125
126 u32 r_prog;
127 u32 r_vers;
128 u32 r_prot;
129 unsigned short r_port;
130 char * r_netid;
131 char r_addr[RPCB_MAXADDRLEN];
132 char * r_owner;
133 };
134
135 static struct rpc_procinfo rpcb_procedures2[];
136 static struct rpc_procinfo rpcb_procedures3[];
137
138 static struct rpcb_info {
139 int rpc_vers;
140 struct rpc_procinfo * rpc_proc;
141 } rpcb_next_version[];
142
143 static void rpcb_getport_prepare(struct rpc_task *task, void *calldata)
144 {
145 struct rpcbind_args *map = calldata;
146 struct rpc_xprt *xprt = map->r_xprt;
147 struct rpc_message msg = {
148 .rpc_proc = rpcb_next_version[xprt->bind_index].rpc_proc,
149 .rpc_argp = map,
150 .rpc_resp = &map->r_port,
151 };
152
153 rpc_call_setup(task, &msg, 0);
154 }
155
156 static void rpcb_map_release(void *data)
157 {
158 struct rpcbind_args *map = data;
159
160 xprt_put(map->r_xprt);
161 kfree(map);
162 }
163
164 static const struct rpc_call_ops rpcb_getport_ops = {
165 .rpc_call_prepare = rpcb_getport_prepare,
166 .rpc_call_done = rpcb_getport_done,
167 .rpc_release = rpcb_map_release,
168 };
169
170 static void rpcb_wake_rpcbind_waiters(struct rpc_xprt *xprt, int status)
171 {
172 xprt_clear_binding(xprt);
173 rpc_wake_up_status(&xprt->binding, status);
174 }
175
176 static struct rpc_clnt *rpcb_create(char *hostname, struct sockaddr *srvaddr,
177 int proto, int version, int privileged)
178 {
179 struct rpc_create_args args = {
180 .protocol = proto,
181 .address = srvaddr,
182 .addrsize = sizeof(struct sockaddr_in),
183 .servername = hostname,
184 .program = &rpcb_program,
185 .version = version,
186 .authflavor = RPC_AUTH_UNIX,
187 .flags = RPC_CLNT_CREATE_NOPING,
188 };
189
190 ((struct sockaddr_in *)srvaddr)->sin_port = htons(RPCBIND_PORT);
191 if (!privileged)
192 args.flags |= RPC_CLNT_CREATE_NONPRIVPORT;
193 return rpc_create(&args);
194 }
195
196 /**
197 * rpcb_register - set or unset a port registration with the local rpcbind svc
198 * @prog: RPC program number to bind
199 * @vers: RPC version number to bind
200 * @prot: transport protocol to use to make this request
201 * @port: port value to register
202 * @okay: result code
203 *
204 * port == 0 means unregister, port != 0 means register.
205 *
206 * This routine supports only rpcbind version 2.
207 */
208 int rpcb_register(u32 prog, u32 vers, int prot, unsigned short port, int *okay)
209 {
210 struct sockaddr_in sin = {
211 .sin_family = AF_INET,
212 .sin_addr.s_addr = htonl(INADDR_LOOPBACK),
213 };
214 struct rpcbind_args map = {
215 .r_prog = prog,
216 .r_vers = vers,
217 .r_prot = prot,
218 .r_port = port,
219 };
220 struct rpc_message msg = {
221 .rpc_proc = &rpcb_procedures2[port ?
222 RPCBPROC_SET : RPCBPROC_UNSET],
223 .rpc_argp = &map,
224 .rpc_resp = okay,
225 };
226 struct rpc_clnt *rpcb_clnt;
227 int error = 0;
228
229 dprintk("RPC: %sregistering (%u, %u, %d, %u) with local "
230 "rpcbind\n", (port ? "" : "un"),
231 prog, vers, prot, port);
232
233 rpcb_clnt = rpcb_create("localhost", (struct sockaddr *) &sin,
234 IPPROTO_UDP, 2, 1);
235 if (IS_ERR(rpcb_clnt))
236 return PTR_ERR(rpcb_clnt);
237
238 error = rpc_call_sync(rpcb_clnt, &msg, 0);
239
240 rpc_shutdown_client(rpcb_clnt);
241 if (error < 0)
242 printk(KERN_WARNING "RPC: failed to contact local rpcbind "
243 "server (errno %d).\n", -error);
244 dprintk("RPC: registration status %d/%d\n", error, *okay);
245
246 return error;
247 }
248
249 #ifdef CONFIG_ROOT_NFS
250 /**
251 * rpcb_getport_external - obtain the port for an RPC service on a given host
252 * @sin: address of remote peer
253 * @prog: RPC program number to bind
254 * @vers: RPC version number to bind
255 * @prot: transport protocol to use to make this request
256 *
257 * Called from outside the RPC client in a synchronous task context.
258 *
259 * For now, this supports only version 2 queries, but is used only by
260 * mount_clnt for NFS_ROOT.
261 */
262 int rpcb_getport_external(struct sockaddr_in *sin, __u32 prog,
263 __u32 vers, int prot)
264 {
265 struct rpcbind_args map = {
266 .r_prog = prog,
267 .r_vers = vers,
268 .r_prot = prot,
269 .r_port = 0,
270 };
271 struct rpc_message msg = {
272 .rpc_proc = &rpcb_procedures2[RPCBPROC_GETPORT],
273 .rpc_argp = &map,
274 .rpc_resp = &map.r_port,
275 };
276 struct rpc_clnt *rpcb_clnt;
277 char hostname[40];
278 int status;
279
280 dprintk("RPC: rpcb_getport_external(%u.%u.%u.%u, %u, %u, %d)\n",
281 NIPQUAD(sin->sin_addr.s_addr), prog, vers, prot);
282
283 sprintf(hostname, "%u.%u.%u.%u", NIPQUAD(sin->sin_addr.s_addr));
284 rpcb_clnt = rpcb_create(hostname, (struct sockaddr *)sin, prot, 2, 0);
285 if (IS_ERR(rpcb_clnt))
286 return PTR_ERR(rpcb_clnt);
287
288 status = rpc_call_sync(rpcb_clnt, &msg, 0);
289 rpc_shutdown_client(rpcb_clnt);
290
291 if (status >= 0) {
292 if (map.r_port != 0)
293 return map.r_port;
294 status = -EACCES;
295 }
296 return status;
297 }
298 #endif
299
300 /**
301 * rpcb_getport - obtain the port for a given RPC service on a given host
302 * @task: task that is waiting for portmapper request
303 *
304 * This one can be called for an ongoing RPC request, and can be used in
305 * an async (rpciod) context.
306 */
307 void rpcb_getport(struct rpc_task *task)
308 {
309 struct rpc_clnt *clnt = task->tk_client;
310 int bind_version;
311 struct rpc_xprt *xprt = task->tk_xprt;
312 struct rpc_clnt *rpcb_clnt;
313 static struct rpcbind_args *map;
314 struct rpc_task *child;
315 struct sockaddr addr;
316 int status;
317
318 dprintk("RPC: %5u rpcb_getport(%s, %u, %u, %d)\n",
319 task->tk_pid, clnt->cl_server,
320 clnt->cl_prog, clnt->cl_vers, xprt->prot);
321
322 /* Autobind on cloned rpc clients is discouraged */
323 BUG_ON(clnt->cl_parent != clnt);
324
325 if (xprt_test_and_set_binding(xprt)) {
326 status = -EACCES; /* tell caller to check again */
327 dprintk("RPC: %5u rpcb_getport waiting for another binder\n",
328 task->tk_pid);
329 goto bailout_nowake;
330 }
331
332 /* Put self on queue before sending rpcbind request, in case
333 * rpcb_getport_done completes before we return from rpc_run_task */
334 rpc_sleep_on(&xprt->binding, task, NULL, NULL);
335
336 /* Someone else may have bound if we slept */
337 if (xprt_bound(xprt)) {
338 status = 0;
339 dprintk("RPC: %5u rpcb_getport already bound\n", task->tk_pid);
340 goto bailout_nofree;
341 }
342
343 if (rpcb_next_version[xprt->bind_index].rpc_proc == NULL) {
344 xprt->bind_index = 0;
345 status = -EACCES; /* tell caller to try again later */
346 dprintk("RPC: %5u rpcb_getport no more getport versions "
347 "available\n", task->tk_pid);
348 goto bailout_nofree;
349 }
350 bind_version = rpcb_next_version[xprt->bind_index].rpc_vers;
351
352 dprintk("RPC: %5u rpcb_getport trying rpcbind version %u\n",
353 task->tk_pid, bind_version);
354
355 map = kzalloc(sizeof(struct rpcbind_args), GFP_ATOMIC);
356 if (!map) {
357 status = -ENOMEM;
358 dprintk("RPC: %5u rpcb_getport no memory available\n",
359 task->tk_pid);
360 goto bailout_nofree;
361 }
362 map->r_prog = clnt->cl_prog;
363 map->r_vers = clnt->cl_vers;
364 map->r_prot = xprt->prot;
365 map->r_port = 0;
366 map->r_xprt = xprt_get(xprt);
367 map->r_netid = (xprt->prot == IPPROTO_TCP) ? RPCB_NETID_TCP :
368 RPCB_NETID_UDP;
369 memcpy(&map->r_addr, rpc_peeraddr2str(clnt, RPC_DISPLAY_ADDR),
370 sizeof(map->r_addr));
371 map->r_owner = RPCB_OWNER_STRING; /* ignored for GETADDR */
372
373 rpc_peeraddr(clnt, (void *)&addr, sizeof(addr));
374 rpcb_clnt = rpcb_create(clnt->cl_server, &addr, xprt->prot, bind_version, 0);
375 if (IS_ERR(rpcb_clnt)) {
376 status = PTR_ERR(rpcb_clnt);
377 dprintk("RPC: %5u rpcb_getport rpcb_create failed, error %ld\n",
378 task->tk_pid, PTR_ERR(rpcb_clnt));
379 goto bailout;
380 }
381
382 child = rpc_run_task(rpcb_clnt, RPC_TASK_ASYNC, &rpcb_getport_ops, map);
383 rpc_destroy_client(rpcb_clnt);
384 if (IS_ERR(child)) {
385 status = -EIO;
386 dprintk("RPC: %5u rpcb_getport rpc_run_task failed\n",
387 task->tk_pid);
388 goto bailout_nofree;
389 }
390 rpc_put_task(child);
391
392 task->tk_xprt->stat.bind_count++;
393 return;
394
395 bailout:
396 kfree(map);
397 xprt_put(xprt);
398 bailout_nofree:
399 rpcb_wake_rpcbind_waiters(xprt, status);
400 bailout_nowake:
401 task->tk_status = status;
402 }
403
404 /*
405 * Rpcbind child task calls this callback via tk_exit.
406 */
407 static void rpcb_getport_done(struct rpc_task *child, void *data)
408 {
409 struct rpcbind_args *map = data;
410 struct rpc_xprt *xprt = map->r_xprt;
411 int status = child->tk_status;
412
413 /* rpcbind server doesn't support this rpcbind protocol version */
414 if (status == -EPROTONOSUPPORT)
415 xprt->bind_index++;
416
417 if (status < 0) {
418 /* rpcbind server not available on remote host? */
419 xprt->ops->set_port(xprt, 0);
420 } else if (map->r_port == 0) {
421 /* Requested RPC service wasn't registered on remote host */
422 xprt->ops->set_port(xprt, 0);
423 status = -EACCES;
424 } else {
425 /* Succeeded */
426 xprt->ops->set_port(xprt, map->r_port);
427 xprt_set_bound(xprt);
428 status = 0;
429 }
430
431 dprintk("RPC: %5u rpcb_getport_done(status %d, port %u)\n",
432 child->tk_pid, status, map->r_port);
433
434 rpcb_wake_rpcbind_waiters(xprt, status);
435 }
436
437 static int rpcb_encode_mapping(struct rpc_rqst *req, __be32 *p,
438 struct rpcbind_args *rpcb)
439 {
440 dprintk("RPC: rpcb_encode_mapping(%u, %u, %d, %u)\n",
441 rpcb->r_prog, rpcb->r_vers, rpcb->r_prot, rpcb->r_port);
442 *p++ = htonl(rpcb->r_prog);
443 *p++ = htonl(rpcb->r_vers);
444 *p++ = htonl(rpcb->r_prot);
445 *p++ = htonl(rpcb->r_port);
446
447 req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
448 return 0;
449 }
450
451 static int rpcb_decode_getport(struct rpc_rqst *req, __be32 *p,
452 unsigned short *portp)
453 {
454 *portp = (unsigned short) ntohl(*p++);
455 dprintk("RPC: rpcb_decode_getport result %u\n",
456 *portp);
457 return 0;
458 }
459
460 static int rpcb_decode_set(struct rpc_rqst *req, __be32 *p,
461 unsigned int *boolp)
462 {
463 *boolp = (unsigned int) ntohl(*p++);
464 dprintk("RPC: rpcb_decode_set result %u\n",
465 *boolp);
466 return 0;
467 }
468
469 static int rpcb_encode_getaddr(struct rpc_rqst *req, __be32 *p,
470 struct rpcbind_args *rpcb)
471 {
472 dprintk("RPC: rpcb_encode_getaddr(%u, %u, %s)\n",
473 rpcb->r_prog, rpcb->r_vers, rpcb->r_addr);
474 *p++ = htonl(rpcb->r_prog);
475 *p++ = htonl(rpcb->r_vers);
476
477 p = xdr_encode_string(p, rpcb->r_netid);
478 p = xdr_encode_string(p, rpcb->r_addr);
479 p = xdr_encode_string(p, rpcb->r_owner);
480
481 req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
482
483 return 0;
484 }
485
486 static int rpcb_decode_getaddr(struct rpc_rqst *req, __be32 *p,
487 unsigned short *portp)
488 {
489 char *addr;
490 int addr_len, c, i, f, first, val;
491
492 *portp = 0;
493 addr_len = (unsigned int) ntohl(*p++);
494 if (addr_len > RPCB_MAXADDRLEN) /* sanity */
495 return -EINVAL;
496
497 dprintk("RPC: rpcb_decode_getaddr returned string: '%s'\n",
498 (char *) p);
499
500 addr = (char *)p;
501 val = 0;
502 first = 1;
503 f = 1;
504 for (i = addr_len - 1; i > 0; i--) {
505 c = addr[i];
506 if (c >= '0' && c <= '9') {
507 val += (c - '0') * f;
508 f *= 10;
509 } else if (c == '.') {
510 if (first) {
511 *portp = val;
512 val = first = 0;
513 f = 1;
514 } else {
515 *portp |= (val << 8);
516 break;
517 }
518 }
519 }
520
521 dprintk("RPC: rpcb_decode_getaddr port=%u\n", *portp);
522 return 0;
523 }
524
525 #define RPCB_program_sz (1u)
526 #define RPCB_version_sz (1u)
527 #define RPCB_protocol_sz (1u)
528 #define RPCB_port_sz (1u)
529 #define RPCB_boolean_sz (1u)
530
531 #define RPCB_netid_sz (1+XDR_QUADLEN(RPCB_MAXNETIDLEN))
532 #define RPCB_addr_sz (1+XDR_QUADLEN(RPCB_MAXADDRLEN))
533 #define RPCB_ownerstring_sz (1+XDR_QUADLEN(RPCB_MAXOWNERLEN))
534
535 #define RPCB_mappingargs_sz RPCB_program_sz+RPCB_version_sz+ \
536 RPCB_protocol_sz+RPCB_port_sz
537 #define RPCB_getaddrargs_sz RPCB_program_sz+RPCB_version_sz+ \
538 RPCB_netid_sz+RPCB_addr_sz+ \
539 RPCB_ownerstring_sz
540
541 #define RPCB_setres_sz RPCB_boolean_sz
542 #define RPCB_getportres_sz RPCB_port_sz
543
544 /*
545 * Note that RFC 1833 does not put any size restrictions on the
546 * address string returned by the remote rpcbind database.
547 */
548 #define RPCB_getaddrres_sz RPCB_addr_sz
549
550 #define PROC(proc, argtype, restype) \
551 [RPCBPROC_##proc] = { \
552 .p_proc = RPCBPROC_##proc, \
553 .p_encode = (kxdrproc_t) rpcb_encode_##argtype, \
554 .p_decode = (kxdrproc_t) rpcb_decode_##restype, \
555 .p_arglen = RPCB_##argtype##args_sz, \
556 .p_replen = RPCB_##restype##res_sz, \
557 .p_statidx = RPCBPROC_##proc, \
558 .p_timer = 0, \
559 .p_name = #proc, \
560 }
561
562 /*
563 * Not all rpcbind procedures described in RFC 1833 are implemented
564 * since the Linux kernel RPC code requires only these.
565 */
566 static struct rpc_procinfo rpcb_procedures2[] = {
567 PROC(SET, mapping, set),
568 PROC(UNSET, mapping, set),
569 PROC(GETADDR, mapping, getport),
570 };
571
572 static struct rpc_procinfo rpcb_procedures3[] = {
573 PROC(SET, mapping, set),
574 PROC(UNSET, mapping, set),
575 PROC(GETADDR, getaddr, getaddr),
576 };
577
578 static struct rpc_procinfo rpcb_procedures4[] = {
579 PROC(SET, mapping, set),
580 PROC(UNSET, mapping, set),
581 PROC(GETVERSADDR, getaddr, getaddr),
582 };
583
584 static struct rpcb_info rpcb_next_version[] = {
585 #ifdef CONFIG_SUNRPC_BIND34
586 { 4, &rpcb_procedures4[RPCBPROC_GETVERSADDR] },
587 { 3, &rpcb_procedures3[RPCBPROC_GETADDR] },
588 #endif
589 { 2, &rpcb_procedures2[RPCBPROC_GETPORT] },
590 { 0, NULL },
591 };
592
593 static struct rpc_version rpcb_version2 = {
594 .number = 2,
595 .nrprocs = RPCB_HIGHPROC_2,
596 .procs = rpcb_procedures2
597 };
598
599 static struct rpc_version rpcb_version3 = {
600 .number = 3,
601 .nrprocs = RPCB_HIGHPROC_3,
602 .procs = rpcb_procedures3
603 };
604
605 static struct rpc_version rpcb_version4 = {
606 .number = 4,
607 .nrprocs = RPCB_HIGHPROC_4,
608 .procs = rpcb_procedures4
609 };
610
611 static struct rpc_version *rpcb_version[] = {
612 NULL,
613 NULL,
614 &rpcb_version2,
615 &rpcb_version3,
616 &rpcb_version4
617 };
618
619 static struct rpc_stat rpcb_stats;
620
621 struct rpc_program rpcb_program = {
622 .name = "rpcbind",
623 .number = RPCBIND_PROGRAM,
624 .nrvers = ARRAY_SIZE(rpcb_version),
625 .version = rpcb_version,
626 .stats = &rpcb_stats,
627 };