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 <sys/statfs.h>
  13 #include <errno.h>
  14 #ifndef MNT_DETACH
  15 #define MNT_DETACH      0x00000002
  16 #endif
  17
  18 //      #define TEST_INTEGRITY
  19
  20 #ifdef TCONFIG_JFFSV1
  21 #define JFFS_NAME       "jffs"
  22 #else
  23 #define JFFS_NAME       "jffs2"
  24 #endif
  25
  26 static void error(const char *message)
  27 {
  28         char s[512];
  29
  30         snprintf(s, sizeof(s), "Error %s JFFS. Check the logs to see if they contain more details about this error.", message);
  31         notice_set("jffs", s);
  32 }
  33
  34 void start_jffs2(void)
  35 {
  36         if (!nvram_match("jffs2_on", "1")) {
  37                 notice_set("jffs", "");
  38                 return;
  39         }
  40
  41         int format = 0;
  42         char s[256];
  43         int size;
  44         int part;
  45         const char *p;
  46         struct statfs sf;
  47
  48         if (!wait_action_idle(10)) return;
  49
  50         if (!mtd_getinfo("jffs2", &part, &size)) return;
  51
  52         if (nvram_match("jffs2_format", "1")) {
  53                 nvram_set("jffs2_format", "0");
  54                 nvram_commit_x();
  55
  56                 if (!mtd_erase("jffs2")) {
  57                         error("formatting");
  58                         return;
  59                 }
  60
  61                 format = 1;
  62         }
  63
  64         sprintf(s, "%d", size);
  65         p = nvram_get("jffs2_size");
  66         if ((p == NULL) || (strcmp(p, s) != 0)) {
  67                 if (format) {
  68                         nvram_set("jffs2_size", s);
  69                         nvram_commit_x();
  70                 }
  71                 else if ((p != NULL) && (*p != 0)) {
  72                         error("verifying known size of");
  73                         return;
  74                 }
  75         }
  76
  77         if ((statfs("/jffs", &sf) == 0) && (sf.f_type != 0x73717368 /* squashfs */)) {
  78                 // already mounted
  79                 notice_set("jffs", format ? "Formatted" : "Loaded");
  80                 return;
  81         }
  82
  83         if (!mtd_unlock("jffs2")) {
  84                 error("unlocking");
  85                 return;
  86         }
  87
  88         modprobe(JFFS_NAME);
  89
  90         sprintf(s, MTD_BLKDEV(%d), part);
  91         if (mount(s, "/jffs", JFFS_NAME, MS_NOATIME, "") != 0) {
  92                 modprobe_r(JFFS_NAME);
  93                 error("mounting");
  94                 return;
  95         }
  96
  97 #ifdef TEST_INTEGRITY
  98         int test;
  99
 100         if (format) {
 101                 if (f_write("/jffs/.tomato_do_not_erase", &size, sizeof(size), 0, 0) != sizeof(size)) {
 102                         stop_jffs2();
 103                         error("setting integrity test for");
 104                         return;
 105                 }
 106         }
 107
 108         if ((f_read("/jffs/.tomato_do_not_erase", &test, sizeof(test)) != sizeof(test)) || (test != size)) {
 109                 stop_jffs2();
 110                 error("testing integrity of");
 111                 return;
 112         }
 113 #endif
 114
 115         notice_set("jffs", format ? "Formatted" : "Loaded");
 116
 117         if (((p = nvram_get("jffs2_exec")) != NULL) && (*p != 0)) {
 118                 chdir("/jffs");
 119                 system(p);
 120                 chdir("/");
 121         }
 122         run_userfile("/jffs", ".autorun", "/jffs", 3);
 123 }
 124
 125 void stop_jffs2(void)
 126 {
 127         struct statfs sf;
 128
 129         if (!wait_action_idle(10)) return;
 130
 131         if ((statfs("/jffs", &sf) == 0) && (sf.f_type != 0x73717368)) {
 132                 // is mounted
 133                 run_userfile("/jffs", ".autostop", "/jffs", 5);
 134                 run_nvscript("script_autostop", "/jffs", 5);
 135         }
 136
 137         notice_set("jffs", "Stopped");
 138         umount2("/jffs", MNT_DETACH);
 139         modprobe_r(JFFS_NAME);
 140 }