commit 28ff92cc3a72d98885e4fa6dc19e9725db5f9ac7
author Tom Cosgrove <tom.cosgrove@arm.com> Mon Dec 12 17:06:27 2022 +0000
committer Tom Cosgrove <tom.cosgrove@arm.com> Thu Dec 15 16:56:36 2022 +0000
tree 1b04297e920113c36eca096308b4c51b2b1f79b2
parent 30f3b4d60112ffe68cc162cc01938b84d18c9960

Add an explicit mbedtls_mpi_core_montmul_working_limbs() function

Signed-off-by: Tom Cosgrove <tom.cosgrove@arm.com>

library/bignum_mod_raw.c

diff --git a/library/bignum_mod_raw.c b/library/bignum_mod_raw.c
index c98a1c1..be8fc86 100644
--- a/library/bignum_mod_raw.c
+++ b/library/bignum_mod_raw.c
@@ -183,7 +183,7 @@
                                      const mbedtls_mpi_mod_modulus *m )
 {
     mbedtls_mpi_uint *T;
-    const size_t t_limbs = m->limbs * 2 + 1;
+    const size_t t_limbs = mbedtls_mpi_core_montmul_working_limbs( m->limbs );
 
     if( ( T = (mbedtls_mpi_uint *) mbedtls_calloc( t_limbs, ciL ) ) == NULL )
         return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
@@ -200,7 +200,7 @@
                                        const mbedtls_mpi_mod_modulus *m )
 {
     const mbedtls_mpi_uint one = 1;
-    const size_t t_limbs = m->limbs * 2 + 1;
+    const size_t t_limbs = mbedtls_mpi_core_montmul_working_limbs( m->limbs );
     mbedtls_mpi_uint *T;
 
     if( ( T = (mbedtls_mpi_uint *) mbedtls_calloc( t_limbs, ciL ) ) == NULL )