tls_init.cpp
1.28 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
// RUN: %clang_cl_asan %s -Fe%t.exe /MD
// RUN: %run %t.exe | FileCheck %s
// CHECK: my_thread_callback
// CHECK: ran_before_main: 1
#include <windows.h>
#include <stdio.h>
#include <string.h>
#pragma comment (lib, "dbghelp")
static bool ran_before_main = false;
extern "C" void __asan_init(void);
static void NTAPI /*__attribute__((no_sanitize_address))*/
my_thread_callback(PVOID module, DWORD reason, PVOID reserved) {
ran_before_main = true;
static const char str[] = "my_thread_callback\n";
// Fail the test if we aren't called for the expected reason or we can't write
// stdout.
if (reason != DLL_PROCESS_ATTACH)
return;
HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
if (!out || out == INVALID_HANDLE_VALUE)
return;
DWORD written = 0;
WriteFile(out, &str[0], sizeof(str), &written, NULL);
}
extern "C" {
#pragma const_seg(".CRT$XLC")
extern const PIMAGE_TLS_CALLBACK p_thread_callback;
const PIMAGE_TLS_CALLBACK p_thread_callback = my_thread_callback;
#pragma const_seg()
}
#ifdef _WIN64
#pragma comment(linker, "/INCLUDE:_tls_used")
#pragma comment(linker, "/INCLUDE:p_thread_callback")
#else
#pragma comment(linker, "/INCLUDE:__tls_used")
#pragma comment(linker, "/INCLUDE:_p_thread_callback")
#endif
int main() {
printf("ran_before_main: %d\n", ran_before_main);
}