atomic.h 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. #ifndef _INTERNAL_ATOMIC_H
  2. #define _INTERNAL_ATOMIC_H
  3. #include <stdint.h>
  4. static inline int a_ctz_l(unsigned long x)
  5. {
  6. static const char debruijn32[32] = {
  7. 0, 1, 23, 2, 29, 24, 19, 3, 30, 27, 25, 11, 20, 8, 4, 13,
  8. 31, 22, 28, 18, 26, 10, 7, 12, 21, 17, 9, 6, 16, 5, 15, 14
  9. };
  10. return debruijn32[(x&-x)*0x076be629 >> 27];
  11. }
  12. static inline int a_ctz_64(uint64_t x)
  13. {
  14. uint32_t y = x;
  15. if (!y) {
  16. y = x>>32;
  17. return 32 + a_ctz_l(y);
  18. }
  19. return a_ctz_l(y);
  20. }
  21. #if __ARM_ARCH_7A__ || __ARM_ARCH_7R__ || __ARM_ARCH >= 7
  22. static inline void a_barrier()
  23. {
  24. __asm__ __volatile__("dmb ish");
  25. }
  26. static inline int a_cas(volatile int *p, int t, int s)
  27. {
  28. int old;
  29. __asm__ __volatile__(
  30. " dmb ish\n"
  31. "1: ldrex %0,%3\n"
  32. " cmp %0,%1\n"
  33. " bne 1f\n"
  34. " strex %0,%2,%3\n"
  35. " cmp %0, #0\n"
  36. " bne 1b\n"
  37. " mov %0, %1\n"
  38. "1: dmb ish\n"
  39. : "=&r"(old)
  40. : "r"(t), "r"(s), "Q"(*p)
  41. : "memory", "cc" );
  42. return old;
  43. }
  44. static inline int a_swap(volatile int *x, int v)
  45. {
  46. int old, tmp;
  47. __asm__ __volatile__(
  48. " dmb ish\n"
  49. "1: ldrex %0,%3\n"
  50. " strex %1,%2,%3\n"
  51. " cmp %1, #0\n"
  52. " bne 1b\n"
  53. " dmb ish\n"
  54. : "=&r"(old), "=&r"(tmp)
  55. : "r"(v), "Q"(*x)
  56. : "memory", "cc" );
  57. return old;
  58. }
  59. static inline int a_fetch_add(volatile int *x, int v)
  60. {
  61. int old, tmp;
  62. __asm__ __volatile__(
  63. " dmb ish\n"
  64. "1: ldrex %0,%3\n"
  65. " add %0,%0,%2\n"
  66. " strex %1,%0,%3\n"
  67. " cmp %1, #0\n"
  68. " bne 1b\n"
  69. " dmb ish\n"
  70. : "=&r"(old), "=&r"(tmp)
  71. : "r"(v), "Q"(*x)
  72. : "memory", "cc" );
  73. return old-v;
  74. }
  75. static inline void a_inc(volatile int *x)
  76. {
  77. int tmp, tmp2;
  78. __asm__ __volatile__(
  79. " dmb ish\n"
  80. "1: ldrex %0,%2\n"
  81. " add %0,%0,#1\n"
  82. " strex %1,%0,%2\n"
  83. " cmp %1, #0\n"
  84. " bne 1b\n"
  85. " dmb ish\n"
  86. : "=&r"(tmp), "=&r"(tmp2)
  87. : "Q"(*x)
  88. : "memory", "cc" );
  89. }
  90. static inline void a_dec(volatile int *x)
  91. {
  92. int tmp, tmp2;
  93. __asm__ __volatile__(
  94. " dmb ish\n"
  95. "1: ldrex %0,%2\n"
  96. " sub %0,%0,#1\n"
  97. " strex %1,%0,%2\n"
  98. " cmp %1, #0\n"
  99. " bne 1b\n"
  100. " dmb ish\n"
  101. : "=&r"(tmp), "=&r"(tmp2)
  102. : "Q"(*x)
  103. : "memory", "cc" );
  104. }
  105. static inline void a_and(volatile int *x, int v)
  106. {
  107. int tmp, tmp2;
  108. __asm__ __volatile__(
  109. " dmb ish\n"
  110. "1: ldrex %0,%3\n"
  111. " and %0,%0,%2\n"
  112. " strex %1,%0,%3\n"
  113. " cmp %1, #0\n"
  114. " bne 1b\n"
  115. " dmb ish\n"
  116. : "=&r"(tmp), "=&r"(tmp2)
  117. : "r"(v), "Q"(*x)
  118. : "memory", "cc" );
  119. }
  120. static inline void a_or(volatile int *x, int v)
  121. {
  122. int tmp, tmp2;
  123. __asm__ __volatile__(
  124. " dmb ish\n"
  125. "1: ldrex %0,%3\n"
  126. " orr %0,%0,%2\n"
  127. " strex %1,%0,%3\n"
  128. " cmp %1, #0\n"
  129. " bne 1b\n"
  130. " dmb ish\n"
  131. : "=&r"(tmp), "=&r"(tmp2)
  132. : "r"(v), "Q"(*x)
  133. : "memory", "cc" );
  134. }
  135. static inline void a_store(volatile int *p, int x)
  136. {
  137. __asm__ __volatile__(
  138. " dmb ish\n"
  139. " str %1,%0\n"
  140. " dmb ish\n"
  141. : "=m"(*p)
  142. : "r"(x)
  143. : "memory", "cc" );
  144. }
  145. #else
  146. int __a_cas(int, int, volatile int *) __attribute__((__visibility__("hidden")));
  147. #define __k_cas __a_cas
  148. static inline void a_barrier()
  149. {
  150. __asm__ __volatile__("bl __a_barrier"
  151. : : : "memory", "cc", "ip", "lr" );
  152. }
  153. static inline int a_cas(volatile int *p, int t, int s)
  154. {
  155. int old;
  156. for (;;) {
  157. if (!__k_cas(t, s, p))
  158. return t;
  159. if ((old=*p) != t)
  160. return old;
  161. }
  162. }
  163. static inline int a_swap(volatile int *x, int v)
  164. {
  165. int old;
  166. do old = *x;
  167. while (__k_cas(old, v, x));
  168. return old;
  169. }
  170. static inline int a_fetch_add(volatile int *x, int v)
  171. {
  172. int old;
  173. do old = *x;
  174. while (__k_cas(old, old+v, x));
  175. return old;
  176. }
  177. static inline void a_inc(volatile int *x)
  178. {
  179. a_fetch_add(x, 1);
  180. }
  181. static inline void a_dec(volatile int *x)
  182. {
  183. a_fetch_add(x, -1);
  184. }
  185. static inline void a_store(volatile int *p, int x)
  186. {
  187. a_barrier();
  188. *p = x;
  189. a_barrier();
  190. }
  191. static inline void a_and(volatile int *p, int v)
  192. {
  193. int old;
  194. do old = *p;
  195. while (__k_cas(old, old&v, p));
  196. }
  197. static inline void a_or(volatile int *p, int v)
  198. {
  199. int old;
  200. do old = *p;
  201. while (__k_cas(old, old|v, p));
  202. }
  203. #endif
  204. static inline void *a_cas_p(volatile void *p, void *t, void *s)
  205. {
  206. return (void *)a_cas(p, (int)t, (int)s);
  207. }
  208. #define a_spin a_barrier
  209. static inline void a_crash()
  210. {
  211. *(volatile char *)0=0;
  212. }
  213. static inline void a_or_l(volatile void *p, long v)
  214. {
  215. a_or(p, v);
  216. }
  217. static inline void a_and_64(volatile uint64_t *p, uint64_t v)
  218. {
  219. union { uint64_t v; uint32_t r[2]; } u = { v };
  220. a_and((int *)p, u.r[0]);
  221. a_and((int *)p+1, u.r[1]);
  222. }
  223. static inline void a_or_64(volatile uint64_t *p, uint64_t v)
  224. {
  225. union { uint64_t v; uint32_t r[2]; } u = { v };
  226. a_or((int *)p, u.r[0]);
  227. a_or((int *)p+1, u.r[1]);
  228. }
  229. #endif