145 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * This times how long it takes to bind to a port when the port already4 * has multiple sockets in its bhash table.5 *6 * In the setup(), we populate the port's bhash table with7 * MAX_THREADS * MAX_CONNECTIONS number of entries.8 */9 10#include <unistd.h>11#include <stdio.h>12#include <netdb.h>13#include <pthread.h>14#include <string.h>15#include <stdbool.h>16 17#define MAX_THREADS 60018#define MAX_CONNECTIONS 4019 20static const char *setup_addr_v6 = "::1";21static const char *setup_addr_v4 = "127.0.0.1";22static const char *setup_addr;23static const char *bind_addr;24static const char *port;25bool use_v6;26int ret;27 28static int fd_array[MAX_THREADS][MAX_CONNECTIONS];29 30static int bind_socket(int opt, const char *addr)31{32 struct addrinfo *res, hint = {};33 int sock_fd, reuse = 1, err;34 int domain = use_v6 ? AF_INET6 : AF_INET;35 36 sock_fd = socket(domain, SOCK_STREAM, 0);37 if (sock_fd < 0) {38 perror("socket fd err");39 return sock_fd;40 }41 42 hint.ai_family = domain;43 hint.ai_socktype = SOCK_STREAM;44 45 err = getaddrinfo(addr, port, &hint, &res);46 if (err) {47 perror("getaddrinfo failed");48 goto cleanup;49 }50 51 if (opt) {52 err = setsockopt(sock_fd, SOL_SOCKET, opt, &reuse, sizeof(reuse));53 if (err) {54 perror("setsockopt failed");55 goto cleanup;56 }57 }58 59 err = bind(sock_fd, res->ai_addr, res->ai_addrlen);60 if (err) {61 perror("failed to bind to port");62 goto cleanup;63 }64 65 return sock_fd;66 67cleanup:68 close(sock_fd);69 return err;70}71 72static void *setup(void *arg)73{74 int sock_fd, i;75 int *array = (int *)arg;76 77 for (i = 0; i < MAX_CONNECTIONS; i++) {78 sock_fd = bind_socket(SO_REUSEADDR | SO_REUSEPORT, setup_addr);79 if (sock_fd < 0) {80 ret = sock_fd;81 pthread_exit(&ret);82 }83 array[i] = sock_fd;84 }85 86 return NULL;87}88 89int main(int argc, const char *argv[])90{91 int listener_fd, sock_fd, i, j;92 pthread_t tid[MAX_THREADS];93 clock_t begin, end;94 95 if (argc != 4) {96 printf("Usage: listener <port> <ipv6 | ipv4> <bind-addr>\n");97 return -1;98 }99 100 port = argv[1];101 use_v6 = strcmp(argv[2], "ipv6") == 0;102 bind_addr = argv[3];103 104 setup_addr = use_v6 ? setup_addr_v6 : setup_addr_v4;105 106 listener_fd = bind_socket(SO_REUSEADDR | SO_REUSEPORT, setup_addr);107 if (listen(listener_fd, 100) < 0) {108 perror("listen failed");109 return -1;110 }111 112 /* Set up threads to populate the bhash table entry for the port */113 for (i = 0; i < MAX_THREADS; i++)114 pthread_create(&tid[i], NULL, setup, fd_array[i]);115 116 for (i = 0; i < MAX_THREADS; i++)117 pthread_join(tid[i], NULL);118 119 if (ret)120 goto done;121 122 begin = clock();123 124 /* Bind to the same port on a different address */125 sock_fd = bind_socket(0, bind_addr);126 if (sock_fd < 0)127 goto done;128 129 end = clock();130 131 printf("time spent = %f\n", (double)(end - begin) / CLOCKS_PER_SEC);132 133 /* clean up */134 close(sock_fd);135 136done:137 close(listener_fd);138 for (i = 0; i < MAX_THREADS; i++) {139 for (j = 0; i < MAX_THREADS; i++)140 close(fd_array[i][j]);141 }142 143 return 0;144}145