The atomic replace feature uses dynamically allocated struct klp_func to handle functions that will not longer be patched. These structures are of the type KLP_FUNC_NOP. They cause the ftrace handler to jump to the original code. But the address of the original code is not known until the patched module is loaded. This patch allows the late initialization. Also it adds a sanity check into the ftrace handler. Alternative solution would be to do not set the address at all. The ftrace handler could just return to the original code when NOP struct klp_func is used. But this would require another changes. For example, in the stack checking. Note that NOP structures might be available even when the patch is being disabled. This would happen when the patch enable transition is reverted. Signed-off-by: Petr Mladek <pmladek@xxxxxxxx> --- kernel/livepatch/core.c | 16 ++++++++++++++-- kernel/livepatch/patch.c | 5 +++++ 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/kernel/livepatch/core.c b/kernel/livepatch/core.c index ad508a86b2f9..da1438d47d83 100644 --- a/kernel/livepatch/core.c +++ b/kernel/livepatch/core.c @@ -945,8 +945,12 @@ static void klp_free_object_loaded(struct klp_object *obj) obj->mod = NULL; - klp_for_each_func(obj, func) + klp_for_each_func(obj, func) { func->old_addr = 0; + + if (klp_is_func_type(func, KLP_FUNC_NOP)) + func->new_func = NULL; + } } /* @@ -984,7 +988,12 @@ static void klp_free_patch(struct klp_patch *patch) static int klp_init_func(struct klp_object *obj, struct klp_func *func) { - if (!func->old_name || !func->new_func) + if (!func->old_name) + return -EINVAL; + + /* NOPs do not know the address until the patched module is loaded. */ + if (!func->new_func && + (!klp_is_func_type(func, KLP_FUNC_NOP) || klp_is_object_loaded(obj))) return -EINVAL; INIT_LIST_HEAD(&func->stack_node); @@ -1039,6 +1048,9 @@ static int klp_init_object_loaded(struct klp_patch *patch, return -ENOENT; } + if (klp_is_func_type(func, KLP_FUNC_NOP)) + func->new_func = (void *)func->old_addr; + ret = kallsyms_lookup_size_offset((unsigned long)func->new_func, &func->new_size, NULL); if (!ret) { diff --git a/kernel/livepatch/patch.c b/kernel/livepatch/patch.c index 54b3c379bb0f..1f5c3eea9ee1 100644 --- a/kernel/livepatch/patch.c +++ b/kernel/livepatch/patch.c @@ -118,7 +118,12 @@ static void notrace klp_ftrace_handler(unsigned long ip, } } + /* Survive ugly mistakes, for example, when handling NOPs. */ + if (WARN_ON_ONCE(!func->new_func)) + goto unlock; + klp_arch_set_pc(regs, (unsigned long)func->new_func); + unlock: preempt_enable_notrace(); } -- 2.13.6 -- To unsubscribe from this list: send the line "unsubscribe live-patching" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html