diff options
author | Jörg Frings-Fürst <debian@jff-webhosting.net> | 2014-10-01 08:07:52 +0200 |
---|---|---|
committer | Jörg Frings-Fürst <debian@jff-webhosting.net> | 2014-10-01 08:07:52 +0200 |
commit | 6c4f0aa3eaa0de86457dbc734ea552cd03a67a0d (patch) | |
tree | 3a5d5161046f3c97d9a18bd16285a807a20c2466 /test/necho.cpp |
Initial import of psocksxx version 0.0.5-1
Diffstat (limited to 'test/necho.cpp')
-rw-r--r-- | test/necho.cpp | 247 |
1 files changed, 247 insertions, 0 deletions
diff --git a/test/necho.cpp b/test/necho.cpp new file mode 100644 index 0000000..df961a0 --- /dev/null +++ b/test/necho.cpp @@ -0,0 +1,247 @@ +/* +* psocksxx - A C++ wrapper for POSIX sockets +* Copyright (C) 2013 Uditha Atukorala +* +* This software library is free software; you can redistribute it and/or modify +* it under the terms of the GNU Lesser General Public License as published by +* the Free Software Foundation; either version 3 of the License, or +* (at your option) any later version. +* +* This software library is distributed in the hope that it will be useful, +* but WITHOUT ANY WARRANTY; without even the implied warranty of +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +* GNU Lesser General Public License for more details. +* +* You should have received a copy of the GNU Lesser General Public License +* along with this software library. If not, see <http://www.gnu.org/licenses/>. +* +*/ + +#include "necho.h" + +#include <iostream> +#include <cstdlib> +#include <cstring> +#include <csignal> +#include <netdb.h> +#include <cerrno> +#include <sys/wait.h> + + +static int sockfd = -1; + +void sigterm_handler( int sig, siginfo_t *siginfo, void *context ) { + + // close any open sockets + if ( sockfd > -1 ) { + close( sockfd ); + } + + exit( EXIT_SUCCESS ); + +} + + +necho::necho( const char * node, const char * service, int socket_type ) : + _sockfd( -1 ), _nsock_node( node ), _nsock_service( service ), + _socket_type( socket_type ) { + + // fork + _cpid = fork(); + + if ( _cpid == -1 ) { + std::cerr << "failed to fork out child process" << std::endl; + } else if ( _cpid == 0 ) { + + // child - initialise signal handlers + init_signal_handlers(); + + // echo socket + init_listener(); + + // serve requests + serve_requests(); + + // exit - should never get here + exit( 0 ); + + } else { + // parent - wait for the child to create the socket and return + wait_connect(); + } + +} + + +necho::~necho() { + + // cleanup + shutdown(); + +} + + +void necho::init_signal_handlers() { + + struct sigaction sa; + + sigfillset( &sa.sa_mask ); + sa.sa_sigaction = &sigterm_handler; + sa.sa_flags = SA_SIGINFO; + + if ( sigaction( SIGTERM, &sa, 0 ) < 0 ) { + std::cerr << "sigaction() failed: " << strerror( errno ) << std::endl; + exit( EXIT_FAILURE ); + } + +} + + +void necho::init_listener() { + + addrinfo hints, * saddr_info; + int status = 0; + + // setup hints + memset( &hints, 0, sizeof ( hints ) ); + hints.ai_family = AF_INET; + hints.ai_socktype = _socket_type; + + if ( ( status = getaddrinfo( _nsock_node, _nsock_service, &hints, &saddr_info ) ) != 0 ) { + std::cerr << "getaddrinfo() failed: " << gai_strerror( status ) << std::endl; + exit( EXIT_FAILURE ); + } + + // initialise socket + if ( ( _sockfd = socket( saddr_info->ai_family, saddr_info->ai_socktype, saddr_info->ai_protocol ) ) < 0 ) { + std::cerr << "failed to setup network socket" << std::endl; + exit( EXIT_FAILURE ); + } + + // update static variable used by signal handlers + sockfd = _sockfd; + + // set SO_REUSEADDR to true so we can reuse the socket + int optval = 1; + if ( setsockopt( _sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof( optval ) ) != 0 ) { + std::cerr << "setsockopt() failed: " << strerror( errno ) << std::endl; + exit( EXIT_FAILURE ); + } + + // bind + if ( bind( _sockfd, saddr_info->ai_addr, saddr_info->ai_addrlen ) != 0 ) { + std::cerr << "bind() failed: " << strerror( errno ) << std::endl; + exit( EXIT_FAILURE ); + } + + + // listen + if ( listen( _sockfd, 5 ) != 0 ) { + std::cerr << "listen() failed: " << strerror( errno ) << std::endl; + exit( EXIT_FAILURE ); + } + + // cleanup + freeaddrinfo( saddr_info ); + +} + + +void necho::serve_requests() { + + int peer_sockfd; + + int n = 0; + char cbuffer[64]; + bool close_peer; + + for (;;) { + + if ( ( peer_sockfd = accept( _sockfd, 0, 0 ) ) == -1 ) { + continue; + } + + close_peer = false; + do { + + // read from the socket + n = recv( peer_sockfd, cbuffer, sizeof( cbuffer ), 0 ); + + // sanity check - did we receive anything? + if ( n <= 0 ) { + close_peer = true; + } else { + + // send back what we received + if ( send( peer_sockfd, cbuffer, n, 0 ) < 0 ) { + close_peer = true; + } + + } + + } while (! close_peer ); + + // close peer + close( peer_sockfd ); + + } + +} + + +void necho::wait_connect() { + + int sockfd = -1; + int status = 0; + unsigned int max_tries = 10; + addrinfo hints, * saddr_info; + + // setup hints + memset( &hints, 0, sizeof ( hints ) ); + hints.ai_family = AF_INET; + hints.ai_socktype = _socket_type; + + if ( ( status = getaddrinfo( _nsock_node, _nsock_service, &hints, &saddr_info ) ) != 0 ) { + std::cerr << "getaddrinfo() failed: " << gai_strerror( status ) << std::endl; + exit( EXIT_FAILURE ); + } + + // loop until we can establish a connection or + // we exhaust our attempts + for ( int i = 0; i < max_tries; i++ ) { + + // try to connect + if ( ( connect( sockfd, saddr_info->ai_addr, saddr_info->ai_addrlen ) == -1 ) ) { + + // connect failed - sleep for a bit + usleep( 500 ); + + } else { + // connect successful - break out of the loop + break; + } + + } + + // close socket + close( sockfd ); + +} + + +void necho::shutdown() { + + if ( _cpid == 0 ) { + // child - should never get here + } else { + + // parent - send kill to child + kill( _cpid, SIGTERM ); + + // wait for the child + waitpid( _cpid, 0, 0 ); + + } + +} + |