123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196 |
- #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
- #include <linux/module.h>
- #include <linux/kernel.h>
- #include <linux/livepatch.h>
- static int pre_patch_ret;
- module_param(pre_patch_ret, int, 0644);
- MODULE_PARM_DESC(pre_patch_ret, "pre_patch_ret (default=0)");
- static const char *const module_state[] = {
- [MODULE_STATE_LIVE] = "[MODULE_STATE_LIVE] Normal state",
- [MODULE_STATE_COMING] = "[MODULE_STATE_COMING] Full formed, running module_init",
- [MODULE_STATE_GOING] = "[MODULE_STATE_GOING] Going away",
- [MODULE_STATE_UNFORMED] = "[MODULE_STATE_UNFORMED] Still setting it up",
- };
- static void callback_info(const char *callback, struct klp_object *obj)
- {
- if (obj->mod)
- pr_info("%s: %s -> %s\n", callback, obj->mod->name,
- module_state[obj->mod->state]);
- else
- pr_info("%s: vmlinux\n", callback);
- }
- static int pre_patch_callback(struct klp_object *obj)
- {
- callback_info(__func__, obj);
- return pre_patch_ret;
- }
- static void post_patch_callback(struct klp_object *obj)
- {
- callback_info(__func__, obj);
- }
- static void pre_unpatch_callback(struct klp_object *obj)
- {
- callback_info(__func__, obj);
- }
- static void post_unpatch_callback(struct klp_object *obj)
- {
- callback_info(__func__, obj);
- }
- static void patched_work_func(struct work_struct *work)
- {
- pr_info("%s\n", __func__);
- }
- static struct klp_func no_funcs[] = {
- { }
- };
- static struct klp_func busymod_funcs[] = {
- {
- .old_name = "busymod_work_func",
- .new_func = patched_work_func,
- }, { }
- };
- static struct klp_object objs[] = {
- {
- .name = NULL,
- .funcs = no_funcs,
- .callbacks = {
- .pre_patch = pre_patch_callback,
- .post_patch = post_patch_callback,
- .pre_unpatch = pre_unpatch_callback,
- .post_unpatch = post_unpatch_callback,
- },
- }, {
- .name = "livepatch_callbacks_mod",
- .funcs = no_funcs,
- .callbacks = {
- .pre_patch = pre_patch_callback,
- .post_patch = post_patch_callback,
- .pre_unpatch = pre_unpatch_callback,
- .post_unpatch = post_unpatch_callback,
- },
- }, {
- .name = "livepatch_callbacks_busymod",
- .funcs = busymod_funcs,
- .callbacks = {
- .pre_patch = pre_patch_callback,
- .post_patch = post_patch_callback,
- .pre_unpatch = pre_unpatch_callback,
- .post_unpatch = post_unpatch_callback,
- },
- }, { }
- };
- static struct klp_patch patch = {
- .mod = THIS_MODULE,
- .objs = objs,
- };
- static int livepatch_callbacks_demo_init(void)
- {
- return klp_enable_patch(&patch);
- }
- static void livepatch_callbacks_demo_exit(void)
- {
- }
- module_init(livepatch_callbacks_demo_init);
- module_exit(livepatch_callbacks_demo_exit);
- MODULE_LICENSE("GPL");
- MODULE_INFO(livepatch, "Y");
|