From cb4137b6468ca54b503a1710301d1468754b518d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20P=C3=A9gouri=C3=A9-Gonnard?= Date: Thu, 4 Sep 2014 14:55:28 +0200 Subject: [PATCH] Add test utility udp_proxy Currently just forwards: will delay, duplicate and drop later. --- programs/.gitignore | 1 + programs/Makefile | 5 + programs/test/CMakeLists.txt | 5 +- programs/test/udp_proxy.c | 278 +++++++++++++++++++++++++++++++++++ 4 files changed, 288 insertions(+), 1 deletion(-) create mode 100644 programs/test/udp_proxy.c diff --git a/programs/.gitignore b/programs/.gitignore index a47ef0662..4dc2c1853 100644 --- a/programs/.gitignore +++ b/programs/.gitignore @@ -41,6 +41,7 @@ test/o_p_test test/selftest test/ssl_cert_test test/ssl_test +test/udp_proxy util/pem2der util/strerror x509/cert_app diff --git a/programs/Makefile b/programs/Makefile index 9238ae8c2..b3d8dbf77 100644 --- a/programs/Makefile +++ b/programs/Makefile @@ -41,6 +41,7 @@ APPS = aes/aescrypt2 aes/crypt_and_hash \ random/gen_random_ctr_drbg \ test/ssl_cert_test test/benchmark \ test/selftest test/ssl_test \ + test/udp_proxy \ util/pem2der util/strerror \ x509/cert_app x509/crl_app \ x509/cert_req @@ -217,6 +218,10 @@ test/ssl_test: test/ssl_test.c ../library/libpolarssl.a echo " CC test/ssl_test.c" $(CC) $(CFLAGS) $(OFLAGS) test/ssl_test.c $(LDFLAGS) -o $@ +test/udp_proxy: test/udp_proxy.c ../library/libpolarssl.a + echo " CC test/udp_proxy.c" + $(CC) $(CFLAGS) $(OFLAGS) test/udp_proxy.c $(LDFLAGS) -o $@ + test/o_p_test: test/o_p_test.c ../library/libpolarssl.a echo " CC test/o_p_test.c" $(CC) $(CFLAGS) $(OFLAGS) test/o_p_test.c $(LDFLAGS) -o $@ -lssl -lcrypto diff --git a/programs/test/CMakeLists.txt b/programs/test/CMakeLists.txt index 758b7a55d..5e7221064 100644 --- a/programs/test/CMakeLists.txt +++ b/programs/test/CMakeLists.txt @@ -24,7 +24,10 @@ target_link_libraries(ssl_test ${libs}) add_executable(ssl_cert_test ssl_cert_test.c) target_link_libraries(ssl_cert_test ${libs}) -install(TARGETS selftest benchmark ssl_test ssl_cert_test +add_executable(udp_proxy udp_proxy.c) +target_link_libraries(udp_proxy ${libs}) + +install(TARGETS selftest benchmark ssl_test ssl_cert_test udp_proxy DESTINATION "bin" PERMISSIONS OWNER_READ OWNER_WRITE OWNER_EXECUTE GROUP_READ GROUP_EXECUTE WORLD_READ WORLD_EXECUTE) diff --git a/programs/test/udp_proxy.c b/programs/test/udp_proxy.c new file mode 100644 index 000000000..586a1b21d --- /dev/null +++ b/programs/test/udp_proxy.c @@ -0,0 +1,278 @@ +/* + * UDP proxy: emulate an unreliable UDP connexion for DTLS testing + * + * Copyright (C) 2006-2014, Brainspark B.V. + * + * This file is part of PolarSSL (http://www.polarssl.org) + * Lead Maintainer: Paul Bakker + * + * All rights reserved. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#if !defined(POLARSSL_CONFIG_FILE) +#include "polarssl/config.h" +#else +#include POLARSSL_CONFIG_FILE +#endif + +#if !defined(POLARSSL_NET_C) +#include +int main( void ) +{ + printf( "POLARSSL_NET_C not defined.\n" ); + return( 0 ); +} +#else + +#include "polarssl/net.h" +#include "polarssl/error.h" + +#include +#include + +/* For select() */ +#if (defined(_WIN32) || defined(_WIN32_WCE)) && !defined(EFIX64) && \ + !defined(EFI32) +#include +#include +#if defined(_MSC_VER) +#if defined(_WIN32_WCE) +#pragma comment( lib, "ws2.lib" ) +#else +#pragma comment( lib, "ws2_32.lib" ) +#endif +#endif /* _MSC_VER */ +#else /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */ +#include +#include +#include +#endif /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */ + +#define MAX_MSG_SIZE 18445 /* 2^14 + 2048 + 13 */ + +#define DFL_SERVER_ADDR "localhost" +#define DFL_SERVER_PORT 4433 +#define DFL_LISTEN_ADDR "localhost" +#define DFL_LISTEN_PORT 5556 + +/* + * global options + */ +struct options +{ + const char *server_addr; /* address to forward packets to */ + int server_port; /* port to forward packets to */ + const char *listen_addr; /* address for accepting client connections */ + int listen_port; /* port for accepting client connections */ +} opt; + +#define USAGE \ + "\n usage: udp_proxy param=<>...\n" \ + "\n acceptable parameters:\n" \ + " server_addr=%%d default: localhost\n" \ + " server_port=%%d default: 4433\n" \ + " listen_addr=%%d default: localhost\n" \ + " listen_port=%%d default: 4433\n" \ + "\n" + +static void exit_usage( void ) +{ + printf( USAGE ); + exit( 1 ); +} + +static void get_options( int argc, char *argv[] ) +{ + int i; + char *p, *q; + + opt.server_addr = DFL_SERVER_ADDR; + opt.server_port = DFL_SERVER_PORT; + opt.listen_addr = DFL_LISTEN_ADDR; + opt.listen_port = DFL_LISTEN_PORT; + + for( i = 1; i < argc; i++ ) + { + p = argv[i]; + if( ( q = strchr( p, '=' ) ) == NULL ) + exit_usage(); + *q++ = '\0'; + + if( strcmp( p, "server_addr" ) == 0 ) + opt.server_addr = q; + else if( strcmp( p, "server_port" ) == 0 ) + { + opt.server_port = atoi( q ); + if( opt.server_port < 1 || opt.server_port > 65535 ) + exit_usage(); + } + else if( strcmp( p, "listen_addr" ) == 0 ) + opt.listen_addr = q; + else if( strcmp( p, "listen_port" ) == 0 ) + { + opt.listen_port = atoi( q ); + if( opt.listen_port < 1 || opt.listen_port > 65535 ) + exit_usage(); + } + else + exit_usage(); + } +} + +int handle_message( const char *way, int dst, int src ) +{ + unsigned char buf[MAX_MSG_SIZE] = { 0 }; + int ret; + size_t len; + + if( ( ret = net_recv( &src, buf, sizeof( buf ) ) ) <= 0 ) + { + printf( " ! net_recv returned %d\n", ret ); + return( ret ); + } + + printf( " .. %s: %d bytes forwarded\n", way, ret ); + + len = (size_t) ret; + + if( ( ret = net_send( &dst, buf, len ) ) <= 0 ) + { + printf( " ! net_send returned %d\n", ret ); + return( ret ); + } + + fflush( stdout ); + return( 0 ); +} + +int main( int argc, char *argv[] ) +{ + int ret; + + int listen_fd = -1; + int client_fd = -1; + int server_fd = -1; + + int nb_fds; + fd_set read_fds; + + get_options( argc, argv ); + + /* + * 0. Connect to the server + */ + printf( " . Connect to server on UDP/%s/%d ...", + opt.server_addr, opt.server_port ); + fflush( stdout ); + + if( ( ret = net_connect( &server_fd, opt.server_addr, opt.server_port, + NET_PROTO_UDP ) ) != 0 ) + { + printf( " failed\n ! net_connect returned %d\n\n", ret ); + goto exit; + } + + printf( " ok\n" ); + + /* + * 1. Setup the "listening" UDP socket + */ + printf( " . Bind on UDP/%s/%d ...", + opt.listen_addr, opt.listen_port ); + fflush( stdout ); + + if( ( ret = net_bind( &listen_fd, opt.listen_addr, opt.listen_port, + NET_PROTO_UDP ) ) != 0 ) + { + printf( " failed\n ! net_bind returned %d\n\n", ret ); + goto exit; + } + + printf( " ok\n" ); + + /* + * 2. Wait until a client connects + */ + printf( " . Waiting for a remote connection ..." ); + fflush( stdout ); + + if( ( ret = net_accept( listen_fd, &client_fd, NULL ) ) != 0 ) + { + printf( " failed\n ! net_accept returned %d\n\n", ret ); + goto exit; + } + + printf( " ok\n" ); + fflush( stdout ); + + /* + * 3. Forward packets forever (kill the process to terminate it) + */ + nb_fds = ( client_fd > server_fd ? client_fd : server_fd ) + 1; + + while( 1 ) + { + FD_ZERO( &read_fds ); + FD_SET( server_fd, &read_fds ); + FD_SET( client_fd, &read_fds ); + + if( ( ret = select( nb_fds, &read_fds, NULL, NULL, NULL ) ) <= 0 ) + { + perror( "select" ); + goto exit; + } + + if( FD_ISSET( client_fd, &read_fds ) ) + { + if( ( ret = handle_message( "c2s", server_fd, client_fd ) ) != 0 ) + goto exit; + } + + if( FD_ISSET( server_fd, &read_fds ) ) + { + if( ( ret = handle_message( "s2c", client_fd, server_fd ) ) != 0 ) + goto exit; + } + } + +exit: + +#ifdef POLARSSL_ERROR_C + if( ret != 0 ) + { + char error_buf[100]; + polarssl_strerror( ret, error_buf, 100 ); + printf( "Last error was: -0x%04X - %s\n\n", - ret, error_buf ); + fflush( stdout ); + } +#endif + + if( client_fd != -1 ) + net_close( client_fd ); + + if( listen_fd != -1 ) + net_close( listen_fd ); + +#if defined(_WIN32) + printf( " Press Enter to exit this program.\n" ); + fflush( stdout ); getchar(); +#endif + + return( ret != 0 ); +} + +#endif /* POLARSSL_NET_C */