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
|
"""Provides an oscilloscope class using the ChipWhisperer-Lite scope."""
from typing import Optional, Tuple, Sequence, Set
import numpy as np
from chipwhisperer.capture.scopes.OpenADC import OpenADC
from public import public
from pyecsca.sca.scope.base import Scope, SampleType
from pyecsca.sca.trace import Trace
@public
class ChipWhispererScope(Scope): # pragma: no cover
"""A ChipWhisperer based scope."""
def __init__(self, scope: OpenADC):
super().__init__()
self.scope = scope
self.triggers: Set[str] = set()
def open(self) -> None:
self.scope.con()
@property
def channels(self) -> Sequence[str]:
return []
def setup_frequency(
self, frequency: int, pretrig: int, posttrig: int
) -> Tuple[int, int]:
if pretrig != 0:
raise ValueError("ChipWhisperer does not support pretrig samples.")
self.scope.clock.clkgen_freq = frequency
self.scope.adc.samples = posttrig
return self.scope.clock.clkgen_freq, self.scope.adc.samples
def setup_channel(
self, channel: str, coupling: str, range: float, offset: float, enable: bool
) -> None:
pass # Nothing to setup
def setup_trigger(
self,
channel: str,
threshold: float,
direction: str,
delay: int,
timeout: int,
enable: bool,
) -> None:
if enable:
self.triggers.add(channel)
elif channel in self.triggers:
self.triggers.remove(channel)
self.scope.adc.basic_mode = direction
self.scope.trigger.triggers = " OR ".join(self.triggers)
def setup_capture(self, channel: str, enable: bool) -> None:
pass # Nothing to setup
def arm(self) -> None:
self.scope.arm()
def capture(self, timeout: Optional[int] = None) -> bool:
return not self.scope.capture()
def retrieve(
self, channel: str, type: SampleType, dtype=np.float16
) -> Optional[Trace]:
data = self.scope.get_last_trace()
if data is None:
return None
return Trace(
np.array(data, dtype=dtype),
{"sampling_frequency": self.scope.clock.clkgen_freq, "channel": channel},
)
def stop(self) -> None:
pass # Nothing to do
def close(self) -> None:
self.scope.dis()
|
