ratelimit.h 961 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #ifndef _LINUX_RATELIMIT_H
  2. #define _LINUX_RATELIMIT_H
  3. #include <linux/param.h>
  4. #include <linux/spinlock.h>
  5. #define DEFAULT_RATELIMIT_INTERVAL (5 * HZ)
  6. #define DEFAULT_RATELIMIT_BURST 10
  7. struct ratelimit_state {
  8. spinlock_t lock; /* protect the state */
  9. int interval;
  10. int burst;
  11. int printed;
  12. int missed;
  13. unsigned long begin;
  14. };
  15. #define DEFINE_RATELIMIT_STATE(name, interval_init, burst_init) \
  16. \
  17. struct ratelimit_state name = { \
  18. .lock = __SPIN_LOCK_UNLOCKED(name.lock), \
  19. .interval = interval_init, \
  20. .burst = burst_init, \
  21. }
  22. static inline void ratelimit_state_init(struct ratelimit_state *rs,
  23. int interval, int burst)
  24. {
  25. spin_lock_init(&rs->lock);
  26. rs->interval = interval;
  27. rs->burst = burst;
  28. rs->printed = 0;
  29. rs->missed = 0;
  30. rs->begin = 0;
  31. }
  32. extern int ___ratelimit(struct ratelimit_state *rs, const char *func);
  33. #define __ratelimit(state) ___ratelimit(state, __func__)
  34. #endif /* _LINUX_RATELIMIT_H */