1 files changed, 13 insertions, 3 deletions

diff --git a/src/initrd/btrfs-create.sh b/src/initrd/btrfs-create.sh
index cdf73ef..9604340 100644
--- a/src/initrd/btrfs-create.sh
+++ b/src/initrd/btrfs-create.sh
@@ -58,7 +58,8 @@ cdrom_has_rootfs()
 
 losetup_layers()
 {
-    if cdrom_has_rootfs; then
+    if cdrom_has_rootfs
+    then
         # TODO: This is some kind of shortcut or short circuit to find these
         # files, that ought to be found through the grok-block system (i.e.,
         # event-driven rather than polling).
@@ -68,14 +69,23 @@ losetup_layers()
             dd if=/dev/zero of="$fs_rw" bs=1M count=10
             losetup_snapshot "$fs" "$fs_rw" || return
         done
+    elif [ -e /dev/disk/by-partlabel/samizdat-rootfs ]
+    then
+        local dev
+        for dev in samizdat-rootfs
+        do
+            umount /dev/disk/by-partlabel/$dev
+            dd if=/dev/zero of=/$dev.rw bs=1M count=10
+            dm_snapshot /dev/disk/by-partlabel/$dev /$dev.rw
+        done
     else
         bootwait samizdat-nbd-dev
         local dev
-        for dev in nbd0 nbd1; do
+        for dev in nbd0 nbd1
+        do
             dd if=/dev/zero of=/$dev.rw bs=1M count=10
             dm_snapshot /dev/$dev /$dev.rw
         done
-        return
     fi
 }