1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright Amazon.com Inc. or its affiliates. */
3
4 #include <sys/socket.h>
5 #include <netinet/in.h>
6
7 #include "../kselftest_harness.h"
8
FIXTURE(bind_timewait)9 FIXTURE(bind_timewait)
10 {
11 struct sockaddr_in addr;
12 socklen_t addrlen;
13 };
14
FIXTURE_VARIANT(bind_timewait)15 FIXTURE_VARIANT(bind_timewait)
16 {
17 __u32 addr_const;
18 };
19
FIXTURE_VARIANT_ADD(bind_timewait,localhost)20 FIXTURE_VARIANT_ADD(bind_timewait, localhost)
21 {
22 .addr_const = INADDR_LOOPBACK
23 };
24
FIXTURE_VARIANT_ADD(bind_timewait,addrany)25 FIXTURE_VARIANT_ADD(bind_timewait, addrany)
26 {
27 .addr_const = INADDR_ANY
28 };
29
FIXTURE_SETUP(bind_timewait)30 FIXTURE_SETUP(bind_timewait)
31 {
32 self->addr.sin_family = AF_INET;
33 self->addr.sin_port = 0;
34 self->addr.sin_addr.s_addr = htonl(variant->addr_const);
35 self->addrlen = sizeof(self->addr);
36 }
37
FIXTURE_TEARDOWN(bind_timewait)38 FIXTURE_TEARDOWN(bind_timewait)
39 {
40 }
41
create_timewait_socket(struct __test_metadata * _metadata,FIXTURE_DATA (bind_timewait)* self)42 void create_timewait_socket(struct __test_metadata *_metadata,
43 FIXTURE_DATA(bind_timewait) *self)
44 {
45 int server_fd, client_fd, child_fd, ret;
46 struct sockaddr_in addr;
47 socklen_t addrlen;
48
49 server_fd = socket(AF_INET, SOCK_STREAM, 0);
50 ASSERT_GT(server_fd, 0);
51
52 ret = bind(server_fd, (struct sockaddr *)&self->addr, self->addrlen);
53 ASSERT_EQ(ret, 0);
54
55 ret = listen(server_fd, 1);
56 ASSERT_EQ(ret, 0);
57
58 ret = getsockname(server_fd, (struct sockaddr *)&self->addr, &self->addrlen);
59 ASSERT_EQ(ret, 0);
60
61 client_fd = socket(AF_INET, SOCK_STREAM, 0);
62 ASSERT_GT(client_fd, 0);
63
64 ret = connect(client_fd, (struct sockaddr *)&self->addr, self->addrlen);
65 ASSERT_EQ(ret, 0);
66
67 addrlen = sizeof(addr);
68 child_fd = accept(server_fd, (struct sockaddr *)&addr, &addrlen);
69 ASSERT_GT(child_fd, 0);
70
71 close(child_fd);
72 close(client_fd);
73 close(server_fd);
74 }
75
76 TEST_F(bind_timewait, 1)
77 {
78 int fd, ret;
79
80 create_timewait_socket(_metadata, self);
81
82 fd = socket(AF_INET, SOCK_STREAM, 0);
83 ASSERT_GT(fd, 0);
84
85 ret = bind(fd, (struct sockaddr *)&self->addr, self->addrlen);
86 ASSERT_EQ(ret, -1);
87 ASSERT_EQ(errno, EADDRINUSE);
88
89 close(fd);
90 }
91
92 TEST_HARNESS_MAIN
93