bignum_common: Adjusted `format_arg` to always size input according to modulo.
Signed-off-by: Minos Galanakis <minos.galanakis@arm.com>
diff --git a/scripts/mbedtls_dev/bignum_common.py b/scripts/mbedtls_dev/bignum_common.py
index 7b2669a..bb64ce0 100644
--- a/scripts/mbedtls_dev/bignum_common.py
+++ b/scripts/mbedtls_dev/bignum_common.py
@@ -297,10 +297,7 @@
return self.format_arg(self.val_n)
def format_arg(self, val: str) -> str:
- if self.input_style == "variable":
- return val.zfill(len(hex(self.int_n)) - 2)
- else:
- return super().format_arg(val)
+ return super().format_arg(val).zfill(self.hex_digits)
def arguments(self) -> List[str]:
return [quote_str(self.arg_n)] + super().arguments()