wibo/test/tls_reloc_dll.c

#include <windows.h>

#ifndef TLS_RELOC_INITIAL_VALUE
#define TLS_RELOC_INITIAL_VALUE 0x2468ACEDu
#endif

__attribute__((section(".tls$AAA"), used)) static int g_tlsInitialValue = (int)TLS_RELOC_INITIAL_VALUE;
__attribute__((section(".tls$ZZZ"), used)) static const int g_tlsTerminator = 0;

static int g_tlsCallbackCount = 0;

static void NTAPI tls_callback(PVOID module, DWORD reason, PVOID reserved) {
	(void)module;
	(void)reserved;
	if (reason == DLL_PROCESS_ATTACH) {
		++g_tlsCallbackCount;
	}
}

__attribute__((section(".CRT$XLB"), used)) static const PIMAGE_TLS_CALLBACK g_tlsCallback = tls_callback;

BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpReserved) {
	(void)lpReserved;
	if (fdwReason == DLL_PROCESS_ATTACH) {
		DisableThreadLibraryCalls(hinstDLL);
	}
	return TRUE;
}

__declspec(dllexport) int __stdcall tls_get_template_value(void) {
	return g_tlsInitialValue;
}

__declspec(dllexport) void *__stdcall tls_template_address(void) {
	return &g_tlsInitialValue;
}

__declspec(dllexport) int __stdcall tls_callback_hits(void) {
	return g_tlsCallbackCount;
}