test/asan/TestCases/Linux/stress_dtls.c

   1 // REQUIRES: asan-64-bits
   2 // UNSUPPORTED: android
   3 // Stress test dynamic TLS + dlopen + threads.
   4 //
   5 // Note that glibc 2.15 seems utterly broken on this test,
   6 // it fails with ~17 DSOs dlopen-ed.
   7 // glibc 2.19 seems fine.
   8 //
   9 //
  10 // RUN: %clangxx_asan -x c -DSO_NAME=f0 %s -shared -o %t-f0.so -fPIC
  11 // RUN: %clangxx_asan -x c -DSO_NAME=f1 %s -shared -o %t-f1.so -fPIC
  12 // RUN: %clangxx_asan -x c -DSO_NAME=f2 %s -shared -o %t-f2.so -fPIC
  13 // RUN: %clangxx_asan %s -ldl -pthread -o %t
  14 // RUN: %run %t 0 3
  15 // RUN: %run %t 2 3
  16 // RUN: %env_asan_opts=verbosity=2 %run %t 10 2 2>&1 | FileCheck %s
  17 // RUN: %env_asan_opts=verbosity=2:intercept_tls_get_addr=1 %run %t 10 2 2>&1 | FileCheck %s
  18 // RUN: %env_asan_opts=verbosity=2:intercept_tls_get_addr=0 %run %t 10 2 2>&1 | FileCheck %s --check-prefix=CHECK0
  19 // CHECK: __tls_get_addr
  20 // CHECK: Creating thread 0
  21 // CHECK: __tls_get_addr
  22 // CHECK: Creating thread 1
  23 // CHECK: __tls_get_addr
  24 // CHECK: Creating thread 2
  25 // CHECK: __tls_get_addr
  26 // CHECK: Creating thread 3
  27 // CHECK: __tls_get_addr
  28 // Make sure that TLS slots don't leak
  29 // CHECK-NOT: num_live_dtls 5
  30 //
  31 // CHECK0-NOT: __tls_get_addr
  32 /*
  33 cc=your-compiler
  34
  35 $cc stress_dtls.c -pthread -ldl
  36 for((i=0;i<100;i++)); do
  37   $cc -fPIC -shared -DSO_NAME=f$i -o a.out-f$i.so stress_dtls.c;
  38 done
  39 ./a.out 2 4  # <<<<<< 2 threads, 4 libs
  40 ./a.out 3 50 # <<<<<< 3 threads, 50 libs
  41 */
  42 #ifndef SO_NAME
  43 #define _GNU_SOURCE
  44 #include <assert.h>
  45 #include <dlfcn.h>
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <pthread.h>
  49 #include <stdint.h>
  50
  51 typedef void **(*f_t)();
  52
  53 __thread int my_tls;
  54
  55 #define MAX_N_FUNCTIONS 1000
  56 f_t Functions[MAX_N_FUNCTIONS];
  57
  58 void *PrintStuff(void *unused) {
  59   uintptr_t stack;
  60   // fprintf(stderr, "STACK: %p TLS: %p SELF: %p\n", &stack, &my_tls,
  61   //        (void *)pthread_self());
  62   int i;
  63   for (i = 0; i < MAX_N_FUNCTIONS; i++) {
  64     if (!Functions[i]) break;
  65     uintptr_t dtls = (uintptr_t)Functions[i]();
  66     fprintf(stderr, "  dtls[%03d]: %lx\n", i, dtls);
  67     *(long*)dtls = 42;  // check that this is writable.
  68   }
  69   return NULL;
  70 }
  71
  72 int main(int argc, char *argv[]) {
  73   int num_threads = 1;
  74   int num_libs = 1;
  75   if (argc >= 2)
  76     num_threads = atoi(argv[1]);
  77   if (argc >= 3)
  78     num_libs = atoi(argv[2]);
  79   assert(num_libs <= MAX_N_FUNCTIONS);
  80
  81   int lib;
  82   for (lib = 0; lib < num_libs; lib++) {
  83     char buf[4096];
  84     snprintf(buf, sizeof(buf), "%s-f%d.so", argv[0], lib);
  85     void *handle = dlopen(buf, RTLD_LAZY);
  86     if (!handle) {
  87       fprintf(stderr, "%s\n", dlerror());
  88       exit(1);
  89     }
  90     snprintf(buf, sizeof(buf), "f%d", lib);
  91     Functions[lib] = (f_t)dlsym(handle, buf);
  92     if (!Functions[lib]) {
  93       fprintf(stderr, "%s\n", dlerror());
  94       exit(1);
  95     }
  96     fprintf(stderr, "LIB[%03d] %s: %p\n", lib, buf, Functions[lib]);
  97     PrintStuff(0);
  98
  99     int i;
 100     for (i = 0; i < num_threads; i++) {
 101       pthread_t t;
 102       fprintf(stderr, "Creating thread %d\n", i);
 103       pthread_create(&t, 0, PrintStuff, 0);
 104       pthread_join(t, 0);
 105     }
 106   }
 107   return 0;
 108 }
 109 #else  // SO_NAME
 110 #ifndef DTLS_SIZE
 111 # define DTLS_SIZE (1 << 17)
 112 #endif
 113 __thread void *huge_thread_local_array[DTLS_SIZE];
 114 void **SO_NAME() {
 115   return &huge_thread_local_array[0];
 116 }
 117 #endif