/**************************************************************************** * examples/tcpblaster/tcpblaster-server.c * * Copyright (C) 2018 Gregory Nutt. All rights reserved. * Author: Gregory Nutt * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * 3. Neither the name Gregory Nutt nor the names of its contributors may be * used to endorse or promote products derived from this software * without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. * ****************************************************************************/ /**************************************************************************** * Included Files ****************************************************************************/ #include "config.h" #include #include #include #include #include #include #include #include #include #include #include #include "tcpblaster.h" /**************************************************************************** * Public Functions ****************************************************************************/ void tcpblaster_server(void) { #ifdef CONFIG_EXAMPLES_TCPBLASTER_IPv6 struct sockaddr_in6 myaddr; #else struct sockaddr_in myaddr; #endif #ifdef TCPBLASTER_HAVE_SOLINGER struct linger ling; #endif struct timespec start; unsigned long recvtotal; socklen_t addrlen; char *buffer; int groupcount; int recvcount; int listensd; int acceptsd; int nbytesread; int optval; char timebuff[100]; setbuf(stdout, NULL); /* Allocate a BIG buffer */ buffer = (FAR char *)malloc(SENDSIZE); if (!buffer) { printf("server: failed to allocate buffer\n"); exit(1); } /* Create a new TCP socket */ listensd = socket(PF_INETX, SOCK_STREAM, 0); if (listensd < 0) { printf("server: socket failure: %d\n", errno); goto errout_with_buffer; } /* Set socket to reuse address */ optval = 1; if (setsockopt(listensd, SOL_SOCKET, SO_REUSEADDR, (FAR void *)&optval, sizeof(int)) < 0) { printf("server: setsockopt SO_REUSEADDR failure: %d\n", errno); goto errout_with_listensd; } /* Bind the socket to a local address */ #ifdef CONFIG_EXAMPLES_TCPBLASTER_IPv6 myaddr.sin6_family = AF_INET6; myaddr.sin6_port = HTONS(CONFIG_EXAMPLES_TCPBLASTER_SERVER_PORTNO); #if defined(CONFIG_EXAMPLES_TCPBLASTER_LOOPBACK) && !defined(CONFIG_NET_LOOPBACK) memcpy(myaddr.sin6_addr.s6_addr, g_tcpblasterserver_ipv6, 16); #else memset(myaddr.sin6_addr.s6_addr, 0, 16); #endif addrlen = sizeof(struct sockaddr_in6); printf("Binding to IPv6 Address: " "%02x%02x:%02x%02x:%02x%02x:%02x%02x:" "%02x%02x:%02x%02x:%02x%02x:%02x%02x\n", myaddr.sin6_addr.s6_addr[0], myaddr.sin6_addr.s6_addr[1], myaddr.sin6_addr.s6_addr[2], myaddr.sin6_addr.s6_addr[3], myaddr.sin6_addr.s6_addr[4], myaddr.sin6_addr.s6_addr[5], myaddr.sin6_addr.s6_addr[6], myaddr.sin6_addr.s6_addr[7], myaddr.sin6_addr.s6_addr[8], myaddr.sin6_addr.s6_addr[9], myaddr.sin6_addr.s6_addr[10], myaddr.sin6_addr.s6_addr[11], myaddr.sin6_addr.s6_addr[12], myaddr.sin6_addr.s6_addr[13], myaddr.sin6_addr.s6_addr[14], myaddr.sin6_addr.s6_addr[15]); #else myaddr.sin_family = AF_INET; myaddr.sin_port = HTONS(CONFIG_EXAMPLES_TCPBLASTER_SERVER_PORTNO); #if defined(CONFIG_EXAMPLES_TCPBLASTER_LOOPBACK) && !defined(CONFIG_NET_LOOPBACK) myaddr.sin_addr.s_addr = (in_addr_t)g_tcpblasterserver_ipv4; #else myaddr.sin_addr.s_addr = INADDR_ANY; #endif addrlen = sizeof(struct sockaddr_in); printf("Binding to IPv4 Address: %08lx\n", (unsigned long)myaddr.sin_addr.s_addr); #endif if (bind(listensd, (FAR struct sockaddr *)&myaddr, addrlen) < 0) { printf("server: bind failure: %d\n", errno); goto errout_with_listensd; } /* Listen for connections on the bound TCP socket */ if (listen(listensd, 5) < 0) { printf("server: listen failure %d\n", errno); goto errout_with_listensd; } /* Accept only one connection */ printf("server: Accepting connections on port %d\n", CONFIG_EXAMPLES_TCPBLASTER_SERVER_PORTNO); acceptsd = accept(listensd, (FAR struct sockaddr *)&myaddr, &addrlen); if (acceptsd < 0) { printf("server: accept failure: %d\n", errno); goto errout_with_listensd; } printf("server: Connection accepted -- receiving\n"); /* Configure to "linger" until all data is sent when the socket is closed */ #ifdef TCPBLASTER_HAVE_SOLINGER ling.l_onoff = 1; ling.l_linger = 30; /* timeout is seconds */ if (setsockopt(acceptsd, SOL_SOCKET, SO_LINGER, &ling, sizeof(ling)) < 0) { printf("server: setsockopt SO_LINGER failure: %d\n", errno); goto errout_with_acceptsd; } #endif /* Then receive data forever */ recvcount = 0; recvtotal = 0; groupcount = 0; clock_gettime(CLOCK_REALTIME, &start); for (; ; ) { #ifdef CONFIG_EXAMPLES_TCPBLASTER_POLLIN struct pollfd fds[1]; int ret; memset(fds, 0, 1 * sizeof(struct pollfd)); fds[0].fd = acceptsd; fds[0].events = POLLIN | POLLHUP; /* Wait until we can receive data or until the connection is lost */ ret = poll(fds, 1, -1); if (ret < 0) { printf("server: ERROR poll failed: %d\n", errno); goto errout_with_acceptsd; } if ((fds[0].revents & POLLHUP) != 0) { printf("server: WARNING poll returned POLLHUP\n"); goto errout_with_acceptsd; } #endif nbytesread = recv(acceptsd, buffer, SENDSIZE, 0); if (nbytesread < 0) { printf("server: recv failed: %d\n", errno); goto errout_with_acceptsd; } else if (nbytesread == 0) { printf("server: The client broke the connection\n"); goto errout_with_acceptsd; } recvtotal += nbytesread; if (++recvcount >= GROUPSIZE) { struct timespec elapsed; struct timespec curr; float fkbsent; float felapsed; clock_gettime(CLOCK_REALTIME, &curr); elapsed.tv_sec = curr.tv_sec - start.tv_sec; if (curr.tv_nsec >= start.tv_nsec) { elapsed.tv_nsec = curr.tv_nsec - start.tv_nsec; } else { unsigned long borrow = 1000000000 - start.tv_nsec; elapsed.tv_sec--; elapsed.tv_nsec = curr.tv_nsec + borrow; } strftime(timebuff, 100, "%Y-%m-%d %H:%M:%S.000", localtime(&curr.tv_sec)); fkbsent = recvtotal / 1024.0f; felapsed = elapsed.tv_sec + elapsed.tv_nsec / 1000000000.0f; printf("[%s] %d: Received %d buffers: %7.1f KB (buffer average" "size: %5.1f KB) in %6.2f seconds (%7.1f KB/second)\n", timebuff, groupcount, recvcount, fkbsent, fkbsent / recvcount, felapsed, fkbsent / felapsed); recvcount = 0; recvtotal = 0; groupcount++; clock_gettime(CLOCK_REALTIME, &start); } } errout_with_acceptsd: close(acceptsd); errout_with_listensd: close(listensd); errout_with_buffer: free(buffer); exit(1); }