Log.hpp

Go to the documentation of this file.

#pragma once
 
#include <format>
#include <source_location>
#include <sstream>
#include <string_view>
 
#include "Common.hpp"
 
// clang-format off
#ifdef PWN_LOG_USE_COLOR
#define PWN_COLOR_RESET               "\033[0m "
#define PWN_COLOR_BOLD                "\033[1m "
#define PWN_COLOR_UNDERLINE           "\033[4m "
#define PWN_COLOR_FG_BLACK            "\033[30m "
#define PWN_COLOR_FG_RED              "\033[31m "
#define PWN_COLOR_FG_GREEN            "\033[32m "
#define PWN_COLOR_FG_YELLOW           "\033[33m "
#define PWN_COLOR_FG_BLUE             "\033[34m "
#define PWN_COLOR_FG_MAGENTA          "\033[35m "
#define PWN_COLOR_FG_CYAN             "\033[36m "
#define PWN_COLOR_FG_WHITE            "\033[37m "
#else
#define PWN_COLOR_RESET                 ""
#define PWN_COLOR_BOLD                  ""
#define PWN_COLOR_UNDERLINE             ""
#define PWN_COLOR_FG_BLACK              ""
#define PWN_COLOR_FG_RED                ""
#define PWN_COLOR_FG_GREEN              ""
#define PWN_COLOR_FG_YELLOW             ""
#define PWN_COLOR_FG_BLUE               ""
#define PWN_COLOR_FG_MAGENTA            ""
#define PWN_COLOR_FG_CYAN               ""
#define PWN_COLOR_FG_WHITE              ""
#endif
 
#define PWN_LOG_STRINGS_DEBUG           "[  DEBUG   ] "
#define PWN_LOG_STRINGS_INFO            "[  INFO    ] "
#define PWN_LOG_STRINGS_SUCCESS         "[ SUCCESS  ] "
#define PWN_LOG_STRINGS_WARN            "[   WARN   ] "
#define PWN_LOG_STRINGS_ERROR           "[  ERROR   ] "
#define PWN_LOG_STRINGS_CRITICAL        "[ CRITICAL ] "
// clang-format on
 
 
namespace pwn::Log
{
enum class LogLevel : u8
{
    Debug,
 
    Info,
 
    Success,
 
    Warning,
 
    Error,
 
    Critical,
};
 
 
void
Log(const LogLevel level, std::source_location const& location, std::ostringstream& msg);
 
 
template<typename... Args>
void
Log(const LogLevel level, std::source_location const& location, std::string_view const& fmt, Args&&... args)
{
    std::ostringstream msg;
    msg << std::vformat(fmt, std::make_format_args(args...)) << '\n';
    Log(level, location, msg);
}
 
 
void
Log(const LogLevel level, std::source_location const& location, std::wostringstream& msg);
 
 
template<typename... Args>
void
Log(const LogLevel level, std::source_location const& location, std::wstring_view const& fmt, Args&&... args)
{
    std::wostringstream msg;
    msg << std::vformat(fmt, std::make_wformat_args(args...)) << std::endl;
    Log::Log(level, location, msg);
}
 
 
#ifdef PWN_BUILD_FOR_WINDOWS
template<typename T = std::wstring>
auto
FormatLastError(const u32 gle) -> T
{
    if constexpr ( std::is_same_v<T, std::wstring> )
    {
        wchar_t msg[1024] {0};
        ::FormatMessageW(
            FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_MAX_WIDTH_MASK,
            nullptr,
            gle,
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
            msg,
            __countof(msg),
            nullptr);
        return std::wstring(msg);
    }
 
    if constexpr ( std::is_same_v<T, std::string> )
    {
        char msg[1024] {0};
        ::FormatMessageA(
            FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_MAX_WIDTH_MASK,
            nullptr,
            gle,
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
            msg,
            __countof(msg),
            nullptr);
        return std::string(msg);
    }
 
    throw std::bad_variant_access();
}
 
void PWNAPI
perror(const std::wstring_view& prefix);
 
 
void PWNAPI
perror(const std::string_view& prefix);
 
 
void PWNAPI
ntperror(const std::wstring_view& prefix, NTSTATUS Status);
 
 
void PWNAPI
ntperror(const std::string_view& prefix, NTSTATUS Status);
#endif
 
} // namespace pwn::Log
 
 
// clang-format off
#define dbg(...)    Log::Log(Log::LogLevel::Debug,    std::source_location::current(), ##__VA_ARGS__)
#define info(...)   Log::Log(Log::LogLevel::Info,     std::source_location::current(), ##__VA_ARGS__)
#define ok(...)     Log::Log(Log::LogLevel::Success,  std::source_location::current(), ##__VA_ARGS__)
#define warn(...)   Log::Log(Log::LogLevel::Warning,  std::source_location::current(), ##__VA_ARGS__)
#define err(...)    Log::Log(Log::LogLevel::Error,    std::source_location::current(), ##__VA_ARGS__)
// clang-format on