ratelimit.h 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. #ifndef _LINUX_RATELIMIT_H
  3. #define _LINUX_RATELIMIT_H
  4. #include <linux/ratelimit_types.h>
  5. #include <linux/sched.h>
  6. #include <linux/spinlock.h>
  7. static inline void ratelimit_state_init(struct ratelimit_state *rs,
  8. int interval, int burst)
  9. {
  10. memset(rs, 0, sizeof(*rs));
  11. raw_spin_lock_init(&rs->lock);
  12. rs->interval = interval;
  13. rs->burst = burst;
  14. }
  15. static inline void ratelimit_default_init(struct ratelimit_state *rs)
  16. {
  17. return ratelimit_state_init(rs, DEFAULT_RATELIMIT_INTERVAL,
  18. DEFAULT_RATELIMIT_BURST);
  19. }
  20. static inline void ratelimit_state_exit(struct ratelimit_state *rs)
  21. {
  22. if (!(rs->flags & RATELIMIT_MSG_ON_RELEASE))
  23. return;
  24. if (rs->missed) {
  25. pr_warn("%s: %d output lines suppressed due to ratelimiting\n",
  26. current->comm, rs->missed);
  27. rs->missed = 0;
  28. }
  29. }
  30. static inline void
  31. ratelimit_set_flags(struct ratelimit_state *rs, unsigned long flags)
  32. {
  33. rs->flags = flags;
  34. }
  35. extern struct ratelimit_state printk_ratelimit_state;
  36. #ifdef CONFIG_PRINTK
  37. #define WARN_ON_RATELIMIT(condition, state) ({ \
  38. bool __rtn_cond = !!(condition); \
  39. WARN_ON(__rtn_cond && __ratelimit(state)); \
  40. __rtn_cond; \
  41. })
  42. #define WARN_RATELIMIT(condition, format, ...) \
  43. ({ \
  44. static DEFINE_RATELIMIT_STATE(_rs, \
  45. DEFAULT_RATELIMIT_INTERVAL, \
  46. DEFAULT_RATELIMIT_BURST); \
  47. int rtn = !!(condition); \
  48. \
  49. if (unlikely(rtn && __ratelimit(&_rs))) \
  50. WARN(rtn, format, ##__VA_ARGS__); \
  51. \
  52. rtn; \
  53. })
  54. #else
  55. #define WARN_ON_RATELIMIT(condition, state) \
  56. WARN_ON(condition)
  57. #define WARN_RATELIMIT(condition, format, ...) \
  58. ({ \
  59. int rtn = WARN(condition, format, ##__VA_ARGS__); \
  60. rtn; \
  61. })
  62. #endif
  63. #endif /* _LINUX_RATELIMIT_H */