1 #include "test/jemalloc_test.h"
2 
3 #define NTHREADS	2
4 #define NINCRS		2000000
5 
TEST_BEGIN(test_mtx_basic)6 TEST_BEGIN(test_mtx_basic) {
7 	mtx_t mtx;
8 
9 	assert_false(mtx_init(&mtx), "Unexpected mtx_init() failure");
10 	mtx_lock(&mtx);
11 	mtx_unlock(&mtx);
12 	mtx_fini(&mtx);
13 }
14 TEST_END
15 
16 typedef struct {
17 	mtx_t		mtx;
18 	unsigned	x;
19 } thd_start_arg_t;
20 
21 static void *
thd_start(void * varg)22 thd_start(void *varg) {
23 	thd_start_arg_t *arg = (thd_start_arg_t *)varg;
24 	unsigned i;
25 
26 	for (i = 0; i < NINCRS; i++) {
27 		mtx_lock(&arg->mtx);
28 		arg->x++;
29 		mtx_unlock(&arg->mtx);
30 	}
31 	return NULL;
32 }
33 
TEST_BEGIN(test_mtx_race)34 TEST_BEGIN(test_mtx_race) {
35 	thd_start_arg_t arg;
36 	thd_t thds[NTHREADS];
37 	unsigned i;
38 
39 	assert_false(mtx_init(&arg.mtx), "Unexpected mtx_init() failure");
40 	arg.x = 0;
41 	for (i = 0; i < NTHREADS; i++) {
42 		thd_create(&thds[i], thd_start, (void *)&arg);
43 	}
44 	for (i = 0; i < NTHREADS; i++) {
45 		thd_join(thds[i], NULL);
46 	}
47 	assert_u_eq(arg.x, NTHREADS * NINCRS,
48 	    "Race-related counter corruption");
49 }
50 TEST_END
51 
52 int
main(void)53 main(void) {
54 	return test(
55 	    test_mtx_basic,
56 	    test_mtx_race);
57 }
58