windows-nt/Source/XPSP1/NT/shell/services/themesrv/themeservice.cpp

226 lines
7 KiB
C++
Raw Permalink Normal View History

2020-09-26 03:20:57 -05:00
// --------------------------------------------------------------------------
// Module Name: ThemeService.cpp
//
// Copyright (c) 2001, Microsoft Corporation
//
// This file contains functions that are called from the shell services DLL
// to interact with the theme service.
//
// History: 2001-01-02 vtan created
// --------------------------------------------------------------------------
#include "StandardHeader.h"
#include "ThemeService.h"
#include <shlwapi.h>
#include <shlwapip.h>
#include "Resource.h"
#include "ThemeManagerAPIRequest.h"
#include "ThemeManagerService.h"
#include "ThemeServerClient.h"
extern HINSTANCE g_hInstance;
// --------------------------------------------------------------------------
// CThemeService::Main
//
// Arguments: See the platform SDK under DllMain.
//
// Returns: See the platform SDK under DllMain.
//
// Purpose: Performs initialization and clean up on process attach and
// detach. Not interested in anything else.
//
// History: 2000-10-12 vtan created
// 2001-01-02 vtan scoped to a C++ class
// --------------------------------------------------------------------------
NTSTATUS CThemeService::Main (DWORD dwReason)
{
NTSTATUS status;
switch (dwReason)
{
case DLL_PROCESS_ATTACH:
status = CThemeManagerAPIRequest::StaticInitialize();
if (NT_SUCCESS(status))
{
status = CThemeServerClient::StaticInitialize();
}
break;
case DLL_PROCESS_DETACH:
TSTATUS(CThemeServerClient::StaticTerminate());
TSTATUS(CThemeManagerAPIRequest::StaticTerminate());
status = STATUS_SUCCESS;
break;
default:
status = STATUS_SUCCESS;
break;
}
return(status);
}
// --------------------------------------------------------------------------
// CThemeService::DllRegisterServer
//
// Arguments: <none>
//
// Returns: NTSTATUS
//
// Purpose: Register entry point to allow the theme server to install
// itself into the registry.
//
// History: 2000-11-28 vtan created
// 2001-01-02 vtan scoped to a C++ class
// --------------------------------------------------------------------------
NTSTATUS CThemeService::RegisterServer (void)
{
NTSTATUS status;
status = STATUS_SUCCESS;
// In upgrade cases, remove our old name service from both 32 & 64 bit systems
(NTSTATUS)CService::Remove(TEXT("ThemeService"));
#ifdef _WIN64
// In upgrade cases for 64-bit, remove our current name service
(NTSTATUS)CService::Remove(CThemeManagerService::GetName());
#else
// This is 32-bit only. Check if this is REALLY 32-bit and not 32-bit on 64-bit.
if (!IsOS(OS_WOW6432))
{
// Prepare the failure actions, in order to get the service to restart automatically
SC_ACTION ac[3];
ac[0].Type = SC_ACTION_RESTART;
ac[0].Delay = 60000;
ac[1].Type = SC_ACTION_RESTART;
ac[1].Delay = 60000;
ac[2].Type = SC_ACTION_NONE;
ac[2].Delay = 0;
SERVICE_FAILURE_ACTIONS sf;
sf.dwResetPeriod = 86400;
sf.lpRebootMsg = NULL;
sf.lpCommand = NULL;
sf.cActions = 3;
sf.lpsaActions = ac;
// Now install the new service by name.
status = CService::Install(CThemeManagerService::GetName(),
TEXT("%SystemRoot%\\System32\\svchost.exe -k netsvcs"),
TEXT("UIGroup"),
NULL,
TEXT("shsvcs.dll"),
NULL,
TEXT("netsvcs"),
TEXT("ThemeServiceMain"),
SERVICE_AUTO_START,
g_hInstance,
IDS_THEMESERVER_DISPLAYNAME,
IDS_THEMESERVER_DESCRIPTION,
&sf);
}
#endif
return(status);
}
// --------------------------------------------------------------------------
// CThemeService::DllUnregisterServer
//
// Arguments: <none>
//
// Returns: NTSTATUS
//
// Purpose: Unregister entry point to allow the theme server to uninstall
// itself from the registry.
//
// History: 2000-11-28 vtan created
// 2001-01-02 vtan scoped to a C++ class
// --------------------------------------------------------------------------
NTSTATUS CThemeService::UnregisterServer (void)
{
// Ignore any "not found", etc errors.
(NTSTATUS)CService::Remove(CThemeManagerService::GetName());
return(STATUS_SUCCESS);
}
// --------------------------------------------------------------------------
// ::ThemeServiceMain
//
// Arguments: dwArgc = Number of arguments.
// lpszArgv = Argument array.
//
// Returns: <none>
//
// Purpose: ServiceMain entry point for theme server.
//
// History: 2000-11-28 vtan created
// 2001-01-02 vtan scoped to the theme service
// --------------------------------------------------------------------------
void WINAPI ThemeServiceMain (DWORD dwArgc, LPWSTR *lpszArgv)
{
UNREFERENCED_PARAMETER(dwArgc);
UNREFERENCED_PARAMETER(lpszArgv);
NTSTATUS status;
status = CThemeManagerAPIRequest::InitializeServerChangeNumber();
if (NT_SUCCESS(status))
{
CThemeManagerAPIServer *pThemeManagerAPIServer;
// Bring in shell32.dll NOW so that when CheckThemeSignature is called
// and it tries to use SHGetFolderPath it won't cause shell32.dll to be
// brought in while impersonating a user. This will cause advapi32.dll
// to leak a key to the user's hive that won't get cleaned up at logoff.
CModule hModule(TEXT("shell32.dll"));
pThemeManagerAPIServer = new CThemeManagerAPIServer;
if (pThemeManagerAPIServer != NULL)
{
CAPIConnection *pAPIConnection;
pAPIConnection = new CAPIConnection(pThemeManagerAPIServer);
if (pAPIConnection != NULL)
{
CThemeManagerService *pThemeManagerService;
pThemeManagerService = new CThemeManagerService(pAPIConnection, pThemeManagerAPIServer);
if (pThemeManagerService != NULL)
{
CThemeManagerSessionData::SetAPIConnection(pAPIConnection);
TSTATUS(CThemeManagerAPIRequest::ArrayInitialize());
pThemeManagerService->Start();
pThemeManagerService->Release();
TSTATUS(CThemeManagerAPIRequest::ArrayTerminate());
CThemeManagerSessionData::ReleaseAPIConnection();
}
pAPIConnection->Release();
}
pThemeManagerAPIServer->Release();
}
}
}