Context.hpp

#pragma once
 
#include "Architecture.hpp"
#include "Common.hpp"
#include "Log.hpp"
#include "Utils.hpp"
 
using namespace pwn;
 
class GlobalContext
{
public:
    u64 CryptoSeed;
    std::mutex m_ConsoleMutex;
    std::mutex m_ConfigMutex;
    Log::LogLevel LogLevel = Log::LogLevel::Info;
 
    Architecture architecture;
    Endianess endianess;
    usize ptrsize;
 
    GlobalContext();
 
 
    template<typename T>
    void
    Set(T const& arg)
    {
        if constexpr ( std::is_same_v<T, std::string_view> )
        {
            warn("Deprecated, prefer using ArchitectureType");
            SetArchitecture(arg);
            return;
        }
 
        if constexpr ( std::is_same_v<T, std::wstring_view> )
        {
            warn("Deprecated, prefer using ArchitectureType");
            SetArchitecture(Utils::StringLib::To<std::string>(arg));
            return;
        }
 
        if constexpr ( std::is_same_v<T, ArchitectureType> )
        {
            SetArchitecture(arg);
            return;
        }
 
        if constexpr ( std::is_same_v<T, Log::LogLevel> )
        {
            SetLogLevel(arg);
            return;
        }
 
        if constexpr ( std::is_same_v<T, Endianess> )
        {
            SetEndianess(arg);
            return;
        }
 
        throw new std::bad_typeid();
    }
 
 
private:
    void
    SetArchitecture(ArchitectureType const& arch);
 
 
    void
    SetArchitecture(std::string_view const& type);
 
 
    void
    SetEndianess(Endianess end);
 
 
    void SetLogLevel(Log::LogLevel);
};
 
 
extern struct GlobalContext Context;