windows-xp/Source/XPSP1/NT/com/ole32/stg/props/utest/chresult.hxx

#ifndef _CHRESULT_HXX_
#define _CHRESULT_HXX_

class CHResult
{
public:

    ~CHResult()
    {
        Initialize();
    }

    CHResult()
    {
        _hr = (HRESULT) E_FAIL;
        _bstrMessage = NULL;
        _pszFile = NULL;
        _ulLine = (ULONG) 0;
    }

    CHResult( CHResult &chr )
    {
        _hr = chr._hr;
        _ulLine = chr._ulLine;

        _bstrMessage = SysAllocString( chr._bstrMessage );
        _pszFile = new CHAR[ strlen(chr._pszFile) + sizeof(CHAR) ];
        if( NULL != _pszFile )
            memcpy( _pszFile, chr._pszFile, strlen(chr._pszFile) + sizeof(CHAR) );
    }

    CHResult( HRESULT hr, const LPOLESTR pwszMessage, const LPSTR pszFile, ULONG ulLine )
    {
        _hr = hr;

        _bstrMessage = SysAllocString( pwszMessage );

        _pszFile = new CHAR[strlen(pszFile) + sizeof(CHAR)];
        if( NULL != _pszFile )
            memcpy( _pszFile, pszFile, strlen(pszFile) + sizeof(CHAR) );

        _ulLine = ulLine;
    }

    CHResult( HRESULT hr, LPOLESTR pwszMessage )
    {
        this->CHResult::CHResult( hr, pwszMessage, NULL, 0 );
    }


public:

    operator HRESULT()
    {
        return( _hr );
    }

    operator LPOLESTR()
    {
        return( _bstrMessage );
    }

    HRESULT operator =(HRESULT hr)
    {
        _hr = hr;

        SysFreeString( _bstrMessage );
        _bstrMessage = NULL;

        return(_hr);
    }

public:

    HRESULT GetLastError()
    {
        Initialize();
        _hr = HRESULT_FROM_WIN32(::GetLastError());
        return( _hr );
    }

    LPSTR GetFile()
    {
        return( _pszFile );
    }

    ULONG GetLine()
    {
        return( _ulLine );
    }

private:

    void Initialize()
    {
        _hr = (HRESULT) E_FAIL;
        _ulLine = 0;

        SysFreeString( _bstrMessage );
        _bstrMessage = NULL;

        CoTaskMemFree( _pszFile );
        _pszFile = NULL;
    }

private:

    HRESULT _hr;
    BSTR    _bstrMessage;
    LPSTR   _pszFile;
    ULONG   _ulLine;
};


#define CHRESULT(hr,pwszMessage) CHResult( hr, pwszMessage, __FILE__, __LINE__ )

#endif