llvm-project

1
// RUN: %clangxx_dfsan %s -o %t && %run %t
2
// RUN: %clangxx_dfsan -mllvm -dfsan-track-origins=1 %s -o %t && %run %t
3
// RUN: %clangxx_dfsan -mllvm -dfsan-track-origins=1 -mllvm -dfsan-instrument-with-call-threshold=0 %s -o %t && %run %t
4
//
5
// Test that the state of shadows from a sigaction handler are consistent.
6

7
#include <signal.h>
8
#include <stdarg.h>
9
#include <sanitizer/dfsan_interface.h>
10
#include <assert.h>
11
#include <sys/time.h>
12
#include <stdio.h>
13

14
const int kSigCnt = 200;
15
int x = 0;
16

17
__attribute__((noinline))
18
int f(int a) {
19
  return a;
20
}
21

22
__attribute__((noinline))
23
void g() {
24
  int r = f(x);
25
  const dfsan_label r_label = dfsan_get_label(r);
26
  assert(r_label == 8 || r_label == 0);
27
  return;
28
}
29

30
int sigcnt;
31

32
void SignalHandler(int signo) {
33
  assert(signo == SIGPROF);
34
  int a = 0;
35
  dfsan_set_label(4, &a, sizeof(a));
36
  (void)f(a);
37
  ++sigcnt;
38
}
39

40
int main() {
41
  struct sigaction psa = {};
42
  psa.sa_handler = SignalHandler;
43
  int r = sigaction(SIGPROF, &psa, NULL);
44

45
  itimerval itv;
46
  itv.it_interval.tv_sec = 0;
47
  itv.it_interval.tv_usec = 100;
48
  itv.it_value.tv_sec = 0;
49
  itv.it_value.tv_usec = 100;
50
  setitimer(ITIMER_PROF, &itv, NULL);
51

52
  dfsan_set_label(8, &x, sizeof(x));
53
  do {
54
    g();
55
  } while (sigcnt < kSigCnt);
56

57
  itv.it_interval.tv_sec = 0;
58
  itv.it_interval.tv_usec = 0;
59
  itv.it_value.tv_sec = 0;
60
  itv.it_value.tv_usec = 0;
61
  setitimer(ITIMER_PROF, &itv, NULL);
62

63
  signal(SIGPROF, SIG_DFL);
64
  return 0;
65
}
66