321 lines
9.1 KiB
C
321 lines
9.1 KiB
C
/*
|
|
* USER initialization code
|
|
*
|
|
* Copyright 2000 Alexandre Julliard
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "wingdi.h"
|
|
#include "winuser.h"
|
|
|
|
#include "controls.h"
|
|
#include "user_private.h"
|
|
#include "win.h"
|
|
#include "wine/unicode.h"
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(graphics);
|
|
|
|
#define DESKTOP_ALL_ACCESS 0x01ff
|
|
|
|
WORD USER_HeapSel = 0; /* USER heap selector */
|
|
HMODULE user32_module = 0;
|
|
|
|
static SYSLEVEL USER_SysLevel;
|
|
static CRITICAL_SECTION_DEBUG critsect_debug =
|
|
{
|
|
0, 0, &USER_SysLevel.crst,
|
|
{ &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
|
|
0, 0, { (DWORD_PTR)(__FILE__ ": USER_SysLevel") }
|
|
};
|
|
static SYSLEVEL USER_SysLevel = { { &critsect_debug, -1, 0, 0, 0, 0 }, 2 };
|
|
|
|
static HPALETTE (WINAPI *pfnGDISelectPalette)( HDC hdc, HPALETTE hpal, WORD bkgnd );
|
|
static UINT (WINAPI *pfnGDIRealizePalette)( HDC hdc );
|
|
static HPALETTE hPrimaryPalette;
|
|
|
|
static DWORD exiting_thread_id;
|
|
|
|
extern void WDML_NotifyThreadDetach(void);
|
|
|
|
|
|
/***********************************************************************
|
|
* USER_Lock
|
|
*/
|
|
void USER_Lock(void)
|
|
{
|
|
_EnterSysLevel( &USER_SysLevel );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* USER_Unlock
|
|
*/
|
|
void USER_Unlock(void)
|
|
{
|
|
_LeaveSysLevel( &USER_SysLevel );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* USER_CheckNotLock
|
|
*
|
|
* Make sure that we don't hold the user lock.
|
|
*/
|
|
void USER_CheckNotLock(void)
|
|
{
|
|
_CheckNotSysLevel( &USER_SysLevel );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* UserSelectPalette (Not a Windows API)
|
|
*/
|
|
static HPALETTE WINAPI UserSelectPalette( HDC hDC, HPALETTE hPal, BOOL bForceBackground )
|
|
{
|
|
WORD wBkgPalette = 1;
|
|
|
|
if (!bForceBackground && (hPal != GetStockObject(DEFAULT_PALETTE)))
|
|
{
|
|
HWND hwnd = WindowFromDC( hDC );
|
|
if (hwnd)
|
|
{
|
|
HWND hForeground = GetForegroundWindow();
|
|
/* set primary palette if it's related to current active */
|
|
if (hForeground == hwnd || IsChild(hForeground,hwnd))
|
|
{
|
|
wBkgPalette = 0;
|
|
hPrimaryPalette = hPal;
|
|
}
|
|
}
|
|
}
|
|
return pfnGDISelectPalette( hDC, hPal, wBkgPalette);
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* UserRealizePalette (USER32.@)
|
|
*/
|
|
UINT WINAPI UserRealizePalette( HDC hDC )
|
|
{
|
|
UINT realized = pfnGDIRealizePalette( hDC );
|
|
|
|
/* do not send anything if no colors were changed */
|
|
if (realized && GetCurrentObject( hDC, OBJ_PAL ) == hPrimaryPalette)
|
|
{
|
|
/* send palette change notification */
|
|
HWND hWnd = WindowFromDC( hDC );
|
|
if (hWnd) SendMessageTimeoutW( HWND_BROADCAST, WM_PALETTECHANGED, (WPARAM)hWnd, 0,
|
|
SMTO_ABORTIFHUNG, 2000, NULL );
|
|
}
|
|
return realized;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* palette_init
|
|
*
|
|
* Patch the function pointers in GDI for SelectPalette and RealizePalette
|
|
*/
|
|
static void palette_init(void)
|
|
{
|
|
void **ptr;
|
|
HMODULE module = GetModuleHandleA( "gdi32" );
|
|
if (!module)
|
|
{
|
|
ERR( "cannot get GDI32 handle\n" );
|
|
return;
|
|
}
|
|
if ((ptr = (void**)GetProcAddress( module, "pfnSelectPalette" )))
|
|
pfnGDISelectPalette = InterlockedExchangePointer( ptr, UserSelectPalette );
|
|
else ERR( "cannot find pfnSelectPalette in GDI32\n" );
|
|
if ((ptr = (void**)GetProcAddress( module, "pfnRealizePalette" )))
|
|
pfnGDIRealizePalette = InterlockedExchangePointer( ptr, UserRealizePalette );
|
|
else ERR( "cannot find pfnRealizePalette in GDI32\n" );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* winstation_init
|
|
*
|
|
* Connect to the process window station and desktop.
|
|
*/
|
|
static void winstation_init(void)
|
|
{
|
|
static const WCHAR WinSta0[] = {'W','i','n','S','t','a','0',0};
|
|
static const WCHAR Default[] = {'D','e','f','a','u','l','t',0};
|
|
|
|
STARTUPINFOW info;
|
|
WCHAR *winstation = NULL, *desktop = NULL, *buffer = NULL;
|
|
HANDLE handle;
|
|
|
|
GetStartupInfoW( &info );
|
|
if (info.lpDesktop && *info.lpDesktop)
|
|
{
|
|
buffer = HeapAlloc( GetProcessHeap(), 0, (strlenW(info.lpDesktop) + 1) * sizeof(WCHAR) );
|
|
strcpyW( buffer, info.lpDesktop );
|
|
if ((desktop = strchrW( buffer, '\\' )))
|
|
{
|
|
*desktop++ = 0;
|
|
winstation = buffer;
|
|
}
|
|
else desktop = buffer;
|
|
}
|
|
|
|
/* set winstation if explicitly specified, or if we don't have one yet */
|
|
if (buffer || !GetProcessWindowStation())
|
|
{
|
|
handle = CreateWindowStationW( winstation ? winstation : WinSta0, 0, WINSTA_ALL_ACCESS, NULL );
|
|
if (handle) SetProcessWindowStation( handle );
|
|
}
|
|
if (buffer || !GetThreadDesktop( GetCurrentThreadId() ))
|
|
{
|
|
handle = CreateDesktopW( desktop ? desktop : Default, NULL, NULL, 0, DESKTOP_ALL_ACCESS, NULL );
|
|
if (handle) SetThreadDesktop( handle );
|
|
}
|
|
HeapFree( GetProcessHeap(), 0, buffer );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* USER initialisation routine
|
|
*/
|
|
static BOOL process_attach(void)
|
|
{
|
|
HINSTANCE16 instance;
|
|
|
|
/* Create USER heap */
|
|
if ((instance = LoadLibrary16( "USER.EXE" )) >= 32) USER_HeapSel = instance | 7;
|
|
else
|
|
{
|
|
USER_HeapSel = GlobalAlloc16( GMEM_FIXED, 65536 );
|
|
LocalInit16( USER_HeapSel, 32, 65534 );
|
|
}
|
|
|
|
/* some Win9x dlls expect keyboard to be loaded */
|
|
if (GetVersion() & 0x80000000) LoadLibrary16( "keyboard.drv" );
|
|
|
|
winstation_init();
|
|
|
|
/* Initialize system colors and metrics */
|
|
SYSPARAMS_Init();
|
|
|
|
/* Setup palette function pointers */
|
|
palette_init();
|
|
|
|
/* Initialize built-in window classes */
|
|
CLASS_RegisterBuiltinClasses();
|
|
|
|
/* Initialize message spying */
|
|
if (!SPY_Init()) return FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* USER_IsExitingThread
|
|
*/
|
|
BOOL USER_IsExitingThread( DWORD tid )
|
|
{
|
|
return (tid == exiting_thread_id);
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* thread_detach
|
|
*/
|
|
static void thread_detach(void)
|
|
{
|
|
exiting_thread_id = GetCurrentThreadId();
|
|
|
|
WDML_NotifyThreadDetach();
|
|
|
|
WIN_DestroyThreadWindows( get_user_thread_info()->desktop );
|
|
CloseHandle( get_user_thread_info()->server_queue );
|
|
|
|
exiting_thread_id = 0;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* UserClientDllInitialize (USER32.@)
|
|
*
|
|
* USER dll initialisation routine (exported as UserClientDllInitialize for compatibility).
|
|
*/
|
|
BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
|
|
{
|
|
BOOL ret = TRUE;
|
|
switch(reason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
user32_module = inst;
|
|
ret = process_attach();
|
|
break;
|
|
case DLL_THREAD_DETACH:
|
|
thread_detach();
|
|
break;
|
|
case DLL_PROCESS_DETACH:
|
|
USER_unload_driver();
|
|
break;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* ExitWindowsEx (USER32.@)
|
|
*/
|
|
BOOL WINAPI ExitWindowsEx( UINT flags, DWORD reason )
|
|
{
|
|
static const WCHAR winebootW[] = { '\\','w','i','n','e','b','o','o','t','.','e','x','e',0 };
|
|
static const WCHAR killW[] = { ' ','-','-','k','i','l','l',0 };
|
|
static const WCHAR end_sessionW[] = { ' ','-','-','e','n','d','-','s','e','s','s','i','o','n',0 };
|
|
static const WCHAR forceW[] = { ' ','-','-','f','o','r','c','e',0 };
|
|
static const WCHAR shutdownW[] = { ' ','-','-','s','h','u','t','d','o','w','n',0 };
|
|
|
|
WCHAR cmdline[MAX_PATH + 64];
|
|
PROCESS_INFORMATION pi;
|
|
STARTUPINFOW si;
|
|
|
|
GetSystemDirectoryW( cmdline, MAX_PATH );
|
|
lstrcatW( cmdline, winebootW );
|
|
|
|
if (flags & EWX_FORCE) lstrcatW( cmdline, killW );
|
|
else
|
|
{
|
|
lstrcatW( cmdline, end_sessionW );
|
|
if (flags & EWX_FORCEIFHUNG) lstrcatW( cmdline, forceW );
|
|
}
|
|
if (!(flags & EWX_REBOOT)) lstrcatW( cmdline, shutdownW );
|
|
|
|
memset( &si, 0, sizeof si );
|
|
si.cb = sizeof si;
|
|
if (!CreateProcessW( NULL, cmdline, NULL, NULL, FALSE, DETACHED_PROCESS, NULL, NULL, &si, &pi ))
|
|
{
|
|
ERR( "Failed to run %s\n", debugstr_w(cmdline) );
|
|
return FALSE;
|
|
}
|
|
CloseHandle( pi.hProcess );
|
|
CloseHandle( pi.hThread );
|
|
return TRUE;
|
|
}
|