cmpxchg.h 6.9 KB

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