#include "driver.h"
#include <iostream>
#include <TlHelp32.h>

std::uint32_t find_process_by_id(const std::string& name)
{
    const auto snap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (snap == INVALID_HANDLE_VALUE) {
        return 0;
    }

    PROCESSENTRY32 proc_entry{};
    proc_entry.dwSize = sizeof proc_entry;

    auto found_process = false;
    if (!!Process32First(snap, &proc_entry)) {
        do {
            if (name == proc_entry.szExeFile) {
                found_process = true;
                break;
            }
        } while (!!Process32Next(snap, &proc_entry));
    }

    CloseHandle(snap);
    return found_process
        ? proc_entry.th32ProcessID
        : 0;
}

void example()
{
    const auto connection = driver::connect();
    if (connection == INVALID_SOCKET)
    {
        std::cout << "Connection failed.\n";
        return;
    }

    // Cheat stuff
    const auto pid = find_process_by_id("notepad.exe");
    std::printf("Process id: %p.\n", pid);

    const auto base_address = driver::get_process_base_address(connection, pid);
    std::printf("Process base address: %p.\n", (void*)base_address);

    const auto dos_magic = driver::read<uint16_t>(connection, pid, base_address);
    std::printf("DOS signature: %X.\n", dos_magic);

    driver::readtype>(connection, pid, address)

    driver::write<type>(connection, pid, address, size)

    driver::disconnect(connection);
}

int main()
{
    driver::initialize();

    example();

    driver::deinitialize();
}