1/*	$NetBSD: tcp_shutdown.c,v 1.1 2022/11/04 08:01:42 ozaki-r Exp $	*/
2
3/*-
4 * Copyright (c) 2022 Internet Initiative Japan 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 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31#include <sys/cdefs.h>
32#ifdef __RCSID
33__RCSID("$NetBSD: tcp_shutdown.c,v 1.1 2022/11/04 08:01:42 ozaki-r Exp $");
34#else
35extern const char *__progname;
36#define getprogname() __progname
37#endif
38
39#include <sys/types.h>
40#include <sys/socket.h>
41#include <netinet/in.h>
42#include <arpa/inet.h>
43
44#include <stdio.h>
45#include <string.h>
46#include <stdlib.h>
47#include <unistd.h>
48#include <err.h>
49#include <errno.h>
50#include <stdbool.h>
51
52static inline bool
53match(const char *a, const char *b)
54{
55
56	return strncmp(a, b, strlen(b)) == 0;
57}
58
59int
60main(int argc, char *argv[])
61{
62	int s, e;
63	char *target;
64
65	if (argc != 2)
66		errx(EXIT_FAILURE, "invalid argument");
67	target = argv[1];
68
69	s = socket(AF_INET, SOCK_STREAM, 0);
70	if (s == -1)
71		err(EXIT_FAILURE, "socket");
72	e = shutdown(s, SHUT_RDWR);
73	if (e == -1)
74		err(EXIT_FAILURE, "shutdown");
75
76	if (match(target, "connect")) {
77		struct sockaddr_in sin;
78
79		memset(&sin, 0, sizeof(sin));
80		sin.sin_port = htons(31522);
81		sin.sin_addr.s_addr = inet_addr("127.0.0.1");
82		sin.sin_family = AF_INET;
83
84		e = connect(s, (struct sockaddr *)&sin, sizeof(sin));
85		if (e == 0)
86			err(EXIT_FAILURE, "connect didn't fail on a shudown socket");
87		if (e == -1 && errno != EINVAL)
88			err(EXIT_FAILURE, "connect failed with unexpected error");
89	} else if (match(target, "setsockopt")) {
90		int opt = 1;
91		e = setsockopt(s, SOL_SOCKET, SO_REUSEPORT, &opt, sizeof(opt));
92		if (e == 0)
93			err(EXIT_FAILURE, "setsockopt didn't fail on a shutdown socket");
94		if (e == -1 && errno != ECONNRESET)
95			err(EXIT_FAILURE, "setsockopt failed with unexpected error");
96	} else if (match(target, "getsockname")) {
97		struct sockaddr_storage ss;
98		socklen_t len;
99		e = getsockname(s, (struct sockaddr *)&ss, &len);
100		if (e == 0)
101			err(EXIT_FAILURE, "getsockname didn't fail on a shutdown socket");
102		if (e == -1 && errno != EINVAL)
103			err(EXIT_FAILURE, "getsockname failed with unexpected error");
104	} else if (match(target, "listen")) {
105		e = listen(s, 5);
106		if (e == 0)
107			err(EXIT_FAILURE, "listen didn't fail on a shutdown socket");
108		if (e == -1 && errno != EINVAL)
109			err(EXIT_FAILURE, "listen failed with unexpected error");
110	} else if (match(target, "bind")) {
111		struct sockaddr_in sin;
112
113		memset(&sin, 0, sizeof(sin));
114		sin.sin_port = htons(31522);
115		sin.sin_addr.s_addr = inet_addr("127.0.0.1");
116		sin.sin_family = AF_INET;
117
118		e = bind(s, (struct sockaddr *)&sin, sizeof(sin));
119		if (e == 0)
120			err(EXIT_FAILURE, "bind didn't fail on a shutdown socket");
121		if (e == -1 && errno != EINVAL)
122			err(EXIT_FAILURE, "bind failed with unexpected error");
123	} else if (match(target, "shutdown")) {
124		e = shutdown(s, SHUT_RDWR);
125		if (e == 0)
126			err(EXIT_FAILURE, "shutdown didn't fail on a shutdown socket");
127		if (e == -1 && errno != EINVAL)
128			err(EXIT_FAILURE, "shutdown failed with unexpected error");
129	} else {
130		errx(EXIT_FAILURE, "unknown target: %s", target);
131	}
132
133	e = close(s);
134	if (e == -1)
135		err(EXIT_FAILURE, "close");
136	return 0;
137}
138