/////////////////////////////////////////////////////////////////////////////////////////////
//
//  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