1 | /* |
2 | * Test for the regression introduced by |
3 | * |
4 | * b9470c27607b ("inet: kill smallest_size and smallest_port") |
5 | * |
6 | * If we open an ipv4 socket on a port with reuseaddr we shouldn't reset the tb |
7 | * when we open the ipv6 conterpart, which is what was happening previously. |
8 | */ |
9 | #include <errno.h> |
10 | #include <error.h> |
11 | #include <arpa/inet.h> |
12 | #include <netinet/in.h> |
13 | #include <stdbool.h> |
14 | #include <stdio.h> |
15 | #include <sys/socket.h> |
16 | #include <sys/types.h> |
17 | #include <unistd.h> |
18 | |
19 | #define PORT 9999 |
20 | |
21 | int open_port(int ipv6, int any) |
22 | { |
23 | int fd = -1; |
24 | int reuseaddr = 1; |
25 | int v6only = 1; |
26 | int addrlen; |
27 | int ret = -1; |
28 | struct sockaddr *addr; |
29 | int family = ipv6 ? AF_INET6 : AF_INET; |
30 | |
31 | struct sockaddr_in6 addr6 = { |
32 | .sin6_family = AF_INET6, |
33 | .sin6_port = htons(PORT), |
34 | .sin6_addr = in6addr_any |
35 | }; |
36 | struct sockaddr_in addr4 = { |
37 | .sin_family = AF_INET, |
38 | .sin_port = htons(PORT), |
39 | .sin_addr.s_addr = any ? htonl(INADDR_ANY) : inet_addr("127.0.0.1" ), |
40 | }; |
41 | |
42 | |
43 | if (ipv6) { |
44 | addr = (struct sockaddr*)&addr6; |
45 | addrlen = sizeof(addr6); |
46 | } else { |
47 | addr = (struct sockaddr*)&addr4; |
48 | addrlen = sizeof(addr4); |
49 | } |
50 | |
51 | if ((fd = socket(family, SOCK_STREAM, IPPROTO_TCP)) < 0) { |
52 | perror("socket" ); |
53 | goto out; |
54 | } |
55 | |
56 | if (ipv6 && setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, (void*)&v6only, |
57 | sizeof(v6only)) < 0) { |
58 | perror("setsockopt IPV6_V6ONLY" ); |
59 | goto out; |
60 | } |
61 | |
62 | if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, |
63 | sizeof(reuseaddr)) < 0) { |
64 | perror("setsockopt SO_REUSEADDR" ); |
65 | goto out; |
66 | } |
67 | |
68 | if (bind(fd, addr, addrlen) < 0) { |
69 | perror("bind" ); |
70 | goto out; |
71 | } |
72 | |
73 | if (any) |
74 | return fd; |
75 | |
76 | if (listen(fd, 1) < 0) { |
77 | perror("listen" ); |
78 | goto out; |
79 | } |
80 | return fd; |
81 | out: |
82 | close(fd); |
83 | return ret; |
84 | } |
85 | |
86 | int main(void) |
87 | { |
88 | int listenfd; |
89 | int fd1, fd2; |
90 | |
91 | fprintf(stderr, "Opening 127.0.0.1:%d\n" , PORT); |
92 | listenfd = open_port(ipv6: 0, any: 0); |
93 | if (listenfd < 0) |
94 | error(1, errno, "Couldn't open listen socket" ); |
95 | fprintf(stderr, "Opening INADDR_ANY:%d\n" , PORT); |
96 | fd1 = open_port(ipv6: 0, any: 1); |
97 | if (fd1 >= 0) |
98 | error(1, 0, "Was allowed to create an ipv4 reuseport on a already bound non-reuseport socket" ); |
99 | fprintf(stderr, "Opening in6addr_any:%d\n" , PORT); |
100 | fd1 = open_port(ipv6: 1, any: 1); |
101 | if (fd1 < 0) |
102 | error(1, errno, "Couldn't open ipv6 reuseport" ); |
103 | fprintf(stderr, "Opening INADDR_ANY:%d\n" , PORT); |
104 | fd2 = open_port(ipv6: 0, any: 1); |
105 | if (fd2 >= 0) |
106 | error(1, 0, "Was allowed to create an ipv4 reuseport on a already bound non-reuseport socket" ); |
107 | close(fd1); |
108 | fprintf(stderr, "Opening INADDR_ANY:%d after closing ipv6 socket\n" , PORT); |
109 | fd1 = open_port(ipv6: 0, any: 1); |
110 | if (fd1 >= 0) |
111 | error(1, 0, "Was allowed to create an ipv4 reuseport on an already bound non-reuseport socket with no ipv6" ); |
112 | fprintf(stderr, "Success\n" ); |
113 | return 0; |
114 | } |
115 | |