X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Fchannels-aio.c;h=f0b3fc030e204b09cca4f9f581d8faa2b490b251;hb=HEAD;hp=a0ce57156516185fad5a1ab5a50d8c7820ee1838;hpb=cb2c4617316df5b467c6a76bf118ee65f6e1c3a2;p=meshlink

diff --git a/test/channels-aio.c b/test/channels-aio.c
index a0ce5715..19472bad 100644
--- a/test/channels-aio.c
+++ b/test/channels-aio.c
@@ -2,6 +2,8 @@
 #undef NDEBUG
 #endif
 
+#define _POSIX_C_SOURCE 200809L
+
 #include <assert.h>
 #include <stdio.h>
 #include <unistd.h>
@@ -45,16 +47,6 @@ static void aio_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const v
 	set_sync_flag(&info->flag, true);
 }
 
-static bool reject_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) {
-	(void)mesh;
-	(void)channel;
-	(void)port;
-	(void)data;
-	(void)len;
-
-	return false;
-}
-
 static void receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
 	(void)mesh;
 	(void)channel;
@@ -86,9 +78,8 @@ static bool accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint
 	return true;
 }
 
-int main(int argc, char *argv[]) {
-	(void)argc;
-	(void)argv;
+int main(void) {
+	init_sync_flag(&b_received_flag);
 
 	meshlink_set_log_cb(NULL, MESHLINK_WARNING, log_cb);
 
@@ -110,6 +101,11 @@ int main(int argc, char *argv[]) {
 	memset(out_infos, 0, sizeof(out_infos));
 
 	for(size_t i = 0; i < nchannels; i++) {
+		init_sync_flag(&in_infos[i].aio_infos[0].flag);
+		init_sync_flag(&in_infos[i].aio_infos[1].flag);
+		init_sync_flag(&out_infos[i].aio_infos[0].flag);
+		init_sync_flag(&out_infos[i].aio_infos[1].flag);
+
 		in_infos[i].data = malloc(size);
 		assert(in_infos[i].data);
 		out_infos[i].data = outdata;
@@ -124,7 +120,6 @@ int main(int argc, char *argv[]) {
 
 	mesh_b->priv = in_infos;
 
-	meshlink_set_channel_accept_cb(mesh_a, reject_cb);
 	meshlink_set_channel_accept_cb(mesh_b, accept_cb);
 
 	// Start both instances
@@ -194,9 +189,12 @@ int main(int argc, char *argv[]) {
 		// The non-AIO transfer should have completed before everything else
 		assert(!timespec_lt(&out_infos[i].aio_infos[0].ts, &b_received_ts));
 		assert(!timespec_lt(&in_infos[i].aio_infos[0].ts, &b_received_ts));
+
+		free(in_infos[i].data);
 	}
 
 	// Clean up.
 
 	close_meshlink_pair(mesh_a, mesh_b);
+	free(outdata);
 }