/**************************************************************************** * examples/nettest/nettest-server.c * * Copyright (C) 2007, 2011-2012, 2015 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 "nettest.h" /**************************************************************************** * Public Functions ****************************************************************************/ void recv_server(void) { #ifdef CONFIG_EXAMPLES_NETTEST_IPv6 struct sockaddr_in6 myaddr; #else struct sockaddr_in myaddr; #endif #ifdef NETTEST_HAVE_SOLINGER struct linger ling; #endif char *buffer; int listensd; int acceptsd; socklen_t addrlen; int nbytesread; #ifndef CONFIG_EXAMPLES_NETTEST_PERFORMANCE int totalbytesread; int nbytessent; int ch; int i; #endif int optval; /* Allocate a BIG buffer */ buffer = (char*)malloc(2*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, (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_NETTEST_IPv6 myaddr.sin6_family = AF_INET6; myaddr.sin6_port = HTONS(PORTNO); #if defined(CONFIG_EXAMPLES_NETTEST_LOOPBACK) && !defined(NET_LOOPBACK) myaddr.sin6_addr.s6_addr16[0] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_1); myaddr.sin6_addr.s6_addr16[1] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_2); myaddr.sin6_addr.s6_addr16[2] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_3); myaddr.sin6_addr.s6_addr16[3] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_4); myaddr.sin6_addr.s6_addr16[4] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_5); myaddr.sin6_addr.s6_addr16[5] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_6); myaddr.sin6_addr.s6_addr16[6] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_7); myaddr.sin6_addr.s6_addr16[7] = HTONS(CONFIG_EXAMPLES_NETTEST_CLIENTIPv6ADDR_8); #else memset(&myaddr.sin6_addr, 0, sizeof(struct in6_addr)); #endif addrlen = sizeof(struct sockaddr_in6); #else myaddr.sin_family = AF_INET; myaddr.sin_port = HTONS(PORTNO); #if defined(CONFIG_EXAMPLES_NETTEST_LOOPBACK) && !defined(NET_LOOPBACK) myaddr.sin_addr.s_addr = HTONL(CONFIG_EXAMPLES_NETTEST_CLIENTIP); #else myaddr.sin_addr.s_addr = INADDR_ANY; #endif addrlen = sizeof(struct sockaddr_in); #endif if (bind(listensd, (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", PORTNO); acceptsd = accept(listensd, (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 NETTEST_HAVE_SOLINGER ling.l_onoff = 1; ling.l_linger = 30; /* timeout is seconds */ if (setsockopt(acceptsd, SOL_SOCKET, SO_LINGER, &ling, sizeof(struct linger)) < 0) { printf("server: setsockopt SO_LINGER failure: %d\n", errno); goto errout_with_acceptsd; } #endif #ifdef CONFIG_EXAMPLES_NETTEST_PERFORMANCE /* Then receive data forever */ for (;;) { nbytesread = recv(acceptsd, buffer, 2*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; } printf("Received %d bytes\n", nbytesread); } #else /* Receive canned message */ totalbytesread = 0; while (totalbytesread < SENDSIZE) { printf("server: Reading...\n"); nbytesread = recv(acceptsd, &buffer[totalbytesread], 2*SENDSIZE - totalbytesread, 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; } totalbytesread += nbytesread; printf("server: Received %d of %d bytes\n", totalbytesread, SENDSIZE); } /* Verify the message */ if (totalbytesread != SENDSIZE) { printf("server: Received %d / Expected %d bytes\n", totalbytesread, SENDSIZE); goto errout_with_acceptsd; } ch = 0x20; for (i = 0; i < SENDSIZE; i++ ) { if (buffer[i] != ch) { printf("server: Byte %d is %02x / Expected %02x\n", i, buffer[i], ch); goto errout_with_acceptsd; } if (++ch > 0x7e) { ch = 0x20; } } /* Then send the same data back to the client */ printf("server: Sending %d bytes\n", totalbytesread); nbytessent = send(acceptsd, buffer, totalbytesread, 0); if (nbytessent <= 0) { printf("server: send failed: %d\n", errno); goto errout_with_acceptsd; } printf("server: Sent %d bytes\n", nbytessent); /* If this platform only does abortive disconnects, then wait a bit to get the * client side a change to receive the data. */ #if 1 /* Do it for all platforms */ printf("server: Wait before closing\n"); sleep(2); #endif printf("server: Terminating\n"); close(listensd); close(acceptsd); free(buffer); return; #endif errout_with_acceptsd: close(acceptsd); errout_with_listensd: close(listensd); errout_with_buffer: free(buffer); exit(1); }