Only allow missing SereverKeyExchange message in bare PSK mode
diff --git a/library/ssl_cli.c b/library/ssl_cli.c
index a4cc3d7..ff467f3 100644
--- a/library/ssl_cli.c
+++ b/library/ssl_cli.c
@@ -948,8 +948,14 @@
 
     if( ssl->in_msg[0] != SSL_HS_SERVER_KEY_EXCHANGE )
     {
-        ssl->record_read = 1;
-        goto exit;
+        if( ciphersuite_info->key_exchange == POLARSSL_KEY_EXCHANGE_PSK )
+        {
+            ssl->record_read = 1;
+            goto exit;
+        }
+
+        SSL_DEBUG_MSG( 1, ( "bad server key exchange message" ) );
+        return( POLARSSL_ERR_SSL_UNEXPECTED_MESSAGE );
     }
 
     SSL_DEBUG_BUF( 3,   "server key exchange", ssl->in_msg + 4, ssl->in_hslen - 4 );