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 /*
32 cc=your-compiler
33
34 $cc stress_dtls.c -pthread -ldl
35 for((i=0;i<100;i++)); do
36 $cc -fPIC -shared -DSO_NAME=f$i -o a.out-f$i.so stress_dtls.c;
37 done
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
50 typedef void **(*f_t)();
51
52 __thread int my_tls;
53
54 #define MAX_N_FUNCTIONS 1000
55 f_t Functions[MAX_N_FUNCTIONS];
56
PrintStuff(void * unused)57 void *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
main(int argc,char * argv[])71 int 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];
SO_NAME()113 void **SO_NAME() {
114 return &huge_thread_local_array[0];
115 }
116 #endif
117