Fix buffer overflow

aws · Feb 3, 2025 · 95d3add · 95d3add
1 parent dc83a25
commit 95d3add
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crypto/fipsmodule/sha/sha3_test.cc b/crypto/fipsmodule/sha/sha3_test.cc
@@ -181,7 +181,7 @@ class SHA3TestVector {
 
     // Test Squeeze
     // Assert success when |EVP_DigestSqueeze| is called in set byte increments
-    for (num_tests = 0, i = 0; num_tests <  sizeof(stride_tests); num_tests++, i = 0) {
+    for (num_tests = 0, i = 0; num_tests <   (sizeof(stride_tests)/sizeof(stride_tests[0])); num_tests++, i = 0) {
       OPENSSL_memset(digest.get(), 0, digest_length);
       ASSERT_TRUE(EVP_DigestInit(ctx.get(), algorithm));
       ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), msg_.data(), msg_.size()));
@@ -235,7 +235,7 @@ class SHA3TestVector {
   // Test Squeeze with random Input
   // Assert success when |EVP_DigestSqueeze| is called on a random message
   // in set byte increments
-  for (num_tests = 0, i = 0; num_tests <  sizeof(stride_tests); num_tests++, i = 0) {
+  for (num_tests = 0, i = 0; num_tests <   (sizeof(stride_tests)/sizeof(stride_tests[0])); num_tests++, i = 0) {
     OPENSSL_memset(digest_stream.get(), 0, expected_output_size);
     ASSERT_TRUE(EVP_DigestInit(ctx.get(), algorithm));
     ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), random_bytes.data(),  num_bytes));