| Dave Young | 717115e | 2008-07-25 01:45:58 -0700 | [diff] [blame] | 1 | #ifndef _LINUX_RATELIMIT_H | 
 | 2 | #define _LINUX_RATELIMIT_H | 
| Dave Young | 717115e | 2008-07-25 01:45:58 -0700 | [diff] [blame] | 3 |  | 
| Ingo Molnar | 979f693 | 2009-09-22 14:44:11 +0200 | [diff] [blame] | 4 | #include <linux/param.h> | 
| OGAWA Hirofumi | f40c396 | 2010-05-24 14:33:11 -0700 | [diff] [blame] | 5 | #include <linux/spinlock.h> | 
| Ingo Molnar | 979f693 | 2009-09-22 14:44:11 +0200 | [diff] [blame] | 6 |  | 
 | 7 | #define DEFAULT_RATELIMIT_INTERVAL	(5 * HZ) | 
 | 8 | #define DEFAULT_RATELIMIT_BURST		10 | 
| Dave Young | 717115e | 2008-07-25 01:45:58 -0700 | [diff] [blame] | 9 |  | 
 | 10 | struct ratelimit_state { | 
| Ingo Molnar | 979f693 | 2009-09-22 14:44:11 +0200 | [diff] [blame] | 11 | 	spinlock_t	lock;		/* protect the state */ | 
 | 12 |  | 
 | 13 | 	int		interval; | 
 | 14 | 	int		burst; | 
 | 15 | 	int		printed; | 
 | 16 | 	int		missed; | 
 | 17 | 	unsigned long	begin; | 
| Dave Young | 717115e | 2008-07-25 01:45:58 -0700 | [diff] [blame] | 18 | }; | 
 | 19 |  | 
| Ingo Molnar | 979f693 | 2009-09-22 14:44:11 +0200 | [diff] [blame] | 20 | #define DEFINE_RATELIMIT_STATE(name, interval_init, burst_init)		\ | 
 | 21 | 									\ | 
 | 22 | 	struct ratelimit_state name = {					\ | 
 | 23 | 		.lock		= __SPIN_LOCK_UNLOCKED(name.lock),	\ | 
 | 24 | 		.interval	= interval_init,			\ | 
 | 25 | 		.burst		= burst_init,				\ | 
 | 26 | 	} | 
| Dave Young | 717115e | 2008-07-25 01:45:58 -0700 | [diff] [blame] | 27 |  | 
| OGAWA Hirofumi | f40c396 | 2010-05-24 14:33:11 -0700 | [diff] [blame] | 28 | static inline void ratelimit_state_init(struct ratelimit_state *rs, | 
 | 29 | 					int interval, int burst) | 
 | 30 | { | 
 | 31 | 	spin_lock_init(&rs->lock); | 
 | 32 | 	rs->interval = interval; | 
 | 33 | 	rs->burst = burst; | 
 | 34 | 	rs->printed = 0; | 
 | 35 | 	rs->missed = 0; | 
 | 36 | 	rs->begin = 0; | 
 | 37 | } | 
 | 38 |  | 
| Namhyung Kim | f5d87d8 | 2010-10-26 14:22:49 -0700 | [diff] [blame] | 39 | extern struct ratelimit_state printk_ratelimit_state; | 
 | 40 |  | 
| Christian Borntraeger | 5c82871 | 2009-10-23 14:58:11 +0200 | [diff] [blame] | 41 | extern int ___ratelimit(struct ratelimit_state *rs, const char *func); | 
 | 42 | #define __ratelimit(state) ___ratelimit(state, __func__) | 
| Ingo Molnar | 979f693 | 2009-09-22 14:44:11 +0200 | [diff] [blame] | 43 |  | 
 | 44 | #endif /* _LINUX_RATELIMIT_H */ |