#include "prx/libc/include/general/VabiMacros.hpp"
#include "prx/libkernel/KernelErrors.hpp"
#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <cstring>

struct SceKernelProsperoSystemSwVersion {
    std::uint64_t size;
    char versionString[0x1C];
    std::uint32_t version;
};

static_assert(offsetof(SceKernelProsperoSystemSwVersion, versionString) == 0x08);
static_assert(offsetof(SceKernelProsperoSystemSwVersion, version) == 0x24);
static_assert(sizeof(SceKernelProsperoSystemSwVersion) == 0x28);

extern "C" {
int* APS5_VABI __error_nid_postfix();
int APS5_VABI sceKernelGetProsperoSystemSwVersion_nid_postfix(SceKernelProsperoSystemSwVersion*);
int APS5_VABI sceKernelGetAllowedSdkVersionOnSystem_nid_postfix(std::uint32_t*);
}

static void Require(bool value) { if (!value) std::abort(); }

static void CheckProsperoSystemSwVersion() {
    *__error_nid_postfix() = 13;
    Require(sceKernelGetProsperoSystemSwVersion_nid_postfix(nullptr) == SCE_KERNEL_ERROR_EFAULT);
    const std::uint64_t sizes[] = {0, 0x18, 0x27, 0x30};
    for (const std::uint64_t size : sizes) {
        SceKernelProsperoSystemSwVersion version{};
        std::memset(&version, 0xaa, sizeof(version));
        version.size = size;
        Require(sceKernelGetProsperoSystemSwVersion_nid_postfix(&version) == SCE_KERNEL_ERROR_EINVAL);
        Require(version.size == size && version.version == 0xaaaaaaaa && static_cast<unsigned char>(version.versionString[0]) == 0xaa);
    }
    SceKernelProsperoSystemSwVersion version{};
    std::memset(&version, 0xaa, sizeof(version));
    version.size = sizeof(version);
    Require(sceKernelGetProsperoSystemSwVersion_nid_postfix(&version) == 0);
    Require(version.size == sizeof(version));
    Require(std::strcmp(version.versionString, "01.000.000") == 0);
    for (std::size_t i = sizeof("01.000.000"); i < sizeof(version.versionString); ++i) Require(version.versionString[i] == 0);
    Require(version.version == 0x01000000);
    Require(*__error_nid_postfix() == 13);
}

static void CheckAllowedSdkVersionOnSystem() {
    *__error_nid_postfix() = 13;
    Require(sceKernelGetAllowedSdkVersionOnSystem_nid_postfix(nullptr) == SCE_KERNEL_ERROR_EINVAL);
    std::uint32_t allowed = 0xaaaaaaaa;
    Require(sceKernelGetAllowedSdkVersionOnSystem_nid_postfix(&allowed) == 0);
    SceKernelProsperoSystemSwVersion system{};
    system.size = sizeof(system);
    Require(sceKernelGetProsperoSystemSwVersion_nid_postfix(&system) == 0);
    Require(allowed == (system.version | 0xfff) && allowed >> 16 == system.version >> 16);
    Require(*__error_nid_postfix() == 13);
}

int main() {
    CheckProsperoSystemSwVersion();
    CheckAllowedSdkVersionOnSystem();
    return 0;
}