if itest.s a${uboot_defconfig} == a; then
        echo "Please set uboot_defconfig to the appropriate value"
        exit
fi

# remove from environment so they won't override local values
setenv offset
setenv erase_size
setenv qspi_offset
setenv a_base
setenv qspi_match
setenv n

offset=0x400
erase_size=0xC0000
qspi_offset=0x0
a_base=0x10100000
block_size=0x200

#grab 1st 2 characters of string
setexpr cpu2 sub "^(..?).*" "\\1" "${imx_cpu}"
setexpr cpu3 sub "^(..?.?).*" "\\1" "${imx_cpu}"
if itest.s x51 == "x${cpu2}"; then
	a_base=0x90100000
elif itest.s x53 == "x${cpu2}"; then
	a_base=0x70100000
elif itest.s x6SX == "x${cpu3}" || itest.s x6U == "x${cpu2}" || itest.s x7D == "x${cpu2}"; then
	a_base=0x80100000
elif itest.s x8M == "x${cpu2}"; then
	a_base=0x42000000
	offset=0x8400
	if itest.s x8MN == "x${cpu3}" || itest.s x8MP == "x${cpu3}"; then
		offset=0x8000
		if itest ${env_part} != 0 ; then
			offset=0x0
		fi
	fi
fi

qspi_match=1
setexpr a_qspi1 ${a_base}
setexpr a_qspi2 ${a_qspi1} + 0x400000
setexpr a_uImage1 ${a_qspi1} + 0x400
setexpr a_uImage2 ${a_qspi2} + 0x400
setexpr a_script ${a_base}

setenv stdout serial,vga

sf probe
if itest.s "x${sfname}" == "x" ; then
# U-Boot resides in (e)MMC
if itest.s "x${env_dev}" == "x" || itest.s "x${env_part}" == "x"; then
        echo "Please set env_dev/part to the appropriate values"
        exit
fi

# Load bootloader binary for this board
if tftp ${a_uImage1} u-boot.${uboot_defconfig} ; then
else
	echo "File u-boot.${uboot_defconfig} not found on TFTP server" ;
	exit
fi

# Compute block count for filesize and offset
setexpr cntoffset ${offset} / ${block_size}
setexpr cntfile ${filesize} / ${block_size}
# Add 1 in case the $filesize is not a multiple of $block_size
setexpr cntfile ${cntfile} + 1

# Select media partition (if different from main partition)
mmc dev ${env_dev} ${env_part}

# Read and compare current U-Boot
mmc read ${a_uImage2} ${cntoffset} ${cntfile}
if cmp.b ${a_uImage1} ${a_uImage2} ${filesize} ; then
	echo "------- U-Boot versions match" ;
	exit ;
fi

echo "Need U-Boot upgrade" ;
echo "Program in 5 seconds" ;
for n in 5 4 3 2 1 ; do
	echo $n ;
	sleep 1 ;
done
mmc write ${a_uImage1} ${cntoffset} ${cntfile}

# Make sure to boot from the proper partition
if itest ${env_part} != 0 ; then
	mmc partconf ${env_dev} 1 ${env_part} 0
fi

# Switch back to main eMMC partition (to avoid confusion)
mmc dev ${env_dev}

else
# U-Boot resides in NOR flash
if sf probe || sf probe || sf probe 1 27000000 || sf probe 1 27000000 ; then
	echo "probed SPI ROM" ;
else
	echo "Error initializing EEPROM"
	exit
fi

if itest.s "x${sfname}" == "xat45db041d" ; then
	erase_size=0x7e000
fi

if itest.s x7D == "x${imx_cpu}"; then
	echo "check qspi parameter block" ;
	if tftp ${a_qspi1} qspi-${sfname}.${uboot_defconfig} ; then
	else
		echo "parameter file qspi-${sfname}.${uboot_defconfig} not found on TFTP server"
		exit
	fi
	if itest ${filesize} != 0x200 ; then
		echo "------- qspi-${sfname}.${uboot_defconfig} 0x${filesize} != 0x200 bytes" ;
		exit
	fi
	setexpr a_marker ${a_qspi1} + 0x1fc
	if itest *${a_marker} != c0ffee01 ; then
		echo "------- qspi-${sfname}.${uboot_defconfig} c0ffee01 marker missing" ;
		exit
	fi
	if sf read ${a_qspi2} ${qspi_offset} 0x200 ; then
	else
		echo "Error reading qspi parameter from EEPROM"
		exit
	fi
	if cmp.b ${a_qspi1} ${a_qspi2} 0x200 ; then
		echo "------- qspi parameters match"
	else
		echo "------- qspi parameters mismatch"
		qspi_match=0
	fi
fi

echo "check U-Boot" ;

if tftp ${a_uImage1} u-boot.${uboot_defconfig} ; then
else
	echo "File u-boot.${uboot_defconfig} not found on TFTP server" ;
	exit
fi
echo "read ${filesize} bytes" ;
if sf read ${a_uImage2} ${offset} ${filesize} ; then
else
	echo "Error reading boot loader from EEPROM" ;
	exit
fi

if cmp.b ${a_uImage1} ${a_uImage2} ${filesize} ; then
	echo "------- U-Boot versions match" ;
	if itest.s "${qspi_match}" == "1" ; then
		echo "------- u-boot upgrade not needed" ;
		if itest.s x != "x${net_upgrade_fs}" ; then
			if tftp ${a_base} ${net_upgrade_fs} ; then
				echo "------- running ${net_upgrade_fs}" ;
				source ${a_base}
			else
				echo "------- ${net_upgrade_fs} not found on TFTP server" ;
			fi
		fi
		exit
	fi
	erase_size=0x1000
	if itest.s xMX25L6405D == "x${sfname}"; then
		erase_size=0x10000
	fi
	setexpr filesize ${erase_size} - ${offset}
fi

echo "Need U-Boot upgrade" ;
echo "Program in 5 seconds" ;
for n in 5 4 3 2 1 ; do
	echo ${n} ;
	sleep 1 ;
done
echo "erasing" ;
sf erase 0 ${erase_size} ;

# two steps to prevent bricking
echo "programming" ;
setexpr a1 ${a_uImage1} + 0x400
setexpr o1 ${offset} + 0x400
setexpr s1 ${filesize} - 0x400
sf write ${a1} ${o1} ${s1} ;
sf write ${a_uImage1} ${offset} 0x400 ;

if itest.s x7D == "x${imx_cpu}"; then
	sf write ${a_qspi1} ${qspi_offset} 0x200
fi

echo "verifying" ;
if sf read ${a_uImage2} ${offset} ${filesize} ; then
else
	echo "Error re-reading EEPROM" ;
	exit
fi
if cmp.b ${a_uImage1} ${a_uImage2} ${filesize} ; then
else
	echo "Read verification error" ;
	exit
fi

if itest.s x7D == "x${imx_cpu}"; then
	if sf read ${a_qspi2} ${qspi_offset} 0x200 ; then
	else
		echo "Error re-reading qspi" ;
		exit
	fi
	if cmp.b ${a_qspi1} ${a_qspi2} 0x200 ; then
	else
		echo "qspi parameter block verification error" ;
		exit
	fi
fi

if itest.s "x" != "x${next}" ; then
	if tftp ${a_script} ${next} ; then
		source ${a_script}
	else
		echo "${next} not found"
	fi
fi
fi

echo "---- U-Boot upgraded. resetting"
sleep 2
reset

