source: MondoRescue/branches/3.0/mondo/src/restore-scripts/mondo/grub-MR@ 2944

#!/bin/sh
#
# $Id: grub-MR 2944 2012-02-03 01:12:47Z bruno $
#
# grub-MR ------ a rudimentary replacement for grub-install
#
#

Die() {
    echo "$1" >> /dev/stderr
    exit 1
}


FindBootPart() {
    local i
    bootpart=""
    BOOTPATHNAME=""
    for sz in /boot / ; do
        bootpart=`grep " $sz " $2 | cut -d' ' -f1 | head -n1`
        [ "$bootpart" ] && [ -e "/mnt/RESTORING/$bootpart" ] && break
    done
    [ ! "$bootpart" ] && Die "Cannot find boot partition in mountlist"

    if [ "$sz" = "/boot" ] ; then
        BOOTPATHNAME=/grub
    else
        BOOTPATHNAME=/boot/grub
    fi
    if [ -e "$MNT_RESTORING/boot/grub/stage1" ] ; then
        echo "All right then."
        return
    fi
    
    cd $MNT_RESTORING
    for i in usr/lib/grub/* ; do
        echo "Checking $i"
        if [ -e "$i/stage1" ] ; then
            BOOTPATHNAME=/$i
            mkdir -p /boot/grub
            cp -au $i/* /boot/grub/
            echo "BOOTPATHNAME is now $BOOTPATHNAME"
            return 0
        fi
    done
    cd /
    echo "Cannot find BOOTPATHNAME"
    return 1
}

FindPathOfRESTExe() {
    local path
    for path in /usr/sbin /usr/bin /usr/local/sbin /usr/local/bin \
/bin /sbin ; do
        if [ -f "/mnt/RESTORING/$path/$1" ] ; then 
            echo "/mnt/RESTORING/$path/$1"
        return
        fi
    done
    which grub
}


try_grub_hack() {
    local driveno extraline partno
    driveno=$1
    partno=$2
    extraline="$3"

    echo -en "\
    device (hd$driveno) $mbrdev\n\
    $extraline\n\
    root (hd$driveno,$partno)\n\
    setup (hd$driveno)\n\
    quit \n" > /tmp/feed.txt
    cat /tmp/feed.txt
    log_file=/tmp/grub.output
    if [ "$MNT_RESTORING" ] ; then
        chroot $MNT_RESTORING grub --batch < /tmp/feed.txt > $log_file
    else
        grub --batch < /tmp/feed.txt > $log_file
    fi
    cat $log_file
    if grep "Error [0-9]*: " $log_file >/dev/null; then
            return 1
    else
            return 0
    fi
}


# ---------------------------------- main -----------------------------------

if [ "$#" -ne "2" ] ; then
    Die "grub-MR <MBR drive> <mountlist filename>; e.g. grub-MR /dev/hda /tmp/mountlist.txt"
fi
[ ! -f "$2" ] && Die "mountlist file not found"

if [ -e "/mnt/RESTORING/boot" ] ; then
    MNT_RESTORING=/mnt/RESTORING
else
    MNT_RESTORING=""
fi

# For some OpenSuSE
res=1
if [ "$MNT_RESTORING" ] ; then
    if [ -x $MNT_RESTORING/usr/sbin/grub-install.unsupported ]; then
        echo "Now I'll use grub-install.unsupported in chroot" >> $LOGFILE
        chroot $MNT_RESTORING /usr/sbin/grub-install.unsupported $1 >> $LOGFILE 2>> $LOGFILE
        res=$?
        echo "grub-install.unsupported in chroot returned $res" >> $LOGFILE
    fi
else
    if [ -x /usr/sbin/grub-install.unsupported ]; then
        echo "Now I'll use grub-install.unsupported locally" >> $LOGFILE
        /usr/sbin/grub-install.unsupported $1 >> $LOGFILE 2>> $LOGFILE
        res=$?
        echo "grub-install.unsupported returned $res" >> $LOGFILE
    fi
fi
[ "$res" -eq "0" ] && exit 0

echo "Now I'll use grub-install" >> $LOGFILE
if [ "$MNT_RESTORING" ] ; then
    chroot $MNT_RESTORING grub-install $1 >> $LOGFILE 2>> $LOGFILE
    res=$?
else
    grub-install $1 >> $LOGFILE 2>> $LOGFILE
    res=$?
fi
echo "grub-install returned $res" >> $LOGFILE
[ "$res" -eq "0" ] && exit 0


echo "Now I'll use grub2-install" >> $LOGFILE
if [ "$MNT_RESTORING" ] ; then
    chroot $MNT_RESTORING grub2-install $1 >> $LOGFILE 2>> $LOGFILE
    res=$?
else
    grub2-install $1 >> $LOGFILE 2>> $LOGFILE
    res=$?
fi
echo "grub2-install returned $res" >> $LOGFILE
[ "$res" -eq "0" ] && exit 0

echo "Trying a hack" >> $LOGFILE
FindBootPart $1 $2 2>&1 | tee -a $LOGFILE
mbrdev=`echo $1 | sed 's/\([^0-9]*\)[0-9]*$/\1/'`
if echo $mbrdev | grep "/cciss/" > /dev/null ; then
    partno=`basename $mbrdev | cut -d'p' -f2`
    mbrdev=`echo $mbrdev | cut -d'p' -f1`
elif echo $mbrdev | grep "/mapper/mpath" > /dev/null ; then
    partno=`basename $mbrdev | cut -d'p' -f3`
    mbrdev=`echo $mbrdev | cut -d'p' -f1-4`
else
    partno=`basename $mbrdev | sed 's/[a-z]*//'`
fi
if [ ! "$partno" ] ; then
    partno=0
else
    partno=$(($partno-1))
fi
if echo $mbrdev | grep "/md" > /dev/null ; then
    # FIXME: Why this if not used later
    base=`basename $bootpart`
    line=`grep $base /proc/mdstat | head -n1`
    echo "mbrdev was $mbrdev" 2>&1 | tee -a $LOGFILE
    mbrdev=`parted2fdisk -l | grep /dev/ | head -n1 | tr ':' ' ' | cut -d' ' -f2`
    echo "mbrdev is $mbrdev" 2>&1 | tee -a $LOGFILE
    partno="0"; # cheating - FIXME    
fi
echo ".............Cool." 2>&1 | tee -a $LOGFILE

grub=`FindPathOfRESTExe grub`
mkdir -p /boot
[ "$MNT_RESTORING" ] && ln -sf /mnt/RESTORING/boot/grub /boot/grub

# ---------------------------------

for driveno in 0 1 2 ; do
    try_grub_hack $driveno $partno "" >> $LOGFILE 2>> $LOGFILE && break
    try_grub_hack $driveno $partno "find $BOOTPATHNAME/stage1" >> $LOGFILE 2>> $LOGFILE && break
done
res=$?
echo "Hack returned res=$res" >> $LOGFILE
[ "$res" -eq "0" ] && exit 0

# ---------------------------------

echo "Now I'll use grub-install.patched" >> $LOGFILE
cp -f `which grub-install.patched` /mnt/RESTORING/sbin
chroot /mnt/RESTORING grub-install.patched $1 >> $LOGFILE 2>> $LOGFILE
res=$?
echo "grub-install.patched returned $res" >> $LOGFILE
[ "$res" -eq "0" ] && exit 0

# ---------------------------------

if [ -f "/mnt/RESTORING/boot/grub/menu.lst" ]; then
    grep -vE '^#' /boot/grub/menu.lst > /mnt/RESTORING/tmp/grub.conf
elif [ -f "/mnt/RESTORING/boot/grub/grub.cfg" ]; then
    grep -vE '^#' /boot/grub/grub.cfg > /mnt/RESTORING/tmp/grub.conf
else
    echo "Unable to find Grub conf file" | tee -a $LOGFILE
    exit -1
fi
echo "Trying to use the existing grub conf file in batch mode" >> $LOGFILE
cat /mnt/RESTORING/tmp/grub.conf >> $LOGFILE
chroot /mnt/RESTORING grub --batch < /mnt/RESTORING/tmp/grub.conf
res=$?
echo "grub conf file approach returned $res" >> $LOGFILE
rm -f /mnt/RESTORING/tmp/grub.conf
[ "$res" -eq "0" ] && exit 0

echo "grub-MR returned res=$res" >> $LOGFILE

exit $res