fstools: allow to compress the filesystem
[project/fstools.git] / libfstools / overlay.c
index 7f696061fa6792405af5366e33d120ac73422c5d..068afdc44392c433294a2c0fbe6a281f54d87e6d 100644 (file)
@@ -67,30 +67,69 @@ handle_rmdir(const char *dir)
 void
 foreachdir(const char *dir, int (*cb)(const char*))
 {
+       static char *globdir = NULL;
+       static size_t globdirlen = 0;
        struct stat s = { 0 };
-       char globdir[256];
+       size_t dirlen = strlen(dir);
        glob_t gl;
        int j;
 
-       if (dir[strlen(dir) - 1] == '/')
-               snprintf(globdir, 256, "%s*", dir);
-       else
-               snprintf(globdir, 256, "%s/*", dir); /**/
+       if (dirlen + sizeof("/*") > globdirlen) {
+               /* Alloc extra 256 B to avoid too many reallocs */
+               size_t len = dirlen + sizeof("/*") + 256;
+               char *tmp;
 
+               tmp = realloc(globdir, len);
+               if (!tmp)
+                       return;
+               globdir = tmp;
+               globdirlen = len;
+       }
+
+       sprintf(globdir, "%s/*", dir);
+
+       /* Include GLOB_MARK as callbacks expect a trailing slash */
        if (!glob(globdir, GLOB_NOESCAPE | GLOB_MARK | GLOB_ONLYDIR, NULL, &gl))
                for (j = 0; j < gl.gl_pathc; j++) {
                        char *dir = gl.gl_pathv[j];
                        int len = strlen(gl.gl_pathv[j]);
+                       int err;
 
-                       if (len > 1 && dir[len - 1] == '/')
+                       /* Quick way of skipping files */
+                       if (dir[len - 1] != '/')
+                               continue;
+
+                       /* lstat needs path without a trailing slash */
+                       if (len > 1)
                                dir[len - 1] = '\0';
+                       err = lstat(gl.gl_pathv[j], &s);
+                       if (len > 1)
+                               dir[len - 1] = '/';
 
-                       if (!lstat(gl.gl_pathv[j], &s) && !S_ISLNK(s.st_mode))
+                       if (!err && !S_ISLNK(s.st_mode))
                                foreachdir(gl.gl_pathv[j], cb);
        }
        cb(dir);
 }
 
