76 lines
2.1 KiB
C
76 lines
2.1 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
#ifndef _LIVEPATCH_CORE_H
|
|
#define _LIVEPATCH_CORE_H
|
|
|
|
#include <linux/livepatch.h>
|
|
|
|
extern struct mutex klp_mutex;
|
|
extern struct list_head klp_patches;
|
|
|
|
#define klp_for_each_patch_safe(patch, tmp_patch) \
|
|
list_for_each_entry_safe(patch, tmp_patch, &klp_patches, list)
|
|
|
|
#define klp_for_each_patch(patch) \
|
|
list_for_each_entry(patch, &klp_patches, list)
|
|
|
|
#ifdef CONFIG_LIVEPATCH_FTRACE
|
|
void klp_free_patch_async(struct klp_patch *patch);
|
|
void klp_free_replaced_patches_async(struct klp_patch *new_patch);
|
|
void klp_unpatch_replaced_patches(struct klp_patch *new_patch);
|
|
void klp_discard_nops(struct klp_patch *new_patch);
|
|
#else
|
|
/*
|
|
* In the enable_patch() process, we do not need to roll back the patch
|
|
* immediately if the patch fails to enabled. In this way, the function that has
|
|
* been successfully patched does not need to be enabled repeatedly during
|
|
* retry. However, if it is the last retry (rollback == true) or not because of
|
|
* stack check failure (patch_err != -EAGAIN), rollback is required immediately.
|
|
*/
|
|
static inline bool klp_need_rollback(int patch_err, bool rollback)
|
|
{
|
|
return patch_err != -EAGAIN || rollback;
|
|
}
|
|
#endif /* CONFIG_LIVEPATCH_FTRACE */
|
|
|
|
static inline bool klp_is_object_loaded(struct klp_object *obj)
|
|
{
|
|
return !obj->name || obj->mod;
|
|
}
|
|
|
|
#ifdef CONFIG_LIVEPATCH_FTRACE
|
|
static inline int klp_pre_patch_callback(struct klp_object *obj)
|
|
{
|
|
int ret = 0;
|
|
|
|
if (obj->callbacks.pre_patch)
|
|
ret = (*obj->callbacks.pre_patch)(obj);
|
|
|
|
obj->callbacks.post_unpatch_enabled = !ret;
|
|
|
|
return ret;
|
|
}
|
|
|
|
static inline void klp_post_patch_callback(struct klp_object *obj)
|
|
{
|
|
if (obj->callbacks.post_patch)
|
|
(*obj->callbacks.post_patch)(obj);
|
|
}
|
|
|
|
static inline void klp_pre_unpatch_callback(struct klp_object *obj)
|
|
{
|
|
if (obj->callbacks.pre_unpatch)
|
|
(*obj->callbacks.pre_unpatch)(obj);
|
|
}
|
|
|
|
static inline void klp_post_unpatch_callback(struct klp_object *obj)
|
|
{
|
|
if (obj->callbacks.post_unpatch_enabled &&
|
|
obj->callbacks.post_unpatch)
|
|
(*obj->callbacks.post_unpatch)(obj);
|
|
|
|
obj->callbacks.post_unpatch_enabled = false;
|
|
}
|
|
#endif /* CONFIG_LIVEPATCH_FTRACE */
|
|
|
|
#endif /* _LIVEPATCH_CORE_H */
|