Subversion Repositories livecd

#!/bin/bash
#
###############################################################
#
# Installes the LiveCD on local harddisk
# 
# Boot LiveCD and run this script as root
#
# Urs Beyerle, PSI
#
###############################################################


###############################################################
# Definitions
###############################################################

# files which should be restored from .ori files
FILES_RESTORE="/etc/init.d/netfs \
           /etc/init.d/autofs \
           /etc/init.d/halt \
           /etc/init.d/network
           /etc/init.d/functions \
           /etc/rc.d/rc.sysinit \
           /etc/sysconfig/afs \
           /etc/motd \
           /etc/redhat-release \
           /etc/rc.d/rc.local \
           /etc/rc.d/rc.sysinit"

LIVECD_INIT_SCRIPTS="runfirst \
                    runveryfirst \
                    runlast \
                    kudzu-auto"

# Main directories which should be not be copied
DIR_NOT_COPY="/proc \
              /dev \
              /livecd \
              /boot \
              /afs \
              /sys \
              /mnt \
              /media"

# Mount point of the new SL system
NEW=/mnt/harddisk

# Name of the script
SCRIPTNAME=$( basename $0 )




###############################################################
# Functions
###############################################################

function usage() {

   ## Usage
   # ----------------------------------------------------------

   cat <<EOF
   
  $SCRIPTNAME [OPTIONS] -mbr=[DEVICE] [PARTITION]

    Installes LiveCD on PARTITION and the bootloader grub into 
    the Master Boot Record (MBR) of DEVICE. The MBR will be 
    backed up to PARTITION.

  OPTIONS:

    -h   --help         Print this screen
    -swap=[partition]   Use [partition] as swap
    -win=[partition]    Your active Windows partition. If not given, 
                        $SCRIPTNAME tries to find it
    -nogrub             Do not install grub. You have to install 
                        manually a bootloader (not recommended)
    -floppy             Needed, if grub should be installed on floppy
    -y                  Answer all questions with yes

  Example:

  $SCRIPTNAME -swap=/dev/sda3 -mbr=/dev/sda /dev/sda2

    Will install LiveCD on /dev/sda2 (= second partition on 
    first SATA disk). All data on /dev/sda2 will be deleted.
    /dev/sda3 has to be a Linux Swap partion. GRUB will be 
    installed in the MBR of /dev/sda (= first SATA disk).
   
EOF
}


function exit_now() {

    local exitcode=$1
    umount $INSTALL_PART 2>/dev/null
    exit $exitcode
}



###############################################################
# Main
###############################################################

### are we root?
### -----------------------------------------------------------
if [ "$( whoami )" != "root" ]; then
    echo "Please run this script as roo/home/l_beyerle/svn-livecd/livecd/trunkt."
    exit_now 1
fi


### read options from command-line
### -----------------------------------------------------------
while [ $# -gt 0 ]; do

    case "$1" in
       -h)
            usage; exit_now;;

       --help)
            usage; exit_now;;

       -swap*)
           if echo $1 | grep -q '=' ; then
               SWAP_PART=$( echo $1 | sed 's/^-swap=//' )
           else
               shift
               SWAP_PART=$1
           fi
           shift; continue;;

       -mbr*)
           if echo $1 | grep -q '=' ; then
               MBR_DEV=$( echo $1 | sed 's/^-mbr=//' )
           else
               shift
               MBR_DEV=$1
           fi
           shift; continue;;

       -win*)
           if echo $1 | grep -q '=' ; then
               WIN_PART=$( echo $1 | sed 's/^-win=//' )
           else
               shift
               WIN_PART=$1
           fi
           shift; continue;;

       -nogrub)
            NOGRUB=$1; shift; continue;;

       -floppy)
            FLOPPY=$1; shift; continue;;

       -y)
            YES=$1; shift; continue;;

       *)
            INSTALL_PART=$1; break;;
    esac

done
echo


