896 lines
37 KiB
Python
896 lines
37 KiB
Python
"""Framework classes for generation of bignum core test cases."""
|
|
# Copyright The Mbed TLS Contributors
|
|
# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
#
|
|
|
|
import random
|
|
|
|
from typing import Dict, Iterator, List, Tuple
|
|
|
|
from . import test_case
|
|
from . import test_data_generation
|
|
from . import bignum_common
|
|
from .bignum_data import ADD_SUB_DATA
|
|
|
|
class BignumCoreTarget(test_data_generation.BaseTarget):
|
|
#pylint: disable=abstract-method, too-few-public-methods
|
|
"""Target for bignum core test case generation."""
|
|
target_basename = 'test_suite_bignum_core.generated'
|
|
|
|
|
|
class BignumCoreShiftR(BignumCoreTarget, test_data_generation.BaseTest):
|
|
"""Test cases for mbedtls_bignum_core_shift_r()."""
|
|
count = 0
|
|
test_function = "mpi_core_shift_r"
|
|
test_name = "Core shift right"
|
|
|
|
DATA = [
|
|
('00', '0', [0, 1, 8]),
|
|
('01', '1', [0, 1, 2, 8, 64]),
|
|
('dee5ca1a7ef10a75', '64-bit',
|
|
list(range(11)) + [31, 32, 33, 63, 64, 65, 71, 72]),
|
|
('002e7ab0070ad57001', '[leading 0 limb]',
|
|
[0, 1, 8, 63, 64]),
|
|
('a1055eb0bb1efa1150ff', '80-bit',
|
|
[0, 1, 8, 63, 64, 65, 72, 79, 80, 81, 88, 128, 129, 136]),
|
|
('020100000000000000001011121314151617', '138-bit',
|
|
[0, 1, 8, 9, 16, 72, 73, 136, 137, 138, 144]),
|
|
]
|
|
|
|
def __init__(self, input_hex: str, descr: str, count: int) -> None:
|
|
self.input_hex = input_hex
|
|
self.number_description = descr
|
|
self.shift_count = count
|
|
self.result = bignum_common.hex_to_int(input_hex) >> count
|
|
|
|
def arguments(self) -> List[str]:
|
|
return ['"{}"'.format(self.input_hex),
|
|
str(self.shift_count),
|
|
'"{:0{}x}"'.format(self.result, len(self.input_hex))]
|
|
|
|
def description(self) -> str:
|
|
return 'Core shift {} >> {}'.format(self.number_description,
|
|
self.shift_count)
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
for input_hex, descr, counts in cls.DATA:
|
|
for count in counts:
|
|
yield cls(input_hex, descr, count).create_test_case()
|
|
|
|
|
|
class BignumCoreShiftL(BignumCoreTarget, bignum_common.ModOperationCommon):
|
|
"""Test cases for mbedtls_bignum_core_shift_l()."""
|
|
|
|
BIT_SHIFT_VALUES = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a',
|
|
'1f', '20', '21', '3f', '40', '41', '47', '48', '4f',
|
|
'50', '51', '58', '80', '81', '88']
|
|
DATA = ["0", "1", "40", "dee5ca1a7ef10a75", "a1055eb0bb1efa1150ff",
|
|
"002e7ab0070ad57001", "020100000000000000001011121314151617",
|
|
"1946e2958a85d8863ae21f4904fcc49478412534ed53eaf321f63f2a222"
|
|
"7a3c63acbf50b6305595f90cfa8327f6db80d986fe96080bcbb5df1bdbe"
|
|
"9b74fb8dedf2bddb3f8215b54dffd66409323bcc473e45a8fe9d08e77a51"
|
|
"1698b5dad0416305db7fcf"]
|
|
arity = 1
|
|
test_function = "mpi_core_shift_l"
|
|
test_name = "Core shift(L)"
|
|
input_style = "arch_split"
|
|
symbol = "<<"
|
|
input_values = BIT_SHIFT_VALUES
|
|
moduli = DATA
|
|
|
|
@property
|
|
def val_n_max_limbs(self) -> int:
|
|
""" Return the limb count required to store the maximum number that can
|
|
fit in a the number of digits used by val_n """
|
|
m = bignum_common.hex_digits_max_int(self.val_n, self.bits_in_limb) - 1
|
|
return bignum_common.limbs_mpi(m, self.bits_in_limb)
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [bignum_common.quote_str(self.val_n),
|
|
str(self.int_a)
|
|
] + self.result()
|
|
|
|
def description(self) -> str:
|
|
""" Format the output as:
|
|
#{count} {hex input} ({input bits} {limbs capacity}) << {bit shift} """
|
|
bits = "({} bits in {} limbs)".format(self.int_n.bit_length(), self.val_n_max_limbs)
|
|
return "{} #{} {} {} {} {}".format(self.test_name,
|
|
self.count,
|
|
self.val_n,
|
|
bits,
|
|
self.symbol,
|
|
self.int_a)
|
|
|
|
def format_result(self, res: int) -> str:
|
|
# Override to match zero-pading for leading digits between the output and input.
|
|
res_str = bignum_common.zfill_match(self.val_n, "{:x}".format(res))
|
|
return bignum_common.quote_str(res_str)
|
|
|
|
def result(self) -> List[str]:
|
|
result = (self.int_n << self.int_a)
|
|
# Calculate if there is space for shifting to the left(leading zero limbs)
|
|
mx = bignum_common.hex_digits_max_int(self.val_n, self.bits_in_limb)
|
|
# If there are empty limbs ahead, adjust the bitmask accordingly
|
|
result = result & (mx - 1)
|
|
return [self.format_result(result)]
|
|
|
|
@property
|
|
def is_valid(self) -> bool:
|
|
return True
|
|
|
|
|
|
class BignumCoreCTLookup(BignumCoreTarget, test_data_generation.BaseTest):
|
|
"""Test cases for mbedtls_mpi_core_ct_uint_table_lookup()."""
|
|
test_function = "mpi_core_ct_uint_table_lookup"
|
|
test_name = "Constant time MPI table lookup"
|
|
|
|
bitsizes = [
|
|
(32, "One limb"),
|
|
(192, "Smallest curve sized"),
|
|
(512, "Largest curve sized"),
|
|
(2048, "Small FF/RSA sized"),
|
|
(4096, "Large FF/RSA sized"),
|
|
]
|
|
|
|
window_sizes = [0, 1, 2, 3, 4, 5, 6]
|
|
|
|
def __init__(self,
|
|
bitsize: int, descr: str, window_size: int) -> None:
|
|
self.bitsize = bitsize
|
|
self.bitsize_description = descr
|
|
self.window_size = window_size
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [str(self.bitsize), str(self.window_size)]
|
|
|
|
def description(self) -> str:
|
|
return '{} - {} MPI with {} bit window'.format(
|
|
BignumCoreCTLookup.test_name,
|
|
self.bitsize_description,
|
|
self.window_size
|
|
)
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
for bitsize, bitsize_description in cls.bitsizes:
|
|
for window_size in cls.window_sizes:
|
|
yield (cls(bitsize, bitsize_description, window_size)
|
|
.create_test_case())
|
|
|
|
|
|
class BignumCoreAddAndAddIf(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for bignum core add and add-if."""
|
|
count = 0
|
|
symbol = "+"
|
|
test_function = "mpi_core_add_and_add_if"
|
|
test_name = "mpi_core_add_and_add_if"
|
|
input_style = "arch_split"
|
|
input_values = ADD_SUB_DATA
|
|
unique_combinations_only = True
|
|
|
|
def result(self) -> List[str]:
|
|
result = self.int_a + self.int_b
|
|
|
|
carry, result = divmod(result, self.limb_boundary)
|
|
|
|
return [
|
|
self.format_result(result),
|
|
str(carry)
|
|
]
|
|
|
|
|
|
class BignumCoreSub(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for bignum core sub."""
|
|
count = 0
|
|
input_style = "arch_split"
|
|
symbol = "-"
|
|
test_function = "mpi_core_sub"
|
|
test_name = "mbedtls_mpi_core_sub"
|
|
input_values = ADD_SUB_DATA
|
|
|
|
def result(self) -> List[str]:
|
|
if self.int_a >= self.int_b:
|
|
result = self.int_a - self.int_b
|
|
carry = 0
|
|
else:
|
|
result = self.limb_boundary + self.int_a - self.int_b
|
|
carry = 1
|
|
return [
|
|
self.format_result(result),
|
|
str(carry)
|
|
]
|
|
|
|
|
|
class BignumCoreMLA(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for fixed-size multiply accumulate."""
|
|
count = 0
|
|
test_function = "mpi_core_mla"
|
|
test_name = "mbedtls_mpi_core_mla"
|
|
|
|
input_values = [
|
|
"0", "1", "fffe", "ffffffff", "100000000", "20000000000000",
|
|
"ffffffffffffffff", "10000000000000000", "1234567890abcdef0",
|
|
"fffffffffffffffffefefefefefefefe",
|
|
"100000000000000000000000000000000",
|
|
"1234567890abcdef01234567890abcdef0",
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff",
|
|
"1234567890abcdef01234567890abcdef01234567890abcdef01234567890abcdef0",
|
|
(
|
|
"4df72d07b4b71c8dacb6cffa954f8d88254b6277099308baf003fab73227f"
|
|
"34029643b5a263f66e0d3c3fa297ef71755efd53b8fb6cb812c6bbf7bcf17"
|
|
"9298bd9947c4c8b14324140a2c0f5fad7958a69050a987a6096e9f055fb38"
|
|
"edf0c5889eca4a0cfa99b45fbdeee4c696b328ddceae4723945901ec02507"
|
|
"6b12b"
|
|
)
|
|
] # type: List[str]
|
|
input_scalars = [
|
|
"0", "3", "fe", "ff", "ffff", "10000", "ffffffff", "100000000",
|
|
"7f7f7f7f7f7f7f7f", "8000000000000000", "fffffffffffffffe"
|
|
] # type: List[str]
|
|
|
|
def __init__(self, val_a: str, val_b: str, val_s: str) -> None:
|
|
super().__init__(val_a, val_b)
|
|
self.arg_scalar = val_s
|
|
self.int_scalar = bignum_common.hex_to_int(val_s)
|
|
if bignum_common.limbs_mpi(self.int_scalar, 32) > 1:
|
|
self.dependencies = ["MBEDTLS_HAVE_INT64"]
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [
|
|
bignum_common.quote_str(self.arg_a),
|
|
bignum_common.quote_str(self.arg_b),
|
|
bignum_common.quote_str(self.arg_scalar)
|
|
] + self.result()
|
|
|
|
def description(self) -> str:
|
|
"""Override and add the additional scalar."""
|
|
if not self.case_description:
|
|
self.case_description = "0x{} + 0x{} * 0x{}".format(
|
|
self.arg_a, self.arg_b, self.arg_scalar
|
|
)
|
|
return super().description()
|
|
|
|
def result(self) -> List[str]:
|
|
result = self.int_a + (self.int_b * self.int_scalar)
|
|
bound_val = max(self.int_a, self.int_b)
|
|
bound_4 = bignum_common.bound_mpi(bound_val, 32)
|
|
bound_8 = bignum_common.bound_mpi(bound_val, 64)
|
|
carry_4, remainder_4 = divmod(result, bound_4)
|
|
carry_8, remainder_8 = divmod(result, bound_8)
|
|
return [
|
|
"\"{:x}\"".format(remainder_4),
|
|
"\"{:x}\"".format(carry_4),
|
|
"\"{:x}\"".format(remainder_8),
|
|
"\"{:x}\"".format(carry_8)
|
|
]
|
|
|
|
@classmethod
|
|
def get_value_pairs(cls) -> Iterator[Tuple[str, str]]:
|
|
"""Generator to yield pairs of inputs.
|
|
|
|
Combinations are first generated from all input values, and then
|
|
specific cases provided.
|
|
"""
|
|
yield from super().get_value_pairs()
|
|
yield from cls.input_cases
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
"""Override for additional scalar input."""
|
|
for a_value, b_value in cls.get_value_pairs():
|
|
for s_value in cls.input_scalars:
|
|
cur_op = cls(a_value, b_value, s_value)
|
|
yield cur_op.create_test_case()
|
|
|
|
|
|
class BignumCoreMul(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for bignum core multiplication."""
|
|
count = 0
|
|
input_style = "arch_split"
|
|
symbol = "*"
|
|
test_function = "mpi_core_mul"
|
|
test_name = "mbedtls_mpi_core_mul"
|
|
arity = 2
|
|
unique_combinations_only = True
|
|
|
|
def format_arg(self, val: str) -> str:
|
|
return val
|
|
|
|
def format_result(self, res: int) -> str:
|
|
res_str = '{:x}'.format(res)
|
|
a_limbs = bignum_common.limbs_mpi(self.int_a, self.bits_in_limb)
|
|
b_limbs = bignum_common.limbs_mpi(self.int_b, self.bits_in_limb)
|
|
hex_digits = bignum_common.hex_digits_for_limb(a_limbs + b_limbs, self.bits_in_limb)
|
|
return bignum_common.quote_str(self.format_arg(res_str).zfill(hex_digits))
|
|
|
|
def result(self) -> List[str]:
|
|
result = self.int_a * self.int_b
|
|
return [self.format_result(result)]
|
|
|
|
|
|
class BignumCoreMontmul(BignumCoreTarget, test_data_generation.BaseTest):
|
|
"""Test cases for Montgomery multiplication."""
|
|
count = 0
|
|
test_function = "mpi_core_montmul"
|
|
test_name = "mbedtls_mpi_core_montmul"
|
|
|
|
start_2_mpi4 = False
|
|
start_2_mpi8 = False
|
|
|
|
replay_test_cases = [
|
|
(2, 1, 1, 1, "19", "1", "1D"), (2, 1, 1, 1, "7", "1", "9"),
|
|
(2, 1, 1, 1, "4", "1", "9"),
|
|
(
|
|
12, 1, 6, 1, (
|
|
"3C246D0E059A93A266288A7718419EC741661B474C58C032C5EDAF92709402"
|
|
"B07CC8C7CE0B781C641A1EA8DB2F4343"
|
|
), "1", (
|
|
"66A198186C18C10B2F5ED9B522752A9830B69916E535C8F047518A889A43A5"
|
|
"94B6BED27A168D31D4A52F88925AA8F5"
|
|
)
|
|
), (
|
|
8, 1, 4, 1,
|
|
"1E442976B0E63D64FCCE74B999E470CA9888165CB75BFA1F340E918CE03C6211",
|
|
"1", "B3A119602EE213CDE28581ECD892E0F592A338655DCE4CA88054B3D124D0E561"
|
|
), (
|
|
22, 1, 11, 1, (
|
|
"7CF5AC97304E0B63C65413F57249F59994B0FED1D2A8D3D83ED5FA38560FFB"
|
|
"82392870D6D08F87D711917FD7537E13B7E125BE407E74157776839B0AC9DB"
|
|
"23CBDFC696104353E4D2780B2B4968F8D8542306BCA7A2366E"
|
|
), "1", (
|
|
"284139EA19C139EBE09A8111926AAA39A2C2BE12ED487A809D3CB5BC558547"
|
|
"25B4CDCB5734C58F90B2F60D99CC1950CDBC8D651793E93C9C6F0EAD752500"
|
|
"A32C56C62082912B66132B2A6AA42ADA923E1AD22CEB7BA0123"
|
|
)
|
|
)
|
|
] # type: List[Tuple[int, int, int, int, str, str, str]]
|
|
|
|
random_test_cases = [
|
|
("2", "2", "3", ""), ("1", "2", "3", ""), ("2", "1", "3", ""),
|
|
("6", "5", "7", ""), ("3", "4", "7", ""), ("1", "6", "7", ""), ("5", "6", "7", ""),
|
|
("3", "4", "B", ""), ("7", "4", "B", ""), ("9", "7", "B", ""), ("2", "a", "B", ""),
|
|
("25", "16", "29", "(0x29 is prime)"), ("8", "28", "29", ""),
|
|
("18", "21", "29", ""), ("15", "f", "29", ""),
|
|
("e2", "ea", "FF", ""), ("43", "72", "FF", ""),
|
|
("d8", "70", "FF", ""), ("3c", "7c", "FF", ""),
|
|
("99", "b9", "101", "(0x101 is prime)"), ("65", "b2", "101", ""),
|
|
("81", "32", "101", ""), ("51", "dd", "101", ""),
|
|
("d5", "143", "38B", "(0x38B is prime)"), ("3d", "387", "38B", ""),
|
|
("160", "2e5", "38B", ""), ("10f", "137", "38B", ""),
|
|
("7dac", "25a", "8003", "(0x8003 is prime)"), ("6f1c", "3286", "8003", ""),
|
|
("59ed", "2f3f", "8003", ""), ("6893", "736d", "8003", ""),
|
|
("d199", "2832", "10001", "(0x10001 is prime)"), ("c3b2", "3e5b", "10001", ""),
|
|
("abe4", "214e", "10001", ""), ("4360", "a05d", "10001", ""),
|
|
("3f5a1", "165b2", "7F7F7", ""), ("3bd29", "37863", "7F7F7", ""),
|
|
("60c47", "64819", "7F7F7", ""), ("16584", "12c49", "7F7F7", ""),
|
|
("1ff03f", "610347", "800009", "(0x800009 is prime)"), ("340fd5", "19812e", "800009", ""),
|
|
("3fe2e8", "4d0dc7", "800009", ""), ("40356", "e6392", "800009", ""),
|
|
("dd8a1d", "266c0e", "100002B", "(0x100002B is prime)"),
|
|
("3fa1cb", "847fd6", "100002B", ""), ("5f439d", "5c3196", "100002B", ""),
|
|
("18d645", "f72dc6", "100002B", ""),
|
|
("20051ad", "37def6e", "37EEE9D", "(0x37EEE9D is prime)"),
|
|
("2ec140b", "3580dbf", "37EEE9D", ""), ("1d91b46", "190d4fc", "37EEE9D", ""),
|
|
("34e488d", "1224d24", "37EEE9D", ""),
|
|
("2a4fe2cb", "263466a9", "8000000B", "(0x8000000B is prime)"),
|
|
("5643fe94", "29a1aefa", "8000000B", ""), ("29633513", "7b007ac4", "8000000B", ""),
|
|
("2439cef5", "5c9d5a47", "8000000B", ""),
|
|
("4de3cfaa", "50dea178", "8CD626B9", "(0x8CD626B9 is prime)"),
|
|
("b8b8563", "10dbbbac", "8CD626B9", ""), ("4e8a6151", "5574ec19", "8CD626B9", ""),
|
|
("69224878", "309cfc23", "8CD626B9", ""),
|
|
("fb6f7fb6", "afb05423", "10000000F", "(0x10000000F is prime)"),
|
|
("8391a243", "26034dcd", "10000000F", ""), ("d26b98c", "14b2d6aa", "10000000F", ""),
|
|
("6b9f1371", "a21daf1d", "10000000F", ""),
|
|
(
|
|
"9f49435ad", "c8264ade8", "174876E7E9",
|
|
"0x174876E7E9 is prime (dec) 99999999977"
|
|
),
|
|
("c402da434", "1fb427acf", "174876E7E9", ""),
|
|
("f6ebc2bb1", "1096d39f2a", "174876E7E9", ""),
|
|
("153b7f7b6b", "878fda8ff", "174876E7E9", ""),
|
|
("2c1adbb8d6", "4384d2d3c6", "8000000017", "(0x8000000017 is prime)"),
|
|
("2e4f9cf5fb", "794f3443d9", "8000000017", ""),
|
|
("149e495582", "3802b8f7b7", "8000000017", ""),
|
|
("7b9d49df82", "69c68a442a", "8000000017", ""),
|
|
("683a134600", "6dd80ea9f6", "864CB9076D", "(0x864CB9076D is prime)"),
|
|
("13a870ff0d", "59b099694a", "864CB9076D", ""),
|
|
("37d06b0e63", "4d2147e46f", "864CB9076D", ""),
|
|
("661714f8f4", "22e55df507", "864CB9076D", ""),
|
|
("2f0a96363", "52693307b4", "F7F7F7F7F7", ""),
|
|
("3c85078e64", "f2275ecb6d", "F7F7F7F7F7", ""),
|
|
("352dae68d1", "707775b4c6", "F7F7F7F7F7", ""),
|
|
("37ae0f3e0b", "912113040f", "F7F7F7F7F7", ""),
|
|
("6dada15e31", "f58ed9eff7", "1000000000F", "(0x1000000000F is prime)"),
|
|
("69627a7c89", "cfb5ebd13d", "1000000000F", ""),
|
|
("a5e1ad239b", "afc030c731", "1000000000F", ""),
|
|
("f1cc45f4c5", "c64ad607c8", "1000000000F", ""),
|
|
("2ebad87d2e31", "4c72d90bca78", "800000000005", "(0x800000000005 is prime)"),
|
|
("a30b3cc50d", "29ac4fe59490", "800000000005", ""),
|
|
("33674e9647b4", "5ec7ee7e72d3", "800000000005", ""),
|
|
("3d956f474f61", "74070040257d", "800000000005", ""),
|
|
("48348e3717d6", "43fcb4399571", "800795D9BA47", "(0x800795D9BA47 is prime)"),
|
|
("5234c03cc99b", "2f3cccb87803", "800795D9BA47", ""),
|
|
("3ed13db194ab", "44b8f4ba7030", "800795D9BA47", ""),
|
|
("1c11e843bfdb", "95bd1b47b08", "800795D9BA47", ""),
|
|
("a81d11cb81fd", "1e5753a3f33d", "1000000000015", "(0x1000000000015 is prime)"),
|
|
("688c4db99232", "36fc0cf7ed", "1000000000015", ""),
|
|
("f0720cc07e07", "fc76140ed903", "1000000000015", ""),
|
|
("2ec61f8d17d1", "d270c85e36d2", "1000000000015", ""),
|
|
(
|
|
"6a24cd3ab63820", "ed4aad55e5e348", "100000000000051",
|
|
"(0x100000000000051 is prime)"
|
|
),
|
|
("e680c160d3b248", "31e0d8840ed510", "100000000000051", ""),
|
|
("a80637e9aebc38", "bb81decc4e1738", "100000000000051", ""),
|
|
("9afa5a59e9d630", "be9e65a6d42938", "100000000000051", ""),
|
|
("ab5e104eeb71c000", "2cffbd639e9fea00", "ABCDEF0123456789", ""),
|
|
("197b867547f68a00", "44b796cf94654800", "ABCDEF0123456789", ""),
|
|
("329f9483a04f2c00", "9892f76961d0f000", "ABCDEF0123456789", ""),
|
|
("4a2e12dfb4545000", "1aa3e89a69794500", "ABCDEF0123456789", ""),
|
|
(
|
|
"8b9acdf013d140f000", "12e4ceaefabdf2b2f00", "25A55A46E5DA99C71C7",
|
|
"0x25A55A46E5DA99C71C7 is the 3rd repunit prime(dec) 11111111111111111111111"
|
|
),
|
|
("1b8d960ea277e3f5500", "14418aa980e37dd000", "25A55A46E5DA99C71C7", ""),
|
|
("7314524977e8075980", "8172fa45618ccd0d80", "25A55A46E5DA99C71C7", ""),
|
|
("ca14f031769be63580", "147a2f3cf2964ca9400", "25A55A46E5DA99C71C7", ""),
|
|
(
|
|
"18532ba119d5cd0cf39735c0000", "25f9838e31634844924733000000",
|
|
"314DC643FB763F2B8C0E2DE00879",
|
|
"0x314DC643FB763F2B8C0E2DE00879 is (dec)99999999977^3"
|
|
),
|
|
(
|
|
"a56e2d2517519e3970e70c40000", "ec27428d4bb380458588fa80000",
|
|
"314DC643FB763F2B8C0E2DE00879", ""
|
|
),
|
|
(
|
|
"1cb5e8257710e8653fff33a00000", "15fdd42fe440fd3a1d121380000",
|
|
"314DC643FB763F2B8C0E2DE00879", ""
|
|
),
|
|
(
|
|
"e50d07a65fc6f93e538ce040000", "1f4b059ca609f3ce597f61240000",
|
|
"314DC643FB763F2B8C0E2DE00879", ""
|
|
),
|
|
(
|
|
"1ea3ade786a095d978d387f30df9f20000000",
|
|
"127c448575f04af5a367a7be06c7da0000000",
|
|
"47BF19662275FA2F6845C74942ED1D852E521",
|
|
"0x47BF19662275FA2F6845C74942ED1D852E521 is (dec) 99999999977^4"
|
|
),
|
|
(
|
|
"16e15b0ca82764e72e38357b1f10a20000000",
|
|
"43e2355d8514bbe22b0838fdc3983a0000000",
|
|
"47BF19662275FA2F6845C74942ED1D852E521", ""
|
|
),
|
|
(
|
|
"be39332529d93f25c3d116c004c620000000",
|
|
"5cccec42370a0a2c89c6772da801a0000000",
|
|
"47BF19662275FA2F6845C74942ED1D852E521", ""
|
|
),
|
|
(
|
|
"ecaa468d90de0eeda474d39b3e1fc0000000",
|
|
"1e714554018de6dc0fe576bfd3b5660000000",
|
|
"47BF19662275FA2F6845C74942ED1D852E521", ""
|
|
),
|
|
(
|
|
"32298816711c5dce46f9ba06e775c4bedfc770e6700000000000000",
|
|
"8ee751fd5fb24f0b4a653cb3a0c8b7d9e724574d168000000000000",
|
|
"97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931",
|
|
(
|
|
"0x97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931"
|
|
" is (dec) 99999999977^6"
|
|
)
|
|
),
|
|
(
|
|
"29213b9df3cfd15f4b428645b67b677c29d1378d810000000000000",
|
|
"6cbb732c65e10a28872394dfdd1936d5171c3c3aac0000000000000",
|
|
"97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931", ""
|
|
),
|
|
(
|
|
"6f18db06ad4abc52c0c50643dd13098abccd4a232f0000000000000",
|
|
"7e6bf41f2a86098ad51f98dfc10490ba3e8081bc830000000000000",
|
|
"97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931", ""
|
|
),
|
|
(
|
|
"62d3286cd706ad9d73caff63f1722775d7e8c731208000000000000",
|
|
"530f7ba02ae2b04c2fe3e3d27ec095925631a6c2528000000000000",
|
|
"97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931", ""
|
|
),
|
|
(
|
|
"a6c6503e3c031fdbf6009a89ed60582b7233c5a85de28b16000000000000000",
|
|
"75c8ed18270b583f16d442a467d32bf95c5e491e9b8523798000000000000000",
|
|
"DD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499",
|
|
(
|
|
"0xDD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499"
|
|
" is (dec) 99999999977^7"
|
|
)
|
|
),
|
|
(
|
|
"bf84d1f85cf6b51e04d2c8f4ffd03532d852053cf99b387d4000000000000000",
|
|
"397ba5a743c349f4f28bc583ecd5f06e0a25f9c6d98f09134000000000000000",
|
|
"DD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499", ""
|
|
),
|
|
(
|
|
"6db11c3a4152ed1a2aa6fa34b0903ec82ea1b88908dcb482000000000000000",
|
|
"ac8ac576a74ad6ca48f201bf89f77350ce86e821358d85920000000000000000",
|
|
"DD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499", ""
|
|
),
|
|
(
|
|
"3001d96d7fe8b733f33687646fc3017e3ac417eb32e0ec708000000000000000",
|
|
"925ddbdac4174e8321a48a32f79640e8cf7ec6f46ea235a80000000000000000",
|
|
"DD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499", ""
|
|
),
|
|
(
|
|
"1029048755f2e60dd98c8de6d9989226b6bb4f0db8e46bd1939de560000000000000000000",
|
|
"51bb7270b2e25cec0301a03e8275213bb6c2f6e6ec93d4d46d36ca0000000000000000000",
|
|
"141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146380E41",
|
|
(
|
|
"0x141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146"
|
|
"380E41 is 99999999977^8"
|
|
)
|
|
),
|
|
(
|
|
"1c5337ff982b3ad6611257dbff5bbd7a9920ba2d4f5838a0cc681ce000000000000000000",
|
|
"520c5d049ca4702031ba728591b665c4d4ccd3b2b86864d4c160fd2000000000000000000",
|
|
"141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146380E41",
|
|
""
|
|
),
|
|
(
|
|
"57074dfa00e42f6555bae624b7f0209f218adf57f73ed34ab0ff90c000000000000000000",
|
|
"41eb14b6c07bfd3d1fe4f4a610c17cc44fcfcda695db040e011065000000000000000000",
|
|
"141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146380E41",
|
|
""
|
|
),
|
|
(
|
|
"d8ed7feed2fe855e6997ad6397f776158573d425031bf085a615784000000000000000000",
|
|
"6f121dcd18c578ab5e229881006007bb6d319b179f11015fe958b9c000000000000000000",
|
|
"141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146380E41",
|
|
""
|
|
),
|
|
(
|
|
(
|
|
"2a462b156180ea5fe550d3758c764e06fae54e626b5f503265a09df76edbdfbf"
|
|
"a1e6000000000000000000000000"
|
|
), (
|
|
"1136f41d1879fd4fb9e49e0943a46b6704d77c068ee237c3121f9071cfd3e6a0"
|
|
"0315800000000000000000000000"
|
|
), (
|
|
"2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E2833EC90"
|
|
"2713E40F51E3B3C214EDFABC451"
|
|
), (
|
|
"0x2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E2833EC"
|
|
"902713E40F51E3B3C214EDFABC451 is (dec) 99999999977^10"
|
|
)
|
|
),
|
|
(
|
|
(
|
|
"c1ac3800dfb3c6954dea391d206200cf3c47f795bf4a5603b4cb88ae7e574de47"
|
|
"40800000000000000000000000"
|
|
), (
|
|
"c0d16eda0549ede42fa0deb4635f7b7ce061fadea02ee4d85cba4c4f709603419"
|
|
"3c800000000000000000000000"
|
|
), (
|
|
"2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E2833EC90"
|
|
"2713E40F51E3B3C214EDFABC451"
|
|
), ""
|
|
),
|
|
(
|
|
(
|
|
"19e45bb7633094d272588ad2e43bcb3ee341991c6731b6fa9d47c4018d7ce7bba"
|
|
"5ee800000000000000000000000"
|
|
), (
|
|
"1e4f83166ae59f6b9cc8fd3e7677ed8bfc01bb99c98bd3eb084246b64c1e18c33"
|
|
"65b800000000000000000000000"
|
|
), (
|
|
"2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E2833EC90"
|
|
"2713E40F51E3B3C214EDFABC451"
|
|
), ""
|
|
),
|
|
(
|
|
(
|
|
"1aa93395fad5f9b7f20b8f9028a054c0bb7c11bb8520e6a95e5a34f06cb70bcdd"
|
|
"01a800000000000000000000000"
|
|
), (
|
|
"54b45afa5d4310192f8d224634242dd7dcfb342318df3d9bd37b4c614788ba13b"
|
|
"8b000000000000000000000000"
|
|
), (
|
|
"2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E2833EC90"
|
|
"2713E40F51E3B3C214EDFABC451"
|
|
), ""
|
|
),
|
|
(
|
|
(
|
|
"544f2628a28cfb5ce0a1b7180ee66b49716f1d9476c466c57f0c4b23089917843"
|
|
"06d48f78686115ee19e25400000000000000000000000000000000"
|
|
), (
|
|
"677eb31ef8d66c120fa872a60cd47f6e10cbfdf94f90501bd7883cba03d185be0"
|
|
"a0148d1625745e9c4c827300000000000000000000000000000000"
|
|
), (
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA1"
|
|
"1DABD6E6144BEF37C6800000000000000000000000000000000051"
|
|
), (
|
|
"0x8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBF"
|
|
"A11DABD6E6144BEF37C6800000000000000000000000000000000051 is prime,"
|
|
" (dec) 10^143 + 3^4"
|
|
)
|
|
),
|
|
(
|
|
(
|
|
"76bb3470985174915e9993522aec989666908f9e8cf5cb9f037bf4aee33d8865c"
|
|
"b6464174795d07e30015b80000000000000000000000000000000"
|
|
), (
|
|
"6aaaf60d5784dcef612d133613b179a317532ecca0eed40b8ad0c01e6d4a6d8c7"
|
|
"9a52af190abd51739009a900000000000000000000000000000000"
|
|
), (
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA1"
|
|
"1DABD6E6144BEF37C6800000000000000000000000000000000051"
|
|
), ""
|
|
),
|
|
(
|
|
(
|
|
"6cfdd6e60912e441d2d1fc88f421b533f0103a5322ccd3f4db84861643ad63fd6"
|
|
"3d1d8cfbc1d498162786ba00000000000000000000000000000000"
|
|
), (
|
|
"1177246ec5e93814816465e7f8f248b350d954439d35b2b5d75d917218e7fd5fb"
|
|
"4c2f6d0667f9467fdcf33400000000000000000000000000000000"
|
|
), (
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA1"
|
|
"1DABD6E6144BEF37C6800000000000000000000000000000000051"
|
|
), ""
|
|
),
|
|
(
|
|
(
|
|
"7a09a0b0f8bbf8057116fb0277a9bdf3a91b5eaa8830d448081510d8973888be5"
|
|
"a9f0ad04facb69aa3715f00000000000000000000000000000000"
|
|
), (
|
|
"764dec6c05a1c0d87b649efa5fd94c91ea28bffb4725d4ab4b33f1a3e8e3b314d"
|
|
"799020e244a835a145ec9800000000000000000000000000000000"
|
|
), (
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA1"
|
|
"1DABD6E6144BEF37C6800000000000000000000000000000000051"
|
|
), ""
|
|
)
|
|
] # type: List[Tuple[str, str, str, str]]
|
|
|
|
def __init__(
|
|
self, val_a: str, val_b: str, val_n: str, case_description: str = ""
|
|
):
|
|
self.case_description = case_description
|
|
self.arg_a = val_a
|
|
self.int_a = bignum_common.hex_to_int(val_a)
|
|
self.arg_b = val_b
|
|
self.int_b = bignum_common.hex_to_int(val_b)
|
|
self.arg_n = val_n
|
|
self.int_n = bignum_common.hex_to_int(val_n)
|
|
|
|
limbs_a4 = bignum_common.limbs_mpi(self.int_a, 32)
|
|
limbs_a8 = bignum_common.limbs_mpi(self.int_a, 64)
|
|
self.limbs_b4 = bignum_common.limbs_mpi(self.int_b, 32)
|
|
self.limbs_b8 = bignum_common.limbs_mpi(self.int_b, 64)
|
|
self.limbs_an4 = bignum_common.limbs_mpi(self.int_n, 32)
|
|
self.limbs_an8 = bignum_common.limbs_mpi(self.int_n, 64)
|
|
|
|
if limbs_a4 > self.limbs_an4 or limbs_a8 > self.limbs_an8:
|
|
raise Exception("Limbs of input A ({}) exceeds N ({})".format(
|
|
self.arg_a, self.arg_n
|
|
))
|
|
|
|
def arguments(self) -> List[str]:
|
|
return [
|
|
str(self.limbs_an4), str(self.limbs_b4),
|
|
str(self.limbs_an8), str(self.limbs_b8),
|
|
bignum_common.quote_str(self.arg_a),
|
|
bignum_common.quote_str(self.arg_b),
|
|
bignum_common.quote_str(self.arg_n)
|
|
] + self.result()
|
|
|
|
def description(self) -> str:
|
|
if self.case_description != "replay":
|
|
if not self.start_2_mpi4 and self.limbs_an4 > 1:
|
|
tmp = "(start of 2-MPI 4-byte bignums) "
|
|
self.__class__.start_2_mpi4 = True
|
|
elif not self.start_2_mpi8 and self.limbs_an8 > 1:
|
|
tmp = "(start of 2-MPI 8-byte bignums) "
|
|
self.__class__.start_2_mpi8 = True
|
|
else:
|
|
tmp = "(gen) "
|
|
self.case_description = tmp + self.case_description
|
|
return super().description()
|
|
|
|
def result(self) -> List[str]:
|
|
"""Get the result of the operation."""
|
|
r4 = bignum_common.bound_mpi_limbs(self.limbs_an4, 32)
|
|
i4 = bignum_common.invmod(r4, self.int_n)
|
|
x4 = self.int_a * self.int_b * i4
|
|
x4 = x4 % self.int_n
|
|
|
|
r8 = bignum_common.bound_mpi_limbs(self.limbs_an8, 64)
|
|
i8 = bignum_common.invmod(r8, self.int_n)
|
|
x8 = self.int_a * self.int_b * i8
|
|
x8 = x8 % self.int_n
|
|
return [
|
|
"\"{:x}\"".format(x4),
|
|
"\"{:x}\"".format(x8)
|
|
]
|
|
|
|
def set_limbs(
|
|
self, limbs_an4: int, limbs_b4: int, limbs_an8: int, limbs_b8: int
|
|
) -> None:
|
|
"""Set number of limbs for each input.
|
|
|
|
Replaces default values set during initialization.
|
|
"""
|
|
self.limbs_an4 = limbs_an4
|
|
self.limbs_b4 = limbs_b4
|
|
self.limbs_an8 = limbs_an8
|
|
self.limbs_b8 = limbs_b8
|
|
|
|
@classmethod
|
|
def generate_function_tests(cls) -> Iterator[test_case.TestCase]:
|
|
"""Generate replay and randomly generated test cases."""
|
|
# Test cases which replay captured invocations during unit test runs.
|
|
for limbs_an4, limbs_b4, limbs_an8, limbs_b8, a, b, n in cls.replay_test_cases:
|
|
cur_op = cls(a, b, n, case_description="replay")
|
|
cur_op.set_limbs(limbs_an4, limbs_b4, limbs_an8, limbs_b8)
|
|
yield cur_op.create_test_case()
|
|
# Random test cases can be generated using mpi_modmul_case_generate()
|
|
# Uses a mixture of primes and odd numbers as N, with four randomly
|
|
# generated cases for each N.
|
|
for a, b, n, description in cls.random_test_cases:
|
|
cur_op = cls(a, b, n, case_description=description)
|
|
yield cur_op.create_test_case()
|
|
|
|
|
|
def mpi_modmul_case_generate() -> None:
|
|
"""Generate valid inputs for montmul tests using moduli.
|
|
|
|
For each modulus, generates random values for A and B and simple descriptions
|
|
for the test case.
|
|
"""
|
|
moduli = [
|
|
("3", ""), ("7", ""), ("B", ""), ("29", ""), ("FF", ""),
|
|
("101", ""), ("38B", ""), ("8003", ""), ("10001", ""),
|
|
("7F7F7", ""), ("800009", ""), ("100002B", ""), ("37EEE9D", ""),
|
|
("8000000B", ""), ("8CD626B9", ""), ("10000000F", ""),
|
|
("174876E7E9", "is prime (dec) 99999999977"),
|
|
("8000000017", ""), ("864CB9076D", ""), ("F7F7F7F7F7", ""),
|
|
("1000000000F", ""), ("800000000005", ""), ("800795D9BA47", ""),
|
|
("1000000000015", ""), ("100000000000051", ""), ("ABCDEF0123456789", ""),
|
|
(
|
|
"25A55A46E5DA99C71C7",
|
|
"is the 3rd repunit prime (dec) 11111111111111111111111"
|
|
),
|
|
("314DC643FB763F2B8C0E2DE00879", "is (dec)99999999977^3"),
|
|
("47BF19662275FA2F6845C74942ED1D852E521", "is (dec) 99999999977^4"),
|
|
(
|
|
"97EDD86E4B5C4592C6D32064AC55C888A7245F07CA3CC455E07C931",
|
|
"is (dec) 99999999977^6"
|
|
),
|
|
(
|
|
"DD15FE80B731872AC104DB37832F7E75A244AA2631BC87885B861E8F20375499",
|
|
"is (dec) 99999999977^7"
|
|
),
|
|
(
|
|
"141B8EBD9009F84C241879A1F680FACCED355DA36C498F73E96E880CF78EA5F96146380E41",
|
|
"is (dec) 99999999977^8"
|
|
),
|
|
(
|
|
(
|
|
"2A94608DE88B6D5E9F8920F5ABB06B24CC35AE1FBACC87D075C621C3E283"
|
|
"3EC902713E40F51E3B3C214EDFABC451"
|
|
),
|
|
"is (dec) 99999999977^10"
|
|
),
|
|
(
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA11"
|
|
"DABD6E6144BEF37C6800000000000000000000000000000000051",
|
|
"is prime, (dec) 10^143 + 3^4"
|
|
)
|
|
] # type: List[Tuple[str, str]]
|
|
primes = [
|
|
"3", "7", "B", "29", "101", "38B", "8003", "10001", "800009",
|
|
"100002B", "37EEE9D", "8000000B", "8CD626B9",
|
|
# From here they require > 1 4-byte MPI
|
|
"10000000F", "174876E7E9", "8000000017", "864CB9076D", "1000000000F",
|
|
"800000000005", "800795D9BA47", "1000000000015", "100000000000051",
|
|
# From here they require > 1 8-byte MPI
|
|
"25A55A46E5DA99C71C7", # this is 11111111111111111111111 decimal
|
|
# 10^143 + 3^4: (which is prime)
|
|
# 100000000000000000000000000000000000000000000000000000000000000000000000000000
|
|
# 000000000000000000000000000000000000000000000000000000000000000081
|
|
(
|
|
"8335616AED761F1F7F44E6BD49E807B82E3BF2BF11BFA6AF813C808DBF33DBFA11"
|
|
"DABD6E6144BEF37C6800000000000000000000000000000000051"
|
|
)
|
|
] # type: List[str]
|
|
generated_inputs = []
|
|
for mod, description in moduli:
|
|
n = bignum_common.hex_to_int(mod)
|
|
mod_read = "{:x}".format(n)
|
|
case_count = 3 if n < 5 else 4
|
|
cases = {} # type: Dict[int, int]
|
|
i = 0
|
|
while i < case_count:
|
|
a = random.randint(1, n)
|
|
b = random.randint(1, n)
|
|
if cases.get(a) == b:
|
|
continue
|
|
cases[a] = b
|
|
if description:
|
|
out_description = "0x{} {}".format(mod_read, description)
|
|
elif i == 0 and len(mod) > 1 and mod in primes:
|
|
out_description = "(0x{} is prime)"
|
|
else:
|
|
out_description = ""
|
|
generated_inputs.append(
|
|
("{:x}".format(a), "{:x}".format(b), mod, out_description)
|
|
)
|
|
i += 1
|
|
print(generated_inputs)
|
|
|
|
|
|
class BignumCoreExpMod(BignumCoreTarget, bignum_common.ModOperationCommon):
|
|
"""Test cases for bignum core exponentiation."""
|
|
symbol = "^"
|
|
test_function = "mpi_core_exp_mod"
|
|
test_name = "Core modular exponentiation (Mongtomery form only)"
|
|
input_style = "fixed"
|
|
montgomery_form_a = True
|
|
|
|
def result(self) -> List[str]:
|
|
# Result has to be given in Montgomery form too
|
|
result = pow(self.int_a, self.int_b, self.int_n)
|
|
mont_result = self.to_montgomery(result)
|
|
return [self.format_result(mont_result)]
|
|
|
|
@property
|
|
def is_valid(self) -> bool:
|
|
# The base needs to be canonical, but the exponent can be larger than
|
|
# the modulus (see for example exponent blinding)
|
|
return bool(self.int_a < self.int_n)
|
|
|
|
|
|
class BignumCoreSubInt(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for bignum core sub int."""
|
|
count = 0
|
|
symbol = "-"
|
|
test_function = "mpi_core_sub_int"
|
|
test_name = "mpi_core_sub_int"
|
|
input_style = "arch_split"
|
|
|
|
@property
|
|
def is_valid(self) -> bool:
|
|
# This is "sub int", so b is only one limb
|
|
if bignum_common.limbs_mpi(self.int_b, self.bits_in_limb) > 1:
|
|
return False
|
|
return True
|
|
|
|
# Overriding because we don't want leading zeros on b
|
|
@property
|
|
def arg_b(self) -> str:
|
|
return self.val_b
|
|
|
|
def result(self) -> List[str]:
|
|
result = self.int_a - self.int_b
|
|
|
|
borrow, result = divmod(result, self.limb_boundary)
|
|
|
|
# Borrow will be -1 if non-zero, but we want it to be 1 in the test data
|
|
return [
|
|
self.format_result(result),
|
|
str(-borrow)
|
|
]
|
|
|
|
class BignumCoreZeroCheckCT(BignumCoreTarget, bignum_common.OperationCommon):
|
|
"""Test cases for bignum core zero check (constant flow)."""
|
|
count = 0
|
|
symbol = "== 0"
|
|
test_function = "mpi_core_check_zero_ct"
|
|
test_name = "mpi_core_check_zero_ct"
|
|
input_style = "variable"
|
|
arity = 1
|
|
suffix = True
|
|
|
|
def result(self) -> List[str]:
|
|
result = 1 if self.int_a == 0 else 0
|
|
return [str(result)]
|