#!/initrd/bin/ash
#
RAMSIZE=1000000
trap "" 1 2 3 15

INSMOD="insmod"
RMMOD="rmmod"
LIBS="/initrd/lib/modules"

mountit(){
for fs in iso9660 ext2; do
test -b $1 && mount -t $fs $3 $1 $2 >/dev/null 2>&1 && return 0
done
return 1
}

FOUND_LIVECD=""
INTERACTIVE=""

# Clean input/output
exec >/dev/console </dev/console 2>&1

RESET="]R"
# Erase to end of line
CRE="
[K"
CLEAR="c"
NORMAL="[0;39m"
RED="[1;31m"
GREEN="[1;32m"
YELLOW="[1;33m"
BLUE="[1;34m"
MAGENTA="[1;35m"
CYAN="[1;36m"
WHITE="[1;37m"

# Be verbose
echo "${WHITE}Welcome to the ${CYAN}L${MAGENTA}I${YELLOW}V${WHITE}E${RED}C${BLUE}D${WHITE} live Linux-on-CD!${NORMAL}"
echo ""
echo ""

test -f /proc/splash && echo "show 1310" > /proc/splash

PATH=/bin:/sbin:/usr/bin:/usr/sbin:/initrd/bin:/livecd/initrd/bin
export PATH

umask 022

mount -t proc /proc /proc
mount -t sysfs none /sys >/dev/null 2>&1

echo "0" > /proc/sys/kernel/printk

CMDLINE="$(cat /proc/cmdline)"

# Check if we are in interactive startup mode
case "$CMDLINE" in *expert*) INTERACTIVE="yes"; ;; esac
case "$CMDLINE" in *modules-disk*) INTERACTIVE="yes"; ;; esac
# Does the user want to skip scsi detection?
NOSCSI=""
case "$CMDLINE" in *noscsi*) NOSCSI="yes"; ;; esac
case "$CMDLINE" in *nousb*) NOUSB="yes"; ;; esac
case "$CMDLINE" in *nofirewire*) NOFIREWIRE="yes"; ;; esac
NOCD=""
case "$CMDLINE" in *fromhd*) NOCD="yes"; ;; esac
case "$CMDLINE" in *idecd*|*atapicd*) IDECD="yes"; ;; esac

LIVECD_DIR="LIVECD"
LIVECD_NAME="base.sqfs"
USR_NAME="usr.sqfs"
case "$CMDLINE" in *livecd_dir=*) LIVECD_DIR="$livecd_dir"; ;; esac
case "$CMDLINE" in *livecd_name=*) LIVECD_NAME="$livecd_name"; ;; esac

DEBUG=""
case "$CMDLINE" in *debug*) echo "6" > /proc/sys/kernel/printk;
DEBUG="yes"; ;; esac

# LIVECD Kernel conig for SCSI and SCSI CDROM is set to M instead of Y on Knoppix.
$INSMOD $LIBS/scsi/scsi_mod.ko >/dev/null 2>&1
$INSMOD $LIBS/scsi/sd_mod.ko >/dev/null 2>&1
$INSMOD $LIBS/scsi/libata.ko >/dev/null 2>&1
$INSMOD $LIBS/scsi/ata_piix.ko >/dev/null 2>&1
$INSMOD $LIBS/scsi/sr_mod.ko >/dev/null 2>&1

# Check for IDE-SCSI capable CD-Rom(s)
test -n "$IDECD" || $INSMOD $LIBS/scsi/ide-scsi.ko >/dev/null 2>&1

stage=1
rundebugshell(){
if [ -n "$DEBUG" ]; then
echo "${CRE}${BLUE}Starting intermediate Shell stage $stage as requested by \"debug\" option.${NORMAL}"
echo "${CRE}${BLUE}Type \"exit\" to continue with normal bootup.${NORMAL}"
[ -x /initrd/bin/ash ] && /initrd/bin/ash || /bin/bash
fi
}

mountmodules(){
TYPE="$1"; shift
echo -n "${CRE}${CYAN}Please insert ${TYPE} modules disk and hit Return. ${NORMAL}"
read a
echo -n "${CRE}${BLUE}Mounting ${TYPE} modules disk... ${NORMAL}"
# We always mount over $LIBS/scsi (because it's there ;-)
if mountit /dev/fd0 $LIBS/scsi "-o ro"; then
echo "${GREEN}OK.${NORMAL}"
return 0
fi
echo "${RED}NOT FOUND.${NORMAL}"
return 1
}

