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