shell_code.hpp

#pragma once
#include <iostream>
#include <Windows.h>

class shell_code {
private:
    uint8_t* buffer{};
    size_t buffer_size = 0;
public:
    shell_code() {
        buffer = reinterpret_cast<uint8_t*>(VirtualAlloc(nullptr, 0x1000, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE));
    }

    ~shell_code() {
        VirtualFree(buffer, 0, MEM_RELEASE);
    }

    template<typename T>
    void make(const T& v) {
        if constexpr (std::is_same<T, int>::value) {
            *reinterpret_cast<uint8_t*>(buffer + buffer_size) = *reinterpret_cast<const uint8_t*>(&v);
            buffer_size += sizeof(uint8_t);
        }
        else {
            *reinterpret_cast<T*>(buffer + buffer_size) = v;
            buffer_size += sizeof(T);
        }
    }

    template<typename F, typename... R>
    void make(const F& first, const R&... rest)
    {
        make(first);
        make(rest...);
    }

    void get(uint8_t* getter) {
        for (size_t i = 0; i < buffer_size; i++) {
            getter[i] = buffer[i];
        }
    }

    void* get() {
        return buffer;
    }

    size_t size() {
        return buffer_size;
    }
};