0cd8967ba1
The class BaseTarget served two purposes: - track test cases and target files for generation - provide an abstract base class for individual test groups Splitting these allows decoupling these two and to have further common superclasses across targets. No intended change in generated test cases. Signed-off-by: Janos Follath <janos.follath@arm.com>
257 lines
9.7 KiB
Python
257 lines
9.7 KiB
Python
"""Framework classes for generation of bignum mod_raw test cases."""
|
|
# Copyright The Mbed TLS Contributors
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from abc import ABCMeta
|
|
from typing import Dict, Iterator, List
|
|
|
|
from . import test_case
|
|
from . import test_data_generation
|
|
from . import bignum_common
|
|
|
|
class BignumModRawTarget(test_data_generation.BaseTarget):
|
|
#pylint: disable=abstract-method, too-few-public-methods
|
|
"""Target for bignum mod_raw test case generation."""
|
|
target_basename = 'test_suite_bignum_mod_raw.generated'
|
|
|
|
class BignumModRawOperation(bignum_common.OperationCommon, BignumModRawTarget, metaclass=ABCMeta):
|
|
#pylint: disable=abstract-method
|
|
"""Target for bignum mod_raw test case generation."""
|
|
|
|
def __init__(self, val_n: str, val_a: str, val_b: str = "0", bits_in_limb: int = 64) -> None:
|
|
super().__init__(val_a=val_a, val_b=val_b)
|
|
self.val_n = val_n
|
|
self.bits_in_limb = bits_in_limb
|
|
|
|
@property
|
|
def int_n(self) -> int:
|
|
return bignum_common.hex_to_int(self.val_n)
|
|
|
|
@property
|
|
def boundary(self) -> int:
|
|
data_in = [self.int_a, self.int_b, self.int_n]
|
|
return max([n for n in data_in if n is not None])
|
|
|
|
@property
|
|
def limbs(self) -> int:
|
|
return bignum_common.limbs_mpi(self.boundary, self.bits_in_limb)
|
|
|
|
@property
|
|
def hex_digits(self) -> int:
|
|
return 2 * (self.limbs * self.bits_in_limb // 8)
|
|
|
|
@property
|
|
def hex_n(self) -> str:
|
|
return "{:x}".format(self.int_n).zfill(self.hex_digits)
|
|
|
|
@property
|
|
def hex_a(self) -> str:
|
|
return "{:x}".format(self.int_a).zfill(self.hex_digits)
|
|
|
|
@property
|
|
def hex_b(self) -> str:
|
|
return "{:x}".format(self.int_b).zfill(self.hex_digits)
|
|
|
|
@property
|
|
def r(self) -> int: # pylint: disable=invalid-name
|
|
l = bignum_common.limbs_mpi(self.int_n, self.bits_in_limb)
|
|
return bignum_common.bound_mpi_limbs(l, self.bits_in_limb)
|
|
|
|
@property
|
|
def r_inv(self) -> int:
|
|
return bignum_common.invmod(self.r, self.int_n)
|
|
|
|
@property
|
|
def r2(self) -> int: # pylint: disable=invalid-name
|
|
return pow(self.r, 2)
|
|
|
|
class BignumModRawOperationArchSplit(BignumModRawOperation):
|
|
#pylint: disable=abstract-method
|
|
"""Common features for bignum mod raw operations where the result depends on
|
|
the limb size."""
|
|
|
|
limb_sizes = [32, 64] # type: List[int]
|
|
|
|
def __init__(self, val_n: str, val_a: str, val_b: str = "0", bits_in_limb: int = 64) -> None:
|
|
super().__init__(val_n=val_n, val_a=val_a, val_b=val_b, bits_in_limb=bits_in_limb)
|
|
|
|
if bits_in_limb not in self.limb_sizes:
|
|
raise ValueError("Invalid number of bits in limb!")
|
|
|
|
self.dependencies = ["MBEDTLS_HAVE_INT{:d}".format(bits_in_limb)]
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
for a_value, b_value in cls.get_value_pairs():
|
|
for bil in cls.limb_sizes:
|
|
yield cls(a_value, b_value, bits_in_limb=bil).create_test_case()
|
|
# BEGIN MERGE SLOT 1
|
|
|
|
# END MERGE SLOT 1
|
|
|
|
# BEGIN MERGE SLOT 2
|
|
|
|
# END MERGE SLOT 2
|
|
|
|
# BEGIN MERGE SLOT 3
|
|
|
|
# END MERGE SLOT 3
|
|
|
|
# BEGIN MERGE SLOT 4
|
|
|
|
# END MERGE SLOT 4
|
|
|
|
# BEGIN MERGE SLOT 5
|
|
|
|
# END MERGE SLOT 5
|
|
|
|
# BEGIN MERGE SLOT 6
|
|
|
|
# END MERGE SLOT 6
|
|
|
|
# BEGIN MERGE SLOT 7
|
|
class BignumModRawConvertToMont(BignumModRawOperationArchSplit):
|
|
""" Test cases for mpi_mod_raw_to_mont_rep(). """
|
|
|
|
test_function = "mpi_mod_raw_to_mont_rep"
|
|
test_name = "Convert into Mont: "
|
|
|
|
test_data_moduli = ["b",
|
|
"fd",
|
|
"eeff99aa37",
|
|
"eeff99aa11",
|
|
"800000000005",
|
|
"7fffffffffffffff",
|
|
"80fe000a10000001",
|
|
"25a55a46e5da99c71c7",
|
|
"1058ad82120c3a10196bb36229c1",
|
|
"7e35b84cb19ea5bc57ec37f5e431462fa962d98c1e63738d4657f"
|
|
"18ad6532e6adc3eafe67f1e5fa262af94cee8d3e7268593942a2a"
|
|
"98df75154f8c914a282f8b",
|
|
"8335616aed761f1f7f44e6bd49e807b82e3bf2bf11bfa63",
|
|
"ffcece570f2f991013f26dd5b03c4c5b65f97be5905f36cb4664f"
|
|
"2c78ff80aa8135a4aaf57ccb8a0aca2f394909a74cef1ef6758a6"
|
|
"4d11e2c149c393659d124bfc94196f0ce88f7d7d567efa5a649e2"
|
|
"deefaa6e10fdc3deac60d606bf63fc540ac95294347031aefd73d"
|
|
"6a9ee10188aaeb7a90d920894553cb196881691cadc51808715a0"
|
|
"7e8b24fcb1a63df047c7cdf084dd177ba368c806f3d51ddb5d389"
|
|
"8c863e687ecaf7d649a57a46264a582f94d3c8f2edaf59f77a7f6"
|
|
"bdaf83c991e8f06abe220ec8507386fce8c3da84c6c3903ab8f3a"
|
|
"d4630a204196a7dbcbd9bcca4e40ec5cc5c09938d49f5e1e6181d"
|
|
"b8896f33bb12e6ef73f12ec5c5ea7a8a337"
|
|
]
|
|
|
|
test_input_numbers = ["0",
|
|
"1",
|
|
"97",
|
|
"f5",
|
|
"6f5c3",
|
|
"745bfe50f7",
|
|
"ffa1f9924123",
|
|
"334a8b983c79bd",
|
|
"5b84f632b58f3461",
|
|
"19acd15bc38008e1",
|
|
"ffffffffffffffff",
|
|
"54ce6a6bb8247fa0427cfc75a6b0599",
|
|
"fecafe8eca052f154ce6a6bb8247fa019558bfeecce9bb9",
|
|
"a87d7a56fa4bfdc7da42ef798b9cf6843d4c54794698cb14d72"
|
|
"851dec9586a319f4bb6d5695acbd7c92e7a42a5ede6972adcbc"
|
|
"f68425265887f2d721f462b7f1b91531bac29fa648facb8e3c6"
|
|
"1bd5ae42d5a59ba1c89a95897bfe541a8ce1d633b98f379c481"
|
|
"6f25e21f6ac49286b261adb4b78274fe5f61c187581f213e84b"
|
|
"2a821e341ef956ecd5de89e6c1a35418cd74a549379d2d4594a"
|
|
"577543147f8e35b3514e62cf3e89d1156cdc91ab5f4c928fbd6"
|
|
"9148c35df5962fed381f4d8a62852a36823d5425f7487c13a12"
|
|
"523473fb823aa9d6ea5f42e794e15f2c1a8785cf6b7d51a4617"
|
|
"947fb3baf674f74a673cf1d38126983a19ed52c7439fab42c2185"
|
|
]
|
|
|
|
descr_tpl = '{} #{} N: \"{}\" A: \"{}\".'
|
|
|
|
def result(self) -> List[str]:
|
|
return [self.hex_x]
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [bignum_common.quote_str(n) for n in [self.hex_n,
|
|
self.hex_a,
|
|
self.hex_x]]
|
|
|
|
def description(self) -> str:
|
|
return self.descr_tpl.format(self.test_name,
|
|
self.count,
|
|
self.int_n,
|
|
self.int_a)
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
for bil in [32, 64]:
|
|
for n in cls.test_data_moduli:
|
|
for i in cls.test_input_numbers:
|
|
# Skip invalid combinations where A.limbs > N.limbs
|
|
if bignum_common.hex_to_int(i) > bignum_common.hex_to_int(n):
|
|
continue
|
|
yield cls(n, i, bits_in_limb=bil).create_test_case()
|
|
|
|
@property
|
|
def x(self) -> int: # pylint: disable=invalid-name
|
|
return (self.int_a * self.r) % self.int_n
|
|
|
|
@property
|
|
def hex_x(self) -> str:
|
|
return "{:x}".format(self.x).zfill(self.hex_digits)
|
|
|
|
class BignumModRawConvertFromMont(BignumModRawConvertToMont):
|
|
""" Test cases for mpi_mod_raw_from_mont_rep(). """
|
|
|
|
test_function = "mpi_mod_raw_from_mont_rep"
|
|
test_name = "Convert from Mont: "
|
|
|
|
test_input_numbers = ["0",
|
|
"1",
|
|
"3ca",
|
|
"539ed428",
|
|
"7dfe5c6beb35a2d6",
|
|
"dca8de1c2adfc6d7aafb9b48e",
|
|
"a7d17b6c4be72f3d5c16bf9c1af6fc933",
|
|
"2fec97beec546f9553142ed52f147845463f579",
|
|
"378dc83b8bc5a7b62cba495af4919578dce6d4f175cadc4f",
|
|
"b6415f2a1a8e48a518345db11f56db3829c8f2c6415ab4a395a"
|
|
"b3ac2ea4cbef4af86eb18a84eb6ded4c6ecbfc4b59c2879a675"
|
|
"487f687adea9d197a84a5242a5cf6125ce19a6ad2e7341f1c57"
|
|
"d43ea4f4c852a51cb63dabcd1c9de2b827a3146a3d175b35bea"
|
|
"41ae75d2a286a3e9d43623152ac513dcdea1d72a7da846a8ab3"
|
|
"58d9be4926c79cfb287cf1cf25b689de3b912176be5dcaf4d4c"
|
|
"6e7cb839a4a3243a6c47c1e2c99d65c59d6fa3672575c2f1ca8"
|
|
"de6a32e854ec9d8ec635c96af7679fce26d7d159e4a9da3bd74"
|
|
"e1272c376cd926d74fe3fb164a5935cff3d5cdb92b35fe2cea32"
|
|
"138a7e6bfbc319ebd1725dacb9a359cbf693f2ecb785efb9d627"
|
|
]
|
|
|
|
@property
|
|
def x(self): # pylint: disable=invalid-name
|
|
return (self.int_a * self.r_inv) % self.int_n
|
|
# END MERGE SLOT 7
|
|
|
|
# BEGIN MERGE SLOT 8
|
|
|
|
# END MERGE SLOT 8
|
|
|
|
# BEGIN MERGE SLOT 9
|
|
|
|
# END MERGE SLOT 9
|
|
|
|
# BEGIN MERGE SLOT 10
|
|
|
|
# END MERGE SLOT 10
|