umountmodules(){
TYPE="$1"; shift
echo -n "${CRE}${BLUE}Unmounting ${TYPE} modules disk... ${NORMAL}"
umount $LIBS/scsi 2>/dev/null
echo "${GREEN}DONE.${NORMAL}"
}

askmodules(){
TYPE="$1"; shift
echo "${BLUE}${TYPE} modules available:${WHITE}"
c=""; for m in "$@"; do
if test -f "$LIBS/scsi/$m"; then
test -z "$c"  && { echo -n "    $m"; c="1"; } || { echo "               $m"; c=""; }
fi
done
[ -n "$c" ] && echo ""
echo "${CYAN}Load ${TYPE} Modules?${NORMAL}"
echo "${CYAN}[Enter full filename(s) (space-separated), Return for autoprobe, ${WHITE}n${CYAN} for none] ${NORMAL}"
echo -n "${CYAN}$INSMOD module(s)> ${NORMAL}"
read MODULES
case "$MODULES" in n|N) MODULES=""; ;; y|"")  MODULES="$*"; ;; esac
}

# Try to load the given modules (full path or current directory)
loadmodules(){
TYPE="$1"; shift
if test -z "$DEBUG"; then
test -n "$INTERACTIVE" && echo "6" > /proc/sys/kernel/printk
fi
for i in "$@"; do
echo -n "${CRE}${BLUE}Probing ${TYPE}... ${MAGENTA}$i${NORMAL}"
if test -f $LIBS/scsi/$i && $INSMOD -f $LIBS/scsi/$i >/dev/null 2>&1
then
case "$i" in *ataraid*) ;; *) echo "${CRE} ${GREEN}Found ${TYPE} device(s) handled by ${MAGENTA}$i${GREEN}.${NORMAL}" ;; esac
fi
done
if test -z "$DEBUG"; then
test -n "$INTERACTIVE" && echo "0" > /proc/sys/kernel/printk
fi
echo -n "${CRE}"
}

# Check for SCSI, use modules on bootfloppy first
# these are the autoprobe-safe modules
SCSI_PROBE=""
if test -z "$NOSCSI"; then
PROCPCI="`cat /proc/pci 2>/dev/null`"
ISA_SCSI="aha1740.ko aha1542.ko aha152x.ko pas16.ko psi240i.ko qlogicfas.ko qlogicfc.ko seagate.ko t128.ko u14-34f.ko wd7000.ko"
SCSI_PROBE="$ISA_SCSI"
case "$PROCPCI" in *[Aa][Ii][Cc]-*|*[Aa][Hh][Aa]-*) SCSI_PROBE="$SCSI_PROBE aic7xxx.ko" ;; esac
case "$PROCPCI" in *[Bb][Uu][Ss][Ll][Oo][Gg][Ii][Cc]*) SCSI_PROBE="$SCSI_PROBE BusLogic.ko" ;; esac
case "$PROCPCI" in *53[Cc]406*) SCSI_PROBE="$SCSI_PROBE NCR53c406a.ko" ;; esac
case "$PROCPCI" in *[Ii][Nn][Ii][Tt][Ii][Oo]\ *|*[Ii][Nn][Ii]-[Aa]100[Uu]2[Ww]*) SCSI_PROBE="$SCSI_PROBE initio.ko" ;; esac
case "$PROCPCI" in *[Mm][Pp][Tt]*[Ss][Cc][Ss][Ii]*) SCSI_PROBE="$SCSI_PROBE mptscsih.ko" ;; esac
case "$PROCPCI" in *[Aa][Dd][Vv][Aa][Nn][Cc][Ee][Dd]\ [Ss][Yy][Ss]*) SCSI_PROBE="$SCSI_PROBE advansys.ko" ;; esac
case "$PROCPCI" in *[Aa][Tt][Pp]8|*[Aa][Ee][Cc]6*) SCSI_PROBE="$SCSI_PROBE atp870u.ko" ;; esac
case "$PROCPCI" in *[Dd][Tt][Cc]*) SCSI_PROBE="$SCSI_PROBE dtc.ko" ;; esac
case "$PROCPCI" in *[Ee][Aa][Tt][Aa]*) SCSI_PROBE="$SCSI_PROBE eata.ko" ;; esac
case "$PROCPCI" in *[Ff]*[Dd][Oo][Mm][Aa][Ii][Nn]*) SCSI_PROBE="$SCSI_PROBE fdomain.ko" ;; esac
case "$PROCPCI" in *[Gg][Dd][Tt]\ *) SCSI_PROBE="$SCSI_PROBE gdth.ko" ;; esac
case "$PROCPCI" in *[Mm][Ee][Gg][Aa][Rr][Aa][Ii][Dd]*) SCSI_PROBE="$SCSI_PROBE megaraid.ko" ;; esac
case "$PROCPCI" in *[Pp][Cc][Ii]-22*) SCSI_PROBE="$SCSI_PROBE pci2220i.ko" ;; esac
case "$PROCPCI" in *[Pp][Cc][Ii]-2000*) SCSI_PROBE="$SCSI_PROBE pci2000.ko" ;; esac
case "$PROCPCI" in *[Qq][Ll][Oo][Gg][Ii][Cc]*) SCSI_PROBE="$SCSI_PROBE qlogicisp.ko" ;; esac
case "$PROCPCI" in *53[Cc]974*) SCSI_PROBE="$SCSI_PROBE tmscsim.ko" ;; esac
case "$PROCPCI" in *[Uu][Ll][Tt][Rr][Aa][Ss][Tt][Oo][Rr]*) SCSI_PROBE="$SCSI_PROBE ultrastor.ko" ;; esac
case "$PROCPCI" in *3[Ww][Aa][Rr][Ee]*) SCSI_PROBE="$SCSI_PROBE 3w-xxxx.ko" ;; esac
fi

