mirror of https://github.com/tongzx/nt5src
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.
139 lines
6.6 KiB
139 lines
6.6 KiB
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c)2000 Microsoft Corporation, All Rights Reserved
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
#ifndef _BVT_UTIL_HEADER
|
|
#define _BVT_UTIL_HEADER
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
//*******************************************************************************************
|
|
//
|
|
// Class definitions & prototypes
|
|
//
|
|
//*******************************************************************************************
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
HRESULT AllocateAndConvertAnsiToUnicode( char * pstr, WCHAR *& pszW);
|
|
int ExecuteScript ( int nTest);
|
|
void LogCLSID ( const char * csFile, const ULONG Line, WCHAR * wcsID, CLSID clsid);
|
|
|
|
BOOL InitMasterList ( const WCHAR * wcsClassesAfterDelete,ItemList & MasterList);
|
|
BOOL InitAndExpandMasterList ( const WCHAR * wcsClassesString, int nWhichTest, ItemList & MasterList);
|
|
BOOL InitMasterListOfAddDeleteClasses ( const WCHAR * wcsClassesAddDelete,int nWhichTest,ItemList & MasterList);
|
|
int CrackClassName ( const WCHAR * wcsString, CHString & sClass, BOOL fExpectedFailure, const char * csFile, const ULONG Line );
|
|
int CrackAssociation ( const WCHAR * wcsString, CHString & sClass, CPropertyList & Properties, int & n, BOOL fExpectedFailure, const char * csFile, const ULONG Line );
|
|
int CrackClass ( WCHAR * wcsInClassString, CHString & sClass, CHString & sParentClass, CHString & sInstance,
|
|
CPropertyList & Properties, int & nResults, BOOL fExpectedFailure,const char * csFile, const ULONG Line );
|
|
int CrackEvent ( WCHAR * wcsInEventString, EventInfo & Event, BOOL fExpectedFailure, const char * csFile, const ULONG Line );
|
|
int CrackNamespace ( WCHAR * wcsInString, CHString & sNamespace, BOOL fExpectedFailure,const char * csFile, const ULONG Line );
|
|
int CrackClassNameAndResults ( WCHAR * wcsInString, CHString & sClass, ItemList & nResults, BOOL fExpectedFailure, const char * csFile, const ULONG Line );
|
|
int CrackMethod ( WCHAR * wcsInString, CHString & sClass, CHString & sInst, CHString & sMethod,CPropertyList & InProp, CPropertyList & OutProps,
|
|
int & nResults, BOOL fExpectedFailure, const char * csFile, const ULONG Line );
|
|
BOOL CompareType ( CVARIANT & Var1, CVARIANT & Var2, long lType );
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
class CLogAndDisplayOnScreen
|
|
{
|
|
protected:
|
|
WCHAR m_wcsFileName[MAX_PATH+2];
|
|
BOOL m_fDisplay;
|
|
CCriticalSection m_CritSec;
|
|
BOOL WriteToFile(WCHAR * pwcsError, WCHAR * pwcsFileAndLine, const WCHAR *wcsString);
|
|
|
|
public:
|
|
CLogAndDisplayOnScreen();
|
|
~CLogAndDisplayOnScreen();
|
|
|
|
BOOL LogError(const char * csFile , const ULONG Line , int nErrorType, const WCHAR *fmt, ...);
|
|
virtual BOOL Log(WCHAR * pwcsError, WCHAR * pwcsFileAndLine, const WCHAR *wcsString) { return FALSE;}
|
|
|
|
inline void SetDisplay(BOOL fDisplay) { CAutoBlock Block(&m_CritSec); m_fDisplay = fDisplay;}
|
|
inline void SetFileName(WCHAR * wcsFile ) { CAutoBlock Block(&m_CritSec); wcsncpy( m_wcsFileName, wcsFile, MAX_PATH); }
|
|
};
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
class CIniFileAndGlobalOptions
|
|
{
|
|
private:
|
|
WCHAR m_wcsFileName[MAX_PATH+2];
|
|
CCriticalSection m_CritSec;
|
|
CFlexArray m_SpecificTests;
|
|
BOOL m_fSpecificTests;
|
|
int m_nThreads;
|
|
int m_nConnections;
|
|
|
|
BOOL ReadIniFile( WCHAR * wcsSection, const WCHAR * wcsKey, WCHAR * wcsDefault, CHString & sBuffer);
|
|
void DeleteList();
|
|
|
|
public:
|
|
|
|
CIniFileAndGlobalOptions();
|
|
~CIniFileAndGlobalOptions();
|
|
|
|
BOOL GetSpecificOptionForAPITest(const WCHAR * wcsClass, CHString & sClass, int nTest);
|
|
|
|
void WriteDefaultIniFile();
|
|
|
|
inline void AddToSpecificTestList(int nTest) { int * p = new int; *p = nTest; m_SpecificTests.Add(p); }
|
|
inline int SpecificTestSize() { return m_SpecificTests.Size(); }
|
|
inline int GetSpecificTest(int n) { int * p = (int*) m_SpecificTests.GetAt(n); return *p;}
|
|
inline void SpecificTests(BOOL f) { m_fSpecificTests = f;}
|
|
inline BOOL RunSpecificTests() { return m_fSpecificTests;}
|
|
|
|
inline void SetFileName(WCHAR * wcsFile ) { CAutoBlock Block(&m_CritSec); wcsncpy( m_wcsFileName, wcsFile, MAX_PATH); }
|
|
inline void SetThreads(int nThreads) { m_nThreads = nThreads;}
|
|
inline void SetConnections(int nConn) { m_nConnections = nConn;}
|
|
inline int GetThreads() { return m_nThreads;}
|
|
inline int GetConnections() { return m_nConnections;}
|
|
|
|
};
|
|
|
|
class CSinkEx : public IWbemObjectSinkEx
|
|
{
|
|
CFlexArray m_aObjects;
|
|
LONG m_lRefCount;
|
|
CRITICAL_SECTION m_cs;
|
|
HANDLE m_hEvent;
|
|
HRESULT m_hres;
|
|
IWbemClassObject* m_pErrorObj;
|
|
IID m_pInterfaceID;
|
|
IUnknown *m_pInterface;
|
|
|
|
public:
|
|
STDMETHOD(QueryInterface)(THIS_ REFIID riid, LPVOID FAR* ppvObj);
|
|
STDMETHOD_(ULONG, AddRef)(THIS);
|
|
STDMETHOD_(ULONG, Release)(THIS);
|
|
|
|
STDMETHOD(Indicate)(long lObjectCount, IWbemClassObject** pObjArray);
|
|
STDMETHOD(SetStatus)(long lFlags, HRESULT hResult, BSTR strParam, IWbemClassObject* pObjPAram);
|
|
|
|
STDMETHOD(Set)(long lFlags, REFIID riid, void *pComObject);
|
|
|
|
// Private to implementation.
|
|
// ==========================
|
|
|
|
CSinkEx(LONG lStartingRefCount = 1);
|
|
~CSinkEx();
|
|
|
|
UINT WaitForSignal(DWORD dwMSec) { return ::WbemWaitForSingleObject(m_hEvent, dwMSec); }
|
|
CFlexArray* GetObjectArray() { return &m_aObjects; }
|
|
int GetNumberOfObjectsReceived() { return m_aObjects.Size();}
|
|
|
|
IUnknown *GetInterface() { return m_pInterface; }
|
|
HRESULT GetStatusCode(IWbemClassObject** ppErrorObj = NULL)
|
|
{
|
|
if(ppErrorObj)
|
|
{
|
|
*ppErrorObj = m_pErrorObj;
|
|
if(m_pErrorObj) m_pErrorObj->AddRef();
|
|
}
|
|
return m_hres;
|
|
}
|
|
|
|
void Lock() { EnterCriticalSection(&m_cs); }
|
|
void Unlock() { LeaveCriticalSection(&m_cs); }
|
|
};
|
|
|
|
|
|
#endif
|