Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Init code for a livepatch kernel module
4 */
5
6#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
7
8#include <linux/kernel.h>
9#include <linux/slab.h>
10#include <linux/livepatch.h>
11
12extern struct klp_object_ext __start_klp_objects[];
13extern struct klp_object_ext __stop_klp_objects[];
14
15static struct klp_patch *patch;
16
17static int __init livepatch_mod_init(void)
18{
19 struct klp_object *objs;
20 unsigned int nr_objs;
21 int ret;
22
23 nr_objs = __stop_klp_objects - __start_klp_objects;
24
25 if (!nr_objs) {
26 pr_err("nothing to patch!\n");
27 ret = -EINVAL;
28 goto err;
29 }
30
31 patch = kzalloc(sizeof(*patch), GFP_KERNEL);
32 if (!patch) {
33 ret = -ENOMEM;
34 goto err;
35 }
36
37 objs = kzalloc(sizeof(struct klp_object) * (nr_objs + 1), GFP_KERNEL);
38 if (!objs) {
39 ret = -ENOMEM;
40 goto err_free_patch;
41 }
42
43 for (int i = 0; i < nr_objs; i++) {
44 struct klp_object_ext *obj_ext = __start_klp_objects + i;
45 struct klp_func_ext *funcs_ext = obj_ext->funcs;
46 unsigned int nr_funcs = obj_ext->nr_funcs;
47 struct klp_func *funcs = objs[i].funcs;
48 struct klp_object *obj = objs + i;
49
50 funcs = kzalloc(sizeof(struct klp_func) * (nr_funcs + 1), GFP_KERNEL);
51 if (!funcs) {
52 ret = -ENOMEM;
53 for (int j = 0; j < i; j++)
54 kfree(objs[i].funcs);
55 goto err_free_objs;
56 }
57
58 for (int j = 0; j < nr_funcs; j++) {
59 funcs[j].old_name = funcs_ext[j].old_name;
60 funcs[j].new_func = funcs_ext[j].new_func;
61 funcs[j].old_sympos = funcs_ext[j].sympos;
62 }
63
64 obj->name = obj_ext->name;
65 obj->funcs = funcs;
66
67 memcpy(&obj->callbacks, &obj_ext->callbacks, sizeof(struct klp_callbacks));
68 }
69
70 patch->mod = THIS_MODULE;
71 patch->objs = objs;
72
73 /* TODO patch->states */
74
75#ifdef KLP_NO_REPLACE
76 patch->replace = false;
77#else
78 patch->replace = true;
79#endif
80
81 return klp_enable_patch(patch);
82
83err_free_objs:
84 kfree(objs);
85err_free_patch:
86 kfree(patch);
87err:
88 return ret;
89}
90
91static void __exit livepatch_mod_exit(void)
92{
93 unsigned int nr_objs;
94
95 nr_objs = __stop_klp_objects - __start_klp_objects;
96
97 for (int i = 0; i < nr_objs; i++)
98 kfree(patch->objs[i].funcs);
99
100 kfree(patch->objs);
101 kfree(patch);
102}
103
104module_init(livepatch_mod_init);
105module_exit(livepatch_mod_exit);
106MODULE_LICENSE("GPL");
107MODULE_INFO(livepatch, "Y");
108MODULE_DESCRIPTION("Livepatch module");