1 /* 2 * Freescale i.MX28 image generator 3 * 4 * Copyright (C) 2011 Marek Vasut <marek.vasut@gmail.com> 5 * on behalf of DENX Software Engineering GmbH 6 * 7 * See file CREDITS for list of people who contributed to this 8 * project. 9 * 10 * This program is free software; you can redistribute it and/or 11 * modify it under the terms of the GNU General Public License as 12 * published by the Free Software Foundation; either version 2 of 13 * the License, or (at your option) any later version. 14 * 15 * This program is distributed in the hope that it will be useful, 16 * but WITHOUT ANY WARRANTY; without even the implied warranty of 17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 18 * GNU General Public License for more details. 19 * 20 * You should have received a copy of the GNU General Public License 21 * along with this program; if not, write to the Free Software 22 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 23 * MA 02111-1307 USA 24 */ 25 26 #include <fcntl.h> 27 #include <sys/stat.h> 28 #include <sys/types.h> 29 #include <unistd.h> 30 31 #include "compiler.h" 32 33 /* 34 * Default BCB layout. 35 * 36 * TWEAK this if you have blown any OCOTP fuses. 37 */ 38 #define STRIDE_PAGES 64 39 #define STRIDE_COUNT 4 40 41 /* 42 * Layout for 256Mb big NAND with 2048b page size, 64b OOB size and 43 * 128kb erase size. 44 * 45 * TWEAK this if you have different kind of NAND chip. 46 */ 47 uint32_t nand_writesize = 2048; 48 uint32_t nand_oobsize = 64; 49 uint32_t nand_erasesize = 128 * 1024; 50 51 /* 52 * Sector on which the SigmaTel boot partition (0x53) starts. 53 */ 54 uint32_t sd_sector = 2048; 55 56 /* 57 * Each of the U-Boot bootstreams is at maximum 1MB big. 58 * 59 * TWEAK this if, for some wild reason, you need to boot bigger image. 60 */ 61 #define MAX_BOOTSTREAM_SIZE (1 * 1024 * 1024) 62 63 /* i.MX28 NAND controller-specific constants. DO NOT TWEAK! */ 64 #define MXS_NAND_DMA_DESCRIPTOR_COUNT 4 65 #define MXS_NAND_CHUNK_DATA_CHUNK_SIZE 512 66 #define MXS_NAND_METADATA_SIZE 10 67 #define MXS_NAND_COMMAND_BUFFER_SIZE 32 68 69 struct mx28_nand_fcb { 70 uint32_t checksum; 71 uint32_t fingerprint; 72 uint32_t version; 73 struct { 74 uint8_t data_setup; 75 uint8_t data_hold; 76 uint8_t address_setup; 77 uint8_t dsample_time; 78 uint8_t nand_timing_state; 79 uint8_t rea; 80 uint8_t rloh; 81 uint8_t rhoh; 82 } timing; 83 uint32_t page_data_size; 84 uint32_t total_page_size; 85 uint32_t sectors_per_block; 86 uint32_t number_of_nands; /* Ignored */ 87 uint32_t total_internal_die; /* Ignored */ 88 uint32_t cell_type; /* Ignored */ 89 uint32_t ecc_block_n_ecc_type; 90 uint32_t ecc_block_0_size; 91 uint32_t ecc_block_n_size; 92 uint32_t ecc_block_0_ecc_type; 93 uint32_t metadata_bytes; 94 uint32_t num_ecc_blocks_per_page; 95 uint32_t ecc_block_n_ecc_level_sdk; /* Ignored */ 96 uint32_t ecc_block_0_size_sdk; /* Ignored */ 97 uint32_t ecc_block_n_size_sdk; /* Ignored */ 98 uint32_t ecc_block_0_ecc_level_sdk; /* Ignored */ 99 uint32_t num_ecc_blocks_per_page_sdk; /* Ignored */ 100 uint32_t metadata_bytes_sdk; /* Ignored */ 101 uint32_t erase_threshold; 102 uint32_t boot_patch; 103 uint32_t patch_sectors; 104 uint32_t firmware1_starting_sector; 105 uint32_t firmware2_starting_sector; 106 uint32_t sectors_in_firmware1; 107 uint32_t sectors_in_firmware2; 108 uint32_t dbbt_search_area_start_address; 109 uint32_t badblock_marker_byte; 110 uint32_t badblock_marker_start_bit; 111 uint32_t bb_marker_physical_offset; 112 }; 113 114 struct mx28_nand_dbbt { 115 uint32_t checksum; 116 uint32_t fingerprint; 117 uint32_t version; 118 uint32_t number_bb; 119 uint32_t number_2k_pages_bb; 120 }; 121 122 struct mx28_nand_bbt { 123 uint32_t nand; 124 uint32_t number_bb; 125 uint32_t badblock[510]; 126 }; 127 128 struct mx28_sd_drive_info { 129 uint32_t chip_num; 130 uint32_t drive_type; 131 uint32_t tag; 132 uint32_t first_sector_number; 133 uint32_t sector_count; 134 }; 135 136 struct mx28_sd_config_block { 137 uint32_t signature; 138 uint32_t primary_boot_tag; 139 uint32_t secondary_boot_tag; 140 uint32_t num_copies; 141 struct mx28_sd_drive_info drv_info[1]; 142 }; 143 144 static inline uint32_t mx28_nand_ecc_size_in_bits(uint32_t ecc_strength) 145 { 146 return ecc_strength * 13; 147 } 148 149 static inline uint32_t mx28_nand_get_ecc_strength(uint32_t page_data_size, 150 uint32_t page_oob_size) 151 { 152 if (page_data_size == 2048) 153 return 8; 154 155 if (page_data_size == 4096) { 156 if (page_oob_size == 128) 157 return 8; 158 159 if (page_oob_size == 218) 160 return 16; 161 } 162 163 return 0; 164 } 165 166 static inline uint32_t mx28_nand_get_mark_offset(uint32_t page_data_size, 167 uint32_t ecc_strength) 168 { 169 uint32_t chunk_data_size_in_bits; 170 uint32_t chunk_ecc_size_in_bits; 171 uint32_t chunk_total_size_in_bits; 172 uint32_t block_mark_chunk_number; 173 uint32_t block_mark_chunk_bit_offset; 174 uint32_t block_mark_bit_offset; 175 176 chunk_data_size_in_bits = MXS_NAND_CHUNK_DATA_CHUNK_SIZE * 8; 177 chunk_ecc_size_in_bits = mx28_nand_ecc_size_in_bits(ecc_strength); 178 179 chunk_total_size_in_bits = 180 chunk_data_size_in_bits + chunk_ecc_size_in_bits; 181 182 /* Compute the bit offset of the block mark within the physical page. */ 183 block_mark_bit_offset = page_data_size * 8; 184 185 /* Subtract the metadata bits. */ 186 block_mark_bit_offset -= MXS_NAND_METADATA_SIZE * 8; 187 188 /* 189 * Compute the chunk number (starting at zero) in which the block mark 190 * appears. 191 */ 192 block_mark_chunk_number = 193 block_mark_bit_offset / chunk_total_size_in_bits; 194 195 /* 196 * Compute the bit offset of the block mark within its chunk, and 197 * validate it. 198 */ 199 block_mark_chunk_bit_offset = block_mark_bit_offset - 200 (block_mark_chunk_number * chunk_total_size_in_bits); 201 202 if (block_mark_chunk_bit_offset > chunk_data_size_in_bits) 203 return 1; 204 205 /* 206 * Now that we know the chunk number in which the block mark appears, 207 * we can subtract all the ECC bits that appear before it. 208 */ 209 block_mark_bit_offset -= 210 block_mark_chunk_number * chunk_ecc_size_in_bits; 211 212 return block_mark_bit_offset; 213 } 214 215 static inline uint32_t mx28_nand_mark_byte_offset(void) 216 { 217 uint32_t ecc_strength; 218 ecc_strength = mx28_nand_get_ecc_strength(nand_writesize, nand_oobsize); 219 return mx28_nand_get_mark_offset(nand_writesize, ecc_strength) >> 3; 220 } 221 222 static inline uint32_t mx28_nand_mark_bit_offset(void) 223 { 224 uint32_t ecc_strength; 225 ecc_strength = mx28_nand_get_ecc_strength(nand_writesize, nand_oobsize); 226 return mx28_nand_get_mark_offset(nand_writesize, ecc_strength) & 0x7; 227 } 228 229 static uint32_t mx28_nand_block_csum(uint8_t *block, uint32_t size) 230 { 231 uint32_t csum = 0; 232 int i; 233 234 for (i = 0; i < size; i++) 235 csum += block[i]; 236 237 return csum ^ 0xffffffff; 238 } 239 240 static struct mx28_nand_fcb *mx28_nand_get_fcb(uint32_t size) 241 { 242 struct mx28_nand_fcb *fcb; 243 uint32_t bcb_size_bytes; 244 uint32_t stride_size_bytes; 245 uint32_t bootstream_size_pages; 246 uint32_t fw1_start_page; 247 uint32_t fw2_start_page; 248 249 fcb = malloc(nand_writesize); 250 if (!fcb) { 251 printf("MX28 NAND: Unable to allocate FCB\n"); 252 return NULL; 253 } 254 255 memset(fcb, 0, nand_writesize); 256 257 fcb->fingerprint = 0x20424346; 258 fcb->version = 0x01000000; 259 260 /* 261 * FIXME: These here are default values as found in kobs-ng. We should 262 * probably retrieve the data from NAND or something. 263 */ 264 fcb->timing.data_setup = 80; 265 fcb->timing.data_hold = 60; 266 fcb->timing.address_setup = 25; 267 fcb->timing.dsample_time = 6; 268 269 fcb->page_data_size = nand_writesize; 270 fcb->total_page_size = nand_writesize + nand_oobsize; 271 fcb->sectors_per_block = nand_erasesize / nand_writesize; 272 273 fcb->num_ecc_blocks_per_page = (nand_writesize / 512) - 1; 274 fcb->ecc_block_0_size = 512; 275 fcb->ecc_block_n_size = 512; 276 fcb->metadata_bytes = 10; 277 278 if (nand_writesize == 2048) { 279 fcb->ecc_block_n_ecc_type = 4; 280 fcb->ecc_block_0_ecc_type = 4; 281 } else if (nand_writesize == 4096) { 282 if (nand_oobsize == 128) { 283 fcb->ecc_block_n_ecc_type = 4; 284 fcb->ecc_block_0_ecc_type = 4; 285 } else if (nand_oobsize == 218) { 286 fcb->ecc_block_n_ecc_type = 8; 287 fcb->ecc_block_0_ecc_type = 8; 288 } 289 } 290 291 if (fcb->ecc_block_n_ecc_type == 0) { 292 printf("MX28 NAND: Unsupported NAND geometry\n"); 293 goto err; 294 } 295 296 fcb->boot_patch = 0; 297 fcb->patch_sectors = 0; 298 299 fcb->badblock_marker_byte = mx28_nand_mark_byte_offset(); 300 fcb->badblock_marker_start_bit = mx28_nand_mark_bit_offset(); 301 fcb->bb_marker_physical_offset = nand_writesize; 302 303 stride_size_bytes = STRIDE_PAGES * nand_writesize; 304 bcb_size_bytes = stride_size_bytes * STRIDE_COUNT; 305 306 bootstream_size_pages = (size + (nand_writesize - 1)) / 307 nand_writesize; 308 309 fw1_start_page = 2 * bcb_size_bytes / nand_writesize; 310 fw2_start_page = (2 * bcb_size_bytes + MAX_BOOTSTREAM_SIZE) / 311 nand_writesize; 312 313 fcb->firmware1_starting_sector = fw1_start_page; 314 fcb->firmware2_starting_sector = fw2_start_page; 315 fcb->sectors_in_firmware1 = bootstream_size_pages; 316 fcb->sectors_in_firmware2 = bootstream_size_pages; 317 318 fcb->dbbt_search_area_start_address = STRIDE_PAGES * STRIDE_COUNT; 319 320 return fcb; 321 322 err: 323 free(fcb); 324 return NULL; 325 } 326 327 static struct mx28_nand_dbbt *mx28_nand_get_dbbt(void) 328 { 329 struct mx28_nand_dbbt *dbbt; 330 331 dbbt = malloc(nand_writesize); 332 if (!dbbt) { 333 printf("MX28 NAND: Unable to allocate DBBT\n"); 334 return NULL; 335 } 336 337 memset(dbbt, 0, nand_writesize); 338 339 dbbt->fingerprint = 0x54424244; 340 dbbt->version = 0x1; 341 342 return dbbt; 343 } 344 345 static inline uint8_t mx28_nand_parity_13_8(const uint8_t b) 346 { 347 uint32_t parity = 0, tmp; 348 349 tmp = ((b >> 6) ^ (b >> 5) ^ (b >> 3) ^ (b >> 2)) & 1; 350 parity |= tmp << 0; 351 352 tmp = ((b >> 7) ^ (b >> 5) ^ (b >> 4) ^ (b >> 2) ^ (b >> 1)) & 1; 353 parity |= tmp << 1; 354 355 tmp = ((b >> 7) ^ (b >> 6) ^ (b >> 5) ^ (b >> 1) ^ (b >> 0)) & 1; 356 parity |= tmp << 2; 357 358 tmp = ((b >> 7) ^ (b >> 4) ^ (b >> 3) ^ (b >> 0)) & 1; 359 parity |= tmp << 3; 360 361 tmp = ((b >> 6) ^ (b >> 4) ^ (b >> 3) ^ 362 (b >> 2) ^ (b >> 1) ^ (b >> 0)) & 1; 363 parity |= tmp << 4; 364 365 return parity; 366 } 367 368 static uint8_t *mx28_nand_fcb_block(struct mx28_nand_fcb *fcb) 369 { 370 uint8_t *block; 371 uint8_t *ecc; 372 int i; 373 374 block = malloc(nand_writesize + nand_oobsize); 375 if (!block) { 376 printf("MX28 NAND: Unable to allocate FCB block\n"); 377 return NULL; 378 } 379 380 memset(block, 0, nand_writesize + nand_oobsize); 381 382 /* Update the FCB checksum */ 383 fcb->checksum = mx28_nand_block_csum(((uint8_t *)fcb) + 4, 508); 384 385 /* Figure 12-11. in iMX28RM, rev. 1, says FCB is at offset 12 */ 386 memcpy(block + 12, fcb, sizeof(struct mx28_nand_fcb)); 387 388 /* ECC is at offset 12 + 512 */ 389 ecc = block + 12 + 512; 390 391 /* Compute the ECC parity */ 392 for (i = 0; i < sizeof(struct mx28_nand_fcb); i++) 393 ecc[i] = mx28_nand_parity_13_8(block[i + 12]); 394 395 return block; 396 } 397 398 static int mx28_nand_write_fcb(struct mx28_nand_fcb *fcb, char *buf) 399 { 400 uint32_t offset; 401 uint8_t *fcbblock; 402 int ret = 0; 403 int i; 404 405 fcbblock = mx28_nand_fcb_block(fcb); 406 if (!fcbblock) 407 return -1; 408 409 for (i = 0; i < STRIDE_PAGES * STRIDE_COUNT; i += STRIDE_PAGES) { 410 offset = i * nand_writesize; 411 memcpy(buf + offset, fcbblock, nand_writesize + nand_oobsize); 412 } 413 414 free(fcbblock); 415 return ret; 416 } 417 418 static int mx28_nand_write_dbbt(struct mx28_nand_dbbt *dbbt, char *buf) 419 { 420 uint32_t offset; 421 int i = STRIDE_PAGES * STRIDE_COUNT; 422 423 for (; i < 2 * STRIDE_PAGES * STRIDE_COUNT; i += STRIDE_PAGES) { 424 offset = i * nand_writesize; 425 memcpy(buf + offset, dbbt, sizeof(struct mx28_nand_dbbt)); 426 } 427 428 return 0; 429 } 430 431 static int mx28_nand_write_firmware(struct mx28_nand_fcb *fcb, int infd, 432 char *buf) 433 { 434 int ret; 435 off_t size; 436 uint32_t offset1, offset2; 437 438 size = lseek(infd, 0, SEEK_END); 439 lseek(infd, 0, SEEK_SET); 440 441 offset1 = fcb->firmware1_starting_sector * nand_writesize; 442 offset2 = fcb->firmware2_starting_sector * nand_writesize; 443 444 ret = read(infd, buf + offset1, size); 445 if (ret != size) 446 return -1; 447 448 memcpy(buf + offset2, buf + offset1, size); 449 450 return 0; 451 } 452 453 void usage(void) 454 { 455 printf( 456 "Usage: mx28image [ops] <type> <infile> <outfile>\n" 457 "Augment BootStream file with a proper header for i.MX28 boot\n" 458 "\n" 459 " <type> type of image:\n" 460 " \"nand\" for NAND image\n" 461 " \"sd\" for SD image\n" 462 " <infile> input file, the u-boot.sb bootstream\n" 463 " <outfile> output file, the bootable image\n" 464 "\n"); 465 printf( 466 "For NAND boot, these options are accepted:\n" 467 " -w <size> NAND page size\n" 468 " -o <size> NAND OOB size\n" 469 " -e <size> NAND erase size\n" 470 "\n" 471 "For SD boot, these options are accepted:\n" 472 " -p <sector> Sector where the SGTL partition starts\n" 473 ); 474 } 475 476 static int mx28_create_nand_image(int infd, int outfd) 477 { 478 struct mx28_nand_fcb *fcb; 479 struct mx28_nand_dbbt *dbbt; 480 int ret = -1; 481 char *buf; 482 int size; 483 ssize_t wr_size; 484 485 size = nand_writesize * 512 + 2 * MAX_BOOTSTREAM_SIZE; 486 487 buf = malloc(size); 488 if (!buf) { 489 printf("Can not allocate output buffer of %d bytes\n", size); 490 goto err0; 491 } 492 493 memset(buf, 0, size); 494 495 fcb = mx28_nand_get_fcb(MAX_BOOTSTREAM_SIZE); 496 if (!fcb) { 497 printf("Unable to compile FCB\n"); 498 goto err1; 499 } 500 501 dbbt = mx28_nand_get_dbbt(); 502 if (!dbbt) { 503 printf("Unable to compile DBBT\n"); 504 goto err2; 505 } 506 507 ret = mx28_nand_write_fcb(fcb, buf); 508 if (ret) { 509 printf("Unable to write FCB to buffer\n"); 510 goto err3; 511 } 512 513 ret = mx28_nand_write_dbbt(dbbt, buf); 514 if (ret) { 515 printf("Unable to write DBBT to buffer\n"); 516 goto err3; 517 } 518 519 ret = mx28_nand_write_firmware(fcb, infd, buf); 520 if (ret) { 521 printf("Unable to write firmware to buffer\n"); 522 goto err3; 523 } 524 525 wr_size = write(outfd, buf, size); 526 if (wr_size != size) { 527 ret = -1; 528 goto err3; 529 } 530 531 ret = 0; 532 533 err3: 534 free(dbbt); 535 err2: 536 free(fcb); 537 err1: 538 free(buf); 539 err0: 540 return ret; 541 } 542 543 static int mx28_create_sd_image(int infd, int outfd) 544 { 545 int ret = -1; 546 uint32_t *buf; 547 int size; 548 off_t fsize; 549 ssize_t wr_size; 550 struct mx28_sd_config_block *cb; 551 552 fsize = lseek(infd, 0, SEEK_END); 553 lseek(infd, 0, SEEK_SET); 554 size = fsize + 512; 555 556 buf = malloc(size); 557 if (!buf) { 558 printf("Can not allocate output buffer of %d bytes\n", size); 559 goto err0; 560 } 561 562 ret = read(infd, (uint8_t *)buf + 512, fsize); 563 if (ret != fsize) { 564 ret = -1; 565 goto err1; 566 } 567 568 cb = (struct mx28_sd_config_block *)buf; 569 570 cb->signature = 0x00112233; 571 cb->primary_boot_tag = 0x1; 572 cb->secondary_boot_tag = 0x1; 573 cb->num_copies = 1; 574 cb->drv_info[0].chip_num = 0x0; 575 cb->drv_info[0].drive_type = 0x0; 576 cb->drv_info[0].tag = 0x1; 577 cb->drv_info[0].first_sector_number = sd_sector + 1; 578 cb->drv_info[0].sector_count = (size - 1) / 512; 579 580 wr_size = write(outfd, buf, size); 581 if (wr_size != size) { 582 ret = -1; 583 goto err1; 584 } 585 586 ret = 0; 587 588 err1: 589 free(buf); 590 err0: 591 return ret; 592 } 593 594 int parse_ops(int argc, char **argv) 595 { 596 int i; 597 int tmp; 598 char *end; 599 enum param { 600 PARAM_WRITE, 601 PARAM_OOB, 602 PARAM_ERASE, 603 PARAM_PART, 604 PARAM_SD, 605 PARAM_NAND 606 }; 607 int type; 608 609 if (argc < 4) 610 return -1; 611 612 for (i = 1; i < argc; i++) { 613 if (!strncmp(argv[i], "-w", 2)) 614 type = PARAM_WRITE; 615 else if (!strncmp(argv[i], "-o", 2)) 616 type = PARAM_OOB; 617 else if (!strncmp(argv[i], "-e", 2)) 618 type = PARAM_ERASE; 619 else if (!strncmp(argv[i], "-p", 2)) 620 type = PARAM_PART; 621 else /* SD/MMC */ 622 break; 623 624 tmp = strtol(argv[++i], &end, 10); 625 if (tmp % 2) 626 return -1; 627 if (tmp <= 0) 628 return -1; 629 630 if (type == PARAM_WRITE) 631 nand_writesize = tmp; 632 if (type == PARAM_OOB) 633 nand_oobsize = tmp; 634 if (type == PARAM_ERASE) 635 nand_erasesize = tmp; 636 if (type == PARAM_PART) 637 sd_sector = tmp; 638 } 639 640 if (strcmp(argv[i], "sd") && strcmp(argv[i], "nand")) 641 return -1; 642 643 if (i + 3 != argc) 644 return -1; 645 646 return i; 647 } 648 649 int main(int argc, char **argv) 650 { 651 int infd, outfd; 652 int ret = 0; 653 int offset; 654 655 offset = parse_ops(argc, argv); 656 if (offset < 0) { 657 usage(); 658 ret = 1; 659 goto err1; 660 } 661 662 infd = open(argv[offset + 1], O_RDONLY); 663 if (infd < 0) { 664 printf("Input BootStream file can not be opened\n"); 665 ret = 2; 666 goto err1; 667 } 668 669 outfd = open(argv[offset + 2], O_CREAT | O_TRUNC | O_WRONLY, 670 S_IRUSR | S_IWUSR); 671 if (outfd < 0) { 672 printf("Output file can not be created\n"); 673 ret = 3; 674 goto err2; 675 } 676 677 if (!strcmp(argv[offset], "sd")) 678 ret = mx28_create_sd_image(infd, outfd); 679 else if (!strcmp(argv[offset], "nand")) 680 ret = mx28_create_nand_image(infd, outfd); 681 682 close(outfd); 683 err2: 684 close(infd); 685 err1: 686 return ret; 687 } 688