// Copyright (c) 1997-1999 Microsoft Corporation
#if !defined(__SdkWbemHelpers_H)
#define      __SdkWbemHelpers_H
#pragma once

#if defined(_SDKWBEMHELPERS) || defined(_BUILD_SDKWBEMHELPERS)
#undef SMSSDK_Export

    #if defined(_BUILD_SDKWBEMHELPERS)
        #define SMSSDK_Export __declspec(dllexport)
    #else
        #define SMSSDK_Export __declspec(dllimport)
    #endif
#else
    #define SMSSDK_Export
#endif

#include <wbemidl.h>
#include <comdef.h>

SMSSDK_Export _bstr_t GetWbemErrorText(HRESULT hr);


class CWbemClassObject;
class CWbemServices;



_COM_SMARTPTR_TYPEDEF(IUnsecuredApartment,          IID_IUnsecuredApartment);
_COM_SMARTPTR_TYPEDEF(IWbemObjectSink,              IID_IWbemObjectSink);
_COM_SMARTPTR_TYPEDEF(IWbemClassObject,             IID_IWbemClassObject);
_COM_SMARTPTR_TYPEDEF(IWbemServices,                IID_IWbemServices);
_COM_SMARTPTR_TYPEDEF(IWbemContext,                 IID_IWbemContext );
_COM_SMARTPTR_TYPEDEF(IWbemCallResult,              IID_IWbemCallResult);
_COM_SMARTPTR_TYPEDEF(IWbemQualifierSet,            IID_IWbemQualifierSet);
_COM_SMARTPTR_TYPEDEF(IWbemLocator,                 IID_IWbemLocator);
_COM_SMARTPTR_TYPEDEF(IWbemObjectAccess,            IID_IWbemObjectAccess);
_COM_SMARTPTR_TYPEDEF(IEnumWbemClassObject,         IID_IEnumWbemClassObject);


//-----------------------------------------------------------------------------
class SMSSDK_Export CWbemException : public _com_error
{
private:
    CWbemClassObject *  m_pWbemError;
    HRESULT             m_hr;
    _bstr_t             m_sDescription;

    static IErrorInfo * GetErrorObject();
    static IErrorInfo * MakeErrorObject(_bstr_t);
    void GetWbemStatusObject();

public:
    CWbemException(HRESULT hr,_bstr_t sMessage);
    CWbemException(_bstr_t sMessage);

    CWbemClassObject GetWbemError();

    _bstr_t GetDescription() { return m_sDescription;  }
    HRESULT GetErrorCode()   { return m_hr;            }
};


//-----------------------------------------------------------------------------
class SMSSDK_Export CWbemClassObject
{
private:
    IWbemClassObjectPtr     m_pWbemObject;
	ULONG ref;
public:
	CWbemClassObject(const CWbemClassObject&  _in);
	CWbemClassObject(IWbemClassObject * const _in);
    CWbemClassObject(IWbemClassObjectPtr& _in);
    CWbemClassObject(IUnknown * _in);
    CWbemClassObject(IUnknownPtr& _in);
	CWbemClassObject();
    ~CWbemClassObject();

    ULONG AddRef();
    ULONG Release();

    void Attach(IWbemClassObject * pWbemObject);
    void Attach(IWbemClassObject * pWbemObject,bool bAddRef);
    IWbemClassObject * Detach();

    IWbemClassObject * operator->();
    operator IWbemClassObject*();
    operator IWbemClassObject**();
    operator IWbemClassObjectPtr();
	operator IUnknown *();
    IWbemClassObject ** operator &();
	IWbemClassObject* operator=(IWbemClassObject* _p);
	IWbemClassObjectPtr operator=(IWbemClassObjectPtr& _p);
	IWbemClassObject* operator=(IUnknown * _p);
	IWbemClassObjectPtr operator=(IUnknownPtr& _p);
	IWbemClassObject* operator=(const CWbemClassObject& _p);

    bool operator<(const CWbemClassObject& _comp);

    bool IsNull() const ;
    bool operator !();
    operator bool();

    unsigned long GetObjectSize();
    _bstr_t GetObjectText();

    HRESULT Clone(CWbemClassObject& _newObject);
    CWbemClassObject SpawnInstance();

	// put overloads
    HRESULT Put(const _bstr_t& _Name,_variant_t _value,CIMTYPE vType = 0);
    HRESULT Put(const _bstr_t& _Name,const _bstr_t& _value,CIMTYPE vType = 0);
	HRESULT Put(const _bstr_t& _Name, const long _value, CIMTYPE vType = 0);
	HRESULT Put(const _bstr_t& _Name, const bool _value,CIMTYPE vType = 0);
    HRESULT Get(const _bstr_t& _Name, _bstr_t& _value);
	HRESULT Get(const _bstr_t& _Name, long& _value);
	HRESULT Get(const _bstr_t& _Name, bool& _value);
    HRESULT Get(const _bstr_t& _Name,_variant_t& _value);
    _variant_t Get(const _bstr_t& _Name,CIMTYPE& vType,long& lFlavor);

    _bstr_t GetString(const _bstr_t& _Name);
    _int64  GetI64(const _bstr_t& _Name);
    long    GetLong(const _bstr_t& _Name);
	bool    GetBool(const _bstr_t& _Name);
    _bstr_t GetCIMTYPE(const _bstr_t& _Name);
	HRESULT GetValueMap (const _bstr_t& _Name, long value, _bstr_t &str);

	// these cast string props to the retval.
    long    GetLongEx(const _bstr_t& _Name);
	bool    GetBoolEx(const _bstr_t& _Name);
	// these cast string props fm the parm.
	HRESULT PutEx(const _bstr_t& _Name, const long _value, CIMTYPE vType = 0);
	HRESULT PutEx(const _bstr_t& _Name, const bool _value,CIMTYPE vType = 0);
	//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

    CWbemClassObject GetEmbeddedObject(const _bstr_t& _Name);
    HRESULT PutEmbeddedObject(const _bstr_t& _Name, CWbemClassObject &obj);

	HRESULT GetBLOB(const _bstr_t& _Name, BYTE **ptr, DWORD *outLen = NULL);
	HRESULT PutBLOB(const _bstr_t& _Name, const BYTE *ptr, const DWORD len);
	HRESULT GetDateTimeFormat(const _bstr_t& _Name, bstr_t &timeStr);
	HRESULT GetDIB(const _bstr_t& _Name, HDC hDC, HBITMAP &hDDBitmap);

    HRESULT BeginEnumeration(long _lFlags = WBEM_FLAG_LOCAL_ONLY);
    HRESULT Next(_bstr_t& _sName,_variant_t& _value);
    HRESULT Next(_bstr_t& _sName,_variant_t& _value,CIMTYPE& _vartype);
    HRESULT Next(_bstr_t& _sName,_variant_t& _value,CIMTYPE& _vartype,long& _flavor);
    HRESULT EndEnumeration();

	HRESULT GetMethod(const IN _bstr_t& _name, CWbemClassObject& coInSignature,
								CWbemClassObject& coOutSignature, long _lFlags = 0);


private:
	int ValidDMTF(bstr_t dmtf);

	// helpers for GetDIB().
	WORD DibNumColors(LPBITMAPINFOHEADER lpbi);
	WORD PaletteSize(LPBITMAPINFOHEADER lpbi);

};


__inline bool operator<(const CWbemClassObject& _X,const CWbemClassObject& _Y)
{
    return _X < _Y;
}

__inline bool operator==(const CWbemClassObject& _X,const CWbemClassObject& _Y)
{
    return _X == _Y;
}

//-----------------------------------------------------------------------------
typedef struct {
	bool currUser;
	COAUTHIDENTITY *authIdent;
	TCHAR fullAcct[100];
} LOGIN_CREDENTIALS;

class SMSSDK_Export CWbemServices 
{
private:

    HRESULT GetInterfacePtr(IWbemServicesPtr & pServices,
                            DWORD _dwProxyCapabilities = EOAC_NONE);
    HRESULT CommonInit(IWbemServicesPtr& pServ);
	bool IsClientNT5OrMore(void);
	HANDLE m_hAccessToken;
	LUID m_luid;
	bool m_fClearToken;
public:
	bool m_cloak;  // protects the cloak from eoac.
	COAUTHIDENTITY *m_authIdent;
	_bstr_t m_path;
    HRESULT m_hr;
    IWbemServicesPtr m_pService;
    IWbemContextPtr     m_pCtx;

	_bstr_t m_User;
	_bstr_t m_Password;
	long m_lFlags;

	CWbemServices(IWbemContext * _pContext = NULL);
    CWbemServices(const CWbemServices& _p, COAUTHIDENTITY *authIdent = 0);
	CWbemServices(const IWbemServicesPtr& _in);
    CWbemServices(const IUnknownPtr& _in);
    CWbemServices(IUnknown * _in);
	CWbemServices(IWbemServices *_in,IWbemContext * _pContext = NULL);
    
