#
# U-Boot bootscript for EMMC/SD images created by Yocto.
#

# Back up environment variables
setenv ORIG_overlays ${overlays}
setenv ORIG_extra_bootargs ${extra_bootargs}

# Set SOC type to "imx8mm" if not already defined by U-Boot
if test ! -n "${soc_type}"; then
	setenv soc_type "imx8mm"
fi

# Reset overlays when using dualboot mode, because we need to
# save the environment in case of an installation fall back.
if test "${dual_boot}" = "yes"; then
        if test -n "${overlays}"; then
                setenv overlays
        fi
fi

#
# Determine overlays to apply depending on the hardware capabilities
# described by the HWID, SOM version, and carrier board version.
#
if test -n "${module_ram}"; then
	setexpr som_hv ${hwid_2} \& 38
	setexpr som_hv ${som_hv} / 8

	setexpr module_has_wifi ${hwid_2} \& 10000
	setexpr module_has_wifi ${module_has_wifi} / 10000
	setexpr module_has_bt ${hwid_2} \& 20000
	setexpr module_has_bt ${module_has_bt} / 20000

	if test "${module_has_bt}" = "1" && test -z "${disable_bt}"; then
		setenv overlays _ov_som_bt_ccimx8m.dtbo,${overlays}
	fi

	if test "${module_has_wifi}" = "1" && test -z "${disable_wifi}"; then
		setenv overlays _ov_som_wifi_ccimx8m.dtbo,${overlays}
	fi
fi

# Dual boot update verification
if test "${dualboot}" = "yes"; then
	if test "${upgrade_available}" = "1"; then
		echo "Update detected; Booting new system (try ${bootcount})"
		if test "${bootcount}" = "${bootlimit}"; then
			echo "## Update failed; Rolling back to previous version."
			if test "${active_system}" = "linux_a"; then
				part number mmc ${mmcbootdev} linux_b pi
				setenv mmcpart ${pi}
				# Save the partition index on variable rootfs_b_index
				part number mmc ${mmcbootdev} rootfs_b rootfs_b_index
				# Save the rootfs_b UUID into mmcroot_b
				part uuid mmc ${mmcbootdev}:${rootfs_b_index} mmcroot_b
				setenv mmcroot PARTUUID=${mmcroot_b}
				setenv active_system linux_b
			else
				part number mmc ${mmcbootdev} linux_a pi
				setenv mmcpart ${pi}
				# Save the partition index on variable rootfs_a_index
				part number mmc ${mmcbootdev} rootfs_a rootfs_a_index
				# Save the rootfs_a UUID into mmcroot_a
				part uuid mmc ${mmcbootdev}:${rootfs_a_index} mmcroot_a
				setenv mmcroot PARTUUID=${mmcroot_a}
				setenv active_system linux_a
			fi
			setenv upgrade_available
			saveenv
		fi
	else
		if test "${active_system}" = "linux_a"; then
			echo "Booting from system A"
			part number mmc ${mmcbootdev} linux_a pi
			setenv mmcpart ${pi}
			# Save the partition index on variable rootfs_a_index
			part number mmc ${mmcbootdev} rootfs_a rootfs_a_index
			# Save the rootfs_a UUID into mmcroot_a
			part uuid mmc ${mmcbootdev}:${rootfs_a_index} mmcroot_a
			setenv mmcroot PARTUUID=${mmcroot_a}
		else
			echo "Booting from system B"
			part number mmc ${mmcbootdev} linux_b pi
			setenv mmcpart ${pi}
			# Save the partition index on variable rootfs_b_index
			part number mmc ${mmcbootdev} rootfs_b rootfs_b_index
			# Save the rootfs_b UUID into mmcroot_b
			part uuid mmc ${mmcbootdev}:${rootfs_b_index} mmcroot_b
			setenv mmcroot PARTUUID=${mmcroot_b}
		fi
	fi
else
	# Get the UUID of the configured boot partition.
	part uuid mmc ${mmcbootdev}:${mmcpart} bootpart
	# Check the boot source.
	if test "${bootpart}" = "${part1_uuid}"; then
		# We are booting from the eMMC using 'linux'.
		true
	elif test "${bootpart}" = "${part2_uuid}"; then
		# We are booting from the eMMC using 'recovery'.
		setenv boot_initrd true
		setenv initrd_file uramdisk-recovery.img
	else
		# We are booting from the SD card.
		setenv mmcroot /dev/mmcblk${mmcbootdev}p2
	fi
fi
setenv extra_bootargs fbcon=logo-pos:center ${extra_bootargs}
dboot linux mmc ${mmcbootdev}:${mmcpart}



# We only get here in case of an error on the dboot command.

# Undo changes to environment variables
setenv overlays ${ORIG_overlays}
setenv ORIG_overlays
setenv extra_bootargs ${ORIG_extra_bootargs}
setenv ORIG_extra_bootargs
