Change log directory to /var/log (FHS).
[kopensolaris-gnu/glibc.git] / nscd / nscd_getgr_r.c
index ec4f5a1..aa9c3fd 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (C) 1998 Free Software Foundation, Inc.
+/* Copyright (C) 1998, 1999 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Thorsten Kukuk <kukuk@uni-paderborn.de>, 1998.
 
@@ -19,6 +19,7 @@
 
 #include <errno.h>
 #include <grp.h>
+#include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/uio.h>
 #include <sys/un.h>
 
-#include "nscd.h"
+#include "nscd-client.h"
 #include "nscd_proto.h"
 
 int __nss_not_use_nscd_group;
 
-static int __nscd_getgr_r (const char *key, request_type type,
-                          struct group *resultbuf, char *buffer,
-                          size_t buflen);
+static int nscd_getgr_r (const char *key, size_t keylen, request_type type,
+                        struct group *resultbuf, char *buffer,
+                        size_t buflen) internal_function;
+
 
 int
 __nscd_getgrnam_r (const char *name, struct group *resultbuf, char *buffer,
                   size_t buflen)
 {
-  if (name == NULL)
-    return 1;
-
-  return __nscd_getgr_r (name, GETGRBYNAME, resultbuf, buffer, buflen);
+  return nscd_getgr_r (name, strlen (name) + 1, GETGRBYNAME, resultbuf,
+                      buffer, buflen);
 }
 
+
 int
 __nscd_getgrgid_r (gid_t gid, struct group *resultbuf, char *buffer,
                   size_t buflen)
 {
-  char *p = buffer;
-  int plen;
+  char buf[12];
+  size_t n;
 
-  plen = __snprintf (buffer, buflen, "%d", gid);
-  if (plen == -1)
-    {
-      __set_errno (ERANGE);
-      return -1;
-    }
-  p = buffer + plen + 1;
+  n = __snprintf (buf, sizeof (buf), "%d", gid) + 1;
 
-  return __nscd_getgr_r (buffer, GETGRBYGID, resultbuf, p, buflen - plen -1);
+  return nscd_getgr_r (buf, n, GETGRBYGID, resultbuf, buffer, buflen);
 }
 
+
 /* Create a socket connected to a name. */
 static int
-nscd_open_socket (void)
+open_socket (void)
 {
   struct sockaddr_un addr;
   int sock;
@@ -91,45 +87,44 @@ nscd_open_socket (void)
   return sock;
 }
 
+
 static int
