1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _LINUX_RATELIMIT_TYPES_H 3 #define _LINUX_RATELIMIT_TYPES_H 4 5 #include <linux/bits.h> 6 #include <linux/param.h> 7 #include <linux/spinlock_types_raw.h> 8 9 #define DEFAULT_RATELIMIT_INTERVAL (5 * HZ) 10 #define DEFAULT_RATELIMIT_BURST 10 11 12 /* issue num suppressed message on exit */ 13 #define RATELIMIT_MSG_ON_RELEASE BIT(0) 14 15 struct ratelimit_state { 16 raw_spinlock_t lock; /* protect the state */ 17 18 int interval; 19 int burst; 20 int printed; 21 int missed; 22 unsigned int flags; 23 unsigned long begin; 24 }; 25 26 #define RATELIMIT_STATE_INIT_FLAGS(name, interval_init, burst_init, flags_init) { \ 27 .lock = __RAW_SPIN_LOCK_UNLOCKED(name.lock), \ 28 .interval = interval_init, \ 29 .burst = burst_init, \ 30 .flags = flags_init, \ 31 } 32 33 #define RATELIMIT_STATE_INIT(name, interval_init, burst_init) \ 34 RATELIMIT_STATE_INIT_FLAGS(name, interval_init, burst_init, 0) 35 36 #define RATELIMIT_STATE_INIT_DISABLED \ 37 RATELIMIT_STATE_INIT(ratelimit_state, 0, DEFAULT_RATELIMIT_BURST) 38 39 #define DEFINE_RATELIMIT_STATE(name, interval_init, burst_init) \ 40 \ 41 struct ratelimit_state name = \ 42 RATELIMIT_STATE_INIT(name, interval_init, burst_init) \ 43 44 extern int ___ratelimit(struct ratelimit_state *rs, const char *func); 45 #define __ratelimit(state) ___ratelimit(state, __func__) 46 47 #endif /* _LINUX_RATELIMIT_TYPES_H */ 48