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 domain, int type, 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(domain, type | 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(inet); 101 ATF_TC_HEAD(inet, tc) 102 { 103 atf_tc_set_md_var(tc, "descr", 104 "socketpair(2) does not work in the PF_INET domain"); 105 } 106 107 ATF_TC_BODY(inet, tc) 108 { 109 int fd[2]; 110 111 ATF_REQUIRE_EQ(socketpair(PF_INET, SOCK_DGRAM, 0, fd), -1); 112 ATF_REQUIRE_EQ(EOPNOTSUPP, errno); 113 ATF_REQUIRE_EQ(socketpair(PF_INET, SOCK_STREAM, 0, fd), -1); 114 ATF_REQUIRE_EQ(EOPNOTSUPP, errno); 115 } 116 117 ATF_TC(null_sv); 118 ATF_TC_HEAD(null_sv, tc) 119 { 120 atf_tc_set_md_var(tc, "descr", 121 "socketpair(2) should fail without return storage"); 122 } 123 124 ATF_TC_BODY(null_sv, tc) 125 { 126 int fd; 127 128 closefrom(3); 129 ATF_REQUIRE_EQ(socketpair(AF_UNIX, SOCK_DGRAM, 0, NULL), -1); 130 ATF_REQUIRE_EQ(EFAULT, errno); 131 fd = open("/", O_RDONLY); 132 ATF_REQUIRE_EQ_MSG(fd, 3, 133 "socketpair(..., NULL) allocated descriptors"); 134 } 135 136 ATF_TC(socketpair_basic); 137 ATF_TC_HEAD(socketpair_basic, tc) 138 { 139 atf_tc_set_md_var(tc, "descr", "A basic test of socketpair(2)"); 140 } 141 142 ATF_TC_BODY(socketpair_basic, tc) 143 { 144 run(AF_UNIX, SOCK_DGRAM, 0); 145 } 146 147 ATF_TC(socketpair_nonblock); 148 ATF_TC_HEAD(socketpair_nonblock, tc) 149 { 150 atf_tc_set_md_var(tc, "descr", "A non-blocking test of socketpair(2)"); 151 } 152 153 ATF_TC_BODY(socketpair_nonblock, tc) 154 { 155 run(AF_UNIX, SOCK_DGRAM, SOCK_NONBLOCK); 156 } 157 158 ATF_TC(socketpair_cloexec); 159 ATF_TC_HEAD(socketpair_cloexec, tc) 160 { 161 atf_tc_set_md_var(tc, "descr", "A close-on-exec of socketpair(2)"); 162 } 163 164 ATF_TC_BODY(socketpair_cloexec, tc) 165 { 166 run(AF_UNIX, SOCK_DGRAM, SOCK_CLOEXEC); 167 } 168 169 ATF_TC(socketpair_stream); 170 ATF_TC_HEAD(socketpair_stream, tc) 171 { 172 atf_tc_set_md_var(tc, "descr", "A stream-oriented socketpair(2)"); 173 } 174 175 ATF_TC_BODY(socketpair_stream, tc) 176 { 177 run(AF_UNIX, SOCK_STREAM, 0); 178 } 179 180 ATF_TP_ADD_TCS(tp) 181 { 182 183 ATF_TP_ADD_TC(tp, inet); 184 ATF_TP_ADD_TC(tp, null_sv); 185 ATF_TP_ADD_TC(tp, socketpair_basic); 186 ATF_TP_ADD_TC(tp, socketpair_nonblock); 187 ATF_TP_ADD_TC(tp, socketpair_cloexec); 188 ATF_TP_ADD_TC(tp, socketpair_stream); 189 190 return atf_no_error(); 191 } 192