[UXTHEME] -Move the global variables for the scrollbars into the WND_CONTEXT. Hook...
[reactos.git] / reactos / dll / win32 / uxtheme / themehooks.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS uxtheme.dll
4 * FILE: dll/win32/uxtheme/themehooks.c
5 * PURPOSE: uxtheme user api hook functions
6 * PROGRAMMER: Giannis Adamopoulos
7 */
8
9 #include "uxthemep.h"
10
11 USERAPIHOOK user32ApiHook;
12 BYTE gabDWPmessages[UAHOWP_MAX_SIZE];
13 BYTE gabMSGPmessages[UAHOWP_MAX_SIZE];
14 BOOL gbThemeHooksActive = FALSE;
15
16 PWND_CONTEXT ThemeGetWndContext(HWND hWnd)
17 {
18 PWND_CONTEXT pcontext;
19
20 pcontext = (PWND_CONTEXT)GetPropW(hWnd, (LPCWSTR)MAKEINTATOM(atWndContext));
21 if(pcontext == NULL)
22 {
23 pcontext = HeapAlloc(GetProcessHeap(),
24 HEAP_ZERO_MEMORY,
25 sizeof(WND_CONTEXT));
26 if(pcontext == NULL)
27 {
28 return NULL;
29 }
30
31 SetPropW( hWnd, (LPCWSTR)MAKEINTATOM(atWndContext), pcontext);
32 }
33
34 return pcontext;
35 }
36
37 void ThemeDestroyWndContext(HWND hWnd)
38 {
39 PWND_CONTEXT pContext;
40 DWORD ProcessId;
41
42 /*Do not destroy WND_CONTEXT of a window that belong to another process */
43 GetWindowThreadProcessId(hWnd, &ProcessId);
44 if(ProcessId != GetCurrentProcessId())
45 {
46 return;
47 }
48
49 pContext = (PWND_CONTEXT)GetPropW(hWnd, (LPCWSTR)MAKEINTATOM(atWndContext));
50 if(pContext == NULL)
51 {
52 return;
53 }
54
55 if(pContext->HasThemeRgn)
56 {
57 user32ApiHook.SetWindowRgn(hWnd, 0, TRUE);
58 }
59
60 HeapFree(GetProcessHeap(), 0, pContext);
61
62 SetPropW( hWnd, (LPCWSTR)MAKEINTATOM(atWndContext), NULL);
63 }
64
65 static BOOL CALLBACK ThemeCleanupChildWndContext (HWND hWnd, LPARAM msg)
66 {
67 ThemeDestroyWndContext(hWnd);
68 return TRUE;
69 }
70
71 static BOOL CALLBACK ThemeCleanupWndContext(HWND hWnd, LPARAM msg)
72 {
73 if (hWnd == NULL)
74 {
75 EnumWindows (ThemeCleanupWndContext, 0);
76 }
77 else
78 {
79 ThemeDestroyWndContext(hWnd);
80 EnumChildWindows (hWnd, ThemeCleanupChildWndContext, 0);
81 }
82
83 return TRUE;
84 }
85
86 void SetThemeRegion(HWND hWnd)
87 {
88 HTHEME hTheme;
89 RECT rcWindow;
90 HRGN hrgn, hrgn1;
91 int CaptionHeight, iPart;
92 WINDOWINFO wi;
93
94 TRACE("SetThemeRegion %d\n", hWnd);
95
96 wi.cbSize = sizeof(wi);
97 GetWindowInfo(hWnd, &wi);
98
99 /* Get the caption part id */
100 if (wi.dwExStyle & WS_EX_TOOLWINDOW)
101 iPart = WP_SMALLCAPTION;
102 else if (wi.dwStyle & WS_MAXIMIZE)
103 iPart = WP_MAXCAPTION;
104 else
105 iPart = WP_CAPTION;
106
107 CaptionHeight = wi.cyWindowBorders;
108 CaptionHeight += GetSystemMetrics(wi.dwExStyle & WS_EX_TOOLWINDOW ? SM_CYSMCAPTION : SM_CYCAPTION );
109
110 GetWindowRect(hWnd, &rcWindow);
111 rcWindow.right -= rcWindow.left;
112 rcWindow.bottom = CaptionHeight;
113 rcWindow.top = 0;
114 rcWindow.left = 0;
115
116 hTheme = MSSTYLES_OpenThemeClass(ActiveThemeFile, NULL, L"WINDOW");
117 GetThemeBackgroundRegion(hTheme, 0, iPart, FS_ACTIVE, &rcWindow, &hrgn);
118 CloseThemeData(hTheme);
119
120 GetWindowRect(hWnd, &rcWindow);
121 rcWindow.right -= rcWindow.left;
122 rcWindow.bottom -= rcWindow.top;
123 rcWindow.top = CaptionHeight;
124 rcWindow.left = 0;
125 hrgn1 = CreateRectRgnIndirect(&rcWindow);
126
127 CombineRgn(hrgn, hrgn, hrgn1, RGN_OR );
128
129 DeleteObject(hrgn1);
130
131 user32ApiHook.SetWindowRgn(hWnd, hrgn, TRUE);
132 }
133
134 int OnPostWinPosChanged(HWND hWnd, WINDOWPOS* pWinPos)
135 {
136 PWND_CONTEXT pcontext;
137 DWORD style;
138
139 /* We only proceed to change the window shape if it has a caption */
140 style = GetWindowLongW(hWnd, GWL_STYLE);
141 if((style & WS_CAPTION)!=WS_CAPTION)
142 return 0;
143
144 /* Get theme data for this window */
145 pcontext = ThemeGetWndContext(hWnd);
146 if (pcontext == NULL)
147 return 0;
148
149 /* Do not change the region of the window if its size wasn't changed */
150 if ((pWinPos->flags & SWP_NOSIZE) != 0 && pcontext->DirtyThemeRegion == FALSE)
151 return 0;
152
153 /* We don't touch the shape of the window if the application sets it on its own */
154 if (pcontext->HasAppDefinedRgn == TRUE)
155 return 0;
156
157 /* Calling SetWindowRgn will call SetWindowPos again so we need to avoid this recursion */
158 if (pcontext->UpdatingRgn == TRUE)
159 return 0;
160
161 if(!IsAppThemed())
162 {
163 if(pcontext->HasThemeRgn)
164 {
165 pcontext->HasThemeRgn = FALSE;
166 user32ApiHook.SetWindowRgn(hWnd, 0, TRUE);
167 }
168 return 0;
169 }
170
171 pcontext->DirtyThemeRegion = FALSE;
172 pcontext->HasThemeRgn = TRUE;
173 pcontext->UpdatingRgn = TRUE;
174 SetThemeRegion(hWnd);
175 pcontext->UpdatingRgn = FALSE;
176
177 return 0;
178 }
179
180 /**********************************************************************
181 * Hook Functions
182 */
183
184 static LRESULT CALLBACK
185 ThemeDefWindowProcW(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
186 {
187 if(!IsAppThemed())
188 {
189 return user32ApiHook.DefWindowProcW(hWnd,
190 Msg,
191 wParam,
192 lParam);
193 }
194
195 return ThemeWndProc(hWnd,
196 Msg,
197 wParam,
198 lParam,
199 user32ApiHook.DefWindowProcW);
200 }
201
202 static LRESULT CALLBACK
203 ThemeDefWindowProcA(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
204 {
205 if(!IsAppThemed())
206 {
207 return user32ApiHook.DefWindowProcA(hWnd,
208 Msg,
209 wParam,
210 lParam);
211 }
212
213 return ThemeWndProc(hWnd,
214 Msg,
215 wParam,
216 lParam,
217 user32ApiHook.DefWindowProcA);
218 }
219
220 static LRESULT CALLBACK
221 ThemePreWindowProc(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam, ULONG_PTR ret,PDWORD unknown)
222 {
223 switch(Msg)
224 {
225 case WM_THEMECHANGED:
226 if (GetAncestor(hWnd, GA_PARENT) == GetDesktopWindow())
227 UXTHEME_LoadTheme(TRUE);
228 case WM_NCCREATE:
229 {
230 PWND_CONTEXT pcontext = ThemeGetWndContext(hWnd);
231 if (pcontext == NULL)
232 return 0;
233 pcontext->DirtyThemeRegion = TRUE;
234 }
235 }
236
237 return 0;
238 }
239
240
241 static LRESULT CALLBACK
242 ThemePostWindowProc(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam, ULONG_PTR ret,PDWORD unknown)
243 {
244 switch(Msg)
245 {
246 case WM_WINDOWPOSCHANGED:
247 {
248 return OnPostWinPosChanged(hWnd, (WINDOWPOS*)lParam);
249 }
250 case WM_DESTROY:
251 {
252 ThemeDestroyWndContext(hWnd);
253 return 0;
254 }
255 }
256
257 return 0;
258 }
259
260 int WINAPI ThemeSetWindowRgn(HWND hWnd, HRGN hRgn, BOOL bRedraw)
261 {
262 PWND_CONTEXT pcontext = ThemeGetWndContext(hWnd);
263 if(pcontext)
264 {
265 pcontext->HasAppDefinedRgn = TRUE;
266 pcontext->HasThemeRgn = FALSE;
267 }
268
269 return user32ApiHook.SetWindowRgn(hWnd, hRgn, bRedraw);
270 }
271
272 BOOL ThemeGetScrollInfo(HWND hwnd, int fnBar, LPSCROLLINFO lpsi)
273 {
274 PWND_CONTEXT pwndContext;
275 DWORD style;
276 BOOL ret;
277
278 /* Avoid creating a window context if it is not needed */
279 if(!IsAppThemed())
280 goto dodefault;
281
282 style = GetWindowLongW(hwnd, GWL_STYLE);
283 if((style & (WS_HSCROLL|WS_VSCROLL))==0)
284 goto dodefault;
285
286 pwndContext = ThemeGetWndContext(hwnd);
287 if (pwndContext == NULL)
288 goto dodefault;
289
290 /*
291 * Uxtheme needs to handle the tracking of the scrollbar itself
292 * This means than if an application needs to get the track position
293 * with GetScrollInfo, it will get wrong data. So uxtheme needs to
294 * hook it and set the correct tracking position itself
295 */
296 ret = user32ApiHook.GetScrollInfo(hwnd, fnBar, lpsi);
297 if ( lpsi &&
298 (lpsi->fMask & SIF_TRACKPOS) &&
299 pwndContext->SCROLL_TrackingWin == hwnd &&
300 pwndContext->SCROLL_TrackingBar == fnBar)
301 {
302 lpsi->nTrackPos = pwndContext->SCROLL_TrackingVal;
303 }
304 return ret;
305
306 dodefault:
307 return user32ApiHook.GetScrollInfo(hwnd, fnBar, lpsi);
308 }
309
310 /**********************************************************************
311 * Exports
312 */
313
314 BOOL CALLBACK
315 ThemeInitApiHook(UAPIHK State, PUSERAPIHOOK puah)
316 {
317 if (!puah || State != uahLoadInit)
318 {
319 UXTHEME_LoadTheme(FALSE);
320 ThemeCleanupWndContext(NULL, 0);
321 gbThemeHooksActive = FALSE;
322 return TRUE;
323 }
324
325 gbThemeHooksActive = TRUE;
326
327 /* Store the original functions from user32 */
328 user32ApiHook = *puah;
329
330 puah->DefWindowProcA = ThemeDefWindowProcA;
331 puah->DefWindowProcW = ThemeDefWindowProcW;
332 puah->PreWndProc = ThemePreWindowProc;
333 puah->PostWndProc = ThemePostWindowProc;
334 puah->PreDefDlgProc = ThemePreWindowProc;
335 puah->PostDefDlgProc = ThemePostWindowProc;
336 puah->DefWndProcArray.MsgBitArray = gabDWPmessages;
337 puah->DefWndProcArray.Size = UAHOWP_MAX_SIZE;
338 puah->WndProcArray.MsgBitArray = gabMSGPmessages;
339 puah->WndProcArray.Size = UAHOWP_MAX_SIZE;
340 puah->DlgProcArray.MsgBitArray = gabMSGPmessages;
341 puah->DlgProcArray.Size = UAHOWP_MAX_SIZE;
342
343 puah->SetWindowRgn = ThemeSetWindowRgn;
344 puah->GetScrollInfo = (FARPROC)ThemeGetScrollInfo;
345
346 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCPAINT);
347 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCACTIVATE);
348 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCMOUSEMOVE);
349 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCMOUSELEAVE);
350 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCHITTEST);
351 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCLBUTTONDOWN);
352 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCUAHDRAWCAPTION);
353 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCUAHDRAWFRAME);
354 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_SETTEXT);
355 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_WINDOWPOSCHANGED);
356 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_CONTEXTMENU);
357 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_STYLECHANGED);
358 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_SETICON);
359 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_NCDESTROY);
360 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_SYSCOMMAND);
361 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_CTLCOLORMSGBOX);
362 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_CTLCOLORBTN);
363 UAH_HOOK_MESSAGE(puah->DefWndProcArray, WM_CTLCOLORSTATIC);
364
365 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_CREATE);
366 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_SETTINGCHANGE);
367 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_DRAWITEM);
368 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_MEASUREITEM);
369 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_WINDOWPOSCHANGING);
370 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_WINDOWPOSCHANGED);
371 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_STYLECHANGING);
372 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_STYLECHANGED);
373 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_NCCREATE);
374 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_NCDESTROY);
375 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_NCPAINT);
376 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_MENUCHAR);
377 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_MDISETMENU);
378 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_THEMECHANGED);
379 UAH_HOOK_MESSAGE(puah->WndProcArray, WM_UAHINIT);
380
381 UXTHEME_LoadTheme(TRUE);
382
383 return TRUE;
384 }
385
386 typedef BOOL (WINAPI * PREGISTER_UAH_WINXP)(HINSTANCE hInstance, USERAPIHOOKPROC CallbackFunc);
387 typedef BOOL (WINAPI * PREGISTER_UUAH_WIN2003)(PUSERAPIHOOKINFO puah);
388
389 BOOL WINAPI
390 ThemeHooksInstall()
391 {
392 PVOID lpFunc;
393 OSVERSIONINFO osvi;
394 BOOL ret;
395
396 lpFunc = GetProcAddress(GetModuleHandle("user32.dll"), "RegisterUserApiHook");
397
398 ZeroMemory(&osvi, sizeof(OSVERSIONINFO));
399 osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
400 GetVersionEx(&osvi);
401
402 if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 1)
403 {
404 PREGISTER_UAH_WINXP lpfuncxp = (PREGISTER_UAH_WINXP)lpFunc;
405 ret = lpfuncxp(hDllInst, ThemeInitApiHook);
406 }
407 else if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 2)
408 {
409 PREGISTER_UUAH_WIN2003 lpfunc2003 = (PREGISTER_UUAH_WIN2003)lpFunc;
410 USERAPIHOOKINFO uah;
411
412 uah.m_size = sizeof(uah);
413 uah.m_dllname1 = L"uxtheme.dll";
414 uah.m_funname1 = L"ThemeInitApiHook";
415 uah.m_dllname2 = NULL;
416 uah.m_funname2 = NULL;
417
418 ret = lpfunc2003(&uah);
419 }
420 else
421 {
422 UNIMPLEMENTED;
423 ret = FALSE;
424 }
425
426 UXTHEME_broadcast_msg (NULL, WM_THEMECHANGED);
427
428 return ret;
429 }
430
431 BOOL WINAPI
432 ThemeHooksRemove()
433 {
434 BOOL ret;
435
436 ret = UnregisterUserApiHook();
437
438 UXTHEME_broadcast_msg (NULL, WM_THEMECHANGED);
439
440 return ret;
441 }
442
443 INT WINAPI ClassicSystemParametersInfoW(UINT uiAction, UINT uiParam, PVOID pvParam, UINT fWinIni)
444 {
445 if (gbThemeHooksActive)
446 {
447 return user32ApiHook.SystemParametersInfoW(uiAction, uiParam, pvParam, fWinIni);
448 }
449
450 return SystemParametersInfoW(uiAction, uiParam, pvParam, fWinIni);
451 }
452
453 INT WINAPI ClassicSystemParametersInfoA(UINT uiAction, UINT uiParam, PVOID pvParam, UINT fWinIni)
454 {
455 if (gbThemeHooksActive)
456 {
457 return user32ApiHook.SystemParametersInfoA(uiAction, uiParam, pvParam, fWinIni);
458 }
459
460 return SystemParametersInfoA(uiAction, uiParam, pvParam, fWinIni);
461 }
462
463 INT WINAPI ClassicGetSystemMetrics(int nIndex)
464 {
465 if (gbThemeHooksActive)
466 {
467 return user32ApiHook.GetSystemMetrics(nIndex);
468 }
469
470 return GetSystemMetrics(nIndex);
471 }
472
473 BOOL WINAPI ClassicAdjustWindowRectEx(LPRECT lpRect, DWORD dwStyle, BOOL bMenu, DWORD dwExStyle)
474 {
475 if (gbThemeHooksActive)
476 {
477 return user32ApiHook.AdjustWindowRectEx(lpRect, dwStyle, bMenu, dwExStyle);
478 }
479
480 return AdjustWindowRectEx(lpRect, dwStyle, bMenu, dwExStyle);
481 }