1 /* 2 * Command for accessing SPI flash. 3 * 4 * Copyright (C) 2008 Atmel Corporation 5 * 6 * SPDX-License-Identifier: GPL-2.0+ 7 */ 8 9 #include <common.h> 10 #include <div64.h> 11 #include <dm.h> 12 #include <malloc.h> 13 #include <mapmem.h> 14 #include <spi.h> 15 #include <spi_flash.h> 16 #include <jffs2/jffs2.h> 17 #include <linux/mtd/mtd.h> 18 19 #include <asm/io.h> 20 #include <dm/device-internal.h> 21 22 static struct spi_flash *flash; 23 24 /* 25 * This function computes the length argument for the erase command. 26 * The length on which the command is to operate can be given in two forms: 27 * 1. <cmd> offset len - operate on <'offset', 'len') 28 * 2. <cmd> offset +len - operate on <'offset', 'round_up(len)') 29 * If the second form is used and the length doesn't fall on the 30 * sector boundary, than it will be adjusted to the next sector boundary. 31 * If it isn't in the flash, the function will fail (return -1). 32 * Input: 33 * arg: length specification (i.e. both command arguments) 34 * Output: 35 * len: computed length for operation 36 * Return: 37 * 1: success 38 * -1: failure (bad format, bad address). 39 */ 40 static int sf_parse_len_arg(char *arg, ulong *len) 41 { 42 char *ep; 43 char round_up_len; /* indicates if the "+length" form used */ 44 ulong len_arg; 45 46 round_up_len = 0; 47 if (*arg == '+') { 48 round_up_len = 1; 49 ++arg; 50 } 51 52 len_arg = simple_strtoul(arg, &ep, 16); 53 if (ep == arg || *ep != '\0') 54 return -1; 55 56 if (round_up_len && flash->sector_size > 0) 57 *len = ROUND(len_arg, flash->sector_size); 58 else 59 *len = len_arg; 60 61 return 1; 62 } 63 64 /** 65 * This function takes a byte length and a delta unit of time to compute the 66 * approximate bytes per second 67 * 68 * @param len amount of bytes currently processed 69 * @param start_ms start time of processing in ms 70 * @return bytes per second if OK, 0 on error 71 */ 72 static ulong bytes_per_second(unsigned int len, ulong start_ms) 73 { 74 /* less accurate but avoids overflow */ 75 if (len >= ((unsigned int) -1) / 1024) 76 return len / (max(get_timer(start_ms) / 1024, 1UL)); 77 else 78 return 1024 * len / max(get_timer(start_ms), 1UL); 79 } 80 81 static int do_spi_flash_probe(int argc, char * const argv[]) 82 { 83 unsigned int bus = CONFIG_SF_DEFAULT_BUS; 84 unsigned int cs = CONFIG_SF_DEFAULT_CS; 85 /* In DM mode, defaults speed and mode will be taken from DT */ 86 unsigned int speed = CONFIG_SF_DEFAULT_SPEED; 87 unsigned int mode = CONFIG_SF_DEFAULT_MODE; 88 char *endp; 89 #ifdef CONFIG_DM_SPI_FLASH 90 struct udevice *new, *bus_dev; 91 int ret; 92 #else 93 struct spi_flash *new; 94 #endif 95 96 if (argc >= 2) { 97 cs = simple_strtoul(argv[1], &endp, 0); 98 if (*argv[1] == 0 || (*endp != 0 && *endp != ':')) 99 return -1; 100 if (*endp == ':') { 101 if (endp[1] == 0) 102 return -1; 103 104 bus = cs; 105 cs = simple_strtoul(endp + 1, &endp, 0); 106 if (*endp != 0) 107 return -1; 108 } 109 } 110 111 if (argc >= 3) { 112 speed = simple_strtoul(argv[2], &endp, 0); 113 if (*argv[2] == 0 || *endp != 0) 114 return -1; 115 } 116 if (argc >= 4) { 117 mode = simple_strtoul(argv[3], &endp, 16); 118 if (*argv[3] == 0 || *endp != 0) 119 return -1; 120 } 121 122 #ifdef CONFIG_DM_SPI_FLASH 123 /* Remove the old device, otherwise probe will just be a nop */ 124 ret = spi_find_bus_and_cs(bus, cs, &bus_dev, &new); 125 if (!ret) { 126 device_remove(new, DM_REMOVE_NORMAL); 127 } 128 flash = NULL; 129 ret = spi_flash_probe_bus_cs(bus, cs, speed, mode, &new); 130 if (ret) { 131 printf("Failed to initialize SPI flash at %u:%u (error %d)\n", 132 bus, cs, ret); 133 return 1; 134 } 135 136 flash = dev_get_uclass_priv(new); 137 #else 138 if (flash) 139 spi_flash_free(flash); 140 141 new = spi_flash_probe(bus, cs, speed, mode); 142 flash = new; 143 144 if (!new) { 145 printf("Failed to initialize SPI flash at %u:%u\n", bus, cs); 146 return 1; 147 } 148 149 flash = new; 150 #endif 151 152 return 0; 153 } 154 155 /** 156 * Write a block of data to SPI flash, first checking if it is different from 157 * what is already there. 158 * 159 * If the data being written is the same, then *skipped is incremented by len. 160 * 161 * @param flash flash context pointer 162 * @param offset flash offset to write 163 * @param len number of bytes to write 164 * @param buf buffer to write from 165 * @param cmp_buf read buffer to use to compare data 166 * @param skipped Count of skipped data (incremented by this function) 167 * @return NULL if OK, else a string containing the stage which failed 168 */ 169 static const char *spi_flash_update_block(struct spi_flash *flash, u32 offset, 170 size_t len, const char *buf, char *cmp_buf, size_t *skipped) 171 { 172 char *ptr = (char *)buf; 173 174 debug("offset=%#x, sector_size=%#x, len=%#zx\n", 175 offset, flash->sector_size, len); 176 /* Read the entire sector so to allow for rewriting */ 177 if (spi_flash_read(flash, offset, flash->sector_size, cmp_buf)) 178 return "read"; 179 /* Compare only what is meaningful (len) */ 180 if (memcmp(cmp_buf, buf, len) == 0) { 181 debug("Skip region %x size %zx: no change\n", 182 offset, len); 183 *skipped += len; 184 return NULL; 185 } 186 /* Erase the entire sector */ 187 if (spi_flash_erase(flash, offset, flash->sector_size)) 188 return "erase"; 189 /* If it's a partial sector, copy the data into the temp-buffer */ 190 if (len != flash->sector_size) { 191 memcpy(cmp_buf, buf, len); 192 ptr = cmp_buf; 193 } 194 /* Write one complete sector */ 195 if (spi_flash_write(flash, offset, flash->sector_size, ptr)) 196 return "write"; 197 198 return NULL; 199 } 200 201 /** 202 * Update an area of SPI flash by erasing and writing any blocks which need 203 * to change. Existing blocks with the correct data are left unchanged. 204 * 205 * @param flash flash context pointer 206 * @param offset flash offset to write 207 * @param len number of bytes to write 208 * @param buf buffer to write from 209 * @return 0 if ok, 1 on error 210 */ 211 static int spi_flash_update(struct spi_flash *flash, u32 offset, 212 size_t len, const char *buf) 213 { 214 const char *err_oper = NULL; 215 char *cmp_buf; 216 const char *end = buf + len; 217 size_t todo; /* number of bytes to do in this pass */ 218 size_t skipped = 0; /* statistics */ 219 const ulong start_time = get_timer(0); 220 size_t scale = 1; 221 const char *start_buf = buf; 222 ulong delta; 223 224 if (end - buf >= 200) 225 scale = (end - buf) / 100; 226 cmp_buf = memalign(ARCH_DMA_MINALIGN, flash->sector_size); 227 if (cmp_buf) { 228 ulong last_update = get_timer(0); 229 230 for (; buf < end && !err_oper; buf += todo, offset += todo) { 231 todo = min_t(size_t, end - buf, flash->sector_size); 232 if (get_timer(last_update) > 100) { 233 printf(" \rUpdating, %zu%% %lu B/s", 234 100 - (end - buf) / scale, 235 bytes_per_second(buf - start_buf, 236 start_time)); 237 last_update = get_timer(0); 238 } 239 err_oper = spi_flash_update_block(flash, offset, todo, 240 buf, cmp_buf, &skipped); 241 } 242 } else { 243 err_oper = "malloc"; 244 } 245 free(cmp_buf); 246 putc('\r'); 247 if (err_oper) { 248 printf("SPI flash failed in %s step\n", err_oper); 249 return 1; 250 } 251 252 delta = get_timer(start_time); 253 printf("%zu bytes written, %zu bytes skipped", len - skipped, 254 skipped); 255 printf(" in %ld.%lds, speed %ld B/s\n", 256 delta / 1000, delta % 1000, bytes_per_second(len, start_time)); 257 258 return 0; 259 } 260 261 static int do_spi_flash_read_write(int argc, char * const argv[]) 262 { 263 unsigned long addr; 264 void *buf; 265 char *endp; 266 int ret = 1; 267 int dev = 0; 268 loff_t offset, len, maxsize; 269 270 if (argc < 3) 271 return -1; 272 273 addr = simple_strtoul(argv[1], &endp, 16); 274 if (*argv[1] == 0 || *endp != 0) 275 return -1; 276 277 if (mtd_arg_off_size(argc - 2, &argv[2], &dev, &offset, &len, 278 &maxsize, MTD_DEV_TYPE_NOR, flash->size)) 279 return -1; 280 281 /* Consistency checking */ 282 if (offset + len > flash->size) { 283 printf("ERROR: attempting %s past flash size (%#x)\n", 284 argv[0], flash->size); 285 return 1; 286 } 287 288 buf = map_physmem(addr, len, MAP_WRBACK); 289 if (!buf) { 290 puts("Failed to map physical memory\n"); 291 return 1; 292 } 293 294 if (strcmp(argv[0], "update") == 0) { 295 ret = spi_flash_update(flash, offset, len, buf); 296 } else if (strncmp(argv[0], "read", 4) == 0 || 297 strncmp(argv[0], "write", 5) == 0) { 298 int read; 299 300 read = strncmp(argv[0], "read", 4) == 0; 301 if (read) 302 ret = spi_flash_read(flash, offset, len, buf); 303 else 304 ret = spi_flash_write(flash, offset, len, buf); 305 306 printf("SF: %zu bytes @ %#x %s: ", (size_t)len, (u32)offset, 307 read ? "Read" : "Written"); 308 if (ret) 309 printf("ERROR %d\n", ret); 310 else 311 printf("OK\n"); 312 } 313 314 unmap_physmem(buf, len); 315 316 return ret == 0 ? 0 : 1; 317 } 318 319 static int do_spi_flash_erase(int argc, char * const argv[]) 320 { 321 int ret; 322 int dev = 0; 323 loff_t offset, len, maxsize; 324 ulong size; 325 326 if (argc < 3) 327 return -1; 328 329 if (mtd_arg_off(argv[1], &dev, &offset, &len, &maxsize, 330 MTD_DEV_TYPE_NOR, flash->size)) 331 return -1; 332 333 ret = sf_parse_len_arg(argv[2], &size); 334 if (ret != 1) 335 return -1; 336 337 /* Consistency checking */ 338 if (offset + size > flash->size) { 339 printf("ERROR: attempting %s past flash size (%#x)\n", 340 argv[0], flash->size); 341 return 1; 342 } 343 344 ret = spi_flash_erase(flash, offset, size); 345 printf("SF: %zu bytes @ %#x Erased: %s\n", (size_t)size, (u32)offset, 346 ret ? "ERROR" : "OK"); 347 348 return ret == 0 ? 0 : 1; 349 } 350 351 static int do_spi_protect(int argc, char * const argv[]) 352 { 353 int ret = 0; 354 loff_t start, len; 355 bool prot = false; 356 357 if (argc != 4) 358 return -1; 359 360 if (!str2off(argv[2], &start)) { 361 puts("start sector is not a valid number\n"); 362 return 1; 363 } 364 365 if (!str2off(argv[3], &len)) { 366 puts("len is not a valid number\n"); 367 return 1; 368 } 369 370 if (strcmp(argv[1], "lock") == 0) 371 prot = true; 372 else if (strcmp(argv[1], "unlock") == 0) 373 prot = false; 374 else 375 return -1; /* Unknown parameter */ 376 377 ret = spi_flash_protect(flash, start, len, prot); 378 379 return ret == 0 ? 0 : 1; 380 } 381 382 #ifdef CONFIG_CMD_SF_TEST 383 enum { 384 STAGE_ERASE, 385 STAGE_CHECK, 386 STAGE_WRITE, 387 STAGE_READ, 388 389 STAGE_COUNT, 390 }; 391 392 static char *stage_name[STAGE_COUNT] = { 393 "erase", 394 "check", 395 "write", 396 "read", 397 }; 398 399 struct test_info { 400 int stage; 401 int bytes; 402 unsigned base_ms; 403 unsigned time_ms[STAGE_COUNT]; 404 }; 405 406 static void show_time(struct test_info *test, int stage) 407 { 408 uint64_t speed; /* KiB/s */ 409 int bps; /* Bits per second */ 410 411 speed = (long long)test->bytes * 1000; 412 if (test->time_ms[stage]) 413 do_div(speed, test->time_ms[stage] * 1024); 414 bps = speed * 8; 415 416 printf("%d %s: %d ticks, %d KiB/s %d.%03d Mbps\n", stage, 417 stage_name[stage], test->time_ms[stage], 418 (int)speed, bps / 1000, bps % 1000); 419 } 420 421 static void spi_test_next_stage(struct test_info *test) 422 { 423 test->time_ms[test->stage] = get_timer(test->base_ms); 424 show_time(test, test->stage); 425 test->base_ms = get_timer(0); 426 test->stage++; 427 } 428 429 /** 430 * Run a test on the SPI flash 431 * 432 * @param flash SPI flash to use 433 * @param buf Source buffer for data to write 434 * @param len Size of data to read/write 435 * @param offset Offset within flash to check 436 * @param vbuf Verification buffer 437 * @return 0 if ok, -1 on error 438 */ 439 static int spi_flash_test(struct spi_flash *flash, uint8_t *buf, ulong len, 440 ulong offset, uint8_t *vbuf) 441 { 442 struct test_info test; 443 int i; 444 int erase_len; 445 446 printf("SPI flash test:\n"); 447 memset(&test, '\0', sizeof(test)); 448 test.base_ms = get_timer(0); 449 test.bytes = len; 450 451 erase_len = roundup(len, 4096); 452 if (spi_flash_erase(flash, offset, erase_len)) { 453 printf("Erase failed\n"); 454 return -1; 455 } 456 spi_test_next_stage(&test); 457 458 if (spi_flash_read(flash, offset, len, vbuf)) { 459 printf("Check read failed\n"); 460 return -1; 461 } 462 for (i = 0; i < len; i++) { 463 if (vbuf[i] != 0xff) { 464 printf("Check failed at %d\n", i); 465 print_buffer(i, vbuf + i, 1, 466 min_t(uint, len - i, 0x40), 0); 467 return -1; 468 } 469 } 470 spi_test_next_stage(&test); 471 472 if (spi_flash_write(flash, offset, len, buf)) { 473 printf("Write failed\n"); 474 return -1; 475 } 476 memset(vbuf, '\0', len); 477 spi_test_next_stage(&test); 478 479 if (spi_flash_read(flash, offset, len, vbuf)) { 480 printf("Read failed\n"); 481 return -1; 482 } 483 spi_test_next_stage(&test); 484 485 for (i = 0; i < len; i++) { 486 if (buf[i] != vbuf[i]) { 487 printf("Verify failed at %d, good data:\n", i); 488 print_buffer(i, buf + i, 1, 489 min_t(uint, len - i, 0x40), 0); 490 printf("Bad data:\n"); 491 print_buffer(i, vbuf + i, 1, 492 min_t(uint, len - i, 0x40), 0); 493 return -1; 494 } 495 } 496 printf("Test passed\n"); 497 for (i = 0; i < STAGE_COUNT; i++) 498 show_time(&test, i); 499 500 return 0; 501 } 502 503 static int do_spi_flash_test(int argc, char * const argv[]) 504 { 505 unsigned long offset; 506 unsigned long len; 507 uint8_t *buf, *from; 508 char *endp; 509 uint8_t *vbuf; 510 int ret; 511 int count; 512 int i; 513 514 if (argc < 3) 515 return -1; 516 offset = simple_strtoul(argv[1], &endp, 16); 517 if (*argv[1] == 0 || *endp != 0) 518 return -1; 519 len = simple_strtoul(argv[2], &endp, 16); 520 if (*argv[2] == 0 || *endp != 0) 521 return -1; 522 523 count = simple_strtoul(argv[3], &endp, 10); 524 if (!count) 525 count = 1; 526 527 vbuf = memalign(ARCH_DMA_MINALIGN, len); 528 if (!vbuf) { 529 printf("Cannot allocate memory (%lu bytes)\n", len); 530 return 1; 531 } 532 buf = memalign(ARCH_DMA_MINALIGN, len); 533 if (!buf) { 534 free(vbuf); 535 printf("Cannot allocate memory (%lu bytes)\n", len); 536 return 1; 537 } 538 539 from = map_sysmem(CONFIG_SYS_TEXT_BASE, 0); 540 memcpy(buf, from, len); 541 for (i = 0; i < count; i++) { 542 ret = spi_flash_test(flash, buf, len, offset, vbuf); 543 if (ret < 0) { 544 printf("Test Failed, passed count:%d\n", i); 545 break; 546 } 547 } 548 free(vbuf); 549 free(buf); 550 if (ret) { 551 printf("Test failed\n"); 552 return 1; 553 } 554 555 return 0; 556 } 557 #endif /* CONFIG_CMD_SF_TEST */ 558 559 static int do_spi_flash(cmd_tbl_t *cmdtp, int flag, int argc, 560 char * const argv[]) 561 { 562 const char *cmd; 563 int ret; 564 565 /* need at least two arguments */ 566 if (argc < 2) 567 goto usage; 568 569 cmd = argv[1]; 570 --argc; 571 ++argv; 572 573 if (strcmp(cmd, "probe") == 0) { 574 ret = do_spi_flash_probe(argc, argv); 575 goto done; 576 } 577 578 /* The remaining commands require a selected device */ 579 if (!flash) { 580 puts("No SPI flash selected. Please run `sf probe'\n"); 581 return 1; 582 } 583 584 if (strcmp(cmd, "read") == 0 || strcmp(cmd, "write") == 0 || 585 strcmp(cmd, "update") == 0) 586 ret = do_spi_flash_read_write(argc, argv); 587 else if (strcmp(cmd, "erase") == 0) 588 ret = do_spi_flash_erase(argc, argv); 589 else if (strcmp(cmd, "protect") == 0) 590 ret = do_spi_protect(argc, argv); 591 #ifdef CONFIG_CMD_SF_TEST 592 else if (!strcmp(cmd, "test")) 593 ret = do_spi_flash_test(argc, argv); 594 #endif 595 else 596 ret = -1; 597 598 done: 599 if (ret != -1) 600 return ret; 601 602 usage: 603 return CMD_RET_USAGE; 604 } 605 606 #ifdef CONFIG_CMD_SF_TEST 607 #define SF_TEST_HELP "\nsf test offset len " \ 608 "- run a very basic destructive test" 609 #else 610 #define SF_TEST_HELP 611 #endif 612 613 U_BOOT_CMD( 614 sf, 5, 1, do_spi_flash, 615 "SPI flash sub-system", 616 "probe [[bus:]cs] [hz] [mode] - init flash device on given SPI bus\n" 617 " and chip select\n" 618 "sf read addr offset|partition len - read `len' bytes starting at\n" 619 " `offset' or from start of mtd\n" 620 " `partition'to memory at `addr'\n" 621 "sf write addr offset|partition len - write `len' bytes from memory\n" 622 " at `addr' to flash at `offset'\n" 623 " or to start of mtd `partition'\n" 624 "sf erase offset|partition [+]len - erase `len' bytes from `offset'\n" 625 " or from start of mtd `partition'\n" 626 " `+len' round up `len' to block size\n" 627 "sf update addr offset|partition len - erase and write `len' bytes from memory\n" 628 " at `addr' to flash at `offset'\n" 629 " or to start of mtd `partition'\n" 630 "sf protect lock/unlock sector len - protect/unprotect 'len' bytes starting\n" 631 " at address 'sector'\n" 632 SF_TEST_HELP 633 ); 634