Add tests for DPA and CPA.

1 files changed, 46 insertions, 0 deletions

diff --git a/test/sca/test_attacks.py b/test/sca/test_attacks.py
new file mode 100644
index 0000000..e213ecf
--- /dev/null
+++ b/test/sca/test_attacks.py
@@ -0,0 +1,46 @@
+import pytest
+
+from pyecsca.ec.mult import LTRMultiplier
+from pyecsca.sca.attack.DPA import DPA
+from pyecsca.sca.attack.CPA import CPA
+from pyecsca.sca.target import LeakageTarget
+from pyecsca.sca.attack import HammingWeight
+
+
+@pytest.fixture()
+def mult(secp128r1):
+    return LTRMultiplier(
+        secp128r1.curve.coordinate_model.formulas["add-2015-rcb"],
+        secp128r1.curve.coordinate_model.formulas["dbl-2015-rcb"],
+    )
+
+
+@pytest.fixture()
+def target(secp128r1, mult):
+    target = LeakageTarget(
+        secp128r1.curve.model, secp128r1.curve.coordinate_model, mult, HammingWeight()
+    )
+    target.set_params(secp128r1)
+    return target
+
+
+def test_dpa(secp128r1, mult, target):
+    scalar = 5
+    pub = secp128r1.curve.affine_multiply(
+        secp128r1.generator.to_affine(), scalar
+    ).to_model(secp128r1.curve.coordinate_model, secp128r1.curve)
+    points, traces = target.simulate_scalar_mult_traces(500, scalar)
+    dpa = DPA(points, traces, mult, secp128r1)
+    res = dpa.perform(3, pub)
+    assert res == 5
+
+
+def test_cpa(secp128r1, mult, target):
+    scalar = 5
+    pub = secp128r1.curve.affine_multiply(
+        secp128r1.generator.to_affine(), scalar
+    ).to_model(secp128r1.curve.coordinate_model, secp128r1.curve)
+    points, traces = target.simulate_scalar_mult_traces(100, scalar)
+    cpa = CPA(points, traces, HammingWeight(), mult, secp128r1)
+    res = cpa.perform(3, pub)
+    assert res == 5