if test -n "$INTERACTIVE"; then
# Let the user select interactively
askmodules SCSI $(cd $LIBS/scsi; echo *.ko)
else
# these are the autoprobe-safe modules
MODULES="$SCSI_PROBE"
fi
test -z "$NOSCSI" && test -n "$MODULES" && loadmodules SCSI $MODULES
# End of SCSI check

if test -z "$NOUSB"; then
echo -n "${CRE}${BLUE}Checking for USB...${NORMAL}"
FOUNDUSB=""
for i in ohci-hcd.ko uhci-hcd.ko ehci-hcd.ko; do
$INSMOD $LIBS/scsi/$i >/dev/null 2>&1 && FOUNDUSB="yes"
done
if test -n "$FOUNDUSB"; then
$INSMOD $LIBS/scsi/usb-storage.ko >/dev/null 2>&1
else
true
fi
echo -n "${CRE}"
fi
# End of USB check

# Check for Firewire, use modules on bootfloppy first
if test -z "$NOFIREWIRE"; then
echo -n "${CRE}${BLUE}Checking for Firewire...${NORMAL}"
echo -n "${CRE}${BLUE}Loading ieee1394...${NORMAL}"
$INSMOD $LIBS/scsi/ieee1394.ko >/dev/null 2>&1
FOUNDFIREWIRE=""
echo -n "${CRE}${BLUE}Loading ohci1394.ko...${NORMAL}"
$INSMOD $LIBS/scsi/ohci1394.ko >/dev/null 2>&1 && FOUNDFIREWIRE="yes"
if test -n "$FOUNDFIREWIRE"; then
echo -n "${CRE}${BLUE}Loading sbp2.ko...${NORMAL}"
$INSMOD $LIBS/scsi/sbp2.ko sbp2_serialize_io=1 >/dev/null 2>&1
else
true
fi
echo -n "${CRE}"
fi
# End of FIREWIRE check

# Unfortunately, hotpluggable devices tend to need some time in order to register
if test -n "$FOUNDUSB" -o -n "$FOUNDFIREWIRE"; then
echo -n "${CRE}${BLUE}Scanning for USB/Firewire devices... ${NORMAL}"
if test -n "$FOUNDFIREWIRE"; then
sleep 2
fi
echo "${BLUE}Done.${NORMAL}"
fi

