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