#
# U-Boot script for installing Linux images created by Yocto from the micro SD
# card into the NAND
#

# Reset temp variables
install_abort=0
BASEFILENAME=0

# Determine U-Boot file to program basing on module variant
if test -n "${module_variant}"; then
	if test "${module_variant}" = "0x02" || test "${module_variant}" = "0x03" || test "${module_variant}" = "0x06" || test "${module_variant}" = "0x0a"; then
		setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##.imx;
	elif test "${module_variant}" = "0x04" || test "${module_variant}" = "0x05" || test "${module_variant}" = "0x07"; then
		setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##1GB.imx;
	elif test "${module_variant}" = "0x08" || test "${module_variant}" = "0x09"; then
		setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##512MB.imx
	fi
fi

# Use 'test -n ...' because 'test -z ...' does not work well on old versions of
# u-boot when the checked value is empty.
if test -n "${INSTALL_UBOOT_FILENAME}"; then
	true;
else
	echo "";
	echo "[ERROR] Cannot determine U-Boot file for this module!";
	echo "";
	echo "1. Set variable 'INSTALL_UBOOT_FILENAME' depending on your ConnectCore 6UL variant:";
	echo "   - For a SOM with 1GB DDR3, run:";
	echo "     => setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##1GB.imx";
	echo "   - For a SOM with 512MB DDR3, run:";
	echo "     => setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##512MB.imx";
	echo "   - For a SOM with 256MB DDR3, run:";
	echo "     => setenv INSTALL_UBOOT_FILENAME u-boot-##MACHINE##.imx";
	echo "";
	echo "2. Run the install script again.";
	echo "";
	echo "Aborted";
	echo "";
	exit;
fi;

setenv INSTALL_MMCDEV 0

if test -z "${image-name}"; then
	setenv image-name ##DEFAULT_IMAGE_NAME##
fi
GRAPHICAL_IMAGES="##GRAPHICAL_IMAGES##"
for g in ${GRAPHICAL_IMAGES}; do
	if test "${image-name}" = "${g}"; then
		BASEFILENAME="${image-name}-##GRAPHICAL_BACKEND##"
	fi
done
if test "${BASEFILENAME}" = "0"; then
	BASEFILENAME="${image-name}"
fi
setenv INSTALL_LINUX_FILENAME ${BASEFILENAME}-##MACHINE##.boot.ubifs
setenv INSTALL_RECOVERY_FILENAME ${BASEFILENAME}-##MACHINE##.recovery.ubifs
setenv INSTALL_ROOTFS_FILENAME ${BASEFILENAME}-##MACHINE##.ubifs

# Check for presence of firmware files on the SD card
for install_f in ${INSTALL_UBOOT_FILENAME} ${INSTALL_LINUX_FILENAME} ${INSTALL_RECOVERY_FILENAME} ${INSTALL_ROOTFS_FILENAME}; do
	if test ! -e mmc ${INSTALL_MMCDEV} ${install_f}; then
		echo "ERROR: Could not find file ${install_f}";
		install_abort=1;
	fi;
done
if test "${install_abort}" = "1"; then
	echo "Aborted.";
	exit;
fi

echo "############################################################"
echo "#           Linux firmware install from micro SD           #"
echo "############################################################"
echo ""
echo " This process will erase your NAND and will install the following files"
echo " on the partitions of the NAND."
echo ""
echo "   PARTITION   FILENAME"
echo "   ---------   --------"
echo "   bootloader  ${INSTALL_UBOOT_FILENAME}"
if test \"\$\{dualboot\}\" = yes; then
	echo "   linux_a     ${INSTALL_LINUX_FILENAME}"
	echo "   linux_b     ${INSTALL_LINUX_FILENAME}"
	echo "   rootfs_a    ${INSTALL_ROOTFS_FILENAME}"
	echo "   rootfs_b    ${INSTALL_ROOTFS_FILENAME}"
else
	echo "   linux       ${INSTALL_LINUX_FILENAME}"
	echo "   recovery    ${INSTALL_RECOVERY_FILENAME}"
	echo "   rootfs      ${INSTALL_ROOTFS_FILENAME}"
fi
echo ""
echo " Press CTRL+C now if you wish to abort or wait 10 seconds"
echo " to continue."

