In preparation for klp-build, enable "classic" objtool to work on
livepatch modules:

  - Avoid duplicate symbol/section warnings for prefix symbols and the
    .static_call_sites and __mcount_loc sections which may have already
    been extracted by klp diff.

  - Add __klp_funcs to the IBT function pointer section whitelist.

  - Prevent KLP symbols from getting incorrectly classified as cold
    subfunctions.

Signed-off-by: Josh Poimboeuf <jpoim...@kernel.org>
---
 tools/objtool/check.c                   | 48 +++++++++++++++++++++----
 tools/objtool/elf.c                     |  5 ++-
 tools/objtool/include/objtool/elf.h     |  1 +
 tools/objtool/include/objtool/objtool.h |  2 +-
 4 files changed, 48 insertions(+), 8 deletions(-)

diff --git a/tools/objtool/check.c b/tools/objtool/check.c
index c6c99e3fb76f..1eb6489ae459 100644
--- a/tools/objtool/check.c
+++ b/tools/objtool/check.c
@@ -3,6 +3,7 @@
  * Copyright (C) 2015-2017 Josh Poimboeuf <jpoim...@redhat.com>
  */
 
+#define _GNU_SOURCE /* memmem() */
 #include <string.h>
 #include <stdlib.h>
 #include <inttypes.h>
@@ -610,6 +611,20 @@ static int init_pv_ops(struct objtool_file *file)
        return 0;
 }
 
