summaryrefslogtreecommitdiffstats
path: root/test/unit/atomic.c
blob: eb6136c79403937b61ecd41b17bcc7abb68191d7 (plain)
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#include "test/jemalloc_test.h"

#define	TEST_STRUCT(p, t)						\
struct p##_test_s {							\
	t	accum0;							\
	t	x;							\
};									\
typedef struct p##_test_s p##_test_t;

#define	TEST_BODY(p, t, PRI) do {					\
	const p##_test_t tests[] = {					\
		{-1, -1},						\
		{-1,  0},						\
		{-1,  1},						\
									\
		{ 0, -1},						\
		{ 0,  0},						\
		{ 0,  1},						\
									\
		{ 1, -1},						\
		{ 1,  0},						\
		{ 1,  1},						\
									\
		{0, -(1 << 22)},					\
		{0, (1 << 22)},						\
		{(1 << 22), -(1 << 22)},				\
		{(1 << 22), (1 << 22)}					\
	};								\
	unsigned i;							\
									\
	for (i = 0; i < sizeof(tests)/sizeof(p##_test_t); i++) {	\
		t accum = tests[i].accum0;				\
		assert_u64_eq(atomic_read_##p(&accum), tests[i].accum0,	\
		    "i=%u", i);						\
		assert_u64_eq(atomic_add_##p(&accum, tests[i].x),	\
		    tests[i].accum0 + tests[i].x,			\
		    "i=%u, accum=%#"PRI", x=%#"PRI,			\
		    i, tests[i].accum0, tests[i].x);			\
		assert_u64_eq(atomic_read_##p(&accum), accum,		\
		    "i=%u", i);						\
									\
		accum = tests[i].accum0;				\
		assert_u64_eq(atomic_sub_##p(&accum, tests[i].x),	\
		    tests[i].accum0 - tests[i].x,			\
		    "i=%u, accum=%#"PRI", x=%#"PRI,			\
		    i, tests[i].accum0, tests[i].x);			\
		assert_u64_eq(atomic_read_##p(&accum), accum,		\
		    "i=%u", i);						\
	}								\
} while (0)

TEST_STRUCT(uint64, uint64_t)
TEST_BEGIN(test_atomic_uint64)
{

#if !(LG_SIZEOF_PTR == 3 || LG_SIZEOF_INT == 3)
	test_skip("64-bit atomic operations not supported");
#else
	TEST_BODY(uint64, uint64_t, PRIx64);
#endif
}
TEST_END

TEST_STRUCT(uint32, uint32_t)
TEST_BEGIN(test_atomic_uint32)
{

	TEST_BODY(uint32, uint32_t, PRIx32);
}
TEST_END

TEST_STRUCT(z, size_t)
TEST_BEGIN(test_atomic_z)
{

	TEST_BODY(z, size_t, "zx");
}
TEST_END

TEST_STRUCT(u, unsigned)
TEST_BEGIN(test_atomic_u)
{

	TEST_BODY(u, unsigned, "x");
}
TEST_END

int
main(void)
{

	return (test(
	    test_atomic_uint64,
	    test_atomic_uint32,
	    test_atomic_z,
	    test_atomic_u));
}