arch/arm/include/asm/spinlock.h

   1 #ifndef __ASM_SPINLOCK_H
   2 #define __ASM_SPINLOCK_H
   3
   4 #if __LINUX_ARM_ARCH__ < 6
   5 #error SMP not supported on pre-ARMv6 CPUs
   6 #endif
   7
   8 #include <asm/processor.h>
   9
  10 /*
  11  * sev and wfe are ARMv6K extensions.  Uniprocessor ARMv6 may not have the K
  12  * extensions, so when running on UP, we have to patch these instructions away.
  13  */
  14 #define ALT_SMP(smp, up)                                        \
  15         "9998:  " smp "\n"                                      \
  16         "       .pushsection \".alt.smp.init\", \"a\"\n"        \
  17         "       .long   9998b\n"                                \
  18         "       " up "\n"                                       \
  19         "       .popsection\n"
  20
  21 #ifdef CONFIG_THUMB2_KERNEL
  22 #define SEV             ALT_SMP("sev.w", "nop.w")
  23 /*
  24  * For Thumb-2, special care is needed to ensure that the conditional WFE
  25  * instruction really does assemble to exactly 4 bytes (as required by
  26  * the SMP_ON_UP fixup code).   By itself "wfene" might cause the
  27  * assembler to insert a extra (16-bit) IT instruction, depending on the
  28  * presence or absence of neighbouring conditional instructions.
  29  *
  30  * To avoid this unpredictableness, an approprite IT is inserted explicitly:
  31  * the assembler won't change IT instructions which are explicitly present
  32  * in the input.
  33  */
  34 #define WFE(cond)       ALT_SMP(                \
  35         "it " cond "\n\t"                       \
  36         "wfe" cond ".n",                        \
  37                                                 \
  38         "nop.w"                                 \
  39 )
  40 #else
  41 #define SEV             ALT_SMP("sev", "nop")
  42 #define WFE(cond)       ALT_SMP("wfe" cond, "nop")
  43 #endif
  44
  45 static inline void dsb_sev(void)
  46 {
  47 #if __LINUX_ARM_ARCH__ >= 7
  48         __asm__ __volatile__ (
  49                 "dsb ishst\n"
  50                 SEV
  51         );
  52 #else
  53         __asm__ __volatile__ (
  54                 "mcr p15, 0, %0, c7, c10, 4\n"
  55                 SEV
  56                 : : "r" (0)
  57         );
  58 #endif
  59 }
  60
  61 /*
  62  * ARMv6 ticket-based spin-locking.
  63  *
  64  * A memory barrier is required after we get a lock, and before we
  65  * release it, because V6 CPUs are assumed to have weakly ordered
  66  * memory.
  67  */
  68
  69 #define arch_spin_unlock_wait(lock) \
  70         do { while (arch_spin_is_locked(lock)) cpu_relax(); } while (0)
  71
  72 #define arch_spin_lock_flags(lock, flags) arch_spin_lock(lock)
  73
  74 static inline void arch_spin_lock(arch_spinlock_t *lock)
  75 {
  76         unsigned long tmp;
  77         u32 newval;
  78         arch_spinlock_t lockval;
  79
  80         __asm__ __volatile__(
  81 "1:     ldrex   %0, [%3]\n"
  82 "       add     %1, %0, %4\n"
  83 "       strex   %2, %1, [%3]\n"
  84 "       teq     %2, #0\n"
  85 "       bne     1b"
  86         : "=&r" (lockval), "=&r" (newval), "=&r" (tmp)
  87         : "r" (&lock->slock), "I" (1 << TICKET_SHIFT)
  88         : "cc");
  89
  90         while (lockval.tickets.next != lockval.tickets.owner) {
  91                 wfe();
  92                 lockval.tickets.owner = ACCESS_ONCE(lock->tickets.owner);
  93         }
  94
  95         smp_mb();
  96 }
  97
  98 static inline int arch_spin_trylock(arch_spinlock_t *lock)
  99 {
 100         unsigned long contended, res;
 101         u32 slock;
 102
 103         do {
 104                 __asm__ __volatile__(
 105                 "       ldrex   %0, [%3]\n"
 106                 "       mov     %2, #0\n"
 107                 "       subs    %1, %0, %0, ror #16\n"
 108                 "       addeq   %0, %0, %4\n"
 109                 "       strexeq %2, %0, [%3]"
 110                 : "=&r" (slock), "=&r" (contended), "=&r" (res)
 111                 : "r" (&lock->slock), "I" (1 << TICKET_SHIFT)
 112                 : "cc");
 113         } while (res);
 114
 115         if (!contended) {
 116                 smp_mb();
 117                 return 1;
 118         } else {
 119                 return 0;
 120         }
 121 }
 122
 123 static inline void arch_spin_unlock(arch_spinlock_t *lock)
 124 {
 125         smp_mb();
 126         lock->tickets.owner++;
 127         dsb_sev();
 128 }
 129
 130 static inline int arch_spin_value_unlocked(arch_spinlock_t lock)
 131 {
 132         return lock.tickets.owner == lock.tickets.next;
 133 }
 134
 135 static inline int arch_spin_is_locked(arch_spinlock_t *lock)
 136 {
 137         return !arch_spin_value_unlocked(ACCESS_ONCE(*lock));
 138 }
 139
 140 static inline int arch_spin_is_contended(arch_spinlock_t *lock)
 141 {
 142         struct __raw_tickets tickets = ACCESS_ONCE(lock->tickets);
 143         return (tickets.next - tickets.owner) > 1;
 144 }
 145 #define arch_spin_is_contended  arch_spin_is_contended
 146
 147 /*
 148  * RWLOCKS
 149  *
 150  *
 151  * Write locks are easy - we just set bit 31.  When unlocking, we can
 152  * just write zero since the lock is exclusively held.
 153  */
 154
 155 static inline void arch_write_lock(arch_rwlock_t *rw)
 156 {
 157         unsigned long tmp;
 158
 159         __asm__ __volatile__(
 160 "1:     ldrex   %0, [%1]\n"
 161 "       teq     %0, #0\n"
 162         WFE("ne")
 163 "       strexeq %0, %2, [%1]\n"
 164 "       teq     %0, #0\n"
 165 "       bne     1b"
 166         : "=&r" (tmp)
 167         : "r" (&rw->lock), "r" (0x80000000)
 168         : "cc");
 169
 170         smp_mb();
 171 }
 172
 173 static inline int arch_write_trylock(arch_rwlock_t *rw)
 174 {
 175         unsigned long contended, res;
 176
 177         do {
 178                 __asm__ __volatile__(
 179                 "       ldrex   %0, [%2]\n"
 180                 "       mov     %1, #0\n"
 181                 "       teq     %0, #0\n"
 182                 "       strexeq %1, %3, [%2]"
 183                 : "=&r" (contended), "=&r" (res)
 184                 : "r" (&rw->lock), "r" (0x80000000)
 185                 : "cc");
 186         } while (res);
 187
 188         if (!contended) {
 189                 smp_mb();
 190                 return 1;
 191         } else {
 192                 return 0;
 193         }
 194 }
 195
 196 static inline void arch_write_unlock(arch_rwlock_t *rw)
 197 {
 198         smp_mb();
 199
 200         __asm__ __volatile__(
 201         "str    %1, [%0]\n"
 202         :
 203         : "r" (&rw->lock), "r" (0)
 204         : "cc");
 205
 206         dsb_sev();
 207 }
 208
 209 /* write_can_lock - would write_trylock() succeed? */
 210 #define arch_write_can_lock(x)          ((x)->lock == 0)
 211
 212 /*
 213  * Read locks are a bit more hairy:
 214  *  - Exclusively load the lock value.
 215  *  - Increment it.
 216  *  - Store new lock value if positive, and we still own this location.
 217  *    If the value is negative, we've already failed.
 218  *  - If we failed to store the value, we want a negative result.
 219  *  - If we failed, try again.
 220  * Unlocking is similarly hairy.  We may have multiple read locks
 221  * currently active.  However, we know we won't have any write
 222  * locks.
 223  */
 224 static inline void arch_read_lock(arch_rwlock_t *rw)
 225 {
 226         unsigned long tmp, tmp2;
 227
 228         __asm__ __volatile__(
 229 "1:     ldrex   %0, [%2]\n"
 230 "       adds    %0, %0, #1\n"
 231 "       strexpl %1, %0, [%2]\n"
 232         WFE("mi")
 233 "       rsbpls  %0, %1, #0\n"
 234 "       bmi     1b"
 235         : "=&r" (tmp), "=&r" (tmp2)
 236         : "r" (&rw->lock)
 237         : "cc");
 238
 239         smp_mb();
 240 }
 241
 242 static inline void arch_read_unlock(arch_rwlock_t *rw)
 243 {
 244         unsigned long tmp, tmp2;
 245
 246         smp_mb();
 247
 248         __asm__ __volatile__(
 249 "1:     ldrex   %0, [%2]\n"
 250 "       sub     %0, %0, #1\n"
 251 "       strex   %1, %0, [%2]\n"
 252 "       teq     %1, #0\n"
 253 "       bne     1b"
 254         : "=&r" (tmp), "=&r" (tmp2)
 255         : "r" (&rw->lock)
 256         : "cc");
 257
 258         if (tmp == 0)
 259                 dsb_sev();
 260 }
 261
 262 static inline int arch_read_trylock(arch_rwlock_t *rw)
 263 {
 264         unsigned long contended, res;
 265
 266         do {
 267                 __asm__ __volatile__(
 268                 "       ldrex   %0, [%2]\n"
 269                 "       mov     %1, #0\n"
 270                 "       adds    %0, %0, #1\n"
 271                 "       strexpl %1, %0, [%2]"
 272                 : "=&r" (contended), "=&r" (res)
 273                 : "r" (&rw->lock)
 274                 : "cc");
 275         } while (res);
 276
 277         /* If the lock is negative, then it is already held for write. */
 278         if (contended < 0x80000000) {
 279                 smp_mb();
 280                 return 1;
 281         } else {
 282                 return 0;
 283         }
 284 }
 285
 286 /* read_can_lock - would read_trylock() succeed? */
 287 #define arch_read_can_lock(x)           ((x)->lock < 0x80000000)
 288
 289 #define arch_read_lock_flags(lock, flags) arch_read_lock(lock)
 290 #define arch_write_lock_flags(lock, flags) arch_write_lock(lock)
 291
 292 #define arch_spin_relax(lock)   cpu_relax()
 293 #define arch_read_relax(lock)   cpu_relax()
 294 #define arch_write_relax(lock)  cpu_relax()
 295
 296 #endif /* __ASM_SPINLOCK_H */