commit 6b0d268bc919e14e0617169052e0ad874f2fca5c
author Manuel Pégourié-Gonnard <mpg@elzevir.fr> Tue Mar 25 11:24:43 2014 +0100
committer Manuel Pégourié-Gonnard <mpg@elzevir.fr> Mon Mar 31 11:28:11 2014 +0200
tree 34f11dde12da22b5fc0149bcec5b60ee869047a3
parent 00d538f8f996601f3ef8b88d0ad855f84ae6f3c4

Add ssl_close_notify() to servers that missed it

programs/ssl/ssl_pthread_server.c

diff --git a/programs/ssl/ssl_pthread_server.c b/programs/ssl/ssl_pthread_server.c
index 7e1b52e..efb360c 100644
--- a/programs/ssl/ssl_pthread_server.c
+++ b/programs/ssl/ssl_pthread_server.c
@@ -270,6 +270,21 @@
     printf( "  [ #%d ]  %d bytes written\n=====\n%s\n=====\n",
             thread_id, len, (char *) buf );
 
+    printf( "  [ #%d ]  . Closing the connection...", thread_id );
+
+    while( ( ret = ssl_close_notify( &ssl ) ) < 0 )
+    {
+        if( ret != POLARSSL_ERR_NET_WANT_READ &&
+            ret != POLARSSL_ERR_NET_WANT_WRITE )
+        {
+            printf( "  [ #%d ]  failed: ssl_close_notify returned -0x%04x\n",
+                    thread_id, ret );
+            goto thread_exit;
+        }
+    }
+
+    printf( " ok\n" );
+
     ret = 0;
 
 thread_exit: