2 * Copyright (c) 2014 Spectra Logic Corporation. All rights reserved.
3 * Redistribution and use in source and binary forms, with or without
4 * modification, are permitted provided that the following conditions
6 * 1. Redistributions of source code must retain the above copyright
7 * notice, this list of conditions and the following disclaimer.
8 * 2. Redistributions in binary form must reproduce the above copyright
9 * notice, this list of conditions and the following disclaimer in the
10 * documentation and/or other materials provided with the distribution.
12 * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND
13 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
14 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
15 * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
16 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
17 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
18 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
19 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
20 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
21 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 #include <sys/cdefs.h>
26 __FBSDID("$FreeBSD$");
32 #include <sys/socket.h>
43 #define MIN(x, y) ((x) < (y) ? (x) : (y))
44 #define MAX(x, y) ((x) > (y) ? (x) : (y))
47 do_socketpair(int *sv)
51 s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv);
53 ATF_REQUIRE(sv[0] >= 0);
54 ATF_REQUIRE(sv[1] >= 0);
55 ATF_REQUIRE(sv[0] != sv[1]);
59 do_socketpair_nonblocking(int *sv)
63 s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv);
65 ATF_REQUIRE(sv[0] >= 0);
66 ATF_REQUIRE(sv[1] >= 0);
67 ATF_REQUIRE(sv[0] != sv[1]);
68 ATF_REQUIRE(-1 != fcntl(sv[0], F_SETFL, O_NONBLOCK));
69 ATF_REQUIRE(-1 != fcntl(sv[1], F_SETFL, O_NONBLOCK));
73 * Returns a pair of sockets made the hard way: bind, listen, connect & accept
74 * @return const char* The path to the socket
77 mk_pair_of_sockets(int *sv)
79 struct sockaddr_un sun;
80 /* ATF's isolation mechanisms will guarantee uniqueness of this file */
81 const char *path = "sock";
84 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
87 bzero(&sun, sizeof(sun));
88 sun.sun_family = AF_LOCAL;
89 sun.sun_len = sizeof(sun);
90 strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
91 err = bind(s, (struct sockaddr *)&sun, sizeof(sun));
96 /* Create the other socket */
97 s2 = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
99 err = connect(s2, (struct sockaddr*)&sun, sizeof(sun));
102 atf_tc_fail("connect(2) failed");
106 s1 = accept(s, NULL, NULL);
109 atf_tc_fail("accept(2) failed");
117 static volatile sig_atomic_t got_sigpipe = 0;
119 shutdown_send_sigpipe_handler(int x)
125 * Parameterized test function bodies
128 test_eagain(size_t sndbufsize, size_t rcvbufsize)
132 const size_t totalsize = (sndbufsize + rcvbufsize) * 2;
133 const size_t pktsize = MIN(sndbufsize, rcvbufsize) / 4;
134 char sndbuf[pktsize];
135 char recv_buf[pktsize];
136 ssize_t ssize, rsize;
138 /* setup the socket pair */
140 /* Setup the buffers */
141 ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize,
142 sizeof(sndbufsize)));
143 ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize,
144 sizeof(rcvbufsize)));
146 bzero(sndbuf, pktsize);
147 /* Send data until we get EAGAIN */
148 for(i=0; i < totalsize / pktsize; i++) {
149 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
155 atf_tc_fail("send returned < 0 but not EAGAIN");
159 atf_tc_fail("Never got EAGAIN");
163 test_sendrecv_symmetric_buffers(size_t bufsize, int blocking) {
166 const size_t pktsize = bufsize / 2;
167 char sndbuf[pktsize];
168 char recv_buf[pktsize];
169 ssize_t ssize, rsize;
171 /* setup the socket pair */
175 do_socketpair_nonblocking(sv);
177 /* Setup the buffers */
178 s = setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof(bufsize));
179 ATF_REQUIRE_EQ(0, s);
180 s = setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize));
181 ATF_REQUIRE_EQ(0, s);
183 /* Fill the send buffer */
184 bzero(sndbuf, pktsize);
186 /* send and receive the packet */
187 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
190 atf_tc_fail("send returned < 0");
192 ATF_CHECK_EQ_MSG(pktsize, ssize, "expected %zd=send(...) but got %zd",
195 rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL);
198 atf_tc_fail("recv returned < 0");
200 ATF_CHECK_EQ_MSG(pktsize, rsize, "expected %zd=send(...) but got %zd",
205 test_pipe_simulator(size_t sndbufsize, size_t rcvbufsize)
207 int s, num_sent, num_received;
209 const size_t pktsize = MIN(sndbufsize, rcvbufsize) / 4;
211 char sndbuf[pktsize];
212 char rcvbuf[pktsize];
213 char comparebuf[pktsize];
214 ssize_t ssize, rsize;
215 bool currently_sending = true;
217 /* setup the socket pair */
218 do_socketpair_nonblocking(sv);
219 /* Setup the buffers */
220 ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize,
221 sizeof(sndbufsize)));
222 ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize,
223 sizeof(rcvbufsize)));
225 /* Send a total amount of data comfortably greater than the buffers */
226 numpkts = MAX(sndbufsize, rcvbufsize) * 8 / pktsize;
227 for (num_sent=0, num_received=0;
228 num_sent < numpkts || num_received < numpkts; ) {
229 if (currently_sending && num_sent < numpkts) {
230 /* The simulated sending process */
231 /* fill the buffer */
232 memset(sndbuf, num_sent, pktsize);
233 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
236 * XXX: This is bug-compatible with the kernel.
237 * The kernel returns EMSGSIZE when it should
240 if (errno == EAGAIN || errno == EMSGSIZE)
241 currently_sending = false;
244 atf_tc_fail("send failed");
247 ATF_CHECK_EQ_MSG(pktsize, ssize,
248 "expected %zd=send(...) but got %zd",
253 /* The simulated receiving process */
254 rsize = recv(sv[1], rcvbuf, pktsize, MSG_WAITALL);
256 if (errno == EAGAIN) {
257 currently_sending = true;
258 ATF_REQUIRE_MSG(num_sent < numpkts,
259 "Packets were lost!");
263 atf_tc_fail("recv failed");
266 ATF_CHECK_EQ_MSG(pktsize, rsize,
267 "expected %zd=recv(...) but got %zd",
269 memset(comparebuf, num_received, pktsize);
270 ATF_CHECK_EQ_MSG(0, memcmp(comparebuf, rcvbuf,
272 "Received data miscompare");
283 } test_pipe_thread_data_t;
286 test_pipe_writer(void* args)
288 test_pipe_thread_data_t* td = args;
289 char sndbuf[td->pktsize];
293 for(i=0; i < td->numpkts; i++) {
294 memset(sndbuf, i, td->pktsize);
295 ssize = send(td->so, sndbuf, td->pktsize, MSG_EOR);
298 atf_tc_fail("send returned < 0");
300 ATF_CHECK_EQ_MSG(td->pktsize, ssize,
301 "expected %zd=send(...) but got %zd",
308 test_pipe_reader(void* args)
310 test_pipe_thread_data_t* td = args;
311 char rcvbuf[td->pktsize];
312 char comparebuf[td->pktsize];
316 for(i=0; i < td->numpkts; i++) {
317 memset(comparebuf, i, td->pktsize);
318 rsize = recv(td->so, rcvbuf, td->pktsize, MSG_WAITALL);
321 atf_tc_fail("recv returned < 0");
323 ATF_CHECK_EQ_MSG(td->pktsize, rsize,
324 "expected %zd=send(...) but got %zd",
326 d = memcmp(comparebuf, rcvbuf, td->pktsize);
327 ATF_CHECK_EQ_MSG(0, d,
328 "Received data miscompare on packet %d", i);
335 test_pipe(size_t sndbufsize, size_t rcvbufsize)
337 test_pipe_thread_data_t writer_data, reader_data;
338 pthread_t writer, reader;
339 int num_sent, num_received;
341 const size_t pktsize = MIN(sndbufsize, rcvbufsize) / 4;
344 /* setup the socket pair */
346 /* Setup the buffers */
347 ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize,
348 sizeof(sndbufsize)));
349 ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize,
350 sizeof(rcvbufsize)));
352 /* Send a total amount of data comfortably greater than the buffers */
353 numpkts = MAX(sndbufsize, rcvbufsize) * 8 / pktsize;
355 /* Start the child threads */
356 writer_data.pktsize = pktsize;
357 writer_data.numpkts = numpkts;
358 writer_data.so = sv[0];
359 reader_data.pktsize = pktsize;
360 reader_data.numpkts = numpkts;
361 reader_data.so = sv[1];
362 ATF_REQUIRE_EQ(0, pthread_create(&writer, NULL, test_pipe_writer,
363 (void*)&writer_data));
364 ATF_REQUIRE_EQ(0, pthread_create(&reader, NULL, test_pipe_reader,
365 (void*)&reader_data));
367 /* Join the children */
368 ATF_REQUIRE_EQ(0, pthread_join(writer, NULL));
369 ATF_REQUIRE_EQ(0, pthread_join(reader, NULL));
377 /* Create a SEQPACKET socket */
378 ATF_TC_WITHOUT_HEAD(create_socket);
379 ATF_TC_BODY(create_socket, tc)
383 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
387 /* Create SEQPACKET sockets using socketpair(2) */
388 ATF_TC_WITHOUT_HEAD(create_socketpair);
389 ATF_TC_BODY(create_socketpair, tc)
394 s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv);
396 ATF_CHECK(sv[0] >= 0);
397 ATF_CHECK(sv[1] >= 0);
398 ATF_CHECK(sv[0] != sv[1]);
401 /* Call listen(2) without first calling bind(2). It should fail */
402 ATF_TC_WITHOUT_HEAD(listen_unbound);
403 ATF_TC_BODY(listen_unbound, tc)
407 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
410 /* expect listen to fail since we haven't called bind(2) */
414 /* Bind the socket to a file */
415 ATF_TC_WITHOUT_HEAD(bind);
416 ATF_TC_BODY(bind, tc)
418 struct sockaddr_un sun;
419 /* ATF's isolation mechanisms will guarantee uniqueness of this file */
420 const char *path = "sock";
423 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
426 bzero(&sun, sizeof(sun));
427 sun.sun_family = AF_LOCAL;
428 sun.sun_len = sizeof(sun);
429 strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
430 r = bind(s, (struct sockaddr *)&sun, sizeof(sun));
434 /* listen(2) a socket that is already bound(2) should succeed */
435 ATF_TC_WITHOUT_HEAD(listen_bound);
436 ATF_TC_BODY(listen_bound, tc)
438 struct sockaddr_un sun;
439 /* ATF's isolation mechanisms will guarantee uniqueness of this file */
440 const char *path = "sock";
443 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
446 bzero(&sun, sizeof(sun));
447 sun.sun_family = AF_LOCAL;
448 sun.sun_len = sizeof(sun);
449 strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
450 r = bind(s, (struct sockaddr *)&sun, sizeof(sun));
456 /* connect(2) can make a connection */
457 ATF_TC_WITHOUT_HEAD(connect);
458 ATF_TC_BODY(connect, tc)
460 struct sockaddr_un sun;
461 /* ATF's isolation mechanisms will guarantee uniqueness of this file */
462 const char *path = "sock";
463 int s, r, err, l, s2;
465 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
468 bzero(&sun, sizeof(sun));
469 sun.sun_family = AF_LOCAL;
470 sun.sun_len = sizeof(sun);
471 strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
472 r = bind(s, (struct sockaddr *)&sun, sizeof(sun));
477 /* Create the other socket */
478 s2 = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
479 ATF_REQUIRE(s2 >= 0);
480 err = connect(s2, (struct sockaddr*)&sun, sizeof(sun));
483 atf_tc_fail("connect(2) failed");
487 /* accept(2) can receive a connection */
488 ATF_TC_WITHOUT_HEAD(accept);
489 ATF_TC_BODY(accept, tc)
493 mk_pair_of_sockets(sv);
497 /* Set O_NONBLOCK on the socket */
498 ATF_TC_WITHOUT_HEAD(fcntl_nonblock);
499 ATF_TC_BODY(fcntl_nonblock, tc)
503 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
505 if (fcntl(s, F_SETFL, O_NONBLOCK) == -1) {
507 atf_tc_fail("fcntl failed");
511 /* Resize the send and receive buffers */
512 ATF_TC_WITHOUT_HEAD(resize_buffers);
513 ATF_TC_BODY(resize_buffers, tc)
519 socklen_t sl = sizeof(xs);
521 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
524 printf(" Socket Buffer Sizes\n");
525 printf(" | SNDBUF | RCVBUF |\n");
526 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl));
527 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl));
528 printf("Default | %7d | %7d |\n", xs, xr);
530 if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof(sndbuf)) != 0){
531 perror("setsockopt");
532 atf_tc_fail("setsockopt(SO_SNDBUF) failed");
534 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl));
535 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl));
536 printf("After changing SNDBUF | %7d | %7d |\n", xs, xr);
538 if (setsockopt(s, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf)) != 0){
539 perror("setsockopt");
540 atf_tc_fail("setsockopt(SO_RCVBUF) failed");
542 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl));
543 ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl));
544 printf("After changing RCVBUF | %7d | %7d |\n", xs, xr);
548 * Resize the send and receive buffers of a connected socketpair
549 * Print some useful debugging info too
551 ATF_TC_WITHOUT_HEAD(resize_connected_buffers);
552 ATF_TC_BODY(resize_connected_buffers, tc)
559 socklen_t sl = sizeof(ls);
561 /* setup the socket pair */
564 printf(" Socket Buffer Sizes\n");
565 printf(" | Left Socket | Right Socket |\n");
566 printf(" | SNDBUF | RCVBUF | SNDBUF | RCVBUF |\n");
567 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl));
568 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl));
569 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl));
570 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl));
571 printf("Default | %7d | %7d | %7d | %7d |\n",
574 /* Update one side's send buffer */
575 err = setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof(sndbuf));
577 perror("setsockopt");
578 atf_tc_fail("setsockopt(SO_SNDBUF) failed");
581 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl));
582 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl));
583 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl));
584 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl));
585 printf("After changing Left's SNDBUF | %7d | %7d | %7d | %7d |\n",
588 /* Update the same side's receive buffer */
589 err = setsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf));
591 perror("setsockopt");
592 atf_tc_fail("setsockopt(SO_RCVBUF) failed");
595 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl));
596 ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl));
597 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl));
598 ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl));
599 printf("After changing Left's RCVBUF | %7d | %7d | %7d | %7d |\n",
604 /* send(2) and recv(2) a single short record */
605 ATF_TC_WITHOUT_HEAD(send_recv);
606 ATF_TC_BODY(send_recv, tc)
610 const int bufsize = 64;
611 const char *data = "data";
612 char recv_buf[bufsize];
614 ssize_t ssize, rsize;
616 /* setup the socket pair */
619 /* send and receive a small packet */
620 datalen = strlen(data) + 1; /* +1 for the null */
621 ssize = send(sv[0], data, datalen, MSG_EOR);
624 atf_tc_fail("send returned < 0");
626 ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd",
629 rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL);
630 ATF_CHECK_EQ(datalen, rsize);
633 /* sendto(2) and recvfrom(2) a single short record
634 * According to The Open Group Base Specifications Issue 6 IEEE Std 1003.1, 2004
635 * Edition, sendto(2) is exactly the same as send(2) on a connection-mode socket
637 * According to the same spec, not all protocols are required to provide the
638 * source addres in recvfrom(2).
640 ATF_TC_WITHOUT_HEAD(sendto_recvfrom);
641 ATF_TC_BODY(sendto_recvfrom, tc)
644 struct sockaddr_storage from;
647 const int bufsize = 64;
648 const char *data = "data";
649 char recv_buf[bufsize];
651 ssize_t ssize, rsize;
654 /* setup the socket pair */
655 path = mk_pair_of_sockets(sv);
657 /* send and receive a small packet */
658 datalen = strlen(data) + 1; /* +1 for the null */
659 ssize = sendto(sv[0], data, datalen, MSG_EOR, NULL, 0);
662 atf_tc_fail("send returned < 0");
664 ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd",
667 fromlen = sizeof(from);
668 rsize = recvfrom(sv[1], recv_buf, bufsize, MSG_WAITALL,
669 (struct sockaddr*)&from, &fromlen);
672 atf_tc_fail("recvfrom returned < 0");
674 ATF_CHECK_EQ(datalen, rsize);
677 * FreeBSD does not currently provide the source address for SEQ_PACKET
678 * AF_UNIX sockets, and POSIX does not require it, so these two checks
679 * are disabled. If FreeBSD gains that feature in the future, then
680 * these checks may be reenabled
682 /* ATF_CHECK_EQ(PF_LOCAL, from.ss_family); */
683 /* ATF_CHECK_STREQ(path, ((struct sockaddr_un*)&from)->sun_path); */
687 * send(2) and recv(2) a single short record with sockets created the
688 * traditional way, involving bind, listen, connect, and accept
690 ATF_TC_WITHOUT_HEAD(send_recv_with_connect);
691 ATF_TC_BODY(send_recv_with_connect, tc)
695 const int bufsize = 64;
696 const char *data = "data";
697 char recv_buf[bufsize];
699 ssize_t ssize, rsize;
701 mk_pair_of_sockets(sv);
703 /* send and receive a small packet */
704 datalen = strlen(data) + 1; /* +1 for the null */
705 ssize = send(sv[0], data, datalen, MSG_EOR);
708 atf_tc_fail("send returned < 0");
710 ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd",
713 rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL);
714 ATF_CHECK_EQ(datalen, rsize);
717 /* send(2) should fail on a shutdown socket */
718 ATF_TC_WITHOUT_HEAD(shutdown_send);
719 ATF_TC_BODY(shutdown_send, tc)
722 const char *data = "data";
725 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
727 ATF_CHECK_EQ(0, shutdown(s, SHUT_RDWR));
728 /* USE MSG_NOSIGNAL so we don't get SIGPIPE */
729 ssize = send(s, data, sizeof(data), MSG_EOR | MSG_NOSIGNAL);
730 ATF_CHECK_EQ(EPIPE, errno);
731 ATF_CHECK_EQ(-1, ssize);
734 /* send(2) should cause SIGPIPE on a shutdown socket */
735 ATF_TC_WITHOUT_HEAD(shutdown_send_sigpipe);
736 ATF_TC_BODY(shutdown_send_sigpipe, tc)
739 const char *data = "data";
742 s = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
744 ATF_CHECK_EQ(0, shutdown(s, SHUT_RDWR));
745 ATF_REQUIRE(SIG_ERR != signal(SIGPIPE, shutdown_send_sigpipe_handler));
746 ssize = send(s, data, sizeof(data), MSG_EOR);
747 ATF_CHECK_EQ(1, got_sigpipe);
750 /* nonblocking send(2) and recv(2) a single short record */
751 ATF_TC_WITHOUT_HEAD(send_recv_nonblocking);
752 ATF_TC_BODY(send_recv_nonblocking, tc)
756 const int bufsize = 64;
757 const char *data = "data";
758 char recv_buf[bufsize];
760 ssize_t ssize, rsize;
762 /* setup the socket pair */
763 do_socketpair_nonblocking(sv);
765 /* Verify that there is nothing to receive */
766 rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL);
767 ATF_CHECK_EQ(EAGAIN, errno);
768 ATF_CHECK_EQ(-1, rsize);
770 /* send and receive a small packet */
771 datalen = strlen(data) + 1; /* +1 for the null */
772 ssize = send(sv[0], data, datalen, MSG_EOR);
775 atf_tc_fail("send returned < 0");
777 ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd",
780 rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL);
781 ATF_CHECK_EQ(datalen, rsize);
785 * We should get EMSGSIZE if we try to send a message larger than the socket
786 * buffer, with blocking sockets
788 ATF_TC_WITHOUT_HEAD(emsgsize);
789 ATF_TC_BODY(emsgsize, tc)
793 const size_t sndbufsize = 8192;
794 const size_t rcvbufsize = 8192;
795 const size_t pktsize = (sndbufsize + rcvbufsize) * 2;
796 char sndbuf[pktsize];
797 char recv_buf[pktsize];
798 ssize_t ssize, rsize;
800 /* setup the socket pair */
802 /* Setup the buffers */
803 ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize,
804 sizeof(sndbufsize)));
805 ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize,
806 sizeof(rcvbufsize)));
808 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
809 ATF_CHECK_EQ(EMSGSIZE, errno);
810 ATF_CHECK_EQ(-1, ssize);
814 * We should get EMSGSIZE if we try to send a message larger than the socket
815 * buffer, with nonblocking sockets
817 ATF_TC_WITHOUT_HEAD(emsgsize_nonblocking);
818 ATF_TC_BODY(emsgsize_nonblocking, tc)
822 const size_t sndbufsize = 8192;
823 const size_t rcvbufsize = 8192;
824 const size_t pktsize = (sndbufsize + rcvbufsize) * 2;
825 char sndbuf[pktsize];
826 char recv_buf[pktsize];
827 ssize_t ssize, rsize;
829 /* setup the socket pair */
830 do_socketpair_nonblocking(sv);
831 /* Setup the buffers */
832 ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize,
833 sizeof(sndbufsize)));
834 ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize,
835 sizeof(rcvbufsize)));
837 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
838 ATF_CHECK_EQ(EMSGSIZE, errno);
839 ATF_CHECK_EQ(-1, ssize);
844 * We should get EAGAIN if we try to send a message larger than the socket
845 * buffer, with nonblocking sockets. Test with several different sockbuf sizes
847 ATF_TC_WITHOUT_HEAD(eagain_8k_8k);
848 ATF_TC_BODY(eagain_8k_8k, tc)
850 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
851 test_eagain(8192, 8192);
853 ATF_TC_WITHOUT_HEAD(eagain_8k_128k);
854 ATF_TC_BODY(eagain_8k_128k, tc)
856 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
857 test_eagain(8192, 131072);
859 ATF_TC_WITHOUT_HEAD(eagain_128k_8k);
860 ATF_TC_BODY(eagain_128k_8k, tc)
862 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
863 test_eagain(131072, 8192);
865 ATF_TC_WITHOUT_HEAD(eagain_128k_128k);
866 ATF_TC_BODY(eagain_128k_128k, tc)
868 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
869 test_eagain(131072, 131072);
874 * nonblocking send(2) and recv(2) of several records, which should collectively
875 * fill up the send buffer but not the receive buffer
877 ATF_TC_WITHOUT_HEAD(rcvbuf_oversized);
878 ATF_TC_BODY(rcvbuf_oversized, tc)
882 const size_t sndbufsize = 8192;
883 const size_t rcvbufsize = 131072;
884 const size_t geom_mean_bufsize = 32768;
885 const int pktsize = 1024;
886 char sndbuf[pktsize];
887 char recv_buf[pktsize];
889 ssize_t ssize, rsize;
891 /* setup the socket pair */
892 do_socketpair_nonblocking(sv);
895 * Send and receive packets that are collectively greater than the send
896 * buffer, but less than the receive buffer
898 for (i=0; i < geom_mean_bufsize / pktsize; i++) {
899 /* Fill the buffer */
900 memset(sndbuf, i, pktsize);
902 /* send the packet */
903 ssize = send(sv[0], sndbuf, pktsize, MSG_EOR);
906 atf_tc_fail("send returned < 0");
908 ATF_CHECK_EQ_MSG(pktsize, ssize,
909 "expected %zd=send(...) but got %zd", pktsize, ssize);
913 rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL);
916 atf_tc_fail("recv returned < 0");
918 ATF_CHECK_EQ_MSG(pktsize, rsize,
919 "expected %zd=send(...) but got %zd", pktsize, rsize);
921 /* Verify the contents */
922 ATF_CHECK_EQ_MSG(0, memcmp(sndbuf, recv_buf, pktsize),
923 "Received data miscompare");
926 /* Trying to receive again should return EAGAIN */
927 rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL);
928 ATF_CHECK_EQ(EAGAIN, errno);
929 ATF_CHECK_EQ(-1, rsize);
933 * Simulate the behavior of a blocking pipe. The sender will send until his
934 * buffer fills up, then we'll simulate a scheduler switch that will allow the
935 * receiver to read until his buffer empties. Repeat the process until the
936 * transfer is complete.
937 * Repeat the test with multiple send and receive buffer sizes
939 ATF_TC_WITHOUT_HEAD(pipe_simulator_8k_8k);
940 ATF_TC_BODY(pipe_simulator_8k_8k, tc)
942 test_pipe_simulator(8192, 8192);
945 ATF_TC_WITHOUT_HEAD(pipe_simulator_8k_128k);
946 ATF_TC_BODY(pipe_simulator_8k_128k, tc)
948 test_pipe_simulator(8192, 131072);
951 ATF_TC_WITHOUT_HEAD(pipe_simulator_128k_8k);
952 ATF_TC_BODY(pipe_simulator_128k_8k, tc)
954 atf_tc_expect_fail("PR kern/185812 SOCK_SEQPACKET AF_UNIX sockets with asymmetrical buffers drop packets");
955 test_pipe_simulator(131072, 8192);
958 ATF_TC_WITHOUT_HEAD(pipe_simulator_128k_128k);
959 ATF_TC_BODY(pipe_simulator_128k_128k, tc)
961 test_pipe_simulator(131072, 131072);
965 * Test blocking I/O by passing data between two threads. The total amount of
966 * data will be >> buffer size to force blocking. Repeat the test with multiple
967 * send and receive buffer sizes
969 ATF_TC_WITHOUT_HEAD(pipe_8k_8k);
970 ATF_TC_BODY(pipe_8k_8k, tc)
972 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
973 test_pipe(8192, 8192);
976 ATF_TC_WITHOUT_HEAD(pipe_8k_128k);
977 ATF_TC_BODY(pipe_8k_128k, tc)
979 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
980 test_pipe(8192, 131072);
983 ATF_TC_WITHOUT_HEAD(pipe_128k_8k);
984 ATF_TC_BODY(pipe_128k_8k, tc)
987 * kern/185812 causes this test case to both fail and timeout. The
988 * atf-c-api(3) doesn't have a way to set such an expectation.
989 * If you use atf_tc_expect_fail, then it will timeout. If you use
990 * atf_tc_expect_timeout, then it will fail. If you use both, then it
991 * will show up as an unexpected pass, which is much worse
993 * https://code.google.com/p/kyua/issues/detail?id=76
995 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
996 test_pipe(131072, 8192);
999 ATF_TC_WITHOUT_HEAD(pipe_128k_128k);
1000 ATF_TC_BODY(pipe_128k_128k, tc)
1002 atf_tc_expect_fail("PR kern/185812 send(2) on a UNIX domain SEQPACKET socket returns EMSGSIZE instead of EAGAIN");
1003 test_pipe(131072, 131072);
1008 * Test single-packet I/O with and without blocking, with symmetric buffers of
1011 ATF_TC_WITHOUT_HEAD(sendrecv_8k);
1012 ATF_TC_BODY(sendrecv_8k, tc)
1014 test_sendrecv_symmetric_buffers(8 * 1024, true);
1016 ATF_TC_WITHOUT_HEAD(sendrecv_16k);
1017 ATF_TC_BODY(sendrecv_16k, tc)
1019 test_sendrecv_symmetric_buffers(16 * 1024, true);
1021 ATF_TC_WITHOUT_HEAD(sendrecv_32k);
1022 ATF_TC_BODY(sendrecv_32k, tc)
1024 test_sendrecv_symmetric_buffers(32 * 1024, true);
1026 ATF_TC_WITHOUT_HEAD(sendrecv_64k);
1027 ATF_TC_BODY(sendrecv_64k, tc)
1029 test_sendrecv_symmetric_buffers(64 * 1024, true);
1031 ATF_TC_WITHOUT_HEAD(sendrecv_128k);
1032 ATF_TC_BODY(sendrecv_128k, tc)
1034 test_sendrecv_symmetric_buffers(128 * 1024, true);
1036 ATF_TC_WITHOUT_HEAD(sendrecv_8k_nonblocking);
1037 ATF_TC_BODY(sendrecv_8k_nonblocking, tc)
1039 test_sendrecv_symmetric_buffers(8 * 1024, false);
1041 ATF_TC_WITHOUT_HEAD(sendrecv_16k_nonblocking);
1042 ATF_TC_BODY(sendrecv_16k_nonblocking, tc)
1044 test_sendrecv_symmetric_buffers(16 * 1024, false);
1046 ATF_TC_WITHOUT_HEAD(sendrecv_32k_nonblocking);
1047 ATF_TC_BODY(sendrecv_32k_nonblocking, tc)
1049 test_sendrecv_symmetric_buffers(32 * 1024, false);
1051 ATF_TC_WITHOUT_HEAD(sendrecv_64k_nonblocking);
1052 ATF_TC_BODY(sendrecv_64k_nonblocking, tc)
1054 test_sendrecv_symmetric_buffers(64 * 1024, false);
1056 ATF_TC_WITHOUT_HEAD(sendrecv_128k_nonblocking);
1057 ATF_TC_BODY(sendrecv_128k_nonblocking, tc)
1059 test_sendrecv_symmetric_buffers(128 * 1024, false);
1069 /* Basic creation and connection tests */
1070 ATF_TP_ADD_TC(tp, create_socket);
1071 ATF_TP_ADD_TC(tp, create_socketpair);
1072 ATF_TP_ADD_TC(tp, listen_unbound);
1073 ATF_TP_ADD_TC(tp, bind);
1074 ATF_TP_ADD_TC(tp, listen_bound);
1075 ATF_TP_ADD_TC(tp, connect);
1076 ATF_TP_ADD_TC(tp, accept);
1077 ATF_TP_ADD_TC(tp, fcntl_nonblock);
1078 ATF_TP_ADD_TC(tp, resize_buffers);
1079 ATF_TP_ADD_TC(tp, resize_connected_buffers);
1081 /* Unthreaded I/O tests */
1082 ATF_TP_ADD_TC(tp, send_recv);
1083 ATF_TP_ADD_TC(tp, send_recv_nonblocking);
1084 ATF_TP_ADD_TC(tp, send_recv_with_connect);
1085 ATF_TP_ADD_TC(tp, sendto_recvfrom);
1086 ATF_TP_ADD_TC(tp, shutdown_send);
1087 ATF_TP_ADD_TC(tp, shutdown_send_sigpipe);
1088 ATF_TP_ADD_TC(tp, emsgsize);
1089 ATF_TP_ADD_TC(tp, emsgsize_nonblocking);
1090 ATF_TP_ADD_TC(tp, eagain_8k_8k);
1091 ATF_TP_ADD_TC(tp, eagain_8k_128k);
1092 ATF_TP_ADD_TC(tp, eagain_128k_8k);
1093 ATF_TP_ADD_TC(tp, eagain_128k_128k);
1094 ATF_TP_ADD_TC(tp, sendrecv_8k);
1095 ATF_TP_ADD_TC(tp, sendrecv_16k);
1096 ATF_TP_ADD_TC(tp, sendrecv_32k);
1097 ATF_TP_ADD_TC(tp, sendrecv_64k);
1098 ATF_TP_ADD_TC(tp, sendrecv_128k);
1099 ATF_TP_ADD_TC(tp, sendrecv_8k_nonblocking);
1100 ATF_TP_ADD_TC(tp, sendrecv_16k_nonblocking);
1101 ATF_TP_ADD_TC(tp, sendrecv_32k_nonblocking);
1102 ATF_TP_ADD_TC(tp, sendrecv_64k_nonblocking);
1103 ATF_TP_ADD_TC(tp, sendrecv_128k_nonblocking);
1104 ATF_TP_ADD_TC(tp, rcvbuf_oversized);
1105 ATF_TP_ADD_TC(tp, pipe_simulator_8k_8k);
1106 ATF_TP_ADD_TC(tp, pipe_simulator_8k_128k);
1107 ATF_TP_ADD_TC(tp, pipe_simulator_128k_8k);
1108 ATF_TP_ADD_TC(tp, pipe_simulator_128k_128k);
1110 /* Threaded I/O tests with blocking sockets */
1111 ATF_TP_ADD_TC(tp, pipe_8k_8k);
1112 ATF_TP_ADD_TC(tp, pipe_8k_128k);
1113 ATF_TP_ADD_TC(tp, pipe_128k_8k);
1114 ATF_TP_ADD_TC(tp, pipe_128k_128k);
1116 return atf_no_error();