include/linux/livepatch_sched.h

   1 /* SPDX-License-Identifier: GPL-2.0-or-later */
   2 #ifndef _LINUX_LIVEPATCH_SCHED_H_
   3 #define _LINUX_LIVEPATCH_SCHED_H_
   4
   5 #include <linux/jump_label.h>
   6 #include <linux/static_call_types.h>
   7
   8 #ifdef CONFIG_LIVEPATCH
   9
  10 void __klp_sched_try_switch(void);
  11
  12 #if !defined(CONFIG_PREEMPT_DYNAMIC) || !defined(CONFIG_HAVE_PREEMPT_DYNAMIC_CALL)
  13
  14 DECLARE_STATIC_KEY_FALSE(klp_sched_try_switch_key);
  15
  16 static __always_inline void klp_sched_try_switch(void)
  17 {
  18         if (static_branch_unlikely(&klp_sched_try_switch_key))
  19                 __klp_sched_try_switch();
  20 }
  21
  22 #endif /* !CONFIG_PREEMPT_DYNAMIC || !CONFIG_HAVE_PREEMPT_DYNAMIC_CALL */
  23
  24 #else /* !CONFIG_LIVEPATCH */
  25 static inline void klp_sched_try_switch(void) {}
  26 static inline void __klp_sched_try_switch(void) {}
  27 #endif /* CONFIG_LIVEPATCH */
  28
  29 #endif /* _LINUX_LIVEPATCH_SCHED_H_ */