commit 95070a828660f8964bd4bd6b52162481873d685d
author k-stachowiak <krzysiek.stachowiak@gmail.com> Wed Dec 19 14:48:37 2018 +0100
committer k-stachowiak <krzysiek.stachowiak@gmail.com> Wed Dec 19 17:34:58 2018 +0100
tree 8b02cd2e77e837d0631705422b2abef4621b7a26
parent 6df25e793009466a857140127732dce01db9b686

Make some cipher parameter validation unconditional

library/cipher.c

diff --git a/library/cipher.c b/library/cipher.c
index 14ff371..b37fc76 100644
--- a/library/cipher.c
+++ b/library/cipher.c
@@ -182,7 +182,8 @@
 int mbedtls_cipher_setup( mbedtls_cipher_context_t *ctx, const mbedtls_cipher_info_t *cipher_info )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( cipher_info != NULL );
+    if( cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     memset( ctx, 0, sizeof( mbedtls_cipher_context_t ) );
 
@@ -211,10 +212,11 @@
                            const mbedtls_operation_t operation )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( key != NULL );
     CIPHER_VALIDATE_RET( operation == MBEDTLS_ENCRYPT ||
                          operation == MBEDTLS_DECRYPT );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     if( ( ctx->cipher_info->flags & MBEDTLS_CIPHER_VARIABLE_KEY_LEN ) == 0 &&
         (int) ctx->cipher_info->key_bitlen != key_bitlen )
@@ -251,8 +253,9 @@
     size_t actual_iv_size;
 
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( iv_len == 0 || iv != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     /* avoid buffer overflow in ctx->iv */
     if( iv_len > MBEDTLS_MAX_IV_LENGTH )
@@ -293,7 +296,8 @@
 int mbedtls_cipher_reset( mbedtls_cipher_context_t *ctx )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     ctx->unprocessed_len = 0;
 
@@ -305,8 +309,9 @@
                       const unsigned char *ad, size_t ad_len )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( ad_len == 0 || ad != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
 #if defined(MBEDTLS_GCM_C)
     if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
@@ -348,10 +353,11 @@
     size_t block_size;
 
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( ilen == 0 || input != NULL );
     CIPHER_VALIDATE_RET( output != NULL );
     CIPHER_VALIDATE_RET( olen != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     *olen = 0;
     block_size = mbedtls_cipher_get_block_size( ctx );
@@ -757,9 +763,10 @@
                    unsigned char *output, size_t *olen )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( output != NULL );
     CIPHER_VALIDATE_RET( olen != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     *olen = 0;
 
@@ -847,7 +854,8 @@
                                      mbedtls_cipher_padding_t mode )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     if( MBEDTLS_MODE_CBC != ctx->cipher_info->mode )
     {
@@ -898,8 +906,9 @@
                       unsigned char *tag, size_t tag_len )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( tag_len == 0 || tag != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     if( MBEDTLS_ENCRYPT != ctx->operation )
         return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
@@ -931,8 +940,9 @@
     int ret;
 
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( tag_len == 0 || tag != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     if( MBEDTLS_DECRYPT != ctx->operation )
     {
@@ -997,11 +1007,12 @@
     size_t finish_olen;
 
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( iv_len == 0 || iv != NULL );
     CIPHER_VALIDATE_RET( ilen == 0 || input != NULL );
     CIPHER_VALIDATE_RET( output != NULL );
     CIPHER_VALIDATE_RET( olen != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
     if( ( ret = mbedtls_cipher_set_iv( ctx, iv, iv_len ) ) != 0 )
         return( ret );
@@ -1032,13 +1043,14 @@
                          unsigned char *tag, size_t tag_len )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( iv != NULL );
     CIPHER_VALIDATE_RET( ad_len == 0 || ad != NULL );
     CIPHER_VALIDATE_RET( ilen == 0 || input != NULL );
     CIPHER_VALIDATE_RET( output != NULL );
     CIPHER_VALIDATE_RET( olen != NULL );
     CIPHER_VALIDATE_RET( tag_len == 0 || tag != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
 #if defined(MBEDTLS_GCM_C)
     if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
@@ -1088,13 +1100,14 @@
                          const unsigned char *tag, size_t tag_len )
 {
     CIPHER_VALIDATE_RET( ctx != NULL );
-    CIPHER_VALIDATE_RET( ctx->cipher_info != NULL );
     CIPHER_VALIDATE_RET( iv != NULL );
     CIPHER_VALIDATE_RET( ad_len == 0 || ad != NULL );
     CIPHER_VALIDATE_RET( ilen == 0 || input != NULL );
     CIPHER_VALIDATE_RET( output != NULL );
     CIPHER_VALIDATE_RET( olen != NULL );
     CIPHER_VALIDATE_RET( tag_len == 0 || tag != NULL );
+    if( ctx->cipher_info == NULL )
+        return MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA;
 
 #if defined(MBEDTLS_GCM_C)
     if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )