X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Futcp-test.c;h=a7c0534f5703530b3d3204bea19012be5a96c93f;hb=f6a9eb00bab9659f28c479dc34c16dc5c28a354d;hp=d78c4b860073c7a0ef0363448a68cad3590c7721;hpb=9506ae59be6efde85bc1b8dd1fe0ebbe009fb1f1;p=meshlink

diff --git a/src/utcp-test.c b/src/utcp-test.c
index d78c4b86..a7c0534f 100644
--- a/src/utcp-test.c
+++ b/src/utcp-test.c
@@ -105,8 +105,8 @@ static void do_accept(struct utcp_connection *nc, uint16_t port) {
 	c = nc;
 
 	if(bufsize) {
-		utcp_set_sndbuf(c, bufsize);
-		utcp_set_rcvbuf(c, bufsize);
+		utcp_set_sndbuf(c, NULL, bufsize);
+		utcp_set_rcvbuf(c, NULL, bufsize);
 	}
 
 	utcp_set_accept_cb(c->utcp, NULL, NULL);
@@ -270,6 +270,11 @@ int main(int argc, char *argv[]) {
 			return 1;
 		}
 	} else {
+#ifdef SO_NOSIGPIPE
+		int nosigpipe = 1;
+		setsockopt(s, SOL_SOCKET, SO_NOSIGPIPE, &nosigpipe, sizeof(nosigpipe));
+#endif
+
 		if(connect(s, ai->ai_addr, ai->ai_addrlen)) {
 			debug("Could not connect: %s\n", strerror(errno));
 			return 1;
@@ -294,8 +299,8 @@ int main(int argc, char *argv[]) {
 		c = utcp_connect_ex(u, 1, do_recv, NULL, flags);
 
 		if(bufsize) {
-			utcp_set_sndbuf(c, bufsize);
-			utcp_set_rcvbuf(c, bufsize);
+			utcp_set_sndbuf(c, NULL, bufsize);
+			utcp_set_rcvbuf(c, NULL, bufsize);
 		}
 	}