core.h 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. #ifndef _LIVEPATCH_CORE_H
  3. #define _LIVEPATCH_CORE_H
  4. #include <linux/livepatch.h>
  5. extern struct mutex klp_mutex;
  6. extern struct list_head klp_patches;
  7. #define klp_for_each_patch_safe(patch, tmp_patch) \
  8. list_for_each_entry_safe(patch, tmp_patch, &klp_patches, list)
  9. #define klp_for_each_patch(patch) \
  10. list_for_each_entry(patch, &klp_patches, list)
  11. void klp_free_patch_async(struct klp_patch *patch);
  12. void klp_free_replaced_patches_async(struct klp_patch *new_patch);
  13. void klp_unpatch_replaced_patches(struct klp_patch *new_patch);
  14. void klp_discard_nops(struct klp_patch *new_patch);
  15. static inline bool klp_is_object_loaded(struct klp_object *obj)
  16. {
  17. return !obj->name || obj->mod;
  18. }
  19. static inline int klp_pre_patch_callback(struct klp_object *obj)
  20. {
  21. int ret = 0;
  22. if (obj->callbacks.pre_patch)
  23. ret = (*obj->callbacks.pre_patch)(obj);
  24. obj->callbacks.post_unpatch_enabled = !ret;
  25. return ret;
  26. }
  27. static inline void klp_post_patch_callback(struct klp_object *obj)
  28. {
  29. if (obj->callbacks.post_patch)
  30. (*obj->callbacks.post_patch)(obj);
  31. }
  32. static inline void klp_pre_unpatch_callback(struct klp_object *obj)
  33. {
  34. if (obj->callbacks.pre_unpatch)
  35. (*obj->callbacks.pre_unpatch)(obj);
  36. }
  37. static inline void klp_post_unpatch_callback(struct klp_object *obj)
  38. {
  39. if (obj->callbacks.post_unpatch_enabled &&
  40. obj->callbacks.post_unpatch)
  41. (*obj->callbacks.post_unpatch)(obj);
  42. obj->callbacks.post_unpatch_enabled = false;
  43. }
  44. #endif /* _LIVEPATCH_CORE_H */