atomic.h 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. #ifndef _INTERNAL_ATOMIC_H
  2. #define _INTERNAL_ATOMIC_H
  3. #include <stdint.h>
  4. #include <endian.h>
  5. static inline int a_ctz_l(unsigned long x)
  6. {
  7. static const char debruijn32[32] = {
  8. 0, 1, 23, 2, 29, 24, 19, 3, 30, 27, 25, 11, 20, 8, 4, 13,
  9. 31, 22, 28, 18, 26, 10, 7, 12, 21, 17, 9, 6, 16, 5, 15, 14
  10. };
  11. return debruijn32[(x&-x)*0x076be629 >> 27];
  12. }
  13. static inline int a_ctz_64(uint64_t x)
  14. {
  15. uint32_t y = x;
  16. if (!y) {
  17. y = x>>32;
  18. return 32 + a_ctz_l(y);
  19. }
  20. return a_ctz_l(y);
  21. }
  22. static inline int a_cas(volatile int *p, int t, int s)
  23. {
  24. __asm__("\n"
  25. " sync\n"
  26. "1: lwarx %0, 0, %4\n"
  27. " cmpw %0, %2\n"
  28. " bne 1f\n"
  29. " stwcx. %3, 0, %4\n"
  30. " bne- 1b\n"
  31. " isync\n"
  32. "1: \n"
  33. : "=&r"(t), "+m"(*p) : "r"(t), "r"(s), "r"(p) : "cc", "memory" );
  34. return t;
  35. }
  36. static inline void *a_cas_p(volatile void *p, void *t, void *s)
  37. {
  38. return (void *)a_cas(p, (int)t, (int)s);
  39. }
  40. static inline long a_cas_l(volatile void *p, long t, long s)
  41. {
  42. return a_cas(p, t, s);
  43. }
  44. static inline int a_swap(volatile int *x, int v)
  45. {
  46. int old;
  47. do old = *x;
  48. while (a_cas(x, old, v) != old);
  49. return old;
  50. }
  51. static inline int a_fetch_add(volatile int *x, int v)
  52. {
  53. int old;
  54. do old = *x;
  55. while (a_cas(x, old, old+v) != old);
  56. return old;
  57. }
  58. static inline void a_inc(volatile int *x)
  59. {
  60. a_fetch_add(x, 1);
  61. }
  62. static inline void a_dec(volatile int *x)
  63. {
  64. a_fetch_add(x, -1);
  65. }
  66. static inline void a_store(volatile int *p, int x)
  67. {
  68. __asm__ __volatile__ ("\n"
  69. " sync\n"
  70. " stw %1, %0\n"
  71. " isync\n"
  72. : "=m"(*p) : "r"(x) : "memory" );
  73. }
  74. static inline void a_spin()
  75. {
  76. }
  77. static inline void a_crash()
  78. {
  79. *(volatile char *)0=0;
  80. }
  81. static inline void a_and(volatile int *p, int v)
  82. {
  83. int old;
  84. do old = *p;
  85. while (a_cas(p, old, old&v) != old);
  86. }
  87. static inline void a_or(volatile int *p, int v)
  88. {
  89. int old;
  90. do old = *p;
  91. while (a_cas(p, old, old|v) != old);
  92. }
  93. static inline void a_or_l(volatile void *p, long v)
  94. {
  95. a_or(p, v);
  96. }
  97. static inline void a_and_64(volatile uint64_t *p, uint64_t v)
  98. {
  99. union { uint64_t v; uint32_t r[2]; } u = { v };
  100. a_and((int *)p, u.r[0]);
  101. a_and((int *)p+1, u.r[1]);
  102. }
  103. static inline void a_or_64(volatile uint64_t *p, uint64_t v)
  104. {
  105. union { uint64_t v; uint32_t r[2]; } u = { v };
  106. a_or((int *)p, u.r[0]);
  107. a_or((int *)p+1, u.r[1]);
  108. }
  109. #endif