release/src/router/rc/jffs2.c

   1 /*
   2
   3         Tomato Firmware
   4         Copyright (C) 2006-2009 Jonathan Zarate
   5
   6 */
   7
   8 #include "rc.h"
   9
  10 #include <sys/mount.h>
  11 #include <sys/stat.h>
  12 #include <errno.h>
  13
  14 //      #define TEST_INTEGRITY
  15
  16 #ifdef LINUX26
  17 #define JFFS_NAME       "jffs2"
  18 #else
  19 #define JFFS_NAME       "jffs"
  20 #endif
  21
  22 static void error(const char *message)
  23 {
  24         char s[512];
  25
  26         snprintf(s, sizeof(s), "Error %s JFFS. Check the logs to see if they contain more details about this error.", message);
  27         notice_set("jffs", s);
  28 }
  29
  30 void start_jffs2(void)
  31 {
  32         if (!nvram_match("jffs2_on", "1")) {
  33                 notice_set("jffs", "");
  34                 return;
  35         }
  36
  37         int format = 0;
  38         char s[256];
  39         int size;
  40         int part;
  41         const char *p;
  42
  43         if (!wait_action_idle(10)) return;
  44
  45         if (!mtd_getinfo("jffs2", &part, &size)) return;
  46
  47         if (nvram_match("jffs2_format", "1")) {
  48                 nvram_set("jffs2_format", "0");
  49
  50                 if (!mtd_erase("jffs2")) {
  51                         error("formatting");
  52                         return;
  53                 }
  54
  55                 format = 1;
  56         }
  57
  58         sprintf(s, "%d", size);
  59         p = nvram_get("jffs2_size");
  60         if ((p == NULL) || (strcmp(p, s) != 0)) {
  61                 if (format) {
  62                         nvram_set("jffs2_size", s);
  63                          nvram_commit_x();
  64                 }
  65                 else if ((p != NULL) && (*p != 0)) {
  66                         error("verifying known size of");
  67                         return;
  68                 }
  69         }
  70
  71         if (!mtd_unlock("jffs2")) {
  72                 error("unlocking");
  73                 return;
  74         }
  75
  76         modprobe(JFFS_NAME);
  77
  78         sprintf(s, MTD_BLKDEV(%d), part);
  79         if (mount(s, "/jffs", JFFS_NAME, MS_NOATIME|MS_NODIRATIME, "") != 0) {
  80                 modprobe_r(JFFS_NAME);
  81                 error("mounting");
  82                 return;
  83         }
  84
  85 #ifdef TEST_INTEGRITY
  86         int test;
  87
  88         if (format) {
  89                 if (f_write("/jffs/.tomato_do_not_erase", &size, sizeof(size), 0, 0) != sizeof(size)) {
  90                         stop_jffs2();
  91                         error("setting integrity test for");
  92                         return;
  93                 }
  94         }
  95
  96         if ((f_read("/jffs/.tomato_do_not_erase", &test, sizeof(test)) != sizeof(test)) || (test != size)) {
  97                 stop_jffs2();
  98                 error("testing integrity of");
  99                 return;
 100         }
 101 #endif
 102
 103         notice_set("jffs", format ? "Formatted." : "Loaded.");
 104
 105         if (((p = nvram_get("jffs2_exec")) != NULL) && (*p != 0)) {
 106                 chdir("/jffs");
 107                 xstart(p);
 108                 chdir("/");
 109         }
 110         run_userfile("/jffs", ".autorun", NULL, 0);
 111 }
 112
 113 void stop_jffs2(void)
 114 {
 115         if (!wait_action_idle(10)) return;
 116
 117         notice_set("jffs", "stopped");
 118         umount("/jffs");
 119         modprobe_r(JFFS_NAME);
 120 }