749 lines
21 KiB
C++
749 lines
21 KiB
C++
|
/********************************************************************
|
||
|
|
|
||
|
|
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);
|
||
|
|
}
|
||
|
|
|