Allow NULL input/output only in CCM mode.

Change-Id: I0758a1f2d8fa7accf8517aec6c93f79cf5f369d5

diff --git a/src/encrypt.c b/src/encrypt.c
index 6dc4527..5cdbeab 100644 (file)
--- a/src/encrypt.c
+++ b/src/encrypt.c
@@ -483,6 +483,10 @@ int encrypt_update(yaca_context_h ctx,
        if (c == NULL || input_len == 0 || output_len == NULL || op_type != c->op_type)
                return YACA_ERROR_INVALID_PARAMETER;
 
+       if (EVP_CIPHER_CTX_mode(c->cipher_ctx) != EVP_CIPH_CCM_MODE)
+               if (input == NULL || output == NULL)
+                       return YACA_ERROR_INVALID_PARAMETER;
+
        switch (op_type) {
        case OP_ENCRYPT:
                ret = EVP_EncryptUpdate(c->cipher_ctx, output, &loutput_len, input, input_len);