]> sigrok.org Git - libsigrokdecode.git/blob - decoders/lin/pd.py
lin: Simplify LinFsm._transition_allowed().
[libsigrokdecode.git] / decoders / lin / pd.py
1 ##
2 ## This file is part of the libsigrokdecode project.
3 ##
4 ## Copyright (C) 2018 Stephan Thiele <stephan.thiele@mailbox.org>
5 ##
6 ## This program is free software; you can redistribute it and/or modify
7 ## it under the terms of the GNU General Public License as published by
8 ## the Free Software Foundation; either version 2 of the License, or
9 ## (at your option) any later version.
10 ##
11 ## This program is distributed in the hope that it will be useful,
12 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 ## GNU General Public License for more details.
15 ##
16 ## You should have received a copy of the GNU General Public License
17 ## along with this program; if not, see <http://www.gnu.org/licenses/>.
18 ##
19
20 import sigrokdecode as srd
21
22 class LinFsm:
23     class State:
24         WaitForBreak = 'WAIT FOR BREAK'
25         Sync = 'SYNC'
26         Pid = 'PID'
27         Data = 'DATA'
28         Checksum = 'CHECKSUM'
29         Error = 'ERROR'
30
31     def transit(self, target_state):
32         if not self._transition_allowed(target_state):
33             return False
34         self.state = target_state
35         return True
36
37     def _transition_allowed(self, target_state):
38         if target_state == LinFsm.State.Error:
39             return True
40         return target_state in self.allowed_state[self.state]
41
42     def reset(self):
43         self.state = LinFsm.State.WaitForBreak
44
45     def __init__(self):
46         self.allowed_state = dict()
47         self.allowed_state[LinFsm.State.WaitForBreak] = (LinFsm.State.Sync,)
48         self.allowed_state[LinFsm.State.Sync]         = (LinFsm.State.Pid,)
49         self.allowed_state[LinFsm.State.Pid]          = (LinFsm.State.Data,)
50         self.allowed_state[LinFsm.State.Data]         = (LinFsm.State.Data, LinFsm.State.Checksum)
51         self.allowed_state[LinFsm.State.Checksum]     = (LinFsm.State.WaitForBreak,)
52         self.allowed_state[LinFsm.State.Error]        = (LinFsm.State.Sync,)
53
54         self.state = None
55         self.reset()
56
57 class Decoder(srd.Decoder):
58     api_version = 3
59     id = 'lin'
60     name = 'LIN'
61     longname = 'Local Interconnect Network'
62     desc = 'Local Interconnect Network (LIN) protocol.'
63     license = 'gplv2+'
64     inputs = ['uart']
65     outputs = ['lin']
66     options = (
67         {'id': 'version', 'desc': 'Protocol version', 'default': 2, 'values': (1, 2)},
68     )
69     annotations = (
70         ('data', 'LIN data'),
71         ('control', 'Protocol info'),
72         ('error', 'Error descriptions'),
73         ('inline_error', 'Protocol violations and errors'),
74     )
75     annotation_rows = (
76         ('data', 'Data', (0, 1, 3)),
77         ('error', 'Error', (2,)),
78     )
79
80     def __init__(self):
81         self.reset()
82
83     def reset(self):
84         self.fsm = LinFsm()
85         self.lin_header = []
86         self.lin_rsp = []
87         self.lin_version = None
88         self.out_ann = None
89         self.ss_block = None
90         self.es_block = None
91
92     def start(self):
93         self.out_ann = self.register(srd.OUTPUT_ANN)
94         self.lin_version = self.options['version']
95
96     def putx(self, data):
97         self.put(self.ss_block, self.es_block, self.out_ann, data)
98
99     def wipe_break_null_byte(self, value):
100         # Upon a break condition a null byte is received which must be ignored.
101         if self.fsm.state != LinFsm.State.WaitForBreak and self.fsm.state != LinFsm.State.Error:
102             if len(self.lin_rsp):
103                 value = self.lin_rsp.pop()[2]
104             else:
105                 self.lin_header.pop()
106
107         if value != 0:
108             self.fsm.transit(LinFsm.State.Error)
109             self.handle_error()
110             return False
111
112         return True
113
114     def handle_wait_for_break(self, value):
115         self.wipe_break_null_byte(value)
116
117     def handle_break(self, value):
118         if self.fsm.state != LinFsm.State.WaitForBreak and self.fsm.state != LinFsm.State.Error:
119             if self.wipe_break_null_byte(value):
120                 self.fsm.transit(LinFsm.State.Checksum)
121                 self.handle_checksum()
122
123         self.fsm.reset()
124         self.fsm.transit(LinFsm.State.Sync)
125
126         data = [1, ['Break condition', 'Break', 'Brk', 'B']]
127         self.put(self.ss_block, self.es_block, self.out_ann, data)
128
129     def handle_sync(self, value):
130         self.fsm.transit(LinFsm.State.Pid)
131         self.lin_header.append((self.ss_block, self.es_block, value))
132
133     def handle_pid(self, value):
134         self.fsm.transit(LinFsm.State.Data)
135         self.lin_header.append((self.ss_block, self.es_block, value))
136
137     def handle_data(self, value):
138         self.lin_rsp.append((self.ss_block, self.es_block, value))
139
140     def handle_checksum(self):
141         sync = self.lin_header.pop(0) if len(self.lin_header) else None
142
143         self.put(sync[0], sync[1], self.out_ann, [0, ['Sync', 'S']])
144
145         if sync[2] != 0x55:
146             self.put(sync[0], sync[1], self.out_ann,
147                      [2, ['Sync is not 0x55', 'Not 0x55', '!= 0x55']])
148
149         pid = self.lin_header.pop(0) if len(self.lin_header) else None
150         checksum = self.lin_rsp.pop() if len(self.lin_rsp) else None
151
152         if pid:
153             id = pid[2] & 0x3F
154             parity = pid[2] >> 6
155
156             expected_parity = self.calc_parity(pid[2])
157             parity_valid = parity == expected_parity
158
159             if not parity_valid:
160                 self.put(pid[0], pid[1], self.out_ann, [2, ['P != %d' % expected_parity]])
161
162             ann_class = 0 if parity_valid else 3
163             self.put(pid[0], pid[1], self.out_ann, [ann_class, [
164                 'ID: %02X Parity: %d (%s)' % (id, parity, 'ok' if parity_valid else 'bad'),
165                 'ID: 0x%02X' % id, 'I: %d' % id
166             ]])
167
168         if len(self.lin_rsp):
169             checksum_valid = self.checksum_is_valid(pid[2], self.lin_rsp, checksum[2])
170
171             for b in self.lin_rsp:
172                 self.put(b[0], b[1], self.out_ann, [0, ['Data: 0x%02X' % b[2], 'D: 0x%02X' % b[2]]])
173
174             ann_class = 0 if checksum_valid else 3
175             self.put(checksum[0], checksum[1], self.out_ann,
176                  [ann_class, ['Checksum: 0x%02X' % checksum[2], 'Checksum', 'Chk', 'C']])
177
178             if not checksum_valid:
179                 self.put(checksum[0], checksum[1], self.out_ann, [2, ['Checksum invalid']])
180         else:
181             pass # No response.
182
183         self.lin_header.clear()
184         self.lin_rsp.clear()
185
186     def handle_error(self):
187         self.putx([3, ['Error', 'Err', 'E']])
188
189     def checksum_is_valid(self, pid, data, checksum):
190         if self.lin_version == 2:
191             id = pid & 0x3F
192
193             if id != 60 and id != 61:
194                 checksum += pid
195
196         for d in data:
197             checksum += d[2]
198
199         carry_bits = int(checksum / 256)
200         checksum += carry_bits
201
202         return checksum & 0xFF == 0xFF
203
204     @staticmethod
205     def calc_parity(pid):
206         id = [((pid & 0x3F) >> i) & 1 for i in range(8)]
207
208         p0 = id[0] ^ id[1] ^ id[2] ^ id[4]
209         p1 = not (id[1] ^ id[3] ^ id[4] ^ id[5])
210
211         return (p0 << 0) | (p1 << 1)
212
213     def decode(self, ss, es, data):
214         ptype, rxtx, pdata = data
215
216         self.ss_block, self.es_block = ss, es
217
218         # Ignore all UART packets except the actual data packets or BREAK.
219         if ptype == 'BREAK':
220             self.handle_break(pdata)
221         if ptype != 'DATA':
222             return
223
224         # We're only interested in the byte value (not individual bits).
225         pdata = pdata[0]
226
227         # Short LIN overview:
228         #  - Message begins with a BREAK (0x00) for at least 13 bittimes.
229         #  - Break is always followed by a SYNC byte (0x55).
230         #  - Sync byte is followed by a PID byte (Protected Identifier).
231         #  - PID byte is followed by 1 - 8 data bytes and a final checksum byte.
232
233         if   self.fsm.state == LinFsm.State.Error:        self.handle_error()
234         elif self.fsm.state == LinFsm.State.WaitForBreak: self.handle_wait_for_break(pdata)
235         elif self.fsm.state == LinFsm.State.Sync:         self.handle_sync(pdata)
236         elif self.fsm.state == LinFsm.State.Pid:          self.handle_pid(pdata)
237         elif self.fsm.state == LinFsm.State.Data:         self.handle_data(pdata)
238         elif self.fsm.state == LinFsm.State.Checksum:     self.handle_checksum()