tools/testing/selftests/net/bind_timewait.c

   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