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