From 52384c56e48e0393f24612788b5edcf2e7381c0e Mon Sep 17 00:00:00 2001
From: Fabio Erculiani <lxnay@sabayon.org>
Date: Sun, 15 Feb 2015 13:03:31 +0000
Subject: [PATCH] 00-livecd.sh: add support for overlay fs.

---
 defaults/initrd.d/00-livecd.sh | 53 +++++++++++++++++++++++++++++++---
 defaults/linuxrc               |  3 ++
 2 files changed, 52 insertions(+), 4 deletions(-)

diff --git a/defaults/initrd.d/00-livecd.sh b/defaults/initrd.d/00-livecd.sh
index 4415c7d..181891d 100755
--- a/defaults/initrd.d/00-livecd.sh
+++ b/defaults/initrd.d/00-livecd.sh
@@ -11,6 +11,11 @@ _is_aufs() {
     return 1
 }
 
+_is_overlayfs() {
+    [ "${USE_OVERLAYFS}" = "1" ] && return 0
+    return 1
+}
+
 _find_loop() {
     local l=
     for loop in ${LOOPS}; do
@@ -30,6 +35,13 @@ _find_looptype() {
     echo "${lt}"
 }
 
+_setup_cdrom_access() {
+    # have handy /mnt/cdrom (CDROOT_PATH) as well
+    local new_cdroot="${NEW_ROOT}${CDROOT_PATH}"
+    [ ! -d "${new_cdroot}" ] && mkdir -p "${new_cdroot}"
+    mount --bind "${CDROOT_PATH}" "${new_cdroot}"
+}
+
 _setup_squashfs_aufs() {
     # Setup aufs directories and vars
     local overlay=/mnt/overlay
@@ -52,10 +64,38 @@ _setup_squashfs_aufs() {
         mount --move "${i}" "${NEW_ROOT}${i}"
     done
 
-    # have handy /mnt/cdrom (CDROOT_PATH) as well
-    local new_cdroot="${NEW_ROOT}${CDROOT_PATH}"
-    [ ! -d "${new_cdroot}" ] && mkdir -p "${new_cdroot}"
-    mount --bind "${CDROOT_PATH}" "${new_cdroot}"
+    _setup_cdrom_access
+}
+
+_setup_squashfs_overlayfs() {
+    # Setup overlayfs directories and vars
+    local overlay=/mnt/overlay
+    local upperdir="${overlay}/.upper"
+    local workdir="${overlay}/.work"
+    local static=/mnt/livecd
+
+    for i in "${overlay}" "${static}"; do
+        [ ! -d "${i}" ] && mkdir -p "${i}"
+    done
+    good_msg "Loading overlayfs"
+    modprobe overlay > /dev/null 2>&1
+
+    mount -t squashfs -o loop,ro "${LOOP_PATH}" "${static}"
+    mount -t tmpfs none "${overlay}"
+    mkdir "${upperdir}" "${workdir}"
+
+    mount -t overlay overlay \
+        -o lowerdir="${static}",upperdir="${upperdir}",workdir="${workdir}" \
+        "${NEW_ROOT}"
+
+    [ ! -d "${NEW_ROOT}${overlay}" ] && mkdir -p "${NEW_ROOT}${overlay}"
+    [ ! -d "${NEW_ROOT}${static}" ] && mkdir -p "${NEW_ROOT}${static}"
+    echo "overlay / overlay defaults 0 0" > "${NEW_ROOT}"/etc/fstab
+    for i in "${overlay}" "${static}"; do
+        mount --bind "${i}" "${NEW_ROOT}${i}"
+    done
+
+    _setup_cdrom_access
 }
 
 _bootstrap_cd() {
@@ -190,6 +230,11 @@ _livecd_mount_squashfs() {
         _setup_squashfs_aufs
         test_success "Mount filesystem"
         return  # preserve old behaviour
+    elif _is_overlayfs; then  # same for overlay fs.
+        good_msg "Mounting squashfs & overlay fs filesystems"
+        _setup_squashfs_overlayfs
+        test_success "Mount filesystem"
+        return  # preserve old behaviour
     fi
 
     good_msg "Mounting squashfs filesystem"
diff --git a/defaults/linuxrc b/defaults/linuxrc
index 1b24dd7..8cfb213 100755
--- a/defaults/linuxrc
+++ b/defaults/linuxrc
@@ -256,6 +256,9 @@ for x in ${CMDLINE}; do
         aufs)
             USE_AUFS=1
         ;;
+        overlayfs)
+            USE_OVERLAYFS=1
+        ;;
     esac
 done