226 lines
7.2 KiB
C++
226 lines
7.2 KiB
C++
// --------------------------------------------------------------------------
|
|
// 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();
|
|
}
|
|
}
|
|
}
|
|
|