ErrLib_CPP.h

Go to the documentation of this file.

//Project: ErrLib
//Author: MSDN.WhiteKnight (https://github.com/MSDN-WhiteKnight)
 
#ifndef ERRLIB_CPP_H_INCLUDED
#define ERRLIB_CPP_H_INCLUDED
#include "ErrLib.h"
#include <string>
 
const DWORD ERRLIB_CPP_EXCEPTION = 0xC0400002;
 
#ifdef __cplusplus
namespace ErrLib{
 
class Exception : public std::exception{
private:
    CONTEXT _context;
    std::wstring _msg;
    std::wstring _stack;
    DWORD _code;
    void* _data;
 
    std::wstring PrintStackTraceImpl(){
        WCHAR buf[ErrLib_StackLen]=L"";
        ErrLib_PrintStack(&_context, buf, ErrLib_StackLen);
        return std::wstring(buf);
    }
 
protected: 
 
    void SetMsg(const std::wstring& msg){_msg=msg;}
 
    void SetCode(DWORD code){_code=code;}
 
    void SetData(void* data){_data=data;}
 
public:
 
    Exception():_msg(L""),_code(ERRLIB_CPP_EXCEPTION),_data(nullptr){
        RtlCaptureContext(&_context);
        _stack = this->PrintStackTraceImpl();
    }
 
    Exception(const std::wstring& message):_msg(message),_code(ERRLIB_CPP_EXCEPTION),_data(nullptr){
        RtlCaptureContext(&_context);
        _stack = this->PrintStackTraceImpl();
    }
 
    Exception(const std::wstring& message, DWORD code, void* data):_msg(message),_code(code),_data(data){
        RtlCaptureContext(&_context);
        _stack = this->PrintStackTraceImpl();
    }
    
    std::wstring GetMsg(){return _msg;}
 
    DWORD GetCode(){return _code;}
 
    void* GetData(){return _data;}
 
    void GetMessageText(WCHAR* pOutput, int cch){
        const WCHAR* pChars = _msg.c_str();
        wcscpy_s(pOutput, cch, pChars);
    }
 
    void GetContext(CONTEXT* pOutput){
        memcpy(pOutput, &_context, sizeof(CONTEXT));
    }
 
    void PrintStackTrace(WCHAR* pOutput, int cch){
        const WCHAR* p = this->_stack.c_str();
        StringCchCopy(pOutput, cch, p);
    }
 
    std::wstring PrintStackTrace(){
        return _stack;
    }
 
    void Log(bool visible){
        BOOL bVisible;
 
        if (visible) bVisible = TRUE;
        else bVisible = FALSE;
 
        ErrLib_LogExceptionInfo(_code, _msg.c_str(), this->PrintStackTrace().c_str(), bVisible);
    }
};
 
class WinapiException : public Exception{
public: 
    WinapiException(DWORD code,const std::wstring& msg){
        this->SetCode(code);
        this->SetMsg(msg);
    }
 
    static WinapiException FromLastError(bool localized){
        DWORD code = GetLastError();
        BOOL fLocalized;
        WCHAR buf[ErrLib_MessageLen]=L"";
 
        if(localized) fLocalized = TRUE;
        else fLocalized = FALSE;
        
        ErrLib_GetWinapiErrorMessage(code, fLocalized, buf, ErrLib_MessageLen);
        return WinapiException(code, buf);
    }
};
 
class ComException : public Exception{
public: 
    ComException(HRESULT hr,const std::wstring& msg){
        this->SetCode((DWORD)hr);
        this->SetMsg(msg);
    }
 
    static ComException FromHResult(HRESULT hr){
        WCHAR buf[ErrLib_MessageLen]=L"";
        
        ErrLib_GetHResultMessage(hr, buf, ErrLib_MessageLen);
        return ComException(hr, buf);
    }
};
 
}
#endif //__cplusplus
#endif