### test if $INSTALL_PART is defined and exists
### -----------------------------------------------------------
if [ ! $INSTALL_PART ]; then
    echo "No partition defined for installation"
    echo "Please see '$SCRIPTNAME -h'."; echo
    exit_now 1
fi


### test if MBR_DEV is given
### -----------------------------------------------------------
if [ ! $NOGRUB ]; then
    if [ ! $MBR_DEV ]; then
        echo "No MBR device defined."
        echo "Please see '$SCRIPTNAME -h'."; echo
        exit_now 1
    fi
fi


### test if $INSTALL_PART exists
### -----------------------------------------------------------
fdisk -l | cut -d" " -f1 | grep -q "^${INSTALL_PART}$"
if [ "$?" != "0" ]; then
    echo "Partition $INSTALL_PART not found! (see 'disk -l')"; echo
    exit_now 1
fi


### set $INSTALL_DEV (eg. /dev/sda)
### -----------------------------------------------------------
INSTALL_DEV=$( echo "$INSTALL_PART" | sed -e 's%\([sh]d[a-z]\)[0-9]*$%\1%' )
INSTALL_PART_NR=$( echo "$INSTALL_PART" | sed -e 's%.*/[sh]d[a-z]\([0-9]*\)$%\1%' )


### test if $SWAP_PART exists
### -----------------------------------------------------------
if [ $SWAP_PART ]; then
    fdisk -l | cut -d" " -f1 | grep -q "^${SWAP_PART}$"
    if [ "$?" != "0" ]; then
        echo "Swap partition $SWAP_PART not found! (see 'disk -l')"; echo
        exit_now 1
    fi
    fdisk -l | grep "Linux swap" | cut -d" " -f1 | grep -q "^${SWAP_PART}$"
    if [ "$?" != "0" ]; then
        echo "Partition $SWAP_PART is not a Linux swap partition! (see 'disk -l')"
        echo "Use fdisk to create a Linux swap partition!"; echo
        exit_now 1
    fi
fi


### print warning
### -----------------------------------------------------------
echo                     "------------------------------------------------------------"
echo                     "   LiveCD will be installed on partition $INSTALL_PART"
[ "$SWAP_PART" ] && echo " Partition $SWAP_PART will be used as swap partition."
[ ! $NOGRUB ]    && echo " GRUB will be installed in Master Boot Record of $MBR_DEV"
echo                     "       !! All data on $INSTALL_PART will be lost !!"
echo                     "------------------------------------------------------------"
echo


### continue?
### -----------------------------------------------------------
if [ ! $YES ]; then
    echo -n "Continue (y/N)? "
    read key
    echo
    [ "$key" != "y" ] && exit_now 0
fi


### format $SWAP_PART
### -----------------------------------------------------------
if [ $SWAP_PART ]; then
    echo "Make swap on $SWAP_PART ..."
    mkswap $SWAP_PART
    echo
fi


### format $INSTALL_PART
### -----------------------------------------------------------
echo -n "Format $INSTALL_PART, please wait ... " 
mkfs.ext3 -q $INSTALL_PART || exit_now 1
echo "done."; echo


### mount $INSTALL_PART
### -----------------------------------------------------------
echo -n "Try to mount $INSTALL_PART to $NEW ... "
mkdir -p $NEW
mount $INSTALL_PART $NEW || exit_now 1
echo "done."; echo


### copy root dirs
### -----------------------------------------------------------
echo "Copy Live System to $INSTALL_PART ..."
root_dirs=$( ls / )
for dir in $root_dirs; do
    # check if dir is not in $DIR_NOT_COPY
    do_not_copy=""
    for not_dir in $DIR_NOT_COPY; do
        if [ "$not_dir" = "/$dir" ]; then 
            do_not_copy="yes"
            break
        fi
    done
    # do not copy links
    [ -L /$dir ] && do_not_copy="yes"

    fail=""
    if [ ! $do_not_copy ]; then
        echo -n "  * Copy  /$dir ... "
        cp -a /$dir $NEW || fail=true
        echo "done."
    fi
