1 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -mllvm -dfsan-combine-taint-lookup-table=remap_to_upper -DLOOKUP_TABLE -o %t && %run %t
2 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -mllvm -dfsan-combine-taint-lookup-table=no_match -o %t && %run %t
3 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -o %t && %run %t
4
5 #include <sanitizer/dfsan_interface.h>
6 #include <assert.h>
7
8 const char remap_to_upper[256] = {
9 '.', '.', '.', '.', '.', '.', '.', '.', '.',
10 '.', '.', '.', '.', '.', '.', '.', '.', '.',
11 '.', '.', '.', '.', '.', '.', '.', '.', '.',
12 '.', '.', '.', '.', '.', '.', '.', '.', '.',
13 '.', '.', '.', '.', '.', '.', '.', '.', '.',
14 '.', '.', '.', '.', '.', '.', '.', '.', '.',
15 '.', '.', '.', '.', '.', '.', '.', '.', '.',
16 '.', '.', '.', '.', '.', '.', '.', '.', '.',
17 '.', '.', '.', '.', '.', '.', '.', '.', '.',
18 '.', '.', '.', '.', '.', '.', '.', '.', '.',
19 '.', '.', '.', '.', '.', '.', '.', 'A', 'B',
20 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
21 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T',
22 'U', 'V', 'W', 'X', 'Y', 'Z', '.', '.', '.',
23 '.', '.', '.', '.', '.', '.', '.', '.', '.',
24 '.', '.', '.', '.', '.', '.', '.', '.', '.',
25 '.', '.', '.', '.', '.', '.', '.', '.', '.',
26 '.', '.', '.', '.', '.', '.', '.', '.', '.',
27 '.', '.', '.', '.', '.', '.', '.', '.', '.',
28 '.', '.', '.', '.', '.', '.', '.', '.', '.',
29 '.', '.', '.', '.', '.', '.', '.', '.', '.',
30 '.', '.', '.', '.', '.', '.', '.', '.', '.',
31 '.', '.', '.', '.', '.', '.', '.', '.', '.',
32 '.', '.', '.', '.', '.', '.', '.', '.', '.',
33 '.', '.', '.', '.', '.', '.', '.', '.', '.',
34 '.', '.', '.', '.', '.', '.', '.', '.', '.',
35 '.', '.', '.', '.', '.', '.', '.', '.', '.',
36 '.', '.', '.', '.', '.', '.', '.', '.', '.',
37 '.', '.', '.', '.',
38 };
39
character_mapping(unsigned char c)40 char character_mapping(unsigned char c) {
41 return remap_to_upper[c];
42 }
43
main(void)44 int main(void) {
45 char a = 'b';
46 dfsan_label i_label = 1;
47 dfsan_set_label(i_label, &a, sizeof(a));
48 assert(dfsan_read_label(&a, sizeof(a)) == i_label);
49
50 char b = character_mapping(a);
51 assert(b == 'B');
52
53 #ifdef LOOKUP_TABLE
54 assert(dfsan_read_label(&b, sizeof(b)) == i_label);
55 #else
56 assert(dfsan_read_label(&b, sizeof(b)) == 0);
57 #endif
58 return 0;
59 }
60