#include <windows.h>
#include "pipe/p_compiler.h"
#include "util/u_memory.h"
#include "stw_tls.h"
static DWORD tlsIndex = TLS_OUT_OF_INDEXES;
boolean
stw_tls_init(void)
{
tlsIndex = TlsAlloc();
if (tlsIndex == TLS_OUT_OF_INDEXES) {
return FALSE;
}
return TRUE;
}
static INLINE struct stw_tls_data *
stw_tls_data_create()
{
struct stw_tls_data *data;
data = CALLOC_STRUCT(stw_tls_data);
if (!data)
goto no_data;
data->hCallWndProcHook = SetWindowsHookEx(WH_CALLWNDPROC,
stw_call_window_proc,
NULL,
GetCurrentThreadId());
if(data->hCallWndProcHook == NULL)
goto no_hook;
TlsSetValue(tlsIndex, data);
return data;
no_hook:
FREE(data);
no_data:
return NULL;
}
boolean
stw_tls_init_thread(void)
{
struct stw_tls_data *data;
if (tlsIndex == TLS_OUT_OF_INDEXES) {
return FALSE;
}
data = stw_tls_data_create();
if(!data)
return FALSE;
return TRUE;
}
void
stw_tls_cleanup_thread(void)
{
struct stw_tls_data *data;
if (tlsIndex == TLS_OUT_OF_INDEXES) {
return;
}
data = (struct stw_tls_data *) TlsGetValue(tlsIndex);
if(data) {
TlsSetValue(tlsIndex, NULL);
if(data->hCallWndProcHook) {
UnhookWindowsHookEx(data->hCallWndProcHook);
data->hCallWndProcHook = NULL;
}
FREE(data);
}
}
void
stw_tls_cleanup(void)
{
if (tlsIndex != TLS_OUT_OF_INDEXES) {
TlsFree(tlsIndex);
tlsIndex = TLS_OUT_OF_INDEXES;
}
}
struct stw_tls_data *
stw_tls_get_data(void)
{
struct stw_tls_data *data;
if (tlsIndex == TLS_OUT_OF_INDEXES) {
return NULL;
}
data = (struct stw_tls_data *) TlsGetValue(tlsIndex);
if(!data) {
data = stw_tls_data_create();
if(!data)
return NULL;
}
return data;
}