1048d19d4SFlorian Westphal // SPDX-License-Identifier: GPL-2.0 2048d19d4SFlorian Westphal 3048d19d4SFlorian Westphal #define _GNU_SOURCE 4048d19d4SFlorian Westphal 5048d19d4SFlorian Westphal #include <errno.h> 6048d19d4SFlorian Westphal #include <limits.h> 7048d19d4SFlorian Westphal #include <fcntl.h> 8048d19d4SFlorian Westphal #include <string.h> 9048d19d4SFlorian Westphal #include <stdbool.h> 10048d19d4SFlorian Westphal #include <stdint.h> 11048d19d4SFlorian Westphal #include <stdio.h> 12048d19d4SFlorian Westphal #include <stdlib.h> 13048d19d4SFlorian Westphal #include <strings.h> 14048d19d4SFlorian Westphal #include <unistd.h> 15048d19d4SFlorian Westphal 16048d19d4SFlorian Westphal #include <sys/poll.h> 17048d19d4SFlorian Westphal #include <sys/sendfile.h> 18048d19d4SFlorian Westphal #include <sys/stat.h> 19048d19d4SFlorian Westphal #include <sys/socket.h> 20048d19d4SFlorian Westphal #include <sys/types.h> 21048d19d4SFlorian Westphal #include <sys/mman.h> 22048d19d4SFlorian Westphal 23048d19d4SFlorian Westphal #include <netdb.h> 24048d19d4SFlorian Westphal #include <netinet/in.h> 25048d19d4SFlorian Westphal 26048d19d4SFlorian Westphal #include <linux/tcp.h> 27048d19d4SFlorian Westphal 28048d19d4SFlorian Westphal extern int optind; 29048d19d4SFlorian Westphal 30048d19d4SFlorian Westphal #ifndef IPPROTO_MPTCP 31048d19d4SFlorian Westphal #define IPPROTO_MPTCP 262 32048d19d4SFlorian Westphal #endif 33048d19d4SFlorian Westphal #ifndef TCP_ULP 34048d19d4SFlorian Westphal #define TCP_ULP 31 35048d19d4SFlorian Westphal #endif 36048d19d4SFlorian Westphal 378a4b910dSFlorian Westphal static int poll_timeout = 10 * 1000; 38048d19d4SFlorian Westphal static bool listen_mode; 39048d19d4SFlorian Westphal 40048d19d4SFlorian Westphal enum cfg_mode { 41048d19d4SFlorian Westphal CFG_MODE_POLL, 42048d19d4SFlorian Westphal CFG_MODE_MMAP, 43048d19d4SFlorian Westphal CFG_MODE_SENDFILE, 44048d19d4SFlorian Westphal }; 45048d19d4SFlorian Westphal 46048d19d4SFlorian Westphal static enum cfg_mode cfg_mode = CFG_MODE_POLL; 47048d19d4SFlorian Westphal static const char *cfg_host; 48048d19d4SFlorian Westphal static const char *cfg_port = "12000"; 49048d19d4SFlorian Westphal static int cfg_sock_proto = IPPROTO_MPTCP; 50048d19d4SFlorian Westphal static bool tcpulp_audit; 51048d19d4SFlorian Westphal static int pf = AF_INET; 52048d19d4SFlorian Westphal static int cfg_sndbuf; 538a4b910dSFlorian Westphal static int cfg_rcvbuf; 54*b08fbf24SPaolo Abeni static bool cfg_join; 55048d19d4SFlorian Westphal 56048d19d4SFlorian Westphal static void die_usage(void) 57048d19d4SFlorian Westphal { 588a4b910dSFlorian Westphal fprintf(stderr, "Usage: mptcp_connect [-6] [-u] [-s MPTCP|TCP] [-p port] [-m mode]" 598a4b910dSFlorian Westphal "[-l] connect_address\n"); 608a4b910dSFlorian Westphal fprintf(stderr, "\t-6 use ipv6\n"); 618a4b910dSFlorian Westphal fprintf(stderr, "\t-t num -- set poll timeout to num\n"); 628a4b910dSFlorian Westphal fprintf(stderr, "\t-S num -- set SO_SNDBUF to num\n"); 638a4b910dSFlorian Westphal fprintf(stderr, "\t-R num -- set SO_RCVBUF to num\n"); 648a4b910dSFlorian Westphal fprintf(stderr, "\t-p num -- use port num\n"); 658a4b910dSFlorian Westphal fprintf(stderr, "\t-m [MPTCP|TCP] -- use tcp or mptcp sockets\n"); 668a4b910dSFlorian Westphal fprintf(stderr, "\t-s [mmap|poll] -- use poll (default) or mmap\n"); 678a4b910dSFlorian Westphal fprintf(stderr, "\t-u -- check mptcp ulp\n"); 68048d19d4SFlorian Westphal exit(1); 69048d19d4SFlorian Westphal } 70048d19d4SFlorian Westphal 71048d19d4SFlorian Westphal static const char *getxinfo_strerr(int err) 72048d19d4SFlorian Westphal { 73048d19d4SFlorian Westphal if (err == EAI_SYSTEM) 74048d19d4SFlorian Westphal return strerror(errno); 75048d19d4SFlorian Westphal 76048d19d4SFlorian Westphal return gai_strerror(err); 77048d19d4SFlorian Westphal } 78048d19d4SFlorian Westphal 79048d19d4SFlorian Westphal static void xgetnameinfo(const struct sockaddr *addr, socklen_t addrlen, 80048d19d4SFlorian Westphal char *host, socklen_t hostlen, 81048d19d4SFlorian Westphal char *serv, socklen_t servlen) 82048d19d4SFlorian Westphal { 83048d19d4SFlorian Westphal int flags = NI_NUMERICHOST | NI_NUMERICSERV; 84048d19d4SFlorian Westphal int err = getnameinfo(addr, addrlen, host, hostlen, serv, servlen, 85048d19d4SFlorian Westphal flags); 86048d19d4SFlorian Westphal 87048d19d4SFlorian Westphal if (err) { 88048d19d4SFlorian Westphal const char *errstr = getxinfo_strerr(err); 89048d19d4SFlorian Westphal 90048d19d4SFlorian Westphal fprintf(stderr, "Fatal: getnameinfo: %s\n", errstr); 91048d19d4SFlorian Westphal exit(1); 92048d19d4SFlorian Westphal } 93048d19d4SFlorian Westphal } 94048d19d4SFlorian Westphal 95048d19d4SFlorian Westphal static void xgetaddrinfo(const char *node, const char *service, 96048d19d4SFlorian Westphal const struct addrinfo *hints, 97048d19d4SFlorian Westphal struct addrinfo **res) 98048d19d4SFlorian Westphal { 99048d19d4SFlorian Westphal int err = getaddrinfo(node, service, hints, res); 100048d19d4SFlorian Westphal 101048d19d4SFlorian Westphal if (err) { 102048d19d4SFlorian Westphal const char *errstr = getxinfo_strerr(err); 103048d19d4SFlorian Westphal 104048d19d4SFlorian Westphal fprintf(stderr, "Fatal: getaddrinfo(%s:%s): %s\n", 105048d19d4SFlorian Westphal node ? node : "", service ? service : "", errstr); 106048d19d4SFlorian Westphal exit(1); 107048d19d4SFlorian Westphal } 108048d19d4SFlorian Westphal } 109048d19d4SFlorian Westphal 1108a4b910dSFlorian Westphal static void set_rcvbuf(int fd, unsigned int size) 1118a4b910dSFlorian Westphal { 1128a4b910dSFlorian Westphal int err; 1138a4b910dSFlorian Westphal 1148a4b910dSFlorian Westphal err = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &size, sizeof(size)); 1158a4b910dSFlorian Westphal if (err) { 1168a4b910dSFlorian Westphal perror("set SO_RCVBUF"); 1178a4b910dSFlorian Westphal exit(1); 1188a4b910dSFlorian Westphal } 1198a4b910dSFlorian Westphal } 1208a4b910dSFlorian Westphal 121048d19d4SFlorian Westphal static void set_sndbuf(int fd, unsigned int size) 122048d19d4SFlorian Westphal { 123048d19d4SFlorian Westphal int err; 124048d19d4SFlorian Westphal 125048d19d4SFlorian Westphal err = setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &size, sizeof(size)); 126048d19d4SFlorian Westphal if (err) { 127048d19d4SFlorian Westphal perror("set SO_SNDBUF"); 128048d19d4SFlorian Westphal exit(1); 129048d19d4SFlorian Westphal } 130048d19d4SFlorian Westphal } 131048d19d4SFlorian Westphal 132048d19d4SFlorian Westphal static int sock_listen_mptcp(const char * const listenaddr, 133048d19d4SFlorian Westphal const char * const port) 134048d19d4SFlorian Westphal { 135048d19d4SFlorian Westphal int sock; 136048d19d4SFlorian Westphal struct addrinfo hints = { 137048d19d4SFlorian Westphal .ai_protocol = IPPROTO_TCP, 138048d19d4SFlorian Westphal .ai_socktype = SOCK_STREAM, 139048d19d4SFlorian Westphal .ai_flags = AI_PASSIVE | AI_NUMERICHOST 140048d19d4SFlorian Westphal }; 141048d19d4SFlorian Westphal 142048d19d4SFlorian Westphal hints.ai_family = pf; 143048d19d4SFlorian Westphal 144048d19d4SFlorian Westphal struct addrinfo *a, *addr; 145048d19d4SFlorian Westphal int one = 1; 146048d19d4SFlorian Westphal 147048d19d4SFlorian Westphal xgetaddrinfo(listenaddr, port, &hints, &addr); 148048d19d4SFlorian Westphal hints.ai_family = pf; 149048d19d4SFlorian Westphal 150048d19d4SFlorian Westphal for (a = addr; a; a = a->ai_next) { 151048d19d4SFlorian Westphal sock = socket(a->ai_family, a->ai_socktype, cfg_sock_proto); 152048d19d4SFlorian Westphal if (sock < 0) 153048d19d4SFlorian Westphal continue; 154048d19d4SFlorian Westphal 155048d19d4SFlorian Westphal if (-1 == setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &one, 156048d19d4SFlorian Westphal sizeof(one))) 157048d19d4SFlorian Westphal perror("setsockopt"); 158048d19d4SFlorian Westphal 159048d19d4SFlorian Westphal if (bind(sock, a->ai_addr, a->ai_addrlen) == 0) 160048d19d4SFlorian Westphal break; /* success */ 161048d19d4SFlorian Westphal 162048d19d4SFlorian Westphal perror("bind"); 163048d19d4SFlorian Westphal close(sock); 164048d19d4SFlorian Westphal sock = -1; 165048d19d4SFlorian Westphal } 166048d19d4SFlorian Westphal 167048d19d4SFlorian Westphal freeaddrinfo(addr); 168048d19d4SFlorian Westphal 169048d19d4SFlorian Westphal if (sock < 0) { 170048d19d4SFlorian Westphal fprintf(stderr, "Could not create listen socket\n"); 171048d19d4SFlorian Westphal return sock; 172048d19d4SFlorian Westphal } 173048d19d4SFlorian Westphal 174048d19d4SFlorian Westphal if (listen(sock, 20)) { 175048d19d4SFlorian Westphal perror("listen"); 176048d19d4SFlorian Westphal close(sock); 177048d19d4SFlorian Westphal return -1; 178048d19d4SFlorian Westphal } 179048d19d4SFlorian Westphal 180048d19d4SFlorian Westphal return sock; 181048d19d4SFlorian Westphal } 182048d19d4SFlorian Westphal 183048d19d4SFlorian Westphal static bool sock_test_tcpulp(const char * const remoteaddr, 184048d19d4SFlorian Westphal const char * const port) 185048d19d4SFlorian Westphal { 186048d19d4SFlorian Westphal struct addrinfo hints = { 187048d19d4SFlorian Westphal .ai_protocol = IPPROTO_TCP, 188048d19d4SFlorian Westphal .ai_socktype = SOCK_STREAM, 189048d19d4SFlorian Westphal }; 190048d19d4SFlorian Westphal struct addrinfo *a, *addr; 191048d19d4SFlorian Westphal int sock = -1, ret = 0; 192048d19d4SFlorian Westphal bool test_pass = false; 193048d19d4SFlorian Westphal 194048d19d4SFlorian Westphal hints.ai_family = AF_INET; 195048d19d4SFlorian Westphal 196048d19d4SFlorian Westphal xgetaddrinfo(remoteaddr, port, &hints, &addr); 197048d19d4SFlorian Westphal for (a = addr; a; a = a->ai_next) { 198048d19d4SFlorian Westphal sock = socket(a->ai_family, a->ai_socktype, IPPROTO_TCP); 199048d19d4SFlorian Westphal if (sock < 0) { 200048d19d4SFlorian Westphal perror("socket"); 201048d19d4SFlorian Westphal continue; 202048d19d4SFlorian Westphal } 203048d19d4SFlorian Westphal ret = setsockopt(sock, IPPROTO_TCP, TCP_ULP, "mptcp", 204048d19d4SFlorian Westphal sizeof("mptcp")); 205048d19d4SFlorian Westphal if (ret == -1 && errno == EOPNOTSUPP) 206048d19d4SFlorian Westphal test_pass = true; 207048d19d4SFlorian Westphal close(sock); 208048d19d4SFlorian Westphal 209048d19d4SFlorian Westphal if (test_pass) 210048d19d4SFlorian Westphal break; 211048d19d4SFlorian Westphal if (!ret) 212048d19d4SFlorian Westphal fprintf(stderr, 213048d19d4SFlorian Westphal "setsockopt(TCP_ULP) returned 0\n"); 214048d19d4SFlorian Westphal else 215048d19d4SFlorian Westphal perror("setsockopt(TCP_ULP)"); 216048d19d4SFlorian Westphal } 217048d19d4SFlorian Westphal return test_pass; 218048d19d4SFlorian Westphal } 219048d19d4SFlorian Westphal 220048d19d4SFlorian Westphal static int sock_connect_mptcp(const char * const remoteaddr, 221048d19d4SFlorian Westphal const char * const port, int proto) 222048d19d4SFlorian Westphal { 223048d19d4SFlorian Westphal struct addrinfo hints = { 224048d19d4SFlorian Westphal .ai_protocol = IPPROTO_TCP, 225048d19d4SFlorian Westphal .ai_socktype = SOCK_STREAM, 226048d19d4SFlorian Westphal }; 227048d19d4SFlorian Westphal struct addrinfo *a, *addr; 228048d19d4SFlorian Westphal int sock = -1; 229048d19d4SFlorian Westphal 230048d19d4SFlorian Westphal hints.ai_family = pf; 231048d19d4SFlorian Westphal 232048d19d4SFlorian Westphal xgetaddrinfo(remoteaddr, port, &hints, &addr); 233048d19d4SFlorian Westphal for (a = addr; a; a = a->ai_next) { 234048d19d4SFlorian Westphal sock = socket(a->ai_family, a->ai_socktype, proto); 235048d19d4SFlorian Westphal if (sock < 0) { 236048d19d4SFlorian Westphal perror("socket"); 237048d19d4SFlorian Westphal continue; 238048d19d4SFlorian Westphal } 239048d19d4SFlorian Westphal 240048d19d4SFlorian Westphal if (connect(sock, a->ai_addr, a->ai_addrlen) == 0) 241048d19d4SFlorian Westphal break; /* success */ 242048d19d4SFlorian Westphal 243048d19d4SFlorian Westphal perror("connect()"); 244048d19d4SFlorian Westphal close(sock); 245048d19d4SFlorian Westphal sock = -1; 246048d19d4SFlorian Westphal } 247048d19d4SFlorian Westphal 248048d19d4SFlorian Westphal freeaddrinfo(addr); 249048d19d4SFlorian Westphal return sock; 250048d19d4SFlorian Westphal } 251048d19d4SFlorian Westphal 252048d19d4SFlorian Westphal static size_t do_rnd_write(const int fd, char *buf, const size_t len) 253048d19d4SFlorian Westphal { 254*b08fbf24SPaolo Abeni static bool first = true; 255048d19d4SFlorian Westphal unsigned int do_w; 256048d19d4SFlorian Westphal ssize_t bw; 257048d19d4SFlorian Westphal 258048d19d4SFlorian Westphal do_w = rand() & 0xffff; 259048d19d4SFlorian Westphal if (do_w == 0 || do_w > len) 260048d19d4SFlorian Westphal do_w = len; 261048d19d4SFlorian Westphal 262*b08fbf24SPaolo Abeni if (cfg_join && first && do_w > 100) 263*b08fbf24SPaolo Abeni do_w = 100; 264*b08fbf24SPaolo Abeni 265048d19d4SFlorian Westphal bw = write(fd, buf, do_w); 266048d19d4SFlorian Westphal if (bw < 0) 267048d19d4SFlorian Westphal perror("write"); 268048d19d4SFlorian Westphal 269*b08fbf24SPaolo Abeni /* let the join handshake complete, before going on */ 270*b08fbf24SPaolo Abeni if (cfg_join && first) { 271*b08fbf24SPaolo Abeni usleep(200000); 272*b08fbf24SPaolo Abeni first = false; 273*b08fbf24SPaolo Abeni } 274*b08fbf24SPaolo Abeni 275048d19d4SFlorian Westphal return bw; 276048d19d4SFlorian Westphal } 277048d19d4SFlorian Westphal 278048d19d4SFlorian Westphal static size_t do_write(const int fd, char *buf, const size_t len) 279048d19d4SFlorian Westphal { 280048d19d4SFlorian Westphal size_t offset = 0; 281048d19d4SFlorian Westphal 282048d19d4SFlorian Westphal while (offset < len) { 283048d19d4SFlorian Westphal size_t written; 284048d19d4SFlorian Westphal ssize_t bw; 285048d19d4SFlorian Westphal 286048d19d4SFlorian Westphal bw = write(fd, buf + offset, len - offset); 287048d19d4SFlorian Westphal if (bw < 0) { 288048d19d4SFlorian Westphal perror("write"); 289048d19d4SFlorian Westphal return 0; 290048d19d4SFlorian Westphal } 291048d19d4SFlorian Westphal 292048d19d4SFlorian Westphal written = (size_t)bw; 293048d19d4SFlorian Westphal offset += written; 294048d19d4SFlorian Westphal } 295048d19d4SFlorian Westphal 296048d19d4SFlorian Westphal return offset; 297048d19d4SFlorian Westphal } 298048d19d4SFlorian Westphal 299048d19d4SFlorian Westphal static ssize_t do_rnd_read(const int fd, char *buf, const size_t len) 300048d19d4SFlorian Westphal { 301048d19d4SFlorian Westphal size_t cap = rand(); 302048d19d4SFlorian Westphal 303048d19d4SFlorian Westphal cap &= 0xffff; 304048d19d4SFlorian Westphal 305048d19d4SFlorian Westphal if (cap == 0) 306048d19d4SFlorian Westphal cap = 1; 307048d19d4SFlorian Westphal else if (cap > len) 308048d19d4SFlorian Westphal cap = len; 309048d19d4SFlorian Westphal 310048d19d4SFlorian Westphal return read(fd, buf, cap); 311048d19d4SFlorian Westphal } 312048d19d4SFlorian Westphal 313048d19d4SFlorian Westphal static void set_nonblock(int fd) 314048d19d4SFlorian Westphal { 315048d19d4SFlorian Westphal int flags = fcntl(fd, F_GETFL); 316048d19d4SFlorian Westphal 317048d19d4SFlorian Westphal if (flags == -1) 318048d19d4SFlorian Westphal return; 319048d19d4SFlorian Westphal 320048d19d4SFlorian Westphal fcntl(fd, F_SETFL, flags | O_NONBLOCK); 321048d19d4SFlorian Westphal } 322048d19d4SFlorian Westphal 323048d19d4SFlorian Westphal static int copyfd_io_poll(int infd, int peerfd, int outfd) 324048d19d4SFlorian Westphal { 325048d19d4SFlorian Westphal struct pollfd fds = { 326048d19d4SFlorian Westphal .fd = peerfd, 327048d19d4SFlorian Westphal .events = POLLIN | POLLOUT, 328048d19d4SFlorian Westphal }; 329048d19d4SFlorian Westphal unsigned int woff = 0, wlen = 0; 330048d19d4SFlorian Westphal char wbuf[8192]; 331048d19d4SFlorian Westphal 332048d19d4SFlorian Westphal set_nonblock(peerfd); 333048d19d4SFlorian Westphal 334048d19d4SFlorian Westphal for (;;) { 335048d19d4SFlorian Westphal char rbuf[8192]; 336048d19d4SFlorian Westphal ssize_t len; 337048d19d4SFlorian Westphal 338048d19d4SFlorian Westphal if (fds.events == 0) 339048d19d4SFlorian Westphal break; 340048d19d4SFlorian Westphal 341048d19d4SFlorian Westphal switch (poll(&fds, 1, poll_timeout)) { 342048d19d4SFlorian Westphal case -1: 343048d19d4SFlorian Westphal if (errno == EINTR) 344048d19d4SFlorian Westphal continue; 345048d19d4SFlorian Westphal perror("poll"); 346048d19d4SFlorian Westphal return 1; 347048d19d4SFlorian Westphal case 0: 348048d19d4SFlorian Westphal fprintf(stderr, "%s: poll timed out (events: " 349048d19d4SFlorian Westphal "POLLIN %u, POLLOUT %u)\n", __func__, 350048d19d4SFlorian Westphal fds.events & POLLIN, fds.events & POLLOUT); 351048d19d4SFlorian Westphal return 2; 352048d19d4SFlorian Westphal } 353048d19d4SFlorian Westphal 354048d19d4SFlorian Westphal if (fds.revents & POLLIN) { 355048d19d4SFlorian Westphal len = do_rnd_read(peerfd, rbuf, sizeof(rbuf)); 356048d19d4SFlorian Westphal if (len == 0) { 357048d19d4SFlorian Westphal /* no more data to receive: 358048d19d4SFlorian Westphal * peer has closed its write side 359048d19d4SFlorian Westphal */ 360048d19d4SFlorian Westphal fds.events &= ~POLLIN; 361048d19d4SFlorian Westphal 362048d19d4SFlorian Westphal if ((fds.events & POLLOUT) == 0) 363048d19d4SFlorian Westphal /* and nothing more to send */ 364048d19d4SFlorian Westphal break; 365048d19d4SFlorian Westphal 366048d19d4SFlorian Westphal /* Else, still have data to transmit */ 367048d19d4SFlorian Westphal } else if (len < 0) { 368048d19d4SFlorian Westphal perror("read"); 369048d19d4SFlorian Westphal return 3; 370048d19d4SFlorian Westphal } 371048d19d4SFlorian Westphal 372048d19d4SFlorian Westphal do_write(outfd, rbuf, len); 373048d19d4SFlorian Westphal } 374048d19d4SFlorian Westphal 375048d19d4SFlorian Westphal if (fds.revents & POLLOUT) { 376048d19d4SFlorian Westphal if (wlen == 0) { 377048d19d4SFlorian Westphal woff = 0; 378048d19d4SFlorian Westphal wlen = read(infd, wbuf, sizeof(wbuf)); 379048d19d4SFlorian Westphal } 380048d19d4SFlorian Westphal 381048d19d4SFlorian Westphal if (wlen > 0) { 382048d19d4SFlorian Westphal ssize_t bw; 383048d19d4SFlorian Westphal 384048d19d4SFlorian Westphal bw = do_rnd_write(peerfd, wbuf + woff, wlen); 385048d19d4SFlorian Westphal if (bw < 0) 386048d19d4SFlorian Westphal return 111; 387048d19d4SFlorian Westphal 388048d19d4SFlorian Westphal woff += bw; 389048d19d4SFlorian Westphal wlen -= bw; 390048d19d4SFlorian Westphal } else if (wlen == 0) { 391048d19d4SFlorian Westphal /* We have no more data to send. */ 392048d19d4SFlorian Westphal fds.events &= ~POLLOUT; 393048d19d4SFlorian Westphal 394048d19d4SFlorian Westphal if ((fds.events & POLLIN) == 0) 395048d19d4SFlorian Westphal /* ... and peer also closed already */ 396048d19d4SFlorian Westphal break; 397048d19d4SFlorian Westphal 398048d19d4SFlorian Westphal /* ... but we still receive. 399*b08fbf24SPaolo Abeni * Close our write side, ev. give some time 400*b08fbf24SPaolo Abeni * for address notification 401048d19d4SFlorian Westphal */ 402*b08fbf24SPaolo Abeni if (cfg_join) 403*b08fbf24SPaolo Abeni usleep(400000); 404048d19d4SFlorian Westphal shutdown(peerfd, SHUT_WR); 405048d19d4SFlorian Westphal } else { 406048d19d4SFlorian Westphal if (errno == EINTR) 407048d19d4SFlorian Westphal continue; 408048d19d4SFlorian Westphal perror("read"); 409048d19d4SFlorian Westphal return 4; 410048d19d4SFlorian Westphal } 411048d19d4SFlorian Westphal } 412048d19d4SFlorian Westphal 413048d19d4SFlorian Westphal if (fds.revents & (POLLERR | POLLNVAL)) { 414048d19d4SFlorian Westphal fprintf(stderr, "Unexpected revents: " 415048d19d4SFlorian Westphal "POLLERR/POLLNVAL(%x)\n", fds.revents); 416048d19d4SFlorian Westphal return 5; 417048d19d4SFlorian Westphal } 418048d19d4SFlorian Westphal } 419048d19d4SFlorian Westphal 420*b08fbf24SPaolo Abeni /* leave some time for late join/announce */ 421*b08fbf24SPaolo Abeni if (cfg_join) 422*b08fbf24SPaolo Abeni usleep(400000); 423*b08fbf24SPaolo Abeni 424048d19d4SFlorian Westphal close(peerfd); 425048d19d4SFlorian Westphal return 0; 426048d19d4SFlorian Westphal } 427048d19d4SFlorian Westphal 428048d19d4SFlorian Westphal static int do_recvfile(int infd, int outfd) 429048d19d4SFlorian Westphal { 430048d19d4SFlorian Westphal ssize_t r; 431048d19d4SFlorian Westphal 432048d19d4SFlorian Westphal do { 433048d19d4SFlorian Westphal char buf[16384]; 434048d19d4SFlorian Westphal 435048d19d4SFlorian Westphal r = do_rnd_read(infd, buf, sizeof(buf)); 436048d19d4SFlorian Westphal if (r > 0) { 437048d19d4SFlorian Westphal if (write(outfd, buf, r) != r) 438048d19d4SFlorian Westphal break; 439048d19d4SFlorian Westphal } else if (r < 0) { 440048d19d4SFlorian Westphal perror("read"); 441048d19d4SFlorian Westphal } 442048d19d4SFlorian Westphal } while (r > 0); 443048d19d4SFlorian Westphal 444048d19d4SFlorian Westphal return (int)r; 445048d19d4SFlorian Westphal } 446048d19d4SFlorian Westphal 447048d19d4SFlorian Westphal static int do_mmap(int infd, int outfd, unsigned int size) 448048d19d4SFlorian Westphal { 449048d19d4SFlorian Westphal char *inbuf = mmap(NULL, size, PROT_READ, MAP_SHARED, infd, 0); 450048d19d4SFlorian Westphal ssize_t ret = 0, off = 0; 451048d19d4SFlorian Westphal size_t rem; 452048d19d4SFlorian Westphal 453048d19d4SFlorian Westphal if (inbuf == MAP_FAILED) { 454048d19d4SFlorian Westphal perror("mmap"); 455048d19d4SFlorian Westphal return 1; 456048d19d4SFlorian Westphal } 457048d19d4SFlorian Westphal 458048d19d4SFlorian Westphal rem = size; 459048d19d4SFlorian Westphal 460048d19d4SFlorian Westphal while (rem > 0) { 461048d19d4SFlorian Westphal ret = write(outfd, inbuf + off, rem); 462048d19d4SFlorian Westphal 463048d19d4SFlorian Westphal if (ret < 0) { 464048d19d4SFlorian Westphal perror("write"); 465048d19d4SFlorian Westphal break; 466048d19d4SFlorian Westphal } 467048d19d4SFlorian Westphal 468048d19d4SFlorian Westphal off += ret; 469048d19d4SFlorian Westphal rem -= ret; 470048d19d4SFlorian Westphal } 471048d19d4SFlorian Westphal 472048d19d4SFlorian Westphal munmap(inbuf, size); 473048d19d4SFlorian Westphal return rem; 474048d19d4SFlorian Westphal } 475048d19d4SFlorian Westphal 476048d19d4SFlorian Westphal static int get_infd_size(int fd) 477048d19d4SFlorian Westphal { 478048d19d4SFlorian Westphal struct stat sb; 479048d19d4SFlorian Westphal ssize_t count; 480048d19d4SFlorian Westphal int err; 481048d19d4SFlorian Westphal 482048d19d4SFlorian Westphal err = fstat(fd, &sb); 483048d19d4SFlorian Westphal if (err < 0) { 484048d19d4SFlorian Westphal perror("fstat"); 485048d19d4SFlorian Westphal return -1; 486048d19d4SFlorian Westphal } 487048d19d4SFlorian Westphal 488048d19d4SFlorian Westphal if ((sb.st_mode & S_IFMT) != S_IFREG) { 489048d19d4SFlorian Westphal fprintf(stderr, "%s: stdin is not a regular file\n", __func__); 490048d19d4SFlorian Westphal return -2; 491048d19d4SFlorian Westphal } 492048d19d4SFlorian Westphal 493048d19d4SFlorian Westphal count = sb.st_size; 494048d19d4SFlorian Westphal if (count > INT_MAX) { 495048d19d4SFlorian Westphal fprintf(stderr, "File too large: %zu\n", count); 496048d19d4SFlorian Westphal return -3; 497048d19d4SFlorian Westphal } 498048d19d4SFlorian Westphal 499048d19d4SFlorian Westphal return (int)count; 500048d19d4SFlorian Westphal } 501048d19d4SFlorian Westphal 502048d19d4SFlorian Westphal static int do_sendfile(int infd, int outfd, unsigned int count) 503048d19d4SFlorian Westphal { 504048d19d4SFlorian Westphal while (count > 0) { 505048d19d4SFlorian Westphal ssize_t r; 506048d19d4SFlorian Westphal 507048d19d4SFlorian Westphal r = sendfile(outfd, infd, NULL, count); 508048d19d4SFlorian Westphal if (r < 0) { 509048d19d4SFlorian Westphal perror("sendfile"); 510048d19d4SFlorian Westphal return 3; 511048d19d4SFlorian Westphal } 512048d19d4SFlorian Westphal 513048d19d4SFlorian Westphal count -= r; 514048d19d4SFlorian Westphal } 515048d19d4SFlorian Westphal 516048d19d4SFlorian Westphal return 0; 517048d19d4SFlorian Westphal } 518048d19d4SFlorian Westphal 519048d19d4SFlorian Westphal static int copyfd_io_mmap(int infd, int peerfd, int outfd, 520048d19d4SFlorian Westphal unsigned int size) 521048d19d4SFlorian Westphal { 522048d19d4SFlorian Westphal int err; 523048d19d4SFlorian Westphal 524048d19d4SFlorian Westphal if (listen_mode) { 525048d19d4SFlorian Westphal err = do_recvfile(peerfd, outfd); 526048d19d4SFlorian Westphal if (err) 527048d19d4SFlorian Westphal return err; 528048d19d4SFlorian Westphal 529048d19d4SFlorian Westphal err = do_mmap(infd, peerfd, size); 530048d19d4SFlorian Westphal } else { 531048d19d4SFlorian Westphal err = do_mmap(infd, peerfd, size); 532048d19d4SFlorian Westphal if (err) 533048d19d4SFlorian Westphal return err; 534048d19d4SFlorian Westphal 535048d19d4SFlorian Westphal shutdown(peerfd, SHUT_WR); 536048d19d4SFlorian Westphal 537048d19d4SFlorian Westphal err = do_recvfile(peerfd, outfd); 538048d19d4SFlorian Westphal } 539048d19d4SFlorian Westphal 540048d19d4SFlorian Westphal return err; 541048d19d4SFlorian Westphal } 542048d19d4SFlorian Westphal 543048d19d4SFlorian Westphal static int copyfd_io_sendfile(int infd, int peerfd, int outfd, 544048d19d4SFlorian Westphal unsigned int size) 545048d19d4SFlorian Westphal { 546048d19d4SFlorian Westphal int err; 547048d19d4SFlorian Westphal 548048d19d4SFlorian Westphal if (listen_mode) { 549048d19d4SFlorian Westphal err = do_recvfile(peerfd, outfd); 550048d19d4SFlorian Westphal if (err) 551048d19d4SFlorian Westphal return err; 552048d19d4SFlorian Westphal 553048d19d4SFlorian Westphal err = do_sendfile(infd, peerfd, size); 554048d19d4SFlorian Westphal } else { 555048d19d4SFlorian Westphal err = do_sendfile(infd, peerfd, size); 556048d19d4SFlorian Westphal if (err) 557048d19d4SFlorian Westphal return err; 558048d19d4SFlorian Westphal err = do_recvfile(peerfd, outfd); 559048d19d4SFlorian Westphal } 560048d19d4SFlorian Westphal 561048d19d4SFlorian Westphal return err; 562048d19d4SFlorian Westphal } 563048d19d4SFlorian Westphal 564048d19d4SFlorian Westphal static int copyfd_io(int infd, int peerfd, int outfd) 565048d19d4SFlorian Westphal { 566048d19d4SFlorian Westphal int file_size; 567048d19d4SFlorian Westphal 568048d19d4SFlorian Westphal switch (cfg_mode) { 569048d19d4SFlorian Westphal case CFG_MODE_POLL: 570048d19d4SFlorian Westphal return copyfd_io_poll(infd, peerfd, outfd); 571048d19d4SFlorian Westphal case CFG_MODE_MMAP: 572048d19d4SFlorian Westphal file_size = get_infd_size(infd); 573048d19d4SFlorian Westphal if (file_size < 0) 574048d19d4SFlorian Westphal return file_size; 575048d19d4SFlorian Westphal return copyfd_io_mmap(infd, peerfd, outfd, file_size); 576048d19d4SFlorian Westphal case CFG_MODE_SENDFILE: 577048d19d4SFlorian Westphal file_size = get_infd_size(infd); 578048d19d4SFlorian Westphal if (file_size < 0) 579048d19d4SFlorian Westphal return file_size; 580048d19d4SFlorian Westphal return copyfd_io_sendfile(infd, peerfd, outfd, file_size); 581048d19d4SFlorian Westphal } 582048d19d4SFlorian Westphal 583048d19d4SFlorian Westphal fprintf(stderr, "Invalid mode %d\n", cfg_mode); 584048d19d4SFlorian Westphal 585048d19d4SFlorian Westphal die_usage(); 586048d19d4SFlorian Westphal return 1; 587048d19d4SFlorian Westphal } 588048d19d4SFlorian Westphal 589048d19d4SFlorian Westphal static void check_sockaddr(int pf, struct sockaddr_storage *ss, 590048d19d4SFlorian Westphal socklen_t salen) 591048d19d4SFlorian Westphal { 592048d19d4SFlorian Westphal struct sockaddr_in6 *sin6; 593048d19d4SFlorian Westphal struct sockaddr_in *sin; 594048d19d4SFlorian Westphal socklen_t wanted_size = 0; 595048d19d4SFlorian Westphal 596048d19d4SFlorian Westphal switch (pf) { 597048d19d4SFlorian Westphal case AF_INET: 598048d19d4SFlorian Westphal wanted_size = sizeof(*sin); 599048d19d4SFlorian Westphal sin = (void *)ss; 600048d19d4SFlorian Westphal if (!sin->sin_port) 601048d19d4SFlorian Westphal fprintf(stderr, "accept: something wrong: ip connection from port 0"); 602048d19d4SFlorian Westphal break; 603048d19d4SFlorian Westphal case AF_INET6: 604048d19d4SFlorian Westphal wanted_size = sizeof(*sin6); 605048d19d4SFlorian Westphal sin6 = (void *)ss; 606048d19d4SFlorian Westphal if (!sin6->sin6_port) 607048d19d4SFlorian Westphal fprintf(stderr, "accept: something wrong: ipv6 connection from port 0"); 608048d19d4SFlorian Westphal break; 609048d19d4SFlorian Westphal default: 610048d19d4SFlorian Westphal fprintf(stderr, "accept: Unknown pf %d, salen %u\n", pf, salen); 611048d19d4SFlorian Westphal return; 612048d19d4SFlorian Westphal } 613048d19d4SFlorian Westphal 614048d19d4SFlorian Westphal if (salen != wanted_size) 615048d19d4SFlorian Westphal fprintf(stderr, "accept: size mismatch, got %d expected %d\n", 616048d19d4SFlorian Westphal (int)salen, wanted_size); 617048d19d4SFlorian Westphal 618048d19d4SFlorian Westphal if (ss->ss_family != pf) 619048d19d4SFlorian Westphal fprintf(stderr, "accept: pf mismatch, expect %d, ss_family is %d\n", 620048d19d4SFlorian Westphal (int)ss->ss_family, pf); 621048d19d4SFlorian Westphal } 622048d19d4SFlorian Westphal 623048d19d4SFlorian Westphal static void check_getpeername(int fd, struct sockaddr_storage *ss, socklen_t salen) 624048d19d4SFlorian Westphal { 625048d19d4SFlorian Westphal struct sockaddr_storage peerss; 626048d19d4SFlorian Westphal socklen_t peersalen = sizeof(peerss); 627048d19d4SFlorian Westphal 628048d19d4SFlorian Westphal if (getpeername(fd, (struct sockaddr *)&peerss, &peersalen) < 0) { 629048d19d4SFlorian Westphal perror("getpeername"); 630048d19d4SFlorian Westphal return; 631048d19d4SFlorian Westphal } 632048d19d4SFlorian Westphal 633048d19d4SFlorian Westphal if (peersalen != salen) { 634048d19d4SFlorian Westphal fprintf(stderr, "%s: %d vs %d\n", __func__, peersalen, salen); 635048d19d4SFlorian Westphal return; 636048d19d4SFlorian Westphal } 637048d19d4SFlorian Westphal 638048d19d4SFlorian Westphal if (memcmp(ss, &peerss, peersalen)) { 639048d19d4SFlorian Westphal char a[INET6_ADDRSTRLEN]; 640048d19d4SFlorian Westphal char b[INET6_ADDRSTRLEN]; 641048d19d4SFlorian Westphal char c[INET6_ADDRSTRLEN]; 642048d19d4SFlorian Westphal char d[INET6_ADDRSTRLEN]; 643048d19d4SFlorian Westphal 644048d19d4SFlorian Westphal xgetnameinfo((struct sockaddr *)ss, salen, 645048d19d4SFlorian Westphal a, sizeof(a), b, sizeof(b)); 646048d19d4SFlorian Westphal 647048d19d4SFlorian Westphal xgetnameinfo((struct sockaddr *)&peerss, peersalen, 648048d19d4SFlorian Westphal c, sizeof(c), d, sizeof(d)); 649048d19d4SFlorian Westphal 650048d19d4SFlorian Westphal fprintf(stderr, "%s: memcmp failure: accept %s vs peername %s, %s vs %s salen %d vs %d\n", 651048d19d4SFlorian Westphal __func__, a, c, b, d, peersalen, salen); 652048d19d4SFlorian Westphal } 653048d19d4SFlorian Westphal } 654048d19d4SFlorian Westphal 655048d19d4SFlorian Westphal static void check_getpeername_connect(int fd) 656048d19d4SFlorian Westphal { 657048d19d4SFlorian Westphal struct sockaddr_storage ss; 658048d19d4SFlorian Westphal socklen_t salen = sizeof(ss); 659048d19d4SFlorian Westphal char a[INET6_ADDRSTRLEN]; 660048d19d4SFlorian Westphal char b[INET6_ADDRSTRLEN]; 661048d19d4SFlorian Westphal 662048d19d4SFlorian Westphal if (getpeername(fd, (struct sockaddr *)&ss, &salen) < 0) { 663048d19d4SFlorian Westphal perror("getpeername"); 664048d19d4SFlorian Westphal return; 665048d19d4SFlorian Westphal } 666048d19d4SFlorian Westphal 667048d19d4SFlorian Westphal xgetnameinfo((struct sockaddr *)&ss, salen, 668048d19d4SFlorian Westphal a, sizeof(a), b, sizeof(b)); 669048d19d4SFlorian Westphal 670048d19d4SFlorian Westphal if (strcmp(cfg_host, a) || strcmp(cfg_port, b)) 671048d19d4SFlorian Westphal fprintf(stderr, "%s: %s vs %s, %s vs %s\n", __func__, 672048d19d4SFlorian Westphal cfg_host, a, cfg_port, b); 673048d19d4SFlorian Westphal } 674048d19d4SFlorian Westphal 675b0519de8SFlorian Westphal static void maybe_close(int fd) 676b0519de8SFlorian Westphal { 677b0519de8SFlorian Westphal unsigned int r = rand(); 678b0519de8SFlorian Westphal 679*b08fbf24SPaolo Abeni if (!cfg_join && (r & 1)) 680b0519de8SFlorian Westphal close(fd); 681b0519de8SFlorian Westphal } 682b0519de8SFlorian Westphal 683048d19d4SFlorian Westphal int main_loop_s(int listensock) 684048d19d4SFlorian Westphal { 685048d19d4SFlorian Westphal struct sockaddr_storage ss; 686048d19d4SFlorian Westphal struct pollfd polls; 687048d19d4SFlorian Westphal socklen_t salen; 688048d19d4SFlorian Westphal int remotesock; 689048d19d4SFlorian Westphal 690048d19d4SFlorian Westphal polls.fd = listensock; 691048d19d4SFlorian Westphal polls.events = POLLIN; 692048d19d4SFlorian Westphal 693048d19d4SFlorian Westphal switch (poll(&polls, 1, poll_timeout)) { 694048d19d4SFlorian Westphal case -1: 695048d19d4SFlorian Westphal perror("poll"); 696048d19d4SFlorian Westphal return 1; 697048d19d4SFlorian Westphal case 0: 698048d19d4SFlorian Westphal fprintf(stderr, "%s: timed out\n", __func__); 699048d19d4SFlorian Westphal close(listensock); 700048d19d4SFlorian Westphal return 2; 701048d19d4SFlorian Westphal } 702048d19d4SFlorian Westphal 703048d19d4SFlorian Westphal salen = sizeof(ss); 704048d19d4SFlorian Westphal remotesock = accept(listensock, (struct sockaddr *)&ss, &salen); 705048d19d4SFlorian Westphal if (remotesock >= 0) { 706b0519de8SFlorian Westphal maybe_close(listensock); 707048d19d4SFlorian Westphal check_sockaddr(pf, &ss, salen); 708048d19d4SFlorian Westphal check_getpeername(remotesock, &ss, salen); 709048d19d4SFlorian Westphal 710048d19d4SFlorian Westphal return copyfd_io(0, remotesock, 1); 711048d19d4SFlorian Westphal } 712048d19d4SFlorian Westphal 713048d19d4SFlorian Westphal perror("accept"); 714048d19d4SFlorian Westphal 715048d19d4SFlorian Westphal return 1; 716048d19d4SFlorian Westphal } 717048d19d4SFlorian Westphal 718048d19d4SFlorian Westphal static void init_rng(void) 719048d19d4SFlorian Westphal { 720048d19d4SFlorian Westphal int fd = open("/dev/urandom", O_RDONLY); 721048d19d4SFlorian Westphal unsigned int foo; 722048d19d4SFlorian Westphal 723048d19d4SFlorian Westphal if (fd > 0) { 724048d19d4SFlorian Westphal int ret = read(fd, &foo, sizeof(foo)); 725048d19d4SFlorian Westphal 726048d19d4SFlorian Westphal if (ret < 0) 727048d19d4SFlorian Westphal srand(fd + foo); 728048d19d4SFlorian Westphal close(fd); 729048d19d4SFlorian Westphal } 730048d19d4SFlorian Westphal 731048d19d4SFlorian Westphal srand(foo); 732048d19d4SFlorian Westphal } 733048d19d4SFlorian Westphal 734048d19d4SFlorian Westphal int main_loop(void) 735048d19d4SFlorian Westphal { 736048d19d4SFlorian Westphal int fd; 737048d19d4SFlorian Westphal 738048d19d4SFlorian Westphal /* listener is ready. */ 739048d19d4SFlorian Westphal fd = sock_connect_mptcp(cfg_host, cfg_port, cfg_sock_proto); 740048d19d4SFlorian Westphal if (fd < 0) 741048d19d4SFlorian Westphal return 2; 742048d19d4SFlorian Westphal 743048d19d4SFlorian Westphal check_getpeername_connect(fd); 744048d19d4SFlorian Westphal 7458a4b910dSFlorian Westphal if (cfg_rcvbuf) 7468a4b910dSFlorian Westphal set_rcvbuf(fd, cfg_rcvbuf); 747048d19d4SFlorian Westphal if (cfg_sndbuf) 748048d19d4SFlorian Westphal set_sndbuf(fd, cfg_sndbuf); 749048d19d4SFlorian Westphal 750048d19d4SFlorian Westphal return copyfd_io(0, fd, 1); 751048d19d4SFlorian Westphal } 752048d19d4SFlorian Westphal 753048d19d4SFlorian Westphal int parse_proto(const char *proto) 754048d19d4SFlorian Westphal { 755048d19d4SFlorian Westphal if (!strcasecmp(proto, "MPTCP")) 756048d19d4SFlorian Westphal return IPPROTO_MPTCP; 757048d19d4SFlorian Westphal if (!strcasecmp(proto, "TCP")) 758048d19d4SFlorian Westphal return IPPROTO_TCP; 759048d19d4SFlorian Westphal 760048d19d4SFlorian Westphal fprintf(stderr, "Unknown protocol: %s\n.", proto); 761048d19d4SFlorian Westphal die_usage(); 762048d19d4SFlorian Westphal 763048d19d4SFlorian Westphal /* silence compiler warning */ 764048d19d4SFlorian Westphal return 0; 765048d19d4SFlorian Westphal } 766048d19d4SFlorian Westphal 767048d19d4SFlorian Westphal int parse_mode(const char *mode) 768048d19d4SFlorian Westphal { 769048d19d4SFlorian Westphal if (!strcasecmp(mode, "poll")) 770048d19d4SFlorian Westphal return CFG_MODE_POLL; 771048d19d4SFlorian Westphal if (!strcasecmp(mode, "mmap")) 772048d19d4SFlorian Westphal return CFG_MODE_MMAP; 773048d19d4SFlorian Westphal if (!strcasecmp(mode, "sendfile")) 774048d19d4SFlorian Westphal return CFG_MODE_SENDFILE; 775048d19d4SFlorian Westphal 776048d19d4SFlorian Westphal fprintf(stderr, "Unknown test mode: %s\n", mode); 777048d19d4SFlorian Westphal fprintf(stderr, "Supported modes are:\n"); 778048d19d4SFlorian Westphal fprintf(stderr, "\t\t\"poll\" - interleaved read/write using poll()\n"); 779048d19d4SFlorian Westphal fprintf(stderr, "\t\t\"mmap\" - send entire input file (mmap+write), then read response (-l will read input first)\n"); 780048d19d4SFlorian Westphal fprintf(stderr, "\t\t\"sendfile\" - send entire input file (sendfile), then read response (-l will read input first)\n"); 781048d19d4SFlorian Westphal 782048d19d4SFlorian Westphal die_usage(); 783048d19d4SFlorian Westphal 784048d19d4SFlorian Westphal /* silence compiler warning */ 785048d19d4SFlorian Westphal return 0; 786048d19d4SFlorian Westphal } 787048d19d4SFlorian Westphal 7888a4b910dSFlorian Westphal static int parse_int(const char *size) 789048d19d4SFlorian Westphal { 790048d19d4SFlorian Westphal unsigned long s; 791048d19d4SFlorian Westphal 792048d19d4SFlorian Westphal errno = 0; 793048d19d4SFlorian Westphal 794048d19d4SFlorian Westphal s = strtoul(size, NULL, 0); 795048d19d4SFlorian Westphal 796048d19d4SFlorian Westphal if (errno) { 797048d19d4SFlorian Westphal fprintf(stderr, "Invalid sndbuf size %s (%s)\n", 798048d19d4SFlorian Westphal size, strerror(errno)); 799048d19d4SFlorian Westphal die_usage(); 800048d19d4SFlorian Westphal } 801048d19d4SFlorian Westphal 802048d19d4SFlorian Westphal if (s > INT_MAX) { 803048d19d4SFlorian Westphal fprintf(stderr, "Invalid sndbuf size %s (%s)\n", 804048d19d4SFlorian Westphal size, strerror(ERANGE)); 805048d19d4SFlorian Westphal die_usage(); 806048d19d4SFlorian Westphal } 807048d19d4SFlorian Westphal 8088a4b910dSFlorian Westphal return (int)s; 809048d19d4SFlorian Westphal } 810048d19d4SFlorian Westphal 811048d19d4SFlorian Westphal static void parse_opts(int argc, char **argv) 812048d19d4SFlorian Westphal { 813048d19d4SFlorian Westphal int c; 814048d19d4SFlorian Westphal 815*b08fbf24SPaolo Abeni while ((c = getopt(argc, argv, "6jlp:s:hut:m:S:R:")) != -1) { 816048d19d4SFlorian Westphal switch (c) { 817*b08fbf24SPaolo Abeni case 'j': 818*b08fbf24SPaolo Abeni cfg_join = true; 819*b08fbf24SPaolo Abeni cfg_mode = CFG_MODE_POLL; 820*b08fbf24SPaolo Abeni break; 821048d19d4SFlorian Westphal case 'l': 822048d19d4SFlorian Westphal listen_mode = true; 823048d19d4SFlorian Westphal break; 824048d19d4SFlorian Westphal case 'p': 825048d19d4SFlorian Westphal cfg_port = optarg; 826048d19d4SFlorian Westphal break; 827048d19d4SFlorian Westphal case 's': 828048d19d4SFlorian Westphal cfg_sock_proto = parse_proto(optarg); 829048d19d4SFlorian Westphal break; 830048d19d4SFlorian Westphal case 'h': 831048d19d4SFlorian Westphal die_usage(); 832048d19d4SFlorian Westphal break; 833048d19d4SFlorian Westphal case 'u': 834048d19d4SFlorian Westphal tcpulp_audit = true; 835048d19d4SFlorian Westphal break; 836048d19d4SFlorian Westphal case '6': 837048d19d4SFlorian Westphal pf = AF_INET6; 838048d19d4SFlorian Westphal break; 839048d19d4SFlorian Westphal case 't': 840048d19d4SFlorian Westphal poll_timeout = atoi(optarg) * 1000; 841048d19d4SFlorian Westphal if (poll_timeout <= 0) 842048d19d4SFlorian Westphal poll_timeout = -1; 843048d19d4SFlorian Westphal break; 844048d19d4SFlorian Westphal case 'm': 845048d19d4SFlorian Westphal cfg_mode = parse_mode(optarg); 846048d19d4SFlorian Westphal break; 8478a4b910dSFlorian Westphal case 'S': 8488a4b910dSFlorian Westphal cfg_sndbuf = parse_int(optarg); 8498a4b910dSFlorian Westphal break; 8508a4b910dSFlorian Westphal case 'R': 8518a4b910dSFlorian Westphal cfg_rcvbuf = parse_int(optarg); 852048d19d4SFlorian Westphal break; 853048d19d4SFlorian Westphal } 854048d19d4SFlorian Westphal } 855048d19d4SFlorian Westphal 856048d19d4SFlorian Westphal if (optind + 1 != argc) 857048d19d4SFlorian Westphal die_usage(); 858048d19d4SFlorian Westphal cfg_host = argv[optind]; 859048d19d4SFlorian Westphal 860048d19d4SFlorian Westphal if (strchr(cfg_host, ':')) 861048d19d4SFlorian Westphal pf = AF_INET6; 862048d19d4SFlorian Westphal } 863048d19d4SFlorian Westphal 864048d19d4SFlorian Westphal int main(int argc, char *argv[]) 865048d19d4SFlorian Westphal { 866048d19d4SFlorian Westphal init_rng(); 867048d19d4SFlorian Westphal 868048d19d4SFlorian Westphal parse_opts(argc, argv); 869048d19d4SFlorian Westphal 870048d19d4SFlorian Westphal if (tcpulp_audit) 871048d19d4SFlorian Westphal return sock_test_tcpulp(cfg_host, cfg_port) ? 0 : 1; 872048d19d4SFlorian Westphal 873048d19d4SFlorian Westphal if (listen_mode) { 874048d19d4SFlorian Westphal int fd = sock_listen_mptcp(cfg_host, cfg_port); 875048d19d4SFlorian Westphal 876048d19d4SFlorian Westphal if (fd < 0) 877048d19d4SFlorian Westphal return 1; 878048d19d4SFlorian Westphal 8798a4b910dSFlorian Westphal if (cfg_rcvbuf) 8808a4b910dSFlorian Westphal set_rcvbuf(fd, cfg_rcvbuf); 881048d19d4SFlorian Westphal if (cfg_sndbuf) 882048d19d4SFlorian Westphal set_sndbuf(fd, cfg_sndbuf); 883048d19d4SFlorian Westphal 884048d19d4SFlorian Westphal return main_loop_s(fd); 885048d19d4SFlorian Westphal } 886048d19d4SFlorian Westphal 887048d19d4SFlorian Westphal return main_loop(); 888048d19d4SFlorian Westphal } 889