1
2
3
4
5
6
7
8
9#ifndef _LINUX_LIVEPATCH_H_
10#define _LINUX_LIVEPATCH_H_
11
12#include <linux/module.h>
13#include <linux/ftrace.h>
14#include <linux/completion.h>
15#include <linux/list.h>
16
17#if IS_ENABLED(CONFIG_LIVEPATCH)
18
19#include <asm/livepatch.h>
20
21
22#define KLP_UNDEFINED -1
23#define KLP_UNPATCHED 0
24#define KLP_PATCHED 1
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58struct klp_func {
59
60 const char *old_name;
61 void *new_func;
62
63
64
65
66
67
68
69 unsigned long old_sympos;
70
71
72 void *old_func;
73 struct kobject kobj;
74 struct list_head node;
75 struct list_head stack_node;
76 unsigned long old_size, new_size;
77 bool nop;
78 bool patched;
79 bool transition;
80};
81
82struct klp_object;
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98struct klp_callbacks {
99 int (*pre_patch)(struct klp_object *obj);
100 void (*post_patch)(struct klp_object *obj);
101 void (*pre_unpatch)(struct klp_object *obj);
102 void (*post_unpatch)(struct klp_object *obj);
103 bool post_unpatch_enabled;
104};
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120struct klp_object {
121
122 const char *name;
123 struct klp_func *funcs;
124 struct klp_callbacks callbacks;
125
126
127 struct kobject kobj;
128 struct list_head func_list;
129 struct list_head node;
130 struct module *mod;
131 bool dynamic;
132 bool patched;
133};
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149struct klp_patch {
150
151 struct module *mod;
152 struct klp_object *objs;
153 bool replace;
154
155
156 struct list_head list;
157 struct kobject kobj;
158 struct list_head obj_list;
159 bool enabled;
160 bool forced;
161 struct work_struct free_work;
162 struct completion finish;
163};
164
165#define klp_for_each_object_static(patch, obj) \
166 for (obj = patch->objs; obj->funcs || obj->name; obj++)
167
168#define klp_for_each_object_safe(patch, obj, tmp_obj) \
169 list_for_each_entry_safe(obj, tmp_obj, &patch->obj_list, node)
170
171#define klp_for_each_object(patch, obj) \
172 list_for_each_entry(obj, &patch->obj_list, node)
173
174#define klp_for_each_func_static(obj, func) \
175 for (func = obj->funcs; \
176 func->old_name || func->new_func || func->old_sympos; \
177 func++)
178
179#define klp_for_each_func_safe(obj, func, tmp_func) \
180 list_for_each_entry_safe(func, tmp_func, &obj->func_list, node)
181
182#define klp_for_each_func(obj, func) \
183 list_for_each_entry(func, &obj->func_list, node)
184
185int klp_enable_patch(struct klp_patch *);
186
187void arch_klp_init_object_loaded(struct klp_patch *patch,
188 struct klp_object *obj);
189
190
191int klp_module_coming(struct module *mod);
192void klp_module_going(struct module *mod);
193
194void klp_copy_process(struct task_struct *child);
195void klp_update_patch_state(struct task_struct *task);
196
197static inline bool klp_patch_pending(struct task_struct *task)
198{
199 return test_tsk_thread_flag(task, TIF_PATCH_PENDING);
200}
201
202static inline bool klp_have_reliable_stack(void)
203{
204 return IS_ENABLED(CONFIG_STACKTRACE) &&
205 IS_ENABLED(CONFIG_HAVE_RELIABLE_STACKTRACE);
206}
207
208typedef int (*klp_shadow_ctor_t)(void *obj,
209 void *shadow_data,
210 void *ctor_data);
211typedef void (*klp_shadow_dtor_t)(void *obj, void *shadow_data);
212
213void *klp_shadow_get(void *obj, unsigned long id);
214void *klp_shadow_alloc(void *obj, unsigned long id,
215 size_t size, gfp_t gfp_flags,
216 klp_shadow_ctor_t ctor, void *ctor_data);
217void *klp_shadow_get_or_alloc(void *obj, unsigned long id,
218 size_t size, gfp_t gfp_flags,
219 klp_shadow_ctor_t ctor, void *ctor_data);
220void klp_shadow_free(void *obj, unsigned long id, klp_shadow_dtor_t dtor);
221void klp_shadow_free_all(unsigned long id, klp_shadow_dtor_t dtor);
222
223#else
224
225static inline int klp_module_coming(struct module *mod) { return 0; }
226static inline void klp_module_going(struct module *mod) {}
227static inline bool klp_patch_pending(struct task_struct *task) { return false; }
228static inline void klp_update_patch_state(struct task_struct *task) {}
229static inline void klp_copy_process(struct task_struct *child) {}
230
231#endif
232
233#endif
234