windows-nt/Source/XPSP1/NT/net/wlbs/wmi/wlbs_provider.cpp
2020-09-26 16:20:57 +08:00

620 lines
16 KiB
C++

////////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 1997-1999 Microsoft Corporation.
//
// File: WLBSProvider.CPP
//
// Module: WLBS Instance Provider
//
// Purpose: Defines the CWLBSProvider class. An object of this class is
// created by the class factory for each connection.
//
// History:
//
////////////////////////////////////////////////////////////////////////////////
#define ENABLE_PROFILE
#include <objbase.h>
#include <process.h>
#include "WLBS_Provider.h"
#include "ClusterWrapper.h"
#include "ControlWrapper.h"
#include "utils.h"
////////////////////////////////////////////////////////////////////////////////
//
//CWLBSProvider::CWLBSProvider
// CWLBSProvider::~CWLBSProvider
//
////////////////////////////////////////////////////////////////////////////////
CWLBSProvider::CWLBSProvider(
BSTR a_strObjectPath,
BSTR a_strUser,
BSTR a_strPassword,
IWbemContext * a_pContex
)
{
m_pNamespace = NULL;
InterlockedIncrement(&g_cComponents);
return;
}
CWLBSProvider::~CWLBSProvider(void)
{
InterlockedDecrement(&g_cComponents);
return;
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::Initialize
//
// Purpose: This is the implementation of IWbemProviderInit. The method
// is called by WinMgMt.
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::Initialize(
LPWSTR a_pszUser,
LONG a_lFlags,
LPWSTR a_pszNamespace,
LPWSTR a_pszLocale,
IWbemServices * a_pNamespace,
IWbemContext * a_pCtx,
IWbemProviderInitSink * a_pInitSink
)
{
PROFILE("CWLBSProvider::Initialize");
try {
//!!!!!!!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
//g_pWlbsControl must be initialized when the first COM instance is invoked
//and it must stay alive for the lifetime of the DLL, i.e. do NOT DESTROY
//it in the destructor of this CLASS until the API cache of the Cluster IP
//and Password are REMOVED.
//DO NOT INITIALIZE g_pWlbsControl in DLLMain. DLLMain is invoked with regsvr32
//and we do not want to initialize WinSock at that time!!! This will BREAK the
//installation process of the provider.
//!!!!!!!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
if( g_pWlbsControl == NULL ) {
g_pWlbsControl = new CWlbsControlWrapper();
if( g_pWlbsControl == NULL)
throw _com_error( WBEM_E_OUT_OF_MEMORY );
}
g_pWlbsControl->Initialize();
return CImpersonatedProvider::Initialize
(
a_pszUser,
a_lFlags,
a_pszNamespace,
a_pszLocale,
a_pNamespace,
a_pCtx,
a_pInitSink
);
}
catch (...)
{
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
return WBEM_E_FAILED;
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::CreateInstanceEnumAsync
//
// Purpose: Asynchronously enumerates the instances.
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::DoCreateInstanceEnumAsync(
BSTR a_strClass,
long a_lFlags,
IWbemContext * a_pIContex,
IWbemObjectSink * a_pIResponseHandler
)
{
ASSERT(g_pWlbsControl);
if (g_pWlbsControl)
{
//
// Re-enumerate all the clusters
//
g_pWlbsControl->ReInitialize();
}
try {
auto_ptr<CWlbs_Root> pMofClass;
//create an instance of the appropriate MOF support class
HRESULT hRes = GetMOFSupportClass(a_strClass, pMofClass, a_pIResponseHandler);
//call the appropriate wrapper class GetInstance method
if( SUCCEEDED( hRes ) && pMofClass.get() != NULL)
hRes = pMofClass->EnumInstances( a_strClass );
return hRes;
}
catch(_com_error HResErr ) {
a_pIResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
return HResErr.Error();
}
catch(...) {
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
a_pIResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, NULL);
return WBEM_E_FAILED;
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::GetObjectAsync
//
// Purpose: Gets an instance for a particular object path
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::DoGetObjectAsync(
BSTR a_strObjectPath,
long a_lFlags,
IWbemContext * a_pIContex,
IWbemObjectSink * a_pIResponseHandler
)
{
ASSERT(g_pWlbsControl);
if (g_pWlbsControl)
{
//
// Re-enumerate all the clusters
//
g_pWlbsControl->ReInitialize();
}
ParsedObjectPath* pParsedPath = NULL;
try {
auto_ptr<CWlbs_Root> pMofClass;
//parse the object path
ParseObjectPath( a_strObjectPath, &pParsedPath );
//create an instance of the appropriate MOF support class
HRESULT hRes = GetMOFSupportClass( pParsedPath->m_pClass, pMofClass, a_pIResponseHandler );
//call the appropriate wrapper class GetInstance method
if( SUCCEEDED( hRes ) && pMofClass.get() != NULL)
hRes = pMofClass->GetInstance( pParsedPath );
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return hRes;
}
catch(_com_error HResErr) {
a_pIResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return HResErr.Error();
}
catch(...) {
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
a_pIResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, NULL);
return WBEM_E_FAILED;
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::DoDeleteInstanceAsync
//
// Purpose: Gets an instance from a particular object path
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::DoDeleteInstanceAsync(
BSTR a_strObjectPath,
long a_lFlags,
IWbemContext * a_pIContex,
IWbemObjectSink * a_pIResponseHandler
)
{
ASSERT(g_pWlbsControl);
if (g_pWlbsControl)
{
//
// Re-enumerate all the clusters
//
g_pWlbsControl->ReInitialize();
}
ParsedObjectPath* pParsedPath = NULL;
try {
auto_ptr<CWlbs_Root> pMofClass;
//parse the object path
ParseObjectPath( a_strObjectPath, &pParsedPath );
//create an instance of the appropriate MOF support class
HRESULT hRes = GetMOFSupportClass( pParsedPath->m_pClass, pMofClass, a_pIResponseHandler );
//call the appropriate wrapper class GetInstance method
if( SUCCEEDED( hRes ) && pMofClass.get() != NULL)
hRes = pMofClass->DeleteInstance( pParsedPath );
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return hRes;
}
catch(_com_error HResErr) {
a_pIResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return HResErr.Error();
}
catch(...) {
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
a_pIResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, NULL);
return WBEM_E_FAILED;
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::ExecMethodAsync
//
// Purpose: Executes a MOF class method.
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::DoExecMethodAsync(
BSTR a_strObjectPath,
BSTR a_strMethodName,
long a_lFlags,
IWbemContext * a_pIContex,
IWbemClassObject * a_pIInParams,
IWbemObjectSink * a_pIResponseHandler
)
{
ASSERT(g_pWlbsControl);
if (g_pWlbsControl)
{
//
// Re-enumerate all the clusters
//
g_pWlbsControl->ReInitialize();
}
ParsedObjectPath* pParsedPath = NULL;
try {
//parse the object path
auto_ptr<CWlbs_Root> pMofClass;
//parse the object path
ParseObjectPath(a_strObjectPath, &pParsedPath);
//create an instance of the appropriate MOF support class
HRESULT hRes = GetMOFSupportClass(pParsedPath->m_pClass, pMofClass, a_pIResponseHandler);
//execute MOF class method
if( SUCCEEDED( hRes ) && pMofClass.get() != NULL)
hRes = pMofClass->ExecMethod( pParsedPath,
a_strMethodName,
0,
NULL,
a_pIInParams );
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return hRes;
}
catch(_com_error HResErr) {
a_pIResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
return HResErr.Error();
}
catch(...) {
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
a_pIResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, NULL);
return WBEM_E_FAILED;
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::PutInstanceAsync
//
// Purpose: Creates or modifies an instance
//
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CWLBSProvider::DoPutInstanceAsync
(
IWbemClassObject* a_pInst,
long a_lFlags,
IWbemContext* a_pIContex,
IWbemObjectSink* a_pIResponseHandler
)
{
ASSERT(g_pWlbsControl);
if (g_pWlbsControl)
{
//
// Re-enumerate all the clusters
//
g_pWlbsControl->ReInitialize();
}
ParsedObjectPath* pParsedPath = NULL;
HRESULT hRes = 0;
try {
wstring szClass;
auto_ptr<CWlbs_Root> pMofClass;
//retrieve the class name
GetClass( a_pInst, szClass );
//create an instance of the appropriate MOF support class
hRes = GetMOFSupportClass( szClass.c_str(), pMofClass, a_pIResponseHandler );
//call the appropriate wrapper class PutInstance method
if( SUCCEEDED( hRes ) && pMofClass.get() != NULL)
hRes = pMofClass->PutInstance( a_pInst );
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
}
catch(_com_error HResErr) {
a_pIResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
hRes = HResErr.Error();
}
catch(...) {
#ifdef DBG // In debug version, popup a message box for AV
MessageBoxExA(NULL,"Wlbs Provider Failure.","Wlbs Provider Access Violation",
MB_TOPMOST | MB_ICONHAND | MB_OK | MB_SERVICE_NOTIFICATION,LANG_USER_DEFAULT);
#endif
if( pParsedPath )
CObjectPathParser().Free( pParsedPath );
a_pIResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, NULL);
hRes = WBEM_E_FAILED;
}
return hRes;
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::GetMOFSupportClass
//
// Purpose: Determines which MOF class is requested and instantiates the
// appropriate internal support class.
//
////////////////////////////////////////////////////////////////////////////////
HRESULT CWLBSProvider::GetMOFSupportClass(
LPCWSTR a_szObjectClass,
auto_ptr<CWlbs_Root>& a_pMofClass,
IWbemObjectSink* a_pResponseHandler )
{
HRESULT hRes = 0;
try {
for( DWORD i = 0; i < MOF_CLASSES::NumClasses; i++ ) {
if( _wcsicmp( a_szObjectClass, MOF_CLASSES::g_szMOFClassList[i] ) == 0) {
a_pMofClass = auto_ptr<CWlbs_Root>
(MOF_CLASSES::g_pCreateFunc[i]( m_pNamespace, a_pResponseHandler ));
break;
}
}
}
catch(CErrorWlbsControl Err) {
IWbemClassObject* pWbemExtStat = NULL;
CWlbs_Root::CreateExtendedStatus( m_pNamespace,
&pWbemExtStat,
Err.Error(),
(PWCHAR)(Err.Description()) );
a_pResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, pWbemExtStat);
//do not return WBEM_E_FAILED, this causes a race condition
hRes = WBEM_S_NO_ERROR;
pWbemExtStat->Release();
}
return hRes;
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::ParseObjectPath
//
// Purpose:
//
////////////////////////////////////////////////////////////////////////////////
void CWLBSProvider::ParseObjectPath(
const BSTR a_strObjectPath,
ParsedObjectPath **a_pParsedObjectPath )
{
CObjectPathParser PathParser;
ASSERT( a_pParsedObjectPath );
//make sure this is NULL
*a_pParsedObjectPath = NULL;
int nStatus = PathParser.Parse(a_strObjectPath, a_pParsedObjectPath);
if(nStatus != 0) {
if( *a_pParsedObjectPath)
PathParser.Free( *a_pParsedObjectPath );
throw _com_error( WBEM_E_FAILED );
}
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBSProvider::GetClass
//
// Purpose: Retrieve the class name from an IWbemClassObject.
//
////////////////////////////////////////////////////////////////////////////////
void CWLBSProvider::GetClass(
IWbemClassObject* a_pClassObject,
wstring& a_szClass )
{
BSTR strClassName = NULL;
VARIANT vClassName;
HRESULT hRes;
try {
VariantInit( &vClassName );
strClassName = SysAllocString( L"__Class" );
if( !strClassName )
throw _com_error( WBEM_E_OUT_OF_MEMORY );
hRes = a_pClassObject->Get( strClassName,
0,
&vClassName,
NULL,
NULL);
if( FAILED( hRes ) )
throw _com_error( hRes );
a_szClass.assign( static_cast<LPWSTR>(vClassName.bstrVal) );
// CLD: Need to check return code for error
if (S_OK != VariantClear( &vClassName ))
throw _com_error( WBEM_E_FAILED );
if( strClassName ) {
SysFreeString( strClassName );
strClassName = NULL;
}
}
catch( ... ) {
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vClassName );
if( strClassName ) {
SysFreeString( strClassName );
strClassName = NULL;
}
throw;
}
}