1 // RUN: %clang_dfsan %s -O1 -o %t && %run %t
2 // RUN: %clang_dfsan %s -O0 -DO0 -o %t && %run %t
3 //
4 // REQUIRES: x86_64-target-arch
5
6 #include <assert.h>
7 #include <sanitizer/dfsan_interface.h>
8
9 typedef struct Pair {
10 int i;
11 char *ptr;
12 } Pair;
13
14 __attribute__((noinline))
make_pair(int i,char * ptr)15 Pair make_pair(int i, char *ptr) {
16 Pair pair;
17 pair.i = i;
18 pair.ptr = ptr;
19 return pair;
20 }
21
22 __attribute__((noinline))
copy_pair1(const Pair * pair0)23 Pair copy_pair1(const Pair *pair0) {
24 Pair pair;
25 pair.i = pair0->i;
26 pair.ptr = pair0->ptr;
27 return pair;
28 }
29
30 __attribute__((noinline))
copy_pair2(const Pair pair0)31 Pair copy_pair2(const Pair pair0) {
32 Pair pair;
33 pair.i = pair0.i;
34 pair.ptr = pair0.ptr;
35 return pair;
36 }
37
main(void)38 int main(void) {
39 int i = 1;
40 char *ptr = NULL;
41 dfsan_label i_label = 1;
42 dfsan_label ptr_label = 2;
43 dfsan_set_label(i_label, &i, sizeof(i));
44 dfsan_set_label(ptr_label, &ptr, sizeof(ptr));
45
46 Pair pair1 = make_pair(i, ptr);
47 int i1 = pair1.i;
48 char *ptr1 = pair1.ptr;
49
50 dfsan_label i1_label = dfsan_read_label(&i1, sizeof(i1));
51 dfsan_label ptr1_label = dfsan_read_label(&ptr1, sizeof(ptr1));
52 #if defined(O0)
53 assert(i1_label == (i_label | ptr_label));
54 assert(ptr1_label == (i_label | ptr_label));
55 #else
56 assert(i1_label == i_label);
57 assert(ptr1_label == ptr_label);
58 #endif
59
60 Pair pair2 = copy_pair1(&pair1);
61 int i2 = pair2.i;
62 char *ptr2 = pair2.ptr;
63
64 dfsan_label i2_label = dfsan_read_label(&i2, sizeof(i2));
65 dfsan_label ptr2_label = dfsan_read_label(&ptr2, sizeof(ptr2));
66 #if defined(O0)
67 assert(i2_label == (i_label | ptr_label));
68 assert(ptr2_label == (i_label | ptr_label));
69 #else
70 assert(i2_label == i_label);
71 assert(ptr2_label == ptr_label);
72 #endif
73
74 Pair pair3 = copy_pair2(pair1);
75 int i3 = pair3.i;
76 char *ptr3 = pair3.ptr;
77
78 dfsan_label i3_label = dfsan_read_label(&i3, sizeof(i3));
79 dfsan_label ptr3_label = dfsan_read_label(&ptr3, sizeof(ptr3));
80 #if defined(O0)
81 assert(i3_label == (i_label | ptr_label));
82 assert(ptr3_label == (i_label | ptr_label));
83 #else
84 assert(i3_label == i_label);
85 assert(ptr3_label == ptr_label);
86 #endif
87
88
89 return 0;
90 }
91