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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
import pytest
from pyecsca.ec.context import (
local,
DefaultContext,
Node,
PathContext, Action
)
from pyecsca.ec.key_generation import KeyGeneration
from pyecsca.ec.mod import RandomModAction
from pyecsca.ec.mult import LTRMultiplier, ScalarMultiplicationAction
def test_walk_by_key():
tree = Node(Action())
a_a = Action()
a = Node(a_a, parent=tree)
one_a = Action()
one = Node(one_a, parent=a)
other_a = Action()
other = Node(other_a, parent=a)
assert tree.get_by_key([]) == tree
assert tree.get_by_key([a_a]) == a
assert tree.get_by_key(([a_a, one_a])) == one
assert tree.get_by_key(([a_a, other_a])) == other
def test_walk_by_index():
tree = Node(Action())
a_a = Action()
a = Node(a_a, parent=tree)
one_a = Action()
one = Node(one_a, parent=a)
other_a = Action()
other = Node(other_a, parent=a)
assert tree.get_by_index([]) == tree
assert tree.get_by_index([0]) == a
assert tree.get_by_index([0, 0]) == one
assert tree.get_by_index([0, 1]) == other
def test_render():
tree = Node(Action())
a_a = Action()
a = Node(a_a, parent=tree)
one_a = Action()
Node(one_a, parent=a)
other_a = Action()
Node(other_a, parent=a)
txt = tree.render()
assert txt == """Action()
└──Action()
├──Action()
└──Action()"""
@pytest.fixture()
def mult(secp128r1):
base = secp128r1.generator
coords = secp128r1.curve.coordinate_model
mult = LTRMultiplier(
coords.formulas["add-1998-cmo"],
coords.formulas["dbl-1998-cmo"],
coords.formulas["z"],
always=True,
)
mult.init(secp128r1, base)
return mult
def test_null(mult):
with local() as ctx:
mult.multiply(59)
assert ctx is None
def test_default(mult):
with local(DefaultContext()) as ctx:
result = mult.multiply(59)
action = ctx.actions[0].action
assert isinstance(action, ScalarMultiplicationAction)
assert result == action.result
def test_default_no_enter():
with local(DefaultContext()) as default, pytest.raises(ValueError):
default.exit_action(RandomModAction(7))
def test_multiple_enter(mult):
default = DefaultContext()
with local(default) as ctx1:
mult.multiply(59)
with local(default) as ctx2:
mult.multiply(135)
assert len(default.actions) == 0
assert len(ctx1.actions) == len(ctx2.actions)
def test_multiple_enter_chained(mult):
default = DefaultContext()
with local(default) as ctx1:
mult.multiply(59)
with local(ctx1) as ctx2:
mult.multiply(135)
assert len(default.actions) == 0
assert 2 * len(ctx1.actions) == len(ctx2.actions)
def test_multiple_enter_no_copy(mult):
default = DefaultContext()
with local(default, copy=False) as ctx1:
mult.multiply(59)
with local(default, copy=False) as ctx2:
mult.multiply(135)
assert len(default.actions) == len(ctx1.actions)
assert len(ctx1.actions) == len(ctx2.actions)
def test_path(mult, secp128r1):
with local(PathContext([0, 1])) as ctx:
key_generator = KeyGeneration(mult, secp128r1, True)
key_generator.generate()
assert isinstance(ctx.value, ScalarMultiplicationAction)
with local(PathContext([0, 1, 7])):
key_generator = KeyGeneration(mult, secp128r1, True)
key_generator.generate()
def test_str(mult):
with local(DefaultContext()) as default:
mult.multiply(59)
assert str(default) is not None
assert str(default.actions) is not None
with local(None):
mult.multiply(59)
|
