commit 1265954689d3825e4f4b902a02362277d969de05
author Raef Coles <raef.coles@arm.com> Fri Jul 14 10:15:21 2023 +0100
committer David Hu <david.hu@arm.com> Mon Jul 17 05:37:59 2023 +0200
tree 270a70d9d2f18644cada1b1cd2d8c6f2b472eb6d
parent 2271523071fb30b21c9185d9b7ecd37f05d3e926

CC312: Update zero-counting check

To correctly handle input buffers smaller than the OTP item.

Change-Id: I544b8b6258998021ece030e7833853872a7b46cc
Signed-off-by: Raef Coles <raef.coles@arm.com>

platform/ext/accelerator/cc312/otp_cc312.c

diff --git a/platform/ext/accelerator/cc312/otp_cc312.c b/platform/ext/accelerator/cc312/otp_cc312.c
index 16bb1e6..321ce1e 100644
--- a/platform/ext/accelerator/cc312/otp_cc312.c
+++ b/platform/ext/accelerator/cc312/otp_cc312.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021-2022, Arm Limited. All rights reserved.
+ * Copyright (c) 2021-2023, Arm Limited. All rights reserved.
  *
  * SPDX-License-Identifier: BSD-3-Clause
  *
@@ -391,6 +391,12 @@
      */
     if (zero_byte_buf != NULL) {
         zero_count = count_buffer_zero_bits(in, in_len);
+        if (in_len < size) {
+            /* If the buffer is smaller than the OTP item, count the remainder
+             * of the OTP item as well.
+             */
+            zero_count += count_otp_zero_bits(addr + in_len, size - in_len);
+        }
 
         err = otp_write(zero_byte_buf, 2, sizeof(zero_count), (uint8_t*)&zero_count, NULL);
         if (err != TFM_PLAT_ERR_SUCCESS) {