+static void foreach_mount(int (*cb)(const char *, const char *))
+{
+       FILE *fp = fopen("/proc/mounts", "r");
+       static char line[256];
+
+       if (!fp)
+               return;
+
+       while (fgets(line, sizeof(line), fp)) {
+               char device[32], mount_point[32];
+
+               if (sscanf(line, "%31s %31s %*s %*s %*u %*u", device, mount_point) == 2)
+                       cb(device, mount_point);
+       }
+
+       fclose(fp);
+}
+
 void
 overlay_delete(const char *dir, bool _keep_sysupgrade)
 {
@@ -102,16 +141,29 @@ static int
 overlay_mount(struct volume *v, char *fs)
 {
        if (mkdir("/tmp/overlay", 0755)) {
-               ULOG_ERR("failed to mkdir /tmp/overlay: %s\n", strerror(errno));
+               ULOG_ERR("failed to mkdir /tmp/overlay: %m\n");
                return -1;
        }
 
        if (mount(v->blk, "/tmp/overlay", fs, MS_NOATIME, NULL)) {
-               ULOG_ERR("failed to mount -t %s %s /tmp/overlay: %s\n", fs, v->blk, strerror(errno));
+               ULOG_ERR("failed to mount -t %s %s /tmp/overlay: %m\n", fs, v->blk);
                return -1;
        }
 
-       return volume_init(v);
+       return 0;
+}
+
+/**
+ * move_mount - move mount point to the new root
+ */
+static int move_mount(const char *device, const char *mount_point)
+{
+       static const char *prefix = "/tmp/root/";
+
+       if (strncmp(mount_point, prefix, strlen(prefix)))
+               return 0;
+
+       return mount_move(prefix, "/", mount_point + strlen(prefix));
 }
 
 static int
@@ -129,31 +181,44 @@ switch2jffs(struct volume *v)
        ret = mount(v->blk, "/rom/overlay", "jffs2", MS_NOATIME, NULL);
        unlink("/tmp/.switch_jffs2");
        if (ret) {
-               ULOG_ERR("failed - mount -t jffs2 %s /rom/overlay: %s\n", v->blk, strerror(errno));
+               ULOG_ERR("failed - mount -t jffs2 %s /rom/overlay: %m\n", v->blk);
                return -1;
        }
 
        if (mount("none", "/", NULL, MS_NOATIME | MS_REMOUNT, 0)) {
-               ULOG_ERR("failed - mount -o remount,ro none: %s\n", strerror(errno));
+               ULOG_ERR("failed - mount -o remount,ro none: %m\n");
                return -1;
        }
 
        if (system("cp -a /tmp/root/* /rom/overlay")) {
-               ULOG_ERR("failed - cp -a /tmp/root/* /rom/overlay: %s\n", strerror(errno));
+               ULOG_ERR("failed - cp -a /tmp/root/* /rom/overlay: %m\n");
                return -1;
        }
 
        if (pivot("/rom", "/mnt")) {
-               ULOG_ERR("failed - pivot /rom /mnt: %s\n", strerror(errno));
+               ULOG_ERR("failed - pivot /rom /mnt: %m\n");
                return -1;
        }
 
        if (mount_move("/mnt", "/tmp/root", "")) {
-               ULOG_ERR("failed - mount -o move /mnt /tmp/root %s\n", strerror(errno));
+               ULOG_ERR("failed - mount -o move /mnt /tmp/root %m\n");
                return -1;
        }
 
-       return fopivot("/overlay", "/rom");
+       ret = fopivot("/overlay", "/rom");
+
+       /*
+        * Besides copying overlay data from "tmpfs" to "jffs2" we should also
+        * move mount points that user could create during JFFS2 formatting.
+        * This has to happen after fopivot call because:
+        * 1) It's trivial to find mount points to move then (/tmp/root/...).
+        * 2) We can't do that earlier using /rom/overlay/upper/ as overlay(fs)
+        *    doesn't support mounts. Mounting to upper dir don't make overlay
+        *    /propagate/ files to the target dir.
+        */
+       foreach_mount(move_mount);
+
+       return ret;
 }
 
 int
@@ -192,64 +257,76 @@ handle_whiteout(const char *dir)
        return 0;
 }
 
