int main (int argc, char **argv) { int ch, i, boot, imagefd = 0, force, unlocked; char *erase[MAX_ARGS], *device = NULL; char *fis_layout = NULL; size_t offset = 0, part_offset = 0, dump_len = 0; enum { CMD_ERASE, CMD_WRITE, CMD_UNLOCK, CMD_JFFS2WRITE, CMD_FIXTRX, CMD_FIXSEAMA, CMD_VERIFY, CMD_DUMP, } cmd = -1; erase[0] = NULL; boot = 0; force = 0; buflen = 0; quiet = 0; no_erase = 0; while ((ch = getopt(argc, argv, #ifdef FIS_SUPPORT "F:" #endif "frnqe:d:s:j:p:o:l:")) != -1) switch (ch) { case 'f': force = 1; break; case 'r': boot = 1; break; case 'n': no_erase = 1; break; case 'j': jffs2file = optarg; break; case 's': errno = 0; jffs2_skip_bytes = strtoul(optarg, 0, 0); if (errno) { fprintf(stderr, "-s: illegal numeric string\n"); usage(); } break; case 'q': quiet++; break; case 'e': i = 0; while ((erase[i] != NULL) && ((i + 1) < MAX_ARGS)) i++; erase[i++] = optarg; erase[i] = NULL; break; case 'd': jffs2dir = optarg; break; case 'p': errno = 0; part_offset = strtoul(optarg, 0, 0); if (errno) { fprintf(stderr, "-p: illegal numeric string\n"); usage(); } break; case 'l': errno = 0; dump_len = strtoul(optarg, 0, 0); if (errno) { fprintf(stderr, "-l: illegal numeric string\n"); usage(); } break; case 'o': errno = 0; offset = strtoul(optarg, 0, 0); if (errno) { fprintf(stderr, "-o: illegal numeric string\n"); usage(); } break; #ifdef FIS_SUPPORT case 'F': fis_layout = optarg; break; #endif case '?': default: usage(); } argc -= optind; argv += optind; if (argc < 2) usage(); if ((strcmp(argv[0], "unlock") == 0) && (argc == 2)) { cmd = CMD_UNLOCK; device = argv[1]; } else if ((strcmp(argv[0], "erase") == 0) && (argc == 2)) { cmd = CMD_ERASE; device = argv[1]; } else if (((strcmp(argv[0], "fixtrx") == 0) && (argc == 2)) && mtd_fixtrx) { cmd = CMD_FIXTRX; device = argv[1]; } else if (((strcmp(argv[0], "fixseama") == 0) && (argc == 2)) && mtd_fixseama) { cmd = CMD_FIXSEAMA; device = argv[1]; } else if ((strcmp(argv[0], "verify") == 0) && (argc == 3)) { cmd = CMD_VERIFY; imagefile = argv[1]; device = argv[2]; } else if ((strcmp(argv[0], "dump") == 0) && (argc == 2)) { cmd = CMD_DUMP; device = argv[1]; } else if ((strcmp(argv[0], "write") == 0) && (argc == 3)) { cmd = CMD_WRITE; device = argv[2]; if (strcmp(argv[1], "-") == 0) { imagefile = "<stdin>"; imagefd = 0; } else { imagefile = argv[1]; if ((imagefd = open(argv[1], O_RDONLY)) < 0) { fprintf(stderr, "Couldn't open image file: %s!\n", imagefile); exit(1); } } if (!mtd_check(device)) { fprintf(stderr, "Can't open device for writing!\n"); exit(1); } /* check trx file before erasing or writing anything */ if (!image_check(imagefd, device) && !force) { fprintf(stderr, "Image check failed.\n"); exit(1); } } else if ((strcmp(argv[0], "jffs2write") == 0) && (argc == 3)) { cmd = CMD_JFFS2WRITE; device = argv[2]; imagefile = argv[1]; if (!mtd_check(device)) { fprintf(stderr, "Can't open device for writing!\n"); exit(1); } } else { usage(); } sync(); i = 0; unlocked = 0; while (erase[i] != NULL) { mtd_unlock(erase[i]); mtd_erase(erase[i]); if (strcmp(erase[i], device) == 0) unlocked = 1; i++; } switch (cmd) { case CMD_UNLOCK: if (!unlocked) mtd_unlock(device); break; case CMD_VERIFY: mtd_verify(device, imagefile); break; case CMD_DUMP: mtd_dump(device, offset, dump_len); break; case CMD_ERASE: if (!unlocked) mtd_unlock(device); mtd_erase(device); break; case CMD_WRITE: if (!unlocked) mtd_unlock(device); mtd_write(imagefd, device, fis_layout, part_offset); break; case CMD_JFFS2WRITE: if (!unlocked) mtd_unlock(device); mtd_write_jffs2(device, imagefile, jffs2dir); break; case CMD_FIXTRX: if (mtd_fixtrx) { mtd_fixtrx(device, offset); } case CMD_FIXSEAMA: if (mtd_fixseama) mtd_fixseama(device, 0); break; } sync(); if (boot) do_reboot(); return 0; }
static int mtd_write(int imagefd, const char *mtd, char *fis_layout, size_t part_offset) { char *next = NULL; char *str = NULL; int fd, result; ssize_t r, w, e; ssize_t skip = 0; uint32_t offset = 0; int jffs2_replaced = 0; int skip_bad_blocks = 0; #ifdef FIS_SUPPORT static struct fis_part new_parts[MAX_ARGS]; static struct fis_part old_parts[MAX_ARGS]; int n_new = 0, n_old = 0; if (fis_layout) { const char *tmp = mtd; char *word, *brkt; int ret; memset(&old_parts, 0, sizeof(old_parts)); memset(&new_parts, 0, sizeof(new_parts)); do { next = strchr(tmp, ':'); if (!next) next = (char *) tmp + strlen(tmp); memcpy(old_parts[n_old].name, tmp, next - tmp); n_old++; tmp = next + 1; } while(*next); for (word = strtok_r(fis_layout, ",", &brkt); word; word = strtok_r(NULL, ",", &brkt)) { tmp = strtok(word, ":"); strncpy((char *) new_parts[n_new].name, tmp, sizeof(new_parts[n_new].name) - 1); tmp = strtok(NULL, ":"); if (!tmp) goto next; new_parts[n_new].size = strtoul(tmp, NULL, 0); tmp = strtok(NULL, ":"); if (!tmp) goto next; new_parts[n_new].loadaddr = strtoul(tmp, NULL, 16); next: n_new++; } ret = fis_validate(old_parts, n_old, new_parts, n_new); if (ret < 0) { fprintf(stderr, "Failed to validate the new FIS partition table\n"); exit(1); } if (ret == 0) fis_layout = NULL; } #endif if (strchr(mtd, ':')) { str = strdup(mtd); mtd = str; } r = 0; resume: next = strchr(mtd, ':'); if (next) { *next = 0; next++; } fd = mtd_check_open(mtd); if(fd < 0) { fprintf(stderr, "Could not open mtd device: %s\n", mtd); exit(1); } if (part_offset > 0) { fprintf(stderr, "Seeking on mtd device '%s' to: %zu\n", mtd, part_offset); lseek(fd, part_offset, SEEK_SET); } indicate_writing(mtd); w = e = 0; for (;;) { /* buffer may contain data already (from trx check or last mtd partition write attempt) */ while (buflen < erasesize) { r = read(imagefd, buf + buflen, erasesize - buflen); if (r < 0) { if ((errno == EINTR) || (errno == EAGAIN)) continue; else { perror("read"); break; } } if (r == 0) break; buflen += r; } if (buflen == 0) break; if (buflen < erasesize) { /* Pad block to eraseblock size */ memset(&buf[buflen], 0xff, erasesize - buflen); buflen = erasesize; } if (skip > 0) { skip -= buflen; buflen = 0; if (skip <= 0) indicate_writing(mtd); continue; } if (jffs2file && w >= jffs2_skip_bytes) { if (memcmp(buf, JFFS2_EOF, sizeof(JFFS2_EOF) - 1) == 0) { if (!quiet) fprintf(stderr, "\b\b\b "); if (quiet < 2) fprintf(stderr, "\nAppending jffs2 data from %s to %s..\n.", jffs2file, mtd); /* got an EOF marker - this is the place to add some jffs2 data */ skip = mtd_replace_jffs2(mtd, fd, e, jffs2file); jffs2_replaced = 1; /* don't add it again */ jffs2file = NULL; w += skip; e += skip; skip -= buflen; buflen = 0; offset = 0; continue; } /* no EOF marker, make sure we figure out the last inode number * before appending some data */ mtd_parse_jffs2data(buf, jffs2dir); } /* need to erase the next block before writing data to it */ if(!no_erase) { while (w + buflen > e - skip_bad_blocks) { if (!quiet) fprintf(stderr, "\b\b\b[e]"); if (mtd_block_is_bad(fd, e)) { if (!quiet) fprintf(stderr, "\nSkipping bad block at 0x%08zx ", e); skip_bad_blocks += erasesize; e += erasesize; // Move the file pointer along over the bad block. lseek(fd, erasesize, SEEK_CUR); continue; } if (mtd_erase_block(fd, e) < 0) { if (next) { if (w < e) { write(fd, buf + offset, e - w); offset = e - w; } w = 0; e = 0; close(fd); mtd = next; fprintf(stderr, "\b\b\b \n"); goto resume; } else { fprintf(stderr, "Failed to erase block\n"); exit(1); } } /* erase the chunk */ e += erasesize; } } if (!quiet) fprintf(stderr, "\b\b\b[w]"); if ((result = write(fd, buf + offset, buflen)) < buflen) { if (result < 0) { fprintf(stderr, "Error writing image.\n"); exit(1); } else { fprintf(stderr, "Insufficient space.\n"); exit(1); } } w += buflen; buflen = 0; offset = 0; } if (jffs2_replaced) { switch (imageformat) { case MTD_IMAGE_FORMAT_TRX: if (trx_fixup) trx_fixup(fd, mtd); break; case MTD_IMAGE_FORMAT_SEAMA: if (mtd_fixseama) mtd_fixseama(mtd, 0, 0); break; case MTD_IMAGE_FORMAT_WRG: if (mtd_fixwrg) mtd_fixwrg(mtd, 0, 0); break; case MTD_IMAGE_FORMAT_WRGG03: if (mtd_fixwrgg) mtd_fixwrgg(mtd, 0, 0); break; default: break; } } if (!quiet) fprintf(stderr, "\b\b\b\b "); if (quiet < 2) fprintf(stderr, "\n"); #ifdef FIS_SUPPORT if (fis_layout) { if (fis_remap(old_parts, n_old, new_parts, n_new) < 0) fprintf(stderr, "Failed to update the FIS partition table\n"); } #endif close(fd); return 0; }