archive
/
windows-server-2003
mirror of https://github.com/selfrender/Windows-Server-2003
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
4 Commits
1 Branch
0 Tags
1.5 GiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-server-2003/admin/pchealth/helpctr/shell/core/hc.cpp

511 lines
16 KiB
Raw Permalink Blame History

/******************************************************************************
Copyright (c) 1999 Microsoft Corporation
Module Name:
HC.cpp
Abstract:
This file contains the start-up code to create and initialize the Help Center.
Revision History:
Davide Massarenti (Dmassare) 08/07/99
created
******************************************************************************/
#include "stdafx.h"
#include <DataCollection.h>
#include "resource.h"
#include <initguid.h>
#include <HelpServiceTypeLib_i.c>
#include <HelpCenterTypeLib_i.c>
#include <NameSpace_Impl.h>
#include "msscript.h"
/////////////////////////////////////////////////////////////////////////////
static const WCHAR MARSCORE_DLL [] = L"pchshell.dll";
static const WCHAR HCP_PROTOCOL [] = L"HCP";
static const WCHAR MSITS_PROTOCOL[] = L"MS-ITS";
static const WCHAR HELPSVC_REGISTRATION[] = HC_ROOT_HELPSVC_BINARIES L"\\HelpSvc.exe /svchost netsvcs /regserver";
/////////////////////////////////////////////////////////////////////////////
EXTERN_C const CLSID CLSID_JavaScript = { 0xF414C260, 0x6AC0, 0x11CF, { 0xB6, 0xD1, 0x00, 0xAA, 0x00, 0xBB, 0xBB, 0x58 } };
/////////////////////////////////////////////////////////////////////////////
// From "mars.h"
typedef int (APIENTRY *PFNMARSMAIN)(HICON hAppIcon,
LPWSTR lpAppTitle,
LPWSTR lpCmdLine,
int nCmdShow);
/////////////////////////////////////////////////////////////////////////////
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_PCHBootstrapper , CPCHBootstrapper )
OBJECT_ENTRY(CLSID_PCHHelpViewerWrapper, CPCHHelpViewerWrapper )
OBJECT_ENTRY(CLSID_PCHToolBar , CPCHToolBar )
OBJECT_ENTRY(CLSID_PCHProgressBar , CPCHProgressBar )
OBJECT_ENTRY(CLSID_PCHJavaScriptWrapper, CPCHScriptWrapper_ClientSide< &CLSID_JavaScript >)
OBJECT_ENTRY(CLSID_PCHVBScriptWrapper , CPCHScriptWrapper_ClientSide< &CLSID_VBScript >)
END_OBJECT_MAP()
/////////////////////////////////////////////////////////////////////////////
#define DEBUG_REGKEY HC_REGISTRY_HELPCTR L"\\Debug"
#define DEBUG_BREAKONSTART L"BREAKONSTART"
#define DEBUG_BLOCKERRORS L"BLOCKERRORS"
#define DEBUG_CONTEXTMENU L"CONTEXTMENU"
#define DEBUG_BUILDTREE L"BUILDTREE"
#define DEBUG_FORCESTYLE L"FORCESTYLE"
#define DEBUG_FORCERTL L"FORCERTL"
bool g_Debug_BLOCKERRORS = true;
bool g_Debug_CONTEXTMENU = false;
bool g_Debug_BUILDTREE = false;
WCHAR g_Debug_FORCESTYLE[64];
/////////////////////////////////////////////////////////////////////////////
static bool FindMarsDLL( HMODULE& hModule ,
PFNMARSTHREADPROC& pMarsThreadProc )
{
hModule = ::LoadLibraryW( MARSCORE_DLL );
if(hModule)
{
pMarsThreadProc = (PFNMARSTHREADPROC)GetProcAddress( hModule, (LPCSTR)ORD_MARSTHREADPROC );
if(pMarsThreadProc == NULL)
{
MPC::LocalizedMessageBox( IDS_HELPCTR_DLG_ERROR, IDS_HELPCTR_MARSCORE_INCORRECT_VERSION, MB_OK );
return false;
}
}
else
{
MPC::LocalizedMessageBox( IDS_HELPCTR_DLG_ERROR, IDS_HELPCTR_MARSCORE_NOT_FOUND, MB_OK );
return false;
}
return true;
}
/////////////////////////////////////////////////////////////////////////////
static HRESULT RegisterProtocols( /*[in/out]*/ CComPtr<IInternetSession>& pSession ,
/*[in/out]*/ CComPtr<IClassFactory> & pFactoryHCP ,
/*[in/out]*/ CComPtr<IClassFactory> & pFactoryMSITS )
{
__HCP_FUNC_ENTRY( "RegisterProtocols" );
HRESULT hr;
CComPtr<CHCPProtocolInfo> objHCP;
CComPtr<CPCHWrapProtocolInfo> objMSITS;
__MPC_EXIT_IF_METHOD_FAILS(hr, _Module.RegisterClassObjects( CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE ));
__MPC_EXIT_IF_METHOD_FAILS(hr, ::CoInternetGetSession( 0, &pSession, 0 ));
//
// Create an instance of CHCPProtocolInfo, that acts also as a class factory for CHCPProtocol.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::CreateInstance( &objHCP ));
__MPC_EXIT_IF_METHOD_FAILS(hr, objHCP->QueryInterface( IID_IClassFactory, (void **)&pFactoryHCP ));
__MPC_EXIT_IF_METHOD_FAILS(hr, pSession->RegisterNameSpace( pFactoryHCP, CLSID_HCPProtocol, HCP_PROTOCOL, 0, NULL, 0 ));
//
// Create an instance of CPCHWrapProtocolInfo, that acts as a wrapper around the real MS-ITS class factory.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::CreateInstance( &objMSITS ));
__MPC_EXIT_IF_METHOD_FAILS(hr, objMSITS->Init( CLSID_MSITSProtocol ));
__MPC_EXIT_IF_METHOD_FAILS(hr, objMSITS->QueryInterface( IID_IClassFactory, (void **)&pFactoryMSITS ));
__MPC_EXIT_IF_METHOD_FAILS(hr, pSession->RegisterNameSpace( pFactoryMSITS, CLSID_MSITSProtocol, MSITS_PROTOCOL, 0, NULL, 0 ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr);
}
static HRESULT UnregisterProtocols( /*[in/out]*/ CComPtr<IInternetSession>& pSession ,
/*[in/out]*/ CComPtr<IClassFactory> & pFactoryHCP ,
/*[in/out]*/ CComPtr<IClassFactory> & pFactoryMSITS )
{
__HCP_FUNC_ENTRY( "UnregisterProtocols" );
HRESULT hr;
if(pSession)
{
if(pFactoryHCP)
{
__MPC_EXIT_IF_METHOD_FAILS(hr, pSession->UnregisterNameSpace( pFactoryHCP, HCP_PROTOCOL ));
pFactoryHCP.Release();
}
if(pFactoryMSITS)
{
__MPC_EXIT_IF_METHOD_FAILS(hr, pSession->UnregisterNameSpace( pFactoryMSITS, MSITS_PROTOCOL ));
pFactoryMSITS.Release();
}
pSession.Release();
}
__MPC_EXIT_IF_METHOD_FAILS(hr, _Module.RevokeClassObjects());
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr);
}
////////////////////////////////////////////////////////////////////////////////
static HRESULT InitAll()
{
__HCP_FUNC_ENTRY( "InitAll" );
HRESULT hr;
// Fix environment block if necessary
if (!GetEnvironmentVariableW(L"windir", NULL, 0))
{
// %windir% not in the env block (e.g. 16-bit apps), add it
WCHAR szWindir[MAX_PATH + 1];
int nChars = GetWindowsDirectory(szWindir, ARRAYSIZE(szWindir));
if (nChars > 0 && nChars < ARRAYSIZE(szWindir))
{
SetEnvironmentVariable(L"windir", szWindir);
}
}
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::LocalizeInit());
__MPC_EXIT_IF_METHOD_FAILS(hr, OfflineCache::Root ::InitializeSystem( /*fMaster*/false ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHContentStore ::InitializeSystem( /*fMaster*/false ));
__MPC_EXIT_IF_METHOD_FAILS(hr, HyperLinks::Lookup ::InitializeSystem( ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHHelpCenterExternal ::InitializeSystem( ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHFavorites ::InitializeSystem( ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHOptions ::InitializeSystem( ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CHCPProtocolEnvironment::InitializeSystem( ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr);
}
static void CleanAll()
{
CHCPProtocolEnvironment::FinalizeSystem();
CPCHFavorites ::FinalizeSystem();
CPCHOptions ::FinalizeSystem();
CPCHHelpCenterExternal ::FinalizeSystem();
HyperLinks::Lookup ::FinalizeSystem();
CPCHContentStore ::FinalizeSystem();
OfflineCache::Root ::FinalizeSystem();
}
////////////////////////////////////////////////////////////////////////////////
static HRESULT ProcessArguments( int argc ,
LPCWSTR* argv )
{
__HCP_FUNC_ENTRY( "ProcessArguments" );
HRESULT hr;
HMODULE hModule = NULL;
PFNMARSTHREADPROC pMarsThreadProc = NULL;
CComPtr<IInternetSession> pSession;
CComPtr<IClassFactory> pFactoryHCP;
CComPtr<IClassFactory> pFactoryMSITS;
bool fCOM_reg = false;
bool fCOM_unreg = false;
int i;
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "ProcessArguments" );
__MPC_EXIT_IF_METHOD_FAILS(hr, InitAll());
for(i=1; i<argc; i++)
{
LPCWSTR szArg = argv[i];
if(szArg[0] == '-' ||
szArg[0] == '/' )
{
szArg++;
if(_wcsicmp( szArg, L"RegServer" ) == 0) fCOM_reg = true;
if(_wcsicmp( szArg, L"UnregServer" ) == 0) fCOM_unreg = true;
}
}
if(fCOM_reg || fCOM_unreg)
{
MPC::wstring strDir( HC_ROOT_HELPSVC_BINARIES ); MPC::SubstituteEnvVariables( strDir );
struct _ATL_REGMAP_ENTRY rgREG[] =
{
{ L"BINARIES", strDir.c_str() },
{ NULL , NULL }
};
_Module.UpdateRegistryFromResource( IDR_HC, fCOM_reg ? TRUE : FALSE, rgREG );
if(fCOM_reg)
{
_Module.RegisterTypeLib( _T("\\1") );
_Module.RegisterTypeLib( _T("\\2") );
}
if(fCOM_unreg)
{
_Module.UnRegisterTypeLib( _T("\\1") );
_Module.UnRegisterTypeLib( _T("\\2") );
}
__MPC_SET_ERROR_AND_EXIT(hr, S_OK);
}
//////////////////////////////////////////////////////////////////////
{
MPC::RegKey rkBase;
bool fFound;
if(SUCCEEDED(rkBase.SetRoot( HKEY_LOCAL_MACHINE )) &&
SUCCEEDED(rkBase.Attach ( DEBUG_REGKEY )) &&
SUCCEEDED(rkBase.Exists ( fFound )) && fFound)
{
CComVariant vValue;
#ifdef DEBUG
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_BREAKONSTART )) && fFound && vValue.vt == VT_I4)
{
if(vValue.lVal) DebugBreak();
}
g_Debug_CONTEXTMENU = true;
g_Debug_BUILDTREE = true;
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_FORCESTYLE )) && fFound && vValue.vt == VT_BSTR && vValue.bstrVal)
{
wcsncpy( g_Debug_FORCESTYLE, vValue.bstrVal, MAXSTRLEN(g_Debug_FORCESTYLE) );
}
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_FORCERTL )) && fFound && vValue.vt == VT_I4)
{
if(vValue.lVal) ::SetProcessDefaultLayout( LAYOUT_RTL );
}
#endif
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_BLOCKERRORS )) && fFound && vValue.vt == VT_I4)
{
g_Debug_BLOCKERRORS = (vValue.lVal != 0);
}
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_CONTEXTMENU )) && fFound && vValue.vt == VT_I4)
{
g_Debug_CONTEXTMENU = (vValue.lVal != 0);
}
if(SUCCEEDED(rkBase.get_Value( vValue, fFound, DEBUG_BUILDTREE )) && fFound && vValue.vt == VT_I4)
{
g_Debug_BUILDTREE = (vValue.lVal != 0);
}
}
}
//////////////////////////////////////////////////////////////////////
__MPC_EXIT_IF_METHOD_FAILS(hr, RegisterProtocols( pSession, pFactoryHCP, pFactoryMSITS ));
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHHelpCenterExternal::s_GLOBAL->Initialize());
//
// Look for the service and warn the user if we cannot find it.
//
if(CPCHHelpCenterExternal::s_GLOBAL->IsServiceRunning() == false)
{
//
// Bug 535303 SVR: Security: Help And Support Service Resets to Automatic after Set to Disabled
// Modified 4/24 gschua
//
//
// Try re-registering the service.
//
/*
{
MPC::wstring strCmdLine( HELPSVC_REGISTRATION ); MPC::SubstituteEnvVariables( strCmdLine );
(void)MPC::ExecuteCommand( strCmdLine );
}
if(CPCHHelpCenterExternal::s_GLOBAL->IsServiceRunning() == false)
{
MPC::LocalizedMessageBox( IDS_HELPCTR_DLG_ERROR, IDS_HELPCTR_SVC_MISSING, MB_OK );
__MPC_SET_ERROR_AND_EXIT(hr, S_OK);
}
*/
MPC::LocalizedMessageBox( IDS_HELPCTR_DLG_ERROR, IDS_HELPCTR_SVC_MISSING, MB_OK );
__MPC_SET_ERROR_AND_EXIT(hr, S_OK);
}
for(i=1; i<argc; i++)
{
LPCWSTR szArg = argv[i];
if(szArg[0] == '-' ||
szArg[0] == '/' )
{
__MPC_EXIT_IF_METHOD_FAILS(hr, CPCHHelpCenterExternal::s_GLOBAL->ProcessArgument( i, &szArg[1], argc, argv ));
}
}
if(CPCHHelpCenterExternal::s_GLOBAL->DoWeNeedUI())
{
MPC::wstring szTitle; MPC::LocalizeString( IDS_MAINWND_TITLE, szTitle );
{
INITCOMMONCONTROLSEX icc;
icc.dwSize = sizeof(INITCOMMONCONTROLSEX);
icc.dwICC = ICC_USEREX_CLASSES | ICC_COOL_CLASSES | ICC_INTERNET_CLASSES | ICC_PAGESCROLLER_CLASS | ICC_NATIVEFNTCTL_CLASS;
::InitCommonControlsEx( &icc );
}
//
// Load the UI.
//
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "UI - Pre FindMarsDLL" );
if(FindMarsDLL( hModule, pMarsThreadProc ))
{
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "UI - Post FindMarsDLL" );
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "UI - Ready to go" );
hr = CPCHHelpCenterExternal::s_GLOBAL->RunUI( szTitle, pMarsThreadProc );
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "UI - Passivating" );
}
}
//////////////////////////////////////////////////////////////////////
hr = S_OK;
__HCP_FUNC_CLEANUP;
if(hModule) ::FreeLibrary( hModule );
(void)UnregisterProtocols( pSession, pFactoryHCP, pFactoryMSITS );
CleanAll();
__HCP_FUNC_EXIT(hr);
}
extern "C" int WINAPI wWinMain( HINSTANCE hInstance ,
HINSTANCE hPrevInstance,
LPWSTR lpCmdLine ,
int nShowCmd )
{
HRESULT hr;
int argc;
LPCWSTR* argv;
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "wWinMain" );
if(SUCCEEDED(hr = ::CoInitialize( NULL )))
{
if(SUCCEEDED(hr = ::CoInitializeSecurity( NULL ,
-1 , // We don't care which authentication service we use.
NULL ,
NULL ,
RPC_C_AUTHN_LEVEL_CONNECT, // We want to identify the callers.
RPC_C_IMP_LEVEL_DELEGATE , // We want to be able to forward the caller's identity.
NULL ,
EOAC_DYNAMIC_CLOAKING , // Let's use the thread token for outbound calls.
NULL )))
{
__MPC_TRACE_INIT();
//
// Parse the command line.
//
if(SUCCEEDED(hr = MPC::CommandLine_Parse( argc, argv )))
{
//
// Initialize ATL modules.
//
_Module.Init( ObjectMap, hInstance, L"helpctr", 0, 0 );
//
// Initialize MPC module.
//
if(SUCCEEDED(hr = MPC::_MPC_Module.Init()))
{
//
// Process arguments.
//
hr = ProcessArguments( argc, argv );
MPC::_MPC_Module.Term();
}
_Module.Term();
MPC::CommandLine_Free( argc, argv );
}
__MPC_TRACE_TERM();
}
::CoUninitialize();
}
DEBUG_AppendPerf( DEBUG_PERF_BASIC, "Shutdown" );
DEBUG_DumpPerf ( L"%WINDIR%\\TEMP\\HC_perf_counters.txt" );
return FAILED(hr) ? 10 : 0;
}