X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=selftest.c;h=8abf4cc6fdd56e3081ac0762c6bef78116ab489a;hb=526b4278d148a1d902ca4144e2f6c6f8436aa06e;hp=2c299437388e3d6cef3700860f7f3397bd6a70ea;hpb=d5888c10a181d6247e407f480524d36be6319b3e;p=utcp

diff --git a/selftest.c b/selftest.c
index 2c29943..8abf4cc 100644
--- a/selftest.c
+++ b/selftest.c
@@ -10,7 +10,7 @@ struct utcp *a;
 struct utcp *b;
 struct utcp_connection *c;
 
-int do_recv(struct utcp_connection *x, const void *data, size_t len) {
+ssize_t do_recv(struct utcp_connection *x, const void *data, size_t len) {
 	if(!len) {
 		if(errno)
 			fprintf(stderr, "%p Error: %s\n", x->utcp, strerror(errno));
@@ -24,11 +24,9 @@ int do_recv(struct utcp_connection *x, const void *data, size_t len) {
 	}
 
 	if(x == c)
-		write(0, data, len);
+		return write(0, data, len);
 	else
-		utcp_send(x, data, len);
-
-	return 0;
+		return utcp_send(x, data, len);
 }
 
 bool do_pre_accept(struct utcp *utcp, uint16_t port) {
@@ -43,7 +41,7 @@ void do_accept(struct utcp_connection *c, uint16_t port) {
 	utcp_accept(c, do_recv, NULL);
 }
 
-int do_send(struct utcp *utcp, const void *data, size_t len) {
+ssize_t do_send(struct utcp *utcp, const void *data, size_t len) {
 	static int count = 0;
 	if(++count > 1000) {
 		fprintf(stderr, "Too many packets!\n");
@@ -84,7 +82,7 @@ int main(int argc, char *argv[]) {
 	ssize_t len = utcp_send(c, "Hello world!\n", 13);
 
 	if(len != 13) {
-		if(len < 0)
+		if(len == -1)
 			fprintf(stderr, "Error: %s\n", strerror(errno));
 		else
 			fprintf(stderr, "Short write %zd!\n", len);
@@ -92,7 +90,7 @@ int main(int argc, char *argv[]) {
 	len = utcp_send(c, "This is a test.\n", 16);
 
 	if(len != 16) {
-		if(len < 0)
+		if(len == -1)
 			fprintf(stderr, "Error: %s\n", strerror(errno));
 		else
 			fprintf(stderr, "Short write %zd!\n", len);