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/module.h>
  16
  17 #include <linux/types.h>
  18 #include <linux/socket.h>
  19 #include <linux/un.h>
  20 #include <linux/in.h>
  21 #include <linux/in6.h>
  22 #include <linux/kernel.h>
  23 #include <linux/errno.h>
  24 #include <linux/mutex.h>
  25 #include <linux/slab.h>
  26 #include <net/ipv6.h>
  27
  28 #include <linux/sunrpc/clnt.h>
  29 #include <linux/sunrpc/sched.h>
  30 #include <linux/sunrpc/xprtsock.h>
  31
  32 #ifdef RPC_DEBUG
  33 # define RPCDBG_FACILITY        RPCDBG_BIND
  34 #endif
  35
  36 #define RPCBIND_SOCK_PATHNAME   "/var/run/rpcbind.sock"
  37
  38 #define RPCBIND_PROGRAM         (100000u)
  39 #define RPCBIND_PORT            (111u)
  40
  41 #define RPCBVERS_2              (2u)
  42 #define RPCBVERS_3              (3u)
  43 #define RPCBVERS_4              (4u)
  44
  45 enum {
  46         RPCBPROC_NULL,
  47         RPCBPROC_SET,
  48         RPCBPROC_UNSET,
  49         RPCBPROC_GETPORT,
  50         RPCBPROC_GETADDR = 3,           /* alias for GETPORT */
  51         RPCBPROC_DUMP,
  52         RPCBPROC_CALLIT,
  53         RPCBPROC_BCAST = 5,             /* alias for CALLIT */
  54         RPCBPROC_GETTIME,
  55         RPCBPROC_UADDR2TADDR,
  56         RPCBPROC_TADDR2UADDR,
  57         RPCBPROC_GETVERSADDR,
  58         RPCBPROC_INDIRECT,
  59         RPCBPROC_GETADDRLIST,
  60         RPCBPROC_GETSTAT,
  61 };
  62
  63 /*
  64  * r_owner
  65  *
  66  * The "owner" is allowed to unset a service in the rpcbind database.
  67  *
  68  * For AF_LOCAL SET/UNSET requests, rpcbind treats this string as a
  69  * UID which it maps to a local user name via a password lookup.
  70  * In all other cases it is ignored.
  71  *
  72  * For SET/UNSET requests, user space provides a value, even for
  73  * network requests, and GETADDR uses an empty string.  We follow
  74  * those precedents here.
  75  */
  76 #define RPCB_OWNER_STRING       "0"
  77 #define RPCB_MAXOWNERLEN        sizeof(RPCB_OWNER_STRING)
  78
  79 /*
  80  * XDR data type sizes
  81  */
  82 #define RPCB_program_sz         (1)
  83 #define RPCB_version_sz         (1)
  84 #define RPCB_protocol_sz        (1)
  85 #define RPCB_port_sz            (1)
  86 #define RPCB_boolean_sz         (1)
  87
  88 #define RPCB_netid_sz           (1 + XDR_QUADLEN(RPCBIND_MAXNETIDLEN))
  89 #define RPCB_addr_sz            (1 + XDR_QUADLEN(RPCBIND_MAXUADDRLEN))
  90 #define RPCB_ownerstring_sz     (1 + XDR_QUADLEN(RPCB_MAXOWNERLEN))
  91
  92 /*
  93  * XDR argument and result sizes
  94  */
  95 #define RPCB_mappingargs_sz     (RPCB_program_sz + RPCB_version_sz + \
  96                                 RPCB_protocol_sz + RPCB_port_sz)
  97 #define RPCB_getaddrargs_sz     (RPCB_program_sz + RPCB_version_sz + \
  98                                 RPCB_netid_sz + RPCB_addr_sz + \
  99                                 RPCB_ownerstring_sz)
 100
 101 #define RPCB_getportres_sz      RPCB_port_sz
 102 #define RPCB_setres_sz          RPCB_boolean_sz
 103
 104 /*
 105  * Note that RFC 1833 does not put any size restrictions on the
 106  * address string returned by the remote rpcbind database.
 107  */
 108 #define RPCB_getaddrres_sz      RPCB_addr_sz
 109
 110 static void                     rpcb_getport_done(struct rpc_task *, void *);
 111 static void                     rpcb_map_release(void *data);
 112 static struct rpc_program       rpcb_program;
 113
 114 static struct rpc_clnt *        rpcb_local_clnt;
 115 static struct rpc_clnt *        rpcb_local_clnt4;
 116
 117 DEFINE_SPINLOCK(rpcb_clnt_lock);
 118 unsigned int                    rpcb_users;
 119
 120 struct rpcbind_args {
 121         struct rpc_xprt *       r_xprt;
 122
 123         u32                     r_prog;
 124         u32                     r_vers;
 125         u32                     r_prot;
 126         unsigned short          r_port;
 127         const char *            r_netid;
 128         const char *            r_addr;
 129         const char *            r_owner;
 130
 131         int                     r_status;
 132 };
 133
 134 static struct rpc_procinfo rpcb_procedures2[];
 135 static struct rpc_procinfo rpcb_procedures3[];
 136 static struct rpc_procinfo rpcb_procedures4[];
 137
 138 struct rpcb_info {
 139         u32                     rpc_vers;
 140         struct rpc_procinfo *   rpc_proc;
 141 };
 142
 143 static struct rpcb_info rpcb_next_version[];
 144 static struct rpcb_info rpcb_next_version6[];
 145
 146 static const struct rpc_call_ops rpcb_getport_ops = {
 147         .rpc_call_done          = rpcb_getport_done,
 148         .rpc_release            = rpcb_map_release,
 149 };
 150
 151 static void rpcb_wake_rpcbind_waiters(struct rpc_xprt *xprt, int status)
 152 {
 153         xprt_clear_binding(xprt);
 154         rpc_wake_up_status(&xprt->binding, status);
 155 }
 156
 157 static void rpcb_map_release(void *data)
 158 {
 159         struct rpcbind_args *map = data;
 160
 161         rpcb_wake_rpcbind_waiters(map->r_xprt, map->r_status);
 162         xprt_put(map->r_xprt);
 163         kfree(map->r_addr);
 164         kfree(map);
 165 }
 166
 167 static int rpcb_get_local(void)
 168 {
 169         int cnt;
 170
 171         spin_lock(&rpcb_clnt_lock);
 172         if (rpcb_users)
 173                 rpcb_users++;
 174         cnt = rpcb_users;
 175         spin_unlock(&rpcb_clnt_lock);
 176
 177         return cnt;
 178 }
 179
 180 void rpcb_put_local(void)
 181 {
 182         struct rpc_clnt *clnt = rpcb_local_clnt;
 183         struct rpc_clnt *clnt4 = rpcb_local_clnt4;
 184         int shutdown;
 185
 186         spin_lock(&rpcb_clnt_lock);
 187         if (--rpcb_users == 0) {
 188                 rpcb_local_clnt = NULL;
 189                 rpcb_local_clnt4 = NULL;
 190         }
 191         shutdown = !rpcb_users;
 192         spin_unlock(&rpcb_clnt_lock);
 193
 194         if (shutdown) {
 195                 /*
 196                  * cleanup_rpcb_clnt - remove xprtsock's sysctls, unregister
 197                  */
 198                 if (clnt4)
 199                         rpc_shutdown_client(clnt4);
 200                 if (clnt)
 201                         rpc_shutdown_client(clnt);
 202         }
 203 }
 204
 205 static void rpcb_set_local(struct rpc_clnt *clnt, struct rpc_clnt *clnt4)
 206 {
 207         /* Protected by rpcb_create_local_mutex */
 208         rpcb_local_clnt = clnt;
 209         rpcb_local_clnt4 = clnt4;
 210         smp_wmb();
 211         rpcb_users = 1;
 212         dprintk("RPC:       created new rpcb local clients (rpcb_local_clnt: "
 213                         "%p, rpcb_local_clnt4: %p)\n", rpcb_local_clnt,
 214                         rpcb_local_clnt4);
 215 }
 216
 217 /*
 218  * Returns zero on success, otherwise a negative errno value
 219  * is returned.
 220  */
 221 static int rpcb_create_local_unix(void)
 222 {
 223         static const struct sockaddr_un rpcb_localaddr_rpcbind = {
 224                 .sun_family             = AF_LOCAL,
 225                 .sun_path               = RPCBIND_SOCK_PATHNAME,
 226         };
 227         struct rpc_create_args args = {
 228                 .net            = &init_net,
 229                 .protocol       = XPRT_TRANSPORT_LOCAL,
 230                 .address        = (struct sockaddr *)&rpcb_localaddr_rpcbind,
 231                 .addrsize       = sizeof(rpcb_localaddr_rpcbind),
 232                 .servername     = "localhost",
 233                 .program        = &rpcb_program,
 234                 .version        = RPCBVERS_2,
 235                 .authflavor     = RPC_AUTH_NULL,
 236         };
 237         struct rpc_clnt *clnt, *clnt4;
 238         int result = 0;
 239
 240         /*
 241          * Because we requested an RPC PING at transport creation time,
 242          * this works only if the user space portmapper is rpcbind, and
 243          * it's listening on AF_LOCAL on the named socket.
 244          */
 245         clnt = rpc_create(&args);
 246         if (IS_ERR(clnt)) {
 247                 dprintk("RPC:       failed to create AF_LOCAL rpcbind "
 248                                 "client (errno %ld).\n", PTR_ERR(clnt));
 249                 result = -PTR_ERR(clnt);
 250                 goto out;
 251         }
 252
 253         clnt4 = rpc_bind_new_program(clnt, &rpcb_program, RPCBVERS_4);
 254         if (IS_ERR(clnt4)) {
 255                 dprintk("RPC:       failed to bind second program to "
 256                                 "rpcbind v4 client (errno %ld).\n",
 257                                 PTR_ERR(clnt4));
 258                 clnt4 = NULL;
 259         }
 260
 261         rpcb_set_local(clnt, clnt4);
 262
 263 out:
 264         return result;
 265 }
 266
 267 /*
 268  * Returns zero on success, otherwise a negative errno value
 269  * is returned.
 270  */
 271 static int rpcb_create_local_net(void)
 272 {
 273         static const struct sockaddr_in rpcb_inaddr_loopback = {
 274                 .sin_family             = AF_INET,
 275                 .sin_addr.s_addr        = htonl(INADDR_LOOPBACK),
 276                 .sin_port               = htons(RPCBIND_PORT),
 277         };
 278         struct rpc_create_args args = {
 279                 .net            = &init_net,
 280                 .protocol       = XPRT_TRANSPORT_TCP,
 281                 .address        = (struct sockaddr *)&rpcb_inaddr_loopback,
 282                 .addrsize       = sizeof(rpcb_inaddr_loopback),
 283                 .servername     = "localhost",
 284                 .program        = &rpcb_program,
 285                 .version        = RPCBVERS_2,
 286                 .authflavor     = RPC_AUTH_UNIX,
 287                 .flags          = RPC_CLNT_CREATE_NOPING,
 288         };
 289         struct rpc_clnt *clnt, *clnt4;
 290         int result = 0;
 291
 292         clnt = rpc_create(&args);
 293         if (IS_ERR(clnt)) {
 294                 dprintk("RPC:       failed to create local rpcbind "
 295                                 "client (errno %ld).\n", PTR_ERR(clnt));
 296                 result = -PTR_ERR(clnt);
 297                 goto out;
 298         }
 299
 300         /*
 301          * This results in an RPC ping.  On systems running portmapper,
 302          * the v4 ping will fail.  Proceed anyway, but disallow rpcb
 303          * v4 upcalls.
 304          */
 305         clnt4 = rpc_bind_new_program(clnt, &rpcb_program, RPCBVERS_4);
 306         if (IS_ERR(clnt4)) {
 307                 dprintk("RPC:       failed to bind second program to "
 308                                 "rpcbind v4 client (errno %ld).\n",
 309                                 PTR_ERR(clnt4));
 310                 clnt4 = NULL;
 311         }
 312
 313         rpcb_set_local(clnt, clnt4);
 314
 315 out:
 316         return result;
 317 }
 318
 319 /*
 320  * Returns zero on success, otherwise a negative errno value
 321  * is returned.
 322  */
 323 int rpcb_create_local(void)
 324 {
 325         static DEFINE_MUTEX(rpcb_create_local_mutex);
 326         int result = 0;
 327
 328         if (rpcb_get_local())
 329                 return result;
 330
 331         mutex_lock(&rpcb_create_local_mutex);
 332         if (rpcb_get_local())
 333                 goto out;
 334
 335         if (rpcb_create_local_unix() != 0)
 336                 result = rpcb_create_local_net();
 337
 338 out:
 339         mutex_unlock(&rpcb_create_local_mutex);
 340         return result;
 341 }
 342
 343 static struct rpc_clnt *rpcb_create(char *hostname, struct sockaddr *srvaddr,
 344                                     size_t salen, int proto, u32 version)
 345 {
 346         struct rpc_create_args args = {
 347                 .net            = &init_net,
 348                 .protocol       = proto,
 349                 .address        = srvaddr,
 350                 .addrsize       = salen,
 351                 .servername     = hostname,
 352                 .program        = &rpcb_program,
 353                 .version        = version,
 354                 .authflavor     = RPC_AUTH_UNIX,
 355                 .flags          = (RPC_CLNT_CREATE_NOPING |
 356                                         RPC_CLNT_CREATE_NONPRIVPORT),
 357         };
 358
 359         switch (srvaddr->sa_family) {
 360         case AF_INET:
 361                 ((struct sockaddr_in *)srvaddr)->sin_port = htons(RPCBIND_PORT);
 362                 break;
 363         case AF_INET6:
 364                 ((struct sockaddr_in6 *)srvaddr)->sin6_port = htons(RPCBIND_PORT);
 365                 break;
 366         default:
 367                 return ERR_PTR(-EAFNOSUPPORT);
 368         }
 369
 370         return rpc_create(&args);
 371 }
 372
 373 static int rpcb_register_call(struct rpc_clnt *clnt, struct rpc_message *msg)
 374 {
 375         int result, error = 0;
 376
 377         msg->rpc_resp = &result;
 378
 379         error = rpc_call_sync(clnt, msg, RPC_TASK_SOFTCONN);
 380         if (error < 0) {
 381                 dprintk("RPC:       failed to contact local rpcbind "
 382                                 "server (errno %d).\n", -error);
 383                 return error;
 384         }
 385
 386         if (!result)
 387                 return -EACCES;
 388         return 0;
 389 }
 390
 391 /**
 392  * rpcb_register - set or unset a port registration with the local rpcbind svc
 393  * @prog: RPC program number to bind
 394  * @vers: RPC version number to bind
 395  * @prot: transport protocol to register
 396  * @port: port value to register
 397  *
 398  * Returns zero if the registration request was dispatched successfully
 399  * and the rpcbind daemon returned success.  Otherwise, returns an errno
 400  * value that reflects the nature of the error (request could not be
 401  * dispatched, timed out, or rpcbind returned an error).
 402  *
 403  * RPC services invoke this function to advertise their contact
 404  * information via the system's rpcbind daemon.  RPC services
 405  * invoke this function once for each [program, version, transport]
 406  * tuple they wish to advertise.
 407  *
 408  * Callers may also unregister RPC services that are no longer
 409  * available by setting the passed-in port to zero.  This removes
 410  * all registered transports for [program, version] from the local
 411  * rpcbind database.
 412  *
 413  * This function uses rpcbind protocol version 2 to contact the
 414  * local rpcbind daemon.
 415  *
 416  * Registration works over both AF_INET and AF_INET6, and services
 417  * registered via this function are advertised as available for any
 418  * address.  If the local rpcbind daemon is listening on AF_INET6,
 419  * services registered via this function will be advertised on
 420  * IN6ADDR_ANY (ie available for all AF_INET and AF_INET6
 421  * addresses).
 422  */
 423 int rpcb_register(u32 prog, u32 vers, int prot, unsigned short port)
 424 {
 425         struct rpcbind_args map = {
 426                 .r_prog         = prog,
 427                 .r_vers         = vers,
 428                 .r_prot         = prot,
 429                 .r_port         = port,
 430         };
 431         struct rpc_message msg = {
 432                 .rpc_argp       = &map,
 433         };
 434         int error;
 435
 436         error = rpcb_create_local();
 437         if (error)
 438                 return error;
 439
 440         dprintk("RPC:       %sregistering (%u, %u, %d, %u) with local "
 441                         "rpcbind\n", (port ? "" : "un"),
 442                         prog, vers, prot, port);
 443
 444         msg.rpc_proc = &rpcb_procedures2[RPCBPROC_UNSET];
 445         if (port)
 446                 msg.rpc_proc = &rpcb_procedures2[RPCBPROC_SET];
 447
 448         return rpcb_register_call(rpcb_local_clnt, &msg);
 449 }
 450
 451 /*
 452  * Fill in AF_INET family-specific arguments to register
 453  */
 454 static int rpcb_register_inet4(const struct sockaddr *sap,
 455                                struct rpc_message *msg)
 456 {
 457         const struct sockaddr_in *sin = (const struct sockaddr_in *)sap;
 458         struct rpcbind_args *map = msg->rpc_argp;
 459         unsigned short port = ntohs(sin->sin_port);
 460         int result;
 461
 462         map->r_addr = rpc_sockaddr2uaddr(sap, GFP_KERNEL);
 463
 464         dprintk("RPC:       %sregistering [%u, %u, %s, '%s'] with "
 465                 "local rpcbind\n", (port ? "" : "un"),
 466                         map->r_prog, map->r_vers,
 467                         map->r_addr, map->r_netid);
 468
 469         msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];
 470         if (port)
 471                 msg->rpc_proc = &rpcb_procedures4[RPCBPROC_SET];
 472
 473         result = rpcb_register_call(rpcb_local_clnt4, msg);
 474         kfree(map->r_addr);
 475         return result;
 476 }
 477
 478 /*
 479  * Fill in AF_INET6 family-specific arguments to register
 480  */
 481 static int rpcb_register_inet6(const struct sockaddr *sap,
 482                                struct rpc_message *msg)
 483 {
 484         const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *)sap;
 485         struct rpcbind_args *map = msg->rpc_argp;
 486         unsigned short port = ntohs(sin6->sin6_port);
 487         int result;
 488
 489         map->r_addr = rpc_sockaddr2uaddr(sap, GFP_KERNEL);
 490
 491         dprintk("RPC:       %sregistering [%u, %u, %s, '%s'] with "
 492                 "local rpcbind\n", (port ? "" : "un"),
 493                         map->r_prog, map->r_vers,
 494                         map->r_addr, map->r_netid);
 495
 496         msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];
 497         if (port)
 498                 msg->rpc_proc = &rpcb_procedures4[RPCBPROC_SET];
 499
 500         result = rpcb_register_call(rpcb_local_clnt4, msg);
 501         kfree(map->r_addr);
 502         return result;
 503 }
 504
 505 static int rpcb_unregister_all_protofamilies(struct rpc_message *msg)
 506 {
 507         struct rpcbind_args *map = msg->rpc_argp;
 508
 509         dprintk("RPC:       unregistering [%u, %u, '%s'] with "
 510                 "local rpcbind\n",
 511                         map->r_prog, map->r_vers, map->r_netid);
 512
 513         map->r_addr = "";
 514         msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];
 515
 516         return rpcb_register_call(rpcb_local_clnt4, msg);
 517 }
 518
 519 /**
 520  * rpcb_v4_register - set or unset a port registration with the local rpcbind
 521  * @program: RPC program number of service to (un)register
 522  * @version: RPC version number of service to (un)register
 523  * @address: address family, IP address, and port to (un)register
 524  * @netid: netid of transport protocol to (un)register
 525  *
 526  * Returns zero if the registration request was dispatched successfully
 527  * and the rpcbind daemon returned success.  Otherwise, returns an errno
 528  * value that reflects the nature of the error (request could not be
 529  * dispatched, timed out, or rpcbind returned an error).
 530  *
 531  * RPC services invoke this function to advertise their contact
 532  * information via the system's rpcbind daemon.  RPC services
 533  * invoke this function once for each [program, version, address,
 534  * netid] tuple they wish to advertise.
 535  *
 536  * Callers may also unregister RPC services that are registered at a
 537  * specific address by setting the port number in @address to zero.
 538  * They may unregister all registered protocol families at once for
 539  * a service by passing a NULL @address argument.  If @netid is ""
 540  * then all netids for [program, version, address] are unregistered.
 541  *
 542  * This function uses rpcbind protocol version 4 to contact the
 543  * local rpcbind daemon.  The local rpcbind daemon must support
 544  * version 4 of the rpcbind protocol in order for these functions
 545  * to register a service successfully.
 546  *
 547  * Supported netids include "udp" and "tcp" for UDP and TCP over
 548  * IPv4, and "udp6" and "tcp6" for UDP and TCP over IPv6,
 549  * respectively.
 550  *
 551  * The contents of @address determine the address family and the
 552  * port to be registered.  The usual practice is to pass INADDR_ANY
 553  * as the raw address, but specifying a non-zero address is also
 554  * supported by this API if the caller wishes to advertise an RPC
 555  * service on a specific network interface.
 556  *
 557  * Note that passing in INADDR_ANY does not create the same service
 558  * registration as IN6ADDR_ANY.  The former advertises an RPC
 559  * service on any IPv4 address, but not on IPv6.  The latter
 560  * advertises the service on all IPv4 and IPv6 addresses.
 561  */
 562 int rpcb_v4_register(const u32 program, const u32 version,
 563                      const struct sockaddr *address, const char *netid)
 564 {
 565         struct rpcbind_args map = {
 566                 .r_prog         = program,
 567                 .r_vers         = version,
 568                 .r_netid        = netid,
 569                 .r_owner        = RPCB_OWNER_STRING,
 570         };
 571         struct rpc_message msg = {
 572                 .rpc_argp       = &map,
 573         };
 574         int error;
 575
 576         error = rpcb_create_local();
 577         if (error)
 578                 return error;
 579         if (rpcb_local_clnt4 == NULL)
 580                 return -EPROTONOSUPPORT;
 581
 582         if (address == NULL)
 583                 return rpcb_unregister_all_protofamilies(&msg);
 584
 585         switch (address->sa_family) {
 586         case AF_INET:
 587                 return rpcb_register_inet4(address, &msg);
 588         case AF_INET6:
 589                 return rpcb_register_inet6(address, &msg);
 590         }
 591
 592         return -EAFNOSUPPORT;
 593 }
 594
 595 static struct rpc_task *rpcb_call_async(struct rpc_clnt *rpcb_clnt, struct rpcbind_args *map, struct rpc_procinfo *proc)
 596 {
 597         struct rpc_message msg = {
 598                 .rpc_proc = proc,
 599                 .rpc_argp = map,
 600                 .rpc_resp = map,
 601         };
 602         struct rpc_task_setup task_setup_data = {
 603                 .rpc_client = rpcb_clnt,
 604                 .rpc_message = &msg,
 605                 .callback_ops = &rpcb_getport_ops,
 606                 .callback_data = map,
 607                 .flags = RPC_TASK_ASYNC | RPC_TASK_SOFTCONN,
 608         };
 609
 610         return rpc_run_task(&task_setup_data);
 611 }
 612
 613 /*
 614  * In the case where rpc clients have been cloned, we want to make
 615  * sure that we use the program number/version etc of the actual
 616  * owner of the xprt. To do so, we walk back up the tree of parents
 617  * to find whoever created the transport and/or whoever has the
 618  * autobind flag set.
 619  */
 620 static struct rpc_clnt *rpcb_find_transport_owner(struct rpc_clnt *clnt)
 621 {
 622         struct rpc_clnt *parent = clnt->cl_parent;
 623
 624         while (parent != clnt) {
 625                 if (parent->cl_xprt != clnt->cl_xprt)
 626                         break;
 627                 if (clnt->cl_autobind)
 628                         break;
 629                 clnt = parent;
 630                 parent = parent->cl_parent;
 631         }
 632         return clnt;
 633 }
 634
 635 /**
 636  * rpcb_getport_async - obtain the port for a given RPC service on a given host
 637  * @task: task that is waiting for portmapper request
 638  *
 639  * This one can be called for an ongoing RPC request, and can be used in
 640  * an async (rpciod) context.
 641  */
 642 void rpcb_getport_async(struct rpc_task *task)
 643 {
 644         struct rpc_clnt *clnt;
 645         struct rpc_procinfo *proc;
 646         u32 bind_version;
 647         struct rpc_xprt *xprt;
 648         struct rpc_clnt *rpcb_clnt;
 649         struct rpcbind_args *map;
 650         struct rpc_task *child;
 651         struct sockaddr_storage addr;
 652         struct sockaddr *sap = (struct sockaddr *)&addr;
 653         size_t salen;
 654         int status;
 655
 656         clnt = rpcb_find_transport_owner(task->tk_client);
 657         xprt = clnt->cl_xprt;
 658
 659         dprintk("RPC: %5u %s(%s, %u, %u, %d)\n",
 660                 task->tk_pid, __func__,
 661                 clnt->cl_server, clnt->cl_prog, clnt->cl_vers, xprt->prot);
 662
 663         /* Put self on the wait queue to ensure we get notified if
 664          * some other task is already attempting to bind the port */
 665         rpc_sleep_on(&xprt->binding, task, NULL);
 666
 667         if (xprt_test_and_set_binding(xprt)) {
 668                 dprintk("RPC: %5u %s: waiting for another binder\n",
 669                         task->tk_pid, __func__);
 670                 return;
 671         }
 672
 673         /* Someone else may have bound if we slept */
 674         if (xprt_bound(xprt)) {
 675                 status = 0;
 676                 dprintk("RPC: %5u %s: already bound\n",
 677                         task->tk_pid, __func__);
 678                 goto bailout_nofree;
 679         }
 680
 681         /* Parent transport's destination address */
 682         salen = rpc_peeraddr(clnt, sap, sizeof(addr));
 683
 684         /* Don't ever use rpcbind v2 for AF_INET6 requests */
 685         switch (sap->sa_family) {
 686         case AF_INET:
 687                 proc = rpcb_next_version[xprt->bind_index].rpc_proc;
 688                 bind_version = rpcb_next_version[xprt->bind_index].rpc_vers;
 689                 break;
 690         case AF_INET6:
 691                 proc = rpcb_next_version6[xprt->bind_index].rpc_proc;
 692                 bind_version = rpcb_next_version6[xprt->bind_index].rpc_vers;
 693                 break;
 694         default:
 695                 status = -EAFNOSUPPORT;
 696                 dprintk("RPC: %5u %s: bad address family\n",
 697                                 task->tk_pid, __func__);
 698                 goto bailout_nofree;
 699         }
 700         if (proc == NULL) {
 701                 xprt->bind_index = 0;
 702                 status = -EPFNOSUPPORT;
 703                 dprintk("RPC: %5u %s: no more getport versions available\n",
 704                         task->tk_pid, __func__);
 705                 goto bailout_nofree;
 706         }
 707
 708         dprintk("RPC: %5u %s: trying rpcbind version %u\n",
 709                 task->tk_pid, __func__, bind_version);
 710
 711         rpcb_clnt = rpcb_create(clnt->cl_server, sap, salen, xprt->prot,
 712                                 bind_version);
 713         if (IS_ERR(rpcb_clnt)) {
 714                 status = PTR_ERR(rpcb_clnt);
 715                 dprintk("RPC: %5u %s: rpcb_create failed, error %ld\n",
 716                         task->tk_pid, __func__, PTR_ERR(rpcb_clnt));
 717                 goto bailout_nofree;
 718         }
 719
 720         map = kzalloc(sizeof(struct rpcbind_args), GFP_ATOMIC);
 721         if (!map) {
 722                 status = -ENOMEM;
 723                 dprintk("RPC: %5u %s: no memory available\n",
 724                         task->tk_pid, __func__);
 725                 goto bailout_release_client;
 726         }
 727         map->r_prog = clnt->cl_prog;
 728         map->r_vers = clnt->cl_vers;
 729         map->r_prot = xprt->prot;
 730         map->r_port = 0;
 731         map->r_xprt = xprt_get(xprt);
 732         map->r_status = -EIO;
 733
 734         switch (bind_version) {
 735         case RPCBVERS_4:
 736         case RPCBVERS_3:
 737                 map->r_netid = rpc_peeraddr2str(clnt, RPC_DISPLAY_NETID);
 738                 map->r_addr = rpc_sockaddr2uaddr(sap, GFP_ATOMIC);
 739                 map->r_owner = "";
 740                 break;
 741         case RPCBVERS_2:
 742                 map->r_addr = NULL;
 743                 break;
 744         default:
 745                 BUG();
 746         }
 747
 748         child = rpcb_call_async(rpcb_clnt, map, proc);
 749         rpc_release_client(rpcb_clnt);
 750         if (IS_ERR(child)) {
 751                 /* rpcb_map_release() has freed the arguments */
 752                 dprintk("RPC: %5u %s: rpc_run_task failed\n",
 753                         task->tk_pid, __func__);
 754                 return;
 755         }
 756
 757         xprt->stat.bind_count++;
 758         rpc_put_task(child);
 759         return;
 760
 761 bailout_release_client:
 762         rpc_release_client(rpcb_clnt);
 763 bailout_nofree:
 764         rpcb_wake_rpcbind_waiters(xprt, status);
 765         task->tk_status = status;
 766 }
 767 EXPORT_SYMBOL_GPL(rpcb_getport_async);
 768
 769 /*
 770  * Rpcbind child task calls this callback via tk_exit.
 771  */
 772 static void rpcb_getport_done(struct rpc_task *child, void *data)
 773 {
 774         struct rpcbind_args *map = data;
 775         struct rpc_xprt *xprt = map->r_xprt;
 776         int status = child->tk_status;
 777
 778         /* Garbage reply: retry with a lesser rpcbind version */
 779         if (status == -EIO)
 780                 status = -EPROTONOSUPPORT;
 781
 782         /* rpcbind server doesn't support this rpcbind protocol version */
 783         if (status == -EPROTONOSUPPORT)
 784                 xprt->bind_index++;
 785
 786         if (status < 0) {
 787                 /* rpcbind server not available on remote host? */
 788                 xprt->ops->set_port(xprt, 0);
 789         } else if (map->r_port == 0) {
 790                 /* Requested RPC service wasn't registered on remote host */
 791                 xprt->ops->set_port(xprt, 0);
 792                 status = -EACCES;
 793         } else {
 794                 /* Succeeded */
 795                 xprt->ops->set_port(xprt, map->r_port);
 796                 xprt_set_bound(xprt);
 797                 status = 0;
 798         }
 799
 800         dprintk("RPC: %5u rpcb_getport_done(status %d, port %u)\n",
 801                         child->tk_pid, status, map->r_port);
 802
 803         map->r_status = status;
 804 }
 805
 806 /*
 807  * XDR functions for rpcbind
 808  */
 809
 810 static void rpcb_enc_mapping(struct rpc_rqst *req, struct xdr_stream *xdr,
 811                              const struct rpcbind_args *rpcb)
 812 {
 813         struct rpc_task *task = req->rq_task;
 814         __be32 *p;
 815
 816         dprintk("RPC: %5u encoding PMAP_%s call (%u, %u, %d, %u)\n",
 817                         task->tk_pid, task->tk_msg.rpc_proc->p_name,
 818                         rpcb->r_prog, rpcb->r_vers, rpcb->r_prot, rpcb->r_port);
 819
 820         p = xdr_reserve_space(xdr, RPCB_mappingargs_sz << 2);
 821         *p++ = cpu_to_be32(rpcb->r_prog);
 822         *p++ = cpu_to_be32(rpcb->r_vers);
 823         *p++ = cpu_to_be32(rpcb->r_prot);
 824         *p   = cpu_to_be32(rpcb->r_port);
 825 }
 826
 827 static int rpcb_dec_getport(struct rpc_rqst *req, struct xdr_stream *xdr,
 828                             struct rpcbind_args *rpcb)
 829 {
 830         struct rpc_task *task = req->rq_task;
 831         unsigned long port;
 832         __be32 *p;
 833
 834         rpcb->r_port = 0;
 835
 836         p = xdr_inline_decode(xdr, 4);
 837         if (unlikely(p == NULL))
 838                 return -EIO;
 839
 840         port = be32_to_cpup(p);
 841         dprintk("RPC: %5u PMAP_%s result: %lu\n", task->tk_pid,
 842                         task->tk_msg.rpc_proc->p_name, port);
 843         if (unlikely(port > USHRT_MAX))
 844                 return -EIO;
 845
 846         rpcb->r_port = port;
 847         return 0;
 848 }
 849
 850 static int rpcb_dec_set(struct rpc_rqst *req, struct xdr_stream *xdr,
 851                         unsigned int *boolp)
 852 {
 853         struct rpc_task *task = req->rq_task;
 854         __be32 *p;
 855
 856         p = xdr_inline_decode(xdr, 4);
 857         if (unlikely(p == NULL))
 858                 return -EIO;
 859
 860         *boolp = 0;
 861         if (*p != xdr_zero)
 862                 *boolp = 1;
 863
 864         dprintk("RPC: %5u RPCB_%s call %s\n",
 865                         task->tk_pid, task->tk_msg.rpc_proc->p_name,
 866                         (*boolp ? "succeeded" : "failed"));
 867         return 0;
 868 }
 869
 870 static void encode_rpcb_string(struct xdr_stream *xdr, const char *string,
 871                                const u32 maxstrlen)
 872 {
 873         __be32 *p;
 874         u32 len;
 875
 876         len = strlen(string);
 877         BUG_ON(len > maxstrlen);
 878         p = xdr_reserve_space(xdr, 4 + len);
 879         xdr_encode_opaque(p, string, len);
 880 }
 881
 882 static void rpcb_enc_getaddr(struct rpc_rqst *req, struct xdr_stream *xdr,
 883                              const struct rpcbind_args *rpcb)
 884 {
 885         struct rpc_task *task = req->rq_task;
 886         __be32 *p;
 887
 888         dprintk("RPC: %5u encoding RPCB_%s call (%u, %u, '%s', '%s')\n",
 889                         task->tk_pid, task->tk_msg.rpc_proc->p_name,
 890                         rpcb->r_prog, rpcb->r_vers,
 891                         rpcb->r_netid, rpcb->r_addr);
 892
 893         p = xdr_reserve_space(xdr, (RPCB_program_sz + RPCB_version_sz) << 2);
 894         *p++ = cpu_to_be32(rpcb->r_prog);
 895         *p = cpu_to_be32(rpcb->r_vers);
 896
 897         encode_rpcb_string(xdr, rpcb->r_netid, RPCBIND_MAXNETIDLEN);
 898         encode_rpcb_string(xdr, rpcb->r_addr, RPCBIND_MAXUADDRLEN);
 899         encode_rpcb_string(xdr, rpcb->r_owner, RPCB_MAXOWNERLEN);
 900 }
 901
 902 static int rpcb_dec_getaddr(struct rpc_rqst *req, struct xdr_stream *xdr,
 903                             struct rpcbind_args *rpcb)
 904 {
 905         struct sockaddr_storage address;
 906         struct sockaddr *sap = (struct sockaddr *)&address;
 907         struct rpc_task *task = req->rq_task;
 908         __be32 *p;
 909         u32 len;
 910
 911         rpcb->r_port = 0;
 912
 913         p = xdr_inline_decode(xdr, 4);
 914         if (unlikely(p == NULL))
 915                 goto out_fail;
 916         len = be32_to_cpup(p);
 917
 918         /*
 919          * If the returned universal address is a null string,
 920          * the requested RPC service was not registered.
 921          */
 922         if (len == 0) {
 923                 dprintk("RPC: %5u RPCB reply: program not registered\n",
 924                                 task->tk_pid);
 925                 return 0;
 926         }
 927
 928         if (unlikely(len > RPCBIND_MAXUADDRLEN))
 929                 goto out_fail;
 930
 931         p = xdr_inline_decode(xdr, len);
 932         if (unlikely(p == NULL))
 933                 goto out_fail;
 934         dprintk("RPC: %5u RPCB_%s reply: %s\n", task->tk_pid,
 935                         task->tk_msg.rpc_proc->p_name, (char *)p);
 936
 937         if (rpc_uaddr2sockaddr((char *)p, len, sap, sizeof(address)) == 0)
 938                 goto out_fail;
 939         rpcb->r_port = rpc_get_port(sap);
 940
 941         return 0;
 942
 943 out_fail:
 944         dprintk("RPC: %5u malformed RPCB_%s reply\n",
 945                         task->tk_pid, task->tk_msg.rpc_proc->p_name);
 946         return -EIO;
 947 }
 948
 949 /*
 950  * Not all rpcbind procedures described in RFC 1833 are implemented
 951  * since the Linux kernel RPC code requires only these.
 952  */
 953
 954 static struct rpc_procinfo rpcb_procedures2[] = {
 955         [RPCBPROC_SET] = {
 956                 .p_proc         = RPCBPROC_SET,
 957                 .p_encode       = (kxdreproc_t)rpcb_enc_mapping,
 958                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
 959                 .p_arglen       = RPCB_mappingargs_sz,
 960                 .p_replen       = RPCB_setres_sz,
 961                 .p_statidx      = RPCBPROC_SET,
 962                 .p_timer        = 0,
 963                 .p_name         = "SET",
 964         },
 965         [RPCBPROC_UNSET] = {
 966                 .p_proc         = RPCBPROC_UNSET,
 967                 .p_encode       = (kxdreproc_t)rpcb_enc_mapping,
 968                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
 969                 .p_arglen       = RPCB_mappingargs_sz,
 970                 .p_replen       = RPCB_setres_sz,
 971                 .p_statidx      = RPCBPROC_UNSET,
 972                 .p_timer        = 0,
 973                 .p_name         = "UNSET",
 974         },
 975         [RPCBPROC_GETPORT] = {
 976                 .p_proc         = RPCBPROC_GETPORT,
 977                 .p_encode       = (kxdreproc_t)rpcb_enc_mapping,
 978                 .p_decode       = (kxdrdproc_t)rpcb_dec_getport,
 979                 .p_arglen       = RPCB_mappingargs_sz,
 980                 .p_replen       = RPCB_getportres_sz,
 981                 .p_statidx      = RPCBPROC_GETPORT,
 982                 .p_timer        = 0,
 983                 .p_name         = "GETPORT",
 984         },
 985 };
 986
 987 static struct rpc_procinfo rpcb_procedures3[] = {
 988         [RPCBPROC_SET] = {
 989                 .p_proc         = RPCBPROC_SET,
 990                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
 991                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
 992                 .p_arglen       = RPCB_getaddrargs_sz,
 993                 .p_replen       = RPCB_setres_sz,
 994                 .p_statidx      = RPCBPROC_SET,
 995                 .p_timer        = 0,
 996                 .p_name         = "SET",
 997         },
 998         [RPCBPROC_UNSET] = {
 999                 .p_proc         = RPCBPROC_UNSET,
1000                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
1001                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
1002                 .p_arglen       = RPCB_getaddrargs_sz,
1003                 .p_replen       = RPCB_setres_sz,
1004                 .p_statidx      = RPCBPROC_UNSET,
1005                 .p_timer        = 0,
1006                 .p_name         = "UNSET",
1007         },
1008         [RPCBPROC_GETADDR] = {
1009                 .p_proc         = RPCBPROC_GETADDR,
1010                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
1011                 .p_decode       = (kxdrdproc_t)rpcb_dec_getaddr,
1012                 .p_arglen       = RPCB_getaddrargs_sz,
1013                 .p_replen       = RPCB_getaddrres_sz,
1014                 .p_statidx      = RPCBPROC_GETADDR,
1015                 .p_timer        = 0,
1016                 .p_name         = "GETADDR",
1017         },
1018 };
1019
1020 static struct rpc_procinfo rpcb_procedures4[] = {
1021         [RPCBPROC_SET] = {
1022                 .p_proc         = RPCBPROC_SET,
1023                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
1024                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
1025                 .p_arglen       = RPCB_getaddrargs_sz,
1026                 .p_replen       = RPCB_setres_sz,
1027                 .p_statidx      = RPCBPROC_SET,
1028                 .p_timer        = 0,
1029                 .p_name         = "SET",
1030         },
1031         [RPCBPROC_UNSET] = {
1032                 .p_proc         = RPCBPROC_UNSET,
1033                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
1034                 .p_decode       = (kxdrdproc_t)rpcb_dec_set,
1035                 .p_arglen       = RPCB_getaddrargs_sz,
1036                 .p_replen       = RPCB_setres_sz,
1037                 .p_statidx      = RPCBPROC_UNSET,
1038                 .p_timer        = 0,
1039                 .p_name         = "UNSET",
1040         },
1041         [RPCBPROC_GETADDR] = {
1042                 .p_proc         = RPCBPROC_GETADDR,
1043                 .p_encode       = (kxdreproc_t)rpcb_enc_getaddr,
1044                 .p_decode       = (kxdrdproc_t)rpcb_dec_getaddr,
1045                 .p_arglen       = RPCB_getaddrargs_sz,
1046                 .p_replen       = RPCB_getaddrres_sz,
1047                 .p_statidx      = RPCBPROC_GETADDR,
1048                 .p_timer        = 0,
1049                 .p_name         = "GETADDR",
1050         },
1051 };
1052
1053 static struct rpcb_info rpcb_next_version[] = {
1054         {
1055                 .rpc_vers       = RPCBVERS_2,
1056                 .rpc_proc       = &rpcb_procedures2[RPCBPROC_GETPORT],
1057         },
1058         {
1059                 .rpc_proc       = NULL,
1060         },
1061 };
1062
1063 static struct rpcb_info rpcb_next_version6[] = {
1064         {
1065                 .rpc_vers       = RPCBVERS_4,
1066                 .rpc_proc       = &rpcb_procedures4[RPCBPROC_GETADDR],
1067         },
1068         {
1069                 .rpc_vers       = RPCBVERS_3,
1070                 .rpc_proc       = &rpcb_procedures3[RPCBPROC_GETADDR],
1071         },
1072         {
1073                 .rpc_proc       = NULL,
1074         },
1075 };
1076
1077 static struct rpc_version rpcb_version2 = {
1078         .number         = RPCBVERS_2,
1079         .nrprocs        = ARRAY_SIZE(rpcb_procedures2),
1080         .procs          = rpcb_procedures2
1081 };
1082
1083 static struct rpc_version rpcb_version3 = {
1084         .number         = RPCBVERS_3,
1085         .nrprocs        = ARRAY_SIZE(rpcb_procedures3),
1086         .procs          = rpcb_procedures3
1087 };
1088
1089 static struct rpc_version rpcb_version4 = {
1090         .number         = RPCBVERS_4,
1091         .nrprocs        = ARRAY_SIZE(rpcb_procedures4),
1092         .procs          = rpcb_procedures4
1093 };
1094
1095 static struct rpc_version *rpcb_version[] = {
1096         NULL,
1097         NULL,
1098         &rpcb_version2,
1099         &rpcb_version3,
1100         &rpcb_version4
1101 };
1102
1103 static struct rpc_stat rpcb_stats;
1104
1105 static struct rpc_program rpcb_program = {
1106         .name           = "rpcbind",
1107         .number         = RPCBIND_PROGRAM,
1108         .nrvers         = ARRAY_SIZE(rpcb_version),
1109         .version        = rpcb_version,
1110         .stats          = &rpcb_stats,
1111 };
1112
1113 /**
1114  * cleanup_rpcb_clnt - remove xprtsock's sysctls, unregister
1115  *
1116  */
1117 void cleanup_rpcb_clnt(void)
1118 {
1119         if (rpcb_local_clnt4)
1120                 rpc_shutdown_client(rpcb_local_clnt4);
1121         if (rpcb_local_clnt)
1122                 rpc_shutdown_client(rpcb_local_clnt);
1123 }