sleep 10
if test $? -eq 1; then
	echo "Aborted by user.";
	exit;
fi

# Skip user confirmation for U-Boot update
setenv forced_update 1

# Set bootdelay to zero so that firmware update is run immediately after
# the first reset.
setenv bootdelay 0

# Update U-Boot
echo ""
echo ""
echo ">> Installing U-Boot boot loader image ${INSTALL_UBOOT_FILENAME} (target will reset)"
echo ""
echo ""
update uboot mmc ${INSTALL_MMCDEV} ${INSTALL_UBOOT_FILENAME}
if test $? -eq 1; then
	# Use old-style update with source file system argument
	update uboot mmc ${INSTALL_MMCDEV} fat ${INSTALL_UBOOT_FILENAME}
	if test $? -eq 1; then
		echo "[ERROR] Failed to update U-Boot boot loader!";
		echo "";
		echo "Aborted.";
		exit;
	fi
fi

# Set 'bootcmd' to the second part of the script that will
#  - Reset environment to defaults
#  - Restore 'singlemtdsys' and 'dualboot' if previously set
#  - Run 'partition_nand_linux' script to re-partition the NAND if needed
#  - Save the environment
#  - If Dual Boot
#     - Update the system partitions: linux_a, linux_b, rootfs_a, rootfs_b
#  - If Normal Boot:
#     - Update the system partitions: linux, recovery, rootfs
#     - Erase the 'update' partition
#     - Configure recovery to wipe 'update' partition
#     - Run 'recovery' and let the system boot after
setenv bootcmd "
	env default -a;
	setenv singlemtdsys ${singlemtdsys};
	setenv dualboot ${dualboot};
	run partition_nand_linux;
	saveenv;
	if test \"\$\{singlemtdsys\}\" = yes; then
		run ubivolscript;
	else
		force_erase="-e"
	fi;
	if test \"\$\{dualboot\}\" = yes; then
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_LINUX_FILENAME} on linux_a\";
		echo \"\";
		echo \"\";
		update linux_a mmc ${INSTALL_MMCDEV} ${INSTALL_LINUX_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update linux_a!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_LINUX_FILENAME} on linux_b\";
		echo \"\";
		echo \"\";
		update linux_b mmc ${INSTALL_MMCDEV} ${INSTALL_LINUX_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update linux_b!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_ROOTFS_FILENAME} on rootfs_a\";
		echo \"\";
		echo \"\";
		update rootfs_a mmc ${INSTALL_MMCDEV} ${INSTALL_ROOTFS_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update rootfs_a partition!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_ROOTFS_FILENAME} on rootfs_b\";
		echo \"\";
		echo \"\";
		update rootfs_b mmc ${INSTALL_MMCDEV} ${INSTALL_ROOTFS_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update rootfs_b partition!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
	else
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_LINUX_FILENAME} on linux\";
		echo \"\";
		echo \"\";
		update linux mmc ${INSTALL_MMCDEV} ${INSTALL_LINUX_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update linux!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_RECOVERY_FILENAME} on recovery\";
		echo \"\";
		echo \"\";
		update recovery mmc ${INSTALL_MMCDEV} ${INSTALL_RECOVERY_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update recovery partition!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		echo \"\";
		echo \">> Installing file ${INSTALL_ROOTFS_FILENAME} on rootfs\";
		echo \"\";
		echo \"\";
		update rootfs mmc ${INSTALL_MMCDEV} ${INSTALL_ROOTFS_FILENAME} ${force_erase};
		if test \$? -eq 1; then
			echo \"[ERROR] Failed to update rootfs partition!\";
			echo \"\";
			echo \"Aborted.\";
			exit;
		fi;
		echo \"\";
		if test \"\$\{singlemtdsys\}\" != yes; then
			nand erase.part update;
		fi;
		setenv boot_recovery yes;
		setenv recovery_command wipe_update;
	fi;
	saveenv;
	echo \"\";
	echo \"\";
	echo \">> Firmware installation complete.\";
	if test \"\$\{dualboot\}\" != yes; then
		echo \"Rebooting into recovery mode for final deployment.\";
	fi;
	echo \"\";
	echo \"\";
	sleep 1;
	reset;
"

saveenv
reset
