test/asan/TestCases/Posix/halt_on_error-signals.c

   1 // Test interaction of Asan recovery mode with asynch signals.
   2 //
   3 // RUN: %clang_asan -fsanitize-recover=address -pthread %s -o %t
   4 //
   5 // RUN: rm -f %t.log
   6 // RUN: %env_asan_opts=halt_on_error=false:suppress_equal_pcs=false %run %t 100 >%t.log 2>&1 || true
   7 // Collision will almost always get triggered but we still need to check the unlikely case:
   8 // RUN: FileCheck --check-prefix=CHECK-COLLISION %s < %t.log || FileCheck --check-prefix=CHECK-NO-COLLISION %s < %t.log
   9
  10 #define _SVID_SOURCE 1  // SA_NODEFER
  11
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <string.h>
  15 #include <pthread.h>
  16 #include <time.h>
  17 #include <signal.h>
  18
  19 #include <sanitizer/asan_interface.h>
  20
  21 void random_delay(unsigned *seed) {
  22   *seed = 1664525 * *seed + 1013904223;
  23   struct timespec delay = { 0, (*seed % 1000) * 1000 };
  24   nanosleep(&delay, 0);
  25 }
  26
  27 volatile char bad[2] = {1, };
  28
  29 void error() {
  30   // CHECK-COLLISION: AddressSanitizer: nested bug in the same thread, aborting
  31   // CHECK-NO-COLLISION: AddressSanitizer: use-after-poison
  32   volatile int idx = 0;
  33   bad[idx] = 0;
  34 }
  35
  36 #define CHECK_CALL(e, msg) do {             \
  37   if (0 != (e)) {                           \
  38     fprintf(stderr, "Failed to " msg "\n"); \
  39     exit(1);                                \
  40   }                                         \
  41 } while (0)
  42
  43 size_t niter = 10;
  44 pthread_t sender_tid, receiver_tid;
  45
  46 pthread_mutex_t keep_alive_mu = PTHREAD_MUTEX_INITIALIZER;
  47
  48 void *sender(void *arg) {
  49   unsigned seed = 0;
  50   for (size_t i = 0; i < niter; ++i) {
  51     random_delay(&seed);
  52     CHECK_CALL(pthread_kill(receiver_tid, SIGUSR1), "send signal");
  53   }
  54   return 0;
  55 }
  56
  57 void handler(int sig) {
  58   // Expect error collisions here
  59   error();
  60 }
  61
  62 void *receiver(void *arg) {
  63   unsigned seed = 1;
  64   for (size_t i = 0; i < niter; ++i) {
  65     random_delay(&seed);
  66     // And here
  67     error();
  68   }
  69   // Parent will release this when it's ok to terminate
  70   CHECK_CALL(pthread_mutex_lock(&keep_alive_mu), "unlock mutex");
  71   return 0;
  72 }
  73
  74 int main(int argc, char **argv) {
  75   if (argc != 2) {
  76     fprintf(stderr, "Syntax: %s niter\n", argv[0]);
  77     exit(1);
  78   }
  79
  80   niter = (size_t)strtoul(argv[1], 0, 0);
  81
  82   struct sigaction sa;
  83   memset(&sa, 0, sizeof(sa));
  84   sa.sa_handler = handler;
  85   sa.sa_flags = SA_NODEFER; // Enable nested handlers to add more stress
  86   CHECK_CALL(sigaction(SIGUSR1, &sa, 0), "set sighandler");
  87
  88   __asan_poison_memory_region(&bad, sizeof(bad));
  89
  90   CHECK_CALL(pthread_mutex_lock(&keep_alive_mu), "lock mutex");
  91   CHECK_CALL(pthread_create(&receiver_tid, 0, receiver, 0), "start thread");
  92   CHECK_CALL(pthread_create(&sender_tid, 0, sender, 0), "start thread");
  93   CHECK_CALL(pthread_join(sender_tid, 0), "join thread");
  94   // Now allow receiver to die
  95   CHECK_CALL(pthread_mutex_unlock(&keep_alive_mu), "unlock mutex");
  96   CHECK_CALL(pthread_join(receiver_tid, 0), "join thread");
  97
  98   // CHECK-NO-COLLISION: All threads terminated
  99   printf("All threads terminated\n");
 100
 101   return 0;
 102 }