# Check for misc modules in expert mode
if test -n "$INTERACTIVE"; then
another=""; answer=""
while test "$answer" != "n" -a "$answer" != "N"; do
echo -n "${CYAN}Do you want to load additional modules from$another floppy disk? [${WHITE}Y${CYAN}/n] ${NORMAL}"
another=" another"
read answer
case "$answer" in n*|N*) break; ;; esac
if mountmodules new; then
askmodules new $(cd $LIBS/scsi; echo *.ko)
test -n "$MODULES" && loadmodules new $MODULES
umountmodules current
fi
done
fi
# All interactively requested modules should be loaded now.

# Disable kernel messages again
if test -z "$DEBUG"; then
echo "0" > /proc/sys/kernel/printk
fi

ls -A1d /sys/block/*/dev /sys/block/*/*/dev 2>/dev/null | grep -v loop | while read BLOCK; do
      DEVICE="/dev/`basename \`dirname $BLOCK\``"
      if [ ! -b $DEVICE ]; then
         MINORMAJOR="`head -n 1 $BLOCK | tr ':' ' '`"
         mknod $DEVICE b $MINORMAJOR
      fi
done

# We now enable DMA right here, for faster reading/writing from/to IDE devices
# in FROMHD or TORAM mode
case "$CMDLINE" in *\ nodma*) ;; *)
for d in $(cd /proc/ide 2>/dev/null && echo hd[a-z]); do
if test -d /proc/ide/$d; then
MODEL="$(cat /proc/ide/$d/model 2>/dev/null)"
test -z "$MODEL" && MODEL="[GENERIC IDE DEVICE]"
echo "${BLUE}Enabling DMA acceleration for: ${MAGENTA}$d 	${YELLOW}[${MODEL}]${NORMAL}"
echo "using_dma:1" >/proc/ide/$d/settings
fi
done
;;
esac

test -f /proc/splash && echo "show 3932" > /proc/splash

stage=1
rundebugshell

DEVICES=""
MOUNTPOINT=""
if test -z "$NOCD"; then
 CDROM_WITH_BLANK=`cat /proc/sys/dev/cdrom/info | head -n 3 | tail -n 1 | cut -d ":" -f 2`;
   CDROM_TYPE=`cat /proc/sys/dev/cdrom/info | grep CD-RW | cut -d ":" -f 2`;
   if [ $CDROM_TYPE == "1" ]; then CDTYPE="cdrecorder"
   else CDTYPE="cdrom"
   fi
   mkdir -p /media/$CDTYPE
     for CDROM in $CDROM_WITH_BLANK; do
        if mount -t iso9660 -o ro /dev/$CDROM /media/$CDTYPE; then
            # Found a CD-ROM drive containing a CD, check it's the Livecd BootCD
            if [ -f /media/$CDTYPE/$LIVECD_DIR/$LIVECD_NAME ]; then
                echo "${CRE}${BLUE}LIVECD is mounted on /dev/$CDROM.${NORMAL}"
		MOUNTPOINT="/media/$CDTYPE"
                ln -s /dev/$CDROM /dev/cdrom
                echo "/dev/$CDROM /media/$CDTYPE auto pamconsole,exec,noauto 0 0" >> /etc/fstab
                echo "/dev/$CDROM /media/$CDTYPE iso9660 ro,nosuid,nodev 0 0" >> /etc/mtab
            else
                umount /media/$CDTYPE
            fi
        fi
      done
        if test -z "$MOUNTPOINT"; then
	    echo "${CRE}${RED}ERROR: Can't find LIVECD${NORMAL}"
	    exec /initrd/bin/ash
	fi
else
  DEVICES="$fromhd"
  PARTITION="${DEVICES##/dev/}"
	mkdir -p /mnt/$PARTITION
	FSTYPE="ext2"
           if mount -t $FSTYPE -o ro $DEVICES /mnt/$PARTITION; then
            if [ -f /mnt/$PARTITION/$LIVECD_DIR/$LIVECD_NAME ]; then
                echo "${CRE}${BLUE}LIVECD is mounted on /mnt/$PARTITION${NORMAL}"
		MOUNTPOINT="/mnt/$PARTITION"
                    echo "$DEVICES /mnt/$PARTITION $FSTYPE noauto 0 0" >> /etc/fstab
                    MOUNTED=TRUE
                    break
            else
                umount /mnt/$PARTITION
	    	echo "${CRE}${RED}ERROR: There is no base.sqfs${NORMAL}"
	    	exec /initrd/bin/ash
            fi
	   else
	    echo "${CRE}${RED}ERROR: Can't mount $DEVICE${NORMAL}"
	    exec /initrd/bin/ash
           fi
