1261081Sasomers/*- 2261081Sasomers * Copyright (c) 2014 Spectra Logic Corporation. All rights reserved. 3261081Sasomers * Redistribution and use in source and binary forms, with or without 4261081Sasomers * modification, are permitted provided that the following conditions 5261081Sasomers * are met: 6261081Sasomers * 1. Redistributions of source code must retain the above copyright 7261081Sasomers * notice, this list of conditions and the following disclaimer. 8261081Sasomers * 2. Redistributions in binary form must reproduce the above copyright 9261081Sasomers * notice, this list of conditions and the following disclaimer in the 10261081Sasomers * documentation and/or other materials provided with the distribution. 11261081Sasomers * 12261081Sasomers * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND 13261081Sasomers * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 14261081Sasomers * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 15261081Sasomers * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE 16261081Sasomers * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 17261081Sasomers * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 18261081Sasomers * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 19261081Sasomers * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 20261081Sasomers * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 21261081Sasomers * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 22261081Sasomers * SUCH DAMAGE. 23261081Sasomers */ 24261081Sasomers 25261081Sasomers#include <sys/cdefs.h> 26261081Sasomers__FBSDID("$FreeBSD: releng/10.2/tests/sys/kern/unix_seqpacket_test.c 264607 2014-04-17 13:11:56Z jmmv $"); 27261081Sasomers 28261081Sasomers#include <errno.h> 29261081Sasomers#include <fcntl.h> 30261081Sasomers#include <pthread.h> 31261081Sasomers#include <signal.h> 32261081Sasomers#include <sys/socket.h> 33261081Sasomers#include <sys/un.h> 34261081Sasomers 35261081Sasomers#include <stdio.h> 36261081Sasomers 37261081Sasomers#include <atf-c.h> 38261081Sasomers 39261081Sasomers/* 40261081Sasomers * Helper functions 41261081Sasomers */ 42261081Sasomers 43261081Sasomers#define MIN(x, y) ((x) < (y) ? (x) : (y)) 44261081Sasomers#define MAX(x, y) ((x) > (y) ? (x) : (y)) 45261081Sasomers 46263572Sasomersstatic void 47261081Sasomersdo_socketpair(int *sv) 48261081Sasomers{ 49261081Sasomers int s; 50261081Sasomers 51261081Sasomers s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv); 52261081Sasomers ATF_REQUIRE_EQ(0, s); 53261081Sasomers ATF_REQUIRE(sv[0] >= 0); 54261081Sasomers ATF_REQUIRE(sv[1] >= 0); 55261081Sasomers ATF_REQUIRE(sv[0] != sv[1]); 56261081Sasomers} 57261081Sasomers 58263572Sasomersstatic void 59261081Sasomersdo_socketpair_nonblocking(int *sv) 60261081Sasomers{ 61261081Sasomers int s; 62261081Sasomers 63261081Sasomers s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv); 64261081Sasomers ATF_REQUIRE_EQ(0, s); 65261081Sasomers ATF_REQUIRE(sv[0] >= 0); 66261081Sasomers ATF_REQUIRE(sv[1] >= 0); 67261081Sasomers ATF_REQUIRE(sv[0] != sv[1]); 68261081Sasomers ATF_REQUIRE(-1 != fcntl(sv[0], F_SETFL, O_NONBLOCK)); 69261081Sasomers ATF_REQUIRE(-1 != fcntl(sv[1], F_SETFL, O_NONBLOCK)); 70261081Sasomers} 71261081Sasomers 72261081Sasomers/* 73261081Sasomers * Returns a pair of sockets made the hard way: bind, listen, connect & accept 74261081Sasomers * @return const char* The path to the socket 75261081Sasomers */ 76263572Sasomersstatic const char* 77261081Sasomersmk_pair_of_sockets(int *sv) 78261081Sasomers{ 79261081Sasomers struct sockaddr_un sun; 80261081Sasomers /* ATF's isolation mechanisms will guarantee uniqueness of this file */ 81261081Sasomers const char *path = "sock"; 82261081Sasomers int s, err, s2, s1; 83261081Sasomers 84261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 85261081Sasomers ATF_REQUIRE(s >= 0); 86261081Sasomers 87261081Sasomers bzero(&sun, sizeof(sun)); 88261081Sasomers sun.sun_family = AF_LOCAL; 89261081Sasomers sun.sun_len = sizeof(sun); 90261081Sasomers strlcpy(sun.sun_path, path, sizeof(sun.sun_path)); 91261081Sasomers err = bind(s, (struct sockaddr *)&sun, sizeof(sun)); 92261081Sasomers err = listen(s, -1); 93261081Sasomers ATF_CHECK_EQ(0, err); 94261081Sasomers ATF_CHECK_EQ(0, err); 95261081Sasomers 96261081Sasomers /* Create the other socket */ 97261081Sasomers s2 = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 98261081Sasomers ATF_REQUIRE(s2 >= 0); 99261081Sasomers err = connect(s2, (struct sockaddr*)&sun, sizeof(sun)); 100261081Sasomers if (err != 0) { 101261081Sasomers perror("connect"); 102261081Sasomers atf_tc_fail("connect(2) failed"); 103261081Sasomers } 104261081Sasomers 105261081Sasomers /* Accept it */ 106261081Sasomers s1 = accept(s, NULL, NULL); 107261081Sasomers if (s1 == -1) { 108261081Sasomers perror("accept"); 109261081Sasomers atf_tc_fail("accept(2) failed"); 110261081Sasomers } 111261081Sasomers 112261081Sasomers sv[0] = s1; 113261081Sasomers sv[1] = s2; 114261081Sasomers return (path); 115261081Sasomers} 116261081Sasomers 117261081Sasomersstatic volatile sig_atomic_t got_sigpipe = 0; 118261081Sasomersstatic void 119263572Sasomersshutdown_send_sigpipe_handler(int __unused x) 120261081Sasomers{ 121261081Sasomers got_sigpipe = 1; 122261081Sasomers} 123261081Sasomers 124261081Sasomers/* 125261081Sasomers * Parameterized test function bodies 126261081Sasomers */ 127263572Sasomersstatic void 128261081Sasomerstest_eagain(size_t sndbufsize, size_t rcvbufsize) 129261081Sasomers{ 130261081Sasomers int i; 131261081Sasomers int sv[2]; 132261081Sasomers const size_t totalsize = (sndbufsize + rcvbufsize) * 2; 133261081Sasomers const size_t pktsize = MIN(sndbufsize, rcvbufsize) / 4; 134263572Sasomers const int numpkts = totalsize / pktsize; 135261081Sasomers char sndbuf[pktsize]; 136263572Sasomers ssize_t ssize; 137261081Sasomers 138261081Sasomers /* setup the socket pair */ 139262871Sasomers do_socketpair_nonblocking(sv); 140261081Sasomers /* Setup the buffers */ 141261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 142261081Sasomers sizeof(sndbufsize))); 143261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 144261081Sasomers sizeof(rcvbufsize))); 145261081Sasomers 146261081Sasomers bzero(sndbuf, pktsize); 147261081Sasomers /* Send data until we get EAGAIN */ 148263572Sasomers for(i=0; i < numpkts; i++) { 149261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 150261081Sasomers if (ssize == -1) { 151261081Sasomers if (errno == EAGAIN) 152261081Sasomers atf_tc_pass(); 153261081Sasomers else { 154261081Sasomers perror("send"); 155261081Sasomers atf_tc_fail("send returned < 0 but not EAGAIN"); 156261081Sasomers } 157261081Sasomers } 158261081Sasomers } 159261081Sasomers atf_tc_fail("Never got EAGAIN"); 160261081Sasomers} 161261081Sasomers 162263572Sasomersstatic void 163261081Sasomerstest_sendrecv_symmetric_buffers(size_t bufsize, int blocking) { 164261081Sasomers int s; 165261081Sasomers int sv[2]; 166263572Sasomers const ssize_t pktsize = bufsize / 2; 167261081Sasomers char sndbuf[pktsize]; 168261081Sasomers char recv_buf[pktsize]; 169261081Sasomers ssize_t ssize, rsize; 170261081Sasomers 171261081Sasomers /* setup the socket pair */ 172261081Sasomers if (blocking) 173261081Sasomers do_socketpair(sv); 174261081Sasomers else 175261081Sasomers do_socketpair_nonblocking(sv); 176261081Sasomers 177261081Sasomers /* Setup the buffers */ 178261081Sasomers s = setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof(bufsize)); 179261081Sasomers ATF_REQUIRE_EQ(0, s); 180261081Sasomers s = setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize)); 181261081Sasomers ATF_REQUIRE_EQ(0, s); 182261081Sasomers 183261081Sasomers /* Fill the send buffer */ 184261081Sasomers bzero(sndbuf, pktsize); 185261081Sasomers 186261081Sasomers /* send and receive the packet */ 187261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 188261081Sasomers if (ssize < 0) { 189261081Sasomers perror("send"); 190261081Sasomers atf_tc_fail("send returned < 0"); 191261081Sasomers } 192261081Sasomers ATF_CHECK_EQ_MSG(pktsize, ssize, "expected %zd=send(...) but got %zd", 193261081Sasomers pktsize, ssize); 194261081Sasomers 195261081Sasomers rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL); 196261081Sasomers if (rsize < 0) { 197261081Sasomers perror("recv"); 198261081Sasomers atf_tc_fail("recv returned < 0"); 199261081Sasomers } 200261081Sasomers ATF_CHECK_EQ_MSG(pktsize, rsize, "expected %zd=send(...) but got %zd", 201261081Sasomers pktsize, rsize); 202261081Sasomers} 203261081Sasomers 204263572Sasomersstatic void 205261081Sasomerstest_pipe_simulator(size_t sndbufsize, size_t rcvbufsize) 206261081Sasomers{ 207263572Sasomers int num_sent, num_received; 208261081Sasomers int sv[2]; 209263572Sasomers const ssize_t pktsize = MIN(sndbufsize, rcvbufsize) / 4; 210261081Sasomers int numpkts; 211261081Sasomers char sndbuf[pktsize]; 212261081Sasomers char rcvbuf[pktsize]; 213261081Sasomers char comparebuf[pktsize]; 214261081Sasomers ssize_t ssize, rsize; 215261081Sasomers bool currently_sending = true; 216261081Sasomers 217261081Sasomers /* setup the socket pair */ 218261081Sasomers do_socketpair_nonblocking(sv); 219261081Sasomers /* Setup the buffers */ 220261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 221261081Sasomers sizeof(sndbufsize))); 222261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 223261081Sasomers sizeof(rcvbufsize))); 224261081Sasomers 225261081Sasomers /* Send a total amount of data comfortably greater than the buffers */ 226261081Sasomers numpkts = MAX(sndbufsize, rcvbufsize) * 8 / pktsize; 227261081Sasomers for (num_sent=0, num_received=0; 228261081Sasomers num_sent < numpkts || num_received < numpkts; ) { 229261081Sasomers if (currently_sending && num_sent < numpkts) { 230261081Sasomers /* The simulated sending process */ 231261081Sasomers /* fill the buffer */ 232261081Sasomers memset(sndbuf, num_sent, pktsize); 233261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 234261081Sasomers if (ssize < 0) { 235261081Sasomers /* 236261081Sasomers * XXX: This is bug-compatible with the kernel. 237261081Sasomers * The kernel returns EMSGSIZE when it should 238261081Sasomers * return EAGAIN 239261081Sasomers */ 240261081Sasomers if (errno == EAGAIN || errno == EMSGSIZE) 241261081Sasomers currently_sending = false; 242261081Sasomers else { 243261081Sasomers perror("send"); 244261081Sasomers atf_tc_fail("send failed"); 245261081Sasomers } 246261081Sasomers } else { 247261081Sasomers ATF_CHECK_EQ_MSG(pktsize, ssize, 248261081Sasomers "expected %zd=send(...) but got %zd", 249261081Sasomers pktsize, ssize); 250261081Sasomers num_sent++; 251261081Sasomers } 252261081Sasomers } else { 253261081Sasomers /* The simulated receiving process */ 254261081Sasomers rsize = recv(sv[1], rcvbuf, pktsize, MSG_WAITALL); 255261081Sasomers if (rsize < 0) { 256261081Sasomers if (errno == EAGAIN) { 257261081Sasomers currently_sending = true; 258261081Sasomers ATF_REQUIRE_MSG(num_sent < numpkts, 259261081Sasomers "Packets were lost!"); 260261081Sasomers } 261261081Sasomers else { 262261081Sasomers perror("recv"); 263261081Sasomers atf_tc_fail("recv failed"); 264261081Sasomers } 265261081Sasomers } else { 266261081Sasomers ATF_CHECK_EQ_MSG(pktsize, rsize, 267261081Sasomers "expected %zd=recv(...) but got %zd", 268261081Sasomers pktsize, rsize); 269261081Sasomers memset(comparebuf, num_received, pktsize); 270261081Sasomers ATF_CHECK_EQ_MSG(0, memcmp(comparebuf, rcvbuf, 271261081Sasomers pktsize), 272261081Sasomers "Received data miscompare"); 273261081Sasomers num_received++; 274261081Sasomers } 275261081Sasomers } 276261081Sasomers } 277261081Sasomers} 278261081Sasomers 279261081Sasomerstypedef struct { 280261081Sasomers ssize_t pktsize; 281261081Sasomers int numpkts; 282261081Sasomers int so; 283261081Sasomers} test_pipe_thread_data_t; 284261081Sasomers 285261081Sasomersstatic void* 286261081Sasomerstest_pipe_writer(void* args) 287261081Sasomers{ 288261081Sasomers test_pipe_thread_data_t* td = args; 289261081Sasomers char sndbuf[td->pktsize]; 290261081Sasomers ssize_t ssize; 291261081Sasomers int i; 292261081Sasomers 293261081Sasomers for(i=0; i < td->numpkts; i++) { 294261081Sasomers memset(sndbuf, i, td->pktsize); 295261081Sasomers ssize = send(td->so, sndbuf, td->pktsize, MSG_EOR); 296261081Sasomers if (ssize < 0) { 297261081Sasomers perror("send"); 298261081Sasomers atf_tc_fail("send returned < 0"); 299261081Sasomers } 300261081Sasomers ATF_CHECK_EQ_MSG(td->pktsize, ssize, 301261081Sasomers "expected %zd=send(...) but got %zd", 302261081Sasomers td->pktsize, ssize); 303261081Sasomers } 304261081Sasomers return (0); 305261081Sasomers} 306261081Sasomers 307261081Sasomersstatic void* 308261081Sasomerstest_pipe_reader(void* args) 309261081Sasomers{ 310261081Sasomers test_pipe_thread_data_t* td = args; 311261081Sasomers char rcvbuf[td->pktsize]; 312261081Sasomers char comparebuf[td->pktsize]; 313261081Sasomers ssize_t rsize; 314261081Sasomers int i, d; 315261081Sasomers 316261081Sasomers for(i=0; i < td->numpkts; i++) { 317261081Sasomers memset(comparebuf, i, td->pktsize); 318261081Sasomers rsize = recv(td->so, rcvbuf, td->pktsize, MSG_WAITALL); 319261081Sasomers if (rsize < 0) { 320261081Sasomers perror("recv"); 321261081Sasomers atf_tc_fail("recv returned < 0"); 322261081Sasomers } 323261081Sasomers ATF_CHECK_EQ_MSG(td->pktsize, rsize, 324261081Sasomers "expected %zd=send(...) but got %zd", 325261081Sasomers td->pktsize, rsize); 326261081Sasomers d = memcmp(comparebuf, rcvbuf, td->pktsize); 327261081Sasomers ATF_CHECK_EQ_MSG(0, d, 328261081Sasomers "Received data miscompare on packet %d", i); 329261081Sasomers } 330261081Sasomers return (0); 331261081Sasomers} 332261081Sasomers 333261081Sasomers 334263572Sasomersstatic void 335261081Sasomerstest_pipe(size_t sndbufsize, size_t rcvbufsize) 336261081Sasomers{ 337261081Sasomers test_pipe_thread_data_t writer_data, reader_data; 338261081Sasomers pthread_t writer, reader; 339261081Sasomers int sv[2]; 340261081Sasomers const size_t pktsize = MIN(sndbufsize, rcvbufsize) / 4; 341261081Sasomers int numpkts; 342261081Sasomers 343261081Sasomers /* setup the socket pair */ 344261081Sasomers do_socketpair(sv); 345261081Sasomers /* Setup the buffers */ 346261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 347261081Sasomers sizeof(sndbufsize))); 348261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 349261081Sasomers sizeof(rcvbufsize))); 350261081Sasomers 351261081Sasomers /* Send a total amount of data comfortably greater than the buffers */ 352261081Sasomers numpkts = MAX(sndbufsize, rcvbufsize) * 8 / pktsize; 353261081Sasomers 354261081Sasomers /* Start the child threads */ 355261081Sasomers writer_data.pktsize = pktsize; 356261081Sasomers writer_data.numpkts = numpkts; 357261081Sasomers writer_data.so = sv[0]; 358261081Sasomers reader_data.pktsize = pktsize; 359261081Sasomers reader_data.numpkts = numpkts; 360261081Sasomers reader_data.so = sv[1]; 361261081Sasomers ATF_REQUIRE_EQ(0, pthread_create(&writer, NULL, test_pipe_writer, 362261081Sasomers (void*)&writer_data)); 363263820Sasomers /* 364263820Sasomers * Give the writer time to start writing, and hopefully block, before 365263820Sasomers * starting the reader. This increases the likelihood of the test case 366263820Sasomers * failing due to PR kern/185812 367263820Sasomers */ 368263820Sasomers usleep(1000); 369261081Sasomers ATF_REQUIRE_EQ(0, pthread_create(&reader, NULL, test_pipe_reader, 370261081Sasomers (void*)&reader_data)); 371261081Sasomers 372261081Sasomers /* Join the children */ 373261081Sasomers ATF_REQUIRE_EQ(0, pthread_join(writer, NULL)); 374261081Sasomers ATF_REQUIRE_EQ(0, pthread_join(reader, NULL)); 375261081Sasomers} 376261081Sasomers 377261081Sasomers 378261081Sasomers/* 379261081Sasomers * Test Cases 380261081Sasomers */ 381261081Sasomers 382261081Sasomers/* Create a SEQPACKET socket */ 383261081SasomersATF_TC_WITHOUT_HEAD(create_socket); 384261081SasomersATF_TC_BODY(create_socket, tc) 385261081Sasomers{ 386261081Sasomers int s; 387261081Sasomers 388261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 389261081Sasomers ATF_CHECK(s >= 0); 390261081Sasomers} 391261081Sasomers 392261081Sasomers/* Create SEQPACKET sockets using socketpair(2) */ 393261081SasomersATF_TC_WITHOUT_HEAD(create_socketpair); 394261081SasomersATF_TC_BODY(create_socketpair, tc) 395261081Sasomers{ 396261081Sasomers int sv[2]; 397261081Sasomers int s; 398261081Sasomers 399261081Sasomers s = socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sv); 400261081Sasomers ATF_CHECK_EQ(0, s); 401261081Sasomers ATF_CHECK(sv[0] >= 0); 402261081Sasomers ATF_CHECK(sv[1] >= 0); 403261081Sasomers ATF_CHECK(sv[0] != sv[1]); 404261081Sasomers} 405261081Sasomers 406261081Sasomers/* Call listen(2) without first calling bind(2). It should fail */ 407261081SasomersATF_TC_WITHOUT_HEAD(listen_unbound); 408261081SasomersATF_TC_BODY(listen_unbound, tc) 409261081Sasomers{ 410261081Sasomers int s, r; 411261081Sasomers 412261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 413261081Sasomers ATF_REQUIRE(s > 0); 414261081Sasomers r = listen(s, -1); 415261081Sasomers /* expect listen to fail since we haven't called bind(2) */ 416261081Sasomers ATF_CHECK(r != 0); 417261081Sasomers} 418261081Sasomers 419261081Sasomers/* Bind the socket to a file */ 420261081SasomersATF_TC_WITHOUT_HEAD(bind); 421261081SasomersATF_TC_BODY(bind, tc) 422261081Sasomers{ 423261081Sasomers struct sockaddr_un sun; 424261081Sasomers /* ATF's isolation mechanisms will guarantee uniqueness of this file */ 425261081Sasomers const char *path = "sock"; 426261081Sasomers int s, r; 427261081Sasomers 428261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 429261081Sasomers ATF_REQUIRE(s >= 0); 430261081Sasomers 431261081Sasomers bzero(&sun, sizeof(sun)); 432261081Sasomers sun.sun_family = AF_LOCAL; 433261081Sasomers sun.sun_len = sizeof(sun); 434261081Sasomers strlcpy(sun.sun_path, path, sizeof(sun.sun_path)); 435261081Sasomers r = bind(s, (struct sockaddr *)&sun, sizeof(sun)); 436261081Sasomers ATF_CHECK_EQ(0, r); 437261081Sasomers} 438261081Sasomers 439261081Sasomers/* listen(2) a socket that is already bound(2) should succeed */ 440261081SasomersATF_TC_WITHOUT_HEAD(listen_bound); 441261081SasomersATF_TC_BODY(listen_bound, tc) 442261081Sasomers{ 443261081Sasomers struct sockaddr_un sun; 444261081Sasomers /* ATF's isolation mechanisms will guarantee uniqueness of this file */ 445261081Sasomers const char *path = "sock"; 446261081Sasomers int s, r, l; 447261081Sasomers 448261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 449261081Sasomers ATF_REQUIRE(s >= 0); 450261081Sasomers 451261081Sasomers bzero(&sun, sizeof(sun)); 452261081Sasomers sun.sun_family = AF_LOCAL; 453261081Sasomers sun.sun_len = sizeof(sun); 454261081Sasomers strlcpy(sun.sun_path, path, sizeof(sun.sun_path)); 455261081Sasomers r = bind(s, (struct sockaddr *)&sun, sizeof(sun)); 456261081Sasomers l = listen(s, -1); 457261081Sasomers ATF_CHECK_EQ(0, r); 458261081Sasomers ATF_CHECK_EQ(0, l); 459261081Sasomers} 460261081Sasomers 461261081Sasomers/* connect(2) can make a connection */ 462261081SasomersATF_TC_WITHOUT_HEAD(connect); 463261081SasomersATF_TC_BODY(connect, tc) 464261081Sasomers{ 465261081Sasomers struct sockaddr_un sun; 466261081Sasomers /* ATF's isolation mechanisms will guarantee uniqueness of this file */ 467261081Sasomers const char *path = "sock"; 468261081Sasomers int s, r, err, l, s2; 469261081Sasomers 470261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 471261081Sasomers ATF_REQUIRE(s >= 0); 472261081Sasomers 473261081Sasomers bzero(&sun, sizeof(sun)); 474261081Sasomers sun.sun_family = AF_LOCAL; 475261081Sasomers sun.sun_len = sizeof(sun); 476261081Sasomers strlcpy(sun.sun_path, path, sizeof(sun.sun_path)); 477261081Sasomers r = bind(s, (struct sockaddr *)&sun, sizeof(sun)); 478261081Sasomers l = listen(s, -1); 479261081Sasomers ATF_CHECK_EQ(0, r); 480261081Sasomers ATF_CHECK_EQ(0, l); 481261081Sasomers 482261081Sasomers /* Create the other socket */ 483261081Sasomers s2 = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 484261081Sasomers ATF_REQUIRE(s2 >= 0); 485261081Sasomers err = connect(s2, (struct sockaddr*)&sun, sizeof(sun)); 486261081Sasomers if (err != 0) { 487261081Sasomers perror("connect"); 488261081Sasomers atf_tc_fail("connect(2) failed"); 489261081Sasomers } 490261081Sasomers} 491261081Sasomers 492261081Sasomers/* accept(2) can receive a connection */ 493261081SasomersATF_TC_WITHOUT_HEAD(accept); 494261081SasomersATF_TC_BODY(accept, tc) 495261081Sasomers{ 496261081Sasomers int sv[2]; 497261081Sasomers 498261081Sasomers mk_pair_of_sockets(sv); 499261081Sasomers} 500261081Sasomers 501261081Sasomers 502261081Sasomers/* Set O_NONBLOCK on the socket */ 503261081SasomersATF_TC_WITHOUT_HEAD(fcntl_nonblock); 504261081SasomersATF_TC_BODY(fcntl_nonblock, tc) 505261081Sasomers{ 506261081Sasomers int s; 507261081Sasomers 508261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 509261081Sasomers ATF_REQUIRE(s >= 0); 510261081Sasomers if (fcntl(s, F_SETFL, O_NONBLOCK) == -1) { 511261081Sasomers perror("fcntl"); 512261081Sasomers atf_tc_fail("fcntl failed"); 513261081Sasomers } 514261081Sasomers} 515261081Sasomers 516261081Sasomers/* Resize the send and receive buffers */ 517261081SasomersATF_TC_WITHOUT_HEAD(resize_buffers); 518261081SasomersATF_TC_BODY(resize_buffers, tc) 519261081Sasomers{ 520261081Sasomers int s; 521261081Sasomers int sndbuf = 12345; 522261081Sasomers int rcvbuf = 23456; 523261081Sasomers int xs, xr; 524261081Sasomers socklen_t sl = sizeof(xs); 525261081Sasomers 526261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 527261081Sasomers ATF_REQUIRE(s >= 0); 528261081Sasomers 529261081Sasomers printf(" Socket Buffer Sizes\n"); 530261081Sasomers printf(" | SNDBUF | RCVBUF |\n"); 531261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl)); 532261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl)); 533261081Sasomers printf("Default | %7d | %7d |\n", xs, xr); 534261081Sasomers 535261081Sasomers if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof(sndbuf)) != 0){ 536261081Sasomers perror("setsockopt"); 537261081Sasomers atf_tc_fail("setsockopt(SO_SNDBUF) failed"); 538261081Sasomers } 539261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl)); 540261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl)); 541261081Sasomers printf("After changing SNDBUF | %7d | %7d |\n", xs, xr); 542261081Sasomers 543261081Sasomers if (setsockopt(s, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf)) != 0){ 544261081Sasomers perror("setsockopt"); 545261081Sasomers atf_tc_fail("setsockopt(SO_RCVBUF) failed"); 546261081Sasomers } 547261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_SNDBUF, &xs, &sl)); 548261081Sasomers ATF_CHECK_EQ(0, getsockopt(s, SOL_SOCKET, SO_RCVBUF, &xr, &sl)); 549261081Sasomers printf("After changing RCVBUF | %7d | %7d |\n", xs, xr); 550261081Sasomers} 551261081Sasomers 552261081Sasomers/* 553261081Sasomers * Resize the send and receive buffers of a connected socketpair 554261081Sasomers * Print some useful debugging info too 555261081Sasomers */ 556261081SasomersATF_TC_WITHOUT_HEAD(resize_connected_buffers); 557261081SasomersATF_TC_BODY(resize_connected_buffers, tc) 558261081Sasomers{ 559261081Sasomers int sv[2]; 560261081Sasomers int sndbuf = 12345; 561261081Sasomers int rcvbuf = 23456; 562261081Sasomers int err; 563261081Sasomers int ls, lr, rs, rr; 564261081Sasomers socklen_t sl = sizeof(ls); 565261081Sasomers 566261081Sasomers /* setup the socket pair */ 567261081Sasomers do_socketpair(sv); 568261081Sasomers 569261081Sasomers printf(" Socket Buffer Sizes\n"); 570261081Sasomers printf(" | Left Socket | Right Socket |\n"); 571261081Sasomers printf(" | SNDBUF | RCVBUF | SNDBUF | RCVBUF |\n"); 572261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl)); 573261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl)); 574261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl)); 575261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl)); 576261081Sasomers printf("Default | %7d | %7d | %7d | %7d |\n", 577261081Sasomers ls, lr, rs, rr); 578261081Sasomers 579261081Sasomers /* Update one side's send buffer */ 580261081Sasomers err = setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof(sndbuf)); 581261081Sasomers if (err != 0){ 582261081Sasomers perror("setsockopt"); 583261081Sasomers atf_tc_fail("setsockopt(SO_SNDBUF) failed"); 584261081Sasomers } 585261081Sasomers 586261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl)); 587261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl)); 588261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl)); 589261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl)); 590261081Sasomers printf("After changing Left's SNDBUF | %7d | %7d | %7d | %7d |\n", 591261081Sasomers ls, lr, rs, rr); 592261081Sasomers 593261081Sasomers /* Update the same side's receive buffer */ 594261081Sasomers err = setsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf)); 595261081Sasomers if (err != 0){ 596261081Sasomers perror("setsockopt"); 597261081Sasomers atf_tc_fail("setsockopt(SO_RCVBUF) failed"); 598261081Sasomers } 599261081Sasomers 600261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &ls, &sl)); 601261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[0], SOL_SOCKET, SO_RCVBUF, &lr, &sl)); 602261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &rs, &sl)); 603261081Sasomers ATF_CHECK_EQ(0, getsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rr, &sl)); 604261081Sasomers printf("After changing Left's RCVBUF | %7d | %7d | %7d | %7d |\n", 605261081Sasomers ls, lr, rs, rr); 606261081Sasomers} 607261081Sasomers 608261081Sasomers 609261081Sasomers/* send(2) and recv(2) a single short record */ 610261081SasomersATF_TC_WITHOUT_HEAD(send_recv); 611261081SasomersATF_TC_BODY(send_recv, tc) 612261081Sasomers{ 613261081Sasomers int sv[2]; 614261081Sasomers const int bufsize = 64; 615261081Sasomers const char *data = "data"; 616261081Sasomers char recv_buf[bufsize]; 617263572Sasomers ssize_t datalen; 618261081Sasomers ssize_t ssize, rsize; 619261081Sasomers 620261081Sasomers /* setup the socket pair */ 621261081Sasomers do_socketpair(sv); 622261081Sasomers 623261081Sasomers /* send and receive a small packet */ 624261081Sasomers datalen = strlen(data) + 1; /* +1 for the null */ 625261081Sasomers ssize = send(sv[0], data, datalen, MSG_EOR); 626261081Sasomers if (ssize < 0) { 627261081Sasomers perror("send"); 628261081Sasomers atf_tc_fail("send returned < 0"); 629261081Sasomers } 630261081Sasomers ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd", 631261081Sasomers datalen, ssize); 632261081Sasomers 633261081Sasomers rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL); 634261081Sasomers ATF_CHECK_EQ(datalen, rsize); 635261081Sasomers} 636261081Sasomers 637261081Sasomers/* sendto(2) and recvfrom(2) a single short record 638261081Sasomers * According to The Open Group Base Specifications Issue 6 IEEE Std 1003.1, 2004 639261081Sasomers * Edition, sendto(2) is exactly the same as send(2) on a connection-mode socket 640261081Sasomers * 641261081Sasomers * According to the same spec, not all protocols are required to provide the 642261081Sasomers * source addres in recvfrom(2). 643261081Sasomers */ 644261081SasomersATF_TC_WITHOUT_HEAD(sendto_recvfrom); 645261081SasomersATF_TC_BODY(sendto_recvfrom, tc) 646261081Sasomers{ 647261081Sasomers const char* path; 648261081Sasomers struct sockaddr_storage from; 649261081Sasomers int sv[2]; 650261081Sasomers const int bufsize = 64; 651261081Sasomers const char *data = "data"; 652261081Sasomers char recv_buf[bufsize]; 653263572Sasomers ssize_t datalen; 654261081Sasomers ssize_t ssize, rsize; 655261081Sasomers socklen_t fromlen; 656261081Sasomers 657261081Sasomers /* setup the socket pair */ 658261081Sasomers path = mk_pair_of_sockets(sv); 659261081Sasomers 660261081Sasomers /* send and receive a small packet */ 661261081Sasomers datalen = strlen(data) + 1; /* +1 for the null */ 662261081Sasomers ssize = sendto(sv[0], data, datalen, MSG_EOR, NULL, 0); 663261081Sasomers if (ssize < 0) { 664261081Sasomers perror("send"); 665261081Sasomers atf_tc_fail("send returned < 0"); 666261081Sasomers } 667261081Sasomers ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd", 668261081Sasomers datalen, ssize); 669261081Sasomers 670261081Sasomers fromlen = sizeof(from); 671261081Sasomers rsize = recvfrom(sv[1], recv_buf, bufsize, MSG_WAITALL, 672261081Sasomers (struct sockaddr*)&from, &fromlen); 673261081Sasomers if (ssize < 0) { 674261081Sasomers perror("recvfrom"); 675261081Sasomers atf_tc_fail("recvfrom returned < 0"); 676261081Sasomers } 677261081Sasomers ATF_CHECK_EQ(datalen, rsize); 678261081Sasomers 679261081Sasomers /* 680261081Sasomers * FreeBSD does not currently provide the source address for SEQ_PACKET 681261081Sasomers * AF_UNIX sockets, and POSIX does not require it, so these two checks 682261081Sasomers * are disabled. If FreeBSD gains that feature in the future, then 683261081Sasomers * these checks may be reenabled 684261081Sasomers */ 685261081Sasomers /* ATF_CHECK_EQ(PF_LOCAL, from.ss_family); */ 686261081Sasomers /* ATF_CHECK_STREQ(path, ((struct sockaddr_un*)&from)->sun_path); */ 687261081Sasomers} 688261081Sasomers 689261081Sasomers/* 690261081Sasomers * send(2) and recv(2) a single short record with sockets created the 691261081Sasomers * traditional way, involving bind, listen, connect, and accept 692261081Sasomers */ 693261081SasomersATF_TC_WITHOUT_HEAD(send_recv_with_connect); 694261081SasomersATF_TC_BODY(send_recv_with_connect, tc) 695261081Sasomers{ 696261081Sasomers int sv[2]; 697261081Sasomers const int bufsize = 64; 698261081Sasomers const char *data = "data"; 699261081Sasomers char recv_buf[bufsize]; 700263572Sasomers ssize_t datalen; 701261081Sasomers ssize_t ssize, rsize; 702261081Sasomers 703261081Sasomers mk_pair_of_sockets(sv); 704261081Sasomers 705261081Sasomers /* send and receive a small packet */ 706261081Sasomers datalen = strlen(data) + 1; /* +1 for the null */ 707261081Sasomers ssize = send(sv[0], data, datalen, MSG_EOR); 708261081Sasomers if (ssize < 0) { 709261081Sasomers perror("send"); 710261081Sasomers atf_tc_fail("send returned < 0"); 711261081Sasomers } 712261081Sasomers ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd", 713261081Sasomers datalen, ssize); 714261081Sasomers 715261081Sasomers rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL); 716261081Sasomers ATF_CHECK_EQ(datalen, rsize); 717261081Sasomers} 718261081Sasomers 719261081Sasomers/* send(2) should fail on a shutdown socket */ 720261081SasomersATF_TC_WITHOUT_HEAD(shutdown_send); 721261081SasomersATF_TC_BODY(shutdown_send, tc) 722261081Sasomers{ 723261081Sasomers int s; 724261081Sasomers const char *data = "data"; 725261081Sasomers ssize_t ssize; 726261081Sasomers 727261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 728261081Sasomers ATF_CHECK(s >= 0); 729261081Sasomers ATF_CHECK_EQ(0, shutdown(s, SHUT_RDWR)); 730261081Sasomers /* USE MSG_NOSIGNAL so we don't get SIGPIPE */ 731261081Sasomers ssize = send(s, data, sizeof(data), MSG_EOR | MSG_NOSIGNAL); 732261081Sasomers ATF_CHECK_EQ(EPIPE, errno); 733261081Sasomers ATF_CHECK_EQ(-1, ssize); 734261081Sasomers} 735261081Sasomers 736261081Sasomers/* send(2) should cause SIGPIPE on a shutdown socket */ 737261081SasomersATF_TC_WITHOUT_HEAD(shutdown_send_sigpipe); 738261081SasomersATF_TC_BODY(shutdown_send_sigpipe, tc) 739261081Sasomers{ 740261081Sasomers int s; 741261081Sasomers const char *data = "data"; 742261081Sasomers ssize_t ssize; 743261081Sasomers 744261081Sasomers s = socket(PF_LOCAL, SOCK_SEQPACKET, 0); 745261081Sasomers ATF_CHECK(s >= 0); 746261081Sasomers ATF_CHECK_EQ(0, shutdown(s, SHUT_RDWR)); 747261081Sasomers ATF_REQUIRE(SIG_ERR != signal(SIGPIPE, shutdown_send_sigpipe_handler)); 748261081Sasomers ssize = send(s, data, sizeof(data), MSG_EOR); 749261081Sasomers ATF_CHECK_EQ(1, got_sigpipe); 750261081Sasomers} 751261081Sasomers 752261081Sasomers/* nonblocking send(2) and recv(2) a single short record */ 753261081SasomersATF_TC_WITHOUT_HEAD(send_recv_nonblocking); 754261081SasomersATF_TC_BODY(send_recv_nonblocking, tc) 755261081Sasomers{ 756261081Sasomers int sv[2]; 757261081Sasomers const int bufsize = 64; 758261081Sasomers const char *data = "data"; 759261081Sasomers char recv_buf[bufsize]; 760263572Sasomers ssize_t datalen; 761261081Sasomers ssize_t ssize, rsize; 762261081Sasomers 763261081Sasomers /* setup the socket pair */ 764261081Sasomers do_socketpair_nonblocking(sv); 765261081Sasomers 766261081Sasomers /* Verify that there is nothing to receive */ 767261081Sasomers rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL); 768261081Sasomers ATF_CHECK_EQ(EAGAIN, errno); 769261081Sasomers ATF_CHECK_EQ(-1, rsize); 770261081Sasomers 771261081Sasomers /* send and receive a small packet */ 772261081Sasomers datalen = strlen(data) + 1; /* +1 for the null */ 773261081Sasomers ssize = send(sv[0], data, datalen, MSG_EOR); 774261081Sasomers if (ssize < 0) { 775261081Sasomers perror("send"); 776261081Sasomers atf_tc_fail("send returned < 0"); 777261081Sasomers } 778261081Sasomers ATF_CHECK_EQ_MSG(datalen, ssize, "expected %zd=send(...) but got %zd", 779261081Sasomers datalen, ssize); 780261081Sasomers 781261081Sasomers rsize = recv(sv[1], recv_buf, bufsize, MSG_WAITALL); 782261081Sasomers ATF_CHECK_EQ(datalen, rsize); 783261081Sasomers} 784261081Sasomers 785261081Sasomers/* 786261081Sasomers * We should get EMSGSIZE if we try to send a message larger than the socket 787261081Sasomers * buffer, with blocking sockets 788261081Sasomers */ 789261081SasomersATF_TC_WITHOUT_HEAD(emsgsize); 790261081SasomersATF_TC_BODY(emsgsize, tc) 791261081Sasomers{ 792261081Sasomers int sv[2]; 793261081Sasomers const size_t sndbufsize = 8192; 794261081Sasomers const size_t rcvbufsize = 8192; 795261081Sasomers const size_t pktsize = (sndbufsize + rcvbufsize) * 2; 796261081Sasomers char sndbuf[pktsize]; 797263572Sasomers ssize_t ssize; 798261081Sasomers 799261081Sasomers /* setup the socket pair */ 800261081Sasomers do_socketpair(sv); 801261081Sasomers /* Setup the buffers */ 802261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 803261081Sasomers sizeof(sndbufsize))); 804261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 805261081Sasomers sizeof(rcvbufsize))); 806261081Sasomers 807261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 808261081Sasomers ATF_CHECK_EQ(EMSGSIZE, errno); 809261081Sasomers ATF_CHECK_EQ(-1, ssize); 810261081Sasomers} 811261081Sasomers 812261081Sasomers/* 813261081Sasomers * We should get EMSGSIZE if we try to send a message larger than the socket 814261081Sasomers * buffer, with nonblocking sockets 815261081Sasomers */ 816261081SasomersATF_TC_WITHOUT_HEAD(emsgsize_nonblocking); 817261081SasomersATF_TC_BODY(emsgsize_nonblocking, tc) 818261081Sasomers{ 819261081Sasomers int sv[2]; 820261081Sasomers const size_t sndbufsize = 8192; 821261081Sasomers const size_t rcvbufsize = 8192; 822261081Sasomers const size_t pktsize = (sndbufsize + rcvbufsize) * 2; 823261081Sasomers char sndbuf[pktsize]; 824263572Sasomers ssize_t ssize; 825261081Sasomers 826261081Sasomers /* setup the socket pair */ 827261081Sasomers do_socketpair_nonblocking(sv); 828261081Sasomers /* Setup the buffers */ 829261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 830261081Sasomers sizeof(sndbufsize))); 831261081Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 832261081Sasomers sizeof(rcvbufsize))); 833261081Sasomers 834261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 835261081Sasomers ATF_CHECK_EQ(EMSGSIZE, errno); 836261081Sasomers ATF_CHECK_EQ(-1, ssize); 837261081Sasomers} 838261081Sasomers 839261081Sasomers 840261081Sasomers/* 841261081Sasomers * We should get EAGAIN if we try to send a message larger than the socket 842261081Sasomers * buffer, with nonblocking sockets. Test with several different sockbuf sizes 843261081Sasomers */ 844261081SasomersATF_TC_WITHOUT_HEAD(eagain_8k_8k); 845261081SasomersATF_TC_BODY(eagain_8k_8k, tc) 846261081Sasomers{ 847261081Sasomers test_eagain(8192, 8192); 848261081Sasomers} 849261081SasomersATF_TC_WITHOUT_HEAD(eagain_8k_128k); 850261081SasomersATF_TC_BODY(eagain_8k_128k, tc) 851261081Sasomers{ 852261081Sasomers test_eagain(8192, 131072); 853261081Sasomers} 854261081SasomersATF_TC_WITHOUT_HEAD(eagain_128k_8k); 855261081SasomersATF_TC_BODY(eagain_128k_8k, tc) 856261081Sasomers{ 857261081Sasomers test_eagain(131072, 8192); 858261081Sasomers} 859261081SasomersATF_TC_WITHOUT_HEAD(eagain_128k_128k); 860261081SasomersATF_TC_BODY(eagain_128k_128k, tc) 861261081Sasomers{ 862261081Sasomers test_eagain(131072, 131072); 863261081Sasomers} 864261081Sasomers 865261081Sasomers 866261081Sasomers/* 867261081Sasomers * nonblocking send(2) and recv(2) of several records, which should collectively 868261081Sasomers * fill up the send buffer but not the receive buffer 869261081Sasomers */ 870261081SasomersATF_TC_WITHOUT_HEAD(rcvbuf_oversized); 871261081SasomersATF_TC_BODY(rcvbuf_oversized, tc) 872261081Sasomers{ 873263572Sasomers int i; 874261081Sasomers int sv[2]; 875264607Sjmmv const ssize_t pktsize = 1024; 876261081Sasomers const size_t sndbufsize = 8192; 877261081Sasomers const size_t rcvbufsize = 131072; 878263572Sasomers const size_t geometric_mean_bufsize = 32768; 879263572Sasomers const int numpkts = geometric_mean_bufsize / pktsize; 880261081Sasomers char sndbuf[pktsize]; 881261081Sasomers char recv_buf[pktsize]; 882261081Sasomers ssize_t ssize, rsize; 883261081Sasomers 884261081Sasomers /* setup the socket pair */ 885261081Sasomers do_socketpair_nonblocking(sv); 886263572Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[0], SOL_SOCKET, SO_SNDBUF, &sndbufsize, 887263572Sasomers sizeof(sndbufsize))); 888263572Sasomers ATF_REQUIRE_EQ(0, setsockopt(sv[1], SOL_SOCKET, SO_RCVBUF, &rcvbufsize, 889263572Sasomers sizeof(rcvbufsize))); 890261081Sasomers 891261081Sasomers /* 892261081Sasomers * Send and receive packets that are collectively greater than the send 893261081Sasomers * buffer, but less than the receive buffer 894261081Sasomers */ 895263572Sasomers for (i=0; i < numpkts; i++) { 896261081Sasomers /* Fill the buffer */ 897261081Sasomers memset(sndbuf, i, pktsize); 898261081Sasomers 899261081Sasomers /* send the packet */ 900261081Sasomers ssize = send(sv[0], sndbuf, pktsize, MSG_EOR); 901261081Sasomers if (ssize < 0) { 902261081Sasomers perror("send"); 903261081Sasomers atf_tc_fail("send returned < 0"); 904261081Sasomers } 905261081Sasomers ATF_CHECK_EQ_MSG(pktsize, ssize, 906261081Sasomers "expected %zd=send(...) but got %zd", pktsize, ssize); 907261081Sasomers 908261081Sasomers /* Receive it */ 909261081Sasomers 910261081Sasomers rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL); 911261081Sasomers if (rsize < 0) { 912261081Sasomers perror("recv"); 913261081Sasomers atf_tc_fail("recv returned < 0"); 914261081Sasomers } 915261081Sasomers ATF_CHECK_EQ_MSG(pktsize, rsize, 916261081Sasomers "expected %zd=send(...) but got %zd", pktsize, rsize); 917261081Sasomers 918261081Sasomers /* Verify the contents */ 919261081Sasomers ATF_CHECK_EQ_MSG(0, memcmp(sndbuf, recv_buf, pktsize), 920261081Sasomers "Received data miscompare"); 921261081Sasomers } 922261081Sasomers 923261081Sasomers /* Trying to receive again should return EAGAIN */ 924261081Sasomers rsize = recv(sv[1], recv_buf, pktsize, MSG_WAITALL); 925261081Sasomers ATF_CHECK_EQ(EAGAIN, errno); 926261081Sasomers ATF_CHECK_EQ(-1, rsize); 927261081Sasomers} 928261081Sasomers 929261081Sasomers/* 930261081Sasomers * Simulate the behavior of a blocking pipe. The sender will send until his 931261081Sasomers * buffer fills up, then we'll simulate a scheduler switch that will allow the 932261081Sasomers * receiver to read until his buffer empties. Repeat the process until the 933261081Sasomers * transfer is complete. 934261081Sasomers * Repeat the test with multiple send and receive buffer sizes 935261081Sasomers */ 936261081SasomersATF_TC_WITHOUT_HEAD(pipe_simulator_8k_8k); 937261081SasomersATF_TC_BODY(pipe_simulator_8k_8k, tc) 938261081Sasomers{ 939261081Sasomers test_pipe_simulator(8192, 8192); 940261081Sasomers} 941261081Sasomers 942261081SasomersATF_TC_WITHOUT_HEAD(pipe_simulator_8k_128k); 943261081SasomersATF_TC_BODY(pipe_simulator_8k_128k, tc) 944261081Sasomers{ 945261081Sasomers test_pipe_simulator(8192, 131072); 946261081Sasomers} 947261081Sasomers 948261081SasomersATF_TC_WITHOUT_HEAD(pipe_simulator_128k_8k); 949261081SasomersATF_TC_BODY(pipe_simulator_128k_8k, tc) 950261081Sasomers{ 951261081Sasomers test_pipe_simulator(131072, 8192); 952261081Sasomers} 953261081Sasomers 954261081SasomersATF_TC_WITHOUT_HEAD(pipe_simulator_128k_128k); 955261081SasomersATF_TC_BODY(pipe_simulator_128k_128k, tc) 956261081Sasomers{ 957261081Sasomers test_pipe_simulator(131072, 131072); 958261081Sasomers} 959261081Sasomers 960261081Sasomers/* 961261081Sasomers * Test blocking I/O by passing data between two threads. The total amount of 962261081Sasomers * data will be >> buffer size to force blocking. Repeat the test with multiple 963261081Sasomers * send and receive buffer sizes 964261081Sasomers */ 965261081SasomersATF_TC_WITHOUT_HEAD(pipe_8k_8k); 966261081SasomersATF_TC_BODY(pipe_8k_8k, tc) 967261081Sasomers{ 968261081Sasomers test_pipe(8192, 8192); 969261081Sasomers} 970261081Sasomers 971261081SasomersATF_TC_WITHOUT_HEAD(pipe_8k_128k); 972261081SasomersATF_TC_BODY(pipe_8k_128k, tc) 973261081Sasomers{ 974261081Sasomers test_pipe(8192, 131072); 975261081Sasomers} 976261081Sasomers 977261081SasomersATF_TC_WITHOUT_HEAD(pipe_128k_8k); 978261081SasomersATF_TC_BODY(pipe_128k_8k, tc) 979261081Sasomers{ 980261081Sasomers test_pipe(131072, 8192); 981261081Sasomers} 982261081Sasomers 983261081SasomersATF_TC_WITHOUT_HEAD(pipe_128k_128k); 984261081SasomersATF_TC_BODY(pipe_128k_128k, tc) 985261081Sasomers{ 986261081Sasomers test_pipe(131072, 131072); 987261081Sasomers} 988261081Sasomers 989261081Sasomers 990261081Sasomers/* 991261081Sasomers * Test single-packet I/O with and without blocking, with symmetric buffers of 992261081Sasomers * various sizes 993261081Sasomers */ 994261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_8k); 995261081SasomersATF_TC_BODY(sendrecv_8k, tc) 996261081Sasomers{ 997261081Sasomers test_sendrecv_symmetric_buffers(8 * 1024, true); 998261081Sasomers} 999261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_16k); 1000261081SasomersATF_TC_BODY(sendrecv_16k, tc) 1001261081Sasomers{ 1002261081Sasomers test_sendrecv_symmetric_buffers(16 * 1024, true); 1003261081Sasomers} 1004261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_32k); 1005261081SasomersATF_TC_BODY(sendrecv_32k, tc) 1006261081Sasomers{ 1007261081Sasomers test_sendrecv_symmetric_buffers(32 * 1024, true); 1008261081Sasomers} 1009261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_64k); 1010261081SasomersATF_TC_BODY(sendrecv_64k, tc) 1011261081Sasomers{ 1012261081Sasomers test_sendrecv_symmetric_buffers(64 * 1024, true); 1013261081Sasomers} 1014261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_128k); 1015261081SasomersATF_TC_BODY(sendrecv_128k, tc) 1016261081Sasomers{ 1017261081Sasomers test_sendrecv_symmetric_buffers(128 * 1024, true); 1018261081Sasomers} 1019261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_8k_nonblocking); 1020261081SasomersATF_TC_BODY(sendrecv_8k_nonblocking, tc) 1021261081Sasomers{ 1022261081Sasomers test_sendrecv_symmetric_buffers(8 * 1024, false); 1023261081Sasomers} 1024261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_16k_nonblocking); 1025261081SasomersATF_TC_BODY(sendrecv_16k_nonblocking, tc) 1026261081Sasomers{ 1027261081Sasomers test_sendrecv_symmetric_buffers(16 * 1024, false); 1028261081Sasomers} 1029261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_32k_nonblocking); 1030261081SasomersATF_TC_BODY(sendrecv_32k_nonblocking, tc) 1031261081Sasomers{ 1032261081Sasomers test_sendrecv_symmetric_buffers(32 * 1024, false); 1033261081Sasomers} 1034261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_64k_nonblocking); 1035261081SasomersATF_TC_BODY(sendrecv_64k_nonblocking, tc) 1036261081Sasomers{ 1037261081Sasomers test_sendrecv_symmetric_buffers(64 * 1024, false); 1038261081Sasomers} 1039261081SasomersATF_TC_WITHOUT_HEAD(sendrecv_128k_nonblocking); 1040261081SasomersATF_TC_BODY(sendrecv_128k_nonblocking, tc) 1041261081Sasomers{ 1042261081Sasomers test_sendrecv_symmetric_buffers(128 * 1024, false); 1043261081Sasomers} 1044261081Sasomers 1045261081Sasomers 1046261081Sasomers/* 1047261081Sasomers * Main. 1048261081Sasomers */ 1049261081Sasomers 1050261081SasomersATF_TP_ADD_TCS(tp) 1051261081Sasomers{ 1052261081Sasomers /* Basic creation and connection tests */ 1053261081Sasomers ATF_TP_ADD_TC(tp, create_socket); 1054261081Sasomers ATF_TP_ADD_TC(tp, create_socketpair); 1055261081Sasomers ATF_TP_ADD_TC(tp, listen_unbound); 1056261081Sasomers ATF_TP_ADD_TC(tp, bind); 1057261081Sasomers ATF_TP_ADD_TC(tp, listen_bound); 1058261081Sasomers ATF_TP_ADD_TC(tp, connect); 1059261081Sasomers ATF_TP_ADD_TC(tp, accept); 1060261081Sasomers ATF_TP_ADD_TC(tp, fcntl_nonblock); 1061261081Sasomers ATF_TP_ADD_TC(tp, resize_buffers); 1062261081Sasomers ATF_TP_ADD_TC(tp, resize_connected_buffers); 1063261081Sasomers 1064261081Sasomers /* Unthreaded I/O tests */ 1065261081Sasomers ATF_TP_ADD_TC(tp, send_recv); 1066261081Sasomers ATF_TP_ADD_TC(tp, send_recv_nonblocking); 1067261081Sasomers ATF_TP_ADD_TC(tp, send_recv_with_connect); 1068261081Sasomers ATF_TP_ADD_TC(tp, sendto_recvfrom); 1069261081Sasomers ATF_TP_ADD_TC(tp, shutdown_send); 1070261081Sasomers ATF_TP_ADD_TC(tp, shutdown_send_sigpipe); 1071261081Sasomers ATF_TP_ADD_TC(tp, emsgsize); 1072261081Sasomers ATF_TP_ADD_TC(tp, emsgsize_nonblocking); 1073261081Sasomers ATF_TP_ADD_TC(tp, eagain_8k_8k); 1074261081Sasomers ATF_TP_ADD_TC(tp, eagain_8k_128k); 1075261081Sasomers ATF_TP_ADD_TC(tp, eagain_128k_8k); 1076261081Sasomers ATF_TP_ADD_TC(tp, eagain_128k_128k); 1077261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_8k); 1078261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_16k); 1079261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_32k); 1080261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_64k); 1081261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_128k); 1082261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_8k_nonblocking); 1083261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_16k_nonblocking); 1084261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_32k_nonblocking); 1085261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_64k_nonblocking); 1086261081Sasomers ATF_TP_ADD_TC(tp, sendrecv_128k_nonblocking); 1087261081Sasomers ATF_TP_ADD_TC(tp, rcvbuf_oversized); 1088261081Sasomers ATF_TP_ADD_TC(tp, pipe_simulator_8k_8k); 1089261081Sasomers ATF_TP_ADD_TC(tp, pipe_simulator_8k_128k); 1090261081Sasomers ATF_TP_ADD_TC(tp, pipe_simulator_128k_8k); 1091261081Sasomers ATF_TP_ADD_TC(tp, pipe_simulator_128k_128k); 1092261081Sasomers 1093261081Sasomers /* Threaded I/O tests with blocking sockets */ 1094261081Sasomers ATF_TP_ADD_TC(tp, pipe_8k_8k); 1095261081Sasomers ATF_TP_ADD_TC(tp, pipe_8k_128k); 1096261081Sasomers ATF_TP_ADD_TC(tp, pipe_128k_8k); 1097261081Sasomers ATF_TP_ADD_TC(tp, pipe_128k_128k); 1098261081Sasomers 1099261081Sasomers return atf_no_error(); 1100261081Sasomers} 1101