X-Git-Url: http://git.tremily.us/?a=blobdiff_plain;f=gen_bootloader.sh;h=5848bd24220c8aeefe6c0b254a3221afde19dda3;hb=refs%2Fheads%2Fmaster;hp=c484b3782a5c72924a0baed954a13316b348424a;hpb=bcc80c316220cc96f13e1e9125cf57730ebf0c19;p=genkernel.git diff --git a/gen_bootloader.sh b/gen_bootloader.sh old mode 100644 new mode 100755 index c484b37..5848bd2 --- a/gen_bootloader.sh +++ b/gen_bootloader.sh @@ -1,100 +1,177 @@ -#!/bin/bash +# $Id$ set_bootloader() { - if [ "x$BOOTLOADER" == 'xgrub' ] - then - set_grub_bootloader - else + case "${BOOTLOADER}" in + grub) + set_bootloader_grub + ;; + grub2) + set_bootloader_grub2 + ;; + *) + print_warning "Bootloader ${BOOTLOADER} is not currently supported" + ;; + esac +} + +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) + + # If ${BOOTDIR} is not defined in /etc/fstab, it must be the same as / + [ -z "${BOOTFS}" ] && BOOTFS=${ROOTFS} + + echo "${ROOTFS} ${BOOTFS}" +} + +set_bootloader_grub_read_device_map() { + # Read GRUB device map + [ ! -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_grub_bootloader() { +set_bootloader_grub() { local GRUB_CONF="${BOOTDIR}/grub/grub.conf" - print_info 1 '' - print_info 1 "Adding kernel to $GRUB_CONF..." - if [ "${BOOTFS}" != '' ] - then - GRUB_BOOTFS=${BOOTFS} - else - # Extract block device information from /etc/fstab - GRUB_ROOTFS=$(awk 'BEGIN{RS="((#[^\n]*)?\n)"}( $2 == "/" ) { print $1; exit }' /etc/fstab) - GRUB_BOOTFS=$(awk 'BEGIN{RS="((#[^\n]*)?\n)"}( $2 == "'${BOOTDIR}'") { print $1; exit }' /etc/fstab) - - # If ${BOOTDIR} is not defined in /etc/fstab, it must be the same as / - [ "x$GRUB_BOOTFS" == 'x' ] && GRUB_BOOTFS=$GRUB_ROOTFS - fi + print_info 1 "Adding kernel to ${GRUB_CONF}..." - # Read GRUB device map - [ ! -d ${TEMP} ] && mkdir ${TEMP} - grub --batch --device-map=${TEMP}/grub.map </dev/null 2>&1 -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;}}' ${TEMP}/grub.map) - - local GRUB_BOOT_PARTITION=$(echo $GRUB_BOOTFS | sed -e 's#/dev/.\+\([[:digit:]]?*\)#\1#') - [ ! -d ${TEMP} ] && rm -r ${TEMP} - - # Create grub configuration directory and file if it doesn't exist. - [ ! -e `basename $GRUB_CONF` ] && mkdir -p `basename $GRUB_CONF` - - if [ ! -e $GRUB_CONF ] + if [ ! -e ${GRUB_CONF} ] then - if [ "${GRUB_BOOT_DISK}" != '' -a "${GRUB_BOOT_PARTITION}" != '' ] + local GRUB_BOOTFS + if [ -n "${BOOTFS}" ] then - GRUB_BOOT_PARTITION=`expr ${GRUB_BOOT_PARTITION} - 1` - # 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=$BOOTFS + else + 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 -f $GRUB_CONF $GRUB_CONF.bak - awk 'BEGIN { RS="\n"; } - { - if(match($0, "kernel-" KNAME "-" ARCH "-" KV)) - { have_k="1" } - if(match($0, "init" TYPE "-" KNAME "-" ARCH "-" KV)) - { have_i="1" } - if(have_k == "1" && have_i == "1") - { exit 1; } - }' KNAME=$KNAME ARCH=$ARCH KV=$KV TYPE=$TYPE $GRUB_CONF.bak - if [ "$?" -eq '0' ] - then - local LIMIT=$(wc -l $GRUB_CONF.bak) - awk -f ${GK_BIN}/gen_bootloader_grub.awk LIMIT=${LIMIT/ */} KNAME=$KNAME ARCH=$ARCH KV=$KV TYPE=$TYPE $GRUB_CONF.bak > $GRUB_CONF - else - print_info 1 "GRUB: Definition found, not duplicating." + # 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}" }