include/linux/jump_label_ratelimit.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _LINUX_JUMP_LABEL_RATELIMIT_H
   3 #define _LINUX_JUMP_LABEL_RATELIMIT_H
   4
   5 #include <linux/jump_label.h>
   6 #include <linux/workqueue.h>
   7
   8 #if defined(CONFIG_JUMP_LABEL)
   9 struct static_key_deferred {
  10         struct static_key key;
  11         unsigned long timeout;
  12         struct delayed_work work;
  13 };
  14
  15 struct static_key_true_deferred {
  16         struct static_key_true key;
  17         unsigned long timeout;
  18         struct delayed_work work;
  19 };
  20
  21 struct static_key_false_deferred {
  22         struct static_key_false key;
  23         unsigned long timeout;
  24         struct delayed_work work;
  25 };
  26
  27 #define static_key_slow_dec_deferred(x)                                 \
  28         __static_key_slow_dec_deferred(&(x)->key, &(x)->work, (x)->timeout)
  29 #define static_branch_slow_dec_deferred(x)                              \
  30         __static_key_slow_dec_deferred(&(x)->key.key, &(x)->work, (x)->timeout)
  31
  32 #define static_key_deferred_flush(x)                                    \
  33         __static_key_deferred_flush((x), &(x)->work)
  34
  35 extern void
  36 __static_key_slow_dec_deferred(struct static_key *key,
  37                                struct delayed_work *work,
  38                                unsigned long timeout);
  39 extern void __static_key_deferred_flush(void *key, struct delayed_work *work);
  40 extern void
  41 jump_label_rate_limit(struct static_key_deferred *key, unsigned long rl);
  42
  43 extern void jump_label_update_timeout(struct work_struct *work);
  44
  45 #define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)                       \
  46         struct static_key_true_deferred name = {                        \
  47                 .key =          { STATIC_KEY_INIT_TRUE },               \
  48                 .timeout =      (rl),                                   \
  49                 .work = __DELAYED_WORK_INITIALIZER((name).work,         \
  50                                                    jump_label_update_timeout, \
  51                                                    0),                  \
  52         }
  53
  54 #define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)                      \
  55         struct static_key_false_deferred name = {                       \
  56                 .key =          { STATIC_KEY_INIT_FALSE },              \
  57                 .timeout =      (rl),                                   \
  58                 .work = __DELAYED_WORK_INITIALIZER((name).work,         \
  59                                                    jump_label_update_timeout, \
  60                                                    0),                  \
  61         }
  62
  63 #define static_branch_deferred_inc(x)   static_branch_inc(&(x)->key)
  64
  65 #else   /* !CONFIG_JUMP_LABEL */
  66 struct static_key_deferred {
  67         struct static_key  key;
  68 };
  69 struct static_key_true_deferred {
  70         struct static_key_true key;
  71 };
  72 struct static_key_false_deferred {
  73         struct static_key_false key;
  74 };
  75 #define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)       \
  76         struct static_key_true_deferred name = { STATIC_KEY_TRUE_INIT }
  77 #define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)      \
  78         struct static_key_false_deferred name = { STATIC_KEY_FALSE_INIT }
  79
  80 #define static_branch_slow_dec_deferred(x)      static_branch_dec(&(x)->key)
  81
  82 static inline void static_key_slow_dec_deferred(struct static_key_deferred *key)
  83 {
  84         STATIC_KEY_CHECK_USE(key);
  85         static_key_slow_dec(&key->key);
  86 }
  87 static inline void static_key_deferred_flush(void *key)
  88 {
  89         STATIC_KEY_CHECK_USE(key);
  90 }
  91 static inline void
  92 jump_label_rate_limit(struct static_key_deferred *key,
  93                 unsigned long rl)
  94 {
  95         STATIC_KEY_CHECK_USE(key);
  96 }
  97 #endif  /* CONFIG_JUMP_LABEL */
  98 #endif  /* _LINUX_JUMP_LABEL_RATELIMIT_H */