Use __readall and __readvall.
authordrepper <drepper>
Tue, 22 Feb 2005 22:47:45 +0000 (22:47 +0000)
committerdrepper <drepper>
Tue, 22 Feb 2005 22:47:45 +0000 (22:47 +0000)
nscd/initgrcache.c
nscd/nscd_getai.c
nscd/nscd_getgr_r.c
nscd/nscd_gethst_r.c
nscd/nscd_getpw_r.c
nscd/nscd_helper.c

index b464337..db01f1b 100644 (file)
@@ -1,5 +1,5 @@
 /* Cache handling for host lookup.
-   Copyright (C) 2004 Free Software Foundation, Inc.
+   Copyright (C) 2004, 2005 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Ulrich Drepper <drepper@redhat.com>, 2004.
 
@@ -343,7 +343,7 @@ addinitgroupsX (struct database_dyn *db, int fd, request_header *req,
             unnecessarily let the receiver wait.  */
          assert (fd != -1);
 
-         written = TEMP_FAILURE_RETRY (write (fd, &dataset->resp, total));
+         written = writeall (fd, &dataset->resp, total);
        }
 
 
index 866f7b2..4e3dfad 100644 (file)
@@ -119,8 +119,7 @@ __nscd_getai (const char *key, struct nscd_ai_result **result, int *h_errnop)
       if (respdata == NULL)
        {
          /* Read the data from the socket.  */
-         if ((size_t) TEMP_FAILURE_RETRY (__read (sock, resultbuf + 1,
-                                                  datalen)) == datalen)
+         if ((size_t) __readall (sock, resultbuf + 1, datalen) == datalen)
            {
              retval = 0;
              *result = resultbuf;
index 282912d..dae1c0d 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (C) 1998, 1999, 2000, 2002, 2003, 2004
+/* Copyright (C) 1998, 1999, 2000, 2002, 2003, 2004, 2005
    Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Thorsten Kukuk <kukuk@uni-paderborn.de>, 1998.
@@ -202,7 +202,7 @@ nscd_getgr_r (const char *key, size_t keylen, request_type type,
          total_len = vec[0].iov_len + vec[1].iov_len;
 
          /* Get this data.  */
-         size_t n = TEMP_FAILURE_RETRY (__readv (sock, vec, 2));
+         size_t n = __readvall (sock, vec, 2);
          if (__builtin_expect (n != total_len, 0))
            goto out_close;
        }
@@ -232,8 +232,7 @@ nscd_getgr_r (const char *key, size_t keylen, request_type type,
       retval = 0;
       if (gr_name == NULL)
        {
-         size_t n = TEMP_FAILURE_RETRY (__read (sock, resultbuf->gr_mem[0],
-                                                total_len));
+         size_t n = __readall (sock, resultbuf->gr_mem[0], total_len);
          if (__builtin_expect (n != total_len, 0))
            {
              /* The `errno' to some value != ERANGE.  */
index ef27e68..70ee38b 100644 (file)
@@ -299,8 +299,7 @@ nscd_gethst_r (const char *key, size_t keylen, request_type type,
              ++n;
            }
 
-         if ((size_t) TEMP_FAILURE_RETRY (__readv (sock, vec, n))
-             != total_len)
+         if ((size_t) __readvall (sock, vec, n) != total_len)
            goto out_close;
        }
       else
@@ -329,9 +328,8 @@ nscd_gethst_r (const char *key, size_t keylen, request_type type,
       /* And finally read the aliases.  */
       if (addr_list == NULL)
        {
-         if ((size_t) TEMP_FAILURE_RETRY (__read (sock,
-                                                  resultbuf->h_aliases[0],
-                                                  total_len)) == total_len)
+         if ((size_t) __readall (sock, resultbuf->h_aliases[0], total_len)
+             == total_len)
            {
              retval = 0;
              *result = resultbuf;
index fe5fb43..61a712c 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (C) 1998, 1999, 2003, 2004 Free Software Foundation, Inc.
+/* Copyright (C) 1998, 1999, 2003, 2004, 2005 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Thorsten Kukuk <kukuk@uni-paderborn.de>, 1998.
 
@@ -172,7 +172,7 @@ nscd_getpw_r (const char *key, size_t keylen, request_type type,
       retval = 0;
       if (pw_name == NULL)
        {
-         ssize_t nbytes = TEMP_FAILURE_RETRY (__read (sock, buffer, total));
+         ssize_t nbytes = __readall (sock, buffer, total);
 
          if (__builtin_expect (nbytes != total, 0))
            {
index ea4fb96..c99cb43 100644 (file)
 #include "nscd-client.h"
 
 
+ssize_t
+__readall (int fd, void *buf, size_t len)
+{
+  size_t n = len;
+  ssize_t ret;
+  do
+    {
+      ret = TEMP_FAILURE_RETRY (__read (fd, buf, n));
+      if (ret <= 0)
+       break;
+      buf = (char *) buf + ret;
+      n -= ret;
+    }
+  while (n > 0);
+  return ret < 0 ? ret : len - n;
+}
+
+
+ssize_t
+__readvall (int fd, const struct iovec *iov, int iovcnt)
+{
+  ssize_t ret = TEMP_FAILURE_RETRY (__readv (fd, iov, iovcnt));
+  if (ret <= 0)
+    return ret;
+
+  size_t total = 0;
+  for (int i = 0; i < iovcnt; ++i)
+    total += iov[i].iov_len;
+
+  if (ret < total)
+    {
+      struct iovec iov_buf[iovcnt];
+      ssize_t r = ret;
+
+      struct iovec *iovp = memcpy (iov_buf, iov, iovcnt * sizeof (*iov));
+      do
+       {
+         while (iovp->iov_len <= r)
+           {
+             r -= iovp->iov_len;
+             --iovcnt;
+             ++iovp;
+           }
+         iovp->iov_base = (char *) iovp->iov_base + r;
+         iovp->iov_len -= r;
+         r = TEMP_FAILURE_RETRY (__readv (fd, iovp, iovcnt));
+         if (r <= 0)
+           break;
+         ret += r;
+       }
+      while (ret < total);
+      if (r < 0)
+       ret = r;
+    }
+  return ret;
+}
+
+
 static int
 open_socket (void)
 {