AnyPS5/core/libs/tests/GuestAccess.cpp

#include "prx/libc/include/Api.hpp"
#include "prx/libc/include/General.hpp"
#include <chrono>
#include <cstdio>
#include <cstdlib>
#include <fstream>
#ifdef _WIN32
#ifndef NOMINMAX
#define NOMINMAX
#endif
#include <windows.h>
#include <aclapi.h>
#endif

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

int main() {
    const auto name = "anyps5-access-test-" + std::to_string(std::chrono::steady_clock::now().time_since_epoch().count());
    const auto directory = std::filesystem::current_path() / name;
    Require(std::filesystem::create_directory(directory));
    const auto file = directory / "sample.txt";
    { std::ofstream output(file); output << "sample"; }
    Require(chdir_nid_postfix(name.c_str()) == 0);
    *__error_nid_postfix() = 34;
    Require(access_nid_postfix("sample.txt", 0) == 0 && *__error_nid_postfix() == 34);
    Require(access_nid_postfix("sample.txt", 6) == 0);
    Require(access_nid_postfix(".", 7) == 0);
    Require(access_nid_postfix(("/" + name + "/sample.txt").c_str(), 4) == 0);
    AddPathAlias_nid_no_patch("access-alias", directory.string().c_str());
    Require(access_nid_postfix("/access-alias/sample.txt", 0) == 0);
    RemovePathAlias_nid_no_patch("access-alias");
    Require(access_nid_postfix("missing", 0) == -1 && *__error_nid_postfix() == 2);
    Require(access_nid_postfix("", 0) == -1 && *__error_nid_postfix() == 2);
    Require(access_nid_postfix(nullptr, 0) == -1 && *__error_nid_postfix() == 14);
    Require(access_nid_postfix("sample.txt", 8) == -1 && *__error_nid_postfix() == 22);
    Require(access_nid_postfix("sample.txt", -1) == -1 && *__error_nid_postfix() == 22);
    Require(access_nid_postfix(nullptr, 8) == -1 && *__error_nid_postfix() == 22);
    Require(access_nid_postfix(nullptr, -1) == -1 && *__error_nid_postfix() == 22);
    Require(access_nid_postfix("", 8) == -1 && *__error_nid_postfix() == 22);
    Require(access_nid_postfix("", -1) == -1 && *__error_nid_postfix() == 22);
#ifdef _WIN32
    Require(SetFileAttributesW(file.c_str(), FILE_ATTRIBUTE_READONLY));
    Require(access_nid_postfix("sample.txt", 4) == 0);
    Require(access_nid_postfix("sample.txt", 2) == -1 && *__error_nid_postfix() == 13);
    Require(SetFileAttributesW(file.c_str(), FILE_ATTRIBUTE_NORMAL));
    PSECURITY_DESCRIPTOR original = nullptr;
    PACL originalAcl = nullptr;
    Require(GetNamedSecurityInfoW(const_cast<wchar_t*>(file.c_str()), SE_FILE_OBJECT,
        DACL_SECURITY_INFORMATION, nullptr, nullptr, &originalAcl, nullptr, &original) == ERROR_SUCCESS);
    char sid[SECURITY_MAX_SID_SIZE];
    DWORD sidBytes = sizeof(sid);
    Require(CreateWellKnownSid(WinWorldSid, nullptr, sid, &sidBytes));
    EXPLICIT_ACCESSW denial{};
    denial.grfAccessPermissions = FILE_EXECUTE;
    denial.grfAccessMode = DENY_ACCESS;
    denial.Trustee.TrusteeForm = TRUSTEE_IS_SID;
    denial.Trustee.ptstrName = reinterpret_cast<wchar_t*>(sid);
    PACL deniedAcl = nullptr;
    Require(SetEntriesInAclW(1, &denial, originalAcl, &deniedAcl) == ERROR_SUCCESS);
    Require(SetNamedSecurityInfoW(const_cast<wchar_t*>(file.c_str()), SE_FILE_OBJECT,
        DACL_SECURITY_INFORMATION, nullptr, nullptr, deniedAcl, nullptr) == ERROR_SUCCESS);
    const int denied = access_nid_postfix("sample.txt", 1);
    const int deniedError = *__error_nid_postfix();
    const int readable = access_nid_postfix("sample.txt", 4);
    Require(SetNamedSecurityInfoW(const_cast<wchar_t*>(file.c_str()), SE_FILE_OBJECT,
        DACL_SECURITY_INFORMATION, nullptr, nullptr, originalAcl, nullptr) == ERROR_SUCCESS);
    LocalFree(deniedAcl);
    LocalFree(original);
    Require(denied == -1 && deniedError == 13 && readable == 0);
#else
    std::filesystem::permissions(file, std::filesystem::perms::owner_read | std::filesystem::perms::owner_write);
    Require(access_nid_postfix("sample.txt", 1) == -1 && *__error_nid_postfix() == 13);
    std::filesystem::permissions(file, std::filesystem::perms::owner_exec, std::filesystem::perm_options::add);
    Require(access_nid_postfix("sample.txt", 1) == 0);
#endif
    const auto dangling = directory / "dangling.link";
    const auto pointing = directory / "sample.link";
    const auto loop = directory / "loop.link";
#ifdef _WIN32
    const DWORD linkFlags = SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE;
    const bool linked = CreateSymbolicLinkW(dangling.c_str(), L"missing.bin", linkFlags) &&
        CreateSymbolicLinkW(pointing.c_str(), L"sample.txt", linkFlags) && CreateSymbolicLinkW(loop.c_str(), L"loop.link", linkFlags);
#else
    std::error_code linkError;
    std::filesystem::create_symlink("missing.bin", dangling, linkError);
    if (!linkError) std::filesystem::create_symlink("sample.txt", pointing, linkError);
    if (!linkError) std::filesystem::create_symlink("loop.link", loop, linkError);
    const bool linked = !linkError;
#endif
    if (linked) {
        Require(access_nid_postfix("dangling.link", 0) == -1 && *__error_nid_postfix() == 2);
        Require(access_nid_postfix("dangling.link", 4) == -1 && *__error_nid_postfix() == 2);
        Require(access_nid_postfix("loop.link", 0) == -1 && *__error_nid_postfix() == 62);
        Require(access_nid_postfix("sample.link", 6) == 0);
#ifdef _WIN32
        Require(SetFileAttributesW(file.c_str(), FILE_ATTRIBUTE_READONLY));
        Require(access_nid_postfix("sample.link", 4) == 0);
        Require(access_nid_postfix("sample.link", 2) == -1 && *__error_nid_postfix() == 13);
        Require(SetFileAttributesW(file.c_str(), FILE_ATTRIBUTE_NORMAL));
        Require(SetFileAttributesW(pointing.c_str(), FILE_ATTRIBUTE_READONLY));
        Require(access_nid_postfix("sample.link", 2) == 0);
        Require(SetFileAttributesW(pointing.c_str(), FILE_ATTRIBUTE_NORMAL));
#endif
    } else {
        std::fprintf(stderr, "symbolic links unavailable: the access link cases not tested\n");
    }
#ifdef _WIN32
    DeleteFileW(dangling.c_str());
    DeleteFileW(pointing.c_str());
    DeleteFileW(loop.c_str());
#else
    std::filesystem::remove(dangling, linkError);
    std::filesystem::remove(pointing, linkError);
    std::filesystem::remove(loop, linkError);
#endif
    Require(chdir_nid_postfix("/") == 0);
    std::filesystem::remove(file);
    std::filesystem::remove(directory);
}