commit 3cbaf1e379900e97c21f5c640496f5971087ed70
author Paul Bakker <p.j.bakker@polarssl.org> Tue Jul 08 12:26:02 2014 +0200
committer Paul Bakker <p.j.bakker@polarssl.org> Tue Jul 08 14:30:35 2014 +0200
tree 43f07d622645fbb9a3bbf4287814dbb8170fc695
parent 875548ce6701c00dd92812e3589df9f9c7c4b4d7

Add ssl_close_notify() to servers that missed it

programs/ssl/ssl_server.c

diff --git a/programs/ssl/ssl_server.c b/programs/ssl/ssl_server.c
index 48e5677..a4247da 100644
--- a/programs/ssl/ssl_server.c
+++ b/programs/ssl/ssl_server.c
@@ -324,7 +324,21 @@
 
     len = ret;
     printf( " %d bytes written\n\n%s\n", len, (char *) buf );
-    
+
+    printf( "  . Closing the connection..." );
+
+    while( ( ret = ssl_close_notify( &ssl ) ) < 0 )
+    {
+        if( ret != POLARSSL_ERR_NET_WANT_READ &&
+            ret != POLARSSL_ERR_NET_WANT_WRITE )
+        {
+            printf( " failed\n  ! ssl_close_notify returned %d\n\n", ret );
+            goto reset;
+        }
+    }
+
+    printf( " ok\n" );
+
     ret = 0;
     goto reset;