+static char *overlay_fs_name(int type)
+{
+       switch (type) {
+               case FS_EXT4:
+                       return "ext4";
+               case FS_F2FS:
+                       return "f2fs";
+               case FS_UBIFS:
+                       return "ubifs";
+               case FS_JFFS2:
+               default:
+                       return "jffs2";
+       }
+}
+
 int
 jffs2_switch(struct volume *v)
 {
-       char *mp;
-       int ret = -1;
+       char *mp, *fs_name;
+       int type;
 
        if (find_overlay_mount("overlayfs:/tmp/root"))
                return -1;
 
        if (find_filesystem("overlay")) {
                ULOG_ERR("overlayfs not supported by kernel\n");
-               return ret;
+               return -1;
        }
 
+       volume_init(v);
        mp = find_mount_point(v->blk, 0);
        if (mp) {
                ULOG_ERR("rootfs_data:%s is already mounted as %s\n", v->blk, mp);
                return -1;
        }
 
-       switch (volume_identify(v)) {
+       type = volume_identify(v);
+       fs_name = overlay_fs_name(type);
+
+       switch (type) {
        case FS_NONE:
                ULOG_ERR("no jffs2 marker found\n");
                /* fall through */
 
        case FS_DEADCODE:
-               ret = switch2jffs(v);
-               if (!ret) {
-                       ULOG_INFO("performing overlay whiteout\n");
-                       umount2("/tmp/root", MNT_DETACH);
-                       foreachdir("/overlay/", handle_whiteout);
-               }
-               break;
+               if (switch2jffs(v))
+                       return -1;
 
-       case FS_JFFS2:
-               ret = overlay_mount(v, "jffs2");
-               if (ret)
-                       break;
-               if (mount_move("/tmp", "", "/overlay") || fopivot("/overlay", "/rom")) {
-                       ULOG_ERR("switching to jffs2 failed\n");
-                       ret = -1;
-               }
+               ULOG_INFO("performing overlay whiteout\n");
+               umount2("/tmp/root", MNT_DETACH);
+               foreachdir("/overlay/", handle_whiteout);
+
+               /* try hard to be in sync */
+               ULOG_INFO("syncronizing overlay\n");
+               if (system("cp -a /tmp/root/upper/* / 2>/dev/null"))
+                       ULOG_ERR("failed to sync jffs2 overlay\n");
                break;
 
+       case FS_EXT4:
+       case FS_F2FS:
        case FS_UBIFS:
-               ret = overlay_mount(v, "ubifs");
-               if (ret)
-                       break;
+               if (overlay_mount(v, fs_name))
+                       return -1;
                if (mount_move("/tmp", "", "/overlay") || fopivot("/overlay", "/rom")) {
-                       ULOG_ERR("switching to ubifs failed\n");
-                       ret = -1;
+                       ULOG_ERR("switching to %s failed\n", fs_name);
+                       return -1;
                }
                break;
        }
 
-       if (ret)
-               return ret;
-
        sync();
        fs_state_set("/overlay", FS_STATE_READY);
        return 0;
@@ -257,30 +334,26 @@ jffs2_switch(struct volume *v)
 
 static int overlay_mount_fs(struct volume *v)
 {
-       char *fstype;
+       char *fstype = overlay_fs_name(volume_identify(v));
 
        if (mkdir("/tmp/overlay", 0755)) {
-               ULOG_ERR("failed to mkdir /tmp/overlay: %s\n", strerror(errno));
+               ULOG_ERR("failed to mkdir /tmp/overlay: %m\n");
                return -1;
        }
 
-       fstype = "jffs2";
-
-       switch (volume_identify(v)) {
-       case FS_UBIFS:
-               fstype = "ubifs";
-               break;
-       }
-
-       volume_init(v);
-
-       if (mount(v->blk, "/tmp/overlay", fstype, MS_NOATIME, NULL)) {
-               ULOG_ERR("failed to mount -t %s %s /tmp/overlay: %s\n",
-                        fstype, v->blk, strerror(errno));
+       if (mount(v->blk, "/tmp/overlay", fstype, MS_NOATIME,
+#ifdef OVL_MOUNT_COMPRESS_ZLIB
+               "compr=zlib"
+#else
+               NULL
+#endif
+               )) {
+               ULOG_ERR("failed to mount -t %s %s /tmp/overlay: %m\n",
+                        fstype, v->blk);
                return -1;
        }
 
-       return -1;
+       return 0;
 }
 
 enum fs_state fs_state_get(const char *dir)
@@ -311,6 +384,9 @@ int fs_state_set(const char *dir, enum fs_state state)
        char valstr[16];
        char *path;
 
+       if (fs_state_get(dir) == state)
+               return 0;
+
        path = alloca(strlen(dir) + 1 + sizeof("/.fs_state"));
        sprintf(path, "%s/.fs_state", dir);
        unlink(path);
@@ -322,7 +398,7 @@ int fs_state_set(const char *dir, enum fs_state state)
 
 int mount_overlay(struct volume *v)
 {
-       char *mp;
+       char *mp, *fs_name;
 
        if (!v)
                return -1;
@@ -356,9 +432,10 @@ int mount_overlay(struct volume *v)
                break;
        }
 
-       ULOG_INFO("switching to jffs2 overlay\n");
+       fs_name = overlay_fs_name(volume_identify(v));
+       ULOG_INFO("switching to %s overlay\n", fs_name);
        if (mount_move("/tmp", "", "/overlay") || fopivot("/overlay", "/rom")) {
-               ULOG_ERR("switching to jffs2 failed - fallback to ramoverlay\n");
+               ULOG_ERR("switching to %s failed - fallback to ramoverlay\n", fs_name);
                return ramoverlay();
        }