AnyPS5/core/libs/tests/GuestGameUpdate.cpp

#include "SceTypes.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <cstdint>
#include <cstdlib>
#include <cstring>
#include <initializer_list>
#include <stdexcept>

extern "C" {
int APS5_VABI sceGameUpdateInitialize(void);
int APS5_VABI sceGameUpdateTerminate(void);
int APS5_VABI sceGameUpdateGetAddcontLatestVersion(std::uint32_t, const NpUnifiedEntitlementLabel*, GameUpdateAddcontVersionInfo*);
int APS5_VABI sceGameUpdateCreateRequest(void);
int APS5_VABI sceGameUpdateCheck(int, const GameUpdateCheckParam*, GameUpdateCheckResult*);
int APS5_VABI sceGameUpdateAbortRequest(int);
int APS5_VABI sceGameUpdateDeleteRequest(int);
}

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

template <typename TFunction>
static void RequireThrows(TFunction function) {
    try {
        function();
    } catch (const std::runtime_error&) {
        return;
    }
    std::abort();
}

static void TestRequestArguments(int notInitialized, int invalidArgument, int invalidSize, int notFound) {
    GameUpdateCheckParam param{};
    param.size = sizeof(param);
    GameUpdateCheckResult result{};
    result.size = sizeof(result);
    constexpr int requestId = 0x20000000;

    for (int id : {0, -1}) {
        Require(sceGameUpdateCheck(id, nullptr, nullptr) == notFound);
        Require(sceGameUpdateAbortRequest(id) == notFound);
        Require(sceGameUpdateDeleteRequest(id) == notFound);
    }
    Require(sceGameUpdateCheck(requestId, nullptr, &result) == invalidArgument);
    Require(sceGameUpdateCheck(requestId, &param, nullptr) == invalidArgument);
    param.size = sizeof(param) - 1;
    Require(sceGameUpdateCheck(requestId, &param, &result) == invalidSize);
    param.size = sizeof(param);
    result.size = sizeof(result) + 8;
    Require(sceGameUpdateCheck(requestId, &param, &result) == invalidSize);
    result.size = sizeof(result);
    param.option = 1;
    Require(sceGameUpdateCheck(requestId, &param, &result) == invalidArgument);
    param.option = 0;
    param.reserved[8] = 1;
    Require(sceGameUpdateCheck(requestId, &param, &result) == invalidArgument);
    param.reserved[8] = 0;

    Require(sceGameUpdateCreateRequest() == notInitialized);
    Require(sceGameUpdateCheck(requestId, &param, &result) == notInitialized);
    Require(sceGameUpdateAbortRequest(requestId) == notInitialized);
    Require(sceGameUpdateDeleteRequest(requestId) == notInitialized);

    Require(sceGameUpdateInitialize() == 0);
    Require(sceGameUpdateCheck(requestId, &param, &result) == notFound);
    Require(sceGameUpdateAbortRequest(requestId) == notFound);
    Require(sceGameUpdateDeleteRequest(requestId) == notFound);
    Require(sceGameUpdateTerminate() == 0);
}

int main() {
    constexpr int notInitialized = static_cast<int>(0x80412801);
    constexpr int invalidArgument = static_cast<int>(0x80412803);
    constexpr int invalidSize = static_cast<int>(0x80412804);

    static_assert(sizeof(GameUpdateAddcontVersionInfo) == 0x30);
    const NpUnifiedEntitlementLabel label{};
    GameUpdateAddcontVersionInfo info{};
    info.size = sizeof(info);

    constexpr int alreadyInitialized = static_cast<int>(0x80412802);
    constexpr int notFound = static_cast<int>(0x80412805);

    Require(sceGameUpdateTerminate() == notInitialized);
    TestRequestArguments(notInitialized, invalidArgument, invalidSize, notFound);

    Require(sceGameUpdateGetAddcontLatestVersion(0, &label, &info) == notInitialized);
    Require(sceGameUpdateGetAddcontLatestVersion(0, &label, nullptr) == invalidArgument);
    Require(sceGameUpdateGetAddcontLatestVersion(0, nullptr, &info) == invalidArgument);
    Require(sceGameUpdateInitialize() == 0);
    Require(sceGameUpdateInitialize() == alreadyInitialized);
    Require(sceGameUpdateGetAddcontLatestVersion(0, &label, nullptr) == invalidArgument);

    Require(sceGameUpdateGetAddcontLatestVersion(0, nullptr, &info) == invalidArgument);

    info.size = sizeof(info) - 1;
    RequireThrows([&] { sceGameUpdateGetAddcontLatestVersion(0, &label, &info); });

    std::memset(&info, 0xAB, sizeof(info));
    info.size = sizeof(info) + 8;
    const auto original = info;
    RequireThrows([&] { sceGameUpdateGetAddcontLatestVersion(0, &label, &info); });
    Require(std::memcmp(&info, &original, sizeof(info)) == 0);
    RequireThrows([] { sceGameUpdateCreateRequest(); });

    Require(sceGameUpdateTerminate() == 0);
    Require(sceGameUpdateTerminate() == notInitialized);
    info.size = sizeof(info);
    Require(sceGameUpdateGetAddcontLatestVersion(0, &label, &info) == notInitialized);
}