snmp.h 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. /*
  2. *
  3. * SNMP MIB entries for the IP subsystem.
  4. *
  5. * Alan Cox <gw4pts@gw4pts.ampr.org>
  6. *
  7. * We don't chose to implement SNMP in the kernel (this would
  8. * be silly as SNMP is a pain in the backside in places). We do
  9. * however need to collect the MIB statistics and export them
  10. * out of /proc (eventually)
  11. *
  12. * This program is free software; you can redistribute it and/or
  13. * modify it under the terms of the GNU General Public License
  14. * as published by the Free Software Foundation; either version
  15. * 2 of the License, or (at your option) any later version.
  16. *
  17. */
  18. #ifndef _SNMP_H
  19. #define _SNMP_H
  20. #include <linux/cache.h>
  21. #include <linux/snmp.h>
  22. #include <linux/smp.h>
  23. /*
  24. * Mibs are stored in array of unsigned long.
  25. */
  26. /*
  27. * struct snmp_mib{}
  28. * - list of entries for particular API (such as /proc/net/snmp)
  29. * - name of entries.
  30. */
  31. struct snmp_mib {
  32. const char *name;
  33. int entry;
  34. };
  35. #define SNMP_MIB_ITEM(_name,_entry) { \
  36. .name = _name, \
  37. .entry = _entry, \
  38. }
  39. #define SNMP_MIB_SENTINEL { \
  40. .name = NULL, \
  41. .entry = 0, \
  42. }
  43. /*
  44. * We use unsigned longs for most mibs but u64 for ipstats.
  45. */
  46. #include <linux/u64_stats_sync.h>
  47. /* IPstats */
  48. #define IPSTATS_MIB_MAX __IPSTATS_MIB_MAX
  49. struct ipstats_mib {
  50. /* mibs[] must be first field of struct ipstats_mib */
  51. u64 mibs[IPSTATS_MIB_MAX];
  52. struct u64_stats_sync syncp;
  53. };
  54. /* ICMP */
  55. #define ICMP_MIB_MAX __ICMP_MIB_MAX
  56. struct icmp_mib {
  57. unsigned long mibs[ICMP_MIB_MAX];
  58. };
  59. #define ICMPMSG_MIB_MAX __ICMPMSG_MIB_MAX
  60. struct icmpmsg_mib {
  61. atomic_long_t mibs[ICMPMSG_MIB_MAX];
  62. };
  63. /* ICMP6 (IPv6-ICMP) */
  64. #define ICMP6_MIB_MAX __ICMP6_MIB_MAX
  65. /* per network ns counters */
  66. struct icmpv6_mib {
  67. unsigned long mibs[ICMP6_MIB_MAX];
  68. };
  69. /* per device counters, (shared on all cpus) */
  70. struct icmpv6_mib_device {
  71. atomic_long_t mibs[ICMP6_MIB_MAX];
  72. };
  73. #define ICMP6MSG_MIB_MAX __ICMP6MSG_MIB_MAX
  74. /* per network ns counters */
  75. struct icmpv6msg_mib {
  76. atomic_long_t mibs[ICMP6MSG_MIB_MAX];
  77. };
  78. /* per device counters, (shared on all cpus) */
  79. struct icmpv6msg_mib_device {
  80. atomic_long_t mibs[ICMP6MSG_MIB_MAX];
  81. };
  82. /* TCP */
  83. #define TCP_MIB_MAX __TCP_MIB_MAX
  84. struct tcp_mib {
  85. unsigned long mibs[TCP_MIB_MAX];
  86. };
  87. /* UDP */
  88. #define UDP_MIB_MAX __UDP_MIB_MAX
  89. struct udp_mib {
  90. unsigned long mibs[UDP_MIB_MAX];
  91. };
  92. /* Linux */
  93. #define LINUX_MIB_MAX __LINUX_MIB_MAX
  94. struct linux_mib {
  95. unsigned long mibs[LINUX_MIB_MAX];
  96. };
  97. /* Linux Xfrm */
  98. #define LINUX_MIB_XFRMMAX __LINUX_MIB_XFRMMAX
  99. struct linux_xfrm_mib {
  100. unsigned long mibs[LINUX_MIB_XFRMMAX];
  101. };
  102. #define SNMP_ARRAY_SZ 1
  103. #define DEFINE_SNMP_STAT(type, name) \
  104. __typeof__(type) __percpu *name[SNMP_ARRAY_SZ]
  105. #define DEFINE_SNMP_STAT_ATOMIC(type, name) \
  106. __typeof__(type) *name
  107. #define DECLARE_SNMP_STAT(type, name) \
  108. extern __typeof__(type) __percpu *name[SNMP_ARRAY_SZ]
  109. #define SNMP_INC_STATS_BH(mib, field) \
  110. __this_cpu_inc(mib[0]->mibs[field])
  111. #define SNMP_INC_STATS_USER(mib, field) \
  112. this_cpu_inc(mib[0]->mibs[field])
  113. #define SNMP_INC_STATS_ATOMIC_LONG(mib, field) \
  114. atomic_long_inc(&mib->mibs[field])
  115. #define SNMP_INC_STATS(mib, field) \
  116. this_cpu_inc(mib[0]->mibs[field])
  117. #define SNMP_DEC_STATS(mib, field) \
  118. this_cpu_dec(mib[0]->mibs[field])
  119. #define SNMP_ADD_STATS_BH(mib, field, addend) \
  120. __this_cpu_add(mib[0]->mibs[field], addend)
  121. #define SNMP_ADD_STATS_USER(mib, field, addend) \
  122. this_cpu_add(mib[0]->mibs[field], addend)
  123. #define SNMP_ADD_STATS(mib, field, addend) \
  124. this_cpu_add(mib[0]->mibs[field], addend)
  125. /*
  126. * Use "__typeof__(*mib[0]) *ptr" instead of "__typeof__(mib[0]) ptr"
  127. * to make @ptr a non-percpu pointer.
  128. */
  129. #define SNMP_UPD_PO_STATS(mib, basefield, addend) \
  130. do { \
  131. this_cpu_inc(mib[0]->mibs[basefield##PKTS]); \
  132. this_cpu_add(mib[0]->mibs[basefield##OCTETS], addend); \
  133. } while (0)
  134. #define SNMP_UPD_PO_STATS_BH(mib, basefield, addend) \
  135. do { \
  136. __this_cpu_inc(mib[0]->mibs[basefield##PKTS]); \
  137. __this_cpu_add(mib[0]->mibs[basefield##OCTETS], addend); \
  138. } while (0)
  139. #if BITS_PER_LONG==32
  140. #define SNMP_ADD_STATS64_BH(mib, field, addend) \
  141. do { \
  142. __typeof__(*mib[0]) *ptr = __this_cpu_ptr((mib)[0]); \
  143. u64_stats_update_begin(&ptr->syncp); \
  144. ptr->mibs[field] += addend; \
  145. u64_stats_update_end(&ptr->syncp); \
  146. } while (0)
  147. #define SNMP_ADD_STATS64_USER(mib, field, addend) \
  148. do { \
  149. local_bh_disable(); \
  150. SNMP_ADD_STATS64_BH(mib, field, addend); \
  151. local_bh_enable(); \
  152. } while (0)
  153. #define SNMP_ADD_STATS64(mib, field, addend) \
  154. SNMP_ADD_STATS64_USER(mib, field, addend)
  155. #define SNMP_INC_STATS64_BH(mib, field) SNMP_ADD_STATS64_BH(mib, field, 1)
  156. #define SNMP_INC_STATS64_USER(mib, field) SNMP_ADD_STATS64_USER(mib, field, 1)
  157. #define SNMP_INC_STATS64(mib, field) SNMP_ADD_STATS64(mib, field, 1)
  158. #define SNMP_UPD_PO_STATS64_BH(mib, basefield, addend) \
  159. do { \
  160. __typeof__(*mib[0]) *ptr; \
  161. ptr = __this_cpu_ptr((mib)[0]); \
  162. u64_stats_update_begin(&ptr->syncp); \
  163. ptr->mibs[basefield##PKTS]++; \
  164. ptr->mibs[basefield##OCTETS] += addend; \
  165. u64_stats_update_end(&ptr->syncp); \
  166. } while (0)
  167. #define SNMP_UPD_PO_STATS64(mib, basefield, addend) \
  168. do { \
  169. local_bh_disable(); \
  170. SNMP_UPD_PO_STATS64_BH(mib, basefield, addend); \
  171. local_bh_enable(); \
  172. } while (0)
  173. #else
  174. #define SNMP_INC_STATS64_BH(mib, field) SNMP_INC_STATS_BH(mib, field)
  175. #define SNMP_INC_STATS64_USER(mib, field) SNMP_INC_STATS_USER(mib, field)
  176. #define SNMP_INC_STATS64(mib, field) SNMP_INC_STATS(mib, field)
  177. #define SNMP_DEC_STATS64(mib, field) SNMP_DEC_STATS(mib, field)
  178. #define SNMP_ADD_STATS64_BH(mib, field, addend) SNMP_ADD_STATS_BH(mib, field, addend)
  179. #define SNMP_ADD_STATS64_USER(mib, field, addend) SNMP_ADD_STATS_USER(mib, field, addend)
  180. #define SNMP_ADD_STATS64(mib, field, addend) SNMP_ADD_STATS(mib, field, addend)
  181. #define SNMP_UPD_PO_STATS64(mib, basefield, addend) SNMP_UPD_PO_STATS(mib, basefield, addend)
  182. #define SNMP_UPD_PO_STATS64_BH(mib, basefield, addend) SNMP_UPD_PO_STATS_BH(mib, basefield, addend)
  183. #endif
  184. #endif