Commit | Line | Data |
---|---|---|
b2441318 | 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
717115e1 DY |
2 | #ifndef _LINUX_RATELIMIT_H |
3 | #define _LINUX_RATELIMIT_H | |
979f693d | 4 | |
b4a461e7 | 5 | #include <linux/ratelimit_types.h> |
6b1d174b | 6 | #include <linux/sched.h> |
f40c396a | 7 | #include <linux/spinlock.h> |
717115e1 | 8 | |
f40c396a OH |
9 | static inline void ratelimit_state_init(struct ratelimit_state *rs, |
10 | int interval, int burst) | |
11 | { | |
6b1d174b BP |
12 | memset(rs, 0, sizeof(*rs)); |
13 | ||
07354eb1 | 14 | raw_spin_lock_init(&rs->lock); |
6b1d174b BP |
15 | rs->interval = interval; |
16 | rs->burst = burst; | |
17 | } | |
18 | ||
19 | static inline void ratelimit_default_init(struct ratelimit_state *rs) | |
20 | { | |
21 | return ratelimit_state_init(rs, DEFAULT_RATELIMIT_INTERVAL, | |
22 | DEFAULT_RATELIMIT_BURST); | |
23 | } | |
24 | ||
25 | static inline void ratelimit_state_exit(struct ratelimit_state *rs) | |
26 | { | |
27 | if (!(rs->flags & RATELIMIT_MSG_ON_RELEASE)) | |
28 | return; | |
29 | ||
30 | if (rs->missed) { | |
31 | pr_warn("%s: %d output lines suppressed due to ratelimiting\n", | |
32 | current->comm, rs->missed); | |
33 | rs->missed = 0; | |
34 | } | |
35 | } | |
36 | ||
37 | static inline void | |
38 | ratelimit_set_flags(struct ratelimit_state *rs, unsigned long flags) | |
39 | { | |
40 | rs->flags = flags; | |
f40c396a OH |
41 | } |
42 | ||
f5d87d85 NK |
43 | extern struct ratelimit_state printk_ratelimit_state; |
44 | ||
86e4ca66 DM |
45 | #ifdef CONFIG_PRINTK |
46 | ||
1b011e2f JS |
47 | #define WARN_ON_RATELIMIT(condition, state) ({ \ |
48 | bool __rtn_cond = !!(condition); \ | |
49 | WARN_ON(__rtn_cond && __ratelimit(state)); \ | |
50 | __rtn_cond; \ | |
51 | }) | |
86e4ca66 | 52 | |
2351a6c6 | 53 | #define WARN_RATELIMIT(condition, format, ...) \ |
86e4ca66 DM |
54 | ({ \ |
55 | static DEFINE_RATELIMIT_STATE(_rs, \ | |
56 | DEFAULT_RATELIMIT_INTERVAL, \ | |
57 | DEFAULT_RATELIMIT_BURST); \ | |
2351a6c6 MT |
58 | int rtn = !!(condition); \ |
59 | \ | |
60 | if (unlikely(rtn && __ratelimit(&_rs))) \ | |
61 | WARN(rtn, format, ##__VA_ARGS__); \ | |
62 | \ | |
63 | rtn; \ | |
86e4ca66 DM |
64 | }) |
65 | ||
66 | #else | |
67 | ||
68 | #define WARN_ON_RATELIMIT(condition, state) \ | |
69 | WARN_ON(condition) | |
70 | ||
2351a6c6 | 71 | #define WARN_RATELIMIT(condition, format, ...) \ |
86e4ca66 | 72 | ({ \ |
2351a6c6 | 73 | int rtn = WARN(condition, format, ##__VA_ARGS__); \ |
86e4ca66 DM |
74 | rtn; \ |
75 | }) | |
76 | ||
77 | #endif | |
78 | ||
979f693d | 79 | #endif /* _LINUX_RATELIMIT_H */ |