#! /bin/sh
# Script to create SD card for Freescale Layerscape LS1021atwr target.
#
# Author: Brijesh Singh, Texas Instruments Inc.
#       : Adapted for dra7xx-evm by Nikhil Devshatwar, Texas Instruments Inc.
#       : Adapted for FSL ls1201atwr by Arun Khandavalli, Mentor Graphics Inc.
#
# Licensed under terms of GPLv2
#

VERSION="0.1"

: ${MACHINE:="ls1021atwr"}
: ${ROOTFS_IMAGE:="console-image-${MACHINE}.tar.bz2"}
: ${sdkdir:="`pwd`/tmp/deploy/images/${MACHINE}"}
: ${dtb:="uImage-ls1021a-twr.dtb"}

execute ()
{
    $* >/dev/null
    if [ $? -ne 0 ]; then
        echo
        echo "ERROR: executing $*"
        echo
        exit 1
    fi
}

version ()
{
  echo
  echo "`basename $1` version $VERSION"
  echo "Script to create bootable SD card for ls1021atwr"
  echo

  exit 0
}

usage ()
{
  echo "
Usage: `basename $1` <options> [ files for install partition ]

Mandatory options:
  --device              SD block device node (e.g /dev/sdd)
  --sdk                 Where is sdk installed ?


Optional options:
  --version             Print version.
  --help                Print this help message.
  --rootfs		Which rootfs would you like to install ? [console-image (or) core-image-sato ]
"
  exit 1
}

check_if_main_drive ()
{
  mount | grep " on / type " > /dev/null
  if [ "$?" != "0" ]
  then
    echo "-- WARNING: not able to determine current filesystem device"
  else
    main_dev=`mount | grep " on / type " | awk '{print $1}'`
    echo "-- Main device is: $main_dev"
    echo $main_dev | grep "$device" > /dev/null
    [ "$?" = "0" ] && echo "++ ERROR: $device seems to be current main drive ++" && exit 1
  fi

}


# Check if the script was started as root or with sudo
user=`id -u`
[ "$user" != "0" ] && echo "++ Must be root/sudo ++" && exit

# Process command line...
while [ $# -gt 0 ]; do
  case $1 in
    --help | -h)
      usage $0
      ;;
    --device) shift; device=$1; shift; ;;
    --sdk) shift; sdkdir=$1; shift; ;;
    --rootfs ) shift; ROOTFS_IMAGE="$1-${MACHINE}.tar.gz"; shift; ;;
    --version) version $0;;
    *) copy="$copy $1"; shift; ;;
  esac
done

test -z $sdkdir && usage $0
test -z $device && usage $0
#test -z $ROOTFS_IMAGE && usage $0


if [ ! -d $sdkdir ]; then
   echo "ERROR: $sdkdir does not exist"
   exit 1;
fi

if [ ! -b $device ]; then
   echo "ERROR: $device is not a block device file"
   exit 1;
fi

check_if_main_drive

echo "************************************************************"
echo "*         THIS WILL DELETE ALL THE DATA ON $device        *"
echo "*                                                          *"
echo "*         WARNING! Make sure your computer does not go     *"
echo "*                  in to idle mode while this script is    *"
echo "*                  running. The script will complete,      *"
echo "*                  but your SD card may be corrupted.      *"
echo "*                                                          *"
echo "*         Press <ENTER> to confirm....                     *"
echo "************************************************************"
read junk

for i in `ls -1 $device?`; do
 echo "unmounting device '$i'"
 umount $i 2>/dev/null
done

execute "dd if=/dev/zero of=$device oflag=sync bs=1M count=1"

echo "copying SD-card u-boot for $machine"
execute "dd if=$sdkdir/u-boot-ls1-sd.bin of=$device bs=512 seek=8"

cat << END | fdisk $device
n
p
1

+62M
n
p
2


t
1
c
a
1
w
END

# handle various device names.
PARTITION1=${device}1

PARTITION2=${device}2

# make partitions.
echo "Formating ${device}1 ..."
if [ -b ${PARTITION1} ]; then
	mkfs.vfat -F 32 -n "boot" ${PARTITION1}
else
	echo "Cant find boot partition in /dev"
fi

if [ -b ${PARITION2} ]; then
	mkfs.ext4 -L "rootfs" ${PARTITION2}
else
	echo "Cant find rootfs partition in /dev"
fi

echo "Copying filesystem on ${device}1,${device}2"
execute "mkdir -p /tmp/sdk/$$/boot"
execute "mkdir -p /tmp/sdk/$$/rootfs"
execute "mount ${device}1 /tmp/sdk/$$/boot"
execute "mount ${device}2 /tmp/sdk/$$/rootfs"

if [ ! -f $sdkdir/${ROOTFS_IMAGE} ]
then
       ROOTFS_IMAGE=`echo $ROOTFS_IMAGE | sed "s:-${MACHINE}::"`
fi

if [ ! -f $sdkdir/${ROOTFS_IMAGE} ]; then
        echo "ERROR: failed to find rootfs [${ROOTFS_IMAGE}] tar in $sdkdir"
        ROOTFS_IMAGE=
        execute "umount /tmp/sdk/$$/boot"
        execute "umount /tmp/sdk/$$/root"
        exit 1
fi

echo "Extracting filesystem [${ROOTFS_IMAGE}] on ${device}2 ..."
root_fs=`ls -1 $sdkdir/${ROOTFS_IMAGE}`
execute "sudo tar -xvf $root_fs -C /tmp/sdk/$$/rootfs"
sync

execute "cp -f $sdkdir/u-boot-ls1-nor.bin /tmp/sdk/$$/boot/"
execute "cp -f $sdkdir/rcw/ls1021atwr/RSR_PPS_70/rcw_1000.bin /tmp/sdk/$$/boot/"
execute "cp -f $sdkdir/uImage /tmp/sdk/$$/boot/"
execute "cp -f $sdkdir/$dtb /tmp/sdk/$$/boot/"
execute "cp -f $sdkdir/boot/iram_Type_A_LS1021a_r1.0.bin   /tmp/sdk/$$/boot/"
sync

echo "unmounting ${device}1,${device}2"
execute "umount /tmp/sdk/$$/boot"
execute "umount /tmp/sdk/$$/rootfs"

execute "rm -rf /tmp/sdk/$$"
echo "completed!"