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