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