51-dracut-rescue.install

   1 #!/bin/bash
   2
   3 export LANG=C
   4
   5 COMMAND="$1"
   6 KERNEL_VERSION="$2"
   7 BOOT_DIR_ABS="${3%/*}/0-rescue"
   8 KERNEL_IMAGE="$4"
   9
  10
  11 dropindirs_sort()
  12 {
  13     suffix=$1; shift
  14     args=("$@")
  15     files=$(
  16         while (( $# > 0 )); do
  17             for i in ${1}/*${suffix}; do
  18                 [[ -f $i ]] && echo ${i##*/}
  19             done
  20             shift
  21         done | sort -Vu
  22     )
  23
  24     for f in $files; do
  25         for d in "${args[@]}"; do
  26             if [[ -f "$d/$f" ]]; then
  27                 echo "$d/$f"
  28                 continue 2
  29             fi
  30         done
  31     done
  32 }
  33
  34 [[ -f /etc/os-release ]] && . /etc/os-release
  35
  36 if [[ ${KERNEL_INSTALL_MACHINE_ID+x} ]]; then
  37     MACHINE_ID=$KERNEL_INSTALL_MACHINE_ID
  38 elif [[ -f /etc/machine-id ]] ; then
  39     read MACHINE_ID < /etc/machine-id
  40 fi
  41
  42 if ! [[ $MACHINE_ID ]]; then
  43     exit 0
  44 fi
  45
  46 if [[ -f /etc/kernel/cmdline ]]; then
  47     read -r -d '' -a BOOT_OPTIONS < /etc/kernel/cmdline
  48 elif [[ -f /usr/lib/kernel/cmdline ]]; then
  49     read -r -d '' -a BOOT_OPTIONS < /usr/lib/kernel/cmdline
  50 else
  51     read -r -d '' -a line < /proc/cmdline
  52     for i in "${line[@]}"; do
  53         [[ "${i#initrd=*}" != "$i" ]] && continue
  54         BOOT_OPTIONS+=("$i")
  55     done
  56 fi
  57
  58 if [[ -d "${BOOT_DIR_ABS%/*}" ]]; then
  59     BOOT_DIR="/${MACHINE_ID}/0-rescue"
  60     BOOT_ROOT=${BOOT_DIR_ABS%$BOOT_DIR}
  61     LOADER_ENTRY="$BOOT_ROOT/loader/entries/${MACHINE_ID}-0-rescue.conf"
  62     KERNEL="linux"
  63     INITRD="initrd"
  64 else
  65     BLS_DIR="/boot/loader/entries"
  66     BOOT_DIR_ABS="/boot"
  67     LOADER_ENTRY="$BLS_DIR/${MACHINE_ID}-0-rescue.conf"
  68     KERNEL="vmlinuz-0-rescue-${MACHINE_ID}"
  69     INITRD="initramfs-0-rescue-${MACHINE_ID}.img"
  70 fi
  71
  72 ret=0
  73
  74 case "$COMMAND" in
  75     add)
  76         [[ -f "$LOADER_ENTRY" ]] && [[ -f "$BOOT_DIR_ABS/$KERNEL" ]] \
  77             && [[ -f "$BOOT_DIR_ABS/$INITRD" ]] && exit 0
  78
  79         # source our config dir
  80         for f in $(dropindirs_sort ".conf" "/etc/dracut.conf.d" "/usr/lib/dracut/dracut.conf.d"); do
  81             [[ -e $f ]] && . "$f"
  82         done
  83
  84         [[ $dracut_rescue_image != "yes" ]] && exit 0
  85
  86         [[ -d "$BOOT_DIR_ABS" ]] || mkdir -p "$BOOT_DIR_ABS"
  87
  88         if ! cp --reflink=auto "$KERNEL_IMAGE" "$BOOT_DIR_ABS/$KERNEL"; then
  89             echo "Can't copy '$KERNEL_IMAGE to '$BOOT_DIR_ABS/$KERNEL'!" >&2
  90         fi
  91
  92         if [[ ! -f "$BOOT_DIR_ABS/$INITRD" ]]; then
  93             dracut -f --no-hostonly -a "rescue" "$BOOT_DIR_ABS/$INITRD" "$KERNEL_VERSION"
  94             ((ret+=$?))
  95         fi
  96
  97         if [[ "${BOOT_DIR_ABS}" != "/boot" ]]; then
  98             {
  99                 echo "title      $PRETTY_NAME - Rescue Image"
 100                 echo "version    $KERNEL_VERSION"
 101                 echo "machine-id $MACHINE_ID"
 102                 echo "options    ${BOOT_OPTIONS[@]} rd.auto=1"
 103                 echo "linux      $BOOT_DIR/linux"
 104                 echo "initrd     $BOOT_DIR/initrd"
 105             } > $LOADER_ENTRY
 106         else
 107             cp -aT "${KERNEL_IMAGE%/*}/bls.conf" $LOADER_ENTRY
 108             sed -i 's/'$KERNEL_VERSION'/0-rescue-'${MACHINE_ID}'/' $LOADER_ENTRY
 109         fi
 110
 111         ((ret+=$?))
 112         ;;
 113
 114     remove)
 115         exit 0
 116         ;;
 117
 118     *)
 119         usage
 120         ret=1;;
 121 esac
 122
 123 exit $ret