| 1 |
|
| 2 | #ifndef _LINUX_LIVEPATCH_HELPERS_H
|
| 3 | #define _LINUX_LIVEPATCH_HELPERS_H
|
| 4 |
|
| 5 | |
| 6 | * Interfaces for use by livepatch patches
|
| 7 |
|
| 8 |
|
| 9 | #include <linux/syscalls.h>
|
| 10 | #include <linux/livepatch.h>
|
| 11 |
|
| 12 | #ifdef MODULE
|
| 13 | #define KLP_OBJNAME __KBUILD_MODNAME
|
| 14 | #else
|
| 15 | #define KLP_OBJNAME vmlinux
|
| 16 | #endif
|
| 17 |
|
| 18 |
|
| 19 |
|
| 20 | #define KLP_CALLBACK_PTRS ".discard.klp_callback_ptrs"
|
| 21 |
|
| 22 | #define KLP_PRE_PATCH_CALLBACK(func) \
|
| 23 | klp_pre_patch_t __used __section(KLP_CALLBACK_PTRS) \
|
| 24 | __PASTE(__KLP_PRE_PATCH_PREFIX, KLP_OBJNAME) = func
|
| 25 |
|
| 26 | #define KLP_POST_PATCH_CALLBACK(func) \
|
| 27 | klp_post_patch_t __used __section(KLP_CALLBACK_PTRS) \
|
| 28 | __PASTE(__KLP_POST_PATCH_PREFIX, KLP_OBJNAME) = func
|
| 29 |
|
| 30 | #define KLP_PRE_UNPATCH_CALLBACK(func) \
|
| 31 | klp_pre_unpatch_t __used __section(KLP_CALLBACK_PTRS) \
|
| 32 | __PASTE(__KLP_PRE_UNPATCH_PREFIX, KLP_OBJNAME) = func
|
| 33 |
|
| 34 | #define KLP_POST_UNPATCH_CALLBACK(func) \
|
| 35 | klp_post_unpatch_t __used __section(KLP_CALLBACK_PTRS) \
|
| 36 | __PASTE(__KLP_POST_UNPATCH_PREFIX, KLP_OBJNAME) = func
|
| 37 |
|
| 38 | |
| 39 | * Replace static_call() usage with this macro when create-diff-object
|
| 40 | * recommends it due to the original static call key living in a module.
|
| 41 | *
|
| 42 | * This converts the static call to a regular indirect call.
|
| 43 |
|
| 44 | #define KLP_STATIC_CALL(name) \
|
| 45 | ((typeof(STATIC_CALL_TRAMP(name))*)(STATIC_CALL_KEY(name).func))
|
| 46 |
|
| 47 |
|
| 48 |
|
| 49 | #define KLP_SYSCALL_DEFINE1(name, ...) KLP_SYSCALL_DEFINEx(1, _##name, __VA_ARGS__)
|
| 50 | #define KLP_SYSCALL_DEFINE2(name, ...) KLP_SYSCALL_DEFINEx(2, _##name, __VA_ARGS__)
|
| 51 | #define KLP_SYSCALL_DEFINE3(name, ...) KLP_SYSCALL_DEFINEx(3, _##name, __VA_ARGS__)
|
| 52 | #define KLP_SYSCALL_DEFINE4(name, ...) KLP_SYSCALL_DEFINEx(4, _##name, __VA_ARGS__)
|
| 53 | #define KLP_SYSCALL_DEFINE5(name, ...) KLP_SYSCALL_DEFINEx(5, _##name, __VA_ARGS__)
|
| 54 | #define KLP_SYSCALL_DEFINE6(name, ...) KLP_SYSCALL_DEFINEx(6, _##name, __VA_ARGS__)
|
| 55 |
|
| 56 | #define KLP_SYSCALL_DEFINEx(x, sname, ...) \
|
| 57 | __KLP_SYSCALL_DEFINEx(x, sname, __VA_ARGS__)
|
| 58 |
|
| 59 | #ifdef CONFIG_X86_64
|
| 60 | |
| 61 | #define __KLP_SYSCALL_DEFINEx(x, name, ...) \
|
| 62 | static long __se_sys##name(__MAP(x,__SC_LONG,__VA_ARGS__)); \
|
| 63 | static inline long __klp_do_sys##name(__MAP(x,__SC_DECL,__VA_ARGS__));\
|
| 64 | __X64_SYS_STUBx(x, name, __VA_ARGS__) \
|
| 65 | __IA32_SYS_STUBx(x, name, __VA_ARGS__) \
|
| 66 | static long __se_sys##name(__MAP(x,__SC_LONG,__VA_ARGS__)) \
|
| 67 | { \
|
| 68 | long ret = __klp_do_sys##name(__MAP(x,__SC_CAST,__VA_ARGS__));\
|
| 69 | __MAP(x,__SC_TEST,__VA_ARGS__); \
|
| 70 | __PROTECT(x, ret,__MAP(x,__SC_ARGS,__VA_ARGS__)); \
|
| 71 | return ret; \
|
| 72 | } \
|
| 73 | static inline long __klp_do_sys##name(__MAP(x,__SC_DECL,__VA_ARGS__))
|
| 74 |
|
| 75 | #endif
|
| 76 |
|
| 77 | #endif
|