#include "prx/libc/include/Api.hpp"
#include "prx/libc/include/ApplicationHeap.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <cstdint>
#include <cstring>
#include <limits>
#include <stdexcept>

namespace {

void Require(bool value) {
    if (!value) throw std::runtime_error("reallocalign check failed");
}

}

int main() {
    void* api[10]{};
    ApplicationHeapRegister_nid_no_patch(api);
    auto* block = static_cast<unsigned char*>(reallocalign_nid_postfix(nullptr, 64, 64));
    Require(block != nullptr && (reinterpret_cast<std::uintptr_t>(block) % 64) == 0);
    std::memset(block, 0x5a, 64);
    auto* grown = static_cast<unsigned char*>(reallocalign_nid_postfix(block, 256, 64));
    Require(grown != nullptr && (reinterpret_cast<std::uintptr_t>(grown) % 64) == 0);
    for (int index = 0; index < 64; ++index) Require(grown[index] == 0x5a);
    auto* shrunk = static_cast<unsigned char*>(reallocalign_nid_postfix(grown, 16, 16));
    Require(shrunk != nullptr && (reinterpret_cast<std::uintptr_t>(shrunk) % 16) == 0);
    for (int index = 0; index < 16; ++index) Require(shrunk[index] == 0x5a);
    shrunk = static_cast<unsigned char*>(reallocalign_nid_postfix(shrunk, 16, 0));
    Require(shrunk != nullptr && (reinterpret_cast<std::uintptr_t>(shrunk) % 32) == 0);
    for (int index = 0; index < 16; ++index) Require(shrunk[index] == 0x5a);
    shrunk = static_cast<unsigned char*>(reallocalign_nid_postfix(shrunk, 40, 100));
    Require(shrunk != nullptr && (reinterpret_cast<std::uintptr_t>(shrunk) % 128) == 0);
    for (int index = 0; index < 16; ++index) Require(shrunk[index] == 0x5a);
    constexpr std::size_t max = std::numeric_limits<std::size_t>::max();
    *__error_nid_postfix() = 77;
    Require(reallocalign_nid_postfix(shrunk, max, 16) == nullptr && *__error_nid_postfix() == 77);
    Require(reallocalign_nid_postfix(shrunk, max - 0x1008f, 16) == nullptr && *__error_nid_postfix() == 22);
    *__error_nid_postfix() = 77;
    Require(reallocalign_nid_postfix(shrunk, max - 0x110ae, 4096) == nullptr && *__error_nid_postfix() == 77);
    Require(reallocalign_nid_postfix(shrunk, max - 0x110af, 4096) == nullptr && *__error_nid_postfix() == 22);
    *__error_nid_postfix() = 77;
    Require(reallocalign_nid_postfix(shrunk, 16, std::size_t{1} << 63) == nullptr && *__error_nid_postfix() == 77);
    for (int index = 0; index < 16; ++index) Require(shrunk[index] == 0x5a);
    Require(reallocalign_nid_postfix(shrunk, 0, 16) == shrunk && malloc_usable_size_nid_postfix(shrunk) == 0);
    free_nid_postfix(shrunk);

    void* empty = reallocalign_nid_postfix(nullptr, 0, 0);
    Require(empty != nullptr);
    free_nid_postfix(empty);

    void* aligned[16]{};
    void* unaligned = nullptr;
    for (int index = 0; index < 16 && unaligned == nullptr; ++index) {
        void* candidate = reallocalign_nid_postfix(nullptr, 24, 64);
        Require(candidate != nullptr && (reinterpret_cast<std::uintptr_t>(candidate) % 64) == 0);
        if ((reinterpret_cast<std::uintptr_t>(candidate) % 256) != 0) unaligned = candidate;
        else aligned[index] = candidate;
    }
    Require(unaligned != nullptr);
    *__error_nid_postfix() = 1234;
    void* moved = reallocalign_nid_postfix(unaligned, 0, 256);
    Require(moved != nullptr && moved != unaligned && (reinterpret_cast<std::uintptr_t>(moved) % 256) == 0);
    Require(malloc_usable_size_nid_postfix(moved) == 0 && *__error_nid_postfix() == 1234);
    free_nid_postfix(moved);
    for (void* block : aligned) free_nid_postfix(block);

    return 0;
}