u64_stats_sync.h 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  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. #if BITS_PER_LONG == 32 && defined(CONFIG_SMP)
  68. # define u64_stats_init(syncp) seqcount_init(syncp.seq)
  69. #else
  70. # define u64_stats_init(syncp) do { } while (0)
  71. #endif
  72. static inline void u64_stats_update_begin(struct u64_stats_sync *syncp)
  73. {
  74. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  75. write_seqcount_begin(&syncp->seq);
  76. #endif
  77. }
  78. static inline void u64_stats_update_end(struct u64_stats_sync *syncp)
  79. {
  80. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  81. write_seqcount_end(&syncp->seq);
  82. #endif
  83. }
  84. static inline unsigned int u64_stats_fetch_begin(const struct u64_stats_sync *syncp)
  85. {
  86. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  87. return read_seqcount_begin(&syncp->seq);
  88. #else
  89. #if BITS_PER_LONG==32
  90. preempt_disable();
  91. #endif
  92. return 0;
  93. #endif
  94. }
  95. static inline bool u64_stats_fetch_retry(const struct u64_stats_sync *syncp,
  96. unsigned int start)
  97. {
  98. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  99. return read_seqcount_retry(&syncp->seq, start);
  100. #else
  101. #if BITS_PER_LONG==32
  102. preempt_enable();
  103. #endif
  104. return false;
  105. #endif
  106. }
  107. /*
  108. * In case irq handlers can update u64 counters, readers can use following helpers
  109. * - SMP 32bit arches use seqcount protection, irq safe.
  110. * - UP 32bit must disable irqs.
  111. * - 64bit have no problem atomically reading u64 values, irq safe.
  112. */
  113. static inline unsigned int u64_stats_fetch_begin_irq(const struct u64_stats_sync *syncp)
  114. {
  115. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  116. return read_seqcount_begin(&syncp->seq);
  117. #else
  118. #if BITS_PER_LONG==32
  119. local_irq_disable();
  120. #endif
  121. return 0;
  122. #endif
  123. }
  124. static inline bool u64_stats_fetch_retry_irq(const struct u64_stats_sync *syncp,
  125. unsigned int start)
  126. {
  127. #if BITS_PER_LONG==32 && defined(CONFIG_SMP)
  128. return read_seqcount_retry(&syncp->seq, start);
  129. #else
  130. #if BITS_PER_LONG==32
  131. local_irq_enable();
  132. #endif
  133. return false;
  134. #endif
  135. }
  136. #endif /* _LINUX_U64_STATS_SYNC_H */