]>
Commit | Line | Data |
---|---|---|
26abbf37 DE |
1 | ## |
2 | ## This file is part of the libsigrokdecode project. | |
3 | ## | |
4 | ## Copyright (C) 2014 Daniel Elstner <daniel.kitta@gmail.com> | |
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 3 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 | from functools import reduce | |
22 | from .tables import instr_table_by_prefix | |
aef3c109 | 23 | import string |
26abbf37 DE |
24 | |
25 | class Ann: | |
26 | ADDR, MEMRD, MEMWR, IORD, IOWR, INSTR, ROP, WOP, WARN = range(9) | |
27 | class Row: | |
28 | ADDRBUS, DATABUS, INSTRUCTIONS, OPERANDS, WARNINGS = range(5) | |
29 | class Pin: | |
30 | D0, D7 = 0, 7 | |
31 | M1, RD, WR, MREQ, IORQ = range(8, 13) | |
32 | A0, A15 = 13, 28 | |
33 | class Cycle: | |
34 | NONE, MEMRD, MEMWR, IORD, IOWR, FETCH, INTACK = range(7) | |
35 | ||
aef3c109 DE |
36 | # Provide custom format type 'H' for hexadecimal output |
37 | # with leading decimal digit (assembler syntax). | |
38 | class AsmFormatter(string.Formatter): | |
39 | def format_field(self, value, format_spec): | |
40 | if format_spec.endswith('H'): | |
41 | result = format(value, format_spec[:-1] + 'X') | |
42 | return result if result[0] in string.digits else '0' + result | |
43 | else: | |
44 | return format(value, format_spec) | |
45 | ||
46 | formatter = AsmFormatter() | |
47 | ||
26abbf37 DE |
48 | ann_data_cycle_map = { |
49 | Cycle.MEMRD: Ann.MEMRD, | |
50 | Cycle.MEMWR: Ann.MEMWR, | |
51 | Cycle.IORD: Ann.IORD, | |
52 | Cycle.IOWR: Ann.IOWR, | |
53 | Cycle.FETCH: Ann.MEMRD, | |
54 | Cycle.INTACK: Ann.IORD, | |
55 | } | |
56 | ||
57 | def reduce_bus(bus): | |
58 | if 0xFF in bus: | |
59 | return None # unassigned bus probes | |
60 | else: | |
61 | return reduce(lambda a, b: (a << 1) | b, reversed(bus)) | |
62 | ||
63 | def signed_byte(byte): | |
64 | return byte if byte < 128 else byte - 256 | |
65 | ||
66 | class Decoder(srd.Decoder): | |
67 | api_version = 1 | |
e029ab1f DE |
68 | id = 'z80' |
69 | name = 'Z80' | |
70 | longname = 'Zilog Z80 CPU' | |
71 | desc = 'Zilog Z80 microprocessor disassembly.' | |
72 | license = 'gplv2+' | |
73 | inputs = ['logic'] | |
74 | outputs = ['z80'] | |
26abbf37 | 75 | probes = [ |
31737ae0 DE |
76 | {'id': 'd%d' % i, 'name': 'D%d' % i, 'desc': 'Data bus line %d' % i} |
77 | for i in range(8) | |
78 | ] + [ | |
26abbf37 DE |
79 | {'id': 'm1', 'name': '/M1', 'desc': 'Machine cycle 1'}, |
80 | {'id': 'rd', 'name': '/RD', 'desc': 'Memory or I/O read'}, | |
81 | {'id': 'wr', 'name': '/WR', 'desc': 'Memory or I/O write'}, | |
82 | ] | |
83 | optional_probes = [ | |
84 | {'id': 'mreq', 'name': '/MREQ', 'desc': 'Memory request'}, | |
85 | {'id': 'iorq', 'name': '/IORQ', 'desc': 'I/O request'}, | |
31737ae0 DE |
86 | ] + [ |
87 | {'id': 'a%d' % i, 'name': 'A%d' % i, 'desc': 'Address bus line %d' % i} | |
88 | for i in range(16) | |
26abbf37 DE |
89 | ] |
90 | options = {} | |
91 | annotations = [ | |
e029ab1f DE |
92 | ['addr', 'Memory or I/O address'], |
93 | ['memrd', 'Byte read from memory'], | |
94 | ['memwr', 'Byte written to memory'], | |
95 | ['iord', 'Byte read from I/O port'], | |
96 | ['iowr', 'Byte written to I/O port'], | |
97 | ['instr', 'Z80 CPU instruction'], | |
98 | ['rop', 'Value of input operand'], | |
99 | ['wop', 'Value of output operand'], | |
100 | ['warn', 'Warning message'], | |
26abbf37 DE |
101 | ] |
102 | annotation_rows = ( | |
103 | ('addrbus', 'Address bus', (Ann.ADDR,)), | |
104 | ('databus', 'Data bus', (Ann.MEMRD, Ann.MEMWR, Ann.IORD, Ann.IOWR)), | |
105 | ('instructions', 'Instructions', (Ann.INSTR,)), | |
106 | ('operands', 'Operands', (Ann.ROP, Ann.WOP)), | |
107 | ('warnings', 'Warnings', (Ann.WARN,)) | |
108 | ) | |
109 | ||
110 | def __init__(self, **kwargs): | |
111 | self.prev_cycle = Cycle.NONE | |
697967f2 | 112 | self.op_state = self.state_IDLE |
26abbf37 DE |
113 | |
114 | def start(self): | |
115 | self.out_ann = self.register(srd.OUTPUT_ANN) | |
116 | self.bus_data = None | |
117 | self.samplenum = None | |
118 | self.addr_start = None | |
119 | self.data_start = None | |
120 | self.dasm_start = None | |
121 | self.pend_addr = None | |
122 | self.pend_data = None | |
123 | self.ann_data = None | |
124 | self.ann_dasm = None | |
125 | self.prev_cycle = Cycle.NONE | |
697967f2 | 126 | self.op_state = self.state_IDLE |
8830db5d | 127 | self.instr_len = 0 |
26abbf37 DE |
128 | |
129 | def decode(self, ss, es, data): | |
130 | for (self.samplenum, pins) in data: | |
131 | cycle = Cycle.NONE | |
132 | if pins[Pin.MREQ] != 1: # default to asserted | |
133 | if pins[Pin.RD] == 0: | |
134 | cycle = Cycle.FETCH if pins[Pin.M1] == 0 else Cycle.MEMRD | |
135 | elif pins[Pin.WR] == 0: | |
136 | cycle = Cycle.MEMWR | |
137 | elif pins[Pin.IORQ] == 0: # default to not asserted | |
138 | if pins[Pin.M1] == 0: | |
139 | cycle = Cycle.INTACK | |
140 | elif pins[Pin.RD] == 0: | |
141 | cycle = Cycle.IORD | |
142 | elif pins[Pin.WR] == 0: | |
143 | cycle = Cycle.IOWR | |
144 | ||
145 | if cycle != Cycle.NONE: | |
146 | self.bus_data = reduce_bus(pins[Pin.D0:Pin.D7+1]) | |
147 | if cycle != self.prev_cycle: | |
148 | if self.prev_cycle == Cycle.NONE: | |
149 | self.on_cycle_begin(reduce_bus(pins[Pin.A0:Pin.A15+1])) | |
150 | elif cycle == Cycle.NONE: | |
151 | self.on_cycle_end() | |
152 | else: | |
153 | self.on_cycle_trans() | |
154 | self.prev_cycle = cycle | |
155 | ||
156 | def on_cycle_begin(self, bus_addr): | |
157 | if self.pend_addr is not None: | |
158 | self.put_text(self.addr_start, Ann.ADDR, | |
159 | '{:04X}'.format(self.pend_addr)) | |
160 | self.addr_start = self.samplenum | |
161 | self.pend_addr = bus_addr | |
162 | ||
163 | def on_cycle_end(self): | |
8830db5d | 164 | self.instr_len += 1 |
697967f2 | 165 | self.op_state = self.op_state() |
26abbf37 DE |
166 | if self.ann_dasm is not None: |
167 | self.put_disasm() | |
697967f2 DE |
168 | if self.op_state == self.state_RESTART: |
169 | self.op_state = self.state_IDLE() | |
26abbf37 DE |
170 | |
171 | if self.ann_data is not None: | |
172 | self.put_text(self.data_start, self.ann_data, | |
173 | '{:02X}'.format(self.pend_data)) | |
174 | self.data_start = self.samplenum | |
175 | self.pend_data = self.bus_data | |
176 | self.ann_data = ann_data_cycle_map[self.prev_cycle] | |
177 | ||
178 | def on_cycle_trans(self): | |
179 | self.put_text(self.samplenum - 1, Ann.WARN, | |
180 | 'Illegal transition between control states') | |
181 | self.pend_addr = None | |
182 | self.ann_data = None | |
183 | self.ann_dasm = None | |
184 | ||
185 | def put_disasm(self): | |
8830db5d DE |
186 | text = formatter.format(self.mnemonic, r=self.arg_reg, d=self.arg_dis, |
187 | j=self.arg_dis+self.instr_len, i=self.arg_imm, | |
aef3c109 | 188 | ro=self.arg_read, wo=self.arg_write) |
26abbf37 DE |
189 | self.put_text(self.dasm_start, self.ann_dasm, text) |
190 | self.ann_dasm = None | |
191 | self.dasm_start = self.samplenum | |
192 | ||
193 | def put_text(self, ss, ann_idx, ann_text): | |
194 | self.put(ss, self.samplenum, self.out_ann, [ann_idx, [ann_text]]) | |
195 | ||
697967f2 DE |
196 | def state_RESTART(self): |
197 | return self.state_IDLE | |
198 | ||
199 | def state_IDLE(self): | |
26abbf37 | 200 | if self.prev_cycle != Cycle.FETCH: |
697967f2 | 201 | return self.state_IDLE |
26abbf37 DE |
202 | self.want_dis = 0 |
203 | self.want_imm = 0 | |
204 | self.want_read = 0 | |
205 | self.want_write = 0 | |
206 | self.want_wr_be = False | |
207 | self.op_repeat = False | |
208 | self.arg_dis = 0 | |
209 | self.arg_imm = 0 | |
210 | self.arg_read = 0 | |
211 | self.arg_write = 0 | |
212 | self.arg_reg = '' | |
213 | self.mnemonic = '' | |
214 | self.instr_pend = False | |
215 | self.read_pend = False | |
216 | self.write_pend = False | |
217 | self.dasm_start = self.samplenum | |
218 | self.op_prefix = 0 | |
8830db5d | 219 | self.instr_len = 0 |
26abbf37 | 220 | if self.bus_data in (0xCB, 0xED, 0xDD, 0xFD): |
697967f2 | 221 | return self.state_PRE1 |
26abbf37 | 222 | else: |
697967f2 | 223 | return self.state_OPCODE |
26abbf37 | 224 | |
697967f2 | 225 | def state_PRE1(self): |
26abbf37 DE |
226 | if self.prev_cycle != Cycle.FETCH: |
227 | self.mnemonic = 'Prefix not followed by fetch' | |
228 | self.ann_dasm = Ann.WARN | |
697967f2 | 229 | return self.state_RESTART |
26abbf37 DE |
230 | self.op_prefix = self.pend_data |
231 | if self.op_prefix in (0xDD, 0xFD): | |
232 | if self.bus_data == 0xCB: | |
697967f2 | 233 | return self.state_PRE2 |
26abbf37 | 234 | if self.bus_data in (0xDD, 0xED, 0xFD): |
697967f2 DE |
235 | return self.state_PRE1 |
236 | return self.state_OPCODE | |
26abbf37 | 237 | |
697967f2 | 238 | def state_PRE2(self): |
26abbf37 DE |
239 | if self.prev_cycle != Cycle.MEMRD: |
240 | self.mnemonic = 'Missing displacement' | |
241 | self.ann_dasm = Ann.WARN | |
697967f2 | 242 | return self.state_RESTART |
26abbf37 | 243 | self.op_prefix = (self.op_prefix << 8) | self.pend_data |
697967f2 | 244 | return self.state_PREDIS |
26abbf37 | 245 | |
697967f2 | 246 | def state_PREDIS(self): |
26abbf37 DE |
247 | if self.prev_cycle != Cycle.MEMRD: |
248 | self.mnemonic = 'Missing opcode' | |
249 | self.ann_dasm = Ann.WARN | |
697967f2 | 250 | return self.state_RESTART |
26abbf37 | 251 | self.arg_dis = signed_byte(self.pend_data) |
697967f2 | 252 | return self.state_OPCODE |
26abbf37 | 253 | |
697967f2 | 254 | def state_OPCODE(self): |
26abbf37 DE |
255 | (table, self.arg_reg) = instr_table_by_prefix[self.op_prefix] |
256 | self.op_prefix = 0 | |
257 | instruction = table.get(self.pend_data, None) | |
258 | if instruction is None: | |
259 | self.mnemonic = 'Invalid instruction' | |
260 | self.ann_dasm = Ann.WARN | |
697967f2 | 261 | return self.state_RESTART |
26abbf37 DE |
262 | (self.want_dis, self.want_imm, self.want_read, want_write, |
263 | self.op_repeat, self.mnemonic) = instruction | |
264 | self.want_write = abs(want_write) | |
265 | self.want_wr_be = (want_write < 0) | |
266 | if self.want_dis > 0: | |
697967f2 | 267 | return self.state_POSTDIS |
26abbf37 | 268 | if self.want_imm > 0: |
697967f2 | 269 | return self.state_IMM1 |
26abbf37 DE |
270 | self.ann_dasm = Ann.INSTR |
271 | if self.want_read > 0 and self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): | |
697967f2 | 272 | return self.state_ROP1 |
26abbf37 | 273 | if self.want_write > 0 and self.prev_cycle in (Cycle.MEMWR, Cycle.IOWR): |
697967f2 DE |
274 | return self.state_WOP1 |
275 | return self.state_RESTART | |
26abbf37 | 276 | |
697967f2 | 277 | def state_POSTDIS(self): |
26abbf37 DE |
278 | self.arg_dis = signed_byte(self.pend_data) |
279 | if self.want_imm > 0: | |
697967f2 | 280 | return self.state_IMM1 |
26abbf37 | 281 | self.ann_dasm = Ann.INSTR |
3831aa89 | 282 | if self.want_read > 0 and self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): |
697967f2 | 283 | return self.state_ROP1 |
3831aa89 | 284 | if self.want_write > 0 and self.prev_cycle in (Cycle.MEMWR, Cycle.IOWR): |
697967f2 DE |
285 | return self.state_WOP1 |
286 | return self.state_RESTART | |
26abbf37 | 287 | |
697967f2 | 288 | def state_IMM1(self): |
26abbf37 DE |
289 | self.arg_imm = self.pend_data |
290 | if self.want_imm > 1: | |
697967f2 | 291 | return self.state_IMM2 |
26abbf37 | 292 | self.ann_dasm = Ann.INSTR |
3831aa89 | 293 | if self.want_read > 0 and self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): |
697967f2 | 294 | return self.state_ROP1 |
3831aa89 | 295 | if self.want_write > 0 and self.prev_cycle in (Cycle.MEMWR, Cycle.IOWR): |
697967f2 DE |
296 | return self.state_WOP1 |
297 | return self.state_RESTART | |
26abbf37 | 298 | |
697967f2 | 299 | def state_IMM2(self): |
26abbf37 DE |
300 | self.arg_imm |= self.pend_data << 8 |
301 | self.ann_dasm = Ann.INSTR | |
3831aa89 | 302 | if self.want_read > 0 and self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): |
697967f2 | 303 | return self.state_ROP1 |
3831aa89 | 304 | if self.want_write > 0 and self.prev_cycle in (Cycle.MEMWR, Cycle.IOWR): |
697967f2 DE |
305 | return self.state_WOP1 |
306 | return self.state_RESTART | |
26abbf37 | 307 | |
697967f2 | 308 | def state_ROP1(self): |
26abbf37 | 309 | self.arg_read = self.pend_data |
739f9313 DE |
310 | if self.want_read < 2: |
311 | self.mnemonic = '{ro:02X}' | |
312 | self.ann_dasm = Ann.ROP | |
26abbf37 | 313 | if self.want_write > 0: |
697967f2 | 314 | return self.state_WOP1 |
26abbf37 | 315 | if self.want_read > 1: |
697967f2 | 316 | return self.state_ROP2 |
26abbf37 | 317 | if self.op_repeat and self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): |
697967f2 DE |
318 | return self.state_ROP1 |
319 | return self.state_RESTART | |
26abbf37 | 320 | |
697967f2 | 321 | def state_ROP2(self): |
26abbf37 DE |
322 | self.arg_read |= self.pend_data << 8 |
323 | self.mnemonic = '{ro:04X}' | |
324 | self.ann_dasm = Ann.ROP | |
3831aa89 | 325 | if self.want_write > 0 and self.prev_cycle in (Cycle.MEMWR, Cycle.IOWR): |
697967f2 DE |
326 | return self.state_WOP1 |
327 | return self.state_RESTART | |
26abbf37 | 328 | |
697967f2 | 329 | def state_WOP1(self): |
26abbf37 DE |
330 | self.arg_write = self.pend_data |
331 | if self.want_read > 1: | |
697967f2 | 332 | return self.state_ROP2 |
26abbf37 | 333 | if self.want_write > 1: |
697967f2 | 334 | return self.state_WOP2 |
739f9313 DE |
335 | self.mnemonic = '{wo:02X}' |
336 | self.ann_dasm = Ann.WOP | |
26abbf37 DE |
337 | if self.want_read > 0 and self.op_repeat and \ |
338 | self.prev_cycle in (Cycle.MEMRD, Cycle.IORD): | |
697967f2 DE |
339 | return self.state_ROP1 |
340 | return self.state_RESTART | |
26abbf37 | 341 | |
697967f2 | 342 | def state_WOP2(self): |
26abbf37 DE |
343 | if self.want_wr_be: |
344 | self.arg_write = (self.arg_write << 8) | self.pend_data | |
345 | else: | |
346 | self.arg_write |= self.pend_data << 8 | |
347 | self.mnemonic = '{wo:04X}' | |
348 | self.ann_dasm = Ann.WOP | |
697967f2 | 349 | return self.state_RESTART |