1 files changed, 16 insertions, 1 deletions

diff --git a/src/parted-usb.sh b/src/parted-usb.sh
index 730dc26..50fcb78 100755
--- a/src/parted-usb.sh
+++ b/src/parted-usb.sh
@@ -34,7 +34,7 @@ initialize_target()
     if [ "$ROOTFS" ]
     then
         get_rootfs_size
-        fallocate -l $(($(KiB 17) + $(MiB $end_keys) + rootfs_size)) "$target"
+        fallocate -l $(($(KiB 17) + $(MiB $end_keys) + rootfs_size + $(KiB 640))) "$target"
     else
         fallocate -l $(($(KiB 17) + $(MiB $end_keys))) "$target"
     fi
@@ -74,6 +74,20 @@ make_target_bootable()
     sync
 }
 
+copy_rootfs()
+{
+    if [ -z "$ROOTFS" ]
+    then
+        return
+    fi
+    [ -e "$ROOTFS" -a -b "$part"3 ]
+    get_rootfs_size
+    part_size=$(blockdev --getsize64 "$part"3)
+    [ "$rootfs_size" -le "$part_size" ] || die "rootfs_size=$rootfs_size > part_size=$part_size"
+
+    dd if="$ROOTFS" of="$part"3
+}
+
 acquire_target_block_devices()
 {
     if [ -b "$target" ]
@@ -143,6 +157,7 @@ globalize()
     partition_target
     acquire_target_block_devices
     make_target_bootable
+    copy_rootfs
 }
 
 sanity_checks()