#!/bin/sh 
#
# This is a main init script for Z
# 
# (C) 2005 Samsung Electronics
#
# $Log: rc,v $
# Revision 1.12  2006/05/30 07:59:41  hcyun
# add dosfsck to correct filesystem corruption due to power off during metadata update
#
# Revision 1.11  2006/04/19 07:26:43  hcyun
# - set hidden on System/System.ver
#
# Revision 1.10  2006/03/23 14:05:54  hcyun
# - support seperate MLC/SLC kernel image.
# - support upgrade from phase1 + SLC
# - add fixup script execution for future upgrade handling
# - handle 7.5MB /ufdrawc limit due to 4GB with pre 2.0 partition table bug
#   now Oasis image must be named as "Oasis-big.cramfs"
# - firmware version update without modifying country code of /mnt/app/System.ver
#
# Revision 1.9  2006/03/06 10:43:44  biglow
# - add detecting s/w version
#
# Revision 1.8  2006/02/22 12:43:45  hcyun
# reset db after the reformatting due to FAT corruption.
#
# Revision 1.7  2006/02/13 10:02:54  hcyun
# handle mount error
#
# - hcyun
#
# Revision 1.6  2006/01/16 13:22:48  hcyun
# retry to write cramfs
#
# Revision 1.5  2006/01/15 08:20:54  hcyun
# handle install fail case
#
# Revision 1.4  2005/12/29 11:43:28  hcyun
# leave System.ver untouched in /mnt/usb
#
# Revision 1.3  2005/12/26 06:41:34  hcyun
# safe System.ver update.
#
# Revision 1.2  2005/12/22 12:37:09  hcyun
# add hostname
#
# Revision 1.1  2005/12/22 05:50:18  hcyun
# Moved from images directory.
#

err_msg()
{
    echo "ERROR $*"
    exit 1 
}

err_install() 
{
    echo "ERROR $*"

    killall -9 install-scr.sh

    rm -f z5-fw.dat fw.tar.gz 
    rm -f Oasis.cramfs Oasis.tar.gz Media.tar.gz System.ver 
    rm -f zImage Image rootfs.cramfs nandboot.sb 
    sync

    # TODO. display firmware upgrade failure. 

    /z-proj/bin/recovery  /share/install-fail.fb

    echo ">> Reset the system" 
    reboot
    echo "main 0" > /proc/powerctl 
}

handle_noformat_ufda()
{
    echo "mount failed"
    echo "Unmounting previous mounts" 
    umount /mnt/app 

    echo "Now re-format the /dev/ufda" 
    ffdisk /dev/ufda -a

    fformat -f12 -c 16 /dev/ufda1 || \
        fformat -f12 -c 32 /dev/ufda1 || \
	    fformat -f16 -c 8 /dev/ufda1 || err_msg "Can't format /dev/ufda1" 

    mount /mnt/app 

    if mount -t cramfs /dev/ufdrawc /Oasis; then 
	if [ -f "/Oasis/OasisData.tar.gz" ]; then 
	    echo "copy OasisData"
	    cat /Oasis/OasisData.tar.gz | ( cd /mnt/app ; tar zxvf - )
	fi
	umount /Oasis 
    fi
}

handle_noformat_ufdb()
{
    echo "mount failed"
    echo "Unmounting previous mounts" 
    umount /mnt/usb 

    echo "Now re-format the /dev/ufdb" 
    ffdisk /dev/ufdb -a

    fformat -f32 -c 32 /dev/ufdb1 || \
	fformat -f16 -c 32 /dev/ufdb1 || err_msg "Can't format /dev/ufdb1" 

    mount /mnt/usb 

    if mount -t cramfs /dev/ufdrawc /Oasis; then 
	echo "Copy wallpapers" 
	(cd /Oasis/usr/Oasis/DataTemplate; tar cf - . ) | ( cd /mnt/usb; tar xvf - )
	(cd /Oasis/usr/Oasis; bin/resetdb.sh )
	umount /Oasis 
    fi
}

# procedure start downloading 
handle_nofw()
{
    echo "[handle_nofw]: Unable to find a firmware"

    umount /mnt/app
    umount /mnt/usb 

    # /dev/ufda2 must be formatted. 
    # TODO : error handling if /dev/ufda1 is not formatted" 

    echo ">> Now ready to connect UMS"
    echo "usbcontrol 1" > /proc/gadget_udc
    insmod g_file_storage file=/dev/ufdb

    echo ">> download firmware to the UMS" 
    echo "after downlod is finished.. !!reboot the target!!"
    echo "it will install new firmware automatically"

    echo 
    echo ">> Start recovery" 
    /z-proj/bin/recovery  /share/recovery.fb

    echo ">> Unloading UMS driver" 
    rmmod g_file_storage 
    sync 

    echo ">> Reset the system" 
    echo "main 0" > /proc/powerctl 
}

