compiler-rt/test/tsan/Linux/check_preinit.cpp

   1 // RUN: %clang_tsan -fno-sanitize=thread -shared -fPIC -O1 -DBUILD_SO=1 %s -o \
   2 // RUN:  %t.so && \
   3 // RUN:   %clang_tsan -O1 %s %t.so -o %t && %run %t 2>&1 | FileCheck %s
   4 // RUN: llvm-objdump -t %t | FileCheck %s --check-prefix=CHECK-DUMP
   5 // CHECK-DUMP:  {{[.]preinit_array.*__local_tsan_preinit}}
   6
   7 // SANITIZER_CAN_USE_PREINIT_ARRAY is undefined on android.
   8 // UNSUPPORTED: android
   9
  10 // Test checks if __tsan_init is called from .preinit_array.
  11 // Without initialization from .preinit_array, __tsan_init will be called from
  12 // constructors of the binary which are called after constructors of shared
  13 // library.
  14
  15 #include <sanitizer/tsan_interface.h>
  16 #include <stdio.h>
  17
  18 #if BUILD_SO
  19
  20 // "volatile" is needed to avoid compiler optimize-out constructors.
  21 volatile int counter = 0;
  22 volatile int lib_constructor_call = 0;
  23 volatile int tsan_init_call = 0;
  24
  25 __attribute__ ((constructor))
  26 void LibConstructor() {
  27   lib_constructor_call = ++counter;
  28 };
  29
  30 #else  // BUILD_SO
  31
  32 extern int counter;
  33 extern int lib_constructor_call;
  34 extern int tsan_init_call;
  35
  36 volatile int bin_constructor_call = 0;
  37
  38 __attribute__ ((constructor))
  39 void BinConstructor() {
  40   bin_constructor_call = ++counter;
  41 };
  42
  43 namespace __tsan {
  44
  45 void OnInitialize() {
  46   tsan_init_call = ++counter;
  47 }
  48
  49 }
  50
  51 int main() {
  52   // CHECK: TSAN_INIT 1
  53   // CHECK: LIB_CONSTRUCTOR 2
  54   // CHECK: BIN_CONSTRUCTOR 3
  55   printf("TSAN_INIT %d\n", tsan_init_call);
  56   printf("LIB_CONSTRUCTOR %d\n", lib_constructor_call);
  57   printf("BIN_CONSTRUCTOR %d\n", bin_constructor_call);
  58   return 0;
  59 }
  60
  61 #endif  // BUILD_SO