1*640235e2SEnji Cooper /* $NetBSD: t_tcp.c,v 1.4 2016/03/04 18:52:01 christos Exp $ */
257718be8SEnji Cooper
357718be8SEnji Cooper /*-
457718be8SEnji Cooper * Copyright (c) 2013 The NetBSD Foundation, Inc.
557718be8SEnji Cooper * All rights reserved.
657718be8SEnji Cooper *
757718be8SEnji Cooper * Redistribution and use in source and binary forms, with or without
857718be8SEnji Cooper * modification, are permitted provided that the following conditions
957718be8SEnji Cooper * are met:
1057718be8SEnji Cooper * 1. Redistributions of source code must retain the above copyright
1157718be8SEnji Cooper * notice, this list of conditions and the following disclaimer.
1257718be8SEnji Cooper * 2. Redistributions in binary form must reproduce the above copyright
1357718be8SEnji Cooper * notice, this list of conditions and the following disclaimer in the
1457718be8SEnji Cooper * documentation and/or other materials provided with the distribution.
1557718be8SEnji Cooper * 3. All advertising materials mentioning features or use of this software
1657718be8SEnji Cooper * must display the following acknowledgement:
1757718be8SEnji Cooper * This product includes software developed by the NetBSD
1857718be8SEnji Cooper * Foundation, Inc. and its contributors.
1957718be8SEnji Cooper * 4. Neither the name of The NetBSD Foundation nor the names of its
2057718be8SEnji Cooper * contributors may be used to endorse or promote products derived
2157718be8SEnji Cooper * from this software without specific prior written permission.
2257718be8SEnji Cooper *
2357718be8SEnji Cooper * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
2457718be8SEnji Cooper * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
2557718be8SEnji Cooper * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
2657718be8SEnji Cooper * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
2757718be8SEnji Cooper * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
2857718be8SEnji Cooper * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
2957718be8SEnji Cooper * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
3057718be8SEnji Cooper * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
3157718be8SEnji Cooper * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
3257718be8SEnji Cooper * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
3357718be8SEnji Cooper * POSSIBILITY OF SUCH DAMAGE.
3457718be8SEnji Cooper */
3557718be8SEnji Cooper
3657718be8SEnji Cooper #include <sys/cdefs.h>
3757718be8SEnji Cooper #ifdef __RCSID
38*640235e2SEnji Cooper __RCSID("$Id: t_tcp.c,v 1.4 2016/03/04 18:52:01 christos Exp $");
3957718be8SEnji Cooper #endif
4057718be8SEnji Cooper
4157718be8SEnji Cooper /* Example code. Should block; does with accept not paccept. */
4257718be8SEnji Cooper /* Original by: Justin Cormack <justin@specialbusrvrervice.com> */
4357718be8SEnji Cooper
4457718be8SEnji Cooper #include <sys/param.h>
4557718be8SEnji Cooper #include <sys/types.h>
4657718be8SEnji Cooper #include <sys/socket.h>
4757718be8SEnji Cooper #include <netinet/in.h>
4857718be8SEnji Cooper #include <stdio.h>
4957718be8SEnji Cooper #include <stdbool.h>
5057718be8SEnji Cooper #include <unistd.h>
5157718be8SEnji Cooper #include <fcntl.h>
5257718be8SEnji Cooper #include <errno.h>
5357718be8SEnji Cooper #include <err.h>
5457718be8SEnji Cooper #include <stdlib.h>
5557718be8SEnji Cooper #include <signal.h>
5657718be8SEnji Cooper
5757718be8SEnji Cooper #ifdef TEST
5857718be8SEnji Cooper #define FAIL(msg, ...) err(EXIT_FAILURE, msg, ## __VA_ARGS__)
5957718be8SEnji Cooper #else
6057718be8SEnji Cooper #include <atf-c.h>
6157718be8SEnji Cooper #define FAIL(msg, ...) ATF_CHECK_MSG(0, msg, ## __VA_ARGS__); goto fail
6257718be8SEnji Cooper #endif
6357718be8SEnji Cooper
64*640235e2SEnji Cooper #ifdef __linux__
65*640235e2SEnji Cooper #define paccept(a, b, c, d, e) accept4((a), (b), (c), (e))
66*640235e2SEnji Cooper #endif
67*640235e2SEnji Cooper
6857718be8SEnji Cooper static void
ding(int al)6957718be8SEnji Cooper ding(int al)
7057718be8SEnji Cooper {
7157718be8SEnji Cooper }
7257718be8SEnji Cooper
7357718be8SEnji Cooper static void
paccept_block(bool pacceptblock,bool fcntlblock)7457718be8SEnji Cooper paccept_block(bool pacceptblock, bool fcntlblock)
7557718be8SEnji Cooper {
7657718be8SEnji Cooper int srvr = -1, clnt = -1, as = -1;
77*640235e2SEnji Cooper int ok, fl;
78*640235e2SEnji Cooper ssize_t n;
7957718be8SEnji Cooper char buf[10];
8057718be8SEnji Cooper struct sockaddr_in sin, ba;
8157718be8SEnji Cooper struct sigaction sa;
8257718be8SEnji Cooper
8357718be8SEnji Cooper srvr = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
8457718be8SEnji Cooper if (srvr == -1)
8557718be8SEnji Cooper FAIL("socket");
8657718be8SEnji Cooper
8757718be8SEnji Cooper memset(&sin, 0, sizeof(sin));
8857718be8SEnji Cooper sin.sin_family = AF_INET;
8957718be8SEnji Cooper #ifdef BSD4_4
9057718be8SEnji Cooper sin.sin_len = sizeof(sin);
9157718be8SEnji Cooper #endif
9257718be8SEnji Cooper sin.sin_port = htons(0);
9357718be8SEnji Cooper sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
9457718be8SEnji Cooper ok = bind(srvr, (const struct sockaddr *)&sin, (socklen_t)sizeof(sin));
9557718be8SEnji Cooper if (ok == -1)
9657718be8SEnji Cooper FAIL("bind");
9757718be8SEnji Cooper
9857718be8SEnji Cooper socklen_t addrlen = sizeof(struct sockaddr_in);
9957718be8SEnji Cooper ok = getsockname(srvr, (struct sockaddr *)&ba, &addrlen);
10057718be8SEnji Cooper if (ok == -1)
10157718be8SEnji Cooper FAIL("getsockname");
10257718be8SEnji Cooper
10357718be8SEnji Cooper ok = listen(srvr, SOMAXCONN);
10457718be8SEnji Cooper if (ok == -1)
10557718be8SEnji Cooper FAIL("listen");
10657718be8SEnji Cooper
10757718be8SEnji Cooper clnt = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
10857718be8SEnji Cooper if (clnt == -1)
10957718be8SEnji Cooper FAIL("socket");
11057718be8SEnji Cooper
11157718be8SEnji Cooper /* may not connect first time */
11257718be8SEnji Cooper ok = connect(clnt, (struct sockaddr *) &ba, addrlen);
113*640235e2SEnji Cooper if (ok != -1 || errno != EINPROGRESS)
114*640235e2SEnji Cooper FAIL("expected connect to fail");
11557718be8SEnji Cooper as = paccept(srvr, NULL, NULL, NULL, pacceptblock ? 0 : SOCK_NONBLOCK);
11657718be8SEnji Cooper ok = connect(clnt, (struct sockaddr *) &ba, addrlen);
11757718be8SEnji Cooper if (ok == -1 && errno != EISCONN)
118*640235e2SEnji Cooper FAIL("connect failed");
11957718be8SEnji Cooper
12057718be8SEnji Cooper #if 0
12157718be8SEnji Cooper fl = fcntl(srvr, F_GETFL, 0);
12257718be8SEnji Cooper if (fl == -1)
12357718be8SEnji Cooper FAIL("fnctl getfl");
12457718be8SEnji Cooper
12557718be8SEnji Cooper ok = fcntl(srvr, F_SETFL, fl & ~O_NONBLOCK);
12657718be8SEnji Cooper if (ok == -1)
12757718be8SEnji Cooper FAIL("fnctl setfl");
12857718be8SEnji Cooper #endif
12957718be8SEnji Cooper
13057718be8SEnji Cooper if (as == -1) { /* not true under NetBSD */
13157718be8SEnji Cooper as = paccept(srvr, NULL, NULL, NULL, pacceptblock ? 0 : SOCK_NONBLOCK);
13257718be8SEnji Cooper if (as == -1)
13357718be8SEnji Cooper FAIL("paccept");
13457718be8SEnji Cooper }
13557718be8SEnji Cooper if (fcntlblock) {
13657718be8SEnji Cooper fl = fcntl(as, F_GETFL, 0);
13757718be8SEnji Cooper if (fl == -1)
13857718be8SEnji Cooper FAIL("fnctl");
13957718be8SEnji Cooper if (fl != (O_RDWR|O_NONBLOCK))
14057718be8SEnji Cooper FAIL("fl 0x%x != 0x%x\n", fl, O_RDWR|O_NONBLOCK);
14157718be8SEnji Cooper ok = fcntl(as, F_SETFL, fl & ~O_NONBLOCK);
14257718be8SEnji Cooper if (ok == -1)
14357718be8SEnji Cooper FAIL("fnctl setfl");
14457718be8SEnji Cooper
14557718be8SEnji Cooper fl = fcntl(as, F_GETFL, 0);
14657718be8SEnji Cooper if (fl & O_NONBLOCK)
14757718be8SEnji Cooper FAIL("fl non blocking after reset");
14857718be8SEnji Cooper }
14957718be8SEnji Cooper sa.sa_handler = ding;
15057718be8SEnji Cooper sa.sa_flags = 0;
15157718be8SEnji Cooper sigemptyset(&sa.sa_mask);
15257718be8SEnji Cooper sigaction(SIGALRM, &sa, NULL);
15357718be8SEnji Cooper alarm(1);
15457718be8SEnji Cooper n = read(as, buf, 10);
15557718be8SEnji Cooper
15657718be8SEnji Cooper if (pacceptblock || fcntlblock) {
15757718be8SEnji Cooper if (n == -1 && errno != EINTR)
15857718be8SEnji Cooper FAIL("read");
15957718be8SEnji Cooper } else {
16057718be8SEnji Cooper if (n != -1 || errno != EWOULDBLOCK)
16157718be8SEnji Cooper FAIL("read");
16257718be8SEnji Cooper }
16357718be8SEnji Cooper return;
16457718be8SEnji Cooper fail:
16557718be8SEnji Cooper close(srvr);
16657718be8SEnji Cooper close(clnt);
16757718be8SEnji Cooper close(as);
16857718be8SEnji Cooper }
16957718be8SEnji Cooper
17057718be8SEnji Cooper #ifndef TEST
17157718be8SEnji Cooper
17257718be8SEnji Cooper ATF_TC(paccept_reset_nonblock);
ATF_TC_HEAD(paccept_reset_nonblock,tc)17357718be8SEnji Cooper ATF_TC_HEAD(paccept_reset_nonblock, tc)
17457718be8SEnji Cooper {
17557718be8SEnji Cooper
17657718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Check that paccept(2) resets "
17757718be8SEnji Cooper "the non-blocking flag on non-blocking sockets");
17857718be8SEnji Cooper }
17957718be8SEnji Cooper
ATF_TC_BODY(paccept_reset_nonblock,tc)18057718be8SEnji Cooper ATF_TC_BODY(paccept_reset_nonblock, tc)
18157718be8SEnji Cooper {
18257718be8SEnji Cooper paccept_block(true, false);
18357718be8SEnji Cooper }
18457718be8SEnji Cooper
18557718be8SEnji Cooper ATF_TC(fcntl_reset_nonblock);
ATF_TC_HEAD(fcntl_reset_nonblock,tc)18657718be8SEnji Cooper ATF_TC_HEAD(fcntl_reset_nonblock, tc)
18757718be8SEnji Cooper {
18857718be8SEnji Cooper
18957718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Check that fcntl(2) resets "
19057718be8SEnji Cooper "the non-blocking flag on non-blocking sockets");
19157718be8SEnji Cooper }
19257718be8SEnji Cooper
ATF_TC_BODY(fcntl_reset_nonblock,tc)19357718be8SEnji Cooper ATF_TC_BODY(fcntl_reset_nonblock, tc)
19457718be8SEnji Cooper {
19557718be8SEnji Cooper paccept_block(false, true);
19657718be8SEnji Cooper }
19757718be8SEnji Cooper
19857718be8SEnji Cooper ATF_TC(paccept_nonblock);
ATF_TC_HEAD(paccept_nonblock,tc)19957718be8SEnji Cooper ATF_TC_HEAD(paccept_nonblock, tc)
20057718be8SEnji Cooper {
20157718be8SEnji Cooper
20257718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Check that fcntl(2) resets "
20357718be8SEnji Cooper "the non-blocking flag on non-blocking sockets");
20457718be8SEnji Cooper }
20557718be8SEnji Cooper
ATF_TC_BODY(paccept_nonblock,tc)20657718be8SEnji Cooper ATF_TC_BODY(paccept_nonblock, tc)
20757718be8SEnji Cooper {
20857718be8SEnji Cooper paccept_block(false, false);
20957718be8SEnji Cooper }
21057718be8SEnji Cooper
ATF_TP_ADD_TCS(tp)21157718be8SEnji Cooper ATF_TP_ADD_TCS(tp)
21257718be8SEnji Cooper {
21357718be8SEnji Cooper
21457718be8SEnji Cooper ATF_TP_ADD_TC(tp, paccept_reset_nonblock);
21557718be8SEnji Cooper ATF_TP_ADD_TC(tp, fcntl_reset_nonblock);
21657718be8SEnji Cooper ATF_TP_ADD_TC(tp, paccept_nonblock);
21757718be8SEnji Cooper return atf_no_error();
21857718be8SEnji Cooper }
21957718be8SEnji Cooper #else
22057718be8SEnji Cooper int
main(int argc,char * argv[])22157718be8SEnji Cooper main(int argc, char *argv[])
22257718be8SEnji Cooper {
22357718be8SEnji Cooper paccept_block(false);
22457718be8SEnji Cooper paccept_block(true);
22557718be8SEnji Cooper return 0;
22657718be8SEnji Cooper }
22757718be8SEnji Cooper #endif
228