commit 6312e0f4e6a0a6de5fbc9922d726495d5a28c79e
author Manuel Pégourié-Gonnard <mpg@elzevir.fr> Tue Sep 23 12:46:33 2014 +0200
committer Paul Bakker <p.j.bakker@polarssl.org> Tue Oct 21 16:32:32 2014 +0200
tree 5cbf5dd3cfc5b766a0e3686780ca0a40b8a5771d
parent 484b8f9ed873b680ea4acc4b72d74cdccd83251b

udp_proxy: allow successive clients

programs/test/udp_proxy.c

diff --git a/programs/test/udp_proxy.c b/programs/test/udp_proxy.c
index 522cedc..cd4974d 100644
--- a/programs/test/udp_proxy.c
+++ b/programs/test/udp_proxy.c
@@ -327,11 +327,17 @@
     return( 0 );
 }
 
+static packet prev;
+
+void clear_pending( void )
+{
+    memset( &prev, 0, sizeof( packet ) );
+}
+
 int handle_message( const char *way, int dst, int src )
 {
     int ret;
     packet cur;
-    static packet prev;
 
     /* receive packet */
     if( ( ret = net_recv( &src, cur.buf, sizeof( cur.buf ) ) ) <= 0 )
@@ -446,6 +452,7 @@
     /*
      * 2. Wait until a client connects
      */
+accept:
     printf( "  . Waiting for a remote connection ..." );
     fflush( stdout );
 
@@ -458,16 +465,35 @@
     printf( " ok\n" );
     fflush( stdout );
 
+    printf( "  . Re-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" );
+
     /*
      * 3. Forward packets forever (kill the process to terminate it)
      */
-    nb_fds = ( client_fd > server_fd ? client_fd : server_fd ) + 1;
+    nb_fds = client_fd;
+    if( nb_fds < server_fd )
+        nb_fds = server_fd;
+    if( nb_fds < listen_fd )
+        nb_fds = listen_fd;
+    ++nb_fds;
 
     while( 1 )
     {
         FD_ZERO( &read_fds );
         FD_SET( server_fd, &read_fds );
         FD_SET( client_fd, &read_fds );
+        FD_SET( listen_fd, &read_fds );
 
         if( ( ret = select( nb_fds, &read_fds, NULL, NULL, NULL ) ) <= 0 )
         {
@@ -475,6 +501,12 @@
             goto exit;
         }
 
+        if( FD_ISSET( listen_fd, &read_fds ) )
+        {
+            clear_pending();
+            goto accept;
+        }
+
         if( FD_ISSET( client_fd, &read_fds ) )
         {
             if( ( ret = handle_message( "S <- C",