1 files changed, 147 insertions, 0 deletions
diff --git a/lib9p/tests/test_server/fs_whoami.c b/lib9p/tests/test_server/fs_whoami.c
new file mode 100644
index 0000000..a07fdba
--- /dev/null
+++ b/lib9p/tests/test_server/fs_whoami.c
@@ -0,0 +1,147 @@
+/* lib9p/tests/test_server/fs_whoami.c - /whoami API endpoint
+ *
+ * Copyright (C) 2024-2025  Luke T. Shumaker <lukeshu@lukeshu.com>
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+#include <inttypes.h> /* for PRI* */
+#include <stdio.h>    /* for snprintf() */
+#include <stdlib.h>   /* for realloc(), free() */
+
+#include <libmisc/alloc.h>
+
+#include "fs_whoami.h"
+
+LO_IMPLEMENTATION_C(lib9p_srv_file, struct whoami_file, whoami_file);
+
+struct whoami_fio {
+	struct whoami_file      *parent;
+	size_t                   buf_len;
+	char                    *buf;
+};
+LO_IMPLEMENTATION_STATIC(lib9p_srv_fio, struct whoami_fio, whoami_fio);
+
+size_t whoami_len(struct lib9p_srv_ctx *ctx) {
+	assert(ctx);
+	assert(ctx->user);
+
+	size_t len = 0;
+	uint32_t uid = ctx->user->num;
+	while (uid) {
+		len++;
+		uid /= 10;
+	}
+	if (!len)
+		len++;
+	len += 2;
+	len += ctx->user->name.len;
+	return len;
+}
+
+/* srv_file *******************************************************************/
+
+void whoami_file_free(struct whoami_file *self) {
+	assert(self);
+}
+struct lib9p_qid whoami_file_qid(struct whoami_file *self) {
+	assert(self);
+	return (struct lib9p_qid){
+		.type = LIB9P_QT_FILE,
+		.vers = 1,
+		.path = self->pathnum,
+	};
+}
+
+error whoami_file_stat(struct whoami_file *self, struct lib9p_srv_ctx *ctx, struct lib9p_srv_stat *out) {
+	assert(self);
+	assert(ctx);
+	assert(out);
+
+	*out = ((struct lib9p_srv_stat){
+		.qid                 = whoami_file_qid(self),
+		.mode                = 0444,
+		.atime_sec           = UTIL9P_ATIME,
+		.mtime_sec           = UTIL9P_MTIME,
+		.size                = whoami_len(ctx),
+		.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 whoami_file_wstat(struct whoami_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 whoami_file_remove(struct whoami_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 whoami_file, whoami_file);
+
+lib9p_srv_fio_or_error whoami_file_fopen(struct whoami_file *self, struct lib9p_srv_ctx *ctx, bool, bool, bool) {
+	assert(self);
+	assert(ctx);
+
+	struct whoami_fio *ret = heap_alloc(1, struct whoami_fio);
+	ret->parent = self;
+	ret->buf_len = 0;
+	ret->buf = NULL;
+
+	return ERROR_NEW_VAL(lib9p_srv_fio, LO_BOX(lib9p_srv_fio, ret));
+}
+
+/* srv_fio ********************************************************************/
+
+static void whoami_fio_iofree(struct whoami_fio *self) {
+	assert(self);
+	if (self->buf)
+		free(self->buf);
+	free(self);
+}
+
+static struct lib9p_qid whoami_fio_ioqid(struct whoami_fio *self) {
+	assert(self);
+	assert(self->parent);
+	return whoami_file_qid(self->parent);
+}
+
+static uint32_t whoami_fio_iounit(struct whoami_fio *self) {
+	assert(self);
+	return 0;
+}
+
+static uint32_t_or_error whoami_fio_pwrite(struct whoami_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 whoami_fio_pread(struct whoami_fio *self, struct lib9p_srv_ctx *ctx,
+                              lo_interface io_writer dst, uint64_t byte_offset, uint32_t byte_count) {
+	assert(self);
+	assert(ctx);
+
+	size_t data_size = whoami_len(ctx);
+	if (self->buf_len < data_size+1) {
+		self->buf = realloc(self->buf, data_size+1);
+		self->buf_len = data_size+1;
+	}
+	snprintf(self->buf, self->buf_len, "%"PRIu32" %.*s\n",
+	         ctx->user->num, ctx->user->name.len, ctx->user->name.utf8);
+
+	if (byte_offset > (uint64_t)data_size)
+		return error_new(E_POSIX_EINVAL, "offset is past end-of-file length");
+
+	size_t beg_off = (size_t)byte_offset;
+	size_t end_off = beg_off + (size_t)byte_count;
+	if (end_off > data_size)
+		end_off = data_size;
+
+	return io_write(dst, &self->buf[beg_off], end_off-beg_off).err;
+}