git.sur5r.net Git - u-boot/blob - tools/mxsboot.c

   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: mxsboot [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 }