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