mirror of
https://github.com/boykopovar/AnyPS5.git
synced 2026-10-28 12:05:06 +00:00
Retr0
3383d303a7
FreeBSD 11.4 kqueue_scan (sys/kern/kern_event.c) puts a reported level knote back at the tail of the queue and takes an EV_CLEAR or EV_DISPATCH knote off it, and knote_enqueue queues that one at the tail when it fires again. kevent moved a reported knote to the tail only while it stayed triggered, so a level socket or pipe knote and a re-fired EV_CLEAR or EV_DISPATCH knote kept their place and could starve later knotes when nevents is smaller than the number of ready ones. Now every reported knote that is not deleted moves to the tail.
1226 lines
59 KiB
C++
1226 lines
59 KiB
C++
#include "SceTypes.hpp"
|
|
#include <algorithm>
|
|
#include <cctype>
|
|
#include <chrono>
|
|
#include <cstdint>
|
|
#include <ctime>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <filesystem>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <vector>
|
|
#ifdef _WIN32
|
|
#include <aclapi.h>
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
extern "C" {
|
|
int APS5_VABI kqueue_nid_postfix();
|
|
int APS5_VABI kevent_nid_postfix(int, const KernelEvent*, int, KernelEvent*, int, const KernelTimespec*);
|
|
int APS5_VABI close_nid_postfix(int);
|
|
int APS5_VABI dup_nid_postfix(int);
|
|
int APS5_VABI pipe_nid_postfix(int*);
|
|
int APS5_VABI fcntl_nid_postfix(int, int, ...);
|
|
int APS5_VABI socketpair_nid_postfix(int, int, int, int*);
|
|
std::int64_t APS5_VABI send_nid_postfix(int, const void*, std::uint64_t, int);
|
|
std::int64_t APS5_VABI recv_nid_postfix(int, void*, std::uint64_t, int);
|
|
std::int64_t APS5_VABI read_nid_postfix(int, void*, std::uint64_t);
|
|
int APS5_VABI open_nid_postfix(const char*, int, int);
|
|
int APS5_VABI mkdir_nid_postfix(const char*, unsigned short);
|
|
int APS5_VABI rmdir_nid_postfix(const char*);
|
|
std::int64_t APS5_VABI write_nid_postfix(int, const char*, std::int64_t);
|
|
int APS5_VABI ftruncate_nid_postfix(int, std::int64_t);
|
|
int APS5_VABI unlink_nid_postfix(const char*);
|
|
int APS5_VABI sceKernelRename(const char*, const char*);
|
|
int APS5_VABI sceKernelCreateEqueue(KernelEqueue* eq, const char* name);
|
|
int APS5_VABI sceKernelDeleteEqueue(KernelEqueue eq);
|
|
int APS5_VABI sceKernelWaitEqueue(KernelEqueue eq, KernelEvent* ev, int num, int* out, const KernelUseconds* timo);
|
|
int APS5_VABI sceKernelAddFileEvent(KernelEqueue eq, int fd, int watch, void* udata);
|
|
int APS5_VABI sceKernelDeleteFileEvent(KernelEqueue eq, int fd);
|
|
int* APS5_VABI __error_nid_postfix();
|
|
}
|
|
|
|
static constexpr int GuestEnoent = 2;
|
|
static constexpr int GuestEbadf = 9;
|
|
static constexpr int GuestEinval = 22;
|
|
static constexpr int GuestEpipe = 32;
|
|
static constexpr int GuestEagain = 35;
|
|
static constexpr int SetStatusFlags = 4;
|
|
static constexpr int NonBlocking = 0x0004;
|
|
static constexpr std::int64_t PipeAtomic = 512;
|
|
static constexpr int Unix = 1;
|
|
static constexpr int Stream = 1;
|
|
static constexpr int FirstSocket = 0x10000000;
|
|
static constexpr std::int16_t EvfiltRead = -1;
|
|
static constexpr std::int16_t EvfiltWrite = -2;
|
|
static constexpr std::int16_t EvfiltVnode = -4;
|
|
static constexpr std::int16_t EvfiltProc = -5;
|
|
static constexpr std::int16_t EvfiltTimer = -7;
|
|
static constexpr std::int16_t EvfiltUser = -11;
|
|
static constexpr std::uint16_t EvAdd = 0x0001;
|
|
static constexpr std::uint16_t EvDelete = 0x0002;
|
|
static constexpr std::uint16_t EvEnable = 0x0004;
|
|
static constexpr std::uint16_t EvDisable = 0x0008;
|
|
static constexpr std::uint16_t EvOneshot = 0x0010;
|
|
static constexpr std::uint16_t EvClear = 0x0020;
|
|
static constexpr std::uint16_t EvReceipt = 0x0040;
|
|
static constexpr std::uint16_t EvDispatch = 0x0080;
|
|
static constexpr std::uint16_t EvForceoneshot = 0x0100;
|
|
static constexpr std::uint16_t EvError = 0x4000;
|
|
static constexpr std::uint16_t EvEof = 0x8000;
|
|
static constexpr std::uint32_t NoteTrigger = 0x01000000;
|
|
static constexpr std::uint32_t NoteLowat = 0x0001;
|
|
static constexpr std::uint32_t NoteFfcopy = 0xc0000000;
|
|
static constexpr std::uint32_t NoteFfor = 0x80000000;
|
|
static constexpr std::uint32_t NoteUseconds = 0x0004;
|
|
static constexpr std::uint32_t NoteNseconds = 0x0008;
|
|
static constexpr std::uint32_t NoteDelete = 0x0001;
|
|
static constexpr std::uint32_t NoteWrite = 0x0002;
|
|
static constexpr std::uint32_t NoteExtend = 0x0004;
|
|
static constexpr std::uint32_t NoteAttrib = 0x0008;
|
|
static constexpr std::uint32_t NoteLink = 0x0010;
|
|
static constexpr std::uint32_t NoteRename = 0x0020;
|
|
static constexpr std::uint32_t NoteRevoke = 0x0040;
|
|
static constexpr int ReadWrite = 0x0002;
|
|
static constexpr int Create = 0x0200;
|
|
static constexpr int Directory = 0x00020000;
|
|
static constexpr int SceOk = 0;
|
|
static constexpr int SceEnoent = static_cast<int>(0x80020002);
|
|
static constexpr int SceEbadf = static_cast<int>(0x80020009);
|
|
static constexpr int SceEinval = static_cast<int>(0x80020016);
|
|
|
|
static void Require(bool value) { if (!value) std::abort(); }
|
|
|
|
static KernelEvent Change(std::uintptr_t ident, std::int16_t filter, std::uint16_t flags, std::uint32_t fflags = 0, std::intptr_t data = 0, void* udata = nullptr) {
|
|
KernelEvent change{};
|
|
change.ident = ident;
|
|
change.filter = filter;
|
|
change.flags = flags;
|
|
change.fflags = fflags;
|
|
change.data = data;
|
|
change.udata = udata;
|
|
return change;
|
|
}
|
|
|
|
static int Apply(int kq, const KernelEvent& change) {
|
|
return kevent_nid_postfix(kq, &change, 1, nullptr, 0, nullptr);
|
|
}
|
|
|
|
static void RequireFailure(int result, int error) {
|
|
Require(result == -1);
|
|
Require(*__error_nid_postfix() == error);
|
|
}
|
|
|
|
static int Poll(int kq, KernelEvent* events, int count) {
|
|
const KernelTimespec zero{0, 0};
|
|
return kevent_nid_postfix(kq, nullptr, 0, events, count, &zero);
|
|
}
|
|
|
|
static int Wait(int kq, KernelEvent* events, int count) {
|
|
const KernelTimespec second{1, 0};
|
|
return kevent_nid_postfix(kq, nullptr, 0, events, count, &second);
|
|
}
|
|
|
|
template <typename TCall> static bool Throws(TCall call) {
|
|
try {
|
|
call();
|
|
} catch (const std::runtime_error&) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template <typename TCall> static std::string ThrownMessage(TCall call) {
|
|
try {
|
|
call();
|
|
} catch (const std::runtime_error& error) {
|
|
return error.what();
|
|
}
|
|
return {};
|
|
}
|
|
|
|
static void CreateAndClose() {
|
|
const int kq = kqueue_nid_postfix();
|
|
Require(kq >= 0 && kq < FirstSocket);
|
|
const int other = kqueue_nid_postfix();
|
|
Require(other >= 0 && other != kq);
|
|
KernelEvent event{};
|
|
Require(Poll(kq, &event, 1) == 0);
|
|
Require(Throws([kq] { dup_nid_postfix(kq); }));
|
|
Require(close_nid_postfix(kq) == 0);
|
|
RequireFailure(Poll(kq, &event, 1), GuestEbadf);
|
|
Require(close_nid_postfix(other) == 0);
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
RequireFailure(Poll(pipe[0], &event, 1), GuestEbadf);
|
|
Require(close_nid_postfix(pipe[0]) == 0 && close_nid_postfix(pipe[1]) == 0);
|
|
}
|
|
|
|
static void Timers() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(1, EvfiltTimer, EvAdd, 0, 10, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Wait(kq, events, 4) == 1);
|
|
Require(events[0].ident == 1 && events[0].filter == EvfiltTimer && events[0].data >= 1 && events[0].udata == &marker);
|
|
Require((events[0].flags & EvClear) != 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == 1);
|
|
Require(Apply(kq, Change(1, EvfiltTimer, EvDelete)) == 0);
|
|
RequireFailure(Apply(kq, Change(1, EvfiltTimer, EvDelete)), GuestEnoent);
|
|
|
|
Require(Apply(kq, Change(2, EvfiltTimer, EvAdd | EvOneshot, NoteUseconds, 0)) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == 2 && events[0].data == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
RequireFailure(Apply(kq, Change(2, EvfiltTimer, EvDelete)), GuestEnoent);
|
|
|
|
Require(Apply(kq, Change(5, EvfiltTimer, EvAdd | EvOneshot, NoteNseconds, 1500000001)) == 0);
|
|
const KernelTimespec twoSeconds{2, 0};
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &twoSeconds) == 1);
|
|
Require(events[0].ident == 5 && events[0].data == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
|
|
RequireFailure(Apply(kq, Change(3, EvfiltTimer, EvAdd, 0, -1)), GuestEinval);
|
|
RequireFailure(Apply(kq, Change(3, EvfiltTimer, EvAdd, 0x10, 5)), GuestEinval);
|
|
RequireFailure(Apply(kq, Change(3, EvfiltTimer, EvAdd, NoteUseconds | 0x1, 5)), GuestEinval);
|
|
Require(Apply(kq, Change(4, EvfiltTimer, EvAdd, 0, 100000)) == 0);
|
|
Require(Throws([kq] { Apply(kq, Change(4, EvfiltTimer, EvAdd, 0, 5)); }));
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void UserEvents() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(5, EvfiltUser, EvAdd | EvClear, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
std::thread trigger([kq, &marker] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(20));
|
|
Require(Apply(kq, Change(5, EvfiltUser, 0, NoteTrigger | NoteFfcopy | 0x123, 77, &marker)) == 0);
|
|
});
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, nullptr) == 1);
|
|
trigger.join();
|
|
Require(events[0].ident == 5 && events[0].filter == EvfiltUser && events[0].fflags == 0x123 && events[0].data == 77);
|
|
Require(events[0].udata == &marker && events[0].flags == EvClear);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(5, EvfiltUser, 0, NoteTrigger | NoteFfor | 0x400, 78)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].fflags == 0x523 && events[0].data == 78);
|
|
|
|
Require(Apply(kq, Change(6, EvfiltUser, EvAdd)) == 0);
|
|
Require(Apply(kq, Change(6, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
for (int i = 0; i < 2; ++i) {
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == 6);
|
|
}
|
|
Require(Apply(kq, Change(6, EvfiltUser, EvClear)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(6, EvfiltUser, EvDelete)) == 0);
|
|
|
|
KernelEvent receipt[2]{};
|
|
const KernelEvent change = Change(7, EvfiltUser, EvAdd | EvReceipt, 0x55, 9, &marker);
|
|
Require(kevent_nid_postfix(kq, &change, 1, receipt, 2, nullptr) == 1);
|
|
Require(receipt[0].ident == 7 && receipt[0].flags == EvError && receipt[0].data == 0 && receipt[0].fflags == 0);
|
|
Require(Apply(kq, Change(7, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].fflags == 0x55 && events[0].data == 0 && events[0].flags == EvReceipt);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void PipeRead() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
const auto readEnd = static_cast<std::uintptr_t>(pipe[0]);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(readEnd, EvfiltRead, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(write_nid_postfix(pipe[1], "abc", 3) == 3);
|
|
Require(Poll(kq, events, 4) == 1);
|
|
Require(events[0].ident == readEnd && events[0].filter == EvfiltRead && events[0].data == 3 && events[0].udata == &marker && (events[0].flags & EvEof) == 0);
|
|
char buffer[4]{};
|
|
Require(read_nid_postfix(pipe[0], buffer, sizeof(buffer)) == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
const KernelTimespec limit{5, 0};
|
|
const auto start = std::chrono::steady_clock::now();
|
|
std::thread writer([&pipe] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(write_nid_postfix(pipe[1], "d", 1) == 1);
|
|
});
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &limit) == 1);
|
|
const auto waited = std::chrono::steady_clock::now() - start;
|
|
writer.join();
|
|
Require(waited >= std::chrono::milliseconds(40) && waited < std::chrono::seconds(4));
|
|
Require(events[0].ident == readEnd && events[0].data == 1);
|
|
Require(close_nid_postfix(pipe[1]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 1 && (events[0].flags & EvEof) != 0);
|
|
Require(read_nid_postfix(pipe[0], buffer, sizeof(buffer)) == 1);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 0 && (events[0].flags & EvEof) != 0);
|
|
Require(close_nid_postfix(pipe[0]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
#ifndef __APPLE__
|
|
static void PipeWrite() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
const auto writeEnd = static_cast<std::uintptr_t>(pipe[1]);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(writeEnd, EvfiltWrite, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 1);
|
|
Require(events[0].ident == writeEnd && events[0].filter == EvfiltWrite && events[0].udata == &marker && (events[0].flags & EvEof) == 0);
|
|
const std::int64_t empty = events[0].data;
|
|
Require(empty >= PipeAtomic);
|
|
Require(write_nid_postfix(pipe[1], "abc", 3) == 3);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == empty - 3);
|
|
#ifndef _WIN32
|
|
Require(fcntl_nid_postfix(pipe[1], SetStatusFlags, NonBlocking) == 0);
|
|
const std::string chunk(4096, 'x');
|
|
std::int64_t queued = 3;
|
|
for (;;) {
|
|
const std::int64_t written = write_nid_postfix(pipe[1], chunk.data(), static_cast<std::int64_t>(chunk.size()));
|
|
if (written < 0) {
|
|
Require(*__error_nid_postfix() == GuestEagain);
|
|
break;
|
|
}
|
|
queued += written;
|
|
}
|
|
Require(queued > PipeAtomic);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
const KernelTimespec limit{5, 0};
|
|
const auto start = std::chrono::steady_clock::now();
|
|
std::thread reader([&pipe, queued] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
std::string drained(static_cast<std::size_t>(queued), '\0');
|
|
std::int64_t done = 0;
|
|
while (done < queued) {
|
|
const std::int64_t got = read_nid_postfix(pipe[0], drained.data() + done, static_cast<std::uint64_t>(queued - done));
|
|
Require(got > 0);
|
|
done += got;
|
|
}
|
|
});
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &limit) == 1);
|
|
const auto waited = std::chrono::steady_clock::now() - start;
|
|
reader.join();
|
|
Require(waited >= std::chrono::milliseconds(40) && waited < std::chrono::seconds(4));
|
|
Require(events[0].ident == writeEnd && events[0].data >= PipeAtomic && (events[0].flags & EvEof) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == empty);
|
|
#else
|
|
char back[3];
|
|
Require(read_nid_postfix(pipe[0], back, 3) == 3);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == empty);
|
|
#endif
|
|
Require(close_nid_postfix(pipe[0]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 0 && (events[0].flags & EvEof) != 0);
|
|
const int late = kqueue_nid_postfix();
|
|
RequireFailure(Apply(late, Change(writeEnd, EvfiltWrite, EvAdd)), GuestEpipe);
|
|
Require(close_nid_postfix(late) == 0);
|
|
Require(close_nid_postfix(pipe[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
#endif
|
|
|
|
static void SocketPair() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(static_cast<std::uintptr_t>(pair[0]), EvfiltRead, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(send_nid_postfix(pair[1], "abc", 3, 0) == 3);
|
|
Require(Wait(kq, events, 4) == 1);
|
|
Require(events[0].ident == static_cast<std::uintptr_t>(pair[0]) && events[0].filter == EvfiltRead && events[0].data == 3 && events[0].udata == &marker);
|
|
Require(Poll(kq, events, 4) == 1);
|
|
char buffer[4]{};
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltWrite, EvAdd | EvOneshot)) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].filter == EvfiltWrite);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltWrite, EvDelete)), GuestEnoent);
|
|
#ifndef __linux__
|
|
Require(Throws([kq, &pair] { Apply(kq, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltRead, EvAdd | EvClear)); }));
|
|
#endif
|
|
const int space = kqueue_nid_postfix();
|
|
Require(Apply(space, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltWrite, EvAdd)) == 0);
|
|
Require(Poll(space, events, 4) == 1 && events[0].filter == EvfiltWrite);
|
|
const std::intptr_t idle = events[0].data;
|
|
Require(idle > 0);
|
|
const std::string payload(1000, 'p');
|
|
Require(send_nid_postfix(pair[1], payload.data(), payload.size(), 0) == static_cast<std::int64_t>(payload.size()));
|
|
Require(Poll(space, events, 4) == 1 && events[0].filter == EvfiltWrite);
|
|
#if defined(__linux__)
|
|
Require(events[0].data <= idle - static_cast<std::intptr_t>(payload.size()));
|
|
#else
|
|
Require(events[0].data > 0 && events[0].data <= idle);
|
|
#endif
|
|
std::string drained(payload.size(), '\0');
|
|
Require(recv_nid_postfix(pair[0], drained.data(), drained.size(), 0) == static_cast<std::int64_t>(drained.size()));
|
|
Require(Poll(space, events, 4) == 1 && events[0].filter == EvfiltWrite && events[0].data == idle);
|
|
Require(Apply(space, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltWrite, EvDelete)) == 0);
|
|
Require(close_nid_postfix(space) == 0);
|
|
Require(close_nid_postfix(pair[0]) == 0);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(pair[0]), EvfiltRead, EvDelete)), GuestEbadf);
|
|
Require(close_nid_postfix(pair[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void WatchKqueue() {
|
|
const int inner = kqueue_nid_postfix();
|
|
const int outer = kqueue_nid_postfix();
|
|
const auto watched = static_cast<std::uintptr_t>(inner);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(inner, Change(5, EvfiltUser, EvAdd | EvClear)) == 0);
|
|
Require(Apply(inner, Change(6, EvfiltUser, EvAdd | EvClear)) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
Require(Poll(inner, events, 4) == 0);
|
|
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(outer, events, 4) == 1);
|
|
Require(events[0].ident == watched && events[0].filter == EvfiltRead && events[0].flags == 0 && events[0].fflags == 0);
|
|
Require(events[0].data == 1 && events[0].udata == &marker);
|
|
Require(Apply(inner, Change(6, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 2);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 2);
|
|
Require(Poll(inner, events, 1) == 1);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 1);
|
|
Require(Poll(inner, events, 4) == 1);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvDisable, 0, 0, &marker)) == 0);
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvEnable, 0, 0, &marker)) == 0);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 1);
|
|
Require(Poll(inner, events, 4) == 1);
|
|
|
|
const KernelTimespec limit{5, 0};
|
|
auto start = std::chrono::steady_clock::now();
|
|
std::thread trigger([inner] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
});
|
|
Require(kevent_nid_postfix(outer, nullptr, 0, events, 4, &limit) == 1);
|
|
auto waited = std::chrono::steady_clock::now() - start;
|
|
trigger.join();
|
|
Require(waited >= std::chrono::milliseconds(40) && waited < std::chrono::seconds(4));
|
|
Require(events[0].ident == watched && events[0].data == 1 && events[0].udata == &marker);
|
|
Require(Poll(inner, events, 4) == 1);
|
|
|
|
start = std::chrono::steady_clock::now();
|
|
std::thread blocked([outer] {
|
|
KernelEvent event{};
|
|
Require(kevent_nid_postfix(outer, nullptr, 0, &event, 1, nullptr) == 1 && event.data == 1);
|
|
});
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(Apply(inner, Change(6, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
blocked.join();
|
|
waited = std::chrono::steady_clock::now() - start;
|
|
Require(waited < std::chrono::seconds(4));
|
|
Require(Poll(inner, events, 4) == 1);
|
|
|
|
Require(Apply(inner, Change(7, EvfiltTimer, EvAdd, 0, 60)) == 0);
|
|
start = std::chrono::steady_clock::now();
|
|
Require(kevent_nid_postfix(outer, nullptr, 0, events, 4, &limit) == 1);
|
|
waited = std::chrono::steady_clock::now() - start;
|
|
Require(waited >= std::chrono::milliseconds(40) && waited < std::chrono::seconds(4));
|
|
Require(events[0].ident == watched && events[0].data == 1);
|
|
Require(Poll(inner, events, 4) == 1 && events[0].ident == 7);
|
|
Require(Apply(inner, Change(7, EvfiltTimer, EvDelete)) == 0);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
Require(Apply(inner, Change(static_cast<std::uintptr_t>(pair[0]), EvfiltRead, EvAdd)) == 0);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
std::thread sender([&pair] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(send_nid_postfix(pair[1], "abc", 3, 0) == 3);
|
|
});
|
|
Require(kevent_nid_postfix(outer, nullptr, 0, events, 4, &limit) == 1);
|
|
sender.join();
|
|
Require(events[0].ident == watched && events[0].data == 1);
|
|
char buffer[4]{};
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 3);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
|
|
|
|
const int top = kqueue_nid_postfix();
|
|
const auto middle = static_cast<std::uintptr_t>(outer);
|
|
Require(Apply(top, Change(middle, EvfiltRead, EvAdd)) == 0);
|
|
Require(Poll(top, events, 4) == 0);
|
|
std::thread deep([inner] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
});
|
|
Require(kevent_nid_postfix(top, nullptr, 0, events, 4, &limit) == 1);
|
|
deep.join();
|
|
Require(events[0].ident == middle && events[0].data == 1);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].ident == watched);
|
|
Require(Poll(inner, events, 4) == 1);
|
|
Require(Poll(top, events, 4) == 0);
|
|
Require(Apply(top, Change(middle, EvfiltRead, EvDelete)) == 0);
|
|
Require(close_nid_postfix(top) == 0);
|
|
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvDelete)) == 0);
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltRead, EvDelete)), GuestEnoent);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvAdd | EvDispatch, 0, 0, &marker)) == 0);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 1 && events[0].flags == EvDispatch);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvEnable, 0, 0, &marker)) == 0);
|
|
Require(Poll(outer, events, 4) == 1);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvDelete)) == 0);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvAdd | EvOneshot)) == 0);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 1 && events[0].flags == EvOneshot);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltRead, EvDelete)), GuestEnoent);
|
|
Require(Poll(inner, events, 4) == 1);
|
|
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltWrite, EvAdd)), GuestEinval);
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltVnode, EvAdd, NoteWrite)), GuestEinval);
|
|
Require(ThrownMessage([outer, watched] { Apply(outer, Change(watched, EvfiltRead, EvAdd | EvClear)); }).find("EV_CLEAR on a kqueue descriptor") != std::string::npos);
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltRead, EvDelete)), GuestEnoent);
|
|
Require(Apply(outer, Change(watched, EvfiltRead, EvAdd)) == 0);
|
|
Require(ThrownMessage([inner, outer] { Apply(inner, Change(static_cast<std::uintptr_t>(outer), EvfiltRead, EvAdd)); }).find("watches the kqueue") != std::string::npos);
|
|
Require(Apply(inner, Change(5, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(outer, events, 4) == 1 && events[0].data == 1);
|
|
|
|
Require(close_nid_postfix(inner) == 0);
|
|
Require(Poll(outer, events, 4) == 0);
|
|
RequireFailure(Apply(outer, Change(watched, EvfiltRead, EvDelete)), GuestEbadf);
|
|
Require(close_nid_postfix(outer) == 0);
|
|
}
|
|
|
|
static void LowWaterMarks() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const auto reader = static_cast<std::uintptr_t>(pair[0]);
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(reader, EvfiltRead, EvAdd, NoteLowat, 4)) == 0);
|
|
Require(send_nid_postfix(pair[1], "ab", 2, 0) == 2);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(send_nid_postfix(pair[1], "cd", 2, 0) == 2);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == reader && events[0].filter == EvfiltRead && events[0].data == 4);
|
|
Require(Apply(kq, Change(reader, EvfiltRead, EvAdd, NoteLowat, 8)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(reader, EvfiltRead, EvAdd)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 4);
|
|
Require(Apply(kq, Change(reader, EvfiltRead, EvAdd, NoteLowat, 8)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(static_cast<std::uintptr_t>(pair[1]), EvfiltWrite, EvAdd | EvOneshot, NoteLowat, 1)) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].filter == EvfiltWrite);
|
|
const auto writer = static_cast<std::uintptr_t>(pair[1]);
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, 2)) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == writer && events[0].filter == EvfiltWrite && events[0].data >= 2);
|
|
const auto freeSpace = events[0].data;
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, freeSpace)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == writer && events[0].data == freeSpace);
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, freeSpace + 1)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Throws([kq, reader] { Apply(kq, Change(reader, EvfiltRead, EvAdd, NoteLowat, 0)); }));
|
|
int datagrams[2];
|
|
Require(socketpair_nid_postfix(Unix, 2, 0, datagrams) == 0);
|
|
const auto datagram = static_cast<std::uintptr_t>(datagrams[0]);
|
|
Require(Throws([kq, datagram] { Apply(kq, Change(datagram, EvfiltRead, EvAdd, NoteLowat, 2)); }));
|
|
Require(Apply(kq, Change(datagram, EvfiltRead, EvAdd, NoteLowat, 1)) == 0);
|
|
Require(send_nid_postfix(datagrams[1], "f", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == datagram && events[0].data == 1);
|
|
Require(close_nid_postfix(datagrams[0]) == 0 && close_nid_postfix(datagrams[1]) == 0);
|
|
Require(close_nid_postfix(pair[1]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == reader && (events[0].flags & EvEof) != 0);
|
|
Require(close_nid_postfix(pair[0]) == 0);
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
Require(Apply(kq, Change(static_cast<std::uintptr_t>(pipe[0]), EvfiltRead, EvAdd, NoteLowat, 64)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(write_nid_postfix(pipe[1], "e", 1) == 1);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 1);
|
|
Require(close_nid_postfix(pipe[0]) == 0 && close_nid_postfix(pipe[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
#ifdef __linux__
|
|
static void EdgeTriggeredSocketRead() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const auto ident = static_cast<std::uintptr_t>(pair[0]);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
char buffer[8]{};
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvClear, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(send_nid_postfix(pair[1], "abc", 3, 0) == 3);
|
|
Require(Poll(kq, events, 4) == 1);
|
|
Require(events[0].ident == ident && events[0].filter == EvfiltRead && events[0].data == 3 && events[0].udata == &marker);
|
|
Require((events[0].flags & EvClear) != 0 && (events[0].flags & EvEof) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
|
|
const KernelTimespec quiet{0, 150000000};
|
|
const auto idleStart = std::chrono::steady_clock::now();
|
|
const std::clock_t idleCpu = std::clock();
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &quiet) == 0);
|
|
Require(std::clock() - idleCpu < CLOCKS_PER_SEC / 20);
|
|
Require(std::chrono::steady_clock::now() - idleStart >= std::chrono::milliseconds(140));
|
|
|
|
Require(send_nid_postfix(pair[1], "d", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 4);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvClear, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 4);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
|
|
Require(recv_nid_postfix(pair[0], buffer, 2, 0) == 2);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 2);
|
|
Require(send_nid_postfix(pair[1], "e", 1, 0) == 1);
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
|
|
const KernelTimespec limit{5, 0};
|
|
const auto start = std::chrono::steady_clock::now();
|
|
std::thread sender([&pair] {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
Require(send_nid_postfix(pair[1], "fg", 2, 0) == 2);
|
|
});
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &limit) == 1);
|
|
const auto waited = std::chrono::steady_clock::now() - start;
|
|
sender.join();
|
|
Require(waited >= std::chrono::milliseconds(40) && waited < std::chrono::seconds(4));
|
|
Require(events[0].ident == ident && events[0].data == 2);
|
|
|
|
Require(close_nid_postfix(pair[1]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 2 && (events[0].flags & EvEof) != 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(close_nid_postfix(pair[0]) == 0);
|
|
|
|
int marked[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, marked) == 0);
|
|
const auto markedReader = static_cast<std::uintptr_t>(marked[0]);
|
|
Require(Apply(kq, Change(markedReader, EvfiltRead, EvAdd | EvClear, NoteLowat, 4)) == 0);
|
|
Require(send_nid_postfix(marked[1], "ab", 2, 0) == 2);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(send_nid_postfix(marked[1], "cd", 2, 0) == 2);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == markedReader && events[0].data == 4);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(close_nid_postfix(marked[0]) == 0 && close_nid_postfix(marked[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void EdgeTriggeredSocketWrite() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const auto ident = static_cast<std::uintptr_t>(pair[1]);
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(ident, EvfiltWrite, EvAdd | EvClear)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].filter == EvfiltWrite && events[0].data > 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
const std::string payload(1000, 'w');
|
|
Require(send_nid_postfix(pair[1], payload.data(), payload.size(), 0) == static_cast<std::int64_t>(payload.size()));
|
|
Require(Poll(kq, events, 4) == 0);
|
|
std::string drained(payload.size(), '\0');
|
|
Require(recv_nid_postfix(pair[0], drained.data(), drained.size(), 0) == static_cast<std::int64_t>(drained.size()));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].filter == EvfiltWrite && events[0].data > 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltWrite, EvAdd | EvClear)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].filter == EvfiltWrite);
|
|
Require(close_nid_postfix(pair[0]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && (events[0].flags & EvEof) != 0);
|
|
Require(close_nid_postfix(pair[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void SocketWriteLowWaterWhileFilling() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const auto writer = static_cast<std::uintptr_t>(pair[1]);
|
|
KernelEvent events[4]{};
|
|
Require(fcntl_nid_postfix(pair[1], SetStatusFlags, NonBlocking) == 0);
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, 2)) == 0);
|
|
Require(Poll(kq, events, 4) == 1);
|
|
const auto empty = events[0].data;
|
|
const std::string chunk(4096, 'x');
|
|
std::int64_t queued = 0;
|
|
Require(send_nid_postfix(pair[1], chunk.data(), chunk.size(), 0) == static_cast<std::int64_t>(chunk.size()));
|
|
queued += static_cast<std::int64_t>(chunk.size());
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data > 0 && events[0].data < empty);
|
|
const auto partial = events[0].data;
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, partial)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == partial);
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, partial + 1)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, empty)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
for (;;) {
|
|
const std::int64_t written = send_nid_postfix(pair[1], chunk.data(), chunk.size(), 0);
|
|
if (written < 0) {
|
|
Require(*__error_nid_postfix() == GuestEagain);
|
|
break;
|
|
}
|
|
queued += written;
|
|
}
|
|
Require(Apply(kq, Change(writer, EvfiltWrite, EvAdd, NoteLowat, 2)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
std::string drained(chunk.size(), '\0');
|
|
while (queued > 0) {
|
|
const std::int64_t received = recv_nid_postfix(pair[0], drained.data(), drained.size(), 0);
|
|
Require(received > 0);
|
|
queued -= received;
|
|
}
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == empty);
|
|
Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void EdgeTriggeredPipeRead() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
const auto ident = static_cast<std::uintptr_t>(pipe[0]);
|
|
KernelEvent events[4]{};
|
|
char buffer[8]{};
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvClear)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(write_nid_postfix(pipe[1], "ab", 2) == 2);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].data == 2 && (events[0].flags & EvEof) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(write_nid_postfix(pipe[1], "c", 1) == 1);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(read_nid_postfix(pipe[0], buffer, sizeof(buffer)) == 3);
|
|
Require(close_nid_postfix(pipe[1]) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 0 && (events[0].flags & EvEof) != 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(close_nid_postfix(pipe[0]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void EdgeTriggeredDispatch() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const auto ident = static_cast<std::uintptr_t>(pair[0]);
|
|
KernelEvent events[4]{};
|
|
char buffer[8]{};
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvClear | EvDispatch)) == 0);
|
|
Require(send_nid_postfix(pair[1], "ab", 2, 0) == 2);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 2);
|
|
Require(send_nid_postfix(pair[1], "c", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvEnable)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 3);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvEnable)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvDisable)) == 0);
|
|
Require(send_nid_postfix(pair[1], "d", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvEnable)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 1);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvDelete)) == 0);
|
|
Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
#endif
|
|
|
|
static void DisableAndDispatch() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(pair[0]);
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvDisable)) == 0);
|
|
Require(send_nid_postfix(pair[1], "a", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvEnable)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].data == 1);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvDisable)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvDelete)) == 0);
|
|
char buffer[4]{};
|
|
Require(recv_nid_postfix(pair[0], buffer, 1, 0) == 1);
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvDispatch, 0, 0, &marker)) == 0);
|
|
Require(send_nid_postfix(pair[1], "bc", 2, 0) == 2);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].udata == &marker && events[0].data == 2);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(send_nid_postfix(pair[1], "d", 1, 0) == 1);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvEnable)) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvDelete)) == 0);
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvForceoneshot)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd | EvForceoneshot)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].data == 3);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
RequireFailure(Apply(kq, Change(ident, EvfiltRead, EvDelete)), GuestEnoent);
|
|
Require(recv_nid_postfix(pair[0], buffer, sizeof(buffer), 0) == 3);
|
|
|
|
Require(Apply(kq, Change(9, EvfiltUser, EvAdd | EvDisable)) == 0);
|
|
Require(Apply(kq, Change(9, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(9, EvfiltUser, EvAdd | EvEnable)) == 0);
|
|
Require(Poll(kq, events, 4) == 1 && events[0].filter == EvfiltUser);
|
|
Require(Apply(kq, Change(9, EvfiltUser, EvDelete)) == 0);
|
|
|
|
Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void RequireNext(int kq, std::uintptr_t ident) {
|
|
KernelEvent event{};
|
|
Require(Poll(kq, &event, 1) == 1 && event.ident == ident);
|
|
}
|
|
|
|
static void ReportedKnotesMoveToTail() {
|
|
const int kq = kqueue_nid_postfix();
|
|
int first[2];
|
|
int second[2];
|
|
int pair[2];
|
|
Require(pipe_nid_postfix(first) == 0 && pipe_nid_postfix(second) == 0);
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
const std::uintptr_t level[3] = {static_cast<std::uintptr_t>(first[0]), static_cast<std::uintptr_t>(second[0]), static_cast<std::uintptr_t>(pair[0])};
|
|
for (const auto ident : level) {
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvAdd)) == 0);
|
|
}
|
|
Require(write_nid_postfix(first[1], "a", 1) == 1 && write_nid_postfix(second[1], "b", 1) == 1);
|
|
Require(send_nid_postfix(pair[1], "c", 1, 0) == 1);
|
|
for (int round = 0; round < 2; ++round) {
|
|
for (const auto ident : level) {
|
|
RequireNext(kq, ident);
|
|
}
|
|
}
|
|
for (const auto ident : level) {
|
|
Require(Apply(kq, Change(ident, EvfiltRead, EvDelete)) == 0);
|
|
}
|
|
|
|
KernelEvent event{};
|
|
for (const std::uintptr_t ident : {11, 12}) {
|
|
Require(Apply(kq, Change(ident, EvfiltUser, EvAdd | EvClear)) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
}
|
|
RequireNext(kq, 11);
|
|
Require(Apply(kq, Change(11, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
RequireNext(kq, 12);
|
|
RequireNext(kq, 11);
|
|
Require(Poll(kq, &event, 1) == 0);
|
|
|
|
for (const std::uintptr_t ident : {21, 22}) {
|
|
Require(Apply(kq, Change(ident, EvfiltUser, EvAdd | EvDispatch)) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltUser, 0, NoteTrigger)) == 0);
|
|
}
|
|
RequireNext(kq, 21);
|
|
Require(Apply(kq, Change(21, EvfiltUser, EvEnable, NoteTrigger)) == 0);
|
|
RequireNext(kq, 22);
|
|
RequireNext(kq, 21);
|
|
Require(Poll(kq, &event, 1) == 0);
|
|
|
|
for (const int descriptor : {first[0], first[1], second[0], second[1], pair[0], pair[1], kq}) {
|
|
Require(close_nid_postfix(descriptor) == 0);
|
|
}
|
|
}
|
|
|
|
static void Errors() {
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const KernelEvent changes[3] = {
|
|
Change(FirstSocket + 4000, EvfiltRead, EvAdd),
|
|
Change(8, EvfiltUser, EvDelete),
|
|
Change(9, EvfiltUser, EvAdd | EvEnable | EvDisable),
|
|
};
|
|
Require(kevent_nid_postfix(kq, changes, 3, events, 4, nullptr) == 3);
|
|
Require(events[0].flags == EvError && events[0].data == GuestEbadf && events[0].ident == FirstSocket + 4000);
|
|
Require(events[1].flags == EvError && events[1].data == GuestEnoent);
|
|
Require(events[2].flags == EvError && events[2].data == GuestEinval);
|
|
RequireFailure(kevent_nid_postfix(kq, changes, 3, events, 1, nullptr), GuestEnoent);
|
|
Require(events[0].flags == EvError && events[0].data == GuestEbadf);
|
|
RequireFailure(kevent_nid_postfix(kq, changes, 3, nullptr, 0, nullptr), GuestEbadf);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(kq), EvfiltRead, EvAdd)), GuestEinval);
|
|
RequireFailure(Apply(kq, Change(1, 1, EvAdd)), GuestEinval);
|
|
Require(Throws([kq] { Apply(kq, Change(1, EvfiltProc, EvAdd)); }));
|
|
Require(Apply(kq, Change(1, EvfiltUser, EvAdd | EvDisable)) == 0);
|
|
Require(Apply(kq, Change(1, EvfiltUser, EvDelete)) == 0);
|
|
RequireFailure(kevent_nid_postfix(kq, nullptr, 0, events, -1, nullptr), GuestEinval);
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
Require(Throws([kq, &pipe] { Apply(kq, Change(static_cast<std::uintptr_t>(pipe[1]), EvfiltRead, EvAdd)); }));
|
|
Require(Throws([kq, &pipe] { Apply(kq, Change(static_cast<std::uintptr_t>(pipe[0]), EvfiltWrite, EvAdd)); }));
|
|
#ifdef __APPLE__
|
|
Require(Throws([kq, &pipe] { Apply(kq, Change(static_cast<std::uintptr_t>(pipe[1]), EvfiltWrite, EvAdd)); }));
|
|
#endif
|
|
#ifndef __linux__
|
|
Require(Throws([kq, &pipe] { Apply(kq, Change(static_cast<std::uintptr_t>(pipe[0]), EvfiltRead, EvAdd | EvClear)); }));
|
|
#endif
|
|
Require(close_nid_postfix(pipe[0]) == 0 && close_nid_postfix(pipe[1]) == 0);
|
|
const KernelEvent skipped = Change(1, 0, EvAdd);
|
|
const KernelTimespec zero{0, 0};
|
|
Require(kevent_nid_postfix(kq, &skipped, 1, events, 4, &zero) == 0);
|
|
const KernelTimespec invalid{0, 1000000000};
|
|
RequireFailure(kevent_nid_postfix(kq, nullptr, 0, events, 4, &invalid), GuestEinval);
|
|
const KernelTimespec negative{-1, 0};
|
|
RequireFailure(kevent_nid_postfix(kq, nullptr, 0, events, 4, &negative), GuestEinval);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static std::string TemporaryName(const char* stem) {
|
|
return std::string("anyps5-kqueue-") + stem + "-" + std::to_string(std::chrono::steady_clock::now().time_since_epoch().count());
|
|
}
|
|
|
|
static void FileWatch() {
|
|
const auto path = TemporaryName("file");
|
|
const int file = open_nid_postfix(path.c_str(), ReadWrite | Create, 0644);
|
|
Require(file >= 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(file);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, 0x25, NoteDelete | NoteWrite | NoteExtend | NoteLink | NoteRename, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(write_nid_postfix(file, "abc", 3) == 3);
|
|
Require(Wait(kq, events, 4) == 1);
|
|
Require(events[0].ident == ident && events[0].filter == EvfiltVnode && events[0].flags == EvClear && events[0].udata == &marker);
|
|
Require(events[0].fflags == (NoteWrite | NoteExtend) && events[0].data == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(ftruncate_nid_postfix(file, 0) == 0);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd, NoteAttrib)) == 0);
|
|
Require(write_nid_postfix(file, "abc", 3) == 3);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(ftruncate_nid_postfix(file, 0) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].fflags == NoteAttrib && events[0].udata == nullptr);
|
|
Require(Throws([kq, ident] { Apply(kq, Change(ident, EvfiltVnode, EvAdd, NoteRevoke)); }));
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvDelete)) == 0);
|
|
RequireFailure(Apply(kq, Change(ident, EvfiltVnode, EvDelete)), GuestEnoent);
|
|
|
|
KernelEqueue eq = 0;
|
|
Require(sceKernelCreateEqueue(&eq, "file") == SceOk);
|
|
Require(sceKernelAddFileEvent(eq, file, static_cast<int>(NoteWrite), &marker) == SceOk);
|
|
Require(write_nid_postfix(file, "defg", 4) == 4);
|
|
int count = 0;
|
|
const KernelUseconds second = 1000000;
|
|
Require(sceKernelWaitEqueue(eq, events, 4, &count, &second) == SceOk && count == 1);
|
|
Require(events[0].ident == ident && events[0].filter == EvfiltVnode && events[0].fflags == (NoteWrite | NoteExtend) && events[0].udata == &marker);
|
|
Require(sceKernelDeleteFileEvent(eq, file) == SceOk);
|
|
Require(sceKernelDeleteFileEvent(eq, file) == SceEnoent);
|
|
Require(sceKernelAddFileEvent(eq, FirstSocket + 4000, static_cast<int>(NoteWrite), nullptr) == SceEbadf);
|
|
int pair[2];
|
|
Require(socketpair_nid_postfix(Unix, Stream, 0, pair) == 0);
|
|
Require(sceKernelAddFileEvent(eq, pair[0], static_cast<int>(NoteWrite), nullptr) == SceEinval);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(pair[0]), EvfiltVnode, EvAdd, NoteWrite)), GuestEinval);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(kq), EvfiltVnode, EvAdd, NoteWrite)), GuestEinval);
|
|
Require(close_nid_postfix(pair[0]) == 0 && close_nid_postfix(pair[1]) == 0);
|
|
int pipe[2];
|
|
Require(pipe_nid_postfix(pipe) == 0);
|
|
RequireFailure(Apply(kq, Change(static_cast<std::uintptr_t>(pipe[0]), EvfiltVnode, EvAdd, NoteWrite)), GuestEinval);
|
|
Require(close_nid_postfix(pipe[0]) == 0 && close_nid_postfix(pipe[1]) == 0);
|
|
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd, NoteWrite)) == 0);
|
|
Require(sceKernelAddFileEvent(eq, file, static_cast<int>(NoteWrite), nullptr) == SceOk);
|
|
Require(close_nid_postfix(file) == 0);
|
|
RequireFailure(Apply(kq, Change(ident, EvfiltVnode, EvDelete)), GuestEbadf);
|
|
Require(sceKernelDeleteFileEvent(eq, file) == SceEbadf);
|
|
Require(sceKernelDeleteEqueue(eq) == SceOk);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
Require(std::filesystem::remove(path));
|
|
}
|
|
|
|
#ifndef _WIN32
|
|
static void RenameAndDeleteWatchedFile() {
|
|
const auto path = TemporaryName("moved");
|
|
const auto renamed = path + "-renamed";
|
|
const int file = open_nid_postfix(path.c_str(), ReadWrite | Create, 0644);
|
|
Require(file >= 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(file);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd | EvClear, NoteDelete | NoteRename)) == 0);
|
|
Require(sceKernelRename(path.c_str(), renamed.c_str()) == SceOk);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == NoteRename);
|
|
Require(unlink_nid_postfix(renamed.c_str()) == 0);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == NoteDelete);
|
|
Require(close_nid_postfix(file) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
#endif
|
|
|
|
#if !defined(_WIN32) && !defined(__APPLE__)
|
|
static std::uint32_t NotesThrough(int kq, std::uintptr_t ident, std::uint32_t last) {
|
|
std::uint32_t notes = 0;
|
|
KernelEvent events[4]{};
|
|
while ((notes & last) == 0) {
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident);
|
|
notes |= events[0].fflags;
|
|
}
|
|
return notes;
|
|
}
|
|
|
|
static void MoveDirectoryEntries() {
|
|
const auto path = TemporaryName("moves");
|
|
const auto outside = path + "-outside";
|
|
Require(mkdir_nid_postfix(path.c_str(), 0755) == 0 && mkdir_nid_postfix(outside.c_str(), 0755) == 0);
|
|
const int child = open_nid_postfix((path + "/first.bin").c_str(), ReadWrite | Create, 0644);
|
|
Require(child >= 0 && close_nid_postfix(child) == 0);
|
|
const int directory = open_nid_postfix(path.c_str(), Directory, 0);
|
|
Require(directory >= 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(directory);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd | EvClear, NoteWrite | NoteExtend | NoteLink)) == 0);
|
|
for (int i = 0; i < 64; ++i) {
|
|
Require(sceKernelRename((path + "/first.bin").c_str(), (path + "/second.bin").c_str()) == SceOk);
|
|
Require(sceKernelRename((path + "/second.bin").c_str(), (path + "/first.bin").c_str()) == SceOk);
|
|
}
|
|
Require(mkdir_nid_postfix((path + "/sub").c_str(), 0755) == 0);
|
|
Require(NotesThrough(kq, ident, NoteLink) == (NoteWrite | NoteLink));
|
|
Require(sceKernelRename((path + "/first.bin").c_str(), (outside + "/first.bin").c_str()) == SceOk);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == (NoteWrite | NoteExtend));
|
|
Require(sceKernelRename((outside + "/first.bin").c_str(), (path + "/first.bin").c_str()) == SceOk);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == (NoteWrite | NoteExtend));
|
|
Require(sceKernelRename((path + "/sub").c_str(), (outside + "/sub").c_str()) == SceOk);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == (NoteWrite | NoteExtend | NoteLink));
|
|
Require(close_nid_postfix(directory) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
Require(std::filesystem::remove_all(path) == 2 && std::filesystem::remove_all(outside) == 2);
|
|
}
|
|
#endif
|
|
|
|
static void DirectoryWatch() {
|
|
const auto path = TemporaryName("directory");
|
|
Require(mkdir_nid_postfix(path.c_str(), 0755) == 0);
|
|
const int directory = open_nid_postfix(path.c_str(), Directory, 0);
|
|
Require(directory >= 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(directory);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd | EvClear, NoteWrite | NoteLink | NoteDelete | NoteRename)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
const int child = open_nid_postfix((path + "/child.bin").c_str(), ReadWrite | Create, 0644);
|
|
Require(child >= 0 && close_nid_postfix(child) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == NoteWrite);
|
|
Require(mkdir_nid_postfix((path + "/sub").c_str(), 0755) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].fflags == (NoteWrite | NoteLink));
|
|
Require(sceKernelRename((path + "/child.bin").c_str(), (path + "/renamed.bin").c_str()) == SceOk);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == NoteWrite);
|
|
Require(unlink_nid_postfix((path + "/renamed.bin").c_str()) == 0);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].fflags == NoteWrite);
|
|
Require(rmdir_nid_postfix((path + "/sub").c_str()) == 0);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 1 && events[0].fflags == (NoteWrite | NoteLink));
|
|
Require(close_nid_postfix(directory) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
Require(std::filesystem::remove_all(path) == 1);
|
|
}
|
|
|
|
static void DeleteWatchedFile() {
|
|
const auto path = TemporaryName("deleted");
|
|
const int file = open_nid_postfix(path.c_str(), ReadWrite | Create, 0644);
|
|
Require(file >= 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(file);
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd | EvClear, NoteDelete | NoteWrite | NoteRename)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(unlink_nid_postfix(path.c_str()) == 0);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident && events[0].fflags == NoteDelete);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
Require(Poll(kq, events, 4) == 0);
|
|
Require(close_nid_postfix(file) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
Require(!std::filesystem::exists(path));
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
static std::string Utf8(const std::wstring& text) {
|
|
const int size = ::WideCharToMultiByte(CP_UTF8, 0, text.c_str(), -1, nullptr, 0, nullptr, nullptr);
|
|
Require(size > 0);
|
|
std::string narrow(static_cast<std::size_t>(size), '\0');
|
|
Require(::WideCharToMultiByte(CP_UTF8, 0, text.c_str(), -1, narrow.data(), size, nullptr, nullptr) == size);
|
|
narrow.pop_back();
|
|
return narrow;
|
|
}
|
|
|
|
static PACL DenyListing(const std::wstring& directory, PSECURITY_DESCRIPTOR& owner) {
|
|
HANDLE token = nullptr;
|
|
Require(::OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token) != FALSE);
|
|
DWORD size = 0;
|
|
::GetTokenInformation(token, TokenUser, nullptr, 0, &size);
|
|
std::vector<unsigned char> user(size);
|
|
const BOOL queried = ::GetTokenInformation(token, TokenUser, user.data(), size, &size);
|
|
::CloseHandle(token);
|
|
Require(queried != FALSE);
|
|
PACL original = nullptr;
|
|
Require(::GetNamedSecurityInfoW(directory.c_str(), SE_FILE_OBJECT, DACL_SECURITY_INFORMATION,
|
|
nullptr, nullptr, &original, nullptr, &owner) == ERROR_SUCCESS);
|
|
_EXPLICIT_ACCESS_W denied{};
|
|
denied.grfAccessPermissions = FILE_LIST_DIRECTORY;
|
|
denied.grfAccessMode = DENY_ACCESS;
|
|
denied.grfInheritance = NO_INHERITANCE;
|
|
denied.Trustee.TrusteeForm = TRUSTEE_IS_SID;
|
|
denied.Trustee.TrusteeType = TRUSTEE_IS_USER;
|
|
denied.Trustee.ptstrName = static_cast<LPWSTR>(reinterpret_cast<TOKEN_USER*>(user.data())->User.Sid);
|
|
PACL blocked = nullptr;
|
|
Require(::SetEntriesInAclW(1, &denied, original, &blocked) == ERROR_SUCCESS);
|
|
const DWORD applied = ::SetNamedSecurityInfoW(const_cast<LPWSTR>(directory.c_str()), SE_FILE_OBJECT,
|
|
DACL_SECURITY_INFORMATION, nullptr, nullptr, blocked, nullptr);
|
|
::LocalFree(blocked);
|
|
Require(applied == ERROR_SUCCESS);
|
|
return original;
|
|
}
|
|
|
|
static void RequireContext(const std::string& message, const std::string& context) {
|
|
if (message.find(context) == std::string::npos) {
|
|
std::fprintf(stderr, "the vnode watch error lacks \"%s\": %s\n", context.c_str(), message.c_str());
|
|
std::abort();
|
|
}
|
|
}
|
|
|
|
static void RequireContextInsensitive(std::string message, std::string context) {
|
|
const auto lower = [](char character) { return static_cast<char>(std::tolower(static_cast<unsigned char>(character))); };
|
|
std::transform(message.begin(), message.end(), message.begin(), lower);
|
|
std::transform(context.begin(), context.end(), context.begin(), lower);
|
|
if (message.find(context) == std::string::npos) {
|
|
std::fprintf(stderr, "the vnode watch error lacks \"%s\"\n", context.c_str());
|
|
std::abort();
|
|
}
|
|
}
|
|
|
|
static void DropPrivilege(HANDLE token, const wchar_t* name) {
|
|
TOKEN_PRIVILEGES privileges{};
|
|
privileges.PrivilegeCount = 1;
|
|
privileges.Privileges[0].Attributes = SE_PRIVILEGE_REMOVED;
|
|
Require(::LookupPrivilegeValueW(nullptr, name, &privileges.Privileges[0].Luid) != FALSE);
|
|
Require(::AdjustTokenPrivileges(token, FALSE, &privileges, 0, nullptr, nullptr) != FALSE);
|
|
}
|
|
|
|
static void DropBackupPrivileges() {
|
|
HANDLE token = nullptr;
|
|
Require(::OpenProcessToken(::GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token) != FALSE);
|
|
DropPrivilege(token, L"SeBackupPrivilege");
|
|
DropPrivilege(token, L"SeRestorePrivilege");
|
|
Require(::CloseHandle(token) != FALSE);
|
|
}
|
|
|
|
static void DeniedParentWatch() {
|
|
DropBackupPrivileges();
|
|
const auto path = TemporaryName("denied");
|
|
Require(mkdir_nid_postfix(path.c_str(), 0755) == 0);
|
|
const int file = open_nid_postfix((path + "/watched.bin").c_str(), ReadWrite | Create, 0644);
|
|
Require(file >= 0);
|
|
PSECURITY_DESCRIPTOR owner = nullptr;
|
|
const auto directory = std::filesystem::absolute(path).wstring();
|
|
const PACL original = DenyListing(directory, owner);
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent events[4]{};
|
|
const auto ident = static_cast<std::uintptr_t>(file);
|
|
const std::string message = ThrownMessage([kq, ident] {
|
|
Apply(kq, Change(ident, EvfiltVnode, EvAdd, NoteWrite));
|
|
});
|
|
const DWORD restored = ::SetNamedSecurityInfoW(const_cast<LPWSTR>(directory.c_str()), SE_FILE_OBJECT,
|
|
DACL_SECURITY_INFORMATION, nullptr, nullptr, original, nullptr);
|
|
Require(restored == ERROR_SUCCESS);
|
|
::LocalFree(owner);
|
|
RequireContext(message, "parent");
|
|
RequireContext(message, "error=5");
|
|
RequireContextInsensitive(message, Utf8(directory));
|
|
Require(Apply(kq, Change(ident, EvfiltVnode, EvAdd | EvClear, NoteWrite)) == 0);
|
|
Require(write_nid_postfix(file, "abc", 3) == 3);
|
|
Require(Wait(kq, events, 4) == 1 && events[0].ident == ident);
|
|
Require((events[0].fflags & NoteWrite) != 0);
|
|
Require(close_nid_postfix(file) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
Require(std::filesystem::remove_all(path) == 2);
|
|
}
|
|
#endif
|
|
static void Timeout() {
|
|
const int kq = kqueue_nid_postfix();
|
|
KernelEvent event{};
|
|
const KernelTimespec timeout{0, 30000000};
|
|
const auto start = std::chrono::steady_clock::now();
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, &event, 1, &timeout) == 0);
|
|
Require(std::chrono::steady_clock::now() - start >= std::chrono::milliseconds(29));
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, &event, 0, nullptr) == 0);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
static void StandardInput() {
|
|
const int copy = dup_nid_postfix(0);
|
|
if (copy < 0) {
|
|
Require(*__error_nid_postfix() == GuestEbadf);
|
|
return;
|
|
}
|
|
Require(close_nid_postfix(copy) == 0);
|
|
const int kq = kqueue_nid_postfix();
|
|
int marker = 0;
|
|
KernelEvent events[4]{};
|
|
Require(Apply(kq, Change(0, EvfiltRead, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Poll(kq, events, 4) == 0);
|
|
const KernelTimespec brief{0, 20000000};
|
|
Require(kevent_nid_postfix(kq, nullptr, 0, events, 4, &brief) == 0);
|
|
Require(Apply(kq, Change(0, EvfiltRead, EvAdd, 0, 0, &marker)) == 0);
|
|
Require(Apply(kq, Change(0, EvfiltRead, EvDelete)) == 0);
|
|
RequireFailure(Apply(kq, Change(0, EvfiltRead, EvDelete)), GuestEnoent);
|
|
Require(ThrownMessage([kq] { Apply(kq, Change(0, EvfiltWrite, EvAdd)); }).find("EVFILT_WRITE on a descriptor that is neither a socket nor the write end of a pipe") != std::string::npos);
|
|
Require(close_nid_postfix(kq) == 0);
|
|
}
|
|
|
|
int main() {
|
|
StandardInput();
|
|
CreateAndClose();
|
|
Timers();
|
|
PipeRead();
|
|
#ifndef __APPLE__
|
|
PipeWrite();
|
|
#endif
|
|
UserEvents();
|
|
WatchKqueue();
|
|
SocketPair();
|
|
#ifdef __linux__
|
|
EdgeTriggeredSocketRead();
|
|
EdgeTriggeredSocketWrite();
|
|
SocketWriteLowWaterWhileFilling();
|
|
EdgeTriggeredPipeRead();
|
|
EdgeTriggeredDispatch();
|
|
#endif
|
|
DisableAndDispatch();
|
|
ReportedKnotesMoveToTail();
|
|
Errors();
|
|
LowWaterMarks();
|
|
FileWatch();
|
|
#ifndef _WIN32
|
|
RenameAndDeleteWatchedFile();
|
|
#endif
|
|
#if !defined(_WIN32) && !defined(__APPLE__)
|
|
MoveDirectoryEntries();
|
|
#endif
|
|
DirectoryWatch();
|
|
DeleteWatchedFile();
|
|
#ifdef _WIN32
|
|
DeniedParentWatch();
|
|
#endif
|
|
Timeout();
|
|
return 0;
|
|
}
|