[RFC][PATCH v5 45/51] objtool: mcount: Move mcount section test to objtool ELF

From: Matt Helsley
Date: Thu Jun 18 2020 - 16:41:42 EST


In preparation for removing this first pass by mcount move
the test for whether a section is mcountable into objtool's
ELF read code, much like we did with the symbol name. A
subsequent patch will eliminate the tot_relsize() function
and thereby reduce the number of passes through the ELF
data.

Signed-off-by: Matt Helsley <mhelsley@xxxxxxxxxx>
---
tools/objtool/elf.c | 16 ++++++++++++++++
tools/objtool/elf.h | 2 +-
tools/objtool/mcount.c | 36 +++++++++++-------------------------
3 files changed, 28 insertions(+), 26 deletions(-)

diff --git a/tools/objtool/elf.c b/tools/objtool/elf.c
index 666cbc01c332..dffdc3264a25 100644
--- a/tools/objtool/elf.c
+++ b/tools/objtool/elf.c
@@ -308,6 +308,20 @@ struct reloc *find_reloc_by_dest(const struct elf *elf, struct section *sec, uns
return find_reloc_by_dest_range(elf, sec, offset, 1);
}

+/* Names of the sections that could contain calls to mcount. */
+static int is_mcounted_section_name(char const *const txtname)
+{
+ return strncmp(".text", txtname, 5) == 0 ||
+ strcmp(".init.text", txtname) == 0 ||
+ strcmp(".ref.text", txtname) == 0 ||
+ strcmp(".sched.text", txtname) == 0 ||
+ strcmp(".spinlock.text", txtname) == 0 ||
+ strcmp(".irqentry.text", txtname) == 0 ||
+ strcmp(".softirqentry.text", txtname) == 0 ||
+ strcmp(".kprobes.text", txtname) == 0 ||
+ strcmp(".cpuidle.text", txtname) == 0;
+}
+
static int read_sections(struct elf *elf)
{
Elf_Scn *s = NULL;
@@ -370,6 +384,8 @@ static int read_sections(struct elf *elf)
}
sec->len = sec->sh.sh_size;

+ sec->mcounted = is_mcounted_section_name(sec->name);
+
list_add_tail(&sec->list, &elf->sections);
elf_hash_add(elf->section_hash, &sec->hash, sec->idx);
elf_hash_add(elf->section_name_hash, &sec->name_hash, str_hash(sec->name));
diff --git a/tools/objtool/elf.h b/tools/objtool/elf.h
index f0b02824f9d3..db207e93a1e6 100644
--- a/tools/objtool/elf.h
+++ b/tools/objtool/elf.h
@@ -39,7 +39,7 @@ struct section {
char *name;
int idx;
unsigned int len;
- bool changed, text, rodata, noinstr;
+ bool changed, text, rodata, noinstr, mcounted;
};

enum symbol_class {
diff --git a/tools/objtool/mcount.c b/tools/objtool/mcount.c
index 7d7843f7cd44..a75d918a9fa5 100644
--- a/tools/objtool/mcount.c
+++ b/tools/objtool/mcount.c
@@ -144,20 +144,6 @@ static int make_nop_arm64(struct section *txts, size_t const offset)
return elf_write_insn(lf, txts, offset, 4, ideal_nop);
}

-/* Names of the sections that could contain calls to mcount. */
-static int is_mcounted_section_name(char const *const txtname)
-{
- return strncmp(".text", txtname, 5) == 0 ||
- strcmp(".init.text", txtname) == 0 ||
- strcmp(".ref.text", txtname) == 0 ||
- strcmp(".sched.text", txtname) == 0 ||
- strcmp(".spinlock.text", txtname) == 0 ||
- strcmp(".irqentry.text", txtname) == 0 ||
- strcmp(".softirqentry.text", txtname) == 0 ||
- strcmp(".kprobes.text", txtname) == 0 ||
- strcmp(".cpuidle.text", txtname) == 0;
-}
-
/*
* MIPS mcount long call has 2 _mcount symbols, only the position of the 1st
* _mcount symbol is needed for dynamic function tracer, with it, to disable
@@ -288,7 +274,7 @@ static int nop_mcount(struct section * const rels,
return 0;
}

-static char const *has_rel_mcount(const struct section * const rels)
+static struct section const *has_rel_mcount(const struct section * const rels)
{
const struct section *txts;
if (rels->sh.sh_type != SHT_REL && rels->sh.sh_type != SHT_RELA)
@@ -297,17 +283,17 @@ static char const *has_rel_mcount(const struct section * const rels)
if ((txts->sh.sh_type != SHT_PROGBITS) ||
!(txts->sh.sh_flags & SHF_EXECINSTR))
return NULL;
- return txts->name;
+ return txts;
}

static void tot_relsize(unsigned int *rel_entsize)
{
const struct section *sec;
- char const *txtname;
+ const struct section *txts;

list_for_each_entry(sec, &lf->sections, list) {
- txtname = has_rel_mcount(sec);
- if (!(txtname && is_mcounted_section_name(txtname)))
+ txts = has_rel_mcount(sec);
+ if (!(txts && txts->mcounted))
continue;
*rel_entsize = sec->sh.sh_entsize;
}
@@ -441,25 +427,25 @@ static int do_mcount(unsigned const reltype, size_t rela_size)
mrels->sh.sh_flags = 0; /* clear SHF_INFO_LINK */

list_for_each_entry(sec, &lf->sections, list) {
- char const *txtname;
+ struct section const *txts;

- txtname = has_rel_mcount(sec);
- if (txtname && is_mcounted_section_name(txtname)) {
+ txts = has_rel_mcount(sec);
+ if (txts && txts->mcounted) {
unsigned long recval = 0;
unsigned int recsym;

if (find_section_sym_index(sec->sh.sh_info,
- txtname, &recval, &recsym))
+ txts->name, &recval, &recsym))
goto out;

sift_rel_mcount(&r_offset, sec, mrels, mlocs,
recsym, recval, reltype, is_rela);
- } else if (txtname && (warn_on_notrace_sect || make_nop)) {
+ } else if (txts && (warn_on_notrace_sect || make_nop)) {
/*
* This section is ignored by ftrace, but still
* has mcount calls. Convert them to nops now.
*/
- if (nop_mcount(sec, txtname) < 0)
+ if (nop_mcount(sec, txts->name) < 0)
goto out;
}
}
--
2.20.1