Home | History | Annotate | Line # | Download | only in sys
t_socketpair.c revision 1.2 
      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 	ATF_REQUIRE(closefrom(3) != -1);
     67 
     68 	ATF_REQUIRE(socketpair(AF_UNIX, SOCK_DGRAM | flags, 0, fd) == 0);
     69 
     70 	ATF_REQUIRE(fd[0] == 3);
     71 	ATF_REQUIRE(fd[1] == 4);
     72 
     73 	connected(fd[0]);
     74 	connected(fd[1]);
     75 
     76 	if (flags & SOCK_CLOEXEC) {
     77 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) != 0);
     78 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) != 0);
     79 	} else {
     80 		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) == 0);
     81 		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) == 0);
     82 	}
     83 
     84 	if (flags & SOCK_NONBLOCK) {
     85 		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) != 0);
     86 		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) != 0);
     87 	} else {
     88 		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) == 0);
     89 		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) == 0);
     90 	}
     91 
     92 	ATF_REQUIRE(close(fd[0]) != -1);
     93 	ATF_REQUIRE(close(fd[1]) != -1);
     94 }
     95 
     96 ATF_TC(socketpair_basic);
     97 ATF_TC_HEAD(socketpair_basic, tc)
     98 {
     99 	atf_tc_set_md_var(tc, "descr", "A basic test of socketpair(2)");
    100 }
    101 
    102 ATF_TC_BODY(socketpair_basic, tc)
    103 {
    104 	run(0);
    105 }
    106 
    107 ATF_TC(socketpair_nonblock);
    108 ATF_TC_HEAD(socketpair_nonblock, tc)
    109 {
    110 	atf_tc_set_md_var(tc, "descr", "A non-blocking test of socketpair(2)");
    111 }
    112 
    113 ATF_TC_BODY(socketpair_nonblock, tc)
    114 {
    115 	run(SOCK_NONBLOCK);
    116 }
    117 
    118 ATF_TC(socketpair_cloexec);
    119 ATF_TC_HEAD(socketpair_cloexec, tc)
    120 {
    121 	atf_tc_set_md_var(tc, "descr", "A close-on-exec of socketpair(2)");
    122 }
    123 
    124 ATF_TC_BODY(socketpair_cloexec, tc)
    125 {
    126 	run(SOCK_CLOEXEC);
    127 }
    128 
    129 ATF_TP_ADD_TCS(tp)
    130 {
    131 
    132 	ATF_TP_ADD_TC(tp, socketpair_basic);
    133 	ATF_TP_ADD_TC(tp, socketpair_nonblock);
    134 	ATF_TP_ADD_TC(tp, socketpair_cloexec);
    135 
    136 	return atf_no_error();
    137 }
    138