[PATCH v3 10/16] objtool: Extract elf_strtab_concat()

[Peter Zijlstra]

Create a common helper to append strings to a strtab.

Signed-off-by: Peter Zijlstra (Intel) <pet...@infradead.org>
---
 tools/objtool/elf.c |   60 ++++++++++++++++++++++++++++++++--------------------
 1 file changed, 38 insertions(+), 22 deletions(-)

--- a/tools/objtool/elf.c
+++ b/tools/objtool/elf.c
@@ -666,13 +666,48 @@ struct elf *elf_open_read(const char *na
        return NULL;
 }
 
+static int elf_add_string(struct elf *elf, struct section *strtab, char *str)
+{
+       Elf_Data *data;
+       Elf_Scn *s;
+       int len;
+
+       if (!strtab)
+               strtab = find_section_by_name(elf, ".strtab");
+       if (!strtab) {
+               WARN("can't find .strtab section");
+               return -1;
+       }
+
+       s = elf_getscn(elf->elf, strtab->idx);
+       if (!s) {
+               WARN_ELF("elf_getscn");
+               return -1;
+       }
+
+       data = elf_newdata(s);
+       if (!data) {
+               WARN_ELF("elf_newdata");
+               return -1;
+       }
+
+       data->d_buf = str;
+       data->d_size = strlen(str) + 1;;
+       data->d_align = 1;
+
+       len = strtab->len;
+       strtab->len += data->d_size;
+       strtab->changed = true;
+
+       return len;
+}
+
 struct section *elf_create_section(struct elf *elf, const char *name,
                                   unsigned int sh_flags, size_t entsize, int 
nr)
 {
        struct section *sec, *shstrtab;
        size_t size = entsize * nr;
        Elf_Scn *s;
-       Elf_Data *data;
 
        sec = malloc(sizeof(*sec));
        if (!sec) {
@@ -729,7 +764,6 @@ struct section *elf_create_section(struc
        sec->sh.sh_addralign = 1;
        sec->sh.sh_flags = SHF_ALLOC | sh_flags;
 
-
        /* Add section name to .shstrtab (or .strtab for Clang) */
        shstrtab = find_section_by_name(elf, ".shstrtab");
        if (!shstrtab)
@@ -738,27 +772,9 @@ struct section *elf_create_section(struc
                WARN("can't find .shstrtab or .strtab section");
                return NULL;
        }
-
-       s = elf_getscn(elf->elf, shstrtab->idx);
-       if (!s) {
-               WARN_ELF("elf_getscn");
-               return NULL;
-       }
-
-       data = elf_newdata(s);
-       if (!data) {
-               WARN_ELF("elf_newdata");
+       sec->sh.sh_name = elf_add_string(elf, shstrtab, sec->name);
+       if (sec->sh.sh_name == -1)
                return NULL;
-       }
-
-       data->d_buf = sec->name;
-       data->d_size = strlen(name) + 1;
-       data->d_align = 1;
-
-       sec->sh.sh_name = shstrtab->len;
-
-       shstrtab->len += strlen(name) + 1;
-       shstrtab->changed = true;
 
        list_add_tail(&sec->list, &elf->sections);
        elf_hash_add(elf->section_hash, &sec->hash, sec->idx);