release/src-rt-6.x/linux/linux-2.6/include/linux/percpu_counter.h

   1 #ifndef _LINUX_PERCPU_COUNTER_H
   2 #define _LINUX_PERCPU_COUNTER_H
   3 /*
   4  * A simple "approximate counter" for use in ext2 and ext3 superblocks.
   5  *
   6  * WARNING: these things are HUGE.  4 kbytes per counter on 32-way P4.
   7  */
   8
   9 #include <linux/spinlock.h>
  10 #include <linux/smp.h>
  11 #include <linux/threads.h>
  12 #include <linux/percpu.h>
  13 #include <linux/types.h>
  14
  15 #ifdef CONFIG_SMP
  16
  17 struct percpu_counter {
  18         spinlock_t lock;
  19         s64 count;
  20         s32 *counters;
  21 };
  22
  23 #if NR_CPUS >= 16
  24 #define FBC_BATCH       (NR_CPUS*2)
  25 #else
  26 #define FBC_BATCH       (NR_CPUS*4)
  27 #endif
  28
  29 static inline void percpu_counter_init(struct percpu_counter *fbc, s64 amount)
  30 {
  31         spin_lock_init(&fbc->lock);
  32         fbc->count = amount;
  33         fbc->counters = alloc_percpu(s32);
  34 }
  35
  36 static inline void percpu_counter_destroy(struct percpu_counter *fbc)
  37 {
  38         free_percpu(fbc->counters);
  39 }
  40
  41 void percpu_counter_mod(struct percpu_counter *fbc, s32 amount);
  42 s64 percpu_counter_sum(struct percpu_counter *fbc);
  43
  44 static inline s64 percpu_counter_read(struct percpu_counter *fbc)
  45 {
  46         return fbc->count;
  47 }
  48
  49 /*
  50  * It is possible for the percpu_counter_read() to return a small negative
  51  * number for some counter which should never be negative.
  52  *
  53  */
  54 static inline s64 percpu_counter_read_positive(struct percpu_counter *fbc)
  55 {
  56         s64 ret = fbc->count;
  57
  58         barrier();              /* Prevent reloads of fbc->count */
  59         if (ret >= 0)
  60                 return ret;
  61         return 1;
  62 }
  63
  64 static inline s64 percpu_counter_sum_positive(struct percpu_counter *fbc)
  65 {
  66         s64 ret = percpu_counter_sum(fbc);
  67         return ret < 0 ? 0 : ret;
  68 }
  69
  70 #else
  71
  72 struct percpu_counter {
  73         s64 count;
  74 };
  75
  76 static inline void percpu_counter_init(struct percpu_counter *fbc, s64 amount)
  77 {
  78         fbc->count = amount;
  79 }
  80
  81 static inline void percpu_counter_destroy(struct percpu_counter *fbc)
  82 {
  83 }
  84
  85 static inline void
  86 percpu_counter_mod(struct percpu_counter *fbc, s32 amount)
  87 {
  88         preempt_disable();
  89         fbc->count += amount;
  90         preempt_enable();
  91 }
  92
  93 static inline s64 percpu_counter_read(struct percpu_counter *fbc)
  94 {
  95         return fbc->count;
  96 }
  97
  98 static inline s64 percpu_counter_read_positive(struct percpu_counter *fbc)
  99 {
 100         return fbc->count;
 101 }
 102
 103 static inline s64 percpu_counter_sum(struct percpu_counter *fbc)
 104 {
 105         return percpu_counter_read(fbc);
 106 }
 107
 108 static inline s64 percpu_counter_sum_positive(struct percpu_counter *fbc)
 109 {
 110         return percpu_counter_read_positive(fbc);
 111 }
 112
 113 #endif  /* CONFIG_SMP */
 114
 115 static inline void percpu_counter_inc(struct percpu_counter *fbc)
 116 {
 117         percpu_counter_mod(fbc, 1);
 118 }
 119
 120 static inline void percpu_counter_dec(struct percpu_counter *fbc)
 121 {
 122         percpu_counter_mod(fbc, -1);
 123 }
 124
 125 #endif /* _LINUX_PERCPU_COUNTER_H */