[RFC][PATCH v5 23/51] objtool: mcount: Move relocation entry size detection

[Matt Helsley]

Move where we detect the size of relocation entries we wish to
use into the first loop over the sections. This will allow us
to allocate the mcount location and relocation sections before
the next loop that collects them.

Signed-off-by: Matt Helsley <mhels...@vmware.com>
---
 tools/objtool/recordmcount.c | 8 +++++---
 tools/objtool/recordmcount.h | 5 ++---
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/tools/objtool/recordmcount.c b/tools/objtool/recordmcount.c
index 918e47217379..29f299e5bb52 100644
--- a/tools/objtool/recordmcount.c
+++ b/tools/objtool/recordmcount.c
@@ -458,7 +458,7 @@ static char const *has_rel_mcount(const struct section * 
const rels)
        return txts->name;
 }
 
-static unsigned tot_relsize(void)
+static unsigned tot_relsize(unsigned int *rel_entsize)
 {
        const struct section *sec;
        unsigned totrelsz = 0;
@@ -466,8 +466,10 @@ static unsigned tot_relsize(void)
 
        list_for_each_entry(sec, &lf->sections, list) {
                txtname = has_rel_mcount(sec);
-               if (txtname && is_mcounted_section_name(txtname))
-                       totrelsz += sec->sh.sh_size;
+               if (!(txtname && is_mcounted_section_name(txtname)))
+                       continue;
+               totrelsz += sec->sh.sh_size;
+               *rel_entsize = sec->sh.sh_entsize;
        }
        return totrelsz;
 }
diff --git a/tools/objtool/recordmcount.h b/tools/objtool/recordmcount.h
index 2733f7fde51d..5ca488f3471c 100644
--- a/tools/objtool/recordmcount.h
+++ b/tools/objtool/recordmcount.h
@@ -171,7 +171,7 @@ static int do_func(unsigned const reltype)
        uint_t *      mloc0;
        uint_t *      mlocp;
 
-       unsigned rel_entsize = 0;
+       unsigned int rel_entsize = 0;
        unsigned symsec_sh_link = 0;
 
        struct section *sec;
@@ -181,7 +181,7 @@ static int do_func(unsigned const reltype)
        if (find_section_by_name(lf, "__mcount_loc") != NULL)
                return 0;
 
-       totrelsz = tot_relsize();
+       totrelsz = tot_relsize(&rel_entsize);
        if (totrelsz == 0)
                return 0;
        mrel0 = umalloc(totrelsz);
@@ -211,7 +211,6 @@ static int do_func(unsigned const reltype)
                        if (result)
                                goto out;
 
-                       rel_entsize = sec->sh.sh_entsize;
                        mlocp = sift_rel_mcount(mlocp,
                                (void *)mlocp - (void *)mloc0, &mrelp,
                                sec, recsym, (uint_t)recval, reltype);
-- 
2.20.1