mirror of
https://github.com/boykopovar/AnyPS5.git
synced 2026-10-23 17:13:13 -05:00
zaryabzip
b9e5dc160f
__wrap_access, __wrap_getcwd and __wrap_getenv forward to libc; __wrap_clock_gettime, __wrap_lstat and __wrap_mmap forward to libkernel.
201 lines
10 KiB
C++
201 lines
10 KiB
C++
#include "prx/libc/include/General.hpp"
|
|
#include "SceTypes.hpp"
|
|
#include <array>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <fstream>
|
|
#include <random>
|
|
#include <source_location>
|
|
#include <string>
|
|
#ifdef _WIN32
|
|
#include <fcntl.h>
|
|
#include <io.h>
|
|
#include <windows.h>
|
|
#else
|
|
#include <fcntl.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
extern "C" {
|
|
int APS5_VABI mkstemp_nid_postfix(char*);
|
|
int APS5_VABI isatty_nid_postfix(int);
|
|
int* APS5_VABI __error_nid_postfix();
|
|
int APS5_VABI close_nid_postfix(int);
|
|
int APS5_VABI pipe_nid_postfix(int*);
|
|
int APS5_VABI socket_nid_postfix(int, int, int);
|
|
std::int64_t APS5_VABI read_nid_postfix(int, void*, std::size_t);
|
|
std::int64_t APS5_VABI write_nid_postfix(int, const void*, std::size_t);
|
|
std::int64_t APS5_VABI lseek_nid_postfix(int, std::int64_t, int);
|
|
int APS5_VABI access_nid_postfix(const char*, int);
|
|
int APS5_VABI __wrap_access(const char*, int);
|
|
int APS5_VABI clock_gettime_nid_postfix(int, KernelTimespec*);
|
|
int APS5_VABI __wrap_clock_gettime(int, KernelTimespec*);
|
|
char* APS5_VABI getcwd_nid_postfix(char*, std::size_t);
|
|
char* APS5_VABI __wrap_getcwd(char*, std::size_t);
|
|
char* APS5_VABI getenv_nid_postfix(const char*);
|
|
char* APS5_VABI __wrap_getenv(const char*);
|
|
int APS5_VABI setenv_nid_postfix(const char*, const char*, int);
|
|
int APS5_VABI lstat_nid_postfix(const char*, FileStat*);
|
|
int APS5_VABI __wrap_lstat(const char*, FileStat*);
|
|
void* APS5_VABI mmap_nid_postfix(void*, std::size_t, int, int, int, std::int64_t) noexcept;
|
|
void* APS5_VABI __wrap_mmap(void*, std::size_t, int, int, int, std::int64_t);
|
|
int APS5_VABI munmap_nid_postfix(void*, std::size_t) noexcept;
|
|
}
|
|
|
|
static void Require(bool condition, std::source_location location = std::source_location::current()) {
|
|
if (!condition) {
|
|
std::fprintf(stderr, "POSIX filesystem check failed at line %u (guest errno %d)\n", location.line(), *__error_nid_postfix());
|
|
std::abort();
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
std::random_device random;
|
|
const auto root = std::filesystem::temp_directory_path() /
|
|
("anyps5-posix-test-" + std::to_string(random()) + "-" + std::to_string(random()));
|
|
Require(std::filesystem::create_directory(root));
|
|
AddPathAlias_nid_no_patch("/anyps5-posix-test", root.string().c_str());
|
|
const std::string prefix = "/anyps5-posix-test/file-";
|
|
std::string pattern = prefix + "XXXXXXXXX";
|
|
const int descriptor = mkstemp_nid_postfix(pattern.data());
|
|
Require(descriptor >= 0);
|
|
Require(pattern.starts_with(prefix) && pattern.size() == prefix.size() + 9);
|
|
Require(pattern.substr(prefix.size()).find_first_not_of("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz") == std::string::npos);
|
|
Require(std::filesystem::file_size(ResolvePath_nid_no_patch(pattern.c_str())) == 0);
|
|
#ifndef _WIN32
|
|
struct stat status{};
|
|
Require(::fstat(descriptor, &status) == 0 && (status.st_mode & 0777) == 0600);
|
|
#endif
|
|
const char payload[] = {'a', '\0', '\r', '\n', '\x1a', 'z'};
|
|
Require(write_nid_postfix(descriptor, payload, sizeof(payload)) == sizeof(payload));
|
|
Require(lseek_nid_postfix(descriptor, 0, SEEK_SET) == 0);
|
|
std::array<char, sizeof(payload)> received{};
|
|
Require(read_nid_postfix(descriptor, received.data(), received.size()) == static_cast<std::int64_t>(received.size()));
|
|
Require(std::memcmp(payload, received.data(), sizeof(payload)) == 0);
|
|
Require(isatty_nid_postfix(descriptor) == 0 && *__error_nid_postfix() == 25);
|
|
Require(close_nid_postfix(descriptor) == 0);
|
|
Require(isatty_nid_postfix(descriptor) == 0 && *__error_nid_postfix() == 9);
|
|
std::string second = prefix + "XXXXXXXXX";
|
|
const int other = mkstemp_nid_postfix(second.data());
|
|
Require(other >= 0 && second != pattern);
|
|
Require(close_nid_postfix(other) == 0);
|
|
const std::string candidates = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
|
|
for (const char character : candidates) {
|
|
std::ofstream stream(root / (std::string("occupied-") + character));
|
|
stream << "preserved";
|
|
Require(stream.good());
|
|
}
|
|
std::string occupied = "/anyps5-posix-test/occupied-X";
|
|
Require(mkstemp_nid_postfix(occupied.data()) == -1 && *__error_nid_postfix() == 17);
|
|
for (const char character : candidates) {
|
|
std::ifstream stream(root / (std::string("occupied-") + character));
|
|
std::string contents;
|
|
stream >> contents;
|
|
Require(contents == "preserved");
|
|
}
|
|
Require(std::filesystem::remove(root / "occupied-7"));
|
|
occupied = "/anyps5-posix-test/occupied-X";
|
|
const int collision = mkstemp_nid_postfix(occupied.data());
|
|
Require(collision >= 0 && occupied == "/anyps5-posix-test/occupied-7");
|
|
Require(close_nid_postfix(collision) == 0);
|
|
std::string fixed = "/anyps5-posix-test/fixed";
|
|
const int unchanged = mkstemp_nid_postfix(fixed.data());
|
|
Require(unchanged >= 0 && fixed == "/anyps5-posix-test/fixed");
|
|
Require(close_nid_postfix(unchanged) == 0);
|
|
Require(mkstemp_nid_postfix(fixed.data()) == -1 && *__error_nid_postfix() == 17);
|
|
std::string missing = "/anyps5-posix-test/missing/file-XXXXXX";
|
|
Require(mkstemp_nid_postfix(missing.data()) == -1 && *__error_nid_postfix() == 2);
|
|
std::string notDirectory = fixed + "/file-XXXXXX";
|
|
Require(mkstemp_nid_postfix(notDirectory.data()) == -1);
|
|
Require(*__error_nid_postfix() == 20 || *__error_nid_postfix() == 2);
|
|
std::string empty;
|
|
Require(mkstemp_nid_postfix(empty.data()) == -1 && *__error_nid_postfix() == 22);
|
|
Require(mkstemp_nid_postfix(nullptr) == -1 && *__error_nid_postfix() == 14);
|
|
std::string tooLong(1024, 'X');
|
|
Require(mkstemp_nid_postfix(tooLong.data()) == -1 && *__error_nid_postfix() == 63);
|
|
Require(tooLong == std::string(1024, 'X'));
|
|
Require(isatty_nid_postfix(-1) == 0 && *__error_nid_postfix() == 9);
|
|
Require(isatty_nid_postfix(100000) == 0 && *__error_nid_postfix() == 9);
|
|
int pipes[2]{};
|
|
Require(pipe_nid_postfix(pipes) == 0);
|
|
for (const int pipe : pipes) {
|
|
Require(isatty_nid_postfix(pipe) == 0 && *__error_nid_postfix() == 25);
|
|
Require(close_nid_postfix(pipe) == 0);
|
|
}
|
|
const int socket = socket_nid_postfix(2, 2, 0);
|
|
Require(socket >= 0 && socket < 1024);
|
|
Require(isatty_nid_postfix(socket) == 0 && *__error_nid_postfix() == 25);
|
|
Require(close_nid_postfix(socket) == 0);
|
|
Require(isatty_nid_postfix(socket) == 0 && *__error_nid_postfix() == 9);
|
|
#ifdef _WIN32
|
|
const int nullDevice = ::_open("NUL", _O_RDWR);
|
|
Require(nullDevice >= 0);
|
|
Require(isatty_nid_postfix(nullDevice) == 0 && *__error_nid_postfix() == 25);
|
|
Require(::_close(nullDevice) == 0);
|
|
const int console = ::_open("CONIN$", _O_RDONLY);
|
|
if (console >= 0) {
|
|
*__error_nid_postfix() = 123;
|
|
Require(isatty_nid_postfix(console) == 1 && *__error_nid_postfix() == 123);
|
|
Require(::_close(console) == 0);
|
|
}
|
|
#else
|
|
const int terminal = ::posix_openpt(O_RDWR | O_NOCTTY);
|
|
Require(terminal >= 0 && ::grantpt(terminal) == 0 && ::unlockpt(terminal) == 0);
|
|
const int slave = ::open(::ptsname(terminal), O_RDWR | O_NOCTTY);
|
|
Require(slave >= 0);
|
|
*__error_nid_postfix() = 123;
|
|
Require(isatty_nid_postfix(slave) == 1 && *__error_nid_postfix() == 123);
|
|
Require(::close(slave) == 0 && ::close(terminal) == 0);
|
|
#endif
|
|
Require(setenv_nid_postfix("WEBKIT_WRAP_SET", "set", 1) == 0);
|
|
char* const variable = getenv_nid_postfix("WEBKIT_WRAP_SET");
|
|
Require(variable != nullptr && std::strcmp(variable, "set") == 0 && __wrap_getenv("WEBKIT_WRAP_SET") == variable);
|
|
Require(getenv_nid_postfix("WEBKIT_WRAP_UNSET") == nullptr && __wrap_getenv("WEBKIT_WRAP_UNSET") == nullptr);
|
|
Require(access_nid_postfix(pattern.c_str(), 0) == 0 && __wrap_access(pattern.c_str(), 0) == 0);
|
|
Require(access_nid_postfix("/anyps5-posix-test/absent", 0) == -1);
|
|
const int accessError = *__error_nid_postfix();
|
|
*__error_nid_postfix() = 0;
|
|
Require(__wrap_access("/anyps5-posix-test/absent", 0) == -1 && *__error_nid_postfix() == accessError && accessError == 2);
|
|
std::array<char, 1024> expectedDirectory{};
|
|
std::array<char, 1024> wrappedDirectory{};
|
|
Require(getcwd_nid_postfix(expectedDirectory.data(), expectedDirectory.size()) == expectedDirectory.data());
|
|
Require(__wrap_getcwd(wrappedDirectory.data(), wrappedDirectory.size()) == wrappedDirectory.data());
|
|
Require(std::strcmp(expectedDirectory.data(), wrappedDirectory.data()) == 0);
|
|
Require(getcwd_nid_postfix(wrappedDirectory.data(), 1) == nullptr);
|
|
const int directoryError = *__error_nid_postfix();
|
|
*__error_nid_postfix() = 0;
|
|
Require(__wrap_getcwd(wrappedDirectory.data(), 1) == nullptr && *__error_nid_postfix() == directoryError && directoryError == 34);
|
|
FileStat expectedStatus{};
|
|
FileStat wrappedStatus{};
|
|
Require(lstat_nid_postfix(pattern.c_str(), &expectedStatus) == 0 && __wrap_lstat(pattern.c_str(), &wrappedStatus) == 0);
|
|
Require((wrappedStatus.st_mode & 0170000) == 0100000 && wrappedStatus.st_size == static_cast<std::int64_t>(sizeof(payload)));
|
|
Require(std::memcmp(&expectedStatus, &wrappedStatus, sizeof(FileStat)) == 0);
|
|
const auto notAfter = [](const KernelTimespec& earlier, const KernelTimespec& later) {
|
|
return earlier.tv_sec < later.tv_sec || (earlier.tv_sec == later.tv_sec && earlier.tv_nsec <= later.tv_nsec);
|
|
};
|
|
KernelTimespec before{};
|
|
KernelTimespec during{};
|
|
KernelTimespec after{};
|
|
Require(clock_gettime_nid_postfix(4, &before) == 0 && __wrap_clock_gettime(4, &during) == 0 && clock_gettime_nid_postfix(4, &after) == 0);
|
|
Require(notAfter(before, during) && notAfter(during, after));
|
|
Require(clock_gettime_nid_postfix(4, nullptr) == -1 && *__error_nid_postfix() == 14);
|
|
*__error_nid_postfix() = 0;
|
|
Require(__wrap_clock_gettime(4, nullptr) == -1 && *__error_nid_postfix() == 14);
|
|
constexpr std::size_t page = 0x4000;
|
|
void* const failed = reinterpret_cast<void*>(static_cast<std::uintptr_t>(-1));
|
|
void* const mapped = __wrap_mmap(nullptr, page, 3, 0x1002, -1, 0);
|
|
Require(mapped != failed);
|
|
auto* const bytes = static_cast<volatile unsigned char*>(mapped);
|
|
bytes[0] = 0x5a;
|
|
bytes[page - 1] = 0xa5;
|
|
Require(bytes[0] == 0x5a && bytes[page - 1] == 0xa5);
|
|
Require(munmap_nid_postfix(mapped, page) == 0);
|
|
Require(mmap_nid_postfix(nullptr, page, 8, 0x1002, -1, 0) == failed && *__error_nid_postfix() == 22);
|
|
*__error_nid_postfix() = 0;
|
|
Require(__wrap_mmap(nullptr, page, 8, 0x1002, -1, 0) == failed && *__error_nid_postfix() == 22);
|
|
RemovePathAlias_nid_no_patch("/anyps5-posix-test");
|
|
Require(std::filesystem::remove_all(root) > 0);
|
|
}
|