dlls/user32/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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "wingdi.h"
  27 #include "winuser.h"
  28
  29 #include "controls.h"
  30 #include "user_private.h"
  31 #include "win.h"
  32 #include "wine/unicode.h"
  33 #include "wine/debug.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(graphics);
  36
  37 #define DESKTOP_ALL_ACCESS 0x01ff
  38
  39 HMODULE user32_module = 0;
  40
  41 static CRITICAL_SECTION user_section;
  42 static CRITICAL_SECTION_DEBUG critsect_debug =
  43 {
  44     0, 0, &user_section,
  45     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
  46       0, 0, { (DWORD_PTR)(__FILE__ ": user_section") }
  47 };
  48 static CRITICAL_SECTION user_section = { &critsect_debug, -1, 0, 0, 0, 0 };
  49
  50 static HPALETTE (WINAPI *pfnGDISelectPalette)( HDC hdc, HPALETTE hpal, WORD bkgnd );
  51 static UINT (WINAPI *pfnGDIRealizePalette)( HDC hdc );
  52 static HPALETTE hPrimaryPalette;
  53
  54 static DWORD exiting_thread_id;
  55
  56 extern void WDML_NotifyThreadDetach(void);
  57
  58
  59 /***********************************************************************
  60  *           USER_Lock
  61  */
  62 void USER_Lock(void)
  63 {
  64     EnterCriticalSection( &user_section );
  65 }
  66
  67
  68 /***********************************************************************
  69  *           USER_Unlock
  70  */
  71 void USER_Unlock(void)
  72 {
  73     LeaveCriticalSection( &user_section );
  74 }
  75
  76
  77 /***********************************************************************
  78  *           USER_CheckNotLock
  79  *
  80  * Make sure that we don't hold the user lock.
  81  */
  82 void USER_CheckNotLock(void)
  83 {
  84     if (RtlIsCriticalSectionLockedByThread(&user_section))
  85     {
  86         ERR( "BUG: holding USER lock\n" );
  87         DebugBreak();
  88     }
  89 }
  90
  91
  92 /***********************************************************************
  93  *              UserSelectPalette (Not a Windows API)
  94  */
  95 static HPALETTE WINAPI UserSelectPalette( HDC hDC, HPALETTE hPal, BOOL bForceBackground )
  96 {
  97     WORD wBkgPalette = 1;
  98
  99     if (!bForceBackground && (hPal != GetStockObject(DEFAULT_PALETTE)))
 100     {
 101         HWND hwnd = WindowFromDC( hDC );
 102         if (hwnd)
 103         {
 104             HWND hForeground = GetForegroundWindow();
 105             /* set primary palette if it's related to current active */
 106             if (hForeground == hwnd || IsChild(hForeground,hwnd))
 107             {
 108                 wBkgPalette = 0;
 109                 hPrimaryPalette = hPal;
 110             }
 111         }
 112     }
 113     return pfnGDISelectPalette( hDC, hPal, wBkgPalette);
 114 }
 115
 116
 117 /***********************************************************************
 118  *              UserRealizePalette (USER32.@)
 119  */
 120 UINT WINAPI UserRealizePalette( HDC hDC )
 121 {
 122     UINT realized = pfnGDIRealizePalette( hDC );
 123
 124     /* do not send anything if no colors were changed */
 125     if (realized && GetCurrentObject( hDC, OBJ_PAL ) == hPrimaryPalette)
 126     {
 127         /* send palette change notification */
 128         HWND hWnd = WindowFromDC( hDC );
 129         if (hWnd) SendMessageTimeoutW( HWND_BROADCAST, WM_PALETTECHANGED, (WPARAM)hWnd, 0,
 130                                        SMTO_ABORTIFHUNG, 2000, NULL );
 131     }
 132     return realized;
 133 }
 134
 135
 136 /***********************************************************************
 137  *           palette_init
 138  *
 139  * Patch the function pointers in GDI for SelectPalette and RealizePalette
 140  */
 141 static void palette_init(void)
 142 {
 143     void **ptr;
 144     HMODULE module = GetModuleHandleA( "gdi32" );
 145     if (!module)
 146     {
 147         ERR( "cannot get GDI32 handle\n" );
 148         return;
 149     }
 150     if ((ptr = (void**)GetProcAddress( module, "pfnSelectPalette" )))
 151         pfnGDISelectPalette = InterlockedExchangePointer( ptr, UserSelectPalette );
 152     else ERR( "cannot find pfnSelectPalette in GDI32\n" );
 153     if ((ptr = (void**)GetProcAddress( module, "pfnRealizePalette" )))
 154         pfnGDIRealizePalette = InterlockedExchangePointer( ptr, UserRealizePalette );
 155     else ERR( "cannot find pfnRealizePalette in GDI32\n" );
 156 }
 157
 158
 159 /***********************************************************************
 160  *           get_default_desktop
 161  *
 162  * Get the name of the desktop to use for this app if not specified explicitly.
 163  */
 164 static const WCHAR *get_default_desktop(void)
 165 {
 166     static const WCHAR defaultW[] = {'D','e','f','a','u','l','t',0};
 167     static const WCHAR desktopW[] = {'D','e','s','k','t','o','p',0};
 168     static const WCHAR explorerW[] = {'\\','E','x','p','l','o','r','e','r',0};
 169     static const WCHAR app_defaultsW[] = {'S','o','f','t','w','a','r','e','\\',
 170                                           'W','i','n','e','\\',
 171                                           'A','p','p','D','e','f','a','u','l','t','s',0};
 172     static WCHAR buffer[MAX_PATH + sizeof(explorerW)/sizeof(WCHAR)];
 173     WCHAR *p, *appname = buffer;
 174     const WCHAR *ret = defaultW;
 175     DWORD len;
 176     HKEY tmpkey, appkey;
 177
 178     len = (GetModuleFileNameW( 0, buffer, MAX_PATH ));
 179     if (!len || len >= MAX_PATH) return ret;
 180     if ((p = strrchrW( appname, '/' ))) appname = p + 1;
 181     if ((p = strrchrW( appname, '\\' ))) appname = p + 1;
 182     p = appname + strlenW(appname);
 183     strcpyW( p, explorerW );
 184
 185     /* @@ Wine registry key: HKCU\Software\Wine\AppDefaults\app.exe\Explorer */
 186     if (!RegOpenKeyW( HKEY_CURRENT_USER, app_defaultsW, &tmpkey ))
 187     {
 188         if (RegOpenKeyW( tmpkey, appname, &appkey )) appkey = 0;
 189         RegCloseKey( tmpkey );
 190         if (appkey)
 191         {
 192             len = sizeof(buffer);
 193             if (!RegQueryValueExW( appkey, desktopW, 0, NULL, (LPBYTE)buffer, &len )) ret = buffer;
 194             RegCloseKey( appkey );
 195             if (ret && strcmpiW( ret, defaultW )) return ret;
 196             ret = defaultW;
 197         }
 198     }
 199
 200     memcpy( buffer, app_defaultsW, 13 * sizeof(WCHAR) );  /* copy only software\\wine */
 201     strcpyW( buffer + 13, explorerW );
 202
 203     /* @@ Wine registry key: HKCU\Software\Wine\Explorer */
 204     if (!RegOpenKeyW( HKEY_CURRENT_USER, buffer, &appkey ))
 205     {
 206         len = sizeof(buffer);
 207         if (!RegQueryValueExW( appkey, desktopW, 0, NULL, (LPBYTE)buffer, &len )) ret = buffer;
 208         RegCloseKey( appkey );
 209     }
 210     return ret;
 211 }
 212
 213
 214 /***********************************************************************
 215  *           winstation_init
 216  *
 217  * Connect to the process window station and desktop.
 218  */
 219 static void winstation_init(void)
 220 {
 221     static const WCHAR WinSta0[] = {'W','i','n','S','t','a','0',0};
 222
 223     STARTUPINFOW info;
 224     WCHAR *winstation = NULL, *desktop = NULL, *buffer = NULL;
 225     HANDLE handle;
 226
 227     GetStartupInfoW( &info );
 228     if (info.lpDesktop && *info.lpDesktop)
 229     {
 230         buffer = HeapAlloc( GetProcessHeap(), 0, (strlenW(info.lpDesktop) + 1) * sizeof(WCHAR) );
 231         strcpyW( buffer, info.lpDesktop );
 232         if ((desktop = strchrW( buffer, '\\' )))
 233         {
 234             *desktop++ = 0;
 235             winstation = buffer;
 236         }
 237         else desktop = buffer;
 238     }
 239
 240     /* set winstation if explicitly specified, or if we don't have one yet */
 241     if (buffer || !GetProcessWindowStation())
 242     {
 243         handle = CreateWindowStationW( winstation ? winstation : WinSta0, 0, WINSTA_ALL_ACCESS, NULL );
 244         if (handle)
 245         {
 246             SetProcessWindowStation( handle );
 247             /* only WinSta0 is visible */
 248             if (!winstation || !strcmpiW( winstation, WinSta0 ))
 249             {
 250                 USEROBJECTFLAGS flags;
 251                 flags.fInherit  = FALSE;
 252                 flags.fReserved = FALSE;
 253                 flags.dwFlags   = WSF_VISIBLE;
 254                 SetUserObjectInformationW( handle, UOI_FLAGS, &flags, sizeof(flags) );
 255             }
 256         }
 257     }
 258     if (buffer || !GetThreadDesktop( GetCurrentThreadId() ))
 259     {
 260         handle = CreateDesktopW( desktop ? desktop : get_default_desktop(),
 261                                  NULL, NULL, 0, DESKTOP_ALL_ACCESS, NULL );
 262         if (handle) SetThreadDesktop( handle );
 263     }
 264     HeapFree( GetProcessHeap(), 0, buffer );
 265
 266     register_desktop_class();
 267 }
 268
 269
 270 /***********************************************************************
 271  *           USER initialisation routine
 272  */
 273 static BOOL process_attach(void)
 274 {
 275     winstation_init();
 276
 277     /* Initialize system colors and metrics */
 278     SYSPARAMS_Init();
 279
 280     /* Setup palette function pointers */
 281     palette_init();
 282
 283     return TRUE;
 284 }
 285
 286
 287 /**********************************************************************
 288  *           USER_IsExitingThread
 289  */
 290 BOOL USER_IsExitingThread( DWORD tid )
 291 {
 292     return (tid == exiting_thread_id);
 293 }
 294
 295
 296 /**********************************************************************
 297  *           thread_detach
 298  */
 299 static void thread_detach(void)
 300 {
 301     struct user_thread_info *thread_info = get_user_thread_info();
 302
 303     exiting_thread_id = GetCurrentThreadId();
 304
 305     WDML_NotifyThreadDetach();
 306     USER_Driver->pThreadDetach();
 307
 308     if (thread_info->top_window) WIN_DestroyThreadWindows( thread_info->top_window );
 309     if (thread_info->msg_window) WIN_DestroyThreadWindows( thread_info->msg_window );
 310     CloseHandle( thread_info->server_queue );
 311     HeapFree( GetProcessHeap(), 0, thread_info->wmchar_data );
 312     HeapFree( GetProcessHeap(), 0, thread_info->key_state );
 313     HeapFree( GetProcessHeap(), 0, thread_info->rawinput );
 314
 315     exiting_thread_id = 0;
 316 }
 317
 318
 319 /***********************************************************************
 320  *           UserClientDllInitialize  (USER32.@)
 321  *
 322  * USER dll initialisation routine (exported as UserClientDllInitialize for compatibility).
 323  */
 324 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
 325 {
 326     static const WCHAR imm32_dllW[] = {'i','m','m','3','2','.','d','l','l',0};
 327     static HMODULE imm32_module;
 328     BOOL ret = TRUE;
 329
 330     switch(reason)
 331     {
 332     case DLL_PROCESS_ATTACH:
 333         user32_module = inst;
 334         ret = process_attach();
 335         if(ret)
 336             imm32_module = LoadLibraryW(imm32_dllW);
 337         break;
 338     case DLL_THREAD_DETACH:
 339         thread_detach();
 340         break;
 341     case DLL_PROCESS_DETACH:
 342         USER_unload_driver();
 343         FreeLibrary(imm32_module);
 344         DeleteCriticalSection(&user_section);
 345         break;
 346     }
 347     return ret;
 348 }
 349
 350
 351 /***********************************************************************
 352  *              ExitWindowsEx (USER32.@)
 353  */
 354 BOOL WINAPI ExitWindowsEx( UINT flags, DWORD reason )
 355 {
 356     static const WCHAR winebootW[]    = { '\\','w','i','n','e','b','o','o','t','.','e','x','e',0 };
 357     static const WCHAR killW[]        = { ' ','-','-','k','i','l','l',0 };
 358     static const WCHAR end_sessionW[] = { ' ','-','-','e','n','d','-','s','e','s','s','i','o','n',0 };
 359     static const WCHAR forceW[]       = { ' ','-','-','f','o','r','c','e',0 };
 360     static const WCHAR shutdownW[]    = { ' ','-','-','s','h','u','t','d','o','w','n',0 };
 361
 362     WCHAR app[MAX_PATH];
 363     WCHAR cmdline[MAX_PATH + 64];
 364     PROCESS_INFORMATION pi;
 365     STARTUPINFOW si;
 366     void *redir;
 367
 368     GetSystemDirectoryW( app, MAX_PATH - sizeof(winebootW)/sizeof(WCHAR) );
 369     strcatW( app, winebootW );
 370     strcpyW( cmdline, app );
 371
 372     if (flags & EWX_FORCE) lstrcatW( cmdline, killW );
 373     else
 374     {
 375         lstrcatW( cmdline, end_sessionW );
 376         if (flags & EWX_FORCEIFHUNG) lstrcatW( cmdline, forceW );
 377     }
 378     if (!(flags & EWX_REBOOT)) lstrcatW( cmdline, shutdownW );
 379
 380     memset( &si, 0, sizeof si );
 381     si.cb = sizeof si;
 382     Wow64DisableWow64FsRedirection( &redir );
 383     if (!CreateProcessW( app, cmdline, NULL, NULL, FALSE, DETACHED_PROCESS, NULL, NULL, &si, &pi ))
 384     {
 385         Wow64RevertWow64FsRedirection( redir );
 386         ERR( "Failed to run %s\n", debugstr_w(cmdline) );
 387         return FALSE;
 388     }
 389     Wow64RevertWow64FsRedirection( redir );
 390     CloseHandle( pi.hProcess );
 391     CloseHandle( pi.hThread );
 392     return TRUE;
 393 }
 394
 395 /***********************************************************************
 396  *              LockWorkStation (USER32.@)
 397  */
 398 BOOL WINAPI LockWorkStation(void)
 399 {
 400     TRACE(": stub\n");
 401     SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
 402     return FALSE;
 403 }
 404
 405 /***********************************************************************
 406  *              RegisterServicesProcess (USER32.@)
 407  */
 408 int WINAPI RegisterServicesProcess(DWORD ServicesProcessId)
 409 {
 410     FIXME("(0x%x): stub\n", ServicesProcessId);
 411     return 0;
 412 }
 413
 414 /***********************************************************************
 415  *              ShutdownBlockReasonCreate (USER32.@)
 416  */
 417 BOOL WINAPI ShutdownBlockReasonCreate(HWND hwnd, LPCWSTR reason)
 418 {
 419     FIXME("(%p, %s): stub\n", hwnd, debugstr_w(reason));
 420     SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
 421     return FALSE;
 422 }
 423
 424 /***********************************************************************
 425  *              ShutdownBlockReasonDestroy (USER32.@)
 426  */
 427 BOOL WINAPI ShutdownBlockReasonDestroy(HWND hwnd)
 428 {
 429     FIXME("(%p): stub\n", hwnd);
 430     SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
 431     return FALSE;
 432 }