[PATCH] gencat: Get rid of alloca.
Joe Simmons-Talbott
josimmon@redhat.com
Mon Aug 28 13:20:08 GMT 2023
Ping.
On Tue, Aug 15, 2023 at 10:48:43AM -0400, Joe Simmons-Talbott via Libc-alpha wrote:
> On Thu, Aug 03, 2023 at 09:10:26AM -0400, Joe Simmons-Talbott via Libc-alpha wrote:
> > On Thu, Jul 13, 2023 at 12:04:10PM -0400, Joe Simmons-Talbott wrote:
> > > Convert to scratch_buffers to avoid potential stack overflow.
> >
> > Ping.
> Ping.
>
> Thanks,
> Joe
> >
> > Thanks,
> > Joe
> > >
> > > Checked on x86_64-linux-gnu and aarch64-linux-gnu.
> > > ---
> > > catgets/gencat.c | 37 +++++++++++++++++++++++++++++++------
> > > 1 file changed, 31 insertions(+), 6 deletions(-)
> > >
> > > diff --git a/catgets/gencat.c b/catgets/gencat.c
> > > index 9cacc801b5..63bdbf86a6 100644
> > > --- a/catgets/gencat.c
> > > +++ b/catgets/gencat.c
> > > @@ -32,6 +32,7 @@
> > > #include <limits.h>
> > > #include <nl_types.h>
> > > #include <obstack.h>
> > > +#include <scratch_buffer.h>
> > > #include <stdint.h>
> > > #include <stdio.h>
> > > #include <stdlib.h>
> > > @@ -854,6 +855,10 @@ write_out (struct catalog *catalog, const char *output_name,
> > > uint32_t *array1, *array2;
> > > size_t cnt;
> > > int fd;
> > > + struct scratch_buffer buf1;
> > > + scratch_buffer_init (&buf1);
> > > + struct scratch_buffer buf2;
> > > + scratch_buffer_init (&buf2);
> > >
> > > /* If not otherwise told try to read file with existing
> > > translations. */
> > > @@ -929,9 +934,19 @@ write_out (struct catalog *catalog, const char *output_name,
> > >
> > > uint32_t array_size = best_size * best_depth * sizeof (uint32_t) * 3;
> > > /* Allocate room for all needed arrays. */
> > > - array1 = (uint32_t *) alloca (array_size);
> > > + if (!scratch_buffer_set_array_size (&buf1, best_size * best_depth * 3,
> > > + sizeof (uint32_t)))
> > > + error (EXIT_FAILURE, ENOMEM, gettext ("cannot allocate memory"));
> > > + array1 = buf1.data;
> > > memset (array1, '\0', array_size);
> > > - array2 = (uint32_t *) alloca (array_size);
> > > +
> > > + if (!scratch_buffer_set_array_size (&buf2, best_size * best_depth * 3,
> > > + sizeof (uint32_t)))
> > > + {
> > > + scratch_buffer_free (&buf1);
> > > + error (EXIT_FAILURE, ENOMEM, gettext ("cannot allocate memory"));
> > > + }
> > > + array2 = buf2.data;
> > > obstack_init (&string_pool);
> > >
> > > set_run = catalog->all_sets;
> > > @@ -979,8 +994,12 @@ write_out (struct catalog *catalog, const char *output_name,
> > > {
> > > fd = creat (output_name, 0666);
> > > if (fd < 0)
> > > - error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"),
> > > - output_name);
> > > + {
> > > + scratch_buffer_free (&buf1);
> > > + scratch_buffer_free (&buf2);
> > > + error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"),
> > > + output_name);
> > > + }
> > > }
> > >
> > > /* Write out header. */
> > > @@ -1019,8 +1038,12 @@ write_out (struct catalog *catalog, const char *output_name,
> > > {
> > > fp = fopen (header_name, "w");
> > > if (fp == NULL)
> > > - error (EXIT_FAILURE, errno,
> > > - gettext ("cannot open output file `%s'"), header_name);
> > > + {
> > > + scratch_buffer_free (&buf1);
> > > + scratch_buffer_free (&buf2);
> > > + error (EXIT_FAILURE, errno,
> > > + gettext ("cannot open output file `%s'"), header_name);
> > > + }
> > > }
> > >
> > > /* Iterate over all sets and all messages. */
> > > @@ -1066,6 +1089,8 @@ write_out (struct catalog *catalog, const char *output_name,
> > > if (fp != stdout)
> > > fclose (fp);
> > > }
> > > + scratch_buffer_free (&buf1);
> > > + scratch_buffer_free (&buf2);
> > > }
> > >
> > >
> > > --
> > > 2.39.2
> > >
> >
>
More information about the Libc-alpha
mailing list