172 lines
4.6 KiB
Bash
Executable File
172 lines
4.6 KiB
Bash
Executable File
#!/bin/bash -e
|
|
|
|
function usage {
|
|
echo "$(basename "$0") [OPTIONS]"
|
|
echo " -h shows usage"
|
|
echo " -g <version> generate UKI image for specified kernel version"
|
|
echo " -a generate UKI images for all available kernels"
|
|
}
|
|
|
|
function check_root {
|
|
[ $EUID -eq 0 ] && return
|
|
echo "dracut-ukify requires root privileges to work" >&2
|
|
exit 1
|
|
}
|
|
|
|
if [[ ${#} -eq 0 ]]; then
|
|
usage
|
|
fi
|
|
|
|
declare -a ukify_global_args=()
|
|
|
|
ESP_PATH=$(bootctl --print-esp-path)
|
|
BOOT_PATH="${ESP_PATH}/EFI/Linux"
|
|
|
|
declare -A kernels
|
|
update_all=0
|
|
|
|
while getopts ":hag:xyz" arg; do
|
|
case ${arg} in
|
|
g)
|
|
found=0
|
|
for line in $(pacman -Qql "$OPTARG"); do
|
|
if [[ $line =~ ^/usr/lib/modules/([^/]+)/pkgbase$ ]]; then
|
|
read -r kernel_name < "/${line}"
|
|
kernels["${kernel_name}"]="${BASH_REMATCH[1]}"
|
|
found=1
|
|
break
|
|
fi
|
|
done
|
|
if (( ! found )); then
|
|
echo "Error occurred during '$OPTARG' package traversal" >&2
|
|
exit 1
|
|
fi
|
|
;;
|
|
a)
|
|
update_all=1
|
|
;;
|
|
x)
|
|
check_root
|
|
# Trigger some IO on ESP path to be sure it's mounted by autofs if it's the case
|
|
# Otherwise upgrading systemd may cause ESP partition not mounted at the time dracut attempt to write new image
|
|
stat "$ESP_PATH" >/dev/null
|
|
;;
|
|
y)
|
|
check_root
|
|
while read -r line; do
|
|
if [[ "$line" == 'usr/lib/modules/'+([^/])'/pkgbase' ]]; then
|
|
read -r kernel_name < "/${line}"
|
|
path="$(grep -lE "^${kernel_name}\$" /usr/lib/modules/*/pkgbase)"
|
|
kernel_version=$(basename "${path%/pkgbase}")
|
|
|
|
efi_image="$BOOT_PATH/$kernel_name.efi"
|
|
efi_image_fallback="$BOOT_PATH/$kernel_name-fallback.efi"
|
|
|
|
declare -a images=()
|
|
images+=($efi_image)
|
|
images+=($efi_image_fallback)
|
|
|
|
for image in "${images[@]}";
|
|
do
|
|
if [ -f ]; then
|
|
echo "==> Removing $image..."
|
|
rm -f $image
|
|
fi
|
|
done
|
|
fi
|
|
done
|
|
exit 0
|
|
;;
|
|
z)
|
|
check_root
|
|
while read -r line; do
|
|
if [[ $line =~ ^usr/lib/modules/([^/]+)/pkgbase$ ]]; then
|
|
read -r kernel_name < "/${line}"
|
|
kernels["${kernel_name}"]="${BASH_REMATCH[1]}"
|
|
else
|
|
update_all=1
|
|
break
|
|
fi
|
|
done
|
|
;;
|
|
h)
|
|
usage
|
|
;;
|
|
*)
|
|
usage
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if (( update_all )); then
|
|
for kernel_path in /usr/lib/modules/*; do
|
|
[ -f "$kernel_path/pkgbase" ] || continue
|
|
|
|
kernel_version=$(basename $kernel_path)
|
|
kernel_dir="/usr/lib/modules/$kernel_version"
|
|
pkgbase="$kernel_dir/pkgbase"
|
|
kernel_name=$(sed -e 's/^[[:space:]]//g' -e 's/[[:space:]]$//g' "$pkgbase")
|
|
|
|
if [ -n "$kernel_name" ]; then
|
|
kernels["$kernel_name"]="$kernel_version"
|
|
fi
|
|
done
|
|
fi
|
|
|
|
function gen_image() {
|
|
check_root
|
|
kernel_name="$1"
|
|
kernel_version="$2"
|
|
|
|
kernel_dir="/usr/lib/modules/$kernel_version"
|
|
|
|
if [ -f "/etc/kernel/cmdline" ]; then
|
|
cmdline=$(sed -e 's/^[[:space:]]//g' -e 's/[[:space:]]$//g' "/etc/kernel/cmdline")
|
|
else
|
|
cmdline=$(sed -e 's/^[[:space:]]//g' -e 's/[[:space:]]$//g' -e 's/initrd.*$//g' "/proc/cmdline")
|
|
fi
|
|
|
|
vmlinuz="$kernel_dir/vmlinuz"
|
|
|
|
initrd_image="$(mktemp)"
|
|
initrd_image_fallback="$(mktemp)"
|
|
|
|
efi_image="$BOOT_PATH/$kernel_name.efi"
|
|
efi_image_fallback="$BOOT_PATH/$kernel_name-fallback.efi"
|
|
|
|
echo "==> Building initrd image $kernel_name $initrd_image"
|
|
dracut --force --hostonly --no-hostonly-cmdline --kver "$kernel_version" "$initrd_image"
|
|
|
|
declare -a ukify_args=("${ukify_global_args[@]}")
|
|
ukify_args+=(--cmdline "$cmdline")
|
|
ukify_args+=(--uname "$kernel_version")
|
|
ukify_args+=(--os-release "/etc/os-release")
|
|
ukify_args+=(--output "$efi_image")
|
|
ukify_args+=("$vmlinuz" "$initrd_image")
|
|
|
|
echo "==> Ukify image $kernel_name $efi_image"
|
|
/usr/lib/systemd/ukify "${ukify_args[@]}"
|
|
|
|
|
|
echo "==> Building initrd image $kernel_name $initrd_image_fallback"
|
|
dracut --force --kver "$kernel_version" "$initrd_image_fallback"
|
|
|
|
declare -a ukify_args=("${ukify_global_args[@]}")
|
|
ukify_args+=(--cmdline "$cmdline")
|
|
ukify_args+=(--uname "$kernel_version")
|
|
ukify_args+=(--os-release "/etc/os-release")
|
|
ukify_args+=(--output "$efi_image_fallback")
|
|
ukify_args+=("$vmlinuz" "$initrd_image_fallback")
|
|
|
|
echo "==> Ukify image $kernel_name $efi_image_fallback"
|
|
/usr/lib/systemd/ukify "${ukify_args[@]}"
|
|
|
|
rm -f "$initrd_image"
|
|
rm -f "$initrd_image_fallback"
|
|
}
|
|
|
|
for kernel_name in "${!kernels[@]}"; do
|
|
kernel_version="${kernels[$kernel_name]}"
|
|
gen_image "$kernel_name" "$kernel_version"
|
|
done
|