livepatch: introduce patch/func-walking helpers

klp_for_each_object and klp_for_each_func are now used all over the
code. One need not think what is the proper condition to check in the
for loop now.

Signed-off-by: Jiri Slaby <jslaby@suse.cz>
Acked-by: Josh Poimboeuf <jpoimboe@redhat.com>
Signed-off-by: Jiri Kosina <jkosina@suse.cz>
This commit is contained in:
Jiri Slaby 2015-05-19 12:01:19 +02:00 committed by Jiri Kosina
parent cad706df7e
commit 8cdd043ab3
2 changed files with 15 additions and 9 deletions

View file

@ -123,6 +123,12 @@ struct klp_patch {
enum klp_state state;
};
#define klp_for_each_object(patch, obj) \
for (obj = patch->objs; obj->funcs; obj++)
#define klp_for_each_func(obj, func) \
for (func = obj->funcs; func->old_name; func++)
int klp_register_patch(struct klp_patch *);
int klp_unregister_patch(struct klp_patch *);
int klp_enable_patch(struct klp_patch *);

View file

@ -422,7 +422,7 @@ static void klp_disable_object(struct klp_object *obj)
{
struct klp_func *func;
for (func = obj->funcs; func->old_name; func++)
klp_for_each_func(obj, func)
if (func->state == KLP_ENABLED)
klp_disable_func(func);
@ -440,7 +440,7 @@ static int klp_enable_object(struct klp_object *obj)
if (WARN_ON(!klp_is_object_loaded(obj)))
return -EINVAL;
for (func = obj->funcs; func->old_name; func++) {
klp_for_each_func(obj, func) {
ret = klp_enable_func(func);
if (ret) {
klp_disable_object(obj);
@ -463,7 +463,7 @@ static int __klp_disable_patch(struct klp_patch *patch)
pr_notice("disabling patch '%s'\n", patch->mod->name);
for (obj = patch->objs; obj->funcs; obj++) {
klp_for_each_object(patch, obj) {
if (obj->state == KLP_ENABLED)
klp_disable_object(obj);
}
@ -523,7 +523,7 @@ static int __klp_enable_patch(struct klp_patch *patch)
pr_notice("enabling patch '%s'\n", patch->mod->name);
for (obj = patch->objs; obj->funcs; obj++) {
klp_for_each_object(patch, obj) {
if (!klp_is_object_loaded(obj))
continue;
@ -689,7 +689,7 @@ static void klp_free_object_loaded(struct klp_object *obj)
obj->mod = NULL;
for (func = obj->funcs; func->old_name; func++)
klp_for_each_func(obj, func)
func->old_addr = 0;
}
@ -738,7 +738,7 @@ static int klp_init_object_loaded(struct klp_patch *patch,
return ret;
}
for (func = obj->funcs; func->old_name; func++) {
klp_for_each_func(obj, func) {
ret = klp_find_verify_func_addr(obj, func);
if (ret)
return ret;
@ -767,7 +767,7 @@ static int klp_init_object(struct klp_patch *patch, struct klp_object *obj)
if (ret)
return ret;
for (func = obj->funcs; func->old_name; func++) {
klp_for_each_func(obj, func) {
ret = klp_init_func(obj, func);
if (ret)
goto free;
@ -804,7 +804,7 @@ static int klp_init_patch(struct klp_patch *patch)
if (ret)
goto unlock;
for (obj = patch->objs; obj->funcs; obj++) {
klp_for_each_object(patch, obj) {
ret = klp_init_object(patch, obj);
if (ret)
goto free;
@ -961,7 +961,7 @@ static int klp_module_notify(struct notifier_block *nb, unsigned long action,
mod->klp_alive = false;
list_for_each_entry(patch, &klp_patches, list) {
for (obj = patch->objs; obj->funcs; obj++) {
klp_for_each_object(patch, obj) {
if (!klp_is_module(obj) || strcmp(obj->name, mod->name))
continue;