aboutsummaryrefslogtreecommitdiff
path: root/test/ec/test_params.py
blob: eb703422b68ac04af011ffb0c37fdce2442c7fa6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
from unittest import TestCase

from parameterized import parameterized

from pyecsca.misc.cfg import TemporaryConfig
from pyecsca.ec.coordinates import AffineCoordinateModel
from pyecsca.ec.error import UnsatisfiedAssumptionError
from pyecsca.ec.params import get_params, load_params, load_category, get_category


class DomainParameterTests(TestCase):
    def setUp(self):
        self.secp128r1 = get_params("secg", "secp128r1", "projective")
        self.curve25519 = get_params("other", "Curve25519", "xz")

    def test_eq(self):
        self.assertEqual(self.secp128r1, self.secp128r1)
        self.assertNotEqual(self.secp128r1, self.curve25519)
        self.assertNotEqual(self.secp128r1, None)

    def test_str(self):
        self.assertEqual(str(self.secp128r1), "DomainParameters(secg/secp128r1)")

    @parameterized.expand([
        ("secg/secp128r1", "projective"),
        ("secg/secp256r1", "projective"),
        ("secg/secp521r1", "projective"),
        ("other/Curve25519", "xz"),
        ("other/Ed25519", "projective"),
        ("other/Ed448", "projective"),
        ("other/E-222", "projective")
    ])
    def test_get_params(self, name, coords):
        params = get_params(*name.split("/"), coords)
        try:
            assert params.curve.is_on_curve(params.generator)
        except NotImplementedError:
            pass

    @parameterized.expand([
        ("anssi", "projective"),
        ("brainpool", lambda name: "projective" if name.endswith("r1") else "jacobian")
    ])
    def test_get_category(self, name, coords):
        get_category(name, coords)

    def test_load_params(self):
        params = load_params("test/data/curve.json", "projective")
        try:
            assert params.curve.is_on_curve(params.generator)
        except NotImplementedError:
            pass

    def test_load_category(self):
        category = load_category("test/data/curves.json", "yz")
        self.assertEqual(len(category), 1)

    @parameterized.expand([
        ("no_category/some", "else"),
        ("secg/no_curve", "else"),
        ("secg/secp128r1", "some")
    ])
    def test_unknown(self, name, coords):
        with self.assertRaises(ValueError):
            get_params(*name.split("/"), coords)

    def test_assumption(self):
        with self.assertRaises(UnsatisfiedAssumptionError):
            get_params("secg", "secp128r1", "projective-1")
        with TemporaryConfig() as cfg:
            cfg.ec.unsatisfied_coordinate_assumption_action = "ignore"
            params = get_params("secg", "secp128r1", "projective-1")
            self.assertIsNotNone(params)
        self.assertIsNotNone(get_params("secg", "secp128r1", "projective-3"))

    def test_infty(self):
        with self.assertRaises(ValueError):
            get_params("other", "Ed25519", "modified", False)
        self.assertIsNotNone(get_params("secg", "secp128r1", "projective", False))

    def test_no_binary(self):
        with self.assertRaises(ValueError):
            get_params("secg", "sect163r1", "something")

    def test_no_extension(self):
        with self.assertRaises(ValueError):
            get_params("other", "Fp254n2BNa", "something")

    def test_affine(self):
        aff = get_params("secg", "secp128r1", "affine")
        self.assertIsInstance(aff.curve.coordinate_model, AffineCoordinateModel)