contrib/netbsd-tests/net/net/t_tcp.c

   1 /*      $NetBSD: t_tcp.c,v 1.4 2016/03/04 18:52:01 christos Exp $       */
   2
   3 /*-
   4  * Copyright (c) 2013 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *        This product includes software developed by the NetBSD
  18  *        Foundation, Inc. and its contributors.
  19  * 4. Neither the name of The NetBSD Foundation nor the names of its
  20  *    contributors may be used to endorse or promote products derived
  21  *    from this software without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  24  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  25  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  26  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  27  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  28  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  29  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  30  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  31  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  32  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  33  * POSSIBILITY OF SUCH DAMAGE.
  34  */
  35
  36 #include <sys/cdefs.h>
  37 #ifdef __RCSID
  38 __RCSID("$Id: t_tcp.c,v 1.4 2016/03/04 18:52:01 christos Exp $");
  39 #endif
  40
  41 /* Example code. Should block; does with accept not paccept. */
  42 /* Original by: Justin Cormack <justin@specialbusrvrervice.com> */
  43
  44 #include <sys/param.h>
  45 #include <sys/types.h>
  46 #include <sys/socket.h>
  47 #include <netinet/in.h>
  48 #include <stdio.h>
  49 #include <stdbool.h>
  50 #include <unistd.h>
  51 #include <fcntl.h>
  52 #include <errno.h>
  53 #include <err.h>
  54 #include <stdlib.h>
  55 #include <signal.h>
  56
  57 #ifdef TEST
  58 #define FAIL(msg, ...)  err(EXIT_FAILURE, msg, ## __VA_ARGS__)
  59 #else
  60 #include <atf-c.h>
  61 #define FAIL(msg, ...)  ATF_CHECK_MSG(0, msg, ## __VA_ARGS__); goto fail
  62 #endif
  63
  64 #ifdef __linux__
  65 #define paccept(a, b, c, d, e) accept4((a), (b), (c), (e))
  66 #endif
  67
  68 static void
  69 ding(int al)
  70 {
  71 }
  72
  73 static void
  74 paccept_block(bool pacceptblock, bool fcntlblock)
  75 {
  76         int srvr = -1, clnt = -1, as = -1;
  77         int ok, fl;
  78         ssize_t n;
  79         char buf[10];
  80         struct sockaddr_in sin, ba;
  81         struct sigaction sa;
  82
  83         srvr = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
  84         if (srvr == -1)
  85                 FAIL("socket");
  86
  87         memset(&sin, 0, sizeof(sin));
  88         sin.sin_family = AF_INET;
  89 #ifdef BSD4_4
  90         sin.sin_len = sizeof(sin);
  91 #endif
  92         sin.sin_port = htons(0);
  93         sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
  94         ok = bind(srvr, (const struct sockaddr *)&sin, (socklen_t)sizeof(sin));
  95         if (ok == -1)
  96                 FAIL("bind");
  97
  98         socklen_t addrlen = sizeof(struct sockaddr_in);
  99         ok = getsockname(srvr, (struct sockaddr *)&ba, &addrlen);
 100         if (ok == -1)
 101                 FAIL("getsockname");
 102
 103         ok = listen(srvr, SOMAXCONN);
 104         if (ok == -1)
 105                 FAIL("listen");
 106
 107         clnt = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
 108         if (clnt == -1)
 109                 FAIL("socket");
 110
 111         /* may not connect first time */
 112         ok = connect(clnt, (struct sockaddr *) &ba, addrlen);
 113         if (ok != -1 || errno != EINPROGRESS)
 114                 FAIL("expected connect to fail");
 115         as = paccept(srvr, NULL, NULL, NULL, pacceptblock ? 0 : SOCK_NONBLOCK);
 116         ok = connect(clnt, (struct sockaddr *) &ba, addrlen);
 117         if (ok == -1 && errno != EISCONN)
 118                 FAIL("connect failed");
 119
 120 #if 0
 121         fl = fcntl(srvr, F_GETFL, 0);
 122         if (fl == -1)
 123                 FAIL("fnctl getfl");
 124
 125         ok = fcntl(srvr, F_SETFL, fl & ~O_NONBLOCK);
 126         if (ok == -1)
 127                 FAIL("fnctl setfl");
 128 #endif
 129
 130         if (as == -1) {         /* not true under NetBSD */
 131                 as = paccept(srvr, NULL, NULL, NULL, pacceptblock ? 0 : SOCK_NONBLOCK);
 132                 if (as == -1)
 133                         FAIL("paccept");
 134         }
 135         if (fcntlblock) {
 136                 fl = fcntl(as, F_GETFL, 0);
 137                 if (fl == -1)
 138                         FAIL("fnctl");
 139                 if (fl != (O_RDWR|O_NONBLOCK))
 140                         FAIL("fl 0x%x != 0x%x\n", fl, O_RDWR|O_NONBLOCK);
 141                 ok = fcntl(as, F_SETFL, fl & ~O_NONBLOCK);
 142                 if (ok == -1)
 143                         FAIL("fnctl setfl");
 144
 145                 fl = fcntl(as, F_GETFL, 0);
 146                 if (fl & O_NONBLOCK)
 147                         FAIL("fl non blocking after reset");
 148         }
 149         sa.sa_handler = ding;
 150         sa.sa_flags = 0;
 151         sigemptyset(&sa.sa_mask);
 152         sigaction(SIGALRM, &sa, NULL);
 153         alarm(1);
 154         n = read(as, buf, 10);
 155
 156         if (pacceptblock || fcntlblock) {
 157                 if (n == -1 && errno != EINTR)
 158                         FAIL("read");
 159         } else {
 160                 if (n != -1 || errno != EWOULDBLOCK)
 161                         FAIL("read");
 162         }
 163         return;
 164 fail:
 165         close(srvr);
 166         close(clnt);
 167         close(as);
 168 }
 169
 170 #ifndef TEST
 171
 172 ATF_TC(paccept_reset_nonblock);
 173 ATF_TC_HEAD(paccept_reset_nonblock, tc)
 174 {
 175
 176         atf_tc_set_md_var(tc, "descr", "Check that paccept(2) resets "
 177             "the non-blocking flag on non-blocking sockets");
 178 }
 179
 180 ATF_TC_BODY(paccept_reset_nonblock, tc)
 181 {
 182         paccept_block(true, false);
 183 }
 184
 185 ATF_TC(fcntl_reset_nonblock);
 186 ATF_TC_HEAD(fcntl_reset_nonblock, tc)
 187 {
 188
 189         atf_tc_set_md_var(tc, "descr", "Check that fcntl(2) resets "
 190             "the non-blocking flag on non-blocking sockets");
 191 }
 192
 193 ATF_TC_BODY(fcntl_reset_nonblock, tc)
 194 {
 195         paccept_block(false, true);
 196 }
 197
 198 ATF_TC(paccept_nonblock);
 199 ATF_TC_HEAD(paccept_nonblock, tc)
 200 {
 201
 202         atf_tc_set_md_var(tc, "descr", "Check that fcntl(2) resets "
 203             "the non-blocking flag on non-blocking sockets");
 204 }
 205
 206 ATF_TC_BODY(paccept_nonblock, tc)
 207 {
 208         paccept_block(false, false);
 209 }
 210
 211 ATF_TP_ADD_TCS(tp)
 212 {
 213
 214         ATF_TP_ADD_TC(tp, paccept_reset_nonblock);
 215         ATF_TP_ADD_TC(tp, fcntl_reset_nonblock);
 216         ATF_TP_ADD_TC(tp, paccept_nonblock);
 217         return atf_no_error();
 218 }
 219 #else
 220 int
 221 main(int argc, char *argv[])
 222 {
 223         paccept_block(false);
 224         paccept_block(true);
 225         return 0;
 226 }
 227 #endif