1 files changed, 128 insertions, 0 deletions
diff --git a/lib9p/tests/test_server/fs_flush.c b/lib9p/tests/test_server/fs_flush.c
new file mode 100644
index 0000000..c8152d4
--- /dev/null
+++ b/lib9p/tests/test_server/fs_flush.c
@@ -0,0 +1,128 @@
+/* lib9p/tests/test_server/fs_flush.c - flush-* API endpoints
+ *
+ * Copyright (C) 2024-2025  Luke T. Shumaker <lukeshu@lukeshu.com>
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+#include <libmisc/alloc.h>
+
+#define IMPLEMENTATION_FOR_LIB9P_SRV_H YES /* for ctx->flush_ch */
+#include "fs_flush.h"
+
+LO_IMPLEMENTATION_C(lib9p_srv_file, struct flush_file, flush_file);
+
+struct flush_fio {
+	struct flush_file *parent;
+};
+LO_IMPLEMENTATION_STATIC(lib9p_srv_fio, struct flush_fio, flush_fio);
+
+/* srv_file *******************************************************************/
+
+void flush_file_free(struct flush_file *self) {
+	assert(self);
+}
+struct lib9p_qid flush_file_qid(struct flush_file *self) {
+	assert(self);
+	return (struct lib9p_qid){
+		.type = LIB9P_QT_FILE,
+		.vers = 1,
+		.path = self->pathnum,
+	};
+}
+
+error flush_file_stat(struct flush_file *self, struct lib9p_srv_ctx *ctx, struct lib9p_srv_stat *out) {
+	assert(self);
+	assert(ctx);
+	assert(out);
+
+	*out =  ((struct lib9p_srv_stat){
+		.qid                 = flush_file_qid(self),
+		.mode                = 0444,
+		.atime_sec           = UTIL9P_ATIME,
+		.mtime_sec           = UTIL9P_MTIME,
+		.size                = 6,
+		.name                = lib9p_str(self->name),
+		.owner_uid           = { .name = lib9p_str("root"), .num = 0 },
+		.owner_gid           = { .name = lib9p_str("root"), .num = 0 },
+		.last_modifier_uid   = { .name = lib9p_str("root"), .num = 0 },
+		.extension           = lib9p_str(NULL),
+	});
+	return ERROR_NULL;
+}
+error flush_file_wstat(struct flush_file *self, struct lib9p_srv_ctx *ctx, struct lib9p_srv_stat) {
+	assert(self);
+	assert(ctx);
+	return error_new(E_POSIX_EROFS, "cannot wstat API file");
+}
+error flush_file_remove(struct flush_file *self, struct lib9p_srv_ctx *ctx) {
+	assert(self);
+	assert(ctx);
+	return error_new(E_POSIX_EROFS, "cannot remove API file");
+}
+
+LIB9P_SRV_NOTDIR(, struct flush_file, flush_file);
+
+lib9p_srv_fio_or_error flush_file_fopen(struct flush_file *self, struct lib9p_srv_ctx *ctx, bool, bool, bool) {
+	assert(self);
+	assert(ctx);
+
+	struct flush_fio *ret = heap_alloc(1, struct flush_fio);
+	ret->parent = self;
+
+	return ERROR_NEW_VAL(lib9p_srv_fio, LO_BOX(lib9p_srv_fio, ret));
+}
+
+/* srv_fio ********************************************************************/
+
+static void flush_fio_iofree(struct flush_fio *self) {
+	assert(self);
+	free(self);
+}
+
+static struct lib9p_qid flush_fio_ioqid(struct flush_fio *self) {
+	assert(self);
+	return flush_file_qid(self->parent);
+}
+
+static uint32_t flush_fio_iounit(struct flush_fio *self) {
+	assert(self);
+	return 0;
+}
+
+static uint32_t_or_error flush_fio_pwrite(struct flush_fio *LM_UNUSED(self), struct lib9p_srv_ctx *LM_UNUSED(ctx),
+                                          const void *LM_UNUSED(buf),
+                                          uint32_t LM_UNUSED(byte_count),
+                                          uint64_t LM_UNUSED(offset)) {
+	assert_notreached("not writable");
+}
+
+static error flush_fio_pread(struct flush_fio *self, struct lib9p_srv_ctx *ctx,
+                             lo_interface io_writer dst, uint64_t LM_UNUSED(src_offset), uint32_t byte_count) {
+	assert(self);
+	assert(ctx);
+
+	/* Wait for first Tflush */
+	while (!lib9p_srv_flush_requested(ctx))
+		cr_yield();
+
+	/* Wait for the specified number of Tflush (may be higher *or*
+	 * lower than 1; lower would mean that the first Tflush needs
+	 * to be flushed itself).  */
+	while (cr_chan_num_waiters(&ctx->flush_ch) != self->parent->flush_cnt)
+		cr_yield();
+
+	/* Yield one more time, just because.  */
+	cr_yield();
+
+	/* Return */
+	switch (self->parent->flush_behavior) {
+	case FLUSH_READ:
+		return io_write(dst, "Sloth\n", 6 < byte_count ? 6 : byte_count).err;
+	case FLUSH_ERROR:
+		return error_new(E_POSIX_EAGAIN, "request canceled by flush");
+	case FLUSH_SILENT:
+		return error_new(E_POSIX_ECANCELED, "request canceled by flush");
+	default:
+		assert_notreached("invalid flush_behavior");
+	}
+}