sunrpc/pm_getport.c

   1 /* @(#)pmap_getport.c   2.2 88/08/01 4.0 RPCSRC */
   2 /*
   3  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
   4  * unrestricted use provided that this legend is included on all tape
   5  * media and as a part of the software program in whole or part.  Users
   6  * may copy or modify Sun RPC without charge, but are not authorized
   7  * to license or distribute it to anyone else except as part of a product or
   8  * program developed by the user.
   9  *
  10  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  11  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  12  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  13  *
  14  * Sun RPC is provided with no support and without any obligation on the
  15  * part of Sun Microsystems, Inc. to assist in its use, correction,
  16  * modification or enhancement.
  17  *
  18  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  19  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  20  * OR ANY PART THEREOF.
  21  *
  22  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  23  * or profits or other special, indirect and consequential damages, even if
  24  * Sun has been advised of the possibility of such damages.
  25  *
  26  * Sun Microsystems, Inc.
  27  * 2550 Garcia Avenue
  28  * Mountain View, California  94043
  29  */
  30 #if !defined(lint) && defined(SCCSIDS)
  31 static char sccsid[] = "@(#)pmap_getport.c 1.9 87/08/11 Copyr 1984 Sun Micro";
  32 #endif
  33
  34 /*
  35  * pmap_getport.c
  36  * Client interface to pmap rpc service.
  37  *
  38  * Copyright (C) 1984, Sun Microsystems, Inc.
  39  */
  40
  41 #include <stdbool.h>
  42 #include <unistd.h>
  43 #include <rpc/rpc.h>
  44 #include <rpc/pmap_prot.h>
  45 #include <rpc/pmap_clnt.h>
  46 #include <sys/socket.h>
  47
  48 static const struct timeval timeout =
  49 {5, 0};
  50 static const struct timeval tottimeout =
  51 {60, 0};
  52
  53 /*
  54  * Create a socket that is locally bound to a non-reserve port. For
  55  * any failures, -1 is returned which will cause the RPC code to
  56  * create the socket.
  57  */
  58 int
  59 internal_function
  60 __get_socket (struct sockaddr_in *saddr)
  61 {
  62   int so = __socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
  63   if (so < 0)
  64     return -1;
  65
  66   struct sockaddr_in laddr;
  67   socklen_t namelen = sizeof (laddr);
  68   laddr.sin_family = AF_INET;
  69   laddr.sin_port = 0;
  70   laddr.sin_addr.s_addr = htonl (INADDR_ANY);
  71
  72   int cc = __bind (so, (struct sockaddr *) &laddr, namelen);
  73   if (__builtin_expect (cc < 0, 0))
  74     {
  75     fail:
  76       __close (so);
  77       return -1;
  78     }
  79
  80   cc = __connect (so, (struct sockaddr *) saddr, namelen);
  81   if (__builtin_expect (cc < 0, 0))
  82     goto fail;
  83
  84   return so;
  85 }
  86
  87
  88 /*
  89  * Find the mapped port for program,version.
  90  * Calls the pmap service remotely to do the lookup.
  91  * Returns 0 if no map exists.
  92  */
  93 u_short
  94 pmap_getport (address, program, version, protocol)
  95      struct sockaddr_in *address;
  96      u_long program;
  97      u_long version;
  98      u_int protocol;
  99 {
 100   u_short port = 0;
 101   int socket = -1;
 102   CLIENT *client;
 103   struct pmap parms;
 104   bool closeit = false;
 105
 106   address->sin_port = htons (PMAPPORT);
 107   if (protocol == IPPROTO_TCP)
 108     {
 109       /* Don't need a reserved port to get ports from the portmapper.  */
 110       socket = __get_socket(address);
 111       if (socket != -1)
 112         closeit = true;
 113       client = INTUSE(clnttcp_create) (address, PMAPPROG, PMAPVERS, &socket,
 114                                        RPCSMALLMSGSIZE, RPCSMALLMSGSIZE);
 115     }
 116   else
 117     client = INTUSE(clntudp_bufcreate) (address, PMAPPROG, PMAPVERS, timeout,
 118                                         &socket, RPCSMALLMSGSIZE,
 119                                         RPCSMALLMSGSIZE);
 120   if (client != (CLIENT *) NULL)
 121     {
 122       struct rpc_createerr *ce = &get_rpc_createerr ();
 123       parms.pm_prog = program;
 124       parms.pm_vers = version;
 125       parms.pm_prot = protocol;
 126       parms.pm_port = 0;        /* not needed or used */
 127       if (CLNT_CALL (client, PMAPPROC_GETPORT, (xdrproc_t)INTUSE(xdr_pmap),
 128                      (caddr_t)&parms, (xdrproc_t)INTUSE(xdr_u_short),
 129                      (caddr_t)&port, tottimeout) != RPC_SUCCESS)
 130         {
 131           ce->cf_stat = RPC_PMAPFAILURE;
 132           clnt_geterr (client, &ce->cf_error);
 133         }
 134       else if (port == 0)
 135         {
 136           ce->cf_stat = RPC_PROGNOTREGISTERED;
 137         }
 138       CLNT_DESTROY (client);
 139     }
 140   /* We only need to close the socket here if we opened  it.  */
 141   if (closeit)
 142     (void) __close (socket);
 143   address->sin_port = 0;
 144   return port;
 145 }
 146 libc_hidden_def (pmap_getport)