Protect USER32 against early graphics driver unloading.
[wine] / dlls / user / user_main.c
1 /*
2  * USER initialization code
3  *
4  * Copyright 2000 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include <stdarg.h>
22 #include <string.h>
23 #include "windef.h"
24 #include "winbase.h"
25 #include "wingdi.h"
26 #include "winuser.h"
27 #include "winreg.h"
28 #include "wine/winbase16.h"
29 #include "wine/winuser16.h"
30
31 #include "controls.h"
32 #include "cursoricon.h"
33 #include "message.h"
34 #include "user.h"
35 #include "win.h"
36 #include "wine/debug.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(graphics);
39
40 USER_DRIVER USER_Driver;
41
42 WORD USER_HeapSel = 0;  /* USER heap selector */
43 HMODULE user32_module = 0;
44
45 extern HPALETTE (WINAPI *pfnGDISelectPalette)(HDC hdc, HPALETTE hpal, WORD bkgnd );
46 extern UINT (WINAPI *pfnGDIRealizePalette)(HDC hdc);
47
48 static HMODULE graphics_driver;
49 static DWORD exiting_thread_id;
50
51 extern void WDML_NotifyThreadDetach(void);
52
53 #define GET_USER_FUNC(name) USER_Driver.p##name = (void*)GetProcAddress( graphics_driver, #name )
54
55 /* load the graphics driver */
56 static BOOL load_driver(void)
57 {
58     char buffer[MAX_PATH], *name, *next;
59     HKEY hkey;
60
61     strcpy( buffer, "x11drv,ttydrv" );  /* default value */
62     if (!RegOpenKeyA( HKEY_LOCAL_MACHINE, "Software\\Wine\\Wine\\Config\\Wine", &hkey ))
63     {
64         DWORD type, count = sizeof(buffer);
65         RegQueryValueExA( hkey, "GraphicsDriver", 0, &type, buffer, &count );
66         RegCloseKey( hkey );
67     }
68
69     name = buffer;
70     while (name)
71     {
72         next = strchr( name, ',' );
73         if (next) *next++ = 0;
74
75         if ((graphics_driver = LoadLibraryA( name )) != 0) break;
76         name = next;
77     }
78     if (!graphics_driver)
79     {
80         MESSAGE( "wine: Could not load graphics driver '%s'.\n", buffer );
81         if (!strcasecmp( buffer, "x11drv" ))
82             MESSAGE( "Make sure that your X server is running and that $DISPLAY is set correctly.\n" );
83         ExitProcess(1);
84     }
85
86     GET_USER_FUNC(InitKeyboard);
87     GET_USER_FUNC(VkKeyScanEx);
88     GET_USER_FUNC(MapVirtualKeyEx);
89     GET_USER_FUNC(GetKeyNameText);
90     GET_USER_FUNC(ToUnicodeEx);
91     GET_USER_FUNC(GetKeyboardLayoutList);
92     GET_USER_FUNC(GetKeyboardLayout);
93     GET_USER_FUNC(GetKeyboardLayoutName);
94     GET_USER_FUNC(LoadKeyboardLayout);
95     GET_USER_FUNC(ActivateKeyboardLayout);
96     GET_USER_FUNC(UnloadKeyboardLayout);
97     GET_USER_FUNC(Beep);
98     GET_USER_FUNC(InitMouse);
99     GET_USER_FUNC(SetCursor);
100     GET_USER_FUNC(GetCursorPos);
101     GET_USER_FUNC(SetCursorPos);
102     GET_USER_FUNC(GetScreenSaveActive);
103     GET_USER_FUNC(SetScreenSaveActive);
104     GET_USER_FUNC(AcquireClipboard);
105     GET_USER_FUNC(EmptyClipboard);
106     GET_USER_FUNC(SetClipboardData);
107     GET_USER_FUNC(GetClipboardData);
108     GET_USER_FUNC(CountClipboardFormats);
109     GET_USER_FUNC(EnumClipboardFormats);
110     GET_USER_FUNC(IsClipboardFormatAvailable);
111     GET_USER_FUNC(RegisterClipboardFormat);
112     GET_USER_FUNC(GetClipboardFormatName);
113     GET_USER_FUNC(EndClipboardUpdate);
114     GET_USER_FUNC(ResetSelectionOwner);
115     GET_USER_FUNC(ChangeDisplaySettingsExW);
116     GET_USER_FUNC(EnumDisplaySettingsExW);
117     GET_USER_FUNC(CreateWindow);
118     GET_USER_FUNC(DestroyWindow);
119     GET_USER_FUNC(GetDC);
120     GET_USER_FUNC(ForceWindowRaise);
121     GET_USER_FUNC(MsgWaitForMultipleObjectsEx);
122     GET_USER_FUNC(ReleaseDC);
123     GET_USER_FUNC(ScrollDC);
124     GET_USER_FUNC(SetFocus);
125     GET_USER_FUNC(SetParent);
126     GET_USER_FUNC(SetWindowPos);
127     GET_USER_FUNC(SetWindowRgn);
128     GET_USER_FUNC(SetWindowIcon);
129     GET_USER_FUNC(SetWindowStyle);
130     GET_USER_FUNC(SetWindowText);
131     GET_USER_FUNC(ShowWindow);
132     GET_USER_FUNC(SysCommandSizeMove);
133
134     return TRUE;
135 }
136
137
138 /***********************************************************************
139  *           palette_init
140  *
141  * Patch the function pointers in GDI for SelectPalette and RealizePalette
142  */
143 static void palette_init(void)
144 {
145     void **ptr;
146     HMODULE module = GetModuleHandleA( "gdi32" );
147     if (!module)
148     {
149         ERR( "cannot get GDI32 handle\n" );
150         return;
151     }
152     if ((ptr = (void**)GetProcAddress( module, "pfnSelectPalette" )))
153         pfnGDISelectPalette = InterlockedExchangePointer( ptr, SelectPalette );
154     else ERR( "cannot find pfnSelectPalette in GDI32\n" );
155     if ((ptr = (void**)GetProcAddress( module, "pfnRealizePalette" )))
156         pfnGDIRealizePalette = InterlockedExchangePointer( ptr, UserRealizePalette );
157     else ERR( "cannot find pfnRealizePalette in GDI32\n" );
158 }
159
160
161 /***********************************************************************
162  *           USER initialisation routine
163  */
164 static BOOL process_attach(void)
165 {
166     HINSTANCE16 instance;
167
168     /* Create USER heap */
169     if ((instance = LoadLibrary16( "USER.EXE" )) >= 32) USER_HeapSel = instance | 7;
170     else
171     {
172         USER_HeapSel = GlobalAlloc16( GMEM_FIXED, 65536 );
173         LocalInit16( USER_HeapSel, 32, 65534 );
174     }
175
176     /* some Win9x dlls expect keyboard to be loaded */
177     if (GetVersion() & 0x80000000) LoadLibrary16( "keyboard.drv" );
178
179     /* Load the graphics driver */
180     if (!load_driver()) return FALSE;
181
182     /* Initialize system colors and metrics */
183     SYSMETRICS_Init();
184     SYSCOLOR_Init();
185
186     /* Setup palette function pointers */
187     palette_init();
188
189     /* Initialize built-in window classes */
190     CLASS_RegisterBuiltinClasses();
191
192     /* Initialize menus */
193     if (!MENU_Init()) return FALSE;
194
195     /* Initialize message spying */
196     if (!SPY_Init()) return FALSE;
197
198     /* Create message queue of initial thread */
199     InitThreadInput16( 0, 0 );
200
201     /* Create desktop window */
202     if (!WIN_CreateDesktopWindow()) return FALSE;
203
204     /* Initialize keyboard driver */
205     if (USER_Driver.pInitKeyboard) USER_Driver.pInitKeyboard( InputKeyStateTable );
206
207     /* Initialize mouse driver */
208     if (USER_Driver.pInitMouse) USER_Driver.pInitMouse( InputKeyStateTable );
209
210     return TRUE;
211 }
212
213
214 /**********************************************************************
215  *           USER_IsExitingThread
216  */
217 BOOL USER_IsExitingThread( DWORD tid )
218 {
219     return (tid == exiting_thread_id);
220 }
221
222
223 /**********************************************************************
224  *           thread_detach
225  */
226 static void thread_detach(void)
227 {
228     exiting_thread_id = GetCurrentThreadId();
229
230     WDML_NotifyThreadDetach();
231
232     TIMER_RemoveThreadTimers();
233     WIN_DestroyThreadWindows( GetDesktopWindow() );
234     QUEUE_DeleteMsgQueue();
235
236     exiting_thread_id = 0;
237 }
238
239
240 /**********************************************************************
241  *           process_detach
242  */
243 static void process_detach(void)
244 {
245     memset(&USER_Driver, 0, sizeof(USER_Driver));
246     FreeLibrary(graphics_driver);
247 }
248
249 /***********************************************************************
250  *           UserClientDllInitialize  (USER32.@)
251  *
252  * USER dll initialisation routine (exported as UserClientDllInitialize for compatibility).
253  */
254 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
255 {
256     BOOL ret = TRUE;
257     switch(reason)
258     {
259     case DLL_PROCESS_ATTACH:
260         user32_module = inst;
261         ret = process_attach();
262         break;
263     case DLL_PROCESS_DETACH:
264         process_detach();
265         break;
266     case DLL_THREAD_DETACH:
267         thread_detach();
268         break;
269     }
270     return ret;
271 }