AnyPS5/core/libs/tests/GuestSendfile.cpp

#include "prx/libkernel/include/Api.hpp"
#include "prx/libc/include/Api.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <cstdint>
#include <cstdlib>
#include <filesystem>
#include <fstream>
#include <string>
#include <thread>
#include <vector>

static void Require(bool value) { if (!value) std::abort(); }
static bool Fails(int result, int error) { return result == -1 && *__error_nid_postfix() == error; }
static std::string Receive(int descriptor, std::size_t length) {
    std::string data(length, 0);
    std::size_t done = 0;
    while (done < length) {
        const auto result = recv_nid_postfix(descriptor, data.data() + done, length - done, 0);
        Require(result > 0);
        done += static_cast<std::size_t>(result);
    }
    return data;
}
static std::string Content(std::size_t size) {
    std::string content(size, 0);
    for (std::size_t i = 0; i < size; ++i) content[i] = static_cast<char>(i * 7 % 251);
    return content;
}
static int OpenWritten(const char* path, const std::string& content) {
    {
        std::ofstream stream(path, std::ios::binary);
        stream.write(content.data(), static_cast<std::streamsize>(content.size()));
    }
    const int file = open_nid_postfix(path, 0, 0);
    Require(file >= 0);
    return file;
}
int main() {
    const std::string content = Content(200000);
    const auto size = static_cast<std::int64_t>(content.size());
    const int file = OpenWritten("sendfile_probe.bin", content);
    int pair[2];
    Require(socketpair_nid_postfix(1, 1, 0, pair) == 0);
    KernelIovec header[] = {{const_cast<char*>("GET"), 3}, {const_cast<char*>(" "), 1}};
    KernelIovec trailer[] = {{const_cast<char*>("END"), 3}};
    const GuestSendfileHeaders parts{header, 2, trailer, 1};
    std::int64_t sent = -1;
    Require(sendfile_nid_postfix(file, pair[0], 5, 10, &parts, &sent, 0) == 0 && sent == 17);
    Require(Receive(pair[1], 17) == "GET " + content.substr(5, 10) + "END");
    Require(lseek_nid_postfix(file, 0, 1) == 0);
    std::string rest;
    std::thread reader([&] { rest = Receive(pair[1], content.size() - 100); });
    Require(sendfile_nid_postfix(file, pair[0], 100, 0, nullptr, &sent, 0) == 0 && sent == size - 100);
    reader.join();
    Require(rest == content.substr(100));
    Require(sendfile_nid_postfix(file, pair[0], size - 4, 1000, nullptr, &sent, 0) == 0 && sent == 4);
    Require(Receive(pair[1], 4) == content.substr(content.size() - 4));
    Require(sendfile_nid_postfix(file, pair[0], size + 10, 0, &parts, &sent, 0) == 0 && sent == 7);
    Require(Receive(pair[1], 7) == "GET END");
    const GuestSendfileHeaders nullHeaders{nullptr, 5, trailer, 1};
    Require(sendfile_nid_postfix(file, pair[0], 0, 3, &nullHeaders, nullptr, 0) == 0);
    Require(Receive(pair[1], 6) == content.substr(0, 3) + "END");
    Require(lseek_nid_postfix(file, 0, 1) == 0);

    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, pair[0], -1, 0, nullptr, &sent, 0), 22) && sent == 77);
    std::vector<KernelIovec> many(1025, KernelIovec{const_cast<char*>("x"), 1});
    const GuestSendfileHeaders tooMany{many.data(), 1025, nullptr, 0};
    Require(Fails(sendfile_nid_postfix(file, pair[0], 0, 0, &tooMany, &sent, 0), 22) && sent == 77);
    const GuestSendfileHeaders negative{nullptr, 0, trailer, -1};
    Require(Fails(sendfile_nid_postfix(file, pair[0], 0, 0, &negative, &sent, 0), 22) && sent == 77);
    Require(Fails(sendfile_nid_postfix(pair[1], pair[0], 0, 0, nullptr, &sent, 0), 22) && sent == 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, file, 0, 0, nullptr, &sent, 0), 38) && sent == 0);
    const int datagram = socket_nid_postfix(2, 2, 0);
    Require(datagram >= 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, datagram, 0, 0, nullptr, &sent, 0), 22) && sent == 0);
    const int unconnected = socket_nid_postfix(2, 1, 0);
    Require(unconnected >= 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, unconnected, size + 10, 0, nullptr, &sent, 0), 57) && sent == 0);
    const int random = open_nid_postfix("/dev/urandom", 0, 0);
    Require(random >= 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(random, pair[0], 0, 16, &parts, &sent, 0), 22) && sent == 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(random, pair[0], 0, 0, nullptr, &sent, 0), 22) && sent == 0);
    const int queue = kqueue_nid_postfix();
    Require(queue >= 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(queue, pair[0], 0, 0, nullptr, &sent, 0), 22) && sent == 0);
    Require(close_nid_postfix(queue) == 0 && close_nid_postfix(random) == 0);

    const std::string large = Content(32 << 20);
    const int largeFile = OpenWritten("sendfile_large.bin", large);
    int on = 1;
    Require(ioctl_nid_postfix(pair[0], 0x8004667e, &on) == 0);
    sent = -1;
    Require(Fails(sendfile_nid_postfix(largeFile, pair[0], 0, 0, nullptr, &sent, 0), 35));
    Require(sent > 0 && sent < static_cast<std::int64_t>(large.size()));
    Require(Receive(pair[1], static_cast<std::size_t>(sent)) == large.substr(0, static_cast<std::size_t>(sent)));

    Require(close_nid_postfix(unconnected) == 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, unconnected, 0, 0, nullptr, &sent, 0), 9) && sent == 0);
    Require(close_nid_postfix(largeFile) == 0 && close_nid_postfix(file) == 0);
    sent = 77;
    Require(Fails(sendfile_nid_postfix(file, pair[0], 0, 0, nullptr, &sent, 0), 9) && sent == 77);
    Require(close_nid_postfix(datagram) == 0);
    Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
    std::filesystem::remove("sendfile_probe.bin");
    std::filesystem::remove("sendfile_large.bin");
}