about summary refs log tree commit diff
path: root/sysdeps/posix/readv.c
diff options
context:
space:
mode:
Diffstat (limited to 'sysdeps/posix/readv.c')
-rw-r--r--sysdeps/posix/readv.c42
1 files changed, 36 insertions, 6 deletions
diff --git a/sysdeps/posix/readv.c b/sysdeps/posix/readv.c
index 6349e242eb..89fe1af7d3 100644
--- a/sysdeps/posix/readv.c
+++ b/sysdeps/posix/readv.c
@@ -1,4 +1,4 @@
-/* Copyright (C) 1991, 1992, 1996, 1997 Free Software Foundation, Inc.
+/* Copyright (C) 1991, 1992, 1996, 1997, 2002 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
 
    The GNU C Library is free software; you can redistribute it and/or
@@ -19,6 +19,9 @@
 #include <stdlib.h>
 #include <unistd.h>
 #include <string.h>
+#include <limits.h>
+#include <stdbool.h>
+#include <sys/param.h>
 #include <sys/uio.h>
 
 /* Read data from file descriptor FD, and put the result in the
@@ -33,17 +36,41 @@ __readv (fd, vector, count)
      int count;
 {
   char *buffer;
+  char *buffer_start;
   size_t bytes;
   int bytes_read;
   int i;
+  bool use_malloc = false;
 
   /* Find the total number of bytes to be read.  */
   bytes = 0;
   for (i = 0; i < count; ++i)
-    bytes += vector[i].iov_len;
+    {
+      /* Check for ssize_t overflow.  */
+      if (SSIZE_MAX - bytes < vector[i].iov_len)
+	{
+	  errno = EINVAL;
+	  return -1;
+	}
+      bytes += vector[i].iov_len;
+    }
+
+  /* Allocate a temporary buffer to hold the data.  We should normally
+     use alloca since it's faster and does not require synchronization
+     with other threads.  But we cannot if the amount of memory
+     required is too large.  Use 512k as the limit.  */
+  if (bytes < 512 * 1024)
+    buffer = (char *) __alloca (bytes);
+  else
+    {
+      buffer = (char *) malloc (bytes);
+      if (buffer == NULL)
+	/* XXX I don't know whether it is acceptable to try reading
+	   the data in chunks.  Probably not so we just fail here.  */
+	return -1;
 
-  /* Allocate a temporary buffer to hold the data.  */
-  buffer = (char *) __alloca (bytes);
+      use_malloc = true;
+    }
 
   /* Read the data.  */
   bytes_read = __read (fd, buffer, bytes);
@@ -52,10 +79,10 @@ __readv (fd, vector, count)
 
   /* Copy the data from BUFFER into the memory specified by VECTOR.  */
   bytes = bytes_read;
+  buffer_start = buffer;
   for (i = 0; i < count; ++i)
     {
-#define	min(a, b)	((a) > (b) ? (b) : (a))
-      size_t copy = min (vector[i].iov_len, bytes);
+      size_t copy = MIN (vector[i].iov_len, bytes);
 
       (void) memcpy ((void *) vector[i].iov_base, (void *) buffer, copy);
 
@@ -65,6 +92,9 @@ __readv (fd, vector, count)
 	break;
     }
 
+  if (use_malloc)
+    free (buffer_start);
+
   return bytes_read;
 }
 #ifndef __readv