1a2818ee4SJoe Lawrence // SPDX-License-Identifier: GPL-2.0
2a2818ee4SJoe Lawrence // Copyright (C) 2018 Joe Lawrence <joe.lawrence@redhat.com>
3a2818ee4SJoe Lawrence
4a2818ee4SJoe Lawrence #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
5a2818ee4SJoe Lawrence
6a2818ee4SJoe Lawrence #include <linux/module.h>
7a2818ee4SJoe Lawrence #include <linux/kernel.h>
8a2818ee4SJoe Lawrence #include <linux/livepatch.h>
9a2818ee4SJoe Lawrence
10a2818ee4SJoe Lawrence static int pre_patch_ret;
11a2818ee4SJoe Lawrence module_param(pre_patch_ret, int, 0644);
12a2818ee4SJoe Lawrence MODULE_PARM_DESC(pre_patch_ret, "pre_patch_ret (default=0)");
13a2818ee4SJoe Lawrence
14a2818ee4SJoe Lawrence static const char *const module_state[] = {
15a2818ee4SJoe Lawrence [MODULE_STATE_LIVE] = "[MODULE_STATE_LIVE] Normal state",
16a2818ee4SJoe Lawrence [MODULE_STATE_COMING] = "[MODULE_STATE_COMING] Full formed, running module_init",
17a2818ee4SJoe Lawrence [MODULE_STATE_GOING] = "[MODULE_STATE_GOING] Going away",
18a2818ee4SJoe Lawrence [MODULE_STATE_UNFORMED] = "[MODULE_STATE_UNFORMED] Still setting it up",
19a2818ee4SJoe Lawrence };
20a2818ee4SJoe Lawrence
callback_info(const char * callback,struct klp_object * obj)21a2818ee4SJoe Lawrence static void callback_info(const char *callback, struct klp_object *obj)
22a2818ee4SJoe Lawrence {
23a2818ee4SJoe Lawrence if (obj->mod)
24a2818ee4SJoe Lawrence pr_info("%s: %s -> %s\n", callback, obj->mod->name,
25a2818ee4SJoe Lawrence module_state[obj->mod->state]);
26a2818ee4SJoe Lawrence else
27a2818ee4SJoe Lawrence pr_info("%s: vmlinux\n", callback);
28a2818ee4SJoe Lawrence }
29a2818ee4SJoe Lawrence
30a2818ee4SJoe Lawrence /* Executed on object patching (ie, patch enablement) */
pre_patch_callback(struct klp_object * obj)31a2818ee4SJoe Lawrence static int pre_patch_callback(struct klp_object *obj)
32a2818ee4SJoe Lawrence {
33a2818ee4SJoe Lawrence callback_info(__func__, obj);
34a2818ee4SJoe Lawrence return pre_patch_ret;
35a2818ee4SJoe Lawrence }
36a2818ee4SJoe Lawrence
37a2818ee4SJoe Lawrence /* Executed on object unpatching (ie, patch disablement) */
post_patch_callback(struct klp_object * obj)38a2818ee4SJoe Lawrence static void post_patch_callback(struct klp_object *obj)
39a2818ee4SJoe Lawrence {
40a2818ee4SJoe Lawrence callback_info(__func__, obj);
41a2818ee4SJoe Lawrence }
42a2818ee4SJoe Lawrence
43a2818ee4SJoe Lawrence /* Executed on object unpatching (ie, patch disablement) */
pre_unpatch_callback(struct klp_object * obj)44a2818ee4SJoe Lawrence static void pre_unpatch_callback(struct klp_object *obj)
45a2818ee4SJoe Lawrence {
46a2818ee4SJoe Lawrence callback_info(__func__, obj);
47a2818ee4SJoe Lawrence }
48a2818ee4SJoe Lawrence
49a2818ee4SJoe Lawrence /* Executed on object unpatching (ie, patch disablement) */
post_unpatch_callback(struct klp_object * obj)50a2818ee4SJoe Lawrence static void post_unpatch_callback(struct klp_object *obj)
51a2818ee4SJoe Lawrence {
52a2818ee4SJoe Lawrence callback_info(__func__, obj);
53a2818ee4SJoe Lawrence }
54a2818ee4SJoe Lawrence
patched_work_func(struct work_struct * work)55a2818ee4SJoe Lawrence static void patched_work_func(struct work_struct *work)
56a2818ee4SJoe Lawrence {
57a2818ee4SJoe Lawrence pr_info("%s\n", __func__);
58a2818ee4SJoe Lawrence }
59a2818ee4SJoe Lawrence
60a2818ee4SJoe Lawrence static struct klp_func no_funcs[] = {
61a2818ee4SJoe Lawrence {}
62a2818ee4SJoe Lawrence };
63a2818ee4SJoe Lawrence
64a2818ee4SJoe Lawrence static struct klp_func busymod_funcs[] = {
65a2818ee4SJoe Lawrence {
66a2818ee4SJoe Lawrence .old_name = "busymod_work_func",
67a2818ee4SJoe Lawrence .new_func = patched_work_func,
68a2818ee4SJoe Lawrence }, {}
69a2818ee4SJoe Lawrence };
70a2818ee4SJoe Lawrence
71a2818ee4SJoe Lawrence static struct klp_object objs[] = {
72a2818ee4SJoe Lawrence {
73a2818ee4SJoe Lawrence .name = NULL, /* vmlinux */
74a2818ee4SJoe Lawrence .funcs = no_funcs,
75a2818ee4SJoe Lawrence .callbacks = {
76a2818ee4SJoe Lawrence .pre_patch = pre_patch_callback,
77a2818ee4SJoe Lawrence .post_patch = post_patch_callback,
78a2818ee4SJoe Lawrence .pre_unpatch = pre_unpatch_callback,
79a2818ee4SJoe Lawrence .post_unpatch = post_unpatch_callback,
80a2818ee4SJoe Lawrence },
81a2818ee4SJoe Lawrence }, {
82a2818ee4SJoe Lawrence .name = "test_klp_callbacks_mod",
83a2818ee4SJoe Lawrence .funcs = no_funcs,
84a2818ee4SJoe Lawrence .callbacks = {
85a2818ee4SJoe Lawrence .pre_patch = pre_patch_callback,
86a2818ee4SJoe Lawrence .post_patch = post_patch_callback,
87a2818ee4SJoe Lawrence .pre_unpatch = pre_unpatch_callback,
88a2818ee4SJoe Lawrence .post_unpatch = post_unpatch_callback,
89a2818ee4SJoe Lawrence },
90a2818ee4SJoe Lawrence }, {
91a2818ee4SJoe Lawrence .name = "test_klp_callbacks_busy",
92a2818ee4SJoe Lawrence .funcs = busymod_funcs,
93a2818ee4SJoe Lawrence .callbacks = {
94a2818ee4SJoe Lawrence .pre_patch = pre_patch_callback,
95a2818ee4SJoe Lawrence .post_patch = post_patch_callback,
96a2818ee4SJoe Lawrence .pre_unpatch = pre_unpatch_callback,
97a2818ee4SJoe Lawrence .post_unpatch = post_unpatch_callback,
98a2818ee4SJoe Lawrence },
99a2818ee4SJoe Lawrence }, { }
100a2818ee4SJoe Lawrence };
101a2818ee4SJoe Lawrence
102a2818ee4SJoe Lawrence static struct klp_patch patch = {
103a2818ee4SJoe Lawrence .mod = THIS_MODULE,
104a2818ee4SJoe Lawrence .objs = objs,
105a2818ee4SJoe Lawrence };
106a2818ee4SJoe Lawrence
test_klp_callbacks_demo_init(void)107a2818ee4SJoe Lawrence static int test_klp_callbacks_demo_init(void)
108a2818ee4SJoe Lawrence {
109a2818ee4SJoe Lawrence return klp_enable_patch(&patch);
110a2818ee4SJoe Lawrence }
111a2818ee4SJoe Lawrence
test_klp_callbacks_demo_exit(void)112a2818ee4SJoe Lawrence static void test_klp_callbacks_demo_exit(void)
113a2818ee4SJoe Lawrence {
114a2818ee4SJoe Lawrence }
115a2818ee4SJoe Lawrence
116a2818ee4SJoe Lawrence module_init(test_klp_callbacks_demo_init);
117a2818ee4SJoe Lawrence module_exit(test_klp_callbacks_demo_exit);
118a2818ee4SJoe Lawrence MODULE_LICENSE("GPL");
119a2818ee4SJoe Lawrence MODULE_INFO(livepatch, "Y");
120a2818ee4SJoe Lawrence MODULE_AUTHOR("Joe Lawrence <joe.lawrence@redhat.com>");
121a2818ee4SJoe Lawrence MODULE_DESCRIPTION("Livepatch test: livepatch demo");
122