+static bool is_livepatch_module(struct objtool_file *file)
+{
+       struct section *sec;
+
+       if (!opts.module)
+               return false;
+
+       sec = find_section_by_name(file->elf, ".modinfo");
+       if (!sec)
+               return false;
+
+       return memmem(sec->data->d_buf, sec_size(sec), "\0livepatch=Y", 12);
+}
+
 static int create_static_call_sections(struct objtool_file *file)
 {
        struct static_call_site *site;
@@ -621,7 +636,14 @@ static int create_static_call_sections(struct objtool_file 
*file)
 
        sec = find_section_by_name(file->elf, ".static_call_sites");
        if (sec) {
-               WARN("file already has .static_call_sites section, skipping");
+               /*
+                * Livepatch modules may have already extracted the static call
+                * site entries to take advantage of vmlinux static call
+                * privileges.
+                */
+               if (!file->klp)
+                       WARN("file already has .static_call_sites section, 
skipping");
+
                return 0;
        }
 
@@ -672,7 +694,7 @@ static int create_static_call_sections(struct objtool_file 
*file)
 
                key_sym = find_symbol_by_name(file->elf, tmp);
                if (!key_sym) {
-                       if (!opts.module) {
+                       if (!opts.module || file->klp) {
                                ERROR("static_call: can't find static_call_key 
symbol: %s", tmp);
                                return -1;
                        }
@@ -891,7 +913,13 @@ static int create_mcount_loc_sections(struct objtool_file 
*file)
 
        sec = find_section_by_name(file->elf, "__mcount_loc");
        if (sec) {
-               WARN("file already has __mcount_loc section, skipping");
+               /*
+                * Livepatch modules have already extracted their __mcount_loc
+                * entries to cover the !CONFIG_FTRACE_MCOUNT_USE_OBJTOOL case.
+                */
+               if (!file->klp)
+                       WARN("file already has __mcount_loc section, skipping");
+
                return 0;
        }
 
@@ -2575,6 +2603,8 @@ static bool validate_branch_enabled(void)
 
 static int decode_sections(struct objtool_file *file)
 {
+       file->klp = is_livepatch_module(file);
+
        mark_rodata(file);
 
        if (init_pv_ops(file))
@@ -4235,8 +4265,13 @@ static int add_prefix_symbol(struct objtool_file *file, 
struct symbol *func)
                if (offset < opts.prefix)
                        continue;
 
+               /*
+                * Ignore attempts to make duplicate symbols in livepatch
+                * modules.  They've already extracted the prefix symbols
+                * except for the newly compiled init.c.
+                */
                sym_pfx = elf_create_prefix_symbol(file->elf, func, 
opts.prefix);
-               if (!sym_pfx) {
+               if (!sym_pfx && !file->klp) {
                        WARN("duplicate prefix symbol for %s\n", func->name);
                        return -1;
                }
@@ -4571,6 +4606,7 @@ static int validate_ibt(struct objtool_file *file)
                    !strncmp(sec->name, ".debug", 6)                    ||
                    !strcmp(sec->name, ".altinstructions")              ||
                    !strcmp(sec->name, ".ibt_endbr_seal")               ||
+                   !strcmp(sec->name, ".kcfi_traps")                   ||
                    !strcmp(sec->name, ".orc_unwind_ip")                ||
                    !strcmp(sec->name, ".retpoline_sites")              ||
                    !strcmp(sec->name, ".smp_locks")                    ||
@@ -4580,12 +4616,12 @@ static int validate_ibt(struct objtool_file *file)
                    !strcmp(sec->name, "__bug_table")                   ||
                    !strcmp(sec->name, "__ex_table")                    ||
                    !strcmp(sec->name, "__jump_table")                  ||
+                   !strcmp(sec->name, "__klp_funcs")                   ||
                    !strcmp(sec->name, "__mcount_loc")                  ||
-                   !strcmp(sec->name, ".kcfi_traps")                   ||
                    !strcmp(sec->name, ".llvm.call-graph-profile")      ||
                    !strcmp(sec->name, ".llvm_bb_addr_map")             ||
                    !strcmp(sec->name, "__tracepoints")                 ||
-                   strstr(sec->name, "__patchable_function_entries"))
+                   !strcmp(sec->name, "__patchable_function_entries"))
                        continue;
 
                for_each_reloc(sec->rsec, reloc)
diff --git a/tools/objtool/elf.c b/tools/objtool/elf.c
index 383183e9cf2d..f9eed5d50de5 100644
--- a/tools/objtool/elf.c
+++ b/tools/objtool/elf.c
@@ -496,7 +496,10 @@ static int elf_add_symbol(struct elf *elf, struct symbol 
*sym)
            (strstarts(sym->name, "__pfx_") || strstarts(sym->name, "__cfi_")))
                sym->prefix = 1;
 
-       if (is_func_sym(sym) && strstr(sym->name, ".cold"))
+       if (strstarts(sym->name, ".klp.sym"))
+               sym->klp = 1;
+
+       if (!sym->klp && is_func_sym(sym) && strstr(sym->name, ".cold"))
                sym->cold = 1;
        sym->pfunc = sym->cfunc = sym;
 
diff --git a/tools/objtool/include/objtool/elf.h 
b/tools/objtool/include/objtool/elf.h
index c2a744afd5d5..1212f81f40e0 100644
--- a/tools/objtool/include/objtool/elf.h
+++ b/tools/objtool/include/objtool/elf.h
@@ -84,6 +84,7 @@ struct symbol {
        u8 debug_checksum    : 1;
        u8 changed           : 1;
        u8 included          : 1;
+       u8 klp               : 1;
        struct list_head pv_target;
        struct reloc *relocs;
        struct section *group_sec;
diff --git a/tools/objtool/include/objtool/objtool.h 
b/tools/objtool/include/objtool/objtool.h
index 37e9fe4492d6..731965a742e9 100644
--- a/tools/objtool/include/objtool/objtool.h
+++ b/tools/objtool/include/objtool/objtool.h
@@ -30,7 +30,7 @@ struct objtool_file {
        struct list_head mcount_loc_list;
        struct list_head endbr_list;
        struct list_head call_list;
-       bool ignore_unreachables, hints, rodata;
+       bool ignore_unreachables, hints, rodata, klp;
 
        unsigned int nr_endbr;
        unsigned int nr_endbr_int;
-- 
2.49.0


Reply via email to