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