done
echo
if [ $fail ]; then
    echo "ERROR: Not everything was copied to $INSTALL_PART"
    exit_now 1
fi


### move /usr/opt back to /opt
### -----------------------------------------------------------
if [ -d $NEW/usr/opt ]; then
    echo -n "Move /opt back ... "
    mv $NEW/usr/opt $NEW/
    echo "done."; echo
fi


### create dirs which were not copied
### -----------------------------------------------------------
for dir in $DIR_NOT_COPY; do
    mkdir $NEW/$dir
done
# we do not need this directories
rmdir $NEW/livecd
# if not yet existing, create
mkdir -p $NEW/srv
mkdir -p $NEW/selinux


### copy back original files
### -----------------------------------------------------------
echo -n "Restore original files ... " 
for file in $FILES_RESTORE; do
    [ -r $NEW/${file}.ori ] && cp -a $NEW/${file}.ori $NEW/${file} 
done
echo "done."; echo


### define kernel version
### -----------------------------------------------------------
rpm --quiet -q kernel     && UP_installed=true
rpm --quiet -q kernel-smp && SMP_installed=true
[ $UP_installed ]  && KERNEL_VERSION=$( rpm -q --qf "%{V}-%{R}" kernel 2>/dev/null )
[ $SMP_installed ] && KERNEL_VERSION=$( rpm -q --qf "%{V}-%{R}" kernel-smp  2>/dev/null )
if [ ! $KERNEL_VERSION ]; then
    echo "ERROR: Kernel version could not be determined - installation failed"; echo
    exit_now 1
fi    



if [ ! $NOGRUB ]; then

    ### Backup Master Boot Record MBR
    ### -----------------------------------------------------------
    if [ ! $NOGRUB ]; then
        # do we have already a backup?
        MBR_FILENAME="MBR$( echo $MBR_DEV | tr / _ ).bak"
        if [ ! -e /tmp/$MBR_FILENAME ]; then
            echo "Backup Master Boot Record (MBR) of $MBR_DEV ..."
            dd if=$MBR_DEV of=/tmp/$MBR_FILENAME bs=512 count=1
        fi
        cp -a /tmp/$MBR_FILENAME $NEW/$MBR_FILENAME
        echo "MBR saved as $MBR_FILENAME on $INSTALL_PART and on /tmp"
        echo
    fi


    ### install grub
    ### -----------------------------------------------------------
    echo "Run grub-install ... "
    mkdir -p $NEW/boot/grub
    if [ $FLOPPY ]; then
        grub-install --root-directory=$NEW $MBR_DEV
    else
        grub-install --no-floppy --root-directory=$NEW $MBR_DEV
    fi
    echo "done."; echo


    ### check for device.map file 
    ### -----------------------------------------------------------
    DEVICE_MAP=$NEW/boot/grub/device.map
    if [ ! -e $NEW/boot/grub/device.map ]; then
        echo "ERROR: $NEW/boot/grub/device.map not found"
        exit_now 1
    fi


    ### convert dev syntax to grub syntax
    ### -----------------------------------------------------------
    GRUB_INSTALL_DEV=$( grep $INSTALL_DEV $DEVICE_MAP | awk '{ print $1 }' )
    GRUB_ROOT_PART=$( echo "$GRUB_INSTALL_DEV" | sed "s%)$%,`expr $INSTALL_PART_NR - 1`)%" )


    ### find active Windows partition
    ### -----------------------------------------------------------
    if [ ! $WIN_PART ]; then
        # try to find active Windows partition
        WIN_PART=$( fdisk -l 2>/dev/null | awk '{ if ($2 == "*" && $7 ~ "NTFS") print $1 }' | head -1 )
    fi

    if [ $WIN_PART ]; then
        WIN_installed=true
        WIN_DISK=$( echo "$WIN_PART" | sed -e 's%\([sh]d[a-z]\)[0-9]*$%\1%' )
        WIN_PART_NR=$( echo "$WIN_PART" | sed -e 's%.*/[sh]d[a-z]\([0-9]*\)$%\1%' )
        # convert dev syntax to grub syntax
        GRUB_WIN_DEV=$( grep $WIN_DISK $DEVICE_MAP | awk '{ print $1 }' )
        GRUB_WIN_PART=$( echo "$GRUB_WIN_DEV" | sed "s%)$%,`expr $WIN_PART_NR - 1`)%" )

        # $GRUB_WIN_PART should be something like (hd0,0)
        echo "Found active Windows partition ( $WIN_PART = $GRUB_WIN_PART )" 
        echo "Will add entry for Windows in GRUB."
        echo
    fi


    ### create grub.conf file
    ### -----------------------------------------------------------
    echo "Create grub.conf ..."

    cat > $NEW/boot/grub/grub.conf <<EOF
