generic: add rt2x00_platform.h

[openwrt.git] / package / mtd / src / jffs2.c

diff --git a/package/mtd/src/jffs2.c b/package/mtd/src/jffs2.c
index 7b68ae5..aaf9be5 100644 (file)
--- a/package/mtd/src/jffs2.c
+++ b/package/mtd/src/jffs2.c
@@ -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 <sys/types.h>
 #include <sys/stat.h>
 #include <stdint.h>


@@ -7,21 +25,26 @@
 #include <string.h>
 #include <dirent.h>
 #include <unistd.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)
 
 #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 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 mtdofs = 0;

+static int target_ino = 0;

 
 static void prep_eraseblock(void);
 
 
 static void prep_eraseblock(void);
 


@@ -65,7 +88,7 @@ static void prep_eraseblock(void)
        add_data(CLEANMARKER, sizeof(CLEANMARKER) - 1);
 }
 
        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;
 
 {
        struct jffs2_raw_dirent *de;
 


@@ -97,7 +120,7 @@ static int add_dirent(char *name, char type, int parent)
        return de->ino;
 }
 
        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;
 {
        struct jffs2_raw_inode ri;
        int inode;


@@ -128,13 +151,13 @@ static int add_dir(char *name, int parent)
        return inode;
 }
 
        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;
 {
        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);
 
        if (stat(name, &st)) {
                fprintf(stderr, "File %s does not exist\n", name);


@@ -147,7 +170,7 @@ static void add_file(char *name, int parent)
        else
                fname = name;
 
        else
                fname = name;
 

-       inode = add_dirent(name, IFTODT(S_IFREG), parent);
+       inode = add_dirent(fname, IFTODT(S_IFREG), parent);

        memset(&ri, 0, sizeof(ri));
        ri.magic = JFFS2_MAGIC_BITMASK;
        ri.nodetype = JFFS2_NODETYPE_INODE;
        memset(&ri, 0, sizeof(ri));
        ri.magic = JFFS2_MAGIC_BITMASK;
        ri.nodetype = JFFS2_NODETYPE_INODE;


@@ -163,7 +186,7 @@ static void add_file(char *name, int parent)
        ri.usercompr = 0;
 
        fd = open(name, 0);
        ri.usercompr = 0;
 
        fd = open(name, 0);

-       if (fd <= 0) {
+       if (fd < 0) {

                fprintf(stderr, "File %s does not exist\n", name);
                return;
        }
                fprintf(stderr, "File %s does not exist\n", name);
                return;
        }


@@ -204,14 +227,62 @@ static void add_file(char *name, int parent)
        close(fd);
 }
 
        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;
        int err = -1, fdeof = 0;

-       off_t offset;

 
        outfd = mtd_check_open(mtd);
 
        outfd = mtd_check_open(mtd);

-       if (!outfd)
+       if (outfd < 0)

                return -1;
 
        if (quiet < 2)
                return -1;
 
        if (quiet < 2)


@@ -223,11 +294,13 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
                goto done;
        }
 
                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;
        /* 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;

-               unsigned int ofs = 0;

 
                if (read(outfd, buf, erasesize) != erasesize) {
                        fdeof = 1;
 
                if (read(outfd, buf, erasesize) != erasesize) {
                        fdeof = 1;


@@ -245,27 +318,7 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
                if (node->magic != 0x1985)
                        break;
 
                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 ((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) {
        }
 
        if (fdeof) {


@@ -294,6 +347,10 @@ int mtd_write_jffs2(char *mtd, char *filename, char *dir)
 
        err = 0;
 
 
        err = 0;
 

+       if (trx_fixup) {
+         trx_fixup(outfd, mtd);
+       }
+

 done:
        close(outfd);
        if (buf)
 done:
        close(outfd);
        if (buf)