mirrors/glibc
1
0
Fork 0
mirror of https://sourceware.org/git/glibc.git synced 2024-11-30 13:13:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

hurd: readv: Get rid of alloca

Browse Source 
Replace alloca with a scratch_buffer to avoid potential stack overflows.

Checked on i686-gnu and x86_64-linux-gnu
Message-Id: <20230619144334.2902429-1-josimmon@redhat.com>
This commit is contained in:
Joe Simmons-Talbott 2023-06-19 10:43:34 -04:00 committed by Samuel Thibault
parent c6957bddb9
commit 9e6863a537
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
sysdeps/posix/readv.c
View File

@ -19,6 +19,7 @@
#include <unistd.h>
#include <string.h>
#include <limits.h>
#include <scratch_buffer.h>
#include <stdbool.h>
#include <sys/param.h>
#include <sys/uio.h>
@ -26,9 +27,9 @@
static void
ifree (char **ptrp)
ifree (struct scratch_buffer *sbuf)
{
free (*ptrp);
scratch_buffer_free (sbuf);
}
/* Read data from file descriptor FD, and put the result in the
@ -52,20 +53,15 @@ __readv (int fd, const struct iovec *vector, int count)
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. */
char *buffer;
char *malloced_buffer __attribute__ ((__cleanup__ (ifree))) = NULL;
if (__libc_use_alloca (bytes))
buffer = (char *) __alloca (bytes);
else
{
malloced_buffer = buffer = (char *) malloc (bytes);
if (buffer == NULL)
return -1;
}
/* Allocate a temporary buffer to hold the data. Use a scratch_buffer
since it's faster for small buffer sizes but can handle larger
allocations as well. */
struct scratch_buffer __attribute__ ((__cleanup__ (ifree))) buf;
scratch_buffer_init (&buf);
if (!scratch_buffer_set_array_size (&buf, 1, bytes))
return -1;
char *buffer = buf.data;
/* Read the data. */
ssize_t bytes_read = __read (fd, buffer, bytes);