[package] busybox: fix hexdump segmentation fault with an empty leading format unit
[openwrt.git] / package / mtd / src / jffs2.c
index f614a33..aaf9be5 100644 (file)
@@ -1,3 +1,21 @@
+/*
+ * jffs2 on-disk structure generator for mtd
+ *
+ * Copyright (C) 2008 Felix Fietkau <nbd@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License v2
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * Based on:
+ *   JFFS2 -- Journalling Flash File System, Version 2.
+ *   Copyright © 2001-2007 Red Hat, Inc.
+ *   Created by David Woodhouse <dwmw2@infradead.org>
+ */
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <stdint.h>
 #include <string.h>
 #include <dirent.h>
 #include <unistd.h>
+#include <endian.h>
 #include "jffs2.h"
 #include "crc32.h"
 #include "mtd.h"
 
 #define PAD(x) (((x)+3)&~3)
 
-#define CLEANMARKER "\x85\x19\x03\x20\x0c\x00\x00\x00\xb1\xb0\x1e\xe4"
-#define JFFS2_EOF "\xde\xad\xc0\xde"
+#if BYTE_ORDER == BIG_ENDIAN
+# define CLEANMARKER "\x19\x85\x20\x03\x00\x00\x00\x0c\xf0\x60\xdc\x98"
+#else
+# define CLEANMARKER "\x85\x19\x03\x20\x0c\x00\x00\x00\xb1\xb0\x1e\xe4"
+#endif
 
 static int last_ino = 0;
 static int last_version = 0;
 static char *buf = NULL;
 static int ofs = 0;
-static int outfd = 0;
+static int outfd = -1;
 static int mtdofs = 0;
+static int target_ino = 0;
 
 static void prep_eraseblock(void);
 
@@ -65,7 +88,7 @@ static void prep_eraseblock(void)
        add_data(CLEANMARKER, sizeof(CLEANMARKER) - 1);
 }
 
-static int add_dirent(char *name, char type, int parent)
+static int add_dirent(const char *name, const char type, int parent)
 {
        struct jffs2_raw_dirent *de;
 
@@ -97,7 +120,7 @@ static int add_dirent(char *name, char type, int parent)
        return de->ino;
 }
 
-static int add_dir(char *name, int parent)
+static int add_dir(const char *name, int parent)
 {
        struct jffs2_raw_inode ri;
        int inode;
@@ -128,13 +151,13 @@ static int add_dir(char *name, int parent)
        return inode;
 }
 
-static void add_file(char *name, int parent)
+static void add_file(const char *name, int parent)
 {
        int inode, f_offset = 0, fd;
        struct jffs2_raw_inode ri;
        struct stat st;
-       char wbuf[4096], *fname;
-       FILE *f;
+       char wbuf[4096];
+       const char *fname;
 
        if (stat(name, &st)) {
                fprintf(stderr, "File %s does not exist\n", name);
@@ -163,7 +186,7 @@ static void add_file(char *name, int parent)
        ri.usercompr = 0;
 
        fd = open(name, 0);
-       if (fd <= 0) {
+       if (fd < 0) {
                fprintf(stderr, "File %s does not exist\n", name);
                return;
        }
@@ -204,14 +227,62 @@ static void add_file(char *name, int parent)
        close(fd);
 }
 
-int mtd_write_jffs2(char *mtd, char *filename, char *dir)
+int mtd_replace_jffs2(const char *mtd, int fd, int ofs, const char *filename)
+{
+       outfd = fd;
+       mtdofs = ofs;
+
+       buf = malloc(erasesize);
+       target_ino = 1;
+       if (!last_ino)
+               last_ino = 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);
+       free(buf);
+
+       if (trx_fixup) {
+         trx_fixup(outfd, mtd);
+       }
+       return (mtdofs - ofs);
+}
+
+void mtd_parse_jffs2data(const char *buf, const char *dir)
+{
+       struct jffs2_unknown_node *node = (struct jffs2_unknown_node *) buf;
+       unsigned int ofs = 0;
+
+       while (ofs < erasesize) {
+               node = (struct jffs2_unknown_node *) (buf + ofs);
+               if (node->magic != 0x1985)
+                       break;
+
+               ofs += PAD(node->totlen);
+               if (node->nodetype == JFFS2_NODETYPE_DIRENT) {
+                       struct jffs2_raw_dirent *de = (struct jffs2_raw_dirent *) node;
+
+                       /* is this the right directory name and is it a subdirectory of / */
+                       if (*dir && (de->pino == 1) && !strncmp((char *) de->name, dir, de->nsize))
+                               target_ino = de->ino;
+
+                       /* store the last inode and version numbers for adding extra files */
+                       if (last_ino < de->ino)
+                               last_ino = de->ino;
+                       if (last_version < de->version)
+                               last_version = de->version;
+               }
+       }
+}
+
+int mtd_write_jffs2(const char *mtd, const char *filename, const char *dir)
 {
-       int target_ino = 0;
        int err = -1, fdeof = 0;
-       off_t offset;
 
        outfd = mtd_check_open(mtd);
-       if (!outfd)
+       if (outfd < 0)
                return -1;
 
        if (quiet < 2)
@@ -230,7 +301,6 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
         * locate the directory that the file is going to be placed in */
        for(;;) {
                struct jffs2_unknown_node *node = (struct jffs2_unknown_node *) buf;
-               unsigned int ofs = 0;
 
                if (read(outfd, buf, erasesize) != erasesize) {
                        fdeof = 1;
@@ -248,27 +318,7 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
                if (node->magic != 0x1985)
                        break;
 
-               while (ofs < erasesize) {
-                       node = (struct jffs2_unknown_node *) (buf + ofs);
-                       if (node->magic == 0x1985) {
-                               ofs += PAD(node->totlen);
-                               if (node->nodetype == JFFS2_NODETYPE_DIRENT) {
-                                       struct jffs2_raw_dirent *de = (struct jffs2_raw_dirent *) node;
-                                       
-                                       /* is this the right directory name and is it a subdirectory of / */
-                                       if (*dir && (de->pino == 1) && !strncmp(de->name, dir, de->nsize))
-                                               target_ino = de->ino;
-
-                                       /* store the last inode and version numbers for adding extra files */
-                                       if (last_ino < de->ino)
-                                               last_ino = de->ino;
-                                       if (last_version < de->version)
-                                               last_version = de->version;
-                               }
-                       } else {
-                               ofs = ~0;
-                       }
-               }
+               mtd_parse_jffs2data(buf, dir);
        }
 
        if (fdeof) {
@@ -297,6 +347,10 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
 
        err = 0;
 
+       if (trx_fixup) {
+         trx_fixup(outfd, mtd);
+       }
+
 done:
        close(outfd);
        if (buf)
This page took 0.023709 seconds and 4 git commands to generate.