1 files changed, 19 insertions, 15 deletions

diff --git a/lib9p/tests/test_server/main.c b/lib9p/tests/test_server/main.c
index 9724e25..4caff16 100644
--- a/lib9p/tests/test_server/main.c
+++ b/lib9p/tests/test_server/main.c
@@ -128,22 +128,26 @@ static COROUTINE init_cr(void *) {
 	cr_exit();
 }
 
-static void log_fct(char character, void *_stream) {
-	FILE *stream = _stream;
-	putc(character, stream);
-	putchar(character);
+struct tstlog_stdout {};
+LO_IMPLEMENTATION_H(fmt_dest, struct tstlog_stdout, tstlog_stdout);
+LO_IMPLEMENTATION_C(fmt_dest, struct tstlog_stdout, tstlog_stdout, static);
+
+static size_t tstlog_bytes = 0;
+
+static void tstlog_stdout_putb(struct tstlog_stdout *, uint8_t b) {
+	putc(b, globals.logstream);
+	putchar(b);
+	tstlog_bytes++;
 }
 
-static void log_msg(struct lib9p_srv_ctx *ctx, enum lib9p_msg_type typ, void *hostmsg) {
-	/* It sucks that %v trips -Wformat and -Wformat-extra-args
-	 * https://gcc.gnu.org/bugzilla/show_bug.cgi?id=47781 */
-#pragma GCC diagnostic push
-#pragma GCC diagnostic ignored "-Wformat"
-#pragma GCC diagnostic ignored "-Wformat-extra-args"
-	fmt_fctprintf(log_fct, globals.logstream,
-	              "%c %v\n", typ % 2 ? '<' : '>',
-	              lo_box_lib9p_msg_as_fmt_formatter(&ctx->basectx, typ, hostmsg));
-#pragma GCC diagnostic pop
+static size_t tstlog_stdout_tell(struct tstlog_stdout *) {
+	return tstlog_bytes;
+}
+
+static lo_interface fmt_dest tstlog_dest = { .vtable = &_lo_tstlog_stdout_fmt_dest_vtable };
+
+static void tstlog_msg(struct lib9p_srv_ctx *ctx, enum lib9p_msg_type typ, void *hostmsg) {
+	fmt_print(tstlog_dest, typ % 2 ? "< " : "> ", (lib9p_msg, &ctx->basectx, typ, hostmsg), "\n");
 	fflush(globals.logstream);
 }
 
@@ -155,7 +159,7 @@ int main(int argc, char *argv[]) {
 	globals.logstream = fopen(argv[2], "w");
 	if (!globals.logstream)
 		error(2, errno, "fopen");
-	globals.srv.msglog = log_msg;
+	globals.srv.msglog = tstlog_msg;
 
 	struct hostclock clock_monotonic = {
 		.clock_id = CLOCK_MONOTONIC,