SPI table updates (some new devices and new info)
[openocd.git] / src / flash / nor / jtagspi.c
1 /***************************************************************************
2 * Copyright (C) 2015 Robert Jordens <jordens@gmail.com> *
3 * *
4 * This program is free software; you can redistribute it and/or modify *
5 * it under the terms of the GNU General Public License as published by *
6 * the Free Software Foundation; either version 2 of the License, or *
7 * (at your option) any later version. *
8 * *
9 * This program is distributed in the hope that it will be useful, *
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
12 * GNU General Public License for more details. *
13 * *
14 * You should have received a copy of the GNU General Public License *
15 * along with this program. If not, see <http://www.gnu.org/licenses/>. *
16 ***************************************************************************/
17
18 #ifdef HAVE_CONFIG_H
19 #include "config.h"
20 #endif
21
22 #include "imp.h"
23 #include <jtag/jtag.h>
24 #include <flash/nor/spi.h>
25 #include <helper/time_support.h>
26
27 #define JTAGSPI_MAX_TIMEOUT 3000
28
29
30 struct jtagspi_flash_bank {
31 struct jtag_tap *tap;
32 const struct flash_device *dev;
33 int probed;
34 uint32_t ir;
35 };
36
37 FLASH_BANK_COMMAND_HANDLER(jtagspi_flash_bank_command)
38 {
39 struct jtagspi_flash_bank *info;
40
41 if (CMD_ARGC < 7)
42 return ERROR_COMMAND_SYNTAX_ERROR;
43
44 info = malloc(sizeof(struct jtagspi_flash_bank));
45 if (info == NULL) {
46 LOG_ERROR("no memory for flash bank info");
47 return ERROR_FAIL;
48 }
49 bank->driver_priv = info;
50
51 info->tap = NULL;
52 info->probed = 0;
53 COMMAND_PARSE_NUMBER(u32, CMD_ARGV[6], info->ir);
54
55 return ERROR_OK;
56 }
57
58 static void jtagspi_set_ir(struct flash_bank *bank)
59 {
60 struct jtagspi_flash_bank *info = bank->driver_priv;
61 struct scan_field field;
62 uint8_t buf[4];
63
64 LOG_DEBUG("loading jtagspi ir");
65 buf_set_u32(buf, 0, info->tap->ir_length, info->ir);
66 field.num_bits = info->tap->ir_length;
67 field.out_value = buf;
68 field.in_value = NULL;
69 jtag_add_ir_scan(info->tap, &field, TAP_IDLE);
70 }
71
72 static void flip_u8(uint8_t *in, uint8_t *out, int len)
73 {
74 for (int i = 0; i < len; i++)
75 out[i] = flip_u32(in[i], 8);
76 }
77
78 static int jtagspi_cmd(struct flash_bank *bank, uint8_t cmd,
79 uint32_t *addr, uint8_t *data, int len)
80 {
81 struct jtagspi_flash_bank *info = bank->driver_priv;
82 struct scan_field fields[6];
83 uint8_t marker = 1;
84 uint8_t xfer_bits_buf[4];
85 uint8_t addr_buf[3];
86 uint8_t *data_buf;
87 uint32_t xfer_bits;
88 int is_read, lenb, n;
89
90 /* LOG_DEBUG("cmd=0x%02x len=%i", cmd, len); */
91
92 is_read = (len < 0);
93 if (is_read)
94 len = -len;
95
96 n = 0;
97
98 fields[n].num_bits = 1;
99 fields[n].out_value = &marker;
100 fields[n].in_value = NULL;
101 n++;
102
103 xfer_bits = 8 + len - 1;
104 /* cmd + read/write - 1 due to the counter implementation */
105 if (addr)
106 xfer_bits += 24;
107 h_u32_to_be(xfer_bits_buf, xfer_bits);
108 flip_u8(xfer_bits_buf, xfer_bits_buf, 4);
109 fields[n].num_bits = 32;
110 fields[n].out_value = xfer_bits_buf;
111 fields[n].in_value = NULL;
112 n++;
113
114 cmd = flip_u32(cmd, 8);
115 fields[n].num_bits = 8;
116 fields[n].out_value = &cmd;
117 fields[n].in_value = NULL;
118 n++;
119
120 if (addr) {
121 h_u24_to_be(addr_buf, *addr);
122 flip_u8(addr_buf, addr_buf, 3);
123 fields[n].num_bits = 24;
124 fields[n].out_value = addr_buf;
125 fields[n].in_value = NULL;
126 n++;
127 }
128
129 lenb = DIV_ROUND_UP(len, 8);
130 data_buf = malloc(lenb);
131 if (lenb > 0) {
132 if (data_buf == NULL) {
133 LOG_ERROR("no memory for spi buffer");
134 return ERROR_FAIL;
135 }
136 if (is_read) {
137 fields[n].num_bits = jtag_tap_count_enabled();
138 fields[n].out_value = NULL;
139 fields[n].in_value = NULL;
140 n++;
141
142 fields[n].out_value = NULL;
143 fields[n].in_value = data_buf;
144 } else {
145 flip_u8(data, data_buf, lenb);
146 fields[n].out_value = data_buf;
147 fields[n].in_value = NULL;
148 }
149 fields[n].num_bits = len;
150 n++;
151 }
152
153 jtagspi_set_ir(bank);
154 /* passing from an IR scan to SHIFT-DR clears BYPASS registers */
155 jtag_add_dr_scan(info->tap, n, fields, TAP_IDLE);
156 jtag_execute_queue();
157
158 if (is_read)
159 flip_u8(data_buf, data, lenb);
160 free(data_buf);
161 return ERROR_OK;
162 }
163
164 static int jtagspi_probe(struct flash_bank *bank)
165 {
166 struct jtagspi_flash_bank *info = bank->driver_priv;
167 struct flash_sector *sectors;
168 uint8_t in_buf[3];
169 uint32_t id, sectorsize;
170
171 if (info->probed)
172 free(bank->sectors);
173 info->probed = 0;
174
175 if (bank->target->tap == NULL) {
176 LOG_ERROR("Target has no JTAG tap");
177 return ERROR_FAIL;
178 }
179 info->tap = bank->target->tap;
180
181 jtagspi_cmd(bank, SPIFLASH_READ_ID, NULL, in_buf, -24);
182 /* the table in spi.c has the manufacturer byte (first) as the lsb */
183 id = le_to_h_u24(in_buf);
184
185 info->dev = NULL;
186 for (const struct flash_device *p = flash_devices; p->name ; p++)
187 if (p->device_id == id) {
188 info->dev = p;
189 break;
190 }
191
192 if (!(info->dev)) {
193 LOG_ERROR("Unknown flash device (ID 0x%08" PRIx32 ")", id);
194 return ERROR_FAIL;
195 }
196
197 LOG_INFO("Found flash device \'%s\' (ID 0x%08" PRIx32 ")",
198 info->dev->name, info->dev->device_id);
199
200 /* Set correct size value */
201 bank->size = info->dev->size_in_bytes;
202 if (bank->size <= (1UL << 16))
203 LOG_WARNING("device needs 2-byte addresses - not implemented");
204 if (bank->size > (1UL << 24))
205 LOG_WARNING("device needs paging or 4-byte addresses - not implemented");
206
207 /* if no sectors, treat whole bank as single sector */
208 sectorsize = info->dev->sectorsize ?
209 info->dev->sectorsize : info->dev->size_in_bytes;
210
211 /* create and fill sectors array */
212 bank->num_sectors = info->dev->size_in_bytes / sectorsize;
213 sectors = malloc(sizeof(struct flash_sector) * bank->num_sectors);
214 if (sectors == NULL) {
215 LOG_ERROR("not enough memory");
216 return ERROR_FAIL;
217 }
218
219 for (int sector = 0; sector < bank->num_sectors; sector++) {
220 sectors[sector].offset = sector * sectorsize;
221 sectors[sector].size = sectorsize;
222 sectors[sector].is_erased = -1;
223 sectors[sector].is_protected = 0;
224 }
225
226 bank->sectors = sectors;
227 info->probed = 1;
228 return ERROR_OK;
229 }
230
231 static void jtagspi_read_status(struct flash_bank *bank, uint32_t *status)
232 {
233 uint8_t buf;
234 if (jtagspi_cmd(bank, SPIFLASH_READ_STATUS, NULL, &buf, -8) == ERROR_OK) {
235 *status = buf;
236 /* LOG_DEBUG("status=0x%08" PRIx32, *status); */
237 }
238 }
239
240 static int jtagspi_wait(struct flash_bank *bank, int timeout_ms)
241 {
242 uint32_t status;
243 int64_t t0 = timeval_ms();
244 int64_t dt;
245
246 do {
247 dt = timeval_ms() - t0;
248 jtagspi_read_status(bank, &status);
249 if ((status & SPIFLASH_BSY_BIT) == 0) {
250 LOG_DEBUG("waited %" PRId64 " ms", dt);
251 return ERROR_OK;
252 }
253 alive_sleep(1);
254 } while (dt <= timeout_ms);
255
256 LOG_ERROR("timeout, device still busy");
257 return ERROR_FAIL;
258 }
259
260 static int jtagspi_write_enable(struct flash_bank *bank)
261 {
262 uint32_t status;
263
264 jtagspi_cmd(bank, SPIFLASH_WRITE_ENABLE, NULL, NULL, 0);
265 jtagspi_read_status(bank, &status);
266 if ((status & SPIFLASH_WE_BIT) == 0) {
267 LOG_ERROR("Cannot enable write to flash. Status=0x%08" PRIx32, status);
268 return ERROR_FAIL;
269 }
270 return ERROR_OK;
271 }
272
273 static int jtagspi_bulk_erase(struct flash_bank *bank)
274 {
275 struct jtagspi_flash_bank *info = bank->driver_priv;
276 int retval;
277 int64_t t0 = timeval_ms();
278
279 if (info->dev->chip_erase_cmd == 0x00)
280 return ERROR_FLASH_OPER_UNSUPPORTED;
281
282 retval = jtagspi_write_enable(bank);
283 if (retval != ERROR_OK)
284 return retval;
285 jtagspi_cmd(bank, info->dev->chip_erase_cmd, NULL, NULL, 0);
286 retval = jtagspi_wait(bank, bank->num_sectors*JTAGSPI_MAX_TIMEOUT);
287 LOG_INFO("took %" PRId64 " ms", timeval_ms() - t0);
288 return retval;
289 }
290
291 static int jtagspi_sector_erase(struct flash_bank *bank, int sector)
292 {
293 struct jtagspi_flash_bank *info = bank->driver_priv;
294 int retval;
295 int64_t t0 = timeval_ms();
296
297 retval = jtagspi_write_enable(bank);
298 if (retval != ERROR_OK)
299 return retval;
300 jtagspi_cmd(bank, info->dev->erase_cmd, &bank->sectors[sector].offset, NULL, 0);
301 retval = jtagspi_wait(bank, JTAGSPI_MAX_TIMEOUT);
302 LOG_INFO("sector %d took %" PRId64 " ms", sector, timeval_ms() - t0);
303 return retval;
304 }
305
306 static int jtagspi_erase(struct flash_bank *bank, int first, int last)
307 {
308 int sector;
309 struct jtagspi_flash_bank *info = bank->driver_priv;
310 int retval = ERROR_OK;
311
312 LOG_DEBUG("erase from sector %d to sector %d", first, last);
313
314 if ((first < 0) || (last < first) || (last >= bank->num_sectors)) {
315 LOG_ERROR("Flash sector invalid");
316 return ERROR_FLASH_SECTOR_INVALID;
317 }
318
319 if (!(info->probed)) {
320 LOG_ERROR("Flash bank not probed");
321 return ERROR_FLASH_BANK_NOT_PROBED;
322 }
323
324 for (sector = first; sector <= last; sector++) {
325 if (bank->sectors[sector].is_protected) {
326 LOG_ERROR("Flash sector %d protected", sector);
327 return ERROR_FAIL;
328 }
329 }
330
331 if (first == 0 && last == (bank->num_sectors - 1)
332 && info->dev->chip_erase_cmd != info->dev->erase_cmd) {
333 LOG_DEBUG("Trying bulk erase.");
334 retval = jtagspi_bulk_erase(bank);
335 if (retval == ERROR_OK)
336 return retval;
337 else
338 LOG_WARNING("Bulk flash erase failed. Falling back to sector erase.");
339 }
340
341 if (info->dev->erase_cmd == 0x00)
342 return ERROR_FLASH_OPER_UNSUPPORTED;
343
344 for (sector = first; sector <= last; sector++) {
345 retval = jtagspi_sector_erase(bank, sector);
346 if (retval != ERROR_OK) {
347 LOG_ERROR("Sector erase failed.");
348 break;
349 }
350 }
351
352 return retval;
353 }
354
355 static int jtagspi_protect(struct flash_bank *bank, int set, int first, int last)
356 {
357 int sector;
358
359 for (sector = first; sector <= last; sector++)
360 bank->sectors[sector].is_protected = set;
361 return ERROR_OK;
362 }
363
364 static int jtagspi_read(struct flash_bank *bank, uint8_t *buffer, uint32_t offset, uint32_t count)
365 {
366 struct jtagspi_flash_bank *info = bank->driver_priv;
367
368 if (!(info->probed)) {
369 LOG_ERROR("Flash bank not yet probed.");
370 return ERROR_FLASH_BANK_NOT_PROBED;
371 }
372
373 jtagspi_cmd(bank, SPIFLASH_READ, &offset, buffer, -count*8);
374 return ERROR_OK;
375 }
376
377 static int jtagspi_page_write(struct flash_bank *bank, const uint8_t *buffer, uint32_t offset, uint32_t count)
378 {
379 int retval;
380
381 retval = jtagspi_write_enable(bank);
382 if (retval != ERROR_OK)
383 return retval;
384 jtagspi_cmd(bank, SPIFLASH_PAGE_PROGRAM, &offset, (uint8_t *) buffer, count*8);
385 return jtagspi_wait(bank, JTAGSPI_MAX_TIMEOUT);
386 }
387
388 static int jtagspi_write(struct flash_bank *bank, const uint8_t *buffer, uint32_t offset, uint32_t count)
389 {
390 struct jtagspi_flash_bank *info = bank->driver_priv;
391 int retval;
392 uint32_t n, pagesize;
393
394 if (!(info->probed)) {
395 LOG_ERROR("Flash bank not yet probed.");
396 return ERROR_FLASH_BANK_NOT_PROBED;
397 }
398
399 /* if no write pagesize, use reasonable default */
400 pagesize = info->dev->pagesize ? info->dev->pagesize : SPIFLASH_DEF_PAGESIZE;
401
402 for (n = 0; n < count; n += pagesize) {
403 retval = jtagspi_page_write(bank, buffer + n, offset + n,
404 MIN(count - n, pagesize));
405 if (retval != ERROR_OK) {
406 LOG_ERROR("page write error");
407 return retval;
408 }
409 LOG_DEBUG("wrote page at 0x%08" PRIx32, offset + n);
410 }
411 return ERROR_OK;
412 }
413
414 static int jtagspi_info(struct flash_bank *bank, char *buf, int buf_size)
415 {
416 struct jtagspi_flash_bank *info = bank->driver_priv;
417
418 if (!(info->probed)) {
419 snprintf(buf, buf_size, "\nJTAGSPI flash bank not probed yet\n");
420 return ERROR_OK;
421 }
422
423 snprintf(buf, buf_size, "\nSPIFI flash information:\n"
424 " Device \'%s\' (ID 0x%08" PRIx32 ")\n",
425 info->dev->name, info->dev->device_id);
426
427 return ERROR_OK;
428 }
429
430 struct flash_driver jtagspi_flash = {
431 .name = "jtagspi",
432 .flash_bank_command = jtagspi_flash_bank_command,
433 .erase = jtagspi_erase,
434 .protect = jtagspi_protect,
435 .write = jtagspi_write,
436 .read = jtagspi_read,
437 .probe = jtagspi_probe,
438 .auto_probe = jtagspi_probe,
439 .erase_check = default_flash_blank_check,
440 .info = jtagspi_info,
441 .free_driver_priv = default_flash_free_driver_priv,
442 };