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 | |
9 | FIXTURE(bind_timewait) |
10 | { |
11 | struct sockaddr_in addr; |
12 | socklen_t addrlen; |
13 | }; |
14 | |
15 | FIXTURE_VARIANT(bind_timewait) |
16 | { |
17 | __u32 addr_const; |
18 | }; |
19 | |
20 | FIXTURE_VARIANT_ADD(bind_timewait, localhost) |
21 | { |
22 | .addr_const = INADDR_LOOPBACK |
23 | }; |
24 | |
25 | FIXTURE_VARIANT_ADD(bind_timewait, addrany) |
26 | { |
27 | .addr_const = INADDR_ANY |
28 | }; |
29 | |
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 | |
38 | FIXTURE_TEARDOWN(bind_timewait) |
39 | { |
40 | } |
41 | |
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 | |