Add test to check for the fix on AArch64.

aws · Jul 24, 2024 · 97bee82 · 97bee82
1 parent 1158d91
commit 97bee82
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/crypto/fipsmodule/aes/aes_test.cc b/crypto/fipsmodule/aes/aes_test.cc
@@ -479,6 +479,13 @@ TEST(AESTest, ABI) {
         SCOPED_TRACE(blocks);
         CHECK_ABI(aes_hw_cbc_encrypt, buf, buf, AES_BLOCK_SIZE * blocks, &key,
                   block, AES_ENCRYPT);
+#if defined(OPENSSL_ARM) || defined(OPENSSL_AARCH64)
+        if (blocks == 0) {
+          Bytes buf_before = Bytes(buf,64);
+          CHECK_ABI(aes_hw_ctr32_encrypt_blocks, buf, buf, blocks, &key, block);
+          EXPECT_EQ(buf_before, Bytes(buf,64));
+        }
+#endif
         CHECK_ABI(aes_hw_ctr32_encrypt_blocks, buf, buf, blocks, &key, block);
 #if defined(HWAES_ECB)
         CHECK_ABI(aes_hw_ecb_encrypt, buf, buf, AES_BLOCK_SIZE * blocks, &key,