# grub.conf generated by $SCRIPTNAME
default=0
timeout=5
splashimage=$GRUB_ROOT_PART/boot/grub/splash.xpm.gz
#hiddenmenu
EOF

    if [ $UP_installed ]; then
        echo "Add entry for UP kernel into grub.conf"
        cat >> $NEW/boot/grub/grub.conf <<EOF
title Scientific Linux (${KERNEL_VERSION})
        root $GRUB_ROOT_PART
        kernel /boot/vmlinuz-$KERNEL_VERSION ro root=$INSTALL_PART
        initrd /boot/initrd-$KERNEL_VERSION.img
EOF
    fi

    if [ $SMP_installed ]; then
        echo "Add entry for SMP kernel into grub.conf"
        cat >> $NEW/boot/grub/grub.conf <<EOF
title Scientific Linux (${KERNEL_VERSION}smp)
        root $GRUB_ROOT_PART
        kernel /boot/vmlinuz-${KERNEL_VERSION}smp ro root=$INSTALL_PART
        initrd /boot/initrd-${KERNEL_VERSION}smp.img
EOF
    fi

    if [ $WIN_installed ]; then
        echo "Add entry for Windows into grub.conf"
        cat >> $NEW/boot/grub/grub.conf <<EOF
title Windows
        rootnoverify $GRUB_WIN_PART
        chainloader +1
EOF
    fi

    chmod 600 $NEW/boot/grub/grub.conf
    ln -s ../boot/grub/grub.conf $NEW/etc/grub.conf
    ln -s ./grub.conf $NEW/boot/grub/menu.lst
    echo "done."; echo

fi


### install kernel etc. into /boot
### -----------------------------------------------------------
echo "Install kernel(s) ..."

[ -e /boot/vmlinuz ]      && BOOT_DIR=/boot
[ -e /boot/boot/vmlinuz ] && BOOT_DIR=/boot/boot

if [ ! $BOOT_DIR ]; then
    echo "ERROR: No kernel found - installation failed"; echo
    exit_now 1
fi

cp -a $BOOT_DIR/vmlinuz            $NEW/boot/vmlinuz-${KERNEL_VERSION}
cp -a $BOOT_DIR/vmlinuzs           $NEW/boot/vmlinuz-${KERNEL_VERSION}smp  2>/dev/null

cp -a $BOOT_DIR/System.map-*       $NEW/boot/
cp -a $BOOT_DIR/config-*           $NEW/boot/

cp -a $BOOT_DIR/symvers-*.gz       $NEW/boot/        2>/dev/null

cp -a $BOOT_DIR/grub/splash.xpm.gz $NEW/boot/grub/

cp -a $BOOT_DIR/message.ja         $NEW/boot/        2>/dev/null
cp -a $BOOT_DIR/message            $NEW/boot/        2>/dev/null

echo "done."; echo


