/* SPDX-License-Identifier: GPL-2.0 */

  #ifndef _LIVEPATCH_CORE_H
  #define _LIVEPATCH_CORE_H

  #include <linux/livepatch.h>

  extern struct mutex klp_mutex;

  extern struct list_head klp_patches;

  #define klp_for_each_patch_safe(patch, tmp_patch)		\
  	list_for_each_entry_safe(patch, tmp_patch, &klp_patches, list)
  
  #define klp_for_each_patch(patch)	\
  	list_for_each_entry(patch, &klp_patches, list)

  void klp_free_patch_async(struct klp_patch *patch);
  void klp_free_replaced_patches_async(struct klp_patch *new_patch);
  void klp_unpatch_replaced_patches(struct klp_patch *new_patch);

  void klp_discard_nops(struct klp_patch *new_patch);

  static inline bool klp_is_object_loaded(struct klp_object *obj)
  {
  	return !obj->name || obj->mod;
  }
  
  static inline int klp_pre_patch_callback(struct klp_object *obj)
  {

  	int ret = 0;

  	if (obj->callbacks.pre_patch)
  		ret = (*obj->callbacks.pre_patch)(obj);

  
  	obj->callbacks.post_unpatch_enabled = !ret;
  
  	return ret;
  }
  
  static inline void klp_post_patch_callback(struct klp_object *obj)
  {
  	if (obj->callbacks.post_patch)
  		(*obj->callbacks.post_patch)(obj);
  }
  
  static inline void klp_pre_unpatch_callback(struct klp_object *obj)
  {
  	if (obj->callbacks.pre_unpatch)
  		(*obj->callbacks.pre_unpatch)(obj);
  }
  
  static inline void klp_post_unpatch_callback(struct klp_object *obj)
  {
  	if (obj->callbacks.post_unpatch_enabled &&
  	    obj->callbacks.post_unpatch)
  		(*obj->callbacks.post_unpatch)(obj);

  
  	obj->callbacks.post_unpatch_enabled = false;

  }

  #endif /* _LIVEPATCH_CORE_H */