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 (user_section.OwningThread == ULongToHandle(GetCurrentThreadId()) && user_section.RecursionCount)
  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
 267
 268 /***********************************************************************
 269  *           USER initialisation routine
 270  */
 271 static BOOL process_attach(void)
 272 {
 273     winstation_init();
 274
 275     /* Initialize system colors and metrics */
 276     SYSPARAMS_Init();
 277
 278     /* Setup palette function pointers */
 279     palette_init();
 280
 281     /* Initialize built-in window classes */
 282     CLASS_RegisterBuiltinClasses();
 283
 284     return TRUE;
 285 }
 286
 287
 288 /**********************************************************************
 289  *           USER_IsExitingThread
 290  */
 291 BOOL USER_IsExitingThread( DWORD tid )
 292 {
 293     return (tid == exiting_thread_id);
 294 }
 295
 296
 297 /**********************************************************************
 298  *           thread_detach
 299  */
 300 static void thread_detach(void)
 301 {
 302     struct user_thread_info *thread_info = get_user_thread_info();
 303
 304     exiting_thread_id = GetCurrentThreadId();
 305
 306     WDML_NotifyThreadDetach();
 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     BOOL ret = TRUE;
 327     switch(reason)
 328     {
 329     case DLL_PROCESS_ATTACH:
 330         user32_module = inst;
 331         ret = process_attach();
 332         break;
 333     case DLL_THREAD_DETACH:
 334         thread_detach();
 335         break;
 336     case DLL_PROCESS_DETACH:
 337         USER_unload_driver();
 338         DeleteCriticalSection(&user_section);
 339         break;
 340     }
 341     return ret;
 342 }
 343
 344
 345 /***********************************************************************
 346  *              ExitWindowsEx (USER32.@)
 347  */
 348 BOOL WINAPI ExitWindowsEx( UINT flags, DWORD reason )
 349 {
 350     static const WCHAR winebootW[]    = { '\\','w','i','n','e','b','o','o','t','.','e','x','e',0 };
 351     static const WCHAR killW[]        = { ' ','-','-','k','i','l','l',0 };
 352     static const WCHAR end_sessionW[] = { ' ','-','-','e','n','d','-','s','e','s','s','i','o','n',0 };
 353     static const WCHAR forceW[]       = { ' ','-','-','f','o','r','c','e',0 };
 354     static const WCHAR shutdownW[]    = { ' ','-','-','s','h','u','t','d','o','w','n',0 };
 355
 356     WCHAR app[MAX_PATH];
 357     WCHAR cmdline[MAX_PATH + 64];
 358     PROCESS_INFORMATION pi;
 359     STARTUPINFOW si;
 360     void *redir;
 361
 362     GetSystemDirectoryW( app, MAX_PATH - sizeof(winebootW)/sizeof(WCHAR) );
 363     strcatW( app, winebootW );
 364     strcpyW( cmdline, app );
 365
 366     if (flags & EWX_FORCE) lstrcatW( cmdline, killW );
 367     else
 368     {
 369         lstrcatW( cmdline, end_sessionW );
 370         if (flags & EWX_FORCEIFHUNG) lstrcatW( cmdline, forceW );
 371     }
 372     if (!(flags & EWX_REBOOT)) lstrcatW( cmdline, shutdownW );
 373
 374     memset( &si, 0, sizeof si );
 375     si.cb = sizeof si;
 376     Wow64DisableWow64FsRedirection( &redir );
 377     if (!CreateProcessW( app, cmdline, NULL, NULL, FALSE, DETACHED_PROCESS, NULL, NULL, &si, &pi ))
 378     {
 379         Wow64RevertWow64FsRedirection( redir );
 380         ERR( "Failed to run %s\n", debugstr_w(cmdline) );
 381         return FALSE;
 382     }
 383     Wow64RevertWow64FsRedirection( redir );
 384     CloseHandle( pi.hProcess );
 385     CloseHandle( pi.hThread );
 386     return TRUE;
 387 }
 388
 389 /***********************************************************************
 390  *              LockWorkStation (USER32.@)
 391  */
 392 BOOL WINAPI LockWorkStation(void)
 393 {
 394     TRACE(": stub\n");
 395     SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
 396     return FALSE;
 397 }
 398
 399 /***********************************************************************
 400  *              RegisterServicesProcess (USER32.@)
 401  */
 402 int WINAPI RegisterServicesProcess(DWORD ServicesProcessId)
 403 {
 404     FIXME("(0x%x): stub\n", ServicesProcessId);
 405     return 0;
 406 }