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 49static void 50connected(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 58static void 59run(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 100ATF_TC(inet); 101ATF_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 107ATF_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 117ATF_TC(null_sv); 118ATF_TC_HEAD(null_sv, tc) 119{ 120 atf_tc_set_md_var(tc, "descr", 121 "socketpair(2) should fail without return storage"); 122} 123 124ATF_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 136ATF_TC(socketpair_basic); 137ATF_TC_HEAD(socketpair_basic, tc) 138{ 139 atf_tc_set_md_var(tc, "descr", "A basic test of socketpair(2)"); 140} 141 142ATF_TC_BODY(socketpair_basic, tc) 143{ 144 run(AF_UNIX, SOCK_DGRAM, 0); 145} 146 147ATF_TC(socketpair_nonblock); 148ATF_TC_HEAD(socketpair_nonblock, tc) 149{ 150 atf_tc_set_md_var(tc, "descr", "A non-blocking test of socketpair(2)"); 151} 152 153ATF_TC_BODY(socketpair_nonblock, tc) 154{ 155 run(AF_UNIX, SOCK_DGRAM, SOCK_NONBLOCK); 156} 157 158ATF_TC(socketpair_cloexec); 159ATF_TC_HEAD(socketpair_cloexec, tc) 160{ 161 atf_tc_set_md_var(tc, "descr", "A close-on-exec of socketpair(2)"); 162} 163 164ATF_TC_BODY(socketpair_cloexec, tc) 165{ 166 run(AF_UNIX, SOCK_DGRAM, SOCK_CLOEXEC); 167} 168 169ATF_TC(socketpair_stream); 170ATF_TC_HEAD(socketpair_stream, tc) 171{ 172 atf_tc_set_md_var(tc, "descr", "A stream-oriented socketpair(2)"); 173} 174 175ATF_TC_BODY(socketpair_stream, tc) 176{ 177 run(AF_UNIX, SOCK_STREAM, 0); 178} 179 180ATF_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