175 lines
5.4 KiB
C
175 lines
5.4 KiB
C
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// Copyright (c) 1997-1999 Microsoft Corporation.
|
||
|
|
//
|
||
|
|
// File: WLBS_Provider.h
|
||
|
|
//
|
||
|
|
// Module: WLBS instance provider main project header file
|
||
|
|
//
|
||
|
|
// Purpose: General purpose include file.
|
||
|
|
//
|
||
|
|
// History:
|
||
|
|
//
|
||
|
|
// Author: drbeck
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#ifndef _WLBSPROVIDER_INCLUDED_
|
||
|
|
#define _WLBSPROVIDER_INCLUDED_
|
||
|
|
|
||
|
|
#include <crtdbg.h>
|
||
|
|
#include <comdef.h>
|
||
|
|
#include <iostream>
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <wbemprov.h>
|
||
|
|
#include <genlex.h> //for wmi object path parser
|
||
|
|
#include <objbase.h>
|
||
|
|
#include <wlbsconfig.h>
|
||
|
|
#include <ntrkcomm.h>
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
#include "objpath.h"
|
||
|
|
#include "WLBS_MOFData.h"
|
||
|
|
#include "WLBS_Root.h"
|
||
|
|
#include "WLBS_Node.h"
|
||
|
|
#include "WLBS_Cluster.h"
|
||
|
|
#include "WLBS_NodeSetting.h"
|
||
|
|
#include "WLBS_ClusterSetting.h"
|
||
|
|
#include "WLBS_PortRule.h"
|
||
|
|
#include "WLBS_ParticipatingNode.h"
|
||
|
|
#include "WLBS_NodeSetPortRule.h"
|
||
|
|
#include "WLBS_ClusClusSetting.h"
|
||
|
|
#include "WLBS_NodeNodeSetting.h"
|
||
|
|
|
||
|
|
typedef LPVOID * PPVOID;
|
||
|
|
|
||
|
|
class CWLBSProvider : public CImpersonatedProvider
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
void GetClass(IWbemClassObject *a_pClassObject, wstring& a_szClass);
|
||
|
|
|
||
|
|
void ParseObjectPath(const BSTR a_strObjectPath, ParsedObjectPath **a_pParsedObjectPath);
|
||
|
|
|
||
|
|
HRESULT GetMOFSupportClass( LPCWSTR a_szObjectClass,
|
||
|
|
auto_ptr<CWlbs_Root> &a_pMofClass,
|
||
|
|
IWbemObjectSink* a_pResponseHandler );
|
||
|
|
BOOL IsCallerAdmin(VOID);
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
CWLBSProvider( BSTR ObjectPath = NULL,
|
||
|
|
BSTR User = NULL,
|
||
|
|
BSTR Password = NULL,
|
||
|
|
IWbemContext *pCtx = NULL );
|
||
|
|
|
||
|
|
virtual ~CWLBSProvider();
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoGetObjectAsync(
|
||
|
|
/* [in] */ BSTR ObjectPath,
|
||
|
|
/* [in] */ long lFlags,
|
||
|
|
/* [in] */ IWbemContext __RPC_FAR *pCtx,
|
||
|
|
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler);
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoPutInstanceAsync(
|
||
|
|
/* [in] */ IWbemClassObject __RPC_FAR *pInst,
|
||
|
|
/* [in] */ long lFlags,
|
||
|
|
/* [in] */ IWbemContext __RPC_FAR *pCtx,
|
||
|
|
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler) ;
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoDeleteInstanceAsync(
|
||
|
|
/* [in] */ BSTR ObjectPath,
|
||
|
|
/* [in] */ long lFlags,
|
||
|
|
/* [in] */ IWbemContext __RPC_FAR *pCtx,
|
||
|
|
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler);
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoCreateInstanceEnumAsync(
|
||
|
|
/* [in] */ BSTR Class,
|
||
|
|
/* [in] */ long lFlags,
|
||
|
|
/* [in] */ IWbemContext __RPC_FAR *pCtx,
|
||
|
|
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler);
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoExecQueryAsync(
|
||
|
|
/* [in] */ BSTR QueryLanguage,
|
||
|
|
/* [in] */ BSTR Query,
|
||
|
|
/* [in] */ long lFlags,
|
||
|
|
/* [in] */ IWbemContext __RPC_FAR *pCtx,
|
||
|
|
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler)
|
||
|
|
{return WBEM_E_NOT_SUPPORTED;}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DoExecMethodAsync(
|
||
|
|
BSTR,
|
||
|
|
BSTR,
|
||
|
|
long,
|
||
|
|
IWbemContext*,
|
||
|
|
IWbemClassObject*,
|
||
|
|
IWbemObjectSink* );
|
||
|
|
|
||
|
|
//IWbemProviderInit
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE Initialize(
|
||
|
|
LPWSTR pszUser,
|
||
|
|
LONG lFlags,
|
||
|
|
LPWSTR pszNamespace,
|
||
|
|
LPWSTR pszLocale,
|
||
|
|
IWbemServices * pNamespace,
|
||
|
|
IWbemContext * pCtx,
|
||
|
|
IWbemProviderInitSink * pInitSink
|
||
|
|
);
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CWLBSProvider * PCWLBSProvider;
|
||
|
|
|
||
|
|
// This class is the class factory for CWLBSProvider objects.
|
||
|
|
|
||
|
|
class CWLBSClassFactory : public IClassFactory
|
||
|
|
{
|
||
|
|
protected:
|
||
|
|
long m_cRef;
|
||
|
|
|
||
|
|
public:
|
||
|
|
CWLBSClassFactory(void);
|
||
|
|
~CWLBSClassFactory(void);
|
||
|
|
|
||
|
|
//IUnknown members
|
||
|
|
STDMETHODIMP QueryInterface(REFIID, PPVOID);
|
||
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
||
|
|
STDMETHODIMP_(ULONG) Release(void);
|
||
|
|
|
||
|
|
//IClassFactory members
|
||
|
|
STDMETHODIMP CreateInstance(LPUNKNOWN, REFIID
|
||
|
|
,PPVOID);
|
||
|
|
STDMETHODIMP LockServer(BOOL);
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CWLBSClassFactory *PCWLBSClassFactory;
|
||
|
|
_COM_SMARTPTR_TYPEDEF(IWbemClassObject, _uuidof(IWbemClassObject));
|
||
|
|
_COM_SMARTPTR_TYPEDEF(IWbemQualifierSet, _uuidof(IWbemQualifierSet));
|
||
|
|
|
||
|
|
//Global Variables
|
||
|
|
|
||
|
|
// These variables keep track of when the module can be unloaded
|
||
|
|
extern long g_cComponents;
|
||
|
|
extern long g_cServerLocks;
|
||
|
|
|
||
|
|
//The one and only instance of the WLBS control class. All
|
||
|
|
//WLBS control commands are implemented via this class.
|
||
|
|
class CWlbsControlWrapper;
|
||
|
|
extern CWlbsControlWrapper* g_pWlbsControl;
|
||
|
|
|
||
|
|
// General purpose utilities.
|
||
|
|
|
||
|
|
|
||
|
|
SCODE CreateInst( IWbemServices * pNamespace,
|
||
|
|
LPWSTR pKey,
|
||
|
|
long lVal,
|
||
|
|
IWbemClassObject ** pNewInst,
|
||
|
|
WCHAR * pwcClassName,
|
||
|
|
IWbemContext * pCtx);
|
||
|
|
|
||
|
|
#endif //_WLBSPROVIDER_INCLUDED_
|