|
|
/********************************************************************
Copyright (c) 1999 Microsoft Corporation
Module Name: Slave.cpp
Abstract: File for Implementation of CPCHMasterSlave and CPCHUserProcess classes, used to establish a connection between the service and a slave process.
Revision History: Davide Massarenti created 03/28/2000
********************************************************************/
#include "stdafx.h"
#include <UserEnv.h>
/////////////////////////////////////////////////////////////////////////////
static const WCHAR c_HelpHost[] = HC_ROOT_HELPSVC_BINARIES L"\\HelpHost.exe"; static const DWORD c_Timeout = 100 * 1000;
/////////////////////////////////////////////////////////////////////////////
CPCHUserProcess::UserEntry::UserEntry() { // CComBSTR m_bstrUser
m_dwSessionID = 0; // DWORD m_dwSessionID;
//
// CComBSTR m_bstrVendorID;
// CComBSTR m_bstrPublicKey;
//
// GUID m_guid;
// CComPtr<IPCHSlaveProcess> m_spConnection;
m_hToken = NULL; // HANDLE m_hToken;
m_hProcess = NULL; // HANDLE m_hProcess;
m_phEvent = NULL; // HANDLE* m_phEvent;
::ZeroMemory( &m_guid, sizeof(m_guid) ); }
CPCHUserProcess::UserEntry::~UserEntry() { Cleanup(); }
void CPCHUserProcess::UserEntry::Cleanup() { m_bstrUser .Empty (); m_dwSessionID = 0;
m_bstrVendorID .Empty (); m_bstrPublicKey.Empty ();
m_spConnection .Release();
if(m_hProcess) { ::CloseHandle( m_hProcess ); m_hProcess = NULL; }
if(m_hToken) { ::CloseHandle( m_hToken ); m_hToken = NULL; } }
////////////////////
bool CPCHUserProcess::UserEntry::operator==( /*[in]*/ const UserEntry& ue ) const { if(ue.m_bstrUser) { if(m_bstrUser == ue.m_bstrUser && m_dwSessionID == ue.m_dwSessionID) return true; }
if(ue.m_bstrVendorID) { if(MPC::StrICmp( m_bstrVendorID, ue.m_bstrVendorID ) == 0) return true; }
return false; }
bool CPCHUserProcess::UserEntry::operator==( /*[in]*/ const GUID& guid ) const { return ::IsEqualGUID( m_guid, guid ) ? true : false; }
////////////////////
HRESULT CPCHUserProcess::UserEntry::Clone( /*[in]*/ const UserEntry& ue ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::UserEntry::Clone" );
HRESULT hr;
Cleanup();
m_bstrUser = ue.m_bstrUser; // CComBSTR m_bstrUser;
m_dwSessionID = ue.m_dwSessionID; // DWORD m_dwSessionID;
//
m_bstrVendorID = ue.m_bstrVendorID; // CComBSTR m_bstrVendorID;
m_bstrPublicKey = ue.m_bstrPublicKey; // CComBSTR m_bstrPublicKey;
//
// GUID m_guid; // Used for establishing the connection.
// CComPtr<IPCHSlaveProcess> m_spConnection; // Live object.
// HANDLE m_hToken; // User token.
// HANDLE m_hProcess; // Process handle.
// HANDLE* m_phEvent; // To notify activator.
if(ue.m_hToken) { __MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::DuplicateTokenEx( ue.m_hToken, 0, NULL, SecurityImpersonation, TokenPrimary, &m_hToken )); }
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHUserProcess::UserEntry::Connect( /*[out]*/ HANDLE& hEvent ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::UserEntry::Connect" );
HRESULT hr;
//
// Logon the user, if not already done.
//
if(m_hToken == NULL) { CPCHAccounts acc; LPCWSTR szUser = SAFEBSTR( m_bstrUser );
//// // DEBUG
//// // DEBUG
//// // DEBUG
//// __MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::OpenProcessToken( ::GetCurrentProcess(), TOKEN_ALL_ACCESS, &m_hToken ));
//
// Only keep the account enabled for the time it takes to create its token.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, acc.ChangeUserStatus( szUser, /*fEnable*/true )); __MPC_EXIT_IF_METHOD_FAILS(hr, acc.LogonUser ( szUser, NULL, m_hToken )); __MPC_EXIT_IF_METHOD_FAILS(hr, acc.ChangeUserStatus( szUser, /*fEnable*/false )); }
if(m_hProcess == NULL || m_spConnection == NULL ) { __MPC_EXIT_IF_METHOD_FAILS(hr, SendActivation( hEvent )); }
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHUserProcess::UserEntry::SendActivation( /*[out]*/ HANDLE& hEvent ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::UserEntry::SendActivation" );
HRESULT hr; DWORD dwRes; PROCESS_INFORMATION piProcessInformation; STARTUPINFOW siStartupInfo; VOID* pEnvBlock = NULL; MPC::wstring strExe( c_HelpHost ); MPC::SubstituteEnvVariables( strExe ); WCHAR rgCommandLine[1024];
::ZeroMemory( (PVOID)&piProcessInformation, sizeof( piProcessInformation ) ); ::ZeroMemory( (PVOID)&siStartupInfo , sizeof( siStartupInfo ) ); siStartupInfo.cb = sizeof( siStartupInfo );
//
// Generate random ID.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, ::CoCreateGuid( &m_guid ));
//
// Create event.
//
__MPC_EXIT_IF_CALL_RETURNS_NULL(hr, (hEvent = ::CreateEvent( NULL, FALSE, FALSE, NULL )));
//
// Create process as user.
//
{ CComBSTR bstrGUID( m_guid );
swprintf( rgCommandLine, L"\"%s\" -guid %s", strExe.c_str(), (BSTR)bstrGUID ); }
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::CreateEnvironmentBlock( &pEnvBlock, m_hToken, TRUE ));
//// // DEBUG
//// // DEBUG
//// // DEBUG
//// __MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::CreateProcessW( NULL,
//// rgCommandLine,
//// NULL,
//// NULL,
//// FALSE,
//// NORMAL_PRIORITY_CLASS,
//// NULL,
//// NULL,
//// &siStartupInfo,
//// &piProcessInformation ));
// REAL
// REAL
// REAL
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::CreateProcessAsUserW( m_hToken , NULL , rgCommandLine , NULL , NULL , FALSE , NORMAL_PRIORITY_CLASS | CREATE_UNICODE_ENVIRONMENT , pEnvBlock , NULL , &siStartupInfo , &piProcessInformation ));
m_hProcess = piProcessInformation.hProcess; piProcessInformation.hProcess = NULL; hr = S_OK;
__HCP_FUNC_CLEANUP;
if(pEnvBlock) ::DestroyEnvironmentBlock( pEnvBlock );
if(piProcessInformation.hProcess) ::CloseHandle( piProcessInformation.hProcess ); if(piProcessInformation.hThread ) ::CloseHandle( piProcessInformation.hThread );
__HCP_FUNC_EXIT(hr); }
////////////////////
HRESULT CPCHUserProcess::UserEntry::InitializeForVendorAccount( /*[in]*/ BSTR bstrUser , /*[in]*/ BSTR bstrVendorID , /*[in]*/ BSTR bstrPublicKey ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::UserEntry::InitializeForVendorAccount" );
HRESULT hr;
Cleanup();
m_bstrUser = bstrUser;
m_bstrVendorID = bstrVendorID; m_bstrPublicKey = bstrPublicKey;
hr = S_OK;
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHUserProcess::UserEntry::InitializeForImpersonation( /*[in]*/ HANDLE hToken ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::UserEntry::InitializeForImpersonation" );
HRESULT hr; MPC::Impersonation imp; MPC::wstring strUser; DWORD dwSize; PSID pUserSid = NULL;
if(hToken == NULL) { __MPC_EXIT_IF_METHOD_FAILS(hr, imp.Initialize( MAXIMUM_ALLOWED ));
hToken = imp; }
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::DuplicateTokenEx( hToken, 0, NULL, SecurityImpersonation, TokenPrimary, &m_hToken ));
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::GetTokenInformation( m_hToken, TokenSessionId, &m_dwSessionID, sizeof(m_dwSessionID), &dwSize ));
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::SecurityDescriptor::GetTokenSids ( m_hToken, &pUserSid, NULL )); __MPC_EXIT_IF_METHOD_FAILS(hr, MPC::SecurityDescriptor::ConvertSIDToPrincipal( pUserSid, strUser )); m_bstrUser = strUser.c_str();
hr = S_OK;
__HCP_FUNC_CLEANUP;
MPC::SecurityDescriptor::ReleaseMemory( pUserSid );
__HCP_FUNC_EXIT(hr); }
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
CPCHUserProcess::CPCHUserProcess() { // List m_lst;
(void)MPC::_MPC_Module.RegisterCallback( this, (void (CPCHUserProcess::*)())Shutdown ); }
CPCHUserProcess::~CPCHUserProcess() { MPC::CallDestructorForAll( m_lst );
MPC::_MPC_Module.UnregisterCallback( this ); }
////////////////////
CPCHUserProcess* CPCHUserProcess::s_GLOBAL( NULL );
HRESULT CPCHUserProcess::InitializeSystem() { if(s_GLOBAL == NULL) { s_GLOBAL = new CPCHUserProcess; }
return s_GLOBAL ? S_OK : E_OUTOFMEMORY; }
void CPCHUserProcess::FinalizeSystem() { if(s_GLOBAL) { delete s_GLOBAL; s_GLOBAL = NULL; } }
////////////////////////////////////////////////////////////////////////////////
void CPCHUserProcess::Shutdown() { __HCP_FUNC_ENTRY( "CPCHUserProcess::Shutdown" );
MPC::SmartLock<_ThreadModel> lock( this );
m_lst.clear(); }
CPCHUserProcess::UserEntry* CPCHUserProcess::Lookup( /*[in]*/ const UserEntry& ue, /*[in]*/ bool fRelease ) { UserEntry* ueReal;
//
// Locate vendor and connect to it.
//
for(Iter it = m_lst.begin(); it != m_lst.end(); ) { ueReal = *it; if(ueReal && *ueReal == ue) { if(fRelease) { delete ueReal;
m_lst.erase( it++ ); continue; } else { return ueReal; } }
it++; }
return NULL; }
HRESULT CPCHUserProcess::Remove( /*[in]*/ const UserEntry& ue ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::Remove" );
HRESULT hr; MPC::SmartLock<_ThreadModel> lock( this );
(void)Lookup( ue, /*fRelease*/true );
hr = S_OK;
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHUserProcess::Connect( /*[in ]*/ const UserEntry& ue , /*[out]*/ IPCHSlaveProcess* *spConnection ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::Connect" );
HRESULT hr; MPC::SmartLock<_ThreadModel> lock( this ); HANDLE hEvent = NULL; UserEntry* ueReal = NULL;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHUserProcess::Connect" );
//
// Locate vendor and connect to it.
//
ueReal = Lookup( ue, /*fRelease*/false ); if(ueReal == NULL) { __MPC_EXIT_IF_ALLOC_FAILS(hr, ueReal, new UserEntry); m_lst.push_back( ueReal );
__MPC_EXIT_IF_METHOD_FAILS(hr, ueReal->Clone( ue )); }
__MPC_EXIT_IF_METHOD_FAILS(hr, ueReal->Connect( hEvent ));
//
// If "Connect" returns an event handle, wait on it.
//
if(hEvent) { ueReal->m_phEvent = &hEvent; // For waiting response...
lock = NULL;
if(::WaitForSingleObject( hEvent, c_Timeout ) != WAIT_OBJECT_0) { __MPC_SET_ERROR_AND_EXIT(hr, E_FAIL); }
lock = this;
//
// Relocate vendor (we release the lock on the object, so "ueReal" is not valid anymore.
//
ueReal = Lookup( ue, /*fRelease*/false ); if(ueReal == NULL) { __MPC_SET_WIN32_ERROR_AND_EXIT(hr, ERROR_FILE_NOT_FOUND); } }
__MPC_EXIT_IF_METHOD_FAILS(hr, ueReal->m_spConnection.QueryInterface( spConnection ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
if(hEvent) ::CloseHandle( hEvent );
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHUserProcess::Connect - done" );
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHUserProcess::RegisterHost( /*[in]*/ BSTR bstrID , /*[in]*/ IPCHSlaveProcess* pObj ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::RegisterHost" );
HRESULT hr; MPC::SmartLock<_ThreadModel> lock( this ); UserEntry* ueReal = NULL; GUID guid;
//
// Validate input.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, ::CLSIDFromString( bstrID, &guid ));
//
// Locate vendor and connect to it.
//
for(Iter it = m_lst.begin(); it != m_lst.end(); it++) { ueReal = *it; if(ueReal && *ueReal == guid) { break; } } if(ueReal == NULL) { __MPC_SET_WIN32_ERROR_AND_EXIT(hr, ERROR_FILE_NOT_FOUND); }
__MPC_EXIT_IF_METHOD_FAILS(hr, pObj->Initialize( ueReal->m_bstrVendorID, ueReal->m_bstrPublicKey ));
ueReal->m_spConnection = pObj;
hr = S_OK;
__HCP_FUNC_CLEANUP;
if(ueReal) { if(ueReal->m_phEvent) { //
// Signal the event handle, to awake the activator.
//
::SetEvent( *(ueReal->m_phEvent) );
ueReal->m_phEvent = NULL; } }
__HCP_FUNC_EXIT(hr); }
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
HRESULT CPCHUserProcess::SendResponse( /*[in]*/ DWORD dwArgc , /*[in]*/ LPCWSTR* lpszArgv ) { __HCP_FUNC_ENTRY( "CPCHUserProcess::SendResponse" );
HRESULT hr; int i; CComBSTR bstrGUID; CComPtr<IPCHService> srv; CComPtr<CPCHSlaveProcess> obj; #ifdef DEBUG
bool fDebug = false; #endif
//
// Parse the arguments.
//
for(i=1; i<dwArgc; i++) { LPCWSTR szArg = lpszArgv[i];
if(szArg[0] == '-' || szArg[0] == '/' ) { szArg++;
if(_wcsicmp( szArg, L"guid" ) == 0 && (i<dwArgc-1)) { bstrGUID = lpszArgv[++i]; continue; }
#ifdef DEBUG
if(_wcsicmp( szArg, L"debug" ) == 0) { fDebug = true; continue; } #endif
}
__MPC_SET_ERROR_AND_EXIT(hr, E_FAIL); }
#ifdef DEBUG
if(fDebug) DebugBreak(); #endif
//
// Create the COM object.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::CreateInstance( &obj ));
//
// Register it with the service.
//
__MPC_EXIT_IF_METHOD_FAILS(hr, ::CoCreateInstance( CLSID_PCHService, NULL, CLSCTX_ALL, IID_IPCHService, (void**)&srv ));
__MPC_EXIT_IF_METHOD_FAILS(hr, srv->RegisterHost( bstrGUID, obj ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr); }
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
CPCHSlaveProcess::CPCHSlaveProcess() { // CComBSTR m_bstrVendorID;
// CComBSTR m_bstrPublicKey;
m_ScriptLauncher = NULL; // CPCHScriptWrapper_Launcher* m_ScriptLauncher;
}
CPCHSlaveProcess::~CPCHSlaveProcess() { delete m_ScriptLauncher; }
HRESULT CPCHSlaveProcess::Initialize( /*[in]*/ BSTR bstrVendorID, /*[in]*/ BSTR bstrPublicKey ) { m_bstrVendorID = bstrVendorID; m_bstrPublicKey = bstrPublicKey;
return S_OK; }
HRESULT CPCHSlaveProcess::CreateInstance( /*[in ]*/ REFCLSID rclsid , /*[in ]*/ IUnknown* pUnkOuter , /*[out]*/ IUnknown* *ppvObject ) { HRESULT hr;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::CreateInstance" );
hr = ::CoCreateInstance( rclsid, pUnkOuter, CLSCTX_ALL, IID_IUnknown, (void**)ppvObject );
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::CreateInstance - done" );
return hr; }
HRESULT CPCHSlaveProcess::CreateScriptWrapper( /*[in ]*/ REFCLSID rclsid , /*[in ]*/ BSTR bstrCode , /*[in ]*/ BSTR bstrURL , /*[out]*/ IUnknown* *ppObj ) { __HCP_FUNC_ENTRY( "CPCHSlaveProcess::CreateScriptWrapper" );
HRESULT hr; MPC::SmartLock<_ThreadModel> lock( this );
if(!m_ScriptLauncher) { __MPC_EXIT_IF_ALLOC_FAILS(hr, m_ScriptLauncher, new CPCHScriptWrapper_Launcher); }
__MPC_EXIT_IF_METHOD_FAILS(hr, m_ScriptLauncher->CreateScriptWrapper( rclsid, bstrCode, bstrURL, ppObj ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHSlaveProcess::OpenBlockingStream( /*[in ]*/ BSTR bstrURL , /*[out]*/ IUnknown* *ppvObject ) { __HCP_FUNC_ENTRY( "CPCHSlaveProcess::OpenBlockingStream" );
HRESULT hr; CComPtr<IStream> stream;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::OpenBlockingStream" );
__MPC_EXIT_IF_METHOD_FAILS(hr, URLOpenBlockingStreamW( NULL, bstrURL, &stream, 0, NULL ));
__MPC_EXIT_IF_METHOD_FAILS(hr, stream.QueryInterface( ppvObject ));
hr = S_OK;
__HCP_FUNC_CLEANUP;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::OpenBlockingStream - done" );
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHSlaveProcess::IsNetworkAlive( /*[out]*/ VARIANT_BOOL* pfRetVal ) { __HCP_FUNC_ENTRY( "CPCHSlaveProcess::IsNetworkAlive" );
HRESULT hr;
__MPC_PARAMCHECK_BEGIN(hr) __MPC_PARAMCHECK_POINTER_AND_SET(pfRetVal,VARIANT_FALSE); __MPC_PARAMCHECK_END();
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::IsNetworkAlive" );
if(SUCCEEDED(MPC::Connectivity::NetworkAlive( HC_TIMEOUT_CONNECTIONCHECK ))) { *pfRetVal = VARIANT_TRUE; }
hr = S_OK;
__HCP_FUNC_CLEANUP;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::IsNetworkAlive - done" );
__HCP_FUNC_EXIT(hr); }
HRESULT CPCHSlaveProcess::IsDestinationReachable( /*[in ]*/ BSTR bstrDestination, /*[out]*/ VARIANT_BOOL *pfRetVal ) { __HCP_FUNC_ENTRY( "CPCHSlaveProcess::IsDestinationReachable" );
HRESULT hr;
__MPC_PARAMCHECK_BEGIN(hr) __MPC_PARAMCHECK_STRING_NOT_EMPTY(bstrDestination); __MPC_PARAMCHECK_POINTER_AND_SET(pfRetVal,VARIANT_FALSE); __MPC_PARAMCHECK_END();
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::IsDestinationReachable" );
if(SUCCEEDED(MPC::Connectivity::DestinationReachable( bstrDestination, HC_TIMEOUT_CONNECTIONCHECK ))) { *pfRetVal = VARIANT_TRUE; }
hr = S_OK;
__HCP_FUNC_CLEANUP;
DEBUG_AppendPerf( DEBUG_PERF_HELPHOST, "CPCHSlaveProcess::IsDestinationReachable - done" );
__HCP_FUNC_EXIT(hr); }
|