Have mbedtls_mpi_montg_init() take the modulus, rather than just its least significant limb
Signed-off-by: Tom Cosgrove <tom.cosgrove@arm.com>
diff --git a/library/bignum_core.c b/library/bignum_core.c
index bc1bca1..ac92853 100644
--- a/library/bignum_core.c
+++ b/library/bignum_core.c
@@ -348,14 +348,14 @@
/*
* Fast Montgomery initialization (thanks to Tom St Denis).
*/
-mbedtls_mpi_uint mbedtls_mpi_montg_init( mbedtls_mpi_uint m0 )
+mbedtls_mpi_uint mbedtls_mpi_montg_init( const mbedtls_mpi_uint *N )
{
- mbedtls_mpi_uint x = m0;
+ mbedtls_mpi_uint x = N[0];
- x += ( ( m0 + 2 ) & 4 ) << 1;
+ x += ( ( N[0] + 2 ) & 4 ) << 1;
for( unsigned int i = biL; i >= 8; i /= 2 )
- x *= ( 2 - ( m0 * x ) );
+ x *= ( 2 - ( N[0] * x ) );
return( ~x + 1 );
}