From 716163e82445dadc39ecb77ff8f4064b7e1a7846 Mon Sep 17 00:00:00 2001 From: Dave Rodgman Date: Mon, 20 Feb 2023 14:46:51 +0000 Subject: [PATCH] Improve allocation bounds in testing Signed-off-by: Dave Rodgman --- tests/suites/test_suite_pkcs7.function | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/suites/test_suite_pkcs7.function b/tests/suites/test_suite_pkcs7.function index 9dce25e04..91fe47b89 100644 --- a/tests/suites/test_suite_pkcs7.function +++ b/tests/suites/test_suite_pkcs7.function @@ -125,8 +125,8 @@ void pkcs7_verify(char *pkcs7_file, TEST_ASSERT(file != NULL); datalen = st.st_size; - /* Add 1 so that data is non-NULL for zero length input */ - ASSERT_ALLOC(data, datalen + 1); + /* Special-case for zero-length input so that data will be non-NULL */ + ASSERT_ALLOC(data, datalen == 0 ? 1 : datalen); buflen = fread((void *) data, sizeof(unsigned char), datalen, file); TEST_EQUAL(buflen, datalen);