X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Fchannels-no-partial.c;fp=test%2Fchannels-no-partial.c;h=caa308fbb69cba4a2b9c5c3a1d5c67d4a28a4e92;hb=a18da7ad8a0d9f28dabf97d15407ece748538888;hp=0000000000000000000000000000000000000000;hpb=57114d942004e8a34ff22aadc0c620a0aabbb423;p=meshlink

diff --git a/test/channels-no-partial.c b/test/channels-no-partial.c
new file mode 100644
index 00000000..caa308fb
--- /dev/null
+++ b/test/channels-no-partial.c
@@ -0,0 +1,67 @@
+#include <assert.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/time.h>
+
+#include "meshlink.h"
+#include "utils.h"
+
+static bool accept_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 true;
+}
+
+int main(int argc, char *argv[]) {
+	(void)argc;
+	(void)argv;
+
+	// Start two new meshlink instance.
+
+	meshlink_handle_t *mesh_a;
+	meshlink_handle_t *mesh_b;
+
+	open_meshlink_pair(&mesh_a, &mesh_b, "channels_no_partial");
+	meshlink_set_channel_accept_cb(mesh_b, accept_cb);
+	start_meshlink_pair(mesh_a, mesh_b);
+
+	// Open a channel
+
+	meshlink_node_t *b = meshlink_get_node(mesh_a, "b");
+	assert(b);
+
+	meshlink_channel_t *channel = meshlink_channel_open_ex(mesh_a, b, 1, NULL, NULL, 0, MESHLINK_CHANNEL_NO_PARTIAL);
+	assert(channel);
+
+	// Verify that no partial sends succeed.
+	// If rejected sends would fit an empty send buffer, 0 should be returned, otherwise -1.
+
+	char buf[256] = "";
+
+	meshlink_set_channel_sndbuf(mesh_a, channel, 256);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == -1);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 256) == 256);
+
+	meshlink_set_channel_sndbuf(mesh_a, channel, 512);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == 0);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 128) == 128);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 129) == 0);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 100) == 100);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 29) == 0);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 513) == -1);
+
+	assert_after(!meshlink_channel_get_sendq(mesh_a, channel), 30);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 512) == 512);
+
+	// Clean up.
+
+	close_meshlink_pair(mesh_a, mesh_b, "channels_no_partial");
+
+	return 0;
+}