handle_cramfs()
{
    image=$1
    dev=$2

    echo "Error on installing cramfs $image on $dev. Try again"
    dd if=$image of=$dev bs=2048  || err_msg "Fail again -_-"
    sync
}


# this is fix for 4GB SLC, phase 1. 
fixup_oasis()
{
    echo "[fixup_oasis]: Found an uninstalled Oasis cramfs image"

    cd /mnt/usb 

    echo "Install UI application (bigger than 7.5MB)"
    dd if=Oasis-big.cramfs of=/dev/ufdrawc bs=2048 || handle_cramfs Oasis-big.cramfs /dev/ufdrawc 
    rm -f Oasis-big.cramfs
    sync
}

# temporary procedure to install correct kernel in case of installed 
# rc script do not distinguish MLC/SLC installation. 
# when this is kicked in, we can assume new rc script is running.
# !! IMPORTANT !! require reboot
fixup_kernel()
{
    echo "[fixup_kernel]: Found an uninstalled kernel image" 

    cd /mnt/usb 

    # TODO : display temporary kernel update notice 
    echo "refresh 0" > /proc/lfbctrl
    cat /share/install.fb > /dev/fb0

    if [ ! -f "/proc/z5-hw" ]; then 
	# this must be a phase1 firmware with SLC attached. 
	echo "phase1 firmware with SLC" 
	dd if=Image-SLC of=/dev/ufdrawa bs=2048 || handle_cramfs Image-SLC /dev/ufdrawa 
    else
	echo "[ERROR] Not a phase1 with SLC?" 
	cat /share/install-fail.fb > /dev/fb0 
    fi

    echo "Removing fixup kernel images"
    rm -f Image-SLC Image-MLC

    sync

    echo "Reboot the system" 
    reboot 
}

# temporary procedure for initial product without fw version update capability 
update_fwver()
{
    echo "[update_fwver]: Found firmware update"

    # TODO : 
    update_version `cat /mnt/usb/z5-fw.ver` /mnt/app/System.ver 
    [ ! -d "/mnt/usb/System" ] && mkdir -p /mnt/usb/System 
    cp /mnt/app/System.ver /mnt/usb/System/System.ver
    chmod 1555 /mnt/usb/System
    chmod 1555 /mnt/usb/System/System.ver 
    rm -f /mnt/usb/z5-fw.ver 
    rm -f /mnt/usb/System.ver 
    sync 
}


