2008-07-25 16:45:58 +08:00
|
|
|
#ifndef _LINUX_RATELIMIT_H
|
|
|
|
#define _LINUX_RATELIMIT_H
|
2009-09-22 20:44:11 +08:00
|
|
|
|
2008-07-25 16:45:58 +08:00
|
|
|
#include <linux/param.h>
|
2010-05-25 05:33:11 +08:00
|
|
|
#include <linux/spinlock.h>
|
2008-07-25 16:45:58 +08:00
|
|
|
|
2009-09-22 20:44:11 +08:00
|
|
|
#define DEFAULT_RATELIMIT_INTERVAL (5 * HZ)
|
|
|
|
#define DEFAULT_RATELIMIT_BURST 10
|
2008-07-25 16:45:58 +08:00
|
|
|
|
|
|
|
struct ratelimit_state {
|
2009-09-22 20:44:11 +08:00
|
|
|
spinlock_t lock; /* protect the state */
|
|
|
|
|
|
|
|
int interval;
|
|
|
|
int burst;
|
|
|
|
int printed;
|
|
|
|
int missed;
|
|
|
|
unsigned long begin;
|
2008-07-25 16:45:58 +08:00
|
|
|
};
|
|
|
|
|
2009-09-22 20:44:11 +08:00
|
|
|
#define DEFINE_RATELIMIT_STATE(name, interval_init, burst_init) \
|
|
|
|
\
|
|
|
|
struct ratelimit_state name = { \
|
|
|
|
.lock = __SPIN_LOCK_UNLOCKED(name.lock), \
|
|
|
|
.interval = interval_init, \
|
|
|
|
.burst = burst_init, \
|
|
|
|
}
|
2008-07-25 16:45:58 +08:00
|
|
|
|
2010-05-25 05:33:11 +08:00
|
|
|
static inline void ratelimit_state_init(struct ratelimit_state *rs,
|
|
|
|
int interval, int burst)
|
|
|
|
{
|
|
|
|
spin_lock_init(&rs->lock);
|
|
|
|
rs->interval = interval;
|
|
|
|
rs->burst = burst;
|
|
|
|
rs->printed = 0;
|
|
|
|
rs->missed = 0;
|
|
|
|
rs->begin = 0;
|
|
|
|
}
|
|
|
|
|
2009-10-23 20:58:11 +08:00
|
|
|
extern int ___ratelimit(struct ratelimit_state *rs, const char *func);
|
|
|
|
#define __ratelimit(state) ___ratelimit(state, __func__)
|
2009-09-22 20:44:11 +08:00
|
|
|
|
|
|
|
#endif /* _LINUX_RATELIMIT_H */
|