cmpxchg.h 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. #ifndef ASM_X86_CMPXCHG_H
  2. #define ASM_X86_CMPXCHG_H
  3. #include <linux/compiler.h>
  4. #include <asm/cpufeatures.h>
  5. #include <asm/alternative.h> /* Provides LOCK_PREFIX */
  6. /*
  7. * Non-existant functions to indicate usage errors at link time
  8. * (or compile-time if the compiler implements __compiletime_error().
  9. */
  10. extern void __xchg_wrong_size(void)
  11. __compiletime_error("Bad argument size for xchg");
  12. extern void __cmpxchg_wrong_size(void)
  13. __compiletime_error("Bad argument size for cmpxchg");
  14. extern void __xadd_wrong_size(void)
  15. __compiletime_error("Bad argument size for xadd");
  16. extern void __add_wrong_size(void)
  17. __compiletime_error("Bad argument size for add");
  18. /*
  19. * Constants for operation sizes. On 32-bit, the 64-bit size it set to
  20. * -1 because sizeof will never return -1, thereby making those switch
  21. * case statements guaranteeed dead code which the compiler will
  22. * eliminate, and allowing the "missing symbol in the default case" to
  23. * indicate a usage error.
  24. */
  25. #define __X86_CASE_B 1
  26. #define __X86_CASE_W 2
  27. #define __X86_CASE_L 4
  28. #ifdef CONFIG_64BIT
  29. #define __X86_CASE_Q 8
  30. #else
  31. #define __X86_CASE_Q -1 /* sizeof will never return -1 */
  32. #endif
  33. /*
  34. * An exchange-type operation, which takes a value and a pointer, and
  35. * returns the old value.
  36. */
  37. #define __xchg_op(ptr, arg, op, lock) \
  38. ({ \
  39. __typeof__ (*(ptr)) __ret = (arg); \
  40. switch (sizeof(*(ptr))) { \
  41. case __X86_CASE_B: \
  42. asm volatile (lock #op "b %b0, %1\n" \
  43. : "+q" (__ret), "+m" (*(ptr)) \
  44. : : "memory", "cc"); \
  45. break; \
  46. case __X86_CASE_W: \
  47. asm volatile (lock #op "w %w0, %1\n" \
  48. : "+r" (__ret), "+m" (*(ptr)) \
  49. : : "memory", "cc"); \
  50. break; \
  51. case __X86_CASE_L: \
  52. asm volatile (lock #op "l %0, %1\n" \
  53. : "+r" (__ret), "+m" (*(ptr)) \
  54. : : "memory", "cc"); \
  55. break; \
  56. case __X86_CASE_Q: \
  57. asm volatile (lock #op "q %q0, %1\n" \
  58. : "+r" (__ret), "+m" (*(ptr)) \
  59. : : "memory", "cc"); \
  60. break; \
  61. default: \
  62. __ ## op ## _wrong_size(); \
  63. } \
  64. __ret; \
  65. })
  66. /*
  67. * Note: no "lock" prefix even on SMP: xchg always implies lock anyway.
  68. * Since this is generally used to protect other memory information, we
  69. * use "asm volatile" and "memory" clobbers to prevent gcc from moving
  70. * information around.
  71. */
  72. #define xchg(ptr, v) __xchg_op((ptr), (v), xchg, "")
  73. /*
  74. * Atomic compare and exchange. Compare OLD with MEM, if identical,
  75. * store NEW in MEM. Return the initial value in MEM. Success is
  76. * indicated by comparing RETURN with OLD.
  77. */
  78. #define __raw_cmpxchg(ptr, old, new, size, lock) \
  79. ({ \
  80. __typeof__(*(ptr)) __ret; \
  81. __typeof__(*(ptr)) __old = (old); \
  82. __typeof__(*(ptr)) __new = (new); \
  83. switch (size) { \
  84. case __X86_CASE_B: \
  85. { \
  86. volatile u8 *__ptr = (volatile u8 *)(ptr); \
  87. asm volatile(lock "cmpxchgb %2,%1" \
  88. : "=a" (__ret), "+m" (*__ptr) \
  89. : "q" (__new), "0" (__old) \
  90. : "memory"); \
  91. break; \
  92. } \
  93. case __X86_CASE_W: \
  94. { \
  95. volatile u16 *__ptr = (volatile u16 *)(ptr); \
  96. asm volatile(lock "cmpxchgw %2,%1" \
  97. : "=a" (__ret), "+m" (*__ptr) \
  98. : "r" (__new), "0" (__old) \
  99. : "memory"); \
  100. break; \
  101. } \
  102. case __X86_CASE_L: \
  103. { \
  104. volatile u32 *__ptr = (volatile u32 *)(ptr); \
  105. asm volatile(lock "cmpxchgl %2,%1" \
  106. : "=a" (__ret), "+m" (*__ptr) \
  107. : "r" (__new), "0" (__old) \
  108. : "memory"); \
  109. break; \
  110. } \
  111. case __X86_CASE_Q: \
  112. { \
  113. volatile u64 *__ptr = (volatile u64 *)(ptr); \
  114. asm volatile(lock "cmpxchgq %2,%1" \
  115. : "=a" (__ret), "+m" (*__ptr) \
  116. : "r" (__new), "0" (__old) \
  117. : "memory"); \
  118. break; \
  119. } \
  120. default: \
  121. __cmpxchg_wrong_size(); \
  122. } \
  123. __ret; \
  124. })
  125. #define __cmpxchg(ptr, old, new, size) \
  126. __raw_cmpxchg((ptr), (old), (new), (size), LOCK_PREFIX)
  127. #define __sync_cmpxchg(ptr, old, new, size) \
  128. __raw_cmpxchg((ptr), (old), (new), (size), "lock; ")
  129. #define __cmpxchg_local(ptr, old, new, size) \
  130. __raw_cmpxchg((ptr), (old), (new), (size), "")
  131. #ifdef CONFIG_X86_32
  132. # include <asm/cmpxchg_32.h>
  133. #else
  134. # include <asm/cmpxchg_64.h>
  135. #endif
  136. #define cmpxchg(ptr, old, new) \
  137. __cmpxchg(ptr, old, new, sizeof(*(ptr)))
  138. #define sync_cmpxchg(ptr, old, new) \
  139. __sync_cmpxchg(ptr, old, new, sizeof(*(ptr)))
  140. #define cmpxchg_local(ptr, old, new) \
  141. __cmpxchg_local(ptr, old, new, sizeof(*(ptr)))
  142. /*
  143. * xadd() adds "inc" to "*ptr" and atomically returns the previous
  144. * value of "*ptr".
  145. *
  146. * xadd() is locked when multiple CPUs are online
  147. */
  148. #define __xadd(ptr, inc, lock) __xchg_op((ptr), (inc), xadd, lock)
  149. #define xadd(ptr, inc) __xadd((ptr), (inc), LOCK_PREFIX)
  150. #define __cmpxchg_double(pfx, p1, p2, o1, o2, n1, n2) \
  151. ({ \
  152. bool __ret; \
  153. __typeof__(*(p1)) __old1 = (o1), __new1 = (n1); \
  154. __typeof__(*(p2)) __old2 = (o2), __new2 = (n2); \
  155. BUILD_BUG_ON(sizeof(*(p1)) != sizeof(long)); \
  156. BUILD_BUG_ON(sizeof(*(p2)) != sizeof(long)); \
  157. VM_BUG_ON((unsigned long)(p1) % (2 * sizeof(long))); \
  158. VM_BUG_ON((unsigned long)((p1) + 1) != (unsigned long)(p2)); \
  159. asm volatile(pfx "cmpxchg%c4b %2; sete %0" \
  160. : "=a" (__ret), "+d" (__old2), \
  161. "+m" (*(p1)), "+m" (*(p2)) \
  162. : "i" (2 * sizeof(long)), "a" (__old1), \
  163. "b" (__new1), "c" (__new2)); \
  164. __ret; \
  165. })
  166. #define cmpxchg_double(p1, p2, o1, o2, n1, n2) \
  167. __cmpxchg_double(LOCK_PREFIX, p1, p2, o1, o2, n1, n2)
  168. #define cmpxchg_double_local(p1, p2, o1, o2, n1, n2) \
  169. __cmpxchg_double(, p1, p2, o1, o2, n1, n2)
  170. #endif /* ASM_X86_CMPXCHG_H */