sunrpc/pm_getport.c

   1 /*
   2  * pmap_getport.c
   3  * Client interface to pmap rpc service.
   4  *
   5  * Copyright (c) 2010, Oracle America, Inc.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions are
   9  * met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above
  14  *       copyright notice, this list of conditions and the following
  15  *       disclaimer in the documentation and/or other materials
  16  *       provided with the distribution.
  17  *     * Neither the name of the "Oracle America, Inc." nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  *   FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  25  *   COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  26  *   INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  *   DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  28  *   GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  29  *   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  30  *   WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  31  *   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  32  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  33  */
  34
  35 #include <stdbool.h>
  36 #include <unistd.h>
  37 #include <rpc/rpc.h>
  38 #include <rpc/pmap_prot.h>
  39 #include <rpc/pmap_clnt.h>
  40 #include <sys/socket.h>
  41
  42 /*
  43  * Create a socket that is locally bound to a non-reserve port. For
  44  * any failures, -1 is returned which will cause the RPC code to
  45  * create the socket.
  46  */
  47 int
  48 internal_function
  49 __get_socket (struct sockaddr_in *saddr)
  50 {
  51   int so = __socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
  52   if (so < 0)
  53     return -1;
  54
  55   struct sockaddr_in laddr;
  56   socklen_t namelen = sizeof (laddr);
  57   laddr.sin_family = AF_INET;
  58   laddr.sin_port = 0;
  59   laddr.sin_addr.s_addr = htonl (INADDR_ANY);
  60
  61   int cc = __bind (so, (struct sockaddr *) &laddr, namelen);
  62   if (__builtin_expect (cc < 0, 0))
  63     {
  64     fail:
  65       __close (so);
  66       return -1;
  67     }
  68
  69   cc = __connect (so, (struct sockaddr *) saddr, namelen);
  70   if (__builtin_expect (cc < 0, 0))
  71     goto fail;
  72
  73   return so;
  74 }
  75
  76
  77 /*
  78  * Find the mapped port for program,version.
  79  * Internal version with additional parameters.
  80  * Calls the pmap service remotely to do the lookup.
  81  * Returns 0 if no map exists.
  82  */
  83 u_short
  84 internal_function
  85 __libc_rpc_getport (address, program, version, protocol, timeout_sec,
  86                     tottimeout_sec)
  87      struct sockaddr_in *address;
  88      u_long program;
  89      u_long version;
  90      u_int protocol;
  91      time_t timeout_sec;
  92      time_t tottimeout_sec;
  93 {
  94   const struct timeval timeout = {timeout_sec, 0};
  95   const struct timeval tottimeout = {tottimeout_sec, 0};
  96
  97   u_short port = 0;
  98   int socket = -1;
  99   CLIENT *client;
 100   struct pmap parms;
 101   bool closeit = false;
 102
 103   address->sin_port = htons (PMAPPORT);
 104   if (protocol == IPPROTO_TCP)
 105     {
 106       /* Don't need a reserved port to get ports from the portmapper.  */
 107       socket = __get_socket(address);
 108       if (socket != -1)
 109         closeit = true;
 110       client = INTUSE(clnttcp_create) (address, PMAPPROG, PMAPVERS, &socket,
 111                                        RPCSMALLMSGSIZE, RPCSMALLMSGSIZE);
 112     }
 113   else
 114     client = INTUSE(clntudp_bufcreate) (address, PMAPPROG, PMAPVERS, timeout,
 115                                         &socket, RPCSMALLMSGSIZE,
 116                                         RPCSMALLMSGSIZE);
 117   if (client != (CLIENT *) NULL)
 118     {
 119       struct rpc_createerr *ce = &get_rpc_createerr ();
 120       parms.pm_prog = program;
 121       parms.pm_vers = version;
 122       parms.pm_prot = protocol;
 123       parms.pm_port = 0;        /* not needed or used */
 124       if (CLNT_CALL (client, PMAPPROC_GETPORT, (xdrproc_t)INTUSE(xdr_pmap),
 125                      (caddr_t)&parms, (xdrproc_t)INTUSE(xdr_u_short),
 126                      (caddr_t)&port, tottimeout) != RPC_SUCCESS)
 127         {
 128           ce->cf_stat = RPC_PMAPFAILURE;
 129           clnt_geterr (client, &ce->cf_error);
 130         }
 131       else if (port == 0)
 132         {
 133           ce->cf_stat = RPC_PROGNOTREGISTERED;
 134         }
 135       CLNT_DESTROY (client);
 136     }
 137   /* We only need to close the socket here if we opened  it.  */
 138   if (closeit)
 139     (void) __close (socket);
 140   address->sin_port = 0;
 141   return port;
 142 }
 143 libc_hidden_def (__libc_rpc_getport)
 144
 145
 146 /*
 147  * Find the mapped port for program,version.
 148  * Calls the pmap service remotely to do the lookup.
 149  * Returns 0 if no map exists.
 150  */
 151 u_short
 152 pmap_getport (address, program, version, protocol)
 153      struct sockaddr_in *address;
 154      u_long program;
 155      u_long version;
 156      u_int protocol;
 157 {
 158   return __libc_rpc_getport (address, program, version, protocol, 5, 60);
 159 }
 160 libc_hidden_def (pmap_getport)