diskimage-builder/diskimage_builder/elements/lvm/finalise.d/60-bootloader

66 lines
2.3 KiB
Bash
Executable File

#!/bin/bash
# Configure grub. Note that the various conditionals here are to handle
# different distributions gracefully.
if [ ${DIB_DEBUG_TRACE:-1} -gt 0 ]; then
set -x
fi
set -eu
set -o pipefail
BOOT_DEV=$IMAGE_BLOCK_DEVICE
# All available devices, handy for some bootloaders...
declare -A DEVICES
eval DEVICES=( $IMAGE_BLOCK_DEVICES )
# This might be better factored out into a per-distro 'install-bootblock'
# helper.
if [ -d /boot/grub2 ]; then
GRUB_CFG=/boot/grub2/grub.cfg
elif [ -d /boot/grub ]; then
GRUB_CFG=/boot/grub/grub.cfg
fi
if type grub2-mkconfig >/dev/null; then
GRUB_MKCONFIG="grub2-mkconfig -o $GRUB_CFG"
else
GRUB_MKCONFIG="grub-mkconfig -o $GRUB_CFG"
fi
# Retrieve root filesystem mount point to check if it is LVM based
ROOTFS=$(awk '$2=="/"{print}' /proc/mounts)
# Standard filesystems are mounted as /dev/mapper/loop* within DIB
# so we need to exclude this case
if echo "$ROOTFS" | grep -qv '/dev/mapper/loop'; then
# LVM based filesystems are mounted as /dev/mapper/<VG_name>-<LV-NAME> within DIB
if echo "$ROOTFS" | grep -q '/dev/mapper/'; then
# Check if initramfs-tools are installed
if [ ! -d /etc/initramfs-tools ]; then
echo "You must have initramfs-tools installed for LVM based root filesystem to work properly"
exit 1
fi
# Check if system supports LVM
if ! type pvs >/dev/null 2>&1; then
echo "You must have lvm2 package installed for LVM based root filesystem to work properly"
exit 1
fi
# Check for appropriate filesystem support (for fsck)
FSTYPE=$(echo $ROOTFS | awk '{print $3}')
if ! type fsck.$FSTYPE >/dev/null 2>&1; then
echo "You must have utility package for $FSTYPE filesystem installed for LVM based root filesystem to work properly (fsck.$FSTYPE is missing)"
exit 1
fi
# Change the current GRUB_DEVICE from LABEL=(cloud)img-rootfs to /dev/mapper/...
sed -i -e "s?^GRUB_DEVICE=.*?GRUB_DEVICE=$(echo $ROOTFS | awk '{print $1}')?" /etc/default/grub
# This is required for LVM2 driver to be included in the
# ramdisk of the image.
echo "lvm2" >> /etc/initramfs-tools/modules
# We need to regenerated the init ramdisk at this point
update-initramfs -u -v
fi
fi
$GRUB_MKCONFIG