Add more tests.
[kopensolaris-gnu/glibc.git] / sunrpc / clnt_simp.c
index 043ce0a..631ec8a 100644 (file)
@@ -6,23 +6,23 @@
  * may copy or modify Sun RPC without charge, but are not authorized
  * to license or distribute it to anyone else except as part of a product or
  * program developed by the user.
- * 
+ *
  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
- * 
+ *
  * Sun RPC is provided with no support and without any obligation on the
  * part of Sun Microsystems, Inc. to assist in its use, correction,
  * modification or enhancement.
- * 
+ *
  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  * OR ANY PART THEREOF.
- * 
+ *
  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  * or profits or other special, indirect and consequential damages, even if
  * Sun has been advised of the possibility of such damages.
- * 
+ *
  * Sun Microsystems, Inc.
  * 2550 Garcia Avenue
  * Mountain View, California  94043
 static char sccsid[] = "@(#)clnt_simple.c 1.35 87/08/11 Copyr 1984 Sun Micro";
 #endif
 
-/* 
+/*
  * clnt_simple.c
  * Simplified front end to rpc.
  *
  * Copyright (C) 1984, Sun Microsystems, Inc.
  */
 
+#include <alloca.h>
+#include <errno.h>
 #include <stdio.h>
+#include <unistd.h>
 #include <rpc/rpc.h>
 #include <sys/socket.h>
 #include <netdb.h>
-#include <strings.h>
+#include <string.h>
 
-static struct callrpc_private {
-       CLIENT  *client;
-       int     socket;
-       int     oldprognum, oldversnum, valid;
-       char    *oldhost;
-} *callrpc_private;
+struct callrpc_private_s
+  {
+    CLIENT *client;
+    int socket;
+    u_long oldprognum, oldversnum, valid;
+    char *oldhost;
+  };
+#ifdef _RPC_THREAD_SAFE_
+#define callrpc_private RPC_THREAD_VARIABLE(callrpc_private_s)
+#else
+static struct callrpc_private_s *callrpc_private;
+#endif
 
