-#!/bin/bash
+# $Id$
set_bootloader() {
- if [ "x$BOOTLOADER" == 'xgrub' ]
- then
- set_grub_bootloader
- else
- return 0
- fi
+ case "${BOOTLOADER}" in
+ grub)
+ set_bootloader_grub
+ ;;
+ grub2)
+ set_bootloader_grub2
+ ;;
+ *)
+ print_warning "Bootloader ${BOOTLOADER} is not currently supported"
+ ;;
+ esac
}
-set_grub_bootloader() {
- local GRUB_CONF='/boot/grub/grub.conf'
+set_bootloader_read_fstab() {
+ local ROOTFS=$(awk 'BEGIN{RS="((#[^\n]*)?\n)"}( $2 == "/" ) { print $1; exit }' /etc/fstab)
+ local BOOTFS=$(awk 'BEGIN{RS="((#[^\n]*)?\n)"}( $2 == "'${BOOTDIR}'") { print $1; exit }' /etc/fstab)
- print_info 1 ''
- print_info 1 "Adding kernel to $GRUB_CONF..."
+ # If ${BOOTDIR} is not defined in /etc/fstab, it must be the same as /
+ [ -z "${BOOTFS}" ] && BOOTFS=${ROOTFS}
- # Extract block device information from /etc/fstab
- local GRUB_ROOTFS=$(awk '/[[:space:]]\/[[:space:]]/ { print $1 }' /etc/fstab)
- local GRUB_BOOTFS=$(awk '/^[^#].+[[:space:]]\/boot[[:space:]]/ { print $1 }' /etc/fstab)
-
- # If /boot is not defined in /etc/fstab, it must be the same as /
- [ "x$GRUB_BOOTFS" == 'x' ] && GRUB_BOOTFS=$GRUB_ROOTFS
+ echo "${ROOTFS} ${BOOTFS}"
+}
+set_bootloader_grub_read_device_map() {
# Read GRUB device map
- [ ! -d ${tmp} ] && mkdir ${tmp}
- grub --batch --device-map=${tmp}/grub.map <<EOF >/dev/null
-quit
-EOF
-
- # Get the GRUB mapping for our device
- local GRUB_BOOT_DISK1=$(echo $GRUB_BOOTFS | sed -e 's#\(/dev/.\+\)[[:digit:]]\+#\1#')
- local GRUB_BOOT_DISK=$(awk '{if ($2 == "'$GRUB_BOOT_DISK1'") {gsub(/(\(|\))/, "", $1); print $1;}}' ${tmp}/grub.map)
-
- local GRUB_BOOT_PARTITION=$(echo $GRUB_BOOTFS | sed -e 's#/dev/.\+\([[:digit:]]\+\)#\1#')
- [ ! -d ${tmp} ] && rm -r ${tmp}
-
- # Create grub configuration directory and file if it doesn't exist.
- [ ! -e `basename $GRUB_CONF` ] && mkdir -p `basename $GRUB_CONF`
-
- if [ ! -e $GRUB_CONF ]
+ [ ! -d ${TEMP} ] && mkdir ${TEMP}
+ echo "quit" | grub --batch --device-map=${TEMP}/grub.map &>/dev/null
+ echo "${TEMP}/grub.map"
+}
+
+set_bootloader_grub2() {
+ local GRUB_CONF
+ for candidate in \
+ "${BOOTDIR}/grub2/grub.cfg" \
+ "${BOOTDIR}/grub/grub.cfg" \
+ ; do
+ if [[ -e "${candidate}" ]]; then
+ GRUB_CONF=${candidate}
+ break
+ fi
+ done
+
+ if [[ -z "${GRUB_CONF}" ]]; then
+ print_error 1 "Error! Grub2 configuration file does not exist, please ensure grub2 is correctly setup first."
+ return 0
+ fi
+
+ print_info 1 "You can customize Grub2 parameters in /etc/default/grub."
+ print_info 1 "Running grub2-mkconfig to create ${GRUB_CONF}..."
+ grub2-mkconfig -o "${GRUB_CONF}"
+ [ "${BUILD_RAMDISK}" -ne 0 ] && sed -i 's/ro single/ro debug/' "${GRUB_CONF}"
+}
+
+set_bootloader_grub() {
+ local GRUB_CONF="${BOOTDIR}/grub/grub.conf"
+
+ print_info 1 "Adding kernel to ${GRUB_CONF}..."
+
+ if [ ! -e ${GRUB_CONF} ]
then
- if [ "${GRUB_BOOT_DISK}" = '' -o "${GRUB_BOOT_PARTITION}" = '' ]
+ local GRUB_BOOTFS
+ if [ -n "${BOOTFS}" ]
then
- print_error 1 'Error! /boot/grub/grub.conf does not exist and the correct settings can\'t be automatically detected.'
- print_error 1 'Please manually create your /boot/grub/grub.conf file.'
+ GRUB_BOOTFS=$BOOTFS
else
- # grub.conf doesn't exist - create it with standard defaults
- touch $GRUB_CONF
- echo 'default 0' >> $GRUB_CONF
- echo 'timeout 5' >> $GRUB_CONF
- echo >> $GRUB_CONF
-
- # Add grub configuration to grub.conf
- echo "# Genkernel generated entry, see GRUB documentation for details" >> $GRUB_CONF
- echo "title=Gentoo Linux ($KV)" >> $GRUB_CONF
- echo -e "\troot ($GRUB_BOOT_DISK,$GRUB_BOOT_PARTITION)" >> $GRUB_CONF
- if [ "${BUILD_INITRD}" -eq '0' ]
+ GRUB_BOOTFS=$(set_bootloader_read_fstab | cut -d' ' -f2)
+ fi
+
+ # Get the GRUB mapping for our device
+ local GRUB_BOOT_DISK1=$(echo ${GRUB_BOOTFS} | sed -e 's#\(/dev/.\+\)[[:digit:]]\+#\1#')
+ local GRUB_BOOT_DISK=$(awk '{if ($2 == "'${GRUB_BOOT_DISK1}'") {gsub(/(\(|\))/, "", $1); print $1;}}' ${TEMP}/grub.map)
+ local GRUB_BOOT_PARTITION=$(($(echo ${GRUB_BOOTFS} | sed -e 's#/dev/.\+\([[:digit:]]?*\)#\1#') - 1))
+
+ if [ -n "${GRUB_BOOT_DISK}" -a -n "${GRUB_BOOT_PARTITION}" ]
+ then
+
+ # Create grub configuration directory and file if it doesn't exist.
+ [ ! -d `dirname ${GRUB_CONF}` ] && mkdir -p `dirname ${GRUB_CONF}`
+
+ touch ${GRUB_CONF}
+ echo 'default 0' >> ${GRUB_CONF}
+ echo 'timeout 5' >> ${GRUB_CONF}
+ echo "root (${GRUB_BOOT_DISK},${GRUB_BOOT_PARTITION})" >> ${GRUB_CONF}
+ echo >> ${GRUB_CONF}
+
+ # Add grub configuration to grub.conf
+ echo "# Genkernel generated entry, see GRUB documentation for details" >> ${GRUB_CONF}
+ echo "title=Gentoo Linux ($KV)" >> ${GRUB_CONF}
+ echo -e "\tkernel /kernel-${KNAME}-${ARCH}-${KV} root=${GRUB_ROOTFS}" >> ${GRUB_CONF}
+ if [ "${BUILD_INITRD}" = '1' ]
then
- echo -e "\tkernel /kernel-${KNAME}-${ARCH}-${KV} root=$GRUB_ROOTFS" >> $GRUB_CONF
- else
- echo -e "\tkernel /kernel-${KNAME}-${ARCH}-${KV} root=/dev/ram0 init=/linuxrc real_root=$GRUB_ROOTFS" >> $GRUB_CONF
- if [ "${PAT}" -gt '4' -a "${CMD_BOOTSPLASH}" != '1' ]
+ if [ "${PAT}" -gt '4' ]
then
- echo -e "\tinitrd /initramfs-${KNAME}-${ARCH}-${KV}" >> $GRUB_CONF
- else
- echo -e "\tinitrd /initrd-${KNAME}-${ARCH}-${KV}" >> $GRUB_CONF
+ echo -e "\tinitrd /initramfs-${KNAME}-${ARCH}-${KV}" >> ${GRUB_CONF}
fi
fi
- echo >> $GRUB_CONF
+ echo >> ${GRUB_CONF}
+ else
+ print_error 1 "Error! ${BOOTDIR}/grub/grub.conf does not exist and the correct settings can not be automatically detected."
+ print_error 1 "Please manually create your ${BOOTDIR}/grub/grub.conf file."
fi
+
else
- # grub.conf already exists; so...
- # ... Clone the first boot definition and change the version.
- local TYPE
- [ "${KERN_24}" -eq '1' ] && TYPE='rd' || TYPE='ramfs'
-
- cp $GRUB_CONF $GRUB_CONF.bak
- awk 'BEGIN { RS="" ; FS="\n" ; OFS="\n" ; ORS="\n\n" }
- NR == 2 {
- ORIG=$0;
- sub(/\(.+\)/,"(" KV ")",$1);
- sub(/kernel-[[:alnum:][:punct:]]+/, "kernel-" KNAME "-" ARCH "-" KV, $3);
- sub(/initr(d|amfs)-[[:alnum:][:punct:]]+/, "init" TYPE "-" KNAME "-" ARCH "-" KV, $4);
- print RS $0;
- print RS ORIG;}
- NR != 2 { print RS $0; }' KNAME=$KNAME ARCH=$ARCH KV=$KV TYPE=$TYPE $GRUB_CONF.bak > $GRUB_CONF
+ # The grub.conf already exists, so let's try to duplicate the default entry
+ if set_bootloader_grub_check_for_existing_entry "${GRUB_CONF}"; then
+ print_warning 1 "An entry was already found for a kernel/initramfs with this name...skipping update"
+ return 0
+ fi
+
+ set_bootloader_grub_duplicate_default "${GRUB_CONF}"
fi
+
+}
+
+set_bootloader_grub_duplicate_default_replace_kernel_initrd() {
+ sed -r -e "/^[[:space:]]*kernel/s/kernel-[[:alnum:][:punct:]]+/kernel-${KNAME}-${ARCH}-${KV}/" - |
+ sed -r -e "/^[[:space:]]*initrd/s/init(rd|ramfs)-[[:alnum:][:punct:]]+/init\1-${KNAME}-${ARCH}-${KV}/"
+}
+
+set_bootloader_grub_check_for_existing_entry() {
+ local GRUB_CONF=$1
+ if grep -q "^[[:space:]]*kernel[[:space:]=]*.*/kernel-${KNAME}-${ARCH}-${KV}\([[:space:]]\|$\)" "${GRUB_CONF}" &&
+ grep -q "^[[:space:]]*initrd[[:space:]=]*.*/initramfs-${KNAME}-${ARCH}-${KV}\([[:space:]]\|$\)" "${GRUB_CONF}"
+ then
+ return 0
+ fi
+ return 1
+}
+
+set_bootloader_grub_duplicate_default() {
+ local GRUB_CONF=$1
+ local GRUB_CONF_TMP="${GRUB_CONF}.tmp"
+
+ line_count=$(wc -l < "${GRUB_CONF}")
+ line_nums="$(grep -n "^title" "${GRUB_CONF}" | cut -d: -f1)"
+ if [ -z "${line_nums}" ]; then
+ print_error 1 "No current 'title' entries found in your grub.conf...skipping update"
+ return 0
+ fi
+ line_nums="${line_nums} $((${line_count}+1))"
+
+ # Find default entry
+ default=$(sed -rn '/^[[:space:]]*default[[:space:]=]/s/^.*default[[:space:]=]+([[:alnum:]]+).*$/\1/p' "${GRUB_CONF}")
+ if [ -z "${default}" ]; then
+ print_warning 1 "No default entry found...assuming 0"
+ default=0
+ fi
+ if ! echo ${default} | grep -q '^[0-9]\+$'; then
+ print_error 1 "We don't support non-numeric (such as 'saved') default values...skipping update"
+ return 0
+ fi
+
+ # Grub defaults are 0 based, cut is 1 based
+ # Figure out where the default entry lives
+ startstop=$(echo ${line_nums} | cut -d" " -f$((${default}+1))-$((${default}+2)))
+ startline=$(echo ${startstop} | cut -d" " -f1)
+ stopline=$(echo ${startstop} | cut -d" " -f2)
+
+ # Write out the bits before the default entry
+ sed -n 1,$((${startline}-1))p "${GRUB_CONF}" > "${GRUB_CONF_TMP}"
+
+ # Put in our title
+ echo "title=Gentoo Linux (${KV})" >> "${GRUB_CONF_TMP}"
+
+ # Pass the default entry (minus the title) through to the replacement function and pipe the output to GRUB_CONF_TMP
+ sed -n $((${startline}+1)),$((${stopline}-1))p "${GRUB_CONF}" | set_bootloader_grub_duplicate_default_replace_kernel_initrd >> "${GRUB_CONF_TMP}"
+
+ # Finish off with everything including the previous default entry
+ sed -n ${startline},${line_count}p "${GRUB_CONF}" >> "${GRUB_CONF_TMP}"
+
+ cp "${GRUB_CONF}" "${GRUB_CONF}.bak"
+ cp "${GRUB_CONF_TMP}" "${GRUB_CONF}"
+ rm "${GRUB_CONF_TMP}"
}