net/sunrpc/pmap_clnt.c

   1 /*
   2  * linux/net/sunrpc/pmap_clnt.c
   3  *
   4  * In-kernel RPC portmapper client.
   5  *
   6  * Portmapper supports version 2 of the rpcbind protocol (RFC 1833).
   7  *
   8  * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
   9  */
  10
  11 #include <linux/types.h>
  12 #include <linux/socket.h>
  13 #include <linux/kernel.h>
  14 #include <linux/errno.h>
  15 #include <linux/uio.h>
  16 #include <linux/in.h>
  17 #include <linux/sunrpc/clnt.h>
  18 #include <linux/sunrpc/xprt.h>
  19 #include <linux/sunrpc/sched.h>
  20
  21 #ifdef RPC_DEBUG
  22 # define RPCDBG_FACILITY        RPCDBG_PMAP
  23 #endif
  24
  25 #define PMAP_SET                1
  26 #define PMAP_UNSET              2
  27 #define PMAP_GETPORT            3
  28
  29 struct portmap_args {
  30         u32                     pm_prog;
  31         u32                     pm_vers;
  32         u32                     pm_prot;
  33         unsigned short          pm_port;
  34         struct rpc_task *       pm_task;
  35 };
  36
  37 static struct rpc_procinfo      pmap_procedures[];
  38 static struct rpc_clnt *        pmap_create(char *, struct sockaddr_in *, int, int);
  39 static void                     pmap_getport_done(struct rpc_task *, void *);
  40 static struct rpc_program       pmap_program;
  41
  42 static void pmap_getport_prepare(struct rpc_task *task, void *calldata)
  43 {
  44         struct portmap_args *map = calldata;
  45         struct rpc_message msg = {
  46                 .rpc_proc       = &pmap_procedures[PMAP_GETPORT],
  47                 .rpc_argp       = map,
  48                 .rpc_resp       = &map->pm_port,
  49         };
  50
  51         rpc_call_setup(task, &msg, 0);
  52 }
  53
  54 static inline struct portmap_args *pmap_map_alloc(void)
  55 {
  56         return kmalloc(sizeof(struct portmap_args), GFP_NOFS);
  57 }
  58
  59 static inline void pmap_map_free(struct portmap_args *map)
  60 {
  61         kfree(map);
  62 }
  63
  64 static void pmap_map_release(void *data)
  65 {
  66         pmap_map_free(data);
  67 }
  68
  69 static const struct rpc_call_ops pmap_getport_ops = {
  70         .rpc_call_prepare       = pmap_getport_prepare,
  71         .rpc_call_done          = pmap_getport_done,
  72         .rpc_release            = pmap_map_release,
  73 };
  74
  75 static inline void pmap_wake_portmap_waiters(struct rpc_xprt *xprt)
  76 {
  77         xprt_clear_binding(xprt);
  78         rpc_wake_up(&xprt->binding);
  79 }
  80
  81 /**
  82  * rpc_getport - obtain the port for a given RPC service on a given host
  83  * @task: task that is waiting for portmapper request
  84  *
  85  * This one can be called for an ongoing RPC request, and can be used in
  86  * an async (rpciod) context.
  87  */
  88 void rpc_getport(struct rpc_task *task)
  89 {
  90         struct rpc_clnt *clnt = task->tk_client;
  91         struct rpc_xprt *xprt = task->tk_xprt;
  92         struct sockaddr_in addr;
  93         struct portmap_args *map;
  94         struct rpc_clnt *pmap_clnt;
  95         struct rpc_task *child;
  96
  97         dprintk("RPC: %4d rpc_getport(%s, %u, %u, %d)\n",
  98                         task->tk_pid, clnt->cl_server,
  99                         clnt->cl_prog, clnt->cl_vers, xprt->prot);
 100
 101         /* Autobind on cloned rpc clients is discouraged */
 102         BUG_ON(clnt->cl_parent != clnt);
 103
 104         if (xprt_test_and_set_binding(xprt)) {
 105                 task->tk_status = -EACCES;      /* tell caller to check again */
 106                 rpc_sleep_on(&xprt->binding, task, NULL, NULL);
 107                 return;
 108         }
 109
 110         /* Someone else may have bound if we slept */
 111         if (xprt_bound(xprt)) {
 112                 task->tk_status = 0;
 113                 goto bailout_nofree;
 114         }
 115
 116         map = pmap_map_alloc();
 117         if (!map) {
 118                 task->tk_status = -ENOMEM;
 119                 goto bailout_nofree;
 120         }
 121         map->pm_prog = clnt->cl_prog;
 122         map->pm_vers = clnt->cl_vers;
 123         map->pm_prot = xprt->prot;
 124         map->pm_port = 0;
 125         map->pm_task = task;
 126
 127         rpc_peeraddr(clnt, (struct sockaddr *) &addr, sizeof(addr));
 128         pmap_clnt = pmap_create(clnt->cl_server, &addr, map->pm_prot, 0);
 129         if (IS_ERR(pmap_clnt)) {
 130                 task->tk_status = PTR_ERR(pmap_clnt);
 131                 goto bailout;
 132         }
 133
 134         child = rpc_run_task(pmap_clnt, RPC_TASK_ASYNC, &pmap_getport_ops, map);
 135         if (IS_ERR(child)) {
 136                 task->tk_status = -EIO;
 137                 goto bailout;
 138         }
 139         rpc_release_task(child);
 140
 141         rpc_sleep_on(&xprt->binding, task, NULL, NULL);
 142
 143         task->tk_xprt->stat.bind_count++;
 144         return;
 145
 146 bailout:
 147         pmap_map_free(map);
 148 bailout_nofree:
 149         pmap_wake_portmap_waiters(xprt);
 150 }
 151
 152 #ifdef CONFIG_ROOT_NFS
 153 /**
 154  * rpc_getport_external - obtain the port for a given RPC service on a given host
 155  * @sin: address of remote peer
 156  * @prog: RPC program number to bind
 157  * @vers: RPC version number to bind
 158  * @prot: transport protocol to use to make this request
 159  *
 160  * This one is called from outside the RPC client in a synchronous task context.
 161  */
 162 int rpc_getport_external(struct sockaddr_in *sin, __u32 prog, __u32 vers, int prot)
 163 {
 164         struct portmap_args map = {
 165                 .pm_prog        = prog,
 166                 .pm_vers        = vers,
 167                 .pm_prot        = prot,
 168                 .pm_port        = 0
 169         };
 170         struct rpc_message msg = {
 171                 .rpc_proc       = &pmap_procedures[PMAP_GETPORT],
 172                 .rpc_argp       = &map,
 173                 .rpc_resp       = &map.pm_port,
 174         };
 175         struct rpc_clnt *pmap_clnt;
 176         char            hostname[32];
 177         int             status;
 178
 179         dprintk("RPC:      rpc_getport_external(%u.%u.%u.%u, %u, %u, %d)\n",
 180                         NIPQUAD(sin->sin_addr.s_addr), prog, vers, prot);
 181
 182         sprintf(hostname, "%u.%u.%u.%u", NIPQUAD(sin->sin_addr.s_addr));
 183         pmap_clnt = pmap_create(hostname, sin, prot, 0);
 184         if (IS_ERR(pmap_clnt))
 185                 return PTR_ERR(pmap_clnt);
 186
 187         /* Setup the call info struct */
 188         status = rpc_call_sync(pmap_clnt, &msg, 0);
 189
 190         if (status >= 0) {
 191                 if (map.pm_port != 0)
 192                         return map.pm_port;
 193                 status = -EACCES;
 194         }
 195         return status;
 196 }
 197 #endif
 198
 199 /*
 200  * Portmapper child task invokes this callback via tk_exit.
 201  */
 202 static void pmap_getport_done(struct rpc_task *child, void *data)
 203 {
 204         struct portmap_args *map = data;
 205         struct rpc_task *task = map->pm_task;
 206         struct rpc_xprt *xprt = task->tk_xprt;
 207         int status = child->tk_status;
 208
 209         if (status < 0) {
 210                 /* Portmapper not available */
 211                 xprt->ops->set_port(xprt, 0);
 212                 task->tk_status = status;
 213         } else if (map->pm_port == 0) {
 214                 /* Requested RPC service wasn't registered */
 215                 xprt->ops->set_port(xprt, 0);
 216                 task->tk_status = -EACCES;
 217         } else {
 218                 /* Succeeded */
 219                 xprt->ops->set_port(xprt, map->pm_port);
 220                 xprt_set_bound(xprt);
 221                 task->tk_status = 0;
 222         }
 223
 224         dprintk("RPC: %4d pmap_getport_done(status %d, port %u)\n",
 225                         child->tk_pid, child->tk_status, map->pm_port);
 226
 227         pmap_wake_portmap_waiters(xprt);
 228 }
 229
 230 /**
 231  * rpc_register - set or unset a port registration with the local portmapper
 232  * @prog: RPC program number to bind
 233  * @vers: RPC version number to bind
 234  * @prot: transport protocol to use to make this request
 235  * @port: port value to register
 236  * @okay: result code
 237  *
 238  * port == 0 means unregister, port != 0 means register.
 239  */
 240 int rpc_register(u32 prog, u32 vers, int prot, unsigned short port, int *okay)
 241 {
 242         struct sockaddr_in      sin = {
 243                 .sin_family     = AF_INET,
 244                 .sin_addr.s_addr = htonl(INADDR_LOOPBACK),
 245         };
 246         struct portmap_args     map = {
 247                 .pm_prog        = prog,
 248                 .pm_vers        = vers,
 249                 .pm_prot        = prot,
 250                 .pm_port        = port,
 251         };
 252         struct rpc_message msg = {
 253                 .rpc_proc       = &pmap_procedures[port ? PMAP_SET : PMAP_UNSET],
 254                 .rpc_argp       = &map,
 255                 .rpc_resp       = okay,
 256         };
 257         struct rpc_clnt         *pmap_clnt;
 258         int error = 0;
 259
 260         dprintk("RPC: registering (%u, %u, %d, %u) with portmapper.\n",
 261                         prog, vers, prot, port);
 262
 263         pmap_clnt = pmap_create("localhost", &sin, IPPROTO_UDP, 1);
 264         if (IS_ERR(pmap_clnt)) {
 265                 error = PTR_ERR(pmap_clnt);
 266                 dprintk("RPC: couldn't create pmap client. Error = %d\n", error);
 267                 return error;
 268         }
 269
 270         error = rpc_call_sync(pmap_clnt, &msg, 0);
 271
 272         if (error < 0) {
 273                 printk(KERN_WARNING
 274                         "RPC: failed to contact portmap (errno %d).\n",
 275                         error);
 276         }
 277         dprintk("RPC: registration status %d/%d\n", error, *okay);
 278
 279         /* Client deleted automatically because cl_oneshot == 1 */
 280         return error;
 281 }
 282
 283 static struct rpc_clnt *pmap_create(char *hostname, struct sockaddr_in *srvaddr, int proto, int privileged)
 284 {
 285         struct rpc_xprt *xprt;
 286         struct rpc_clnt *clnt;
 287
 288         xprt = xprt_create_proto(proto, srvaddr, NULL);
 289         if (IS_ERR(xprt))
 290                 return (struct rpc_clnt *)xprt;
 291         xprt->ops->set_port(xprt, RPC_PMAP_PORT);
 292         xprt_set_bound(xprt);
 293         if (!privileged)
 294                 xprt->resvport = 0;
 295
 296         clnt = rpc_new_client(xprt, hostname,
 297                                 &pmap_program, RPC_PMAP_VERSION,
 298                                 RPC_AUTH_UNIX);
 299         if (!IS_ERR(clnt)) {
 300                 clnt->cl_softrtry = 1;
 301                 clnt->cl_oneshot  = 1;
 302         }
 303         return clnt;
 304 }
 305
 306 /*
 307  * XDR encode/decode functions for PMAP
 308  */
 309 static int xdr_encode_mapping(struct rpc_rqst *req, u32 *p, struct portmap_args *map)
 310 {
 311         dprintk("RPC: xdr_encode_mapping(%u, %u, %u, %u)\n",
 312                 map->pm_prog, map->pm_vers, map->pm_prot, map->pm_port);
 313         *p++ = htonl(map->pm_prog);
 314         *p++ = htonl(map->pm_vers);
 315         *p++ = htonl(map->pm_prot);
 316         *p++ = htonl(map->pm_port);
 317
 318         req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
 319         return 0;
 320 }
 321
 322 static int xdr_decode_port(struct rpc_rqst *req, u32 *p, unsigned short *portp)
 323 {
 324         *portp = (unsigned short) ntohl(*p++);
 325         return 0;
 326 }
 327
 328 static int xdr_decode_bool(struct rpc_rqst *req, u32 *p, unsigned int *boolp)
 329 {
 330         *boolp = (unsigned int) ntohl(*p++);
 331         return 0;
 332 }
 333
 334 static struct rpc_procinfo      pmap_procedures[] = {
 335 [PMAP_SET] = {
 336           .p_proc               = PMAP_SET,
 337           .p_encode             = (kxdrproc_t) xdr_encode_mapping,
 338           .p_decode             = (kxdrproc_t) xdr_decode_bool,
 339           .p_bufsiz             = 4,
 340           .p_count              = 1,
 341           .p_statidx            = PMAP_SET,
 342           .p_name               = "SET",
 343         },
 344 [PMAP_UNSET] = {
 345           .p_proc               = PMAP_UNSET,
 346           .p_encode             = (kxdrproc_t) xdr_encode_mapping,
 347           .p_decode             = (kxdrproc_t) xdr_decode_bool,
 348           .p_bufsiz             = 4,
 349           .p_count              = 1,
 350           .p_statidx            = PMAP_UNSET,
 351           .p_name               = "UNSET",
 352         },
 353 [PMAP_GETPORT] = {
 354           .p_proc               = PMAP_GETPORT,
 355           .p_encode             = (kxdrproc_t) xdr_encode_mapping,
 356           .p_decode             = (kxdrproc_t) xdr_decode_port,
 357           .p_bufsiz             = 4,
 358           .p_count              = 1,
 359           .p_statidx            = PMAP_GETPORT,
 360           .p_name               = "GETPORT",
 361         },
 362 };
 363
 364 static struct rpc_version       pmap_version2 = {
 365         .number         = 2,
 366         .nrprocs        = 4,
 367         .procs          = pmap_procedures
 368 };
 369
 370 static struct rpc_version *     pmap_version[] = {
 371         NULL,
 372         NULL,
 373         &pmap_version2
 374 };
 375
 376 static struct rpc_stat          pmap_stats;
 377
 378 static struct rpc_program       pmap_program = {
 379         .name           = "portmap",
 380         .number         = RPC_PMAP_PROGRAM,
 381         .nrvers         = ARRAY_SIZE(pmap_version),
 382         .version        = pmap_version,
 383         .stats          = &pmap_stats,
 384 };