-callrpc(host, prognum, versnum, procnum, inproc, in, outproc, out)
-       char *host;
-       xdrproc_t inproc, outproc;
-       char *in, *out;
+int
+callrpc (const char *host, u_long prognum, u_long versnum, u_long procnum,
+        xdrproc_t inproc, const char *in, xdrproc_t outproc, char *out)
 {
-       register struct callrpc_private *crp = callrpc_private;
-       struct sockaddr_in server_addr;
-       enum clnt_stat clnt_stat;
-       struct hostent *hp;
-       struct timeval timeout, tottimeout;
+  struct callrpc_private_s *crp = callrpc_private;
+  struct sockaddr_in server_addr;
+  enum clnt_stat clnt_stat;
+  struct hostent hostbuf, *hp;
+  struct timeval timeout, tottimeout;
 
-       if (crp == 0) {
-               crp = (struct callrpc_private *)calloc(1, sizeof (*crp));
-               if (crp == 0)
-                       return (0);
-               callrpc_private = crp;
+  if (crp == 0)
+    {
+      crp = (struct callrpc_private_s *) calloc (1, sizeof (*crp));
+      if (crp == 0)
+       return 0;
+      callrpc_private = crp;
+    }
+  if (crp->oldhost == NULL)
+    {
+      crp->oldhost = malloc (256);
+      crp->oldhost[0] = 0;
+      crp->socket = RPC_ANYSOCK;
+    }
+  if (crp->valid && crp->oldprognum == prognum && crp->oldversnum == versnum
+      && strcmp (crp->oldhost, host) == 0)
+    {
+      /* reuse old client */
+    }
+  else
+    {
+      size_t buflen;
+      char *buffer;
+      int herr;
+
+      crp->valid = 0;
+      if (crp->socket != RPC_ANYSOCK)
+       {
+         (void) __close (crp->socket);
+         crp->socket = RPC_ANYSOCK;
        }
-       if (crp->oldhost == NULL) {
-               crp->oldhost = malloc(256);
-               crp->oldhost[0] = 0;
-               crp->socket = RPC_ANYSOCK;
+      if (crp->client)
+       {
+         clnt_destroy (crp->client);
+         crp->client = NULL;
        }
-       if (crp->valid && crp->oldprognum == prognum && crp->oldversnum == versnum
-               && strcmp(crp->oldhost, host) == 0) {
-               /* reuse old client */          
-       } else {
-               crp->valid = 0;
-               (void)close(crp->socket);
-               crp->socket = RPC_ANYSOCK;
-               if (crp->client) {
-                       clnt_destroy(crp->client);
-                       crp->client = NULL;
-               }
-               if ((hp = gethostbyname(host)) == NULL)
-                       return ((int) RPC_UNKNOWNHOST);
-               timeout.tv_usec = 0;
-               timeout.tv_sec = 5;
-               bcopy(hp->h_addr, (char *)&server_addr.sin_addr, hp->h_length);
-               server_addr.sin_family = AF_INET;
-               server_addr.sin_port =  0;
-               if ((crp->client = clntudp_create(&server_addr, (u_long)prognum,
-                   (u_long)versnum, timeout, &crp->socket)) == NULL)
-                       return ((int) rpc_createerr.cf_stat);
-               crp->valid = 1;
-               crp->oldprognum = prognum;
-               crp->oldversnum = versnum;
-               (void) strcpy(crp->oldhost, host);
+
+      buflen = 1024;
+      buffer = __alloca (buflen);
+      while (__gethostbyname_r (host, &hostbuf, buffer, buflen,
+                               &hp, &herr) != 0
+            || hp == NULL)
+       if (herr != NETDB_INTERNAL || errno != ERANGE)
+         return (int) RPC_UNKNOWNHOST;
+       else
+         {
+           /* Enlarge the buffer.  */
+           buflen *= 2;
+           buffer = __alloca (buflen);
+         }
+
+      timeout.tv_usec = 0;
+      timeout.tv_sec = 5;
+      memcpy ((char *) &server_addr.sin_addr, hp->h_addr, hp->h_length);
+      server_addr.sin_family = AF_INET;
+      server_addr.sin_port = 0;
+      if ((crp->client = INTUSE(clntudp_create) (&server_addr, (u_long) prognum,
+                         (u_long) versnum, timeout, &crp->socket)) == NULL)
+       return (int) get_rpc_createerr().cf_stat;
+      crp->valid = 1;
+      crp->oldprognum = prognum;
+      crp->oldversnum = versnum;
+      (void) strncpy (crp->oldhost, host, 255);
+      crp->oldhost[255] = '\0';
+    }
+  tottimeout.tv_sec = 25;
+  tottimeout.tv_usec = 0;
+  clnt_stat = clnt_call (crp->client, procnum, inproc, (char *) in,
+                        outproc, out, tottimeout);
+  /*
+   * if call failed, empty cache
+   */
+  if (clnt_stat != RPC_SUCCESS)
+    crp->valid = 0;
+  return (int) clnt_stat;
+}
+
+#ifdef _RPC_THREAD_SAFE_
+void
+__rpc_thread_clnt_cleanup (void)
+{
+       struct callrpc_private_s *rcp = RPC_THREAD_VARIABLE(callrpc_private_s);
+
+       if (rcp) {
+               if (rcp->client)
+                       CLNT_DESTROY (rcp->client);
+               free (rcp);
        }
-       tottimeout.tv_sec = 25;
-       tottimeout.tv_usec = 0;
-       clnt_stat = clnt_call(crp->client, procnum, inproc, in,
-           outproc, out, tottimeout);
-       /* 
-        * if call failed, empty cache
-        */
-       if (clnt_stat != RPC_SUCCESS)
-               crp->valid = 0;
-       return ((int) clnt_stat);
 }
+#endif /* _RPC_THREAD_SAFE_ */