Home | History | Annotate | Line # | Download | only in sys 
      1 /* $NetBSD: t_socketpair.c,v 1.3 2025/07/17 19:50:40 kre 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.3 2025/07/17 19:50:40 kre 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 #ifndef	SOCK_CLOFORK
     50 #define	SOCK_CLOFORK	0
     51 #endif
     52 #ifndef	FD_CLOFORK
     53 #define	FD_CLOFORK	0
     54 #endif
     55 
     56 static void
     57 connected(int fd)
     58 {
     59 	struct sockaddr_un addr;
     60 	socklen_t len = (socklen_t)sizeof(addr);
     61 	ATF_REQUIRE(getpeername(fd, (struct sockaddr*)(void *)&addr,
     62 	    &len) == 0);
     63 }
     64 
     65 static void
     66 run(int flags)
     67 {
     68 	int fd[2], i;
     69 
     70 	while ((i = open("/", O_RDONLY)) < 3)
     71 		ATF_REQUIRE(i != -1);
     72 
     73 	ATF_REQUIRE(closefrom(3) != -1);
     74 
     75 	ATF_REQUIRE(socketpair(AF_UNIX, SOCK_DGRAM | flags, 0, fd) == 0);
     76 
     77 	ATF_REQUIRE(fd[0] == 3);
     78 	ATF_REQUIRE(fd[1] == 4);
     79 
     80 	connected(fd[0]);
     81 	connected(fd[1]);
     82 
     83 	if (flags & SOCK_CLOEXEC) {
     84 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) != 0);
     85 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) != 0);
     86 	} else {
     87 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) == 0);
     88 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) == 0);
     89 	}
     90 
     91 	if (flags & SOCK_CLOFORK) {
     92 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOFORK) != 0);
     93 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOFORK) != 0);
     94 	} else {
     95 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOFORK) == 0);
     96 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOFORK) == 0);
     97 	}
     98 
     99 	if (flags & SOCK_NONBLOCK) {
    100 		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) != 0);
    101 		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) != 0);
    102 	} else {
    103 		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) == 0);
    104 		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) == 0);
    105 	}
    106 
    107 	ATF_REQUIRE(close(fd[0]) != -1);
    108 	ATF_REQUIRE(close(fd[1]) != -1);
    109 }
    110 
    111 ATF_TC(socketpair_basic);
    112 ATF_TC_HEAD(socketpair_basic, tc)
    113 {
    114 	atf_tc_set_md_var(tc, "descr", "A basic test of socketpair(2)");
    115 }
    116 
    117 ATF_TC_BODY(socketpair_basic, tc)
    118 {
    119 	run(0);
    120 }
    121 
    122 ATF_TC(socketpair_nonblock);
    123 ATF_TC_HEAD(socketpair_nonblock, tc)
    124 {
    125 	atf_tc_set_md_var(tc, "descr", "A non-blocking test of socketpair(2)");
    126 }
    127 
    128 ATF_TC_BODY(socketpair_nonblock, tc)
    129 {
    130 	run(SOCK_NONBLOCK);
    131 }
    132 
    133 ATF_TC(socketpair_cloexec);
    134 ATF_TC_HEAD(socketpair_cloexec, tc)
    135 {
    136 	atf_tc_set_md_var(tc, "descr", "A close-on-exec of socketpair(2)");
    137 }
    138 
    139 ATF_TC_BODY(socketpair_cloexec, tc)
    140 {
    141 	run(SOCK_CLOEXEC);
    142 }
    143 
    144 ATF_TC(socketpair_clofork);
    145 ATF_TC_HEAD(socketpair_clofork, tc)
    146 {
    147 	atf_tc_set_md_var(tc, "descr", "A close-on-fork of socketpair(2)");
    148 }
    149 
    150 ATF_TC_BODY(socketpair_clofork, tc)
    151 {
    152 #if defined(SOCK_CLOFORK) && SOCK_CLOFORK != 0
    153 	run(SOCK_CLOFORK);
    154 #else
    155 	atf_tc_skip("SOCK_CLOFORK not defined");
    156 #endif
    157 }
    158 
    159 ATF_TP_ADD_TCS(tp)
    160 {
    161 
    162 	ATF_TP_ADD_TC(tp, socketpair_basic);
    163 	ATF_TP_ADD_TC(tp, socketpair_nonblock);
    164 	ATF_TP_ADD_TC(tp, socketpair_cloexec);
    165 	ATF_TP_ADD_TC(tp, socketpair_clofork);
    166 
    167 	return atf_no_error();
    168 }
    169