[PATCH] elfcompress: Add sanity checks to make sure to not override variable

[Mark Wielaard]

The process_file code is a little tricky. Add sanity checks to make
sure shstrtab_name, shstrtab_newname and symstrents are only set once.

Signed-off-by: Mark Wielaard <m...@klomp.org>
---
 src/ChangeLog     |  5 +++++
 src/elfcompress.c | 24 ++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/src/ChangeLog b/src/ChangeLog
index b978f9ef..d08381e8 100644
--- a/src/ChangeLog
+++ b/src/ChangeLog
@@ -1,3 +1,8 @@
+2022-05-10  Mark Wielaard  <m...@klomp.org>
+
+       * elfcompress.c (process_file): Sanity check shstrtab_name,
+       shstrtab_newname and symstrents are only set once.
+
 2022-05-09  Mark Wielaard  <m...@klomp.org>
 
        * strip.c (remove_debug_relocations): Check gelf_getshdr, gelf_getrela,
diff --git a/src/elfcompress.c b/src/elfcompress.c
index 92f2fac8..6f18e47d 100644
--- a/src/elfcompress.c
+++ b/src/elfcompress.c
@@ -730,6 +730,15 @@ process_file (const char *fname)
                        {
                          shstrtab_size = size;
                          shstrtab_compressed = T_COMPRESS_GNU;
+                         if (shstrtab_name != NULL
+                             || shstrtab_newname != NULL)
+                           {
+                             error (0, 0, "Internal error,"
+                                          " shstrtab_name already set,"
+                                          " while handling section [%zd] %s",
+                                    ndx, sname);
+                             goto cleanup;
+                           }
                          shstrtab_name = xstrdup (sname);
                          shstrtab_newname = xstrdup (newname);
                        }
@@ -786,6 +795,15 @@ process_file (const char *fname)
                        {
                          shstrtab_size = size;
                          shstrtab_compressed = T_COMPRESS_ZLIB;
+                         if (shstrtab_name != NULL
+                             || shstrtab_newname != NULL)
+                           {
+                             error (0, 0, "Internal error,"
+                                          " shstrtab_name already set,"
+                                          " while handling section [%zd] %s",
+                                    ndx, sname);
+                             goto cleanup;
+                           }
                          shstrtab_name = xstrdup (sname);
                          shstrtab_newname = (newname == NULL
                                              ? NULL : xstrdup (newname));
@@ -919,6 +937,12 @@ process_file (const char *fname)
                }
              size_t elsize = gelf_fsize (elfnew, ELF_T_SYM, 1, EV_CURRENT);
              size_t syms = symd->d_size / elsize;
+             if (symstrents != NULL)
+               {
+                 error (0, 0, "Internal error, symstrents already set,"
+                        " while handling section [%zd] %s", ndx, name);
+                 goto cleanup;
+               }
              symstrents = xmalloc (syms * sizeof (Dwelf_Strent *));
              for (size_t i = 0; i < syms; i++)
                {
-- 
2.18.4