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