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.
 
 
 
 
 
 

370 lines
8.9 KiB

//***************************************************************************
//
// TestInfo.h
//
// Module: CDM Provider
//
// Purpose: Defines the CClassPro class. An object of this class is
// created by the class factory for each connection.
//
// Copyright (c) 2000 Microsoft Corporation
//
//***************************************************************************
class CResultList
{
public:
CResultList();
~CResultList();
HRESULT Add(IWbemClassObject *ResultInstance,
BSTR ResultRelPath,
BSTR ResultForMSERelPath,
BSTR ResultForTestRelPath);
void Clear(void);
HRESULT GetResultsList(ULONG *Count,
IWbemClassObject ***Objects);
HRESULT GetResultByResultRelPath(PWCHAR ObjectPath,
IWbemClassObject **ppResult);
HRESULT GetResultByResultForMSERelPath(PWCHAR ObjectPath,
IWbemClassObject **ppResult);
HRESULT GetResultByResultForTestRelPath(PWCHAR ObjectPath,
IWbemClassObject **ppResult);
private:
typedef struct
{
IWbemClassObject *ResultInstance;
BSTR ResultRelPath;
BSTR ResultForMSERelPath;
BSTR ResultForTestRelPath;
} RESULTENTRY, *PRESULTENTRY;
ULONG ListSize;
ULONG ListEntries;
PRESULTENTRY List;
};
class CBstrArray
{
public:
CBstrArray();
~CBstrArray();
HRESULT Initialize(ULONG ListCount);
void Set(ULONG Index, BSTR s);
BSTR /* NOFREE */ Get(ULONG Index);
ULONG GetListSize();
private:
BOOLEAN IsInitialized();
BSTR *Array;
ULONG ListSize;
};
class CWbemObjectList
{
public:
CWbemObjectList();
~CWbemObjectList();
HRESULT Initialize(ULONG ListCount);
HRESULT Set(ULONG Index, IWbemClassObject *Pointer, BOOLEAN KeepRelPath);
IWbemClassObject *Get(ULONG Index);
BSTR /* NOFREE */ GetRelPath(ULONG Index);
ULONG GetListSize(void);
private:
BOOLEAN IsInitialized(
);
ULONG ListSize;
IWbemClassObject **List;
BSTR *RelPaths;
};
class CTestServices
{
public:
CTestServices();
~CTestServices();
//
// Linked list management routines for the benefit of the
// provider
//
CTestServices *GetNext();
CTestServices *GetPrev();
void InsertSelf(CTestServices **Head);
HRESULT QueryWdmTest(IWbemClassObject *pCdmTest,
int RelPathIndex);
HRESULT ExecuteWdmTest( IWbemClassObject *pCdmSettings,
IWbemClassObject *pCdmResult,
int RelPathIndex,
ULONG *Result,
BSTR *ExecutionID);
HRESULT StopWdmTest( int RelPathIndex,
ULONG *Result,
BOOLEAN *TestingStopped);
HRESULT GetRelPathIndex(BSTR CimRelPath,
int *RelPathIndex);
ULONG GetInstanceCount(void) { return(RelPathCount); };
LONG GetTestEstimatedTime(int RelPathIndex) { return(0); };
BOOLEAN GetTestIsExclusiveForMSE(int RelPathIndex) { return(FALSE); };
HRESULT FillInCdmResult(
IWbemClassObject *pCdmResult,
IWbemClassObject *pCdmSettings,
int RelPathIndex,
BSTR ExecutionID
);
BOOLEAN IsThisInitialized(void);
HRESULT InitializeCdmClasses(PWCHAR CdmClassName);
BOOLEAN ClaimCdmClassName(PWCHAR CdmClassName);
HRESULT AddResultToList(IWbemClassObject *pCdmResult,
BSTR ExecutionID,
int RelPathIndex);
void ClearResultsList(int RelPathIndex);
HRESULT GetResultsList(int RelPathIndex,
ULONG *ResultsCount,
IWbemClassObject ***Results);
HRESULT GetCdmResultByResultRelPath(int RelPathIndex,
PWCHAR ObjectPath,
IWbemClassObject **ppCdmResult);
HRESULT GetCdmResultByResultForMSERelPath(int RelPathIndex,
PWCHAR ObjectPath,
IWbemClassObject **ppCdmResult);
HRESULT GetCdmResultByResultForTestRelPath(int RelPathIndex,
PWCHAR ObjectPath,
IWbemClassObject **ppCdmResult);
//
// Accessors
BSTR GetCimRelPath(int RelPathIndex);
BSTR GetCdmTestClassName(void);
BSTR GetCdmTestRelPath(void);
BSTR GetCdmResultClassName(void);
BSTR GetCdmSettingClassName(void);
BSTR GetCdmSettingRelPath(int RelPathIndex, ULONG SettingIndex);
ULONG GetCdmSettingCount(int RelPathIndex);
IWbemClassObject *GetCdmSettingObject(int RelPathIndex, ULONG SettingIndex);
BSTR GetCdmTestForMSEClassName(void);
BSTR GetCdmTestForMSERelPath(int RelPathIndex);
BSTR GetCdmSettingForTestClassName(void);
BSTR GetCdmSettingForTestRelPath(int RelPathIndex, ULONG SettingIndex);
BSTR GetCdmResultForMSEClassName(void);
BSTR GetCdmResultForMSERelPath(int RelPathIndex);
BSTR GetCdmResultForTestClassName(void);
BSTR GetCdmResultForTestRelPath(int RelPathIndex);
BSTR GetCdmTestForSoftwareClassName(void);
BSTR GetCdmTestForSoftwareRelPath(void);
BSTR GetCdmTestInPackageClassName(void);
BSTR GetCdmTestInPackageRelPath(void);
BSTR GetCdmResultInPackageClassName(void);
BSTR GetCdmResultInPackageRelPath(void);
private:
HRESULT WdmPropertyToCdmProperty(
IWbemClassObject *pCdmClassInstance,
IWbemClassObject *pWdmClassInstance,
BSTR PropertyName,
VARIANT *PropertyValue,
CIMTYPE CdmCimType,
CIMTYPE WdmCimType
);
HRESULT CdmPropertyToWdmProperty(
IWbemClassObject *pWdmClassInstance,
IWbemClassObject *pCdmClassInstance,
BSTR PropertyName,
VARIANT *PropertyValue,
CIMTYPE WdmCimType,
CIMTYPE CdmCimType
);
HRESULT CopyBetweenCdmAndWdmClasses(
IWbemClassObject *pDestinationClass,
IWbemClassObject *pSourceClass,
BOOLEAN WdmToCdm
);
HRESULT ConnectToWdmClass(int RelPathIndex,
IWbemClassObject **ppWdmClassObject);
HRESULT GetCdmClassNamesFromOne(
PWCHAR CdmClass
);
HRESULT BuildResultRelPaths(
IN int RelPathIndex,
IN BSTR ExecutionId,
OUT BSTR *ResultRelPath,
OUT BSTR *ResultForMSERelPath,
OUT BSTR *ResultForTestRelPath
);
HRESULT BuildTestRelPaths(
void
);
HRESULT ParseSettingList(
VARIANT *SettingList,
CWbemObjectList *CdmSettings,
CBstrArray *CdmSettingForTestRelPath,
int RelPathIndex
);
HRESULT BuildSettingForTestRelPath(
OUT BSTR *RelPath,
IN IWbemClassObject *pCdmSettingInstance
);
HRESULT GetCdmTestSettings(void);
HRESULT QueryOfflineResult(
OUT IWbemClassObject *pCdmResult,
IN BSTR ExecutionID,
IN int RelPathIndex
);
// @@BEGIN_DDKSPLIT
HRESULT GatherRebootResults(
void
);
HRESULT PersistResultInSchema(
IWbemClassObject *pCdmResult,
BSTR ExecutionID,
int RelPathIndex
);
// @@END_DDKSPLIT
HRESULT GetTestOutParams(
IN IWbemClassObject *OutParams,
OUT IWbemClassObject *pCdmResult,
OUT ULONG *Result
);
HRESULT OfflineDeviceForTest(IWbemClassObject *pCdmResult,
BSTR ExecutionID,
int RelPathIndex);
BSTR GetExecutionID(
void
);
HRESULT FillTestInParams(
OUT IWbemClassObject *pInParamInstance,
IN IWbemClassObject *pCdmSettings,
IN BSTR ExecutionID
);
IWbemServices *GetWdmServices(void);
IWbemServices *GetCdmServices(void);
CTestServices *Next;
CTestServices *Prev;
//
// WDM Class Names
//
BSTR WdmTestClassName;
BSTR WdmSettingClassName;
BSTR WdmResultClassName;
BSTR WdmOfflineResultClassName;
BSTR WdmSettingListClassName;
//
// CDM Class and RelPath Names
//
BSTR CdmTestClassName;
BSTR CdmTestRelPath;
BSTR CdmResultClassName;
BSTR CdmSettingClassName;
BSTR CdmTestForMSEClassName;
BSTR *CdmTestForMSERelPath;
BSTR CdmSettingForTestClassName;
CBstrArray **CdmSettingForTestRelPath;
BSTR CdmResultForMSEClassName;
BSTR CdmResultForTestClassName;
BSTR CdmTestForSoftwareClassName;
BSTR CdmTestForSoftwareRelPath;
BSTR CdmTestInPackageClassName;
BSTR CdmTestInPackageRelPath;
BSTR CdmResultInPackageClassName;
BSTR CdmResultInPackageRelPath;
//
// Mapping class from Cim to WDM
//
BSTR CimClassMappingClassName;
//
// List of mappings between Cim and WDM
//
int RelPathCount;
BSTR *CimRelPaths;
BSTR *WdmRelPaths;
BSTR *WdmInstanceNames;
BSTR *PnPDeviceIdsX;
//
// Results for test executions. Each relpathindex maintains a
// list of results.
//
CResultList *CdmResultsList;
//
// Settings for test execution, we can have many settings for
// each test.
//
CWbemObjectList **CdmSettingsList;
};