1 /* $NetBSD: t_socketpair.c,v 1.2 2017/01/13 20:04:52 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2011 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Christos Zoulas. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. All advertising materials mentioning features or use of this software 19 * must display the following acknowledgement: 20 * This product includes software developed by the NetBSD 21 * Foundation, Inc. and its contributors. 22 * 4. Neither the name of The NetBSD Foundation nor the names of its 23 * contributors may be used to endorse or promote products derived 24 * from this software without specific prior written permission. 25 * 26 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 28 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 29 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 30 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 31 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 32 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 33 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 34 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 35 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 36 * POSSIBILITY OF SUCH DAMAGE. 37 */ 38 #include <sys/cdefs.h> 39 __RCSID("$NetBSD: t_socketpair.c,v 1.2 2017/01/13 20:04:52 christos Exp $"); 40 41 #include <atf-c.h> 42 #include <fcntl.h> 43 #include <unistd.h> 44 #include <stdlib.h> 45 #include <sys/socket.h> 46 #include <sys/un.h> 47 #include <errno.h> 48 49 static void 50 connected(int fd) 51 { 52 struct sockaddr_un addr; 53 socklen_t len = (socklen_t)sizeof(addr); 54 ATF_REQUIRE(getpeername(fd, (struct sockaddr*)(void *)&addr, 55 &len) == 0); 56 } 57 58 static void 59 run(int flags) 60 { 61 int fd[2], i; 62 63 while ((i = open("/", O_RDONLY)) < 3) 64 ATF_REQUIRE(i != -1); 65 66 #ifdef __FreeBSD__ 67 closefrom(3); 68 #else 69 ATF_REQUIRE(closefrom(3) != -1); 70 #endif 71 72 ATF_REQUIRE(socketpair(AF_UNIX, SOCK_DGRAM | flags, 0, fd) == 0); 73 74 ATF_REQUIRE(fd[0] == 3); 75 ATF_REQUIRE(fd[1] == 4); 76 77 connected(fd[0]); 78 connected(fd[1]); 79 80 if (flags & SOCK_CLOEXEC) { 81 ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) != 0); 82 ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) != 0); 83 } else { 84 ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) == 0); 85 ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) == 0); 86 } 87 88 if (flags & SOCK_NONBLOCK) { 89 ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) != 0); 90 ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) != 0); 91 } else { 92 ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) == 0); 93 ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) == 0); 94 } 95 96 ATF_REQUIRE(close(fd[0]) != -1); 97 ATF_REQUIRE(close(fd[1]) != -1); 98 } 99 100 ATF_TC(socketpair_basic); 101 ATF_TC_HEAD(socketpair_basic, tc) 102 { 103 atf_tc_set_md_var(tc, "descr", "A basic test of socketpair(2)"); 104 } 105 106 ATF_TC_BODY(socketpair_basic, tc) 107 { 108 run(0); 109 } 110 111 ATF_TC(socketpair_nonblock); 112 ATF_TC_HEAD(socketpair_nonblock, tc) 113 { 114 atf_tc_set_md_var(tc, "descr", "A non-blocking test of socketpair(2)"); 115 } 116 117 ATF_TC_BODY(socketpair_nonblock, tc) 118 { 119 run(SOCK_NONBLOCK); 120 } 121 122 ATF_TC(socketpair_cloexec); 123 ATF_TC_HEAD(socketpair_cloexec, tc) 124 { 125 atf_tc_set_md_var(tc, "descr", "A close-on-exec of socketpair(2)"); 126 } 127 128 ATF_TC_BODY(socketpair_cloexec, tc) 129 { 130 run(SOCK_CLOEXEC); 131 } 132 133 ATF_TP_ADD_TCS(tp) 134 { 135 136 ATF_TP_ADD_TC(tp, socketpair_basic); 137 ATF_TP_ADD_TC(tp, socketpair_nonblock); 138 ATF_TP_ADD_TC(tp, socketpair_cloexec); 139 140 return atf_no_error(); 141 } 142