lib/libc/rpc/clnt_simple.c

   1 /*      $NetBSD: clnt_simple.c,v 1.21 2000/07/06 03:10:34 christos Exp $        */
   2
   3 /*-
   4  * Copyright (c) 2009, Sun Microsystems, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions are met:
   9  * - Redistributions of source code must retain the above copyright notice,
  10  *   this list of conditions and the following disclaimer.
  11  * - Redistributions in binary form must reproduce the above copyright notice,
  12  *   this list of conditions and the following disclaimer in the documentation
  13  *   and/or other materials provided with the distribution.
  14  * - Neither the name of Sun Microsystems, Inc. nor the names of its
  15  *   contributors may be used to endorse or promote products derived
  16  *   from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  */
  30 /*
  31  * Copyright (c) 1986-1991 by Sun Microsystems Inc.
  32  */
  33
  34 #if defined(LIBC_SCCS) && !defined(lint)
  35 static char *sccsid2 = "from: @(#)clnt_simple.c 1.35 87/08/11 Copyr 1984 Sun Micro";
  36 static char *sccsid = "from: @(#)clnt_simple.c  2.2 88/08/01 4.0 RPCSRC";
  37 #endif
  38 #include <sys/cdefs.h>
  39 __FBSDID("$FreeBSD$");
  40
  41 /*
  42  * clnt_simple.c
  43  * Simplified front end to client rpc.
  44  *
  45  */
  46
  47 #include "namespace.h"
  48 #include "reentrant.h"
  49 #include <sys/param.h>
  50 #include <stdio.h>
  51 #include <errno.h>
  52 #include <rpc/rpc.h>
  53 #include <string.h>
  54 #include <stdlib.h>
  55 #include <fcntl.h>
  56 #include <unistd.h>
  57 #include "un-namespace.h"
  58 #include "mt_misc.h"
  59
  60 #ifndef MAXHOSTNAMELEN
  61 #define MAXHOSTNAMELEN 64
  62 #endif
  63
  64 #ifndef NETIDLEN
  65 #define NETIDLEN 32
  66 #endif
  67
  68 struct rpc_call_private {
  69         int     valid;                  /* Is this entry valid ? */
  70         CLIENT  *client;                /* Client handle */
  71         pid_t   pid;                    /* process-id at moment of creation */
  72         rpcprog_t prognum;              /* Program */
  73         rpcvers_t versnum;              /* Version */
  74         char    host[MAXHOSTNAMELEN];   /* Servers host */
  75         char    nettype[NETIDLEN];      /* Network type */
  76 };
  77 static struct rpc_call_private *rpc_call_private_main;
  78 static thread_key_t rpc_call_key;
  79 static once_t rpc_call_once = ONCE_INITIALIZER;
  80 static int rpc_call_key_error;
  81
  82 static void rpc_call_key_init(void);
  83 static void rpc_call_destroy(void *);
  84
  85 static void
  86 rpc_call_destroy(void *vp)
  87 {
  88         struct rpc_call_private *rcp = (struct rpc_call_private *)vp;
  89
  90         if (rcp) {
  91                 if (rcp->client)
  92                         CLNT_DESTROY(rcp->client);
  93                 free(rcp);
  94         }
  95 }
  96
  97 static void
  98 rpc_call_key_init(void)
  99 {
 100
 101         rpc_call_key_error = thr_keycreate(&rpc_call_key, rpc_call_destroy);
 102 }
 103
 104 /*
 105  * This is the simplified interface to the client rpc layer.
 106  * The client handle is not destroyed here and is reused for
 107  * the future calls to same prog, vers, host and nettype combination.
 108  *
 109  * The total time available is 25 seconds.
 110  */
 111 enum clnt_stat
 112 rpc_call(host, prognum, versnum, procnum, inproc, in, outproc, out, nettype)
 113         const char *host;                       /* host name */
 114         rpcprog_t prognum;                      /* program number */
 115         rpcvers_t versnum;                      /* version number */
 116         rpcproc_t procnum;                      /* procedure number */
 117         xdrproc_t inproc, outproc;      /* in/out XDR procedures */
 118         const char *in;
 119         char  *out;                     /* recv/send data */
 120         const char *nettype;                    /* nettype */
 121 {
 122         struct rpc_call_private *rcp = (struct rpc_call_private *) 0;
 123         enum clnt_stat clnt_stat;
 124         struct timeval timeout, tottimeout;
 125         int main_thread = 1;
 126
 127         if ((main_thread = thr_main())) {
 128                 rcp = rpc_call_private_main;
 129         } else {
 130                 if (thr_once(&rpc_call_once, rpc_call_key_init) != 0 ||
 131                     rpc_call_key_error != 0) {
 132                         rpc_createerr.cf_stat = RPC_SYSTEMERROR;
 133                         rpc_createerr.cf_error.re_errno = rpc_call_key_error;
 134                         return (rpc_createerr.cf_stat);
 135                 }
 136                 rcp = (struct rpc_call_private *)thr_getspecific(rpc_call_key);
 137         }
 138         if (rcp == NULL) {
 139                 rcp = malloc(sizeof (*rcp));
 140                 if (rcp == NULL) {
 141                         rpc_createerr.cf_stat = RPC_SYSTEMERROR;
 142                         rpc_createerr.cf_error.re_errno = errno;
 143                         return (rpc_createerr.cf_stat);
 144                 }
 145                 if (main_thread)
 146                         rpc_call_private_main = rcp;
 147                 else
 148                         thr_setspecific(rpc_call_key, (void *) rcp);
 149                 rcp->valid = 0;
 150                 rcp->client = NULL;
 151         }
 152         if ((nettype == NULL) || (nettype[0] == 0))
 153                 nettype = "netpath";
 154         if (!(rcp->valid && rcp->pid == getpid() &&
 155                 (rcp->prognum == prognum) &&
 156                 (rcp->versnum == versnum) &&
 157                 (!strcmp(rcp->host, host)) &&
 158                 (!strcmp(rcp->nettype, nettype)))) {
 159                 int fd;
 160
 161                 rcp->valid = 0;
 162                 if (rcp->client)
 163                         CLNT_DESTROY(rcp->client);
 164                 /*
 165                  * Using the first successful transport for that type
 166                  */
 167                 rcp->client = clnt_create(host, prognum, versnum, nettype);
 168                 rcp->pid = getpid();
 169                 if (rcp->client == NULL) {
 170                         return (rpc_createerr.cf_stat);
 171                 }
 172                 /*
 173                  * Set time outs for connectionless case.  Do it
 174                  * unconditionally.  Faster than doing a t_getinfo()
 175                  * and then doing the right thing.
 176                  */
 177                 timeout.tv_usec = 0;
 178                 timeout.tv_sec = 5;
 179                 (void) CLNT_CONTROL(rcp->client,
 180                                 CLSET_RETRY_TIMEOUT, (char *)(void *)&timeout);
 181                 if (CLNT_CONTROL(rcp->client, CLGET_FD, (char *)(void *)&fd))
 182                         _fcntl(fd, F_SETFD, 1); /* make it "close on exec" */
 183                 rcp->prognum = prognum;
 184                 rcp->versnum = versnum;
 185                 if ((strlen(host) < (size_t)MAXHOSTNAMELEN) &&
 186                     (strlen(nettype) < (size_t)NETIDLEN)) {
 187                         (void) strcpy(rcp->host, host);
 188                         (void) strcpy(rcp->nettype, nettype);
 189                         rcp->valid = 1;
 190                 } else {
 191                         rcp->valid = 0;
 192                 }
 193         } /* else reuse old client */
 194         tottimeout.tv_sec = 25;
 195         tottimeout.tv_usec = 0;
 196         /*LINTED const castaway*/
 197         clnt_stat = CLNT_CALL(rcp->client, procnum, inproc, (char *) in,
 198             outproc, out, tottimeout);
 199         /*
 200          * if call failed, empty cache
 201          */
 202         if (clnt_stat != RPC_SUCCESS)
 203                 rcp->valid = 0;
 204         return (clnt_stat);
 205 }