mirror/glibc
2
0
Fork 0
mirror of git://sourceware.org/git/glibc.git synced 2025-01-30 12:31:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

getipv4sourcefilter: Get rid of alloca

Browse Source 
Use a scratch_buffer rather than alloca to avoid potential stack
overflows.
Reviewed-by: Adhemerval Zanella  <adhemerval.zanella@linaro.org>
This commit is contained in:
Joe Simmons-Talbott 2023-05-30 14:32:55 -04:00 committed by Adhemerval Zanella
parent d1eaab5a79
commit 884012db20
1 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
sysdeps/unix/sysv/linux/getipv4sourcefilter.c
View File

@ -16,9 +16,9 @@
License along with the GNU C Library; if not, see
<https://www.gnu.org/licenses/>. */
#include <alloca.h>
#include <errno.h>
#include <stdlib.h>
#include <scratch_buffer.h>
#include <string.h>
#include <stdint.h>
#include <netinet/in.h>
@ -33,17 +33,12 @@ getipv4sourcefilter (int s, struct in_addr interface, struct in_addr group,
/* We have to create an struct ip_msfilter object which we can pass
to the kernel. */
socklen_t needed = IP_MSFILTER_SIZE (*numsrc);
int use_alloca = __libc_use_alloca (needed);
struct ip_msfilter *imsf;
if (use_alloca)
imsf = (struct ip_msfilter *) alloca (needed);
else
{
imsf = (struct ip_msfilter *) malloc (needed);
if (imsf == NULL)
return -1;
}
struct scratch_buffer buf;
scratch_buffer_init (&buf);
if (!scratch_buffer_set_array_size (&buf, 1, needed))
return -1;
struct ip_msfilter *imsf = buf.data;
imsf->imsf_multiaddr = group;
imsf->imsf_interface = interface;
@ -61,12 +56,7 @@ getipv4sourcefilter (int s, struct in_addr interface, struct in_addr group,
*numsrc = imsf->imsf_numsrc;
}
if (! use_alloca)
{
int save_errno = errno;
free (imsf);
__set_errno (save_errno);
}
scratch_buffer_free (&buf);
return result;
}