commit c240600f2490fb7fd01db78dfa8a0e2aec003633
author Tom Cosgrove <tom.cosgrove@arm.com> Tue Dec 06 12:20:43 2022 +0000
committer Tom Cosgrove <tom.cosgrove@arm.com> Tue Dec 06 12:20:43 2022 +0000
tree 2fffa016a4df02ed1fbdc2aab2de9321ad619db1
parent ecda1868934796d0f423feaceadc07ce360f3bd3

Separate out to_montgomery and from_montgomery for bignum tests

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

scripts/mbedtls_dev/bignum_common.py

diff --git a/scripts/mbedtls_dev/bignum_common.py b/scripts/mbedtls_dev/bignum_common.py
index 67ea78d..81bc28e 100644
--- a/scripts/mbedtls_dev/bignum_common.py
+++ b/scripts/mbedtls_dev/bignum_common.py
@@ -251,6 +251,12 @@
         # provides earlier/more robust input validation.
         self.int_n = hex_to_int(val_n)
 
+    def to_montgomery(self, val) -> int:
+        return (val * self.r) % self.int_n
+
+    def from_montgomery(self, val) -> int:
+        return (val * self.r_inv) % self.int_n
+
     @property
     def boundary(self) -> int:
         return self.int_n