### create /etc/fstab
### -----------------------------------------------------------
cat > $NEW/etc/fstab <<EOF
$INSTALL_PART         /                    ext3    defaults        1 1
devpts            /dev/pts             devpts  gid=5,mode=620  0 0
tmpfs             /dev/shm             tmpfs   defaults        0 0
proc              /proc                proc    defaults        0 0
sysfs             /sys                 sysfs   defaults        0 0
EOF
if [ $SWAP_PART ]; then
    echo "$SWAP_PART         swap                 swap    defaults        0 0" >> $NEW/etc/fstab
fi


### make initrd 
### (needs $NEW/etc/fstab to find correct modules for root filesystem !!)
### -----------------------------------------------------------
echo "Create initrd(s) ..."
# initrd should not be build on tmpfs (we take $NEW/tmp instead of /tmp)
sed -i "s|^TMPDIR=.*|TMPDIR=\"$NEW/tmp\"|" /usr/sbin/livecd-mkinitrd
 
if [ $UP_installed ]; then
    depmod -a ${KERNEL_VERSION}
    /usr/sbin/livecd-mkinitrd --fstab=$NEW/etc/fstab \
                    $NEW//boot/initrd-${KERNEL_VERSION}.img ${KERNEL_VERSION}
    if [ ! -e $NEW/boot/initrd-${KERNEL_VERSION}.img ]; then
        echo "ERROR: Failed to create $NEW/boot/initrd-${KERNEL_VERSION}.img"
        exit_now 1
    fi
fi
if [ $SMP_installed ]; then
    depmod -a ${KERNEL_VERSION}smp
    /usr/sbin/livecd-mkinitrd --fstab=$NEW/etc/fstab \
                    $NEW/boot/initrd-${KERNEL_VERSION}smp.img ${KERNEL_VERSION}smp
    if [ ! -e $NEW/boot/initrd-${KERNEL_VERSION}smp.img ]; then
        echo "ERROR: Failed to create $NEW/boot/initrd-${KERNEL_VERSION}smp.img"
        exit_now 1
    fi
fi
echo "done."; echo


### remove LiveCD init.d scripts
### -----------------------------------------------------------
for file in $LIVECD_INIT_SCRIPTS; do
    rm -f $NEW/etc/init.d/$file
    for n in 0 1 2 3 4 5 6; do
        rm -f $NEW/etc/rc.d/rc${n}.d/*$file
    done
done


### restore cronjobs
### -----------------------------------------------------------
mv $NEW/etc/cron_backup/sysstat $NEW/etc/cron.d/ 2>/dev/null
mv $NEW/etc/cron_backup/00-makewhatis.cron.weekly $NEW/etc/cron.weekly/00-makewhatis.cron 2>/dev/null
mv $NEW/etc/cron_backup/* $NEW/etc/cron.daily/


### turn on kudzu again
### -----------------------------------------------------------
chroot $NEW chkconfig kudzu on


### umount $INSTALL_PART
### -----------------------------------------------------------
umount $INSTALL_PART


### print summary
### -----------------------------------------------------------
echo                     "--------------------------------------------------------------"
echo                     "  LiveCD installed on partition $INSTALL_PART"
[ "$SWAP_PART" ] && echo "  Partition $SWAP_PART will be used as swap partition"
echo
[ ! $NOGRUB ]    && echo "  GRUB installed in Master Boot Record (MBR) of $MBR_DEV."
[ ! $NOGRUB ]    && echo "  MBR saved as $MBR_FILENAME on $INSTALL_PART and on /tmp"
[ ! $NOGRUB ]    && echo "  If you have to restore MBR, execute under Linux:"
[ ! $NOGRUB ]    && echo "  # dd if=$MBR_FILENAME of=$MBR_DEV bs=512 count=1"
echo 
[ $WIN_PART ]    && echo "  Entry created in grub.conf for Windows partition $WIN_PART"
echo                     "--------------------------------------------------------------"
echo                     "End of $SCRIPTNAME"
echo