commit 4131ec1260cc97e923832c38b396d016badd5c81
author Jerry Yu <jerry.h.yu@arm.com> Wed Jan 19 10:36:30 2022 +0800
committer Jerry Yu <jerry.h.yu@arm.com> Tue Jan 25 12:46:17 2022 +0800
tree f290463dca09e78c7298f8d6a2872a2215f01384
parent a69269a7117f799fcbe9ccde756b582e543d7e67

Add signature algorithm length check

Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>

library/ssl_tls.c

diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index bb2b47e..e398f48 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -3160,7 +3160,7 @@
     {
         const int *md;
         const int *sig_hashes = ssl->conf->sig_hashes;
-        size_t sig_algs_len = sizeof( uint16_t );
+        size_t sig_algs_len = 0;
         uint16_t *p;
 
         for( md = sig_hashes; *md != MBEDTLS_MD_NONE; md++ )
@@ -3175,10 +3175,13 @@
     #endif
         }
 
-        if( sig_algs_len == sizeof( uint16_t ) )
+        if( sig_algs_len < MBEDTLS_SSL_MIN_SIG_ALG_LIST_LEN ||
+            sig_algs_len > MBEDTLS_SSL_MAX_SIG_ALG_LIST_LEN )
+        {
             return( MBEDTLS_ERR_SSL_BAD_CONFIG );
+        }
 
-        ssl->handshake->sig_algs = mbedtls_calloc( 1, sig_algs_len );
+        ssl->handshake->sig_algs = mbedtls_calloc( 1, sig_algs_len + 2 );
         if( ssl->handshake->sig_algs == NULL )
             return( MBEDTLS_ERR_SSL_ALLOC_FAILED );