# procedure to install new firmware image.. 
install_newfw()
{
    echo "[install_newfw]: Found a new firmware " 

    cd /mnt/usb 

    echo "refresh 0" > /proc/lfbctrl
    /z-proj/bin/install-scr.sh &  # spinning update screen. 

    echo ">> Start firmware installation" 

    echo "Decrypting firmware"
    if ! z5decrypt z5-fw.dat fw.tar.gz z5-fw.ver; then 
	err_install "Decryption failed"
    fi 

    echo "Decompressing firmware" 
    if ! tar zxvf fw.tar.gz ; then 
	err_install "Corrupted firmware" 
    fi

    echo "Removing decrypted firmware Image" 
    rm -f fw.tar.gz


    if [ -f "z5_fixup.sh" ]; then 
	echo "Found a fixup script... run it first"
	sh z5-fixup.sh 
	rm -f z5-fixup.sh 
	sync
    fi 

    if [ ! -f "/mnt/app/System.ver" ]; then 
	echo "Initial System.ver installation to /mnt/app" 
	mv System.ver /mnt/app 
	sync
    fi 

    if [ -f "Media.tar.gz" ]; then 
	echo "Installing Media Files" 
	cat Media.tar.gz | (cd /mnt/usb; tar zxvf - )
	rm -f Media.tar.gz 
	sync
    fi 

    if [ -f "Oasis.cramfs" ]; then 
	echo "Install UI application" 
	dd if=Oasis.cramfs of=/dev/ufdrawc bs=2048 || handle_cramfs Oasis.cramfs /dev/ufdrawc 
	rm -f Oasis.cramfs
	sync
    fi 

    if [ -f "Oasis-big.cramfs" ]; then 
	echo "Install UI application (bigger than 7.5MB)" 
	dd if=Oasis-big.cramfs of=/dev/ufdrawc bs=2048 || handle_cramfs Oasis-big.cramfs /dev/ufdrawc 
	rm -f Oasis-big.cramfs
	sync
    fi 

    if [ -f "Oasis.tar.gz" ]; then 
	echo "Installing OasisData (or Devel version of Oasis)" 
	if [ -d "/mnt/app/usr/Oasis" ]; then 
	    echo "Removing previously installed Oasis" 
	    rm -rf /mnt/app/usr/Oasis 
	    rm -f /mnt/app/start.sh
	fi 
	cat Oasis.tar.gz | (cd /mnt/app; tar zxvf - )
	rm -f Oasis.tar.gz 
	sync
    fi 

    if [ -f "zImage" ]; then 
        echo "Installing zImage" 
	dd if=zImage of=/dev/ufdrawa bs=2048 || handle_cramfs zImage /dev/ufdrawa 
	rm -f zImage 
	sync
    fi 

    if [ -f "Image" ]; then 
        echo "Installing Image" 
	dd if=Image of=/dev/ufdrawa bs=2048 || handle_cramfs Image /dev/ufdrawa 
	rm -f Image 
	sync
    fi 

    if [ -f "Image-SLC" -o -f "Image-MLC" ]; then
	if grep "SLC" /proc/z5-hw ; then 
	    echo "SLC flash is connected" 
	    dd if=Image-SLC of=/dev/ufdrawa bs=2048 || handle_cramfs Image-SLC /dev/ufdrawa 
	elif grep "MLC" /proc/z5-hw ; then 
	    echo "MLC flash is connected" 
	    dd if=Image-MLC of=/dev/ufdrawa bs=2048 || handle_cramfs Image-MLC /dev/ufdrawa 
	else 
	    echo "[ERROR] Incorrect z5-hw info.."
	fi
	echo "Removing fixup kernel images" 
	rm -f Image-SLC Image-MLC
	sync
    fi

    if [ -f "nandboot.sb" ]; then 
	echo "Installing bootloader"
	nsect=`ls -l nandboot.sb | awk '{ print $5 }'` 
	echo "bootloader num sectors : $nsect"
	dd if=nandboot.sb of=/dev/misc/sram 
	echo "$nsect" > /proc/lld/nandsb
	rm -f nandboot.sb 
	sync 
    fi

    if [ -f "rootfs.cramfs" ]; then 
	echo "Install cramfs" 
	dd if=rootfs.cramfs of=/dev/ufdrawb bs=2048 || handle_cramfs rootfs.cramfs /dev/ufdrawb
	rm -f rootfs.cramfs
    fi 

    echo "Removing encrypted firmware Image" 
    rm -f z5-fw.dat
    sync

    echo ">>> Firmware Installation was successful" 
    
    echo ">>> Reset the system"
    /sbin/reboot
    echo "reboot is not working.. force power off" 
    echo "main 0" > /proc/powerctl # should not reach here
}

export PATH=$PATH:/z-proj/bin

mount -t proc none /proc
mount -t tmpfs none /mnt/tmp

/z-proj/bin/dosfsck -a /dev/ufda1 

mount /mnt/app \
    || handle_noformat_ufda \
    && echo "mount /dev/ufda1 is successful"
mount /mnt/usb \
    || handle_noformat_ufdb \
    && echo "mount /dev/ufdb1 is successful"

[ -f "/mnt/usb/z5-fw.dat" ] && install_newfw

# Ugly fix for initial product (SLC with pre 2.0 firmware ) - require reboot 
[ -f "/mnt/usb/Image-SLC" ] && fixup_kernel

# Ugly for for initial product (4GB SLC with pre 2.0 firmware) 
[ -f "/mnt/usb/Oasis-big.cramfs" ] && fixup_oasis

[ -f "/mnt/usb/z5-fw.ver" ] && update_fwver 

# read S/W version information
cat /mnt/app/System.ver > /proc/z5-ver

# RFS Oasis (only for development) 
if [ -f "/mnt/app/start.sh" ]; then 
    echo "[DEVEL] RFS /dev/ufda1 contain oasis distribution"
    echo "[DEVEL] Starting main UI" 
    /mnt/app/start.sh 
    exit
fi 

# CRAMFS Oasis (for production)
mount -t cramfs /dev/ufdrawc /Oasis \
    || handle_nofw
[ -f "/Oasis/start.sh" ] \
    || handle_nofw

echo "Start main UI" 
/Oasis/start.sh

hostname "Z5"