compiler-rt/test/dfsan/sigaction_stress_test.c

   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 // REQUIRES: x86_64-target-arch
   8
   9 #include <signal.h>
  10 #include <stdarg.h>
  11 #include <sanitizer/dfsan_interface.h>
  12 #include <assert.h>
  13 #include <sys/time.h>
  14 #include <stdio.h>
  15
  16 const int kSigCnt = 200;
  17 int x = 0;
  18
  19 __attribute__((noinline))
  20 int f(int a) {
  21   return a;
  22 }
  23
  24 __attribute__((noinline))
  25 void g() {
  26   int r = f(x);
  27   const dfsan_label r_label = dfsan_get_label(r);
  28   assert(r_label == 8 || r_label == 0);
  29   return;
  30 }
  31
  32 int sigcnt;
  33
  34 void SignalHandler(int signo) {
  35   assert(signo == SIGPROF);
  36   int a = 0;
  37   dfsan_set_label(4, &a, sizeof(a));
  38   (void)f(a);
  39   ++sigcnt;
  40 }
  41
  42 int main() {
  43   struct sigaction psa = {};
  44   psa.sa_handler = SignalHandler;
  45   int r = sigaction(SIGPROF, &psa, NULL);
  46
  47   itimerval itv;
  48   itv.it_interval.tv_sec = 0;
  49   itv.it_interval.tv_usec = 100;
  50   itv.it_value.tv_sec = 0;
  51   itv.it_value.tv_usec = 100;
  52   setitimer(ITIMER_PROF, &itv, NULL);
  53
  54   dfsan_set_label(8, &x, sizeof(x));
  55   do {
  56     g();
  57   } while (sigcnt < kSigCnt);
  58
  59   itv.it_interval.tv_sec = 0;
  60   itv.it_interval.tv_usec = 0;
  61   itv.it_value.tv_sec = 0;
  62   itv.it_value.tv_usec = 0;
  63   setitimer(ITIMER_PROF, &itv, NULL);
  64
  65   signal(SIGPROF, SIG_DFL);
  66   return 0;
  67 }