[PATCH v2] getaddrinfo: Get rid of alloca
Joe Simmons-Talbott
josimmon@redhat.com
Tue Aug 29 14:56:06 GMT 2023
Ping.
On Fri, Jun 30, 2023 at 10:30:50AM -0400, Joe Simmons-Talbott wrote:
> Use a scratch_buffer rather than alloca to avoid potential stack
> overflow.
> ---
> Changes to v1:
> * Don't use alloc_size but rather rely on * scratch_buffer_set_array_size's
> overflow checks.
>
> sysdeps/posix/getaddrinfo.c | 24 +++++++++---------------
> 1 file changed, 9 insertions(+), 15 deletions(-)
>
> diff --git a/sysdeps/posix/getaddrinfo.c b/sysdeps/posix/getaddrinfo.c
> index 0356b622be..be3b3f6e45 100644
> --- a/sysdeps/posix/getaddrinfo.c
> +++ b/sysdeps/posix/getaddrinfo.c
> @@ -2404,22 +2404,17 @@ getaddrinfo (const char *name, const char *service,
> struct addrinfo *q;
> struct addrinfo *last = NULL;
> char *canonname = NULL;
> - bool malloc_results;
> - size_t alloc_size = nresults * (sizeof (*results) + sizeof (size_t));
> + struct scratch_buffer buf;
> + scratch_buffer_init (&buf);
>
> - malloc_results
> - = !__libc_use_alloca (alloc_size);
> - if (malloc_results)
> + if (!scratch_buffer_set_array_size (&buf, nresults,
> + sizeof (*results) + sizeof (size_t)))
> {
> - results = malloc (alloc_size);
> - if (results == NULL)
> - {
> - __free_in6ai (in6ai);
> - return EAI_MEMORY;
> - }
> + __free_in6ai (in6ai);
> + return EAI_MEMORY;
> }
> - else
> - results = alloca (alloc_size);
> + results = buf.data;
> +
> order = (size_t *) (results + nresults);
>
> /* Now we definitely need the interface information. */
> @@ -2590,8 +2585,7 @@ getaddrinfo (const char *name, const char *service,
> /* Fill in the canonical name into the new first entry. */
> p->ai_canonname = canonname;
>
> - if (malloc_results)
> - free (results);
> + scratch_buffer_free (&buf);
> }
>
> __free_in6ai (in6ai);
> --
> 2.39.2
>
More information about the Libc-alpha
mailing list