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