llvm-project

1
// Checks that dfsan works with trace-cmp instrumentation, even if some hooks
2
// are not defined (relies on week hooks implemented in dfsan).
3
//
4
// RUN: %clang_dfsan -fsanitize-coverage=trace-pc-guard,pc-table,func,trace-cmp %s -o %t
5
// RUN: %run %t 2>&1 | FileCheck %s
6

7
#include <stdio.h>
8
#include <stdint.h>
9
#include <stdlib.h>
10

11
#include <sanitizer/dfsan_interface.h>
12

13
uint32_t a4, b4;
14
uint64_t a8, b8;
15

16
// Define just two hooks, and leave others undefined.
17
void __dfsw___sanitizer_cov_trace_const_cmp4(uint8_t a, uint8_t b,
18
                                             dfsan_label l1, dfsan_label l2) {
19
  printf("const_cmp4 %d %d\n", a, b);
20
}
21
void __dfsw___sanitizer_cov_trace_cmp8(uint8_t a, uint8_t b, dfsan_label l1,
22
                                       dfsan_label l2) {
23
  printf("cmp8 %d %d\n", a, b);
24
}
25

26
int main(int argc, char **argv) {
27
  printf("MAIN\n");
28
  // CHECK: MAIN
29

30
  if (a4 != b4) abort();
31
  if (a4 == 42) abort();
32
  // CHECK: const_cmp4 42 0
33
  if (a8 != b8) abort();
34
  // CHECK: cmp8 0 0
35
  if (a8 == 66) abort();
36

37
  switch (10 / (a4 + 2)) {
38
    case 1: abort();
39
    case 2: exit(1);
40
    case 5:
41
            printf("SWITCH OK\n");
42
            break;
43
  }
44
  // CHECK: SWITCH OK
45

46

47
  printf("DONE\n");
48
  // CHECK: DONE
49
  return 0;
50
}
51