Return-Path: Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S1751681AbaLISGM (ORCPT ); Tue, 9 Dec 2014 13:06:12 -0500 Received: from cantor2.suse.de ([195.135.220.15]:53883 "EHLO mx2.suse.de" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1751608AbaLISGJ (ORCPT ); Tue, 9 Dec 2014 13:06:09 -0500 From: Petr Mladek To: Seth Jennings , Josh Poimboeuf , Jiri Kosina , Vojtech Pavlik , Steven Rostedt , Miroslav Benes , Masami Hiramatsu Cc: Christoph Hellwig , Greg KH , Andy Lutomirski , live-patching@vger.kernel.org, x86@kernel.org, kpatch@redhat.com, linux-kernel@vger.kernel.org, Petr Mladek Subject: [PATCH 5/6] livepatch v5: split init and free code that is done only for loaded modules Date: Tue, 9 Dec 2014 19:05:06 +0100 Message-Id: <1418148307-21434-6-git-send-email-pmladek@suse.cz> X-Mailer: git-send-email 1.8.5.2 In-Reply-To: <1418148307-21434-1-git-send-email-pmladek@suse.cz> References: <1418148307-21434-1-git-send-email-pmladek@suse.cz> Sender: linux-kernel-owner@vger.kernel.org List-ID: X-Mailing-List: linux-kernel@vger.kernel.org This patch makes it clear what initialization and freeing steps need to be done when an object (module) is being loaded or removed. It will help to maintain the module coming and going handlers. Also it will remove duplicated code from these handlers. Signed-off-by: Petr Mladek --- kernel/livepatch/core.c | 92 ++++++++++++++++++++++++++++++++----------------- 1 file changed, 61 insertions(+), 31 deletions(-) diff --git a/kernel/livepatch/core.c b/kernel/livepatch/core.c index 97a8d4a3d6d8..fe312b9ada78 100644 --- a/kernel/livepatch/core.c +++ b/kernel/livepatch/core.c @@ -590,6 +590,12 @@ static struct kobj_type klp_ktype_func = { .sysfs_ops = &kobj_sysfs_ops, }; +/* Clean up when a patched object is unloaded */ +static void klp_free_func_loaded(struct klp_func *func) +{ + func->old_addr = 0; +} + /* * Free all functions' kobjects in the array up to some limit. When limit is * NULL, all kobjects are freed. @@ -603,6 +609,17 @@ static void klp_free_funcs_limited(struct klp_object *obj, kobject_put(&func->kobj); } +/* Clean up when a patched object is unloaded */ +static void klp_free_object_loaded(struct klp_object *obj) +{ + struct klp_func *func; + + obj->mod = NULL; + + for (func = obj->funcs; func->old_name; func++) + klp_free_func_loaded(func); +} + /* * Free all objects' kobjects in the array up to some limit. When limit is * NULL, all kobjects are freed. @@ -626,6 +643,12 @@ static void klp_free_patch(struct klp_patch *patch) kobject_put(&patch->kobj); } +/* parts of the initialization that is done only when the object is loaded */ +static int klp_init_func_loaded(struct klp_object *obj, struct klp_func *func) +{ + return klp_find_verify_func_addr(obj, func); +} + static int klp_init_func(struct klp_object *obj, struct klp_func *func) { struct ftrace_ops *ops; @@ -633,10 +656,6 @@ static int klp_init_func(struct klp_object *obj, struct klp_func *func) func->state = KLP_DISABLED; - ret = klp_find_verify_func_addr(obj, func); - if (ret) - return ret; - ops = kzalloc(sizeof(*ops), GFP_KERNEL); if (!ops) ret = -ENOMEM; @@ -656,6 +675,28 @@ static int klp_init_func(struct klp_object *obj, struct klp_func *func) return 0; } +/* parts of the initialization that is done only when the object is loaded */ +static int klp_init_object_loaded(struct klp_patch *patch, + struct klp_object *obj) +{ + struct klp_func *func; + int ret; + + if (obj->relocs) { + ret = klp_write_object_relocations(patch->mod, obj); + if (ret) + return ret; + } + + for (func = obj->funcs; func->old_name; func++) { + ret = klp_init_func_loaded(obj, func); + if (ret) + return ret; + } + + return 0; +} + static int klp_init_object(struct klp_patch *patch, struct klp_object *obj) { struct klp_func *func; @@ -669,12 +710,6 @@ static int klp_init_object(struct klp_patch *patch, struct klp_object *obj) klp_find_object_module(obj); - if (obj->relocs && klp_is_object_loaded(obj)) { - ret = klp_write_object_relocations(patch->mod, obj); - if (ret) - return ret; - } - name = klp_is_module(obj) ? obj->name : "vmlinux"; obj->kobj = kobject_create_and_add(name, &patch->kobj); if (!obj->kobj) @@ -686,6 +721,12 @@ static int klp_init_object(struct klp_patch *patch, struct klp_object *obj) goto free; } + if (klp_is_object_loaded(obj)) { + ret = klp_init_object_loaded(patch, obj); + if (ret) + goto free; + } + return 0; free: @@ -802,27 +843,19 @@ int klp_register_patch(struct klp_patch *patch) } EXPORT_SYMBOL_GPL(klp_register_patch); -static void klp_module_notify_coming(struct module *pmod, +static void klp_module_notify_coming(struct klp_patch *patch, struct klp_object *obj) { - struct klp_func *func; + struct module *pmod = patch->mod; struct module *mod = obj->mod; int ret; pr_notice("applying patch '%s' to loading module '%s'\n", pmod->name, mod->name); - if (obj->relocs) { - ret = klp_write_object_relocations(pmod, obj); - if (ret) - goto err; - } - - for (func = obj->funcs; func->old_name; func++) { - ret = klp_find_verify_func_addr(obj, func); - if (ret) - goto err; - } + ret = klp_init_object_loaded(patch, obj); + if (ret) + goto err; ret = klp_enable_object(obj); if (!ret) @@ -833,10 +866,10 @@ err: pmod->name, mod->name, ret); } -static void klp_module_notify_going(struct module *pmod, +static void klp_module_notify_going(struct klp_patch *patch, struct klp_object *obj) { - struct klp_func *func; + struct module *pmod = patch->mod; struct module *mod = obj->mod; int ret; @@ -848,10 +881,7 @@ static void klp_module_notify_going(struct module *pmod, pr_warn("failed to revert patch '%s' on module '%s' (%d)\n", pmod->name, mod->name, ret); - for (func = obj->funcs; func->old_name; func++) - func->old_addr = 0; - - obj->mod = NULL; + klp_free_object_loaded(obj); } static int klp_module_notify(struct notifier_block *nb, unsigned long action, @@ -876,9 +906,9 @@ static int klp_module_notify(struct notifier_block *nb, unsigned long action, if (action == MODULE_STATE_COMING) { obj->mod = mod; - klp_module_notify_coming(patch->mod, obj); + klp_module_notify_coming(patch, obj); } else /* MODULE_STATE_GOING */ - klp_module_notify_going(patch->mod, obj); + klp_module_notify_going(patch, obj); break; } -- 1.8.5.2 -- To unsubscribe from this list: send the line "unsubscribe linux-kernel" in the body of a message to majordomo@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html Please read the FAQ at http://www.tux.org/lkml/