]>
Commit | Line | Data |
---|---|---|
1a4d82fc JJ |
1 | #include "test/jemalloc_test.h" |
2 | ||
3 | #define THREAD_DATA 0x72b65c10 | |
4 | ||
5 | typedef unsigned int data_t; | |
6 | ||
7 | static bool data_cleanup_executed; | |
8 | ||
54a0048b | 9 | malloc_tsd_types(data_, data_t) |
1a4d82fc JJ |
10 | malloc_tsd_protos(, data_, data_t) |
11 | ||
12 | void | |
13 | data_cleanup(void *arg) | |
14 | { | |
15 | data_t *data = (data_t *)arg; | |
16 | ||
17 | if (!data_cleanup_executed) { | |
18 | assert_x_eq(*data, THREAD_DATA, | |
19 | "Argument passed into cleanup function should match tsd " | |
20 | "value"); | |
21 | } | |
22 | data_cleanup_executed = true; | |
23 | ||
24 | /* | |
25 | * Allocate during cleanup for two rounds, in order to assure that | |
26 | * jemalloc's internal tsd reinitialization happens. | |
27 | */ | |
28 | switch (*data) { | |
29 | case THREAD_DATA: | |
30 | *data = 1; | |
31 | data_tsd_set(data); | |
32 | break; | |
33 | case 1: | |
34 | *data = 2; | |
35 | data_tsd_set(data); | |
36 | break; | |
37 | case 2: | |
38 | return; | |
39 | default: | |
40 | not_reached(); | |
41 | } | |
42 | ||
43 | { | |
44 | void *p = mallocx(1, 0); | |
45 | assert_ptr_not_null(p, "Unexpeced mallocx() failure"); | |
46 | dallocx(p, 0); | |
47 | } | |
48 | } | |
49 | ||
50 | malloc_tsd_externs(data_, data_t) | |
51 | #define DATA_INIT 0x12345678 | |
52 | malloc_tsd_data(, data_, data_t, DATA_INIT) | |
53 | malloc_tsd_funcs(, data_, data_t, DATA_INIT, data_cleanup) | |
54 | ||
55 | static void * | |
56 | thd_start(void *arg) | |
57 | { | |
58 | data_t d = (data_t)(uintptr_t)arg; | |
54a0048b SL |
59 | void *p; |
60 | ||
1a4d82fc JJ |
61 | assert_x_eq(*data_tsd_get(), DATA_INIT, |
62 | "Initial tsd get should return initialization value"); | |
63 | ||
54a0048b SL |
64 | p = malloc(1); |
65 | assert_ptr_not_null(p, "Unexpected malloc() failure"); | |
66 | ||
1a4d82fc JJ |
67 | data_tsd_set(&d); |
68 | assert_x_eq(*data_tsd_get(), d, | |
69 | "After tsd set, tsd get should return value that was set"); | |
70 | ||
71 | d = 0; | |
72 | assert_x_eq(*data_tsd_get(), (data_t)(uintptr_t)arg, | |
73 | "Resetting local data should have no effect on tsd"); | |
74 | ||
54a0048b | 75 | free(p); |
1a4d82fc JJ |
76 | return (NULL); |
77 | } | |
78 | ||
79 | TEST_BEGIN(test_tsd_main_thread) | |
80 | { | |
81 | ||
82 | thd_start((void *) 0xa5f3e329); | |
83 | } | |
84 | TEST_END | |
85 | ||
86 | TEST_BEGIN(test_tsd_sub_thread) | |
87 | { | |
88 | thd_t thd; | |
89 | ||
90 | data_cleanup_executed = false; | |
91 | thd_create(&thd, thd_start, (void *)THREAD_DATA); | |
92 | thd_join(thd, NULL); | |
93 | assert_true(data_cleanup_executed, | |
94 | "Cleanup function should have executed"); | |
95 | } | |
96 | TEST_END | |
97 | ||
98 | int | |
99 | main(void) | |
100 | { | |
101 | ||
102 | data_tsd_boot(); | |
103 | ||
104 | return (test( | |
105 | test_tsd_main_thread, | |
106 | test_tsd_sub_thread)); | |
107 | } |