arch/s390/include/asm/atomic.h

   1 /*
   2  * Copyright IBM Corp. 1999, 2016
   3  * Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>,
   4  *            Denis Joseph Barrow,
   5  *            Arnd Bergmann,
   6  */
   7
   8 #ifndef __ARCH_S390_ATOMIC__
   9 #define __ARCH_S390_ATOMIC__
  10
  11 #include <linux/compiler.h>
  12 #include <linux/types.h>
  13 #include <asm/atomic_ops.h>
  14 #include <asm/barrier.h>
  15 #include <asm/cmpxchg.h>
  16
  17 #define ATOMIC_INIT(i)  { (i) }
  18
  19 static inline int atomic_read(const atomic_t *v)
  20 {
  21         int c;
  22
  23         asm volatile(
  24                 "       l       %0,%1\n"
  25                 : "=d" (c) : "Q" (v->counter));
  26         return c;
  27 }
  28
  29 static inline void atomic_set(atomic_t *v, int i)
  30 {
  31         asm volatile(
  32                 "       st      %1,%0\n"
  33                 : "=Q" (v->counter) : "d" (i));
  34 }
  35
  36 static inline int atomic_add_return(int i, atomic_t *v)
  37 {
  38         return __atomic_add_barrier(i, &v->counter) + i;
  39 }
  40
  41 static inline int atomic_fetch_add(int i, atomic_t *v)
  42 {
  43         return __atomic_add_barrier(i, &v->counter);
  44 }
  45
  46 static inline void atomic_add(int i, atomic_t *v)
  47 {
  48 #ifdef CONFIG_HAVE_MARCH_Z196_FEATURES
  49         if (__builtin_constant_p(i) && (i > -129) && (i < 128)) {
  50                 __atomic_add_const(i, &v->counter);
  51                 return;
  52         }
  53 #endif
  54         __atomic_add(i, &v->counter);
  55 }
  56
  57 #define atomic_add_negative(_i, _v)     (atomic_add_return(_i, _v) < 0)
  58 #define atomic_inc(_v)                  atomic_add(1, _v)
  59 #define atomic_inc_return(_v)           atomic_add_return(1, _v)
  60 #define atomic_inc_and_test(_v)         (atomic_add_return(1, _v) == 0)
  61 #define atomic_sub(_i, _v)              atomic_add(-(int)(_i), _v)
  62 #define atomic_sub_return(_i, _v)       atomic_add_return(-(int)(_i), _v)
  63 #define atomic_fetch_sub(_i, _v)        atomic_fetch_add(-(int)(_i), _v)
  64 #define atomic_sub_and_test(_i, _v)     (atomic_sub_return(_i, _v) == 0)
  65 #define atomic_dec(_v)                  atomic_sub(1, _v)
  66 #define atomic_dec_return(_v)           atomic_sub_return(1, _v)
  67 #define atomic_dec_and_test(_v)         (atomic_sub_return(1, _v) == 0)
  68
  69 #define ATOMIC_OPS(op)                                                  \
  70 static inline void atomic_##op(int i, atomic_t *v)                      \
  71 {                                                                       \
  72         __atomic_##op(i, &v->counter);                                  \
  73 }                                                                       \
  74 static inline int atomic_fetch_##op(int i, atomic_t *v)                 \
  75 {                                                                       \
  76         return __atomic_##op##_barrier(i, &v->counter);                 \
  77 }
  78
  79 ATOMIC_OPS(and)
  80 ATOMIC_OPS(or)
  81 ATOMIC_OPS(xor)
  82
  83 #undef ATOMIC_OPS
  84
  85 #define atomic_xchg(v, new) (xchg(&((v)->counter), new))
  86
  87 static inline int atomic_cmpxchg(atomic_t *v, int old, int new)
  88 {
  89         return __atomic_cmpxchg(&v->counter, old, new);
  90 }
  91
  92 static inline int __atomic_add_unless(atomic_t *v, int a, int u)
  93 {
  94         int c, old;
  95         c = atomic_read(v);
  96         for (;;) {
  97                 if (unlikely(c == u))
  98                         break;
  99                 old = atomic_cmpxchg(v, c, c + a);
 100                 if (likely(old == c))
 101                         break;
 102                 c = old;
 103         }
 104         return c;
 105 }
 106
 107 #define ATOMIC64_INIT(i)  { (i) }
 108
 109 static inline long atomic64_read(const atomic64_t *v)
 110 {
 111         long c;
 112
 113         asm volatile(
 114                 "       lg      %0,%1\n"
 115                 : "=d" (c) : "Q" (v->counter));
 116         return c;
 117 }
 118
 119 static inline void atomic64_set(atomic64_t *v, long i)
 120 {
 121         asm volatile(
 122                 "       stg     %1,%0\n"
 123                 : "=Q" (v->counter) : "d" (i));
 124 }
 125
 126 static inline long atomic64_add_return(long i, atomic64_t *v)
 127 {
 128         return __atomic64_add_barrier(i, &v->counter) + i;
 129 }
 130
 131 static inline long atomic64_fetch_add(long i, atomic64_t *v)
 132 {
 133         return __atomic64_add_barrier(i, &v->counter);
 134 }
 135
 136 static inline void atomic64_add(long i, atomic64_t *v)
 137 {
 138 #ifdef CONFIG_HAVE_MARCH_Z196_FEATURES
 139         if (__builtin_constant_p(i) && (i > -129) && (i < 128)) {
 140                 __atomic64_add_const(i, &v->counter);
 141                 return;
 142         }
 143 #endif
 144         __atomic64_add(i, &v->counter);
 145 }
 146
 147 #define atomic64_xchg(v, new) (xchg(&((v)->counter), new))
 148
 149 static inline long atomic64_cmpxchg(atomic64_t *v, long old, long new)
 150 {
 151         return __atomic64_cmpxchg(&v->counter, old, new);
 152 }
 153
 154 #define ATOMIC64_OPS(op)                                                \
 155 static inline void atomic64_##op(long i, atomic64_t *v)                 \
 156 {                                                                       \
 157         __atomic64_##op(i, &v->counter);                                \
 158 }                                                                       \
 159 static inline long atomic64_fetch_##op(long i, atomic64_t *v)           \
 160 {                                                                       \
 161         return __atomic64_##op##_barrier(i, &v->counter);               \
 162 }
 163
 164 ATOMIC64_OPS(and)
 165 ATOMIC64_OPS(or)
 166 ATOMIC64_OPS(xor)
 167
 168 #undef ATOMIC64_OPS
 169
 170 static inline int atomic64_add_unless(atomic64_t *v, long i, long u)
 171 {
 172         long c, old;
 173
 174         c = atomic64_read(v);
 175         for (;;) {
 176                 if (unlikely(c == u))
 177                         break;
 178                 old = atomic64_cmpxchg(v, c, c + i);
 179                 if (likely(old == c))
 180                         break;
 181                 c = old;
 182         }
 183         return c != u;
 184 }
 185
 186 static inline long atomic64_dec_if_positive(atomic64_t *v)
 187 {
 188         long c, old, dec;
 189
 190         c = atomic64_read(v);
 191         for (;;) {
 192                 dec = c - 1;
 193                 if (unlikely(dec < 0))
 194                         break;
 195                 old = atomic64_cmpxchg((v), c, dec);
 196                 if (likely(old == c))
 197                         break;
 198                 c = old;
 199         }
 200         return dec;
 201 }
 202
 203 #define atomic64_add_negative(_i, _v)   (atomic64_add_return(_i, _v) < 0)
 204 #define atomic64_inc(_v)                atomic64_add(1, _v)
 205 #define atomic64_inc_return(_v)         atomic64_add_return(1, _v)
 206 #define atomic64_inc_and_test(_v)       (atomic64_add_return(1, _v) == 0)
 207 #define atomic64_sub_return(_i, _v)     atomic64_add_return(-(long)(_i), _v)
 208 #define atomic64_fetch_sub(_i, _v)      atomic64_fetch_add(-(long)(_i), _v)
 209 #define atomic64_sub(_i, _v)            atomic64_add(-(long)(_i), _v)
 210 #define atomic64_sub_and_test(_i, _v)   (atomic64_sub_return(_i, _v) == 0)
 211 #define atomic64_dec(_v)                atomic64_sub(1, _v)
 212 #define atomic64_dec_return(_v)         atomic64_sub_return(1, _v)
 213 #define atomic64_dec_and_test(_v)       (atomic64_sub_return(1, _v) == 0)
 214 #define atomic64_inc_not_zero(v)        atomic64_add_unless((v), 1, 0)
 215
 216 #endif /* __ARCH_S390_ATOMIC__  */