1 files changed, 89 insertions, 0 deletions
diff --git a/disk-expand/usr/share/dracut/modules.d/50growroot/growroot.sh b/disk-expand/usr/share/dracut/modules.d/50growroot/growroot.sh
new file mode 100755
index 0000000..e09c809
--- /dev/null
+++ b/disk-expand/usr/share/dracut/modules.d/50growroot/growroot.sh
@@ -0,0 +1,89 @@
+#!/bin/sh
+
+# Environment variables that this script relies upon:
+# - root
+
+_info() {
+        echo "growroot: $*"
+}
+
+_warning() {
+        echo "growroot Warning: $*" >&2
+}
+
+# This will drop us into an emergency shell
+_fatal() {
+        echo "growroot Fatal: $*" >&2
+        exit 1
+}
+
+_growroot() {
+        # Remove 'block:' prefix and find the root device
+        rootdev=$(readlink -f "${root#block:}")
+        if [ -z "${rootdev}" ] ; then
+                _warning "unable to find root device"
+                return
+        fi
+
+        # If the basename of the root device (ie 'xvda1', 'sda1', 'vda') exists
+        # in /sys/block/ then it is a block device, not a partition
+        if [ -e "/sys/block/${rootdev##*/}" ] ; then
+                _info "${rootdev} is not a partition"
+                return
+        fi
+
+        # Check if the root device is a partition (name ends with a digit)
+        if [ "${rootdev%[0-9]}" = "${rootdev}" ] ; then
+                _warning "${rootdev} is not a partition"
+                return
+        fi
+
+        # Remove all numbers from the end of rootdev to get the rootdisk and
+        # partition number
+        rootdisk=${rootdev}
+        while [ "${rootdisk%[0-9]}" != "${rootdisk}" ] ; do
+                rootdisk=${rootdisk%[0-9]}
+        done
+        partnum=${rootdev#${rootdisk}}
+
+        # Check if we need to strip a trailing 'p' from the rootdisk name (for
+        # device names like /dev/mmcblk0p)
+        tmp=${rootdisk%[0-9]p}
+        if [ "${#tmp}" != "${#rootdisk}" ] ; then
+                rootdisk=${rootdisk%p}
+        fi
+
+        # Do a growpart dry run and exit if it fails or doesn't have anything
+        # to do
+        if ! out=$(growpart --dry-run "${rootdisk}" "${partnum}") ; then
+                _info "${out}"
+                return
+        fi
+
+        # Wait for any of the initial udev events to finish otherwise growpart
+        # might fail
+        udevadm settle --timeout=30
+
+        # Resize the root partition
+        if out=$(growpart --update off "${rootdisk}" "${partnum}" 2>&1) ; then
+                _info "${out}"
+        else
+                _warning "${out}"
+                _warning "growpart failed"
+        fi
+
+        # Wait for the partition re-read events to complete so that the root
+        # partition is available for mounting
+        udevadm settle --timeout=30
+
+        # Add the root partition if it didn't come back on its own
+        if ! [ -e "${rootdev}" ] ; then
+                partx --add --nr "${partnum}" "${rootdisk}" || \
+                        _warning "failed to add root device ${rootdev}"
+                udevadm settle --timeout=30
+        fi
+}
+
+_growroot
+
+# vi: ts=4 noexpandtab