fi

if test -n "$MOUNTPOINT"; then
    $INSMOD $LIBS/loop.ko 
    $INSMOD $LIBS/unionfs.ko 2>/dev/null
    mkdir -p /mnt/livecd /tmp/livecd_cow
    mount -t squashfs -o loop,ro $MOUNTPOINT/$LIVECD_DIR/$LIVECD_NAME /mnt/livecd
    mount -t unionfs -o dirs=/tmp/livecd_cow:/mnt/livecd none /livecd

FOUNDMEM="$(awk '/MemTotal/{print $2}' /proc/meminfo)"
TOTALMEM="$(awk 'BEGIN{m=0};/MemFree|Cached/{m+=$2};END{print m}' /proc/meminfo)"
echo "${CRE}${BLUE}Total memory found: ${YELLOW}${FOUNDMEM}${BLUE} kB${NORMAL}"

MINSIZE=20000
MINLEFT=16000
MAXSIZE="$(expr $TOTALMEM - $MINLEFT)"
RAMSIZE="$(expr $TOTALMEM / 5)"

test -z "$RAMSIZE" -o "$RAMSIZE" -lt "$MINSIZE" && RAMSIZE="$MINSIZE"

# Harddisk-installed script part version has been removed
# (LIVECD can be booted directly from HD now).

RAMSIZE=$(expr $RAMSIZE \* 4)
echo "${CRE}${BLUE}Creating ${YELLOW}/ramdisk${BLUE} (dynamic size=${RAMSIZE}k) on ${MAGENTA}shared memory${BLUE}...${NORMAL}"

mount -t tmpfs -o "size=${RAMSIZE}k" none /var

test -f /proc/splash && echo "show 6553" > /proc/splash

cp -af /livecd/var/* /var
rm -Rf /var/root/.[A-z]*
cp -a /livecd/var/etc/skel/.[a-z]* /var/root/
cp -af /initrd/etc/inittab /initrd/etc/fstab /initrd/etc/filesystems /initrd/etc/mtab /var/etc/
cp -af /initrd/etc/rc.d/booyo-init /initrd/etc/rc.d/booyo-halt /initrd/etc/rc.d/booyo-reboot /initrd/etc/rc.d/booyo-services /var/etc/rc.d/

umount /livecd
umount /mnt/livecd

rm -f etc; rm -rf boot root tmp
ln -s var/boot var/etc var/home var/root var/tmp /

$INSMOD $LIBS/unionfs.ko 2>/dev/null
mkdir -p /mnt/livecd /tmp/livecd_cow
mount -t squashfs -o loop,ro $MOUNTPOINT/$LIVECD_DIR/$LIVECD_NAME /mnt/livecd
mount -t unionfs -o dirs=/tmp/livecd_cow:/mnt/livecd none /livecd

FOUND_USR=""
if test -f $MOUNTPOINT/$LIVECD_DIR/$USR_NAME; then
    FOUND_USR="yes"
    busybox rm -f /usr
    busybox mkdir /usr
    busybox mkdir -p /mnt/usr /tmp/usr_cow
    /initrd/bin/mount -t squashfs -o loop,ro $MOUNTPOINT/$LIVECD_DIR/$USR_NAME /mnt/usr
    /initrd/bin/mount -t unionfs -o dirs=/tmp/usr_cow:/mnt/usr none /usr
    echo "Mounting usr.sqfs on /usr"
fi

busybox rm -f bin lib sbin opt #remove links
busybox ln -s livecd/bin livecd/lib livecd/sbin livecd/opt /


chmod 755 /var

test -f /proc/splash && echo "show 9174" > /proc/splash

stage=2
rundebugshell

cat /initrd/etc/fstab > /etc/fstab

ln -s /dev/ram0 /dev/root
echo "0x100" > /proc/sys/kernel/real-root-dev

echo "${CRE}${BLUE}Starting init process.${NORMAL}"
exec /initrd/sbin/init "$@" < /dev/console > /dev/console 2>&1
else
echo "failed"
exec /initrd/bin/ash
fi
