int mtd_write_jffs2(const char *mtd, const char *filename, const char *dir) { int err = -1, fdeof = 0; outfd = mtd_check_open(mtd); if (outfd < 0) return -1; if (quiet < 2) fprintf(stderr, "Appending %s to jffs2 partition %s\n", filename, mtd); buf = malloc(erasesize); if (!buf) { fprintf(stderr, "Out of memory!\n"); goto done; } if (!*dir) target_ino = 1; /* parse the structure of the jffs2 first * locate the directory that the file is going to be placed in */ for(;;) { struct jffs2_unknown_node *node = (struct jffs2_unknown_node *) buf; if (read(outfd, buf, erasesize) != erasesize) { fdeof = 1; break; } mtdofs += erasesize; if (node->magic == 0x8519) { fprintf(stderr, "Error: wrong endianness filesystem\n"); goto done; } /* assume no magic == end of filesystem * the filesystem will probably end with be32(0xdeadc0de) */ if (node->magic != 0x1985) break; mtd_parse_jffs2data(buf, dir); } if (fdeof) { fprintf(stderr, "Error: No room for additional data\n"); goto done; } /* jump back one eraseblock */ mtdofs -= erasesize; lseek(outfd, mtdofs, SEEK_SET); ofs = 0; if (!last_ino) last_ino = 1; if (!target_ino) target_ino = add_dir(dir, 1); add_file(filename, target_ino); pad(erasesize); /* add eof marker, pad to eraseblock size and write the data */ add_data(JFFS2_EOF, sizeof(JFFS2_EOF) - 1); pad(erasesize); err = 0; #ifdef target_brcm trx_fixup(outfd, mtd); #endif done: close(outfd); if (buf) free(buf); return err; }
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...", 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 && trx_fixup) { trx_fixup(fd, mtd); } 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; }
static int mtd_write(int imagefd, const char *mtd) { int fd, result; ssize_t r, w, e; fd = mtd_check_open(mtd); if(fd < 0) { fprintf(stderr, "Could not open mtd device: %s\n", mtd); exit(1); } if (quiet < 2) fprintf(stderr, "Writing from %s to %s ... ", imagefile, mtd); r = w = e = 0; if (!quiet) fprintf(stderr, " [ ]"); for (;;) { /* buffer may contain data already (from trx check) */ do { 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; } while (buflen < erasesize); if (buflen == 0) break; if (jffs2file) { if (memcmp(buf, JFFS2_EOF, sizeof(JFFS2_EOF)) == 0) { if (!quiet) fprintf(stderr, "\b\b\b "); if (quiet < 2) fprintf(stderr, "\nAppending jffs2 data to from %s to %s...", jffs2file, mtd); /* got an EOF marker - this is the place to add some jffs2 data */ mtd_replace_jffs2(mtd, fd, e, jffs2file); goto done; } /* 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 */ while (w + buflen > e) { if (!quiet) fprintf(stderr, "\b\b\b[e]"); mtd_erase_block(fd, e); /* erase the chunk */ e += erasesize; } if (!quiet) fprintf(stderr, "\b\b\b[w]"); if ((result = write(fd, buf, 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; } if (!quiet) fprintf(stderr, "\b\b\b\b"); done: if (quiet < 2) fprintf(stderr, "\n"); close(fd); return 0; }