install-image.sh

#!/bin/sh

install_spl()
{
    echo "Installing SPL ..."
    flash_erase /dev/mtd0 0 0 2>/dev/null
    kobs-ng init -x $SPL_IMAGE --search_exponent=1 -v > /dev/null
}

install_uboot()
{
    echo "Installing u-boot ..."
    flash_erase /dev/mtd1  0 0  2>/dev/null
    nandwrite -p /dev/mtd1 $UBOOT_IMAGE
}

install_kernel()
{
    echo "Installing Kernel ..."
    flash_erase /dev/mtd2 0 0 2>/dev/null
    nandwrite -p /dev/mtd2 $KERNEL_IMAGE > /dev/null
}

install_devtree()
{
    echo "Installing device tree ..."
    nandwrite -p /dev/mtd1 -s 0x1e0000 $KERNEL_DTB > /dev/null
}

install_rootfs()
{
    echo "Installing rootfs ..."
    flash_erase /dev/mtd3 0 0 3>/dev/null
    ubiformat /dev/mtd3 -f $ROOTFS_IMAGE -s 2048 -O 2048
}

usage()
{
    echo "Usage: $0 -s SPL_FILE -u UBOOT_FILE -k KERNEL_FILE -d DEVTREE_FILE -r ROOTFS_FILE"
}

while getopts s:u:k:d:r: OPTION
do
    case $OPTION in
        s)
            SPL_IMAGE=$OPTARG
            ;;
        u)
            UBOOT_IMAGE=$OPTARG
            ;;
        k)
            KERNEL_IMAGE=$OPTARG
            ;;
        d)
            KERNEL_DTB=$OPTARG
            ;;
        r)
            ROOTFS_IMAGE=$OPTARG
            ;;
        ?)
        usage
    ;;
    esac
done

if test -z "${SPL_IMAGE}" -a    \
        -z "${UBOOT_IMAGE}" -a  \
        -z "${KERNEL_IMAGE}" -a \
        -z "${KERNEL_DTB}" -a   \
        -z "${ROOTFS_IMAGE}"; then
    usage
    exit 1
fi

if test -n "${SPL_IMAGE}"; then
    install_spl
fi

if test -n "${UBOOT_IMAGE}"; then
    install_uboot
fi

if test -n "${KERNEL_IMAGE}"; then
    install_kernel
fi

if test -n "${KERNEL_DTB}"; then
    install_devtree
fi

if test -n "${ROOTFS_IMAGE}"; then
    install_rootfs
fi