ratelimit.h 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  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. raw_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 = __RAW_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. raw_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 struct ratelimit_state printk_ratelimit_state;
  33. extern int ___ratelimit(struct ratelimit_state *rs, const char *func);
  34. #define __ratelimit(state) ___ratelimit(state, __func__)
  35. #ifdef CONFIG_PRINTK
  36. #define WARN_ON_RATELIMIT(condition, state) \
  37. WARN_ON((condition) && __ratelimit(state))
  38. #define WARN_RATELIMIT(condition, format, ...) \
  39. ({ \
  40. static DEFINE_RATELIMIT_STATE(_rs, \
  41. DEFAULT_RATELIMIT_INTERVAL, \
  42. DEFAULT_RATELIMIT_BURST); \
  43. int rtn = !!(condition); \
  44. \
  45. if (unlikely(rtn && __ratelimit(&_rs))) \
  46. WARN(rtn, format, ##__VA_ARGS__); \
  47. \
  48. rtn; \
  49. })
  50. #else
  51. #define WARN_ON_RATELIMIT(condition, state) \
  52. WARN_ON(condition)
  53. #define WARN_RATELIMIT(condition, format, ...) \
  54. ({ \
  55. int rtn = WARN(condition, format, ##__VA_ARGS__); \
  56. rtn; \
  57. })
  58. #endif
  59. #endif /* _LINUX_RATELIMIT_H */