LXR linux/tools/testing/selftests/net/reuseaddr

   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
  21int 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;
  81out:
  82        close(fd);
  83        return ret;
  84}
  85
  86int 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(0, 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(0, 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(1, 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(0, 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(0, 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");
 113        return 0;
 114}
 115