selftests/net/fin_ack_lat.c

0001 // SPDX-License-Identifier: GPL-2.0
0002
0003 #include <arpa/inet.h>
0004 #include <errno.h>
0005 #include <error.h>
0006 #include <netinet/in.h>
0007 #include <netinet/tcp.h>
0008 #include <signal.h>
0009 #include <stdio.h>
0010 #include <stdlib.h>
0011 #include <sys/socket.h>
0012 #include <sys/time.h>
0013 #include <unistd.h>
0014
0015 static int child_pid;
0016
0017 static unsigned long timediff(struct timeval s, struct timeval e)
0018 {
0019     unsigned long s_us, e_us;
0020
0021     s_us = s.tv_sec * 1000000 + s.tv_usec;
0022     e_us = e.tv_sec * 1000000 + e.tv_usec;
0023     if (s_us > e_us)
0024         return 0;
0025     return e_us - s_us;
0026 }
0027
0028 static void client(int port)
0029 {
0030     int sock = 0;
0031     struct sockaddr_in addr, laddr;
0032     socklen_t len = sizeof(laddr);
0033     struct linger sl;
0034     int flag = 1;
0035     int buffer;
0036     struct timeval start, end;
0037     unsigned long lat, sum_lat = 0, nr_lat = 0;
0038
0039     while (1) {
0040         gettimeofday(&start, NULL);
0041
0042         sock = socket(AF_INET, SOCK_STREAM, 0);
0043         if (sock < 0)
0044             error(-1, errno, "socket creation");
0045
0046         sl.l_onoff = 1;
0047         sl.l_linger = 0;
0048         if (setsockopt(sock, SOL_SOCKET, SO_LINGER, &sl, sizeof(sl)))
0049             error(-1, errno, "setsockopt(linger)");
0050
0051         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY,
0052                     &flag, sizeof(flag)))
0053             error(-1, errno, "setsockopt(nodelay)");
0054
0055         addr.sin_family = AF_INET;
0056         addr.sin_port = htons(port);
0057
0058         if (inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr) <= 0)
0059             error(-1, errno, "inet_pton");
0060
0061         if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0)
0062             error(-1, errno, "connect");
0063
0064         send(sock, &buffer, sizeof(buffer), 0);
0065         if (read(sock, &buffer, sizeof(buffer)) == -1)
0066             error(-1, errno, "waiting read");
0067
0068         gettimeofday(&end, NULL);
0069         lat = timediff(start, end);
0070         sum_lat += lat;
0071         nr_lat++;
0072         if (lat < 100000)
0073             goto close;
0074
0075         if (getsockname(sock, (struct sockaddr *)&laddr, &len) == -1)
0076             error(-1, errno, "getsockname");
0077         printf("port: %d, lat: %lu, avg: %lu, nr: %lu\n",
0078                 ntohs(laddr.sin_port), lat,
0079                 sum_lat / nr_lat, nr_lat);
0080 close:
0081         fflush(stdout);
0082         close(sock);
0083     }
0084 }
0085
0086 static void server(int sock, struct sockaddr_in address)
0087 {
0088     int accepted;
0089     int addrlen = sizeof(address);
0090     int buffer;
0091
0092     while (1) {
0093         accepted = accept(sock, (struct sockaddr *)&address,
0094                 (socklen_t *)&addrlen);
0095         if (accepted < 0)
0096             error(-1, errno, "accept");
0097
0098         if (read(accepted, &buffer, sizeof(buffer)) == -1)
0099             error(-1, errno, "read");
0100         close(accepted);
0101     }
0102 }
0103
0104 static void sig_handler(int signum)
0105 {
0106     kill(SIGTERM, child_pid);
0107     exit(0);
0108 }
0109
0110 int main(int argc, char const *argv[])
0111 {
0112     int sock;
0113     int opt = 1;
0114     struct sockaddr_in address;
0115     struct sockaddr_in laddr;
0116     socklen_t len = sizeof(laddr);
0117
0118     if (signal(SIGTERM, sig_handler) == SIG_ERR)
0119         error(-1, errno, "signal");
0120
0121     sock = socket(AF_INET, SOCK_STREAM, 0);
0122     if (sock < 0)
0123         error(-1, errno, "socket");
0124
0125     if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT,
0126                 &opt, sizeof(opt)) == -1)
0127         error(-1, errno, "setsockopt");
0128
0129     address.sin_family = AF_INET;
0130     address.sin_addr.s_addr = INADDR_ANY;
0131     /* dynamically allocate unused port */
0132     address.sin_port = 0;
0133
0134     if (bind(sock, (struct sockaddr *)&address, sizeof(address)) < 0)
0135         error(-1, errno, "bind");
0136
0137     if (listen(sock, 3) < 0)
0138         error(-1, errno, "listen");
0139
0140     if (getsockname(sock, (struct sockaddr *)&laddr, &len) == -1)
0141         error(-1, errno, "getsockname");
0142
0143     fprintf(stderr, "server port: %d\n", ntohs(laddr.sin_port));
0144     child_pid = fork();
0145     if (!child_pid)
0146         client(ntohs(laddr.sin_port));
0147     else
0148         server(sock, laddr);
0149
0150     return 0;
0151 }