From 7acfaa0a687530ca195769644dbbc82b9bf3ed08 Mon Sep 17 00:00:00 2001 From: andrr3j Date: Tue, 18 Jul 2023 19:53:00 +0200 Subject: simulator tests --- test/test_impl.py | 2 +- test/test_simulator.py | 224 +++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 225 insertions(+), 1 deletion(-) create mode 100644 test/test_simulator.py (limited to 'test') diff --git a/test/test_impl.py b/test/test_impl.py index ea92f8a..b054567 100644 --- a/test/test_impl.py +++ b/test/test_impl.py @@ -255,4 +255,4 @@ class ECDSATests(ImplTests): self.do_ecdsa_test(runner, self.secp128r1, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], "ltr", complete=False, always=True) self.do_ecdsa_test(runner, self.secp128r1, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], - "ltr", complete=True, always=True) + "ltr", complete=True, always=True) \ No newline at end of file diff --git a/test/test_simulator.py b/test/test_simulator.py new file mode 100644 index 0000000..eb7f238 --- /dev/null +++ b/test/test_simulator.py @@ -0,0 +1,224 @@ +from copy import copy +from os.path import join +from unittest import TestCase + +from click.testing import CliRunner +from pyecsca.ec.key_agreement import ECDH_SHA1 +from pyecsca.ec.mult import LTRMultiplier, RTLMultiplier, CoronMultiplier, BinaryNAFMultiplier +from pyecsca.ec.signature import ECDSA_SHA1, SignatureResult + +from pyecsca.codegen.builder import build_impl +from pyecsca.codegen.client import SimulatorTarget + +from pyecsca.ec.curve import EllipticCurve +from pyecsca.ec.mod import Mod +from pyecsca.ec.model import ShortWeierstrassModel +from pyecsca.ec.params import DomainParameters +from pyecsca.ec.point import InfinityPoint, Point + + +class SimulatorTest(TestCase): + + def setUp(self): + model = ShortWeierstrassModel() + coords = model.coordinates["projective"] + p = 0xd7d1247f + a = Mod(0xa4a44016, p) + b = Mod(0x73f76716, p) + n = 0xd7d2a475 + h = 1 + gx, gy, gz = Mod(0x54eed6d7, p), Mod(0x6f1e55ac, p), Mod(1, p) + generator = Point(coords, X=gx, Y=gy, Z=gz) + neutral = InfinityPoint(coords) + + curve = EllipticCurve(model, coords, p, neutral, {"a": a, "b": b}) + self.curve32 = DomainParameters(curve, generator, n, h) + self.base = self.curve32.generator + self.coords = self.curve32.curve.coordinate_model + + def do_basic_test(self, callback, runner, params, mult_class, formulas, mult_name, + ecdsa, ecdh): + with runner.isolated_filesystem() as tmpdir: + runner.invoke(build_impl, + ["--platform", "STM32F3", + "--ecdsa" if ecdsa else "--no-ecdsa", + "--ecdh" if ecdh else "--no-ecdh", + params.curve.model.shortname, params.curve.coordinate_model.name, + *formulas, + f"{mult_name}()", + "."]) + target = SimulatorTarget(params.curve.model, params.curve.coordinate_model) + target.connect(binary=join(tmpdir, "pyecsca-codegen-CW308_STM32F3.elf")) + target.set_params(params) + formula_instances = [params.curve.coordinate_model.formulas[formula] for formula + in formulas] + mult = mult_class(*formula_instances) + mult.init(params, params.generator) + callback(target, mult, params) + target.disconnect() + + +class PRNGTests(SimulatorTest): + + def test_init(self): + runner = CliRunner() + + def callback(target, mult, params): + target.init_prng(bytes([0x12, 0x34, 0x56, 0x78])) + + self.do_basic_test(callback, runner, self.curve32, LTRMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "ltr", False, False) + + +class SetupTests(SimulatorTest): + + def test_setup(self): + runner = CliRunner() + + def callback(target, mult, params): + priv = 57 + pub = mult.multiply(priv).to_affine() + target.set_privkey(priv) + target.set_pubkey(pub) + + self.do_basic_test(callback, runner, self.curve32, LTRMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "ltr", False, False) + + def test_debug(self): + runner = CliRunner() + + def callback(target, mult, params): + model, coords = target.debug() + self.assertEqual(model, params.curve.model.shortname) + self.assertEqual(coords, params.curve.coordinate_model.name) + + self.do_basic_test(callback, runner, self.curve32, LTRMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "ltr", False, False) + + +class KeyGenerationTests(SimulatorTest): + + def do_keygen_test(self, runner, params, mult_class, formulas, mult_name): + def callback(target, mult, params): + for _ in range(1): + priv, pub = target.generate() + self.assertTrue(params.curve.is_on_curve(pub)) + expected = mult.multiply(priv).to_affine() + self.assertEqual(pub, expected) + + self.do_basic_test(callback, runner, params, mult_class, formulas, mult_name, False, False) + + def test_ltr(self): + runner = CliRunner() + self.do_keygen_test(runner, self.curve32, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "ltr") + + def test_rtl(self): + runner = CliRunner() + self.do_keygen_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "rtl") + + def test_coron(self): + runner = CliRunner() + self.do_keygen_test(runner, self.curve32, CoronMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "coron") + + def test_bnaf(self): + runner = CliRunner() + self.do_keygen_test(runner, self.curve32, BinaryNAFMultiplier, + ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") + + +class ScalarMultiplicationTests(SimulatorTest): + + def do_mult_test(self, runner, params, mult_class, formulas, mult_name): + #values = [15, 2355498743, 3253857901321912443757746] + values = [15] + + def callback(target, mult, params): + for value in values: + result = target.scalar_mult(value, params.generator) + expected = mult.multiply(value) + self.assertEqual(result, expected) + + self.do_basic_test(callback, runner, params, mult_class, formulas, mult_name, False, False) + + def test_ltr(self): + runner = CliRunner() + self.do_mult_test(runner, self.curve32, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "ltr") + + def test_rtl(self): + runner = CliRunner() + self.do_mult_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "rtl") + + def test_coron(self): + runner = CliRunner() + self.do_mult_test(runner, self.curve32, CoronMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "coron") + + def test_bnaf(self): + runner = CliRunner() + self.do_mult_test(runner, self.curve32, BinaryNAFMultiplier, + ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") + + +class ECDHTests(SimulatorTest): + def do_ecdh_test(self, runner, params, mult_class, formulas, mult_name): + #other_privs = [15, 2355498743, 3253857901321912443757746] + other_privs = [15] + + def callback(target, mult, params): + for other_priv in other_privs: + priv, pub = target.generate() + other_pub = mult.multiply(other_priv) + ecdh = ECDH_SHA1(copy(mult), params, other_pub, priv) + result = target.ecdh(other_pub) + expected = ecdh.perform() + self.assertEqual(result, expected) + + self.do_basic_test(callback, runner, params, mult_class, formulas, mult_name, False, True) + + def test_ltr(self): + runner = CliRunner() + self.do_ecdh_test(runner, self.curve32, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "ltr") + + def test_rtl(self): + runner = CliRunner() + self.do_ecdh_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], + "rtl") + + def test_coron(self): + runner = CliRunner() + self.do_ecdh_test(runner, self.curve32, CoronMultiplier, + ["add-1998-cmo", "dbl-1998-cmo"], "coron") + + def test_bnaf(self): + runner = CliRunner() + self.do_ecdh_test(runner, self.curve32, BinaryNAFMultiplier, + ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") + + +class ECDSATests(SimulatorTest): + def do_ecdsa_test(self, runner, params, mult_class, formulas, mult_name): + data = b"something" + + def callback(target, mult, params): + priv, pub = target.generate() + ecdsa = ECDSA_SHA1(copy(mult), params, mult.formulas["add"], + pub.to_model(params.curve.coordinate_model, params.curve), priv) + + signature_data = target.ecdsa_sign(data) + result = SignatureResult.from_DER(bytes(signature_data)) + self.assertTrue(ecdsa.verify_data(result, data)) + + expected = ecdsa.sign_data(data).to_DER() + self.assertTrue(target.ecdsa_verify(data, expected)) + + self.do_basic_test(callback, runner, params, mult_class, formulas, mult_name, True, False) + + def test_ltr(self): + runner = CliRunner() + self.do_ecdsa_test(runner, self.curve32, LTRMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], "ltr") \ No newline at end of file -- cgit v1.3.1 From 1252f69a677b432c9d19c2075c113ebae13924bc Mon Sep 17 00:00:00 2001 From: andrr3j Date: Wed, 19 Jul 2023 10:42:45 +0200 Subject: small fix in simulator tests --- test/test_simulator.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'test') diff --git a/test/test_simulator.py b/test/test_simulator.py index eb7f238..56349c8 100644 --- a/test/test_simulator.py +++ b/test/test_simulator.py @@ -100,7 +100,7 @@ class KeyGenerationTests(SimulatorTest): def do_keygen_test(self, runner, params, mult_class, formulas, mult_name): def callback(target, mult, params): - for _ in range(1): + for _ in range(3): priv, pub = target.generate() self.assertTrue(params.curve.is_on_curve(pub)) expected = mult.multiply(priv).to_affine() @@ -132,8 +132,7 @@ class KeyGenerationTests(SimulatorTest): class ScalarMultiplicationTests(SimulatorTest): def do_mult_test(self, runner, params, mult_class, formulas, mult_name): - #values = [15, 2355498743, 3253857901321912443757746] - values = [15] + values = [15, 2355498743, 3253857901321912443757746] def callback(target, mult, params): for value in values: @@ -166,8 +165,7 @@ class ScalarMultiplicationTests(SimulatorTest): class ECDHTests(SimulatorTest): def do_ecdh_test(self, runner, params, mult_class, formulas, mult_name): - #other_privs = [15, 2355498743, 3253857901321912443757746] - other_privs = [15] + other_privs = [15, 2355498743, 3253857901321912443757746] def callback(target, mult, params): for other_priv in other_privs: -- cgit v1.3.1 From 9cc17da21c191c4e1b3c652d98ce4e493f811844 Mon Sep 17 00:00:00 2001 From: andrr3j Date: Wed, 19 Jul 2023 14:11:48 +0200 Subject: test fix --- test/test_simulator.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) (limited to 'test') diff --git a/test/test_simulator.py b/test/test_simulator.py index 56349c8..2e363e7 100644 --- a/test/test_simulator.py +++ b/test/test_simulator.py @@ -100,11 +100,10 @@ class KeyGenerationTests(SimulatorTest): def do_keygen_test(self, runner, params, mult_class, formulas, mult_name): def callback(target, mult, params): - for _ in range(3): - priv, pub = target.generate() - self.assertTrue(params.curve.is_on_curve(pub)) - expected = mult.multiply(priv).to_affine() - self.assertEqual(pub, expected) + priv, pub = target.generate() + self.assertTrue(params.curve.is_on_curve(pub)) + expected = mult.multiply(priv).to_affine() + self.assertEqual(pub, expected) self.do_basic_test(callback, runner, params, mult_class, formulas, mult_name, False, False) @@ -132,7 +131,7 @@ class KeyGenerationTests(SimulatorTest): class ScalarMultiplicationTests(SimulatorTest): def do_mult_test(self, runner, params, mult_class, formulas, mult_name): - values = [15, 2355498743, 3253857901321912443757746] + values = [2355498743] def callback(target, mult, params): for value in values: @@ -165,7 +164,7 @@ class ScalarMultiplicationTests(SimulatorTest): class ECDHTests(SimulatorTest): def do_ecdh_test(self, runner, params, mult_class, formulas, mult_name): - other_privs = [15, 2355498743, 3253857901321912443757746] + other_privs = [2355498743] def callback(target, mult, params): for other_priv in other_privs: -- cgit v1.3.1 From fd8e1c3c40b216bc6beada39e86d07136e9f09a2 Mon Sep 17 00:00:00 2001 From: andrr3j Date: Wed, 19 Jul 2023 14:21:31 +0200 Subject: simulator test fix --- test/test_simulator.py | 30 ------------------------------ 1 file changed, 30 deletions(-) (limited to 'test') diff --git a/test/test_simulator.py b/test/test_simulator.py index 2e363e7..1bce829 100644 --- a/test/test_simulator.py +++ b/test/test_simulator.py @@ -117,16 +117,6 @@ class KeyGenerationTests(SimulatorTest): self.do_keygen_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], "rtl") - def test_coron(self): - runner = CliRunner() - self.do_keygen_test(runner, self.curve32, CoronMultiplier, - ["add-1998-cmo", "dbl-1998-cmo"], "coron") - - def test_bnaf(self): - runner = CliRunner() - self.do_keygen_test(runner, self.curve32, BinaryNAFMultiplier, - ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") - class ScalarMultiplicationTests(SimulatorTest): @@ -151,16 +141,6 @@ class ScalarMultiplicationTests(SimulatorTest): self.do_mult_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], "rtl") - def test_coron(self): - runner = CliRunner() - self.do_mult_test(runner, self.curve32, CoronMultiplier, - ["add-1998-cmo", "dbl-1998-cmo"], "coron") - - def test_bnaf(self): - runner = CliRunner() - self.do_mult_test(runner, self.curve32, BinaryNAFMultiplier, - ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") - class ECDHTests(SimulatorTest): def do_ecdh_test(self, runner, params, mult_class, formulas, mult_name): @@ -187,16 +167,6 @@ class ECDHTests(SimulatorTest): self.do_ecdh_test(runner, self.curve32, RTLMultiplier, ["add-1998-cmo", "dbl-1998-cmo"], "rtl") - def test_coron(self): - runner = CliRunner() - self.do_ecdh_test(runner, self.curve32, CoronMultiplier, - ["add-1998-cmo", "dbl-1998-cmo"], "coron") - - def test_bnaf(self): - runner = CliRunner() - self.do_ecdh_test(runner, self.curve32, BinaryNAFMultiplier, - ["add-1998-cmo", "dbl-1998-cmo", "neg"], "bnaf") - class ECDSATests(SimulatorTest): def do_ecdsa_test(self, runner, params, mult_class, formulas, mult_name): -- cgit v1.3.1 From 533c5766441ff6a9355441874781645d23b61e90 Mon Sep 17 00:00:00 2001 From: andrr3j Date: Wed, 19 Jul 2023 14:48:30 +0200 Subject: simulator test fix 2 --- test/test_simulator.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'test') diff --git a/test/test_simulator.py b/test/test_simulator.py index 1bce829..d741cf4 100644 --- a/test/test_simulator.py +++ b/test/test_simulator.py @@ -15,6 +15,7 @@ from pyecsca.ec.mod import Mod from pyecsca.ec.model import ShortWeierstrassModel from pyecsca.ec.params import DomainParameters from pyecsca.ec.point import InfinityPoint, Point +import gc class SimulatorTest(TestCase): @@ -56,6 +57,8 @@ class SimulatorTest(TestCase): mult.init(params, params.generator) callback(target, mult, params) target.disconnect() + del target + gc.collect() class PRNGTests(SimulatorTest): -- cgit v1.3.1