1/**************************************************************************
2 *
3 * Copyright 2009 VMware, Inc.
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28#include <windows.h>
29
30#include "pipe/p_compiler.h"
31#include "util/u_memory.h"
32#include "stw_tls.h"
33
34static DWORD tlsIndex = TLS_OUT_OF_INDEXES;
35
36boolean
37stw_tls_init(void)
38{
39   tlsIndex = TlsAlloc();
40   if (tlsIndex == TLS_OUT_OF_INDEXES) {
41      return FALSE;
42   }
43
44   return TRUE;
45}
46
47static INLINE struct stw_tls_data *
48stw_tls_data_create()
49{
50   struct stw_tls_data *data;
51
52   data = CALLOC_STRUCT(stw_tls_data);
53   if (!data)
54      goto no_data;
55
56   data->hCallWndProcHook = SetWindowsHookEx(WH_CALLWNDPROC,
57                                             stw_call_window_proc,
58                                             NULL,
59                                             GetCurrentThreadId());
60   if(data->hCallWndProcHook == NULL)
61      goto no_hook;
62
63   TlsSetValue(tlsIndex, data);
64
65   return data;
66
67no_hook:
68   FREE(data);
69no_data:
70   return NULL;
71}
72
73boolean
74stw_tls_init_thread(void)
75{
76   struct stw_tls_data *data;
77
78   if (tlsIndex == TLS_OUT_OF_INDEXES) {
79      return FALSE;
80   }
81
82   data = stw_tls_data_create();
83   if(!data)
84      return FALSE;
85
86   return TRUE;
87}
88
89void
90stw_tls_cleanup_thread(void)
91{
92   struct stw_tls_data *data;
93
94   if (tlsIndex == TLS_OUT_OF_INDEXES) {
95      return;
96   }
97
98   data = (struct stw_tls_data *) TlsGetValue(tlsIndex);
99   if(data) {
100      TlsSetValue(tlsIndex, NULL);
101
102      if(data->hCallWndProcHook) {
103         UnhookWindowsHookEx(data->hCallWndProcHook);
104         data->hCallWndProcHook = NULL;
105      }
106
107      FREE(data);
108   }
109}
110
111void
112stw_tls_cleanup(void)
113{
114   if (tlsIndex != TLS_OUT_OF_INDEXES) {
115      TlsFree(tlsIndex);
116      tlsIndex = TLS_OUT_OF_INDEXES;
117   }
118}
119
120struct stw_tls_data *
121stw_tls_get_data(void)
122{
123   struct stw_tls_data *data;
124
125   if (tlsIndex == TLS_OUT_OF_INDEXES) {
126      return NULL;
127   }
128
129   data = (struct stw_tls_data *) TlsGetValue(tlsIndex);
130   if(!data) {
131      /* DllMain is called with DLL_THREAD_ATTACH only by threads created after
132       * the DLL is loaded by the process */
133      data = stw_tls_data_create();
134      if(!data)
135         return NULL;
136   }
137
138   return data;
139}
140