	~CWbemServices();

    CWbemServices& operator=(IUnknown * _p);
    CWbemServices& operator=(IUnknownPtr& _p);
	CWbemServices& operator=(IWbemServices *_p);
	CWbemServices& operator=(const CWbemServices& _p);

    bool IsNull()  ;
    operator bool();

    HRESULT GetServices(IWbemServices ** ppServices);
	HRESULT SetBlanket(IUnknown *service,
                       DWORD _dwProxyCapabilities = EOAC_NONE);
	void SetPriv(LPCTSTR privName = SE_SYSTEM_ENVIRONMENT_NAME);
	DWORD EnablePriv(bool fEnable );
	void ClearPriv(void);

    // Login as guest...
    HRESULT ConnectServer(_bstr_t sNetworkResource);
    // Login as user...
    HRESULT ConnectServer(_bstr_t sNetworkResource,
							_bstr_t sUser,
							_bstr_t sPassword,
							long SecurityFlags = 0);

    HRESULT ConnectServer(_bstr_t sNetworkResource,
							LOGIN_CREDENTIALS *user,
							long lSecurityFlags = 0);

    CWbemServices OpenNamespace(_bstr_t sNetworkResource);
	void DisconnectServer(void);

	CWbemClassObject CreateInstance(_bstr_t _sClass, IWbemCallResultPtr& _cr);
	CWbemClassObject CreateInstance(_bstr_t _sClass);

	HRESULT DeleteInstance(_bstr_t _sClass);

    CWbemClassObject GetObject(_bstr_t _sName, IWbemCallResultPtr &_cr, long flags = 0);
    CWbemClassObject GetObject(_bstr_t _sName, long flags = 0);
	IWbemClassObject *FirstInstanceOf(bstr_t className);

    HRESULT PutInstance(CWbemClassObject&   _object, IWbemCallResultPtr& _cr,
						long _lFlags = WBEM_FLAG_CREATE_OR_UPDATE);
	HRESULT PutInstance(CWbemClassObject&   _object,
                        IWbemContext *pContext,
                        long _lFlags = WBEM_FLAG_CREATE_OR_UPDATE,
                        DWORD _dwProxyCapabilities = EOAC_NONE);

    HRESULT PutInstance(CWbemClassObject&   _object,
                        long _lFlags = WBEM_FLAG_CREATE_OR_UPDATE,
                        DWORD _dwProxyCapabilities = EOAC_NONE);

    HRESULT CreateInstanceEnum(_bstr_t Class, long lFlags, IEnumWbemClassObject **ppEnum);
    HRESULT CreateInstanceEnumAsync(_bstr_t Class, IWbemObjectSink * ppSink, long lFlags = 0);

    HRESULT CreateClassEnum(_bstr_t Class, long lFlags, IEnumWbemClassObject **ppEnum);
	
    HRESULT ExecQuery(_bstr_t QueryLanguage,_bstr_t Query, long lFlags, IEnumWbemClassObject **ppEnum);
    HRESULT ExecQuery(_bstr_t Query, long lFlags, IEnumWbemClassObject **ppEnum);
    HRESULT ExecQuery(_bstr_t Query, IEnumWbemClassObject **ppEnum) ;
    HRESULT ExecQueryAsync(_bstr_t Query, IWbemObjectSink *pSink, long lFlags = 0);

    HRESULT GetMethodSignatures(const _bstr_t& _sObjectName,
								const _bstr_t& _sMethodName,
								CWbemClassObject& _in,
								CWbemClassObject& _out);
    HRESULT ExecMethod(_bstr_t sPath,
						_bstr_t sMethod,
						CWbemClassObject& inParams,
						CWbemClassObject& outParams);

    HRESULT CancelAsyncCall(IWbemObjectSink * pSink);

    HRESULT SetContextValue(_bstr_t sName,_variant_t value);
    HRESULT GetContextValue(_bstr_t sName,_variant_t& value);
	HRESULT DeleteContextValue(_bstr_t sName);
    HRESULT DeleteAllContextValues();
	HRESULT SetContext(IWbemContext * pWbemContext);
    HRESULT GetContext(IWbemContext ** ppWbemContext);
	HRESULT CreateClassEnumAsync(_bstr_t Class,
								IWbemObjectSink *ppSink,
								long lFlags /*= 0*/);

};


#endif //__SdkWbemHelpers_H