u64_stats_sync.h 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. #ifndef _LINUX_U64_STATS_SYNC_H
  2. #define _LINUX_U64_STATS_SYNC_H
  3. /*
  4. * To properly implement 64bits network statistics on 32bit and 64bit hosts,
  5. * we provide a synchronization point, that is a noop on 64bit or UP kernels.
  6. *
  7. * Key points :
  8. * 1) Use a seqcount on SMP 32bits, with low overhead.
  9. * 2) Whole thing is a noop on 64bit arches or UP kernels.
  10. * 3) Write side must ensure mutual exclusion or one seqcount update could
  11. * be lost, thus blocking readers forever.
  12. * If this synchronization point is not a mutex, but a spinlock or
  13. * spinlock_bh() or disable_bh() :
  14. * 3.1) Write side should not sleep.
  15. * 3.2) Write side should not allow preemption.
  16. * 3.3) If applicable, interrupts should be disabled.
  17. *
  18. * 4) If reader fetches several counters, there is no guarantee the whole values
  19. * are consistent (remember point 1) : this is a noop on 64bit arches anyway)
  20. *
  21. * 5) readers are allowed to sleep or be preempted/interrupted : They perform
  22. * pure reads. But if they have to fetch many values, it's better to not allow
  23. * preemptions/interruptions to avoid many retries.
  24. *
  25. * 6) If counter might be written by an interrupt, readers should block interrupts.
  26. * (On UP, there is no seqcount_t protection, a reader allowing interrupts could
  27. * read partial values)
  28. *
  29. * 7) For irq and softirq uses, readers can use u64_stats_fetch_begin_irq() and
  30. * u64_stats_fetch_retry_irq() helpers
  31. *
  32. * Usage :
  33. *
  34. * Stats producer (writer) should use following template granted it already got
  35. * an exclusive access to counters (a lock is already taken, or per cpu
  36. * data is used [in a non preemptable context])
  37. *
  38. * spin_lock_bh(...) or other synchronization to get exclusive access
  39. * ...
  40. * u64_stats_update_begin(&stats->syncp);
  41. * stats->bytes64 += len; // non atomic operation
  42. * stats->packets64++; // non atomic operation
  43. * u64_stats_update_end(&stats->syncp);
  44. *
  45. * While a consumer (reader) should use following template to get consistent
  46. * snapshot for each variable (but no guarantee on several ones)
  47. *
  48. * u64 tbytes, tpackets;
  49. * unsigned int start;
  50. *
  51. * do {
  52. * start = u64_stats_fetch_begin(&stats->syncp);
  53. * tbytes = stats->bytes64; // non atomic operation
  54. * tpackets = stats->packets64; // non atomic operation
  55. * } while (u64_stats_fetch_retry(&stats->syncp, start));
  56. *
  57. *
  58. * Example of use in drivers/net/loopback.c, using per_cpu containers,
  59. * in BH disabled context.
  60. */
  61. #include <linux/seqlock.h>
  62. struct u64_stats_sync {
  63. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  64. seqcount_t seq;
  65. #endif
  66. };
  67. static inline void u64_stats_init(struct u64_stats_sync *syncp)
  68. {
  69. #if BITS_PER_LONG == 32 && defined(CONFIG_SMP)
  70. seqcount_init(&syncp->seq);
  71. #endif
  72. }
  73. static inline void u64_stats_update_begin(struct u64_stats_sync *syncp)
  74. {
  75. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  76. write_seqcount_begin(&syncp->seq);
  77. #endif
  78. }
  79. static inline void u64_stats_update_end(struct u64_stats_sync *syncp)
  80. {
  81. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  82. write_seqcount_end(&syncp->seq);
  83. #endif
  84. }
  85. static inline void u64_stats_update_begin_raw(struct u64_stats_sync *syncp)
  86. {
  87. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  88. raw_write_seqcount_begin(&syncp->seq);
  89. #endif
  90. }
  91. static inline void u64_stats_update_end_raw(struct u64_stats_sync *syncp)
  92. {
  93. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  94. raw_write_seqcount_end(&syncp->seq);
  95. #endif
  96. }
  97. static inline unsigned int __u64_stats_fetch_begin(const struct u64_stats_sync *syncp)
  98. {
  99. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  100. return read_seqcount_begin(&syncp->seq);
  101. #else
  102. return 0;
  103. #endif
  104. }
  105. static inline unsigned int u64_stats_fetch_begin(const struct u64_stats_sync *syncp)
  106. {
  107. #if BITS_PER_LONG==32 && !defined(CONFIG_SMP)
  108. preempt_disable();
  109. #endif
  110. return __u64_stats_fetch_begin(syncp);
  111. }
  112. static inline bool __u64_stats_fetch_retry(const struct u64_stats_sync *syncp,
  113. unsigned int start)
  114. {
  115. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  116. return read_seqcount_retry(&syncp->seq, start);
  117. #else
  118. return false;
  119. #endif
  120. }
  121. static inline bool u64_stats_fetch_retry(const struct u64_stats_sync *syncp,
  122. unsigned int start)
  123. {
  124. #if BITS_PER_LONG==32 && !defined(CONFIG_SMP)
  125. preempt_enable();
  126. #endif
  127. return __u64_stats_fetch_retry(syncp, start);
  128. }
  129. /*
  130. * In case irq handlers can update u64 counters, readers can use following helpers
  131. * - SMP 32bit arches use seqcount protection, irq safe.
  132. * - UP 32bit must disable irqs.
  133. * - 64bit have no problem atomically reading u64 values, irq safe.
  134. */
  135. static inline unsigned int u64_stats_fetch_begin_irq(const struct u64_stats_sync *syncp)
  136. {
  137. #if BITS_PER_LONG==32 && !defined(CONFIG_SMP)
  138. local_irq_disable();
  139. #endif
  140. return __u64_stats_fetch_begin(syncp);
  141. }
  142. static inline bool u64_stats_fetch_retry_irq(const struct u64_stats_sync *syncp,
  143. unsigned int start)
  144. {
  145. #if BITS_PER_LONG==32 && !defined(CONFIG_SMP)
  146. local_irq_enable();
  147. #endif
  148. return __u64_stats_fetch_retry(syncp, start);
  149. }
  150. #endif /* _LINUX_U64_STATS_SYNC_H */