lib9p: Test+fix that io handles get freed

1 files changed, 26 insertions, 11 deletions

diff --git a/lib9p/tests/test_server/fs_shutdown.c b/lib9p/tests/test_server/fs_shutdown.c
index 3f88985..26a8a10 100644
--- a/lib9p/tests/test_server/fs_shutdown.c
+++ b/lib9p/tests/test_server/fs_shutdown.c
@@ -4,12 +4,17 @@
  * SPDX-License-Identifier: AGPL-3.0-or-later
  */
 
+#include <stdlib.h>
+
 #include "fs_shutdown.h"
 
 LO_IMPLEMENTATION_C(lib9p_srv_file, struct shutdown_file, shutdown_file, static);
 
-LO_IMPLEMENTATION_H(lib9p_srv_fio, struct shutdown_file, shutdown_file);
-LO_IMPLEMENTATION_C(lib9p_srv_fio, struct shutdown_file, shutdown_file, static);
+struct shutdown_fio {
+	struct shutdown_file *parent;
+};
+LO_IMPLEMENTATION_H(lib9p_srv_fio, struct shutdown_fio, shutdown_fio);
+LO_IMPLEMENTATION_C(lib9p_srv_fio, struct shutdown_fio, shutdown_fio, static);
 
 /* srv_file *******************************************************************/
 
@@ -62,32 +67,42 @@ LIB9P_SRV_NOTDIR(struct shutdown_file, shutdown_file)
 static lo_interface lib9p_srv_fio shutdown_file_fopen(struct shutdown_file *self, struct lib9p_srv_ctx *ctx, bool, bool, bool) {
 	assert(self);
 	assert(ctx);
-	return lo_box_shutdown_file_as_lib9p_srv_fio(self);
+
+	struct shutdown_fio *ret = malloc(sizeof(struct shutdown_fio));
+	ret->parent = self;
+
+	return lo_box_shutdown_fio_as_lib9p_srv_fio(ret);
 }
 
 /* srv_fio ********************************************************************/
 
-static void shutdown_file_iofree(struct shutdown_file *self) {
+static void shutdown_fio_iofree(struct shutdown_fio *self) {
+	assert(self);
+	free(self);
+}
+
+static struct lib9p_qid shutdown_fio_qid(struct shutdown_fio *self) {
 	assert(self);
+	return shutdown_file_qid(self->parent);
 }
 
-static uint32_t shutdown_file_iounit(struct shutdown_file *self) {
+static uint32_t shutdown_fio_iounit(struct shutdown_fio *self) {
 	assert(self);
 	return 0;
 }
 
-static uint32_t shutdown_file_pwrite(struct shutdown_file *self, struct lib9p_srv_ctx *ctx, void *buf, uint32_t byte_count, uint64_t LM_UNUSED(offset)) {
+static uint32_t shutdown_fio_pwrite(struct shutdown_fio *self, struct lib9p_srv_ctx *ctx, void *buf, uint32_t byte_count, uint64_t LM_UNUSED(offset)) {
 	assert(self);
 	assert(ctx);
 	assert(buf);
 	if (byte_count == 0)
 		return 0;
-	for (size_t i = 0; i < self->nlisteners; i++)
-		LO_CALL(lo_box_hostnet_tcplist_as_net_stream_listener(&self->listeners[i]), close);
+	for (size_t i = 0; i < self->parent->nlisteners; i++)
+		LO_CALL(lo_box_hostnet_tcplist_as_net_stream_listener(&self->parent->listeners[i]), close);
 	return byte_count;
 }
-static void shutdown_file_pread(struct shutdown_file *LM_UNUSED(self), struct lib9p_srv_ctx *LM_UNUSED(ctx),
-                                uint32_t LM_UNUSED(byte_count), uint64_t LM_UNUSED(byte_offset),
-                                struct iovec *LM_UNUSED(ret)) {
+static void shutdown_fio_pread(struct shutdown_fio *LM_UNUSED(self), struct lib9p_srv_ctx *LM_UNUSED(ctx),
+                               uint32_t LM_UNUSED(byte_count), uint64_t LM_UNUSED(byte_offset),
+                               struct iovec *LM_UNUSED(ret)) {
 	assert_notreached("not readable");
 }