mirror/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-04-23 11:42:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bignum tests: automate modulo test object generation

Browse Source 
Signed-off-by: Janos Follath <janos.follath@arm.com>
This commit is contained in:
Janos Follath 2022-11-19 10:42:20 +00:00
parent 939621f8ed
commit c4fca5de3e
1 changed files with 33 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
scripts/mbedtls_dev/bignum_common.py
View File

@ -209,12 +209,12 @@ class OperationCommon(test_data_generation.BaseTest):
if cls.arity not in cls.arities: if cls.arity not in cls.arities:
raise ValueError("Unsupported number of operands!") raise ValueError("Unsupported number of operands!")
if cls.input_style == "arch_split": if cls.input_style == "arch_split":
test_objects = (cls(a_value, b_value, bits_in_limb=bil) test_objects = (cls(a, b, bits_in_limb=bil)
for a_value, b_value in cls.get_value_pairs() for a, b in cls.get_value_pairs()
for bil in cls.limb_sizes) for bil in cls.limb_sizes)
else: else:
test_objects = (cls(a_value, b_value) for test_objects = (cls(a, b)
a_value, b_value in cls.get_value_pairs()) for a, b in cls.get_value_pairs())
yield from (valid_test_object.create_test_case() yield from (valid_test_object.create_test_case()
for valid_test_object in filter( for valid_test_object in filter(
lambda test_object: test_object.is_valid, lambda test_object: test_object.is_valid,
@ -225,6 +225,7 @@ class OperationCommon(test_data_generation.BaseTest):
class ModOperationCommon(OperationCommon): class ModOperationCommon(OperationCommon):
#pylint: disable=abstract-method #pylint: disable=abstract-method
"""Target for bignum mod_raw test case generation.""" """Target for bignum mod_raw test case generation."""
moduli = [] # type: List[str]
def __init__(self, val_n: str, val_a: str, val_b: str = "0", def __init__(self, val_n: str, val_a: str, val_b: str = "0",
bits_in_limb: int = 64) -> None: bits_in_limb: int = 64) -> None:
@ -258,6 +259,34 @@ class ModOperationCommon(OperationCommon):
def r2(self) -> int: # pylint: disable=invalid-name def r2(self) -> int: # pylint: disable=invalid-name
return pow(self.r, 2) return pow(self.r, 2)
@property
def is_valid(self) -> bool:
if self.int_a >= self.int_n:
return False
if self.arity == 2 and self.int_b >= self.int_n:
return False
return True
@classmethod
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
if cls.input_style not in cls.input_styles:
raise ValueError("Unknown input style!")
if cls.arity not in cls.arities:
raise ValueError("Unsupported number of operands!")
if cls.input_style == "arch_split":
test_objects = (cls(n, a, b, bits_in_limb=bil)
for n in cls.moduli
for a, b in cls.get_value_pairs()
for bil in cls.limb_sizes)
else:
test_objects = (cls(n, a, b)
for n in cls.moduli
for a, b in cls.get_value_pairs())
yield from (valid_test_object.create_test_case()
for valid_test_object in filter(
lambda test_object: test_object.is_valid,
test_objects
))
# BEGIN MERGE SLOT 1 # BEGIN MERGE SLOT 1