183 lines
5.9 KiB
Python
183 lines
5.9 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 typing import Dict, List
|
|
|
|
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'
|
|
|
|
# BEGIN MERGE SLOT 1
|
|
|
|
# END MERGE SLOT 1
|
|
|
|
# BEGIN MERGE SLOT 2
|
|
|
|
class BignumModRawSub(BignumModRawOperation):
|
|
"""Test cases for bignum mod raw sub."""
|
|
count = 0
|
|
symbol = "-"
|
|
test_function = "mpi_mod_raw_sub"
|
|
test_name = "mbedtls_mpi_mod_raw_sub"
|
|
unique_combinations_only = False
|
|
|
|
input_values = [
|
|
"0", "1", "fe", "ff", "fffe", "ffff",
|
|
"fffffffffffffffe", "ffffffffffffffff",
|
|
"fffffffffffffffffffffffffffffffe",
|
|
"ffffffffffffffffffffffffffffffff",
|
|
"1234567890abcdef01234567890abcdef0",
|
|
"3653f8dd9b1f282e4067c3584ee207f8da94e3e8ab73738f",
|
|
"fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffe",
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff",
|
|
"1234567890abcdef01234567890abcdef01234567890abcdef01234567890abcdef0",
|
|
(
|
|
"14c15c910b11ad28cc21ce88d0060cc54278c2614e1bcb383bb4a570294c4ea3"
|
|
"738d243a6e58d5ca49c7b59b995253fd6c79a3de69f85e3131f3b9238224b122"
|
|
"c3e4a892d9196ada4fcfa583e1df8af9b474c7e89286a1754abcb06ae8abb93f"
|
|
"01d89a024cdce7a6d7288ff68c320f89f1347e0cdd905ecfd160c5d0ef412ed6"
|
|
)
|
|
]
|
|
|
|
modulus_values = [
|
|
"7", "ff",
|
|
"d1c127a667786703830500038ebaef20e5a3e2dc378fb75b"
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff43",
|
|
"fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff67",
|
|
(
|
|
"c93ba7ec74d96f411ba008bdb78e63ff11bb5df46a51e16b2c9d156f8e4e18ab"
|
|
"f5e052cb01f47d0d1925a77f60991577e128fb6f52f34a27950a594baadd3d80"
|
|
"57abeb222cf3cca962db16abf79f2ada5bd29ab2f51244bf295eff9f6aaba130"
|
|
"2efc449b128be75eeaca04bc3c1a155d11d14e8be32a2c8287b3996cf6ad5223"
|
|
),
|
|
(
|
|
"5c083126e978d4fdf3b645a1cac083126e978d4fdf3b645a1cac083126e978d4"
|
|
"fdf3b645a1cac083126e978d4fdf3b645a1cac083126e978d4fdf3b645a1cac0"
|
|
"83126e978d4fdf3b645a1cac083126e978d4fdf3b645a1cac083126e978d4fdf"
|
|
"3b645a1cac083126e978d4fdf3b645a1cac083126e978d4fdf3b645a1cac05d2"
|
|
)
|
|
]
|
|
|
|
descr_tpl = '{} #{} \"{}\" - \"{}\" % \"{}\".'
|
|
|
|
BITS_IN_LIMB = 32
|
|
|
|
@property
|
|
def boundary(self) -> int:
|
|
return self.int_n
|
|
|
|
@property
|
|
def x(self): # pylint: disable=invalid-name
|
|
return (self.int_a - self.int_b) % self.int_n if self.int_n > 0 else 0
|
|
|
|
@property
|
|
def hex_x(self) -> str:
|
|
return format(self.x, 'x').zfill(self.hex_digits)
|
|
|
|
def description(self) -> str:
|
|
return self.descr_tpl.format(self.test_name,
|
|
self.count,
|
|
self.int_a,
|
|
self.int_b,
|
|
self.int_n)
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [bignum_common.quote_str(n) for n in [self.hex_a,
|
|
self.hex_b,
|
|
self.hex_n,
|
|
self.hex_x]]
|
|
|
|
def result(self) -> List[str]:
|
|
return [self.hex_x]
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
for a_value, b_value in cls.get_value_pairs():
|
|
int_a = bignum_common.hex_to_int(a_value)
|
|
int_b = bignum_common.hex_to_int(b_value)
|
|
highest = max(int_a, int_b)
|
|
|
|
# Choose a modulus bigger then the arguments
|
|
for n_value in cls.modulus_values:
|
|
int_n = bignum_common.hex_to_int(n_value)
|
|
if highest < int_n:
|
|
yield cls(n_value, a_value, b_value, cls.BITS_IN_LIMB).create_test_case()
|
|
|
|
# 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(bignum_common.ModOperationCommon,
|
|
BignumModRawTarget):
|
|
""" Test cases for mpi_mod_raw_to_mont_rep(). """
|
|
test_function = "mpi_mod_raw_to_mont_rep"
|
|
test_name = "Convert into Mont: "
|
|
symbol = "R *"
|
|
input_style = "arch_split"
|
|
arity = 1
|
|
|
|
def result(self) -> List[str]:
|
|
result = (self.int_a * self.r) % self.int_n
|
|
return [self.format_result(result)]
|
|
|
|
|
|
class BignumModRawConvertFromMont(bignum_common.ModOperationCommon,
|
|
BignumModRawTarget):
|
|
""" Test cases for mpi_mod_raw_from_mont_rep(). """
|
|
test_function = "mpi_mod_raw_from_mont_rep"
|
|
test_name = "Convert from Mont: "
|
|
symbol = "1/R *"
|
|
input_style = "arch_split"
|
|
arity = 1
|
|
|
|
def result(self) -> List[str]:
|
|
result = (self.int_a * self.r_inv) % self.int_n
|
|
return [self.format_result(result)]
|
|
|
|
|
|
# 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
|