-__nscd_getgr_r (const char *key, request_type type, struct group *resultbuf,
-               char *buffer, size_t buflen)
+internal_function
+nscd_getgr_r (const char *key, size_t keylen, request_type type,
+             struct group *resultbuf, char *buffer, size_t buflen)
 {
-  int sock = nscd_open_socket ();
+  int sock = open_socket ();
   request_header req;
   gr_response_header gr_resp;
   ssize_t nbytes;
-  size_t maxiov;
-  size_t sum;
+  struct iovec vec[2];
 
   if (sock == -1)
     {
       __nss_not_use_nscd_group = 1;
-      return 1;
+      return -1;
     }
 
   req.version = NSCD_VERSION;
   req.type = type;
-  req.key_len = strlen (key);
-  nbytes = __write (sock, &req, sizeof (request_header));
-  if (nbytes != sizeof (request_header))
-    {
-      __close (sock);
-      return 1;
-    }
+  req.key_len = keylen;
+
+  vec[0].iov_base = &req;
+  vec[0].iov_len = sizeof (request_header);
+  vec[1].iov_base = (void *) key;
+  vec[1].iov_len = keylen;
 
-  nbytes = __write (sock, key, req.key_len);
-  if (nbytes != req.key_len)
+  if (__writev (sock, vec, 2) != sizeof (request_header) + keylen)
     {
       __close (sock);
-      return 1;
+      return -1;
     }
 
   nbytes = __read (sock, &gr_resp, sizeof (gr_response_header));
   if (nbytes != sizeof (gr_response_header))
     {
       __close (sock);
-      return 1;
+      return -1;
     }
 
   if (gr_resp.found == -1)
@@ -137,129 +132,93 @@ __nscd_getgr_r (const char *key, request_type type, struct group *resultbuf,
       /* The daemon does not cache this database.  */
       __close (sock);
       __nss_not_use_nscd_group = 1;
-      return 1;
+      return -1;
     }
 
   if (gr_resp.found == 1)
     {
-      struct iovec *vec;
       size_t *len;
       char *p = buffer;
-      int nblocks;
       size_t total_len;
       uintptr_t align;
+      size_t cnt;
 
-      /* A first check whether the buffer is sufficently large is possible.  */
-      if (buflen < gr_resp.gr_name_len + 1 + gr_resp.gr_passwd_len + 1)
+      /* Now allocate the buffer the array for the group members.  We must
+        align the pointer.  */
+      align = ((__alignof__ (char *) - (p - ((char *) 0)))
+              & (__alignof__ (char *) - 1));
+      total_len = align + (1 + gr_resp.gr_mem_cnt) * sizeof (char *)
+                 + gr_resp.gr_name_len + gr_resp.gr_passwd_len;
+      if (buflen < total_len)
        {
+       no_room:
          __set_errno (ERANGE);
          __close (sock);
-         return -1;
+         return ERANGE;
        }
+      buflen -= total_len;
 
-      /* Allocate the IOVEC.  */
-      vec = alloca ((2 + gr_resp.gr_mem_len) * sizeof (struct iovec));
-      len = alloca (gr_resp.gr_mem_len * sizeof (size_t));
+      p += align;
+      resultbuf->gr_mem = (char **) p;
+      p += (1 + gr_resp.gr_mem_cnt) * sizeof (char *);
 
-      vec[0].iov_base = resultbuf->gr_name = p;
-      vec[0].iov_len = gr_resp.gr_name_len;
-      total_len = gr_resp.gr_name_len;
-      p += gr_resp.gr_name_len + 1;
+      /* Set pointers for strings.  */
+      resultbuf->gr_name = p;
+      p += gr_resp.gr_name_len;
+      resultbuf->gr_passwd = p;
+      p += gr_resp.gr_passwd_len;
 
-      vec[1].iov_base = resultbuf->gr_passwd = p;
-      vec[1].iov_len = gr_resp.gr_passwd_len;
-      total_len += gr_resp.gr_passwd_len;
-      p += gr_resp.gr_passwd_len + 1;
-      buflen -= total_len;
-      nblocks = 2;
+      /* Fill in what we know now.  */
+      resultbuf->gr_gid = gr_resp.gr_gid;
 
-      if (gr_resp.gr_mem_len > 0)
-       {
-         vec[2].iov_base = len;
-         vec[2].iov_len = gr_resp.gr_mem_len * sizeof (size_t);
-         total_len += gr_resp.gr_mem_len * sizeof (size_t);
-         nblocks = 3;
-       }
+      /* Allocate array to store lengths.  */
+      len = alloca (gr_resp.gr_mem_cnt * sizeof (size_t));
+
+      total_len = gr_resp.gr_mem_cnt * sizeof (size_t);
+      vec[0].iov_base = len;
+      vec[0].iov_len = total_len;
+      vec[1].iov_base = resultbuf->gr_name;
+      vec[1].iov_len = gr_resp.gr_name_len + gr_resp.gr_passwd_len;
+      total_len += gr_resp.gr_name_len + gr_resp.gr_passwd_len;
 
       /* Get this data.  */
-      if (__readv (sock, vec, nblocks) != total_len)
+      if (__readv (sock, vec, 2) != total_len)
        {
          __close (sock);
-         return 1;
+         return -1;
        }
 
-      /* Now we know the sizes.  First terminate the strings we just read. */
-      resultbuf->gr_name[gr_resp.gr_name_len] = '\0';
-      resultbuf->gr_passwd[gr_resp.gr_passwd_len] = '\0';
+      /* Clear the terminating entry.  */
+      resultbuf->gr_mem[gr_resp.gr_mem_cnt] = NULL;
 
-      resultbuf->gr_gid = gr_resp.gr_gid;
-
-      /* Now allocate the buffer the array for the group members.  We must
-        align the pointer.  */
-      align = ((__alignof__ (char *) - (p - ((char *) 0)))
-              & (__alignof__ (char *) - 1));
-      if (align + (1 + gr_resp.gr_mem_len) * sizeof (char *) > buflen)
+      /* Prepare reading the group members.  */
+      total_len = 0;
+      for (cnt = 0; cnt < gr_resp.gr_mem_cnt; ++cnt)
        {
-         __set_errno (ERANGE);
-         __close (sock);
-         return -1;
+         resultbuf->gr_mem[cnt] = p;
+         total_len += len[cnt];
+         p += len[cnt];
        }
-      p += align;
-      resultbuf->gr_mem = (char **) p;
-      p += (1 + gr_resp.gr_mem_len) * sizeof (char *);
-      buflen -= align + (1 + gr_resp.gr_mem_len) * sizeof (char *);
 
-      resultbuf->gr_mem[gr_resp.gr_mem_len] = NULL;
+      if (total_len > buflen)
+       goto no_room;
 
-      if (gr_resp.gr_mem_len > 0)
+      if (__read (sock, resultbuf->gr_mem[0], total_len) != total_len)
        {
-         /* Prepare reading the group members.  */
-         size_t i;
-
-         total_len = 0;
-         for (i = 0; i < gr_resp.gr_mem_len; ++i)
-           {
-             if (len[i] >= buflen)
-               {
-                 __set_errno (ERANGE);
-                 __close (sock);
-                 return -1;
-               }
-
-             vec[i].iov_base = resultbuf->gr_mem[i] = p;
-             vec[i].iov_len = len[i];
-             total_len += len[i];
-             buflen -= len[i];
-             p += len[i];
-             *p++ = '\0';
-           }
-
-#ifdef UIO_MAXIOV
-         maxiov = UIO_MAXIOV;
-#else
-         maxiov = sysconf (_SC_UIO_MAXIOV);
-#endif
-
-         sum = 0;
-         while (i > maxiov)
-           {
-             sum += __readv (sock, vec, maxiov);
-             vec += maxiov;
-             i -= maxiov;
-           }
-
-         if (sum + __readv (sock, vec, i) != total_len)
-           {
-             __close (sock);
-             return -1;
-           }
+         __close (sock);
+         /* The `errno' to some value != ERANGE.  */
+         __set_errno (ENOENT);
+         return ENOENT;
        }
+
       __close (sock);
       return 0;
     }
   else
     {
       __close (sock);
-      return -1;
+      /* The `errno' to some value != ERANGE.  */
+      __set_errno (ENOENT);
+      return ENOENT;
     }
 }