commit 396333e0a3fec1259a5813561cc3221957725577
author Paul Bakker <p.j.bakker@polarssl.org> Thu Sep 26 13:32:19 2013 +0200
committer Paul Bakker <p.j.bakker@polarssl.org> Sat Sep 28 11:08:43 2013 +0200
tree f30e1e9f7ccf785344bfa62fe93e3af13135cc11
parent c13aab18dc69b383a888ac3e088e0707f5610a85

Updated ssl_test to handle EOF return value

programs/test/ssl_test.c

diff --git a/programs/test/ssl_test.c b/programs/test/ssl_test.c
index 97e308f..20907d8 100644
--- a/programs/test/ssl_test.c
+++ b/programs/test/ssl_test.c
@@ -329,7 +329,7 @@
 
         if( opt->command & COMMAND_READ )
         {
-            if( bytes_to_read == 0 )
+            while( bytes_to_read == 0 )
             {
                 bytes_to_read = rand() % opt->buffer_size;
                 offset_to_read = 0;
@@ -338,7 +338,7 @@
             ret = ssl_read( &ssl, read_buf + offset_to_read,
                             bytes_to_read );
 
-            if( ret >= 0 )
+            if( ret > 0 )
             {
                 for( i = 0; i < ret; i++ )
                 {
@@ -356,7 +356,9 @@
                 offset_to_read += ret;
             }
 
-            if( ret == POLARSSL_ERR_SSL_PEER_CLOSE_NOTIFY ||
+            if( ret == 0 ||
+                ret == POLARSSL_ERR_SSL_CONN_EOF ||
+                ret == POLARSSL_ERR_SSL_PEER_CLOSE_NOTIFY ||
                 ret == POLARSSL_ERR_NET_CONN_RESET )
             {
                 ret = 0;