tests/test_uatomic.c

   1 #include <stdio.h>
   2 #include <assert.h>
   3 #include <urcu/uatomic.h>
   4
   5 struct testvals {
   6         unsigned char c;
   7         unsigned short s;
   8         unsigned int i;
   9         unsigned long l;
  10 };
  11
  12 static struct testvals vals;
  13
  14 #define do_test(ptr)                            \
  15 do {                                            \
  16         __typeof__(*(ptr)) v;                   \
  17                                                 \
  18         uatomic_add(ptr, 10);                   \
  19         assert(uatomic_read(ptr) == 10);                \
  20         uatomic_add(ptr, -11UL);                        \
  21         assert(uatomic_read(ptr) == (__typeof__(*(ptr)))-1UL);  \
  22         v = uatomic_cmpxchg(ptr, -1UL, 22);             \
  23         assert(uatomic_read(ptr) == 22);                \
  24         assert(v == (__typeof__(*(ptr)))-1UL);  \
  25         v = uatomic_cmpxchg(ptr, 33, 44);               \
  26         assert(uatomic_read(ptr) == 22);                \
  27         assert(v == 22);                        \
  28         v = uatomic_xchg(ptr, 55);                      \
  29         assert(uatomic_read(ptr) == 55);                \
  30         assert(v == 22);                        \
  31         uatomic_set(ptr, 22);                   \
  32         uatomic_inc(ptr);                       \
  33         assert(uatomic_read(ptr) == 23);                \
  34         uatomic_dec(ptr);                       \
  35         assert(uatomic_read(ptr) == 22);                \
  36         v = uatomic_add_return(ptr, 74);        \
  37         assert(v == 96);                        \
  38         assert(uatomic_read(ptr) == 96);        \
  39         uatomic_or(ptr, 58);                    \
  40         assert(uatomic_read(ptr) == 122);       \
  41         v = uatomic_sub_return(ptr, 1);         \
  42         assert(v == 121);                       \
  43         uatomic_sub(ptr, (unsigned int) 2);     \
  44         assert(uatomic_read(ptr) == 119);       \
  45         uatomic_inc(ptr);                       \
  46         uatomic_inc(ptr);                       \
  47         assert(uatomic_read(ptr) == 121);       \
  48         uatomic_and(ptr, 129);                  \
  49         assert(uatomic_read(ptr) == 1);         \
  50 } while (0)
  51
  52 int main(int argc, char **argv)
  53 {
  54 #ifdef UATOMIC_HAS_ATOMIC_BYTE
  55         do_test(&vals.c);
  56 #endif
  57 #ifdef UATOMIC_HAS_ATOMIC_SHORT
  58         do_test(&vals.s);
  59 #endif
  60         do_test(&vals.i);
  61         do_test(&vals.l);
  62         printf("Atomic ops test OK\n");
  63
  64         return 0;
  65 }