Allow stub implementation of the context_save for now
diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index 4235913..415caa0 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -2371,8 +2371,11 @@
         opt.serialize = 0;
         mbedtls_printf( " Serializing live connection..." );
 
-        if( ( ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len ) )
-                != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL )
+        ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len );
+
+        /* Allow stub implementation returning 0 for now */
+        if( ret != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL &&
+            ret != 0 )
         {
             mbedtls_printf( " failed\n  ! mbedtls_ssl_context_save returned "
                             "-0x%x\n\n", -ret );
diff --git a/programs/ssl/ssl_server2.c b/programs/ssl/ssl_server2.c
index 7b2c184..6103688 100644
--- a/programs/ssl/ssl_server2.c
+++ b/programs/ssl/ssl_server2.c
@@ -3356,8 +3356,11 @@
         opt.serialize = 0;
         mbedtls_printf( " Serializing live connection..." );
 
-        if( ( ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len ) )
-                != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL )
+        ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len );
+
+        /* Allow stub implementation returning 0 for now */
+        if( ret != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL &&
+            ret != 0 )
         {
             mbedtls_printf( " failed\n  ! mbedtls_ssl_context_save returned "
                             "-0x%x\n\n", -ret );