1// RUN: %clang_dfsan -g3 -DDATA_BYTES=3 %s -fno-exceptions %libatomic -o %t && %run %t
2// RUN: %clang_dfsan -g3 -DDATA_BYTES=3 -DORIGIN_TRACKING -mllvm -dfsan-track-origins=1 %s -fno-exceptions %libatomic -o %t && %run %t
3// RUN: %clang_dfsan -g3 -DDATA_BYTES=32 %s -fno-exceptions %libatomic -o %t && %run %t
4// RUN: %clang_dfsan -g3 -DDATA_BYTES=32 -DORIGIN_TRACKING -mllvm -dfsan-track-origins=1 %s -fno-exceptions %libatomic -o %t && %run %t
5
6#include <assert.h>
7#include <sanitizer/dfsan_interface.h>
8#include <stdatomic.h>
9
10typedef struct __attribute((packed)) {
11 uint8_t val[DATA_BYTES];
12} idata;
13
14void test_idata_load() {
15 idata dest = {-1};
16 idata init = {0};
17
18 dfsan_label i_label = 2;
19 dfsan_set_label(label: i_label, addr: &init, size: sizeof(init));
20
21 __atomic_load(&init, &dest, __ATOMIC_RELAXED);
22
23 dfsan_label read_label = dfsan_read_label(addr: &dest, size: sizeof(dest));
24 assert(read_label == i_label);
25#ifdef ORIGIN_TRACKING
26 dfsan_origin read_origin =
27 dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
28 assert(read_origin != 0);
29#endif
30}
31
32void test_idata_store() {
33 idata dest = {-1};
34 idata init = {0};
35
36 dfsan_label i_label = 2;
37 dfsan_set_label(label: i_label, addr: &init, size: sizeof(init));
38
39 __atomic_store(&init, &dest, __ATOMIC_RELAXED);
40
41 dfsan_label read_label = dfsan_read_label(addr: &dest, size: sizeof(dest));
42 assert(read_label == i_label);
43#ifdef ORIGIN_TRACKING
44 dfsan_origin read_origin =
45 dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
46 assert(read_origin != 0);
47#endif
48}
49
50void test_idata_exchange() {
51 idata target = {-1};
52 idata init = {0};
53 idata dest = {3};
54
55 dfsan_label i_label = 1;
56 dfsan_set_label(label: i_label, addr: &init, size: sizeof(init));
57 dfsan_label j_label = 2;
58 dfsan_set_label(label: j_label, addr: &target, size: sizeof(target));
59
60 dfsan_label dest0_label = dfsan_read_label(addr: &dest, size: sizeof(dest));
61 assert(dest0_label == 0);
62#ifdef ORIGIN_TRACKING
63 dfsan_origin dest0_origin =
64 dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
65 assert(dest0_origin == 0);
66#endif
67
68 __atomic_exchange(&target, &init, &dest, __ATOMIC_RELAXED);
69
70 dfsan_label dest_label = dfsan_read_label(addr: &dest, size: sizeof(dest));
71 assert(dest_label == j_label);
72#ifdef ORIGIN_TRACKING
73 dfsan_origin dest_origin =
74 dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
75 assert(dest_origin != 0);
76#endif
77
78 dfsan_label target_label = dfsan_read_label(addr: &target, size: sizeof(target));
79 assert(target_label == i_label);
80#ifdef ORIGIN_TRACKING
81 dfsan_origin target_origin =
82 dfsan_read_origin_of_first_taint(&target, sizeof(target));
83 assert(target_origin != 0);
84#endif
85}
86
87void test_idata_cmp_exchange_1() {
88 idata target = {0};
89 idata expected = {0}; // Target matches expected
90 idata desired = {3};
91
92 dfsan_label i_label = 1;
93 dfsan_set_label(label: i_label, addr: &expected, size: sizeof(expected));
94 dfsan_label j_label = 2;
95 dfsan_set_label(label: j_label, addr: &target, size: sizeof(target));
96 dfsan_label k_label = 4;
97 dfsan_set_label(label: k_label, addr: &desired, size: sizeof(desired));
98
99 int r =
100 __atomic_compare_exchange(&target, &expected, &desired, /*weak=false*/ 0,
101 __ATOMIC_RELAXED, __ATOMIC_RELAXED);
102 // Target matches expected => true
103 assert(r);
104
105 // Copy desired to target.
106 dfsan_label target_label = dfsan_read_label(addr: &target, size: sizeof(target));
107 assert(target_label == k_label);
108#ifdef ORIGIN_TRACKING
109 dfsan_origin target_origin =
110 dfsan_read_origin_of_first_taint(&target, sizeof(target));
111 assert(target_origin != 0);
112#endif
113}
114
115void test_idata_cmp_exchange_2() {
116 idata target = {0};
117 idata expected = {-1}; // Target does not match expected
118 idata desired = {3};
119
120 dfsan_label i_label = 1;
121 dfsan_set_label(label: i_label, addr: &expected, size: sizeof(expected));
122 dfsan_label j_label = 2;
123 dfsan_set_label(label: j_label, addr: &target, size: sizeof(target));
124 dfsan_label k_label = 4;
125 dfsan_set_label(label: k_label, addr: &desired, size: sizeof(desired));
126
127 int r =
128 __atomic_compare_exchange(&target, &expected, &desired, /*weak=false*/ 0,
129 __ATOMIC_RELAXED, __ATOMIC_RELAXED);
130 // Target does not match expected => false
131 assert(!r);
132
133 // Copy target to expected
134 dfsan_label expected_label = dfsan_read_label(addr: &expected, size: sizeof(expected));
135 assert(expected_label == j_label);
136#ifdef ORIGIN_TRACKING
137 dfsan_origin expected_origin =
138 dfsan_read_origin_of_first_taint(&expected, sizeof(expected));
139 assert(expected_origin != 0);
140#endif
141}
142
143int main() {
144 test_idata_load();
145 test_idata_store();
146 test_idata_exchange();
147 test_idata_cmp_exchange_1();
148 test_idata_cmp_exchange_2();
149
150 return 0;
151}
152

source code of compiler-rt/test/dfsan/libatomic.c