#!/bin/sh
# wget2nand
# This script can be used to download a TGZ file from your build system which
# contains the files to be installed on the NAND flash on your RB1xx card.
# The one parameter is the URL of the TGZ file to be downloaded.
# Licence GPL V2
# Author david.goodenough@linkchoose.co.uk
# Based on cf2nand from RB532 support
. /etc/functions.sh

[ -d /tmp/wget2nand ] && {
	echo "/tmp/wget2nand already exists"
	exit 1
}

# first get an address for br-lan using udhcpc
killall udhcpc
/sbin/udhcpc -i br-lan

# need to find the wget server from the command line
url=$1
[ -z "$url" ] && {
	echo "No URL specified for image TGZ"
	echo "Usage : $0 URL"
	exit 1
}

boot="$(find_mtd_part 'Routerboard NAND Boot')"
main="$(find_mtd_part 'rootfs')"
[ -z "$boot" -o -z "$main" ] && {
	echo "Cannot find NAND Flash partitions"
	exit 1
}

echo "Erasing filesystem."
mtd erase Boot 2>/dev/null >/dev/null
mtd erase Main 2>/dev/null >/dev/null

echo "Mounting $main as new root and $boot as boot partition"

mkdir /tmp/wget2nand/
mkdir /tmp/wget2nand-boot
mount -t yaffs2 "$main" /tmp/wget2nand/
mount -t yaffs2 "$boot" /tmp/wget2nand-boot

echo "Copying filesystem..."
( wget -O - $url/openwrt-rb532-rootfs.tgz) | ( cd /tmp/wget2nand/; tar xvz )
wget -O /tmp/wget2nand-boot/kernel $url/openwrt-rb532-vmlinux

# No need to patch the kernel, this was done during the build process
chmod +x /tmp/wget2nand-boot/kernel

# make sure everything is written before we unmount the partitions
echo "chmod ugo+x /" > /tmp/wget2nand/etc/uci-defaults/set_root_permission
sync
ls /tmp/wget2nand-boot/
ls /tmp/wget2nand/
# use kexec if present
[ -x /usr/bin/kexec ] && {
	kexec -l /tmp/wget2nand-boot/kernel --command-line="$(cat /proc/cmdline) root=$main rootfstype=yaffs2"
	kexec -e
}

# unmount the partitions and remove the directories into which they were mounted
umount /tmp/wget2nand-boot
umount /tmp/wget2nand
rmdir /tmp/wget2nand-boot
rmdir /tmp/wget2nand

# all done
echo "Image written, you can now reboot.  Remember to change the boot source to Boot from Nand"
