1 files changed, 22 insertions, 6 deletions

diff --git a/include/net/snmp.h b/include/net/snmp.h
index ea206bff0dc4..ce2f48507510 100644
--- a/include/net/snmp.h
+++ b/include/net/snmp.h
@@ -23,6 +23,7 @@
 
 #include <linux/cache.h>
 #include <linux/snmp.h>
+#include <linux/smp.h>
 
 /*
  * Mibs are stored in array of unsigned long.
@@ -117,6 +118,11 @@ struct linux_mib {
         unsigned long   mibs[LINUX_MIB_MAX];
 };
 
+/* Linux Xfrm */
+#define LINUX_MIB_XFRMMAX       __LINUX_MIB_XFRMMAX
+struct linux_xfrm_mib {
+        unsigned long   mibs[LINUX_MIB_XFRMMAX];
+};
 
 /* 
  * FIXME: On x86 and some other CPUs the split into user and softirq parts
@@ -134,17 +140,27 @@ struct linux_mib {
 
 #define SNMP_INC_STATS_BH(mib, field)   \
         (per_cpu_ptr(mib[0], raw_smp_processor_id())->mibs[field]++)
-#define SNMP_INC_STATS_OFFSET_BH(mib, field, offset)    \
-        (per_cpu_ptr(mib[0], raw_smp_processor_id())->mibs[field + (offset)]++)
 #define SNMP_INC_STATS_USER(mib, field) \
-        (per_cpu_ptr(mib[1], raw_smp_processor_id())->mibs[field]++)
+        do { \
+                per_cpu_ptr(mib[1], get_cpu())->mibs[field]++; \
+                put_cpu(); \
+        } while (0)
 #define SNMP_INC_STATS(mib, field)      \
-        (per_cpu_ptr(mib[!in_softirq()], raw_smp_processor_id())->mibs[field]++)
+        do { \
+                per_cpu_ptr(mib[!in_softirq()], get_cpu())->mibs[field]++; \
+                put_cpu(); \
+        } while (0)
 #define SNMP_DEC_STATS(mib, field)      \
-        (per_cpu_ptr(mib[!in_softirq()], raw_smp_processor_id())->mibs[field]--)
+        do { \
+                per_cpu_ptr(mib[!in_softirq()], get_cpu())->mibs[field]--; \
+                put_cpu(); \
+        } while (0)
 #define SNMP_ADD_STATS_BH(mib, field, addend)   \
         (per_cpu_ptr(mib[0], raw_smp_processor_id())->mibs[field] += addend)
 #define SNMP_ADD_STATS_USER(mib, field, addend)         \
-        (per_cpu_ptr(mib[1], raw_smp_processor_id())->mibs[field] += addend)
+        do { \
+                per_cpu_ptr(mib[1], get_cpu())->mibs[field] += addend; \
+                put_cpu(); \
+        } while (0)
 
 #endif