#!/bin/bash
#

# shellcheck disable=SC2154,SC2086,SC2059

# Variables used only in this file
usb_device=""

#######################
# Download Files List #
#######################
function download_files() {
	local array_name=$1
	local base_url="$2"
	
	# Download all files in the array
	# Use eval for compatibility with older bash versions that don't support nameref (-n)
	local files
	eval "files=(\"\${${array_name}[@]}\")"
	
	for file in "${files[@]}"; do
		if ! $CURL -#LO "${base_url}${file}"; then
			echo_red "Error downloading ${file}; cannot continue"
			return 1
		fi
	done
	#line break
	echo -e ""
}

###################
# flash RW_LEGACY #
###################
function flash_rwlegacy()
{
	#set working dir
	cd /tmp || { exit_red "Error changing to tmp dir; cannot proceed"; return 1; }

	# set dev mode legacy boot / AltFw flags
	if [ "${isChromeOS}" = true ]; then
		crossystem dev_boot_legacy=1 > /dev/null 2>&1
		crossystem dev_boot_altfw=1 > /dev/null 2>&1
	fi

	#determine proper file
	if [ "$device" = "link" ]; then
		rwlegacy_file=$seabios_link
	elif [[ "$isHswBox" = true || "$isBdwBox" = true ]]; then
		rwlegacy_file=$seabios_hswbdw_box
	elif [[ "$isHswBook" = true || "$isBdwBook" = true ]]; then
		rwlegacy_file=$seabios_hswbdw_book
	elif [ "$isByt" = true ]; then
		rwlegacy_file=$seabios_baytrail
	elif [ "$isBsw" = true ]; then
		rwlegacy_file=$seabios_braswell
	elif [ "$isSkl" = true ]; then
		rwlegacy_file=$seabios_skylake
	elif [ "$isApl" = true ]; then
		# prompt for SeaBIOS/edk2 selection
		echo -e ""
		echo_yellow "Firmware Type Selection"
		echo -e "Your device has the option of two RW_LEGACY firmware types."
		REPLY=""
		while [[ "$REPLY" != "L" && "$REPLY" != "l" && "$REPLY" != "U" && "$REPLY" != "u"  ]]
		do
			read -rep "Enter 'L' for Legacy BIOS (SeaBIOS), 'U' for UEFI (edk2/Tianocore): "
			if [[ "$REPLY" = "U" || "$REPLY" = "u" ]]; then
				rwlegacy_file=$rwl_altfw_apl
			else
				rwlegacy_file=$seabios_apl
			fi
		done
	elif [ "$isKbl" = true ]; then
		# prompt for SeaBIOS/edk2 selection
		echo -e ""
		echo_yellow "Firmware Type Selection"
		echo -e "Your device has the option of two RW_LEGACY firmware types."
		REPLY=""
		while [[ "$REPLY" != "L" && "$REPLY" != "l" && "$REPLY" != "U" && "$REPLY" != "u"  ]]
		do
			read -rep "Enter 'L' for Legacy BIOS (SeaBIOS), 'U' for UEFI (edk2/Tianocore): "
			if [[ "$REPLY" = "U" || "$REPLY" = "u" ]]; then
				if [ "$kbl_rwl18" = true ]; then
				rwlegacy_file=$rwl_altfw_kbl_18
				else
				rwlegacy_file=$rwl_altfw_kbl
				fi
			else
				if [ "$kbl_rwl18" = true ]; then
				rwlegacy_file=$seabios_kbl_18
				else
				rwlegacy_file=$seabios_kbl
				fi
			fi
		done
	elif [ "$isWhl" = true ]; then
		rwlegacy_file=$rwl_altfw_whl
	elif [ "$device" = "drallion" ]; then
		rwlegacy_file=$rwl_altfw_drallion
	elif [ "$isCmlBox" = true ]; then
		rwlegacy_file=$rwl_altfw_cml
	elif [ "$isJsl" = true ]; then
		rwlegacy_file=$rwl_altfw_jsl
	elif [ "$isTgl" = true ]; then
		rwlegacy_file=$rwl_altfw_tgl
	elif [ "$isGlk" = true ]; then
		rwlegacy_file=$rwl_altfw_glk
	elif [ "$isAdl_fixed_rwl" = true ]; then
		rwlegacy_file=$rwl_altfw_adl_fixed
	elif [ "$isAdl" = true ]; then
		rwlegacy_file=$rwl_altfw_adl
	elif [ "$isAdlN" = true ]; then
		rwlegacy_file=$rwl_altfw_adl_n
	elif [ "$isMtl" = true ]; then
		rwlegacy_file=$rwl_altfw_mtl
	elif [ "$isStr" = true ]; then
		rwlegacy_file=$rwl_altfw_stoney
	elif [ "$isPco" = true ]; then
		rwlegacy_file=$rwl_altfw_pco
	elif [ "$isCzn" = true ]; then
		rwlegacy_file=$rwl_altfw_czn
	elif [ "$isMdn" = true ]; then
		rwlegacy_file=$rwl_altfw_mdn
	else
		echo_red "Unknown or unsupported device (${device}); cannot update RW_LEGACY firmware."
		echo_red "If your device is listed as supported on https://mrchromebox.tech/#devices,\n
then email MrChromebox@gmail.com  and include a screenshot of the main menu."
		read -rep "Press enter to return to the main menu"
		return 1
	fi
	if [[ "$rwlegacy_file" = *"altfw"* ]]; then
		echo_green "\nInstall/Update RW_LEGACY Firmware (AltFw / edk2)"
	else
		echo_green "\nInstall/Update RW_LEGACY Firmware (Legacy BIOS / SeaBIOS)"
	fi
	echo_yellow "
NOTE: RW_LEGACY firmware cannot be used to run Windows. Period.
If you are looking to run Windows, see the documentation on coolstar.org.
MrChromebox does not provide any support for running Windows."
	REPLY=""
	read -rep "Press Y to continue or any other key to return to the main menu. "
	[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return

	local preferUSB=false
	if [[ "$rwlegacy_file" != *"altfw"* ]]; then
		echo -e ""
		#USB boot priority
		echo_yellow "Default to booting from USB?"
		read -rep "If N, always boot from internal storage unless selected from boot menu. [y/N] "
		[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] && preferUSB=true
		echo -e ""
	fi

	#download RW_LEGACY update
	echo_yellow "\nDownloading RW_LEGACY firmware update\n(${rwlegacy_file})"
	
	rwlegacy_files=(
		"${rwlegacy_file}.md5"
		"${rwlegacy_file}"
	)
	download_files rwlegacy_files "${rwlegacy_source}" || return 1

	#verify checksum on downloaded file
	if ! md5sum -c "${rwlegacy_file}.md5" > /dev/null 2>&1; then
		exit_red "RW_LEGACY download checksum fail; download corrupted, cannot flash"
		return 1
	fi

	#preferUSB?
	if [ "$preferUSB" = true  ]; then
		if ! $CURL -sLo bootorder "${cbfs_source}bootorder.usb"; then
			echo_red "Unable to download bootorder file; boot order cannot be changed."
		else
			${cbfstoolcmd} "${rwlegacy_file}" remove -n bootorder > /dev/null 2>&1
			${cbfstoolcmd} "${rwlegacy_file}" add -n bootorder -f /tmp/bootorder -t raw > /dev/null 2>&1
		fi
	fi

	#flash updated RW_LEGACY firmware
	echo_yellow "Installing RW_LEGACY firmware"
	[[ "$isChromeOS" = false ]] && FMAP="--fmap"
	if ! ${flashromcmd} -w $FMAP -i RW_LEGACY:${rwlegacy_file} ${noverify} -o /tmp/flashrom.log >/dev/null 2>&1; then
		cat /tmp/flashrom.log
		echo_red "An error occurred flashing the RW_LEGACY firmware."
	else
		echo_green "RW_LEGACY firmware successfully installed/updated."
		# update firmware type
		firmwareType="Stock ChromeOS w/RW_LEGACY"
		#Prevent from trying to boot stock ChromeOS install
		[[ "$boot_mounted" = true ]] && rm -rf /tmp/boot/syslinux > /dev/null 2>&1
	fi

	read -rep "Press [Enter] to return to the main menu."
}

#############################
# Install Full ROM Firmware #
#############################
function flash_full_rom()
{
	# ensure hardware write protect disabled
	[[ "$wpEnabled" = true ]] && { exit_red "\nHardware write-protect enabled, cannot flash Full ROM firmware."; return 1; }

	echo_green "\nInstall/Update UEFI Full ROM Firmware"
	echo_yellow "IMPORTANT: flashing the firmware has the potential to brick your device,
requiring relatively inexpensive hardware and some technical knowledge to
recover.Not all boards can be tested prior to release, and even then slight
differences in hardware can lead to unforseen failures.
If you don't have the ability to recover from a bad flash, you're taking a risk.

You have been warned."

	[[ "$isChromeOS" = true ]] && echo_yellow "Also, flashing Full ROM firmware will remove your ability 
to run ChromeOS."

	read -rep "Do you wish to continue? [y/N] "
	[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return

	#spacing
	echo -e ""

	#special warning for CR50 devices
	if [[ "$isStock" = true && "$hasCR50" = true ]]; then
		echo_yellow "NOTICE: flashing your Chromebook is serious business.
To ensure recovery in case something goes wrong when flashing,
be sure to set the ccd capability 'FlashAP Always' using your
USB-C debug cable, otherwise recovery will involve disassembling
your device (which is very difficult in some cases)."

		echo_yellow "If you wish to continue, type: 'I ACCEPT' and press enter."
		read -re
		[[ "$REPLY" = "I ACCEPT" ]] || return
	fi

	#UEFI notice if flashing from ChromeOS or Legacy
	if [[ ! -d /sys/firmware/efi ]]; then
		[[ "$isChromeOS" = true ]] && currOS="ChromeOS" || currOS="Your Legacy-installed OS"
		echo_yellow "
NOTE: After flashing UEFI firmware, you will need to install a UEFI-compatible
OS; ${currOS} will no longer be bootable. See https://mrchromebox.tech/#faq"
		REPLY=""
		read -rep "Press Y to continue or any other key to abort. "
		[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return
	fi

	#determine correct file / URL
	firmware_source=${fullrom_source}
	eval coreboot_file="$`echo "coreboot_uefi_${device}"`"

	# ensure we have a file to flash
	if [[ "$coreboot_file" = "" ]]; then
		exit_red "The script does not currently have a firmware file for your device (${device^^}); cannot continue."; return 1
	fi

	#extract device serial if present in cbfs
	${cbfstoolcmd} /tmp/bios.bin extract -n serial_number -f /tmp/serial.txt >/dev/null 2>&1

	#extract device HWID
	if [[ "$isStock" = "true" ]]; then
		${gbbutilitycmd} /tmp/bios.bin --get --hwid | sed 's/[^ ]* //' > /tmp/hwid.txt 2>/dev/null
	else
		${cbfstoolcmd} /tmp/bios.bin extract -n hwid -f /tmp/hwid.txt >/dev/null 2>&1
	fi

	# create backup if existing firmware is stock
	if [[ "$isStock" = "true" ]]; then
		echo_yellow "\nCreate a backup copy of your stock firmware?"
		echo_yellow "This is highly recommended in case you wish to return your device to stock
configuration/run ChromeOS, or in the (unlikely) event that things go south
and you need to recover using an external EEPROM programmer."
		echo_yellow "If you have already created a backup using the menu option, you can skip this."
		read -erp "Create backup now? [Y/n] "
		if [[ "$REPLY" = "n" || "$REPLY" = "N" ]]; then
			echo_yellow "Skipping backup - ensure you have a backup stored safely!"
		else
			if ! backup_firmware; then
				exit_red "Error creating stock firmware backup; cannot continue."; return 1
			fi
		fi
	fi

	#download firmware file
	cd /tmp || { exit_red "Error changing to tmp dir; cannot proceed"; return 1; }
	echo_yellow "\nDownloading Full ROM firmware\n(${coreboot_file})"
	
	fullrom_files=(
		"${coreboot_file}"
		"${coreboot_file}.sha1"
	)
	download_files fullrom_files "${firmware_source}" || return 1

	#verify checksum on downloaded file
	if ! sha1sum -c "${coreboot_file}.sha1" > /dev/null 2>&1; then
		exit_red "Firmware image checksum verification failed; download corrupted, cannot flash."; return 1
	fi

	#persist serial number?
	if [ -f /tmp/serial.txt ]; then
		echo_yellow "Persisting device serial number"
		${cbfstoolcmd} "${coreboot_file}" add -n serial_number -f /tmp/serial.txt -t raw > /dev/null 2>&1
	fi

	#persist device HWID?
	if [ ! -f /tmp/hwid.txt ] || [ ! -s /tmp/hwid.txt ]; then
		echo_yellow "Creating device HWID from board name"
		echo "${boardName^^}" > /tmp/hwid.txt
	fi
	if [ -f /tmp/hwid.txt ]; then
		echo_yellow "Persisting device HWID"
		${cbfstoolcmd} "${coreboot_file}" add -n hwid -f /tmp/hwid.txt -t raw > /dev/null 2>&1
	fi

	#Persist RW_MRC_CACHE UEFI Full ROM firmware
	${cbfstoolcmd} /tmp/bios.bin read -r RW_MRC_CACHE -f /tmp/mrc.cache > /dev/null 2>&1
	if [[ $isFullRom = "true" && $? -eq 0 ]]; then
		${cbfstoolcmd} "${coreboot_file}" write -r RW_MRC_CACHE -f /tmp/mrc.cache > /dev/null 2>&1
	fi

	#Persist SMMSTORE if exists
	if ${cbfstoolcmd} /tmp/bios.bin read -r SMMSTORE -f /tmp/smmstore > /dev/null 2>&1; then
		${cbfstoolcmd} "${coreboot_file}" write -r SMMSTORE -f /tmp/smmstore > /dev/null 2>&1
	fi

	# persist VPD if possible
	if extract_vpd /tmp/bios.bin; then
		# try writing to RO_VPD FMAP region
		if ! ${cbfstoolcmd} "${coreboot_file}" write -r RO_VPD -f /tmp/vpd.bin > /dev/null 2>&1; then
		# fall back to vpd.bin in CBFS
			${cbfstoolcmd} "${coreboot_file}" add -n vpd.bin -f /tmp/vpd.bin -t raw > /dev/null 2>&1
		fi
	fi

	#disable software write-protect
	echo_yellow "Disabling software write-protect and clearing the WP range"
	if ! ${flashromcmd} --wp-disable > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
		exit_red "Error disabling software write-protect; unable to flash firmware."; return 1
	fi

	#clear SW WP range
	if ! ${flashromcmd} --wp-range 0 0 > /dev/null 2>&1; then
		# use new command format as of commit 99b9550
		if ! ${flashromcmd} --wp-range 0,0 > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
			exit_red "Error clearing software write-protect range; unable to flash firmware."; return 1
		fi
	fi

	#flash Full ROM firmware

	# clear log file
	rm -f /tmp/flashrom.log

	echo_yellow "Installing Full ROM firmware (may take up to 90s)"
	#check if flashrom supports logging to file
	if ${flashromcmd} -V -o /dev/null > /dev/null 2>&1; then
		output_params=">/dev/null 2>&1 -o /tmp/flashrom.log"
		${flashromcmd} ${flashrom_params} ${noverify} -w ${coreboot_file} >/dev/null 2>&1 -o /tmp/flashrom.log
	else
		output_params=">/tmp/flashrom.log 2>&1"
		${flashromcmd} ${flashrom_params} ${noverify} -w ${coreboot_file} >/tmp/flashrom.log 2>&1
	fi
	if [ $? -ne 0 ]; then
		echo_red "Error running cmd: ${flashromcmd} ${flashrom_params} ${noverify} -w ${coreboot_file} ${output_params}"
		if [ -f /tmp/flashrom.log ]; then
			read -rp "Press enter to view the flashrom log file, then space for next page, q to quit"
			more /tmp/flashrom.log
		fi
		exit_red "An error occurred flashing the Full ROM firmware. DO NOT REBOOT!"; return 1
	else
		echo_green "Full ROM firmware successfully installed/updated."

		#Prevent from trying to boot stock ChromeOS install
		if [[ "$isStock" = true && "$isChromeOS" = true && "$boot_mounted" = true ]]; then
			rm -rf /tmp/boot/efi > /dev/null 2>&1
			rm -rf /tmp/boot/syslinux > /dev/null 2>&1
		fi

		#Warn about long RAM training time
		echo_yellow "IMPORTANT:\nThe first boot after flashing may take substantially
longer than subsequent boots -- up to 30s or more.
Be patient and eventually your device will boot :)"

		# Add note on touchpad firmware for EVE
		if [[ "${device^^}" = "EVE" && "$isStock" = true ]]; then
			echo_yellow "IMPORTANT:\n
If you're going to run Windows on your Pixelbook, you must downgrade
the touchpad firmware now (before rebooting) otherwise it will not work.
Select the D option from the main main in order to do so."
		fi
		#set vars to indicate new firmware type
		isStock=false
		isFullRom=true
		# Add NVRAM reset note for 4.12 release
		if [[ "$isUEFI" = true && "$useUEFI" = true ]]; then
			echo_yellow "IMPORTANT:\n
This update uses a new format to store UEFI NVRAM data, and
will reset your BootOrder and boot entries. You may need to
manually Boot From File and reinstall your bootloader if
booting from the internal storage device fails."
		fi
	firmwareType="Full ROM / UEFI (pending reboot)"
	isUEFI=true
	fi

	read -rep "Press [Enter] to return to the main menu."
}

#############################
# Set Touchpad type in SSFC #
#############################
function set_touchpad_in_ssfc()
{
	echo_green "\nSet Touchpad type in SSFC"
	echo_yellow "NOTE: This operation only needs to be done once for GALTIC-based devices
on which you want to run Windows; Linux is not affected either way.
Setting the touchpad type in SSFC requires hardware WP to be disabled."

	read -rep "Do you wish to continue? [y/N] "
	[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return

	# GALTIC boards update SSFC if needed for touchpad type
	echo -e ""
	echo_yellow "Checking if touchpad type set in SSFC"
	echo_yellow "Downloading ectool"
	if ! get_ectool; then
		echo_red "Unable to download ectool; cannot continue"
		read -rep "Press enter to return to the main menu"
		return 1
	fi
	if ! $ectoolcmd cbi get 8 >/dev/null 2>&1; then
		# SSFC not initialized
		echo_yellow "Initializing SSFC"
		if ! $ectoolcmd cbi set 8 0x0 4 1; then
			echo_red "Unable to initialize SSFC; if HW WP is enabled, please disable and retry"
			read -rep "Press enter to return to the main menu"
			return 1
		fi
	fi
	ssfc_val=$($ectoolcmd cbi get 8 | grep -m1 'uint' | cut -f3 -d ' ')
	echo_yellow "Current SSFC value is $ssfc_val"
	
	# TOUCHPAD_OPTION is bits 44-45 in SSFC, so bits 12-13 here
	if [[ $((ssfc_val & 0x3000)) == 0 ]]; then
		#touchpad unset, so detect and set it
		if dmesg | grep "input:" | grep -q "ELAN0000"; then
			#ELAN0000 touchpad, set bit 12
			ssfc_val=$((ssfc_val | 0x1000))
		else
			#ELAN2712 touchpad, set bit 13
			ssfc_val=$((ssfc_val | 0x2000))
		fi

		echo_yellow "Setting new SSFC value $ssfc_val"
		if ! $ectoolcmd cbi set 8 $ssfc_val 4; then
			echo_red "Error setting new SSFC value; if HW WP is enabled, please disable and retry"
			read -rep "Press enter to return to the main menu"
			return 1
		fi
		echo_green "Touchpad type successfully set in SSFC"
	else
		echo_yellow "Touchpad type is already set in SSFC; nothing to do"
	fi

	read -rep "Press [Enter] to return to the main menu."
}

##############################
# Set Storage type in FW_CONFIG #
##############################
function set_storage_in_fw_config()
{
	echo_green "\nSet Storage type in FW_CONFIG"
	echo_yellow "NOTE: This operation sets the storage type (NVMe or eMMC) in FW_CONFIG (CBI tag 6) for taeko/taniks boards.
Setting the storage type in FW_CONFIG requires hardware WP to be disabled."

	# Check if this is a taeko or taniks board
	if [[ "${device^^}" != "TAEKO" && "${device^^}" != "TANIKS" ]]; then
		echo_red "This function is only for taeko/taniks boards"
		read -rep "Press enter to return to the main menu"
		return 1
	fi

	read -rep "Do you wish to continue? [y/N] "
	[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return

	echo -e ""
	echo_yellow "Checking storage type in FW_CONFIG (CBI tag 6)"
	echo_yellow "Downloading ectool"
	if ! get_ectool; then
		echo_red "Unable to download ectool; cannot continue"
		read -rep "Press enter to return to the main menu"
		return 1
	fi
	fw_config_val=$($ectoolcmd cbi get 6 | grep -m1 'uint' | cut -f3 -d ' ')
	if [ -z "$fw_config_val" ]; then
		echo_red "Unable to read FW_CONFIG; cannot continue"
		read -rep "Press enter to return to the main menu"
		return 1
	fi
	echo_yellow "Current FW_CONFIG value is $fw_config_val"
	
	# BOOT_NVME_MASK is bit 12 in FW_CONFIG (0x1000)
	# BOOT_EMMC_MASK is bit 13 in FW_CONFIG (0x2000)
	# These are separate enable/disable bits
	nvme_bit=$((fw_config_val & 0x1000))
	emmc_bit=$((fw_config_val & 0x2000))
	current_storage=""
	
	if [[ $nvme_bit != 0 ]]; then
		current_storage="NVMe"
	elif [[ $emmc_bit != 0 ]]; then
		current_storage="eMMC"
	else
		current_storage="Not set"
	fi
	
	echo_green "Current storage type: $current_storage"
	echo -e "Storage type options:"
	echo -e "  0 = NVMe"
	echo -e "  1 = eMMC"
	echo -e ""
	REPLY=""
	while [[ "$REPLY" != "0" && "$REPLY" != "1" && "$REPLY" != "q" && "$REPLY" != "Q" ]]
	do
		read -rep "Enter storage type to set [0=NVMe, 1=eMMC, q=quit]: "
		if [[ "$REPLY" != "0" && "$REPLY" != "1" && "$REPLY" != "q" && "$REPLY" != "Q" ]]; then
			echo_red "Invalid choice. Please enter 0, 1, or q"
		fi
	done

	echo -e ""
	
	if [[ "$REPLY" = "q" || "$REPLY" = "Q" ]]; then
		echo_yellow "Quitting without changes"
		read -rep "Press [Enter] to return to the main menu."
		return 0
	fi
	
	# Clear both storage bits (bits 12 and 13)
	fw_config_val=$((fw_config_val & ~0x3000))
	
	# Set new storage type
	if [[ "$REPLY" = "1" ]]; then
		# Set eMMC enabled (bit 13 = 0x2000)
		fw_config_val=$((fw_config_val | 0x2000))
		new_storage="eMMC"
	else
		# Set NVMe enabled (bit 12 = 0x1000)
		fw_config_val=$((fw_config_val | 0x1000))
		new_storage="NVMe"
	fi
	
	if [[ "$current_storage" = "$new_storage" ]]; then
		echo_yellow "Storage type is already set to $new_storage; nothing to do"
		read -rep "Press [Enter] to return to the main menu."
		return 0
	fi
	
	echo_yellow "Setting storage type to $new_storage (new FW_CONFIG value: $fw_config_val)"
	if ! $ectoolcmd cbi set 6 $fw_config_val 4; then
		echo_red "Error setting new FW_CONFIG value; if HW WP is enabled, please disable and retry"
		read -rep "Press enter to return to the main menu"
		return 1
	fi
	echo_green "Storage type successfully set to $new_storage in FW_CONFIG"

	read -rep "Press [Enter] to return to the main menu."
}

#########################
# Downgrade Touchpad FW #
#########################
function downgrade_touchpad_fw()
{
	# offer to downgrade touchpad firmware on EVE
	if [[ "${device^^}" = "EVE" ]]; then
		echo_green "\nDowngrade Touchpad Firmware"
		echo_yellow "If you plan to run Windows on your Pixelbook, it is necessary to downgrade
the touchpad firmware, otherwise the touchpad will not work."
		echo_yellow "You should do this after flashing the UEFI firmware, but before rebooting."
		read -rep "Do you wish to downgrade the touchpad firmware now? [y/N] "
		if [[ "$REPLY" = "y" || "$REPLY" = "Y" ]] ; then
			# ensure firmware write protect disabled
			[[ "$wpEnabled" = true ]] && { exit_red "\nHardware write-protect enabled, cannot downgrade touchpad firmware."; return 1; }
			# download TP firmware
			echo_yellow "\nDownloading touchpad firmware\n(${touchpad_eve_fw})"
			
			touchpad_downgrade_files=(
				"${touchpad_eve_fw}"
				"${touchpad_eve_fw}.sha1"
			)
			download_files touchpad_downgrade_files "${other_source}" || return 1

			#verify checksum on downloaded file
			if sha1sum -c ${touchpad_eve_fw}.sha1 > /dev/null 2>&1; then
				# flash TP firmware
				echo_green "Flashing touchpad firmware -- do not touch the touchpad while updating!"
				if ${flashromcmd/${flashrom_programmer}} -p ec:type=tp -i EC_RW -w ${touchpad_eve_fw} -o /tmp/flashrom.log >/dev/null 2>&1; then
					echo_green "Touchpad firmware successfully downgraded."
					echo_yellow "Please reboot your Pixelbook now."
				else
					# try with older eve flashrom
					[[ "$isChromeOS" == "true" ]] && tpPath="/usr/local/bin" || tpPath="/tmp"
					(
						cd $tpPath
						flashrom_eve_files=(
							"${flashrom_eve_tp}"
							"${flashrom_eve_tp}.sha1"
						)
						download_files flashrom_eve_files "${util_source}" || return 1
						sha1sum -c ${flashrom_eve_tp}.sha1 > /dev/null 2>&1 || \
							{ echo_red "Flashrom Eve TP checksum fail; download corrupted, cannot flash."; return 1; }
						chmod +x ${flashrom_eve_tp}
					)
					if $tpPath/${flashrom_eve_tp} -p ec:type=tp -i EC_RW -w ${touchpad_eve_fw} -o /tmp/flashrom.log >/dev/null 2>&1; then
						echo_green "Touchpad firmware successfully downgraded."
						echo_yellow "Please reboot your Pixelbook now."
					else
						echo_red "Error flashing touchpad firmware:"
						cat /tmp/flashrom.log
						echo_yellow "\nThis function sometimes doesn't work under Linux, in which case it is
recommended to try under ChromiumOS."
					fi
				fi
			else
				echo_red "Touchpad firmware download checksum fail; download corrupted, cannot flash."
			fi
		fi
		read -rep "Press [Enter] to return to the main menu."
	fi
}

#######################
# Upgrade Touchpad FW #
#######################
function upgrade_touchpad_fw()
{
	# offer to upgrade touchpad firmware on EVE
	if [[ "${device^^}" = "EVE" ]]; then
		echo_green "\nUpgrade Touchpad Firmware"
		echo_yellow "
If you plan to restore ChromeOS on your Pixelbook, it is necessary to upgrade
the touchpad firmware, otherwise the touchpad will not work."
		echo_yellow "You should do this after restoring the stock firmware, but before rebooting."
		read -rep "Do you wish to upgrade the touchpad firmware now? [y/N] "
		if [[ "$REPLY" = "y" || "$REPLY" = "Y" ]] ; then
			# ensure firmware write protect disabled
			[[ "$wpEnabled" = true ]] && { exit_red "\nHardware write-protect enabled, cannot upgrade touchpad firmware."; return 1; }
			# download TP firmware
			echo_yellow "\nDownloading touchpad firmware\n(${touchpad_eve_fw_stock})"
			
			# Array of touchpad upgrade files to download
			touchpad_upgrade_files=(
				"${touchpad_eve_fw_stock}"
				"${touchpad_eve_fw_stock}.sha1"
			)
			
			# Download touchpad upgrade files
			if ! download_files touchpad_upgrade_files "${other_source}"; then
				return 1
			fi
			#verify checksum on downloaded file
			if sha1sum -c ${touchpad_eve_fw_stock}.sha1 > /dev/null 2>&1; then
				# flash TP firmware
				echo_green "Flashing touchpad firmware -- do not touch the touchpad while updating!"
				if ${flashromcmd/${flashrom_programmer}} -p ec:type=tp -i EC_RW -w ${touchpad_eve_fw_stock} -o /tmp/flashrom.log >/dev/null 2>&1; then
					echo_green "Touchpad firmware successfully upgraded."
					echo_yellow "Please reboot your Pixelbook now."
				else
				# try with older eve flashrom
				[[ "$isChromeOS" == "true" ]] && tpPath="/usr/local/bin" || tpPath="/tmp"
				(
					cd $tpPath
					if ! $CURL -sLO "${util_source}flashrom_eve_tp"; then
						echo_red "Error downloading flashrom_eve_tp; cannot continue"
						exit 1
					fi
					chmod +x flashrom_eve_tp
				)
				if $tpPath/flashrom_eve_tp -p ec:type=tp -i EC_RW -w ${touchpad_eve_fw_stock} -o /tmp/flashrom.log >/dev/null 2>&1; then
					echo_green "Touchpad firmware successfully upgraded."
					echo_yellow "Please reboot your Pixelbook now."
				else
					echo_red "Error flashing touchpad firmware:"
					cat /tmp/flashrom.log
					echo_yellow "\nThis function sometimes doesn't work under Linux, in which case it is
recommended to try under ChromeOS."
				fi
			fi
			else
				echo_red "Touchpad firmware download checksum fail; download corrupted, cannot flash."
			fi
		fi
	read -rep "Press [Enter] to return to the main menu."
	fi
}

##########################
# Flash Custom Firmware #
########################
function flash_custom_firmware()
{
	# ensure hardware write protect disabled
	[[ "$wpEnabled" = true ]] && { exit_red "\nHardware write-protect enabled, cannot flash custom firmware."; return 1; }

	echo_green "\nFlash Custom Firmware Image"
	echo_yellow "IMPORTANT: flashing custom firmware has the potential to brick your device,
requiring relatively inexpensive hardware and some technical knowledge to
recover. Make sure you have a backup of your current firmware before proceeding.
If you don't have the ability to recover from a bad flash, you're taking a risk.

You have been warned."

	read -rep "Do you wish to continue? [y/N] "
	[[ "$REPLY" = "Y" || "$REPLY" = "y" ]] || return

	echo -e ""
	echo_yellow "Please select firmware source:"
	echo -e "1) Flash from local filesystem"
	echo -e "2) Flash from USB device"
	echo -e "Q) Quit and return to main menu"
	echo -e ""
	flash_option=-1
	while :
	do
		read -rep "? " flash_option
		case $flash_option in
		1)  flash_firmware_from_local || return 1;
			break;
			;;
		2)  flash_firmware_from_usb || return 1;
			break;
			;;
		Q|q) flash_option="Q";
			break;
			;;
		esac
	done
	[[ "$flash_option" = "Q" ]] && return
}

function flash_firmware_from_local()
{
	echo_yellow "\nFlashing firmware from local filesystem"
	read -rep "Enter the full path to the custom firmware file: " firmware_path
	
	local hasError=false
	if [ -z "$firmware_path" ]; then
		echo_red "No firmware path provided"
		hasError=true
	fi
	
	if [ ! -f "$firmware_path" ]; then
		echo_red "Firmware file not found: $firmware_path"
		hasError=true
	fi
	
	# Copy firmware to /tmp for processing
	cp "$firmware_path" /tmp/custom-firmware.rom || {
		echo_red "Failed to copy firmware file to /tmp"
		hasError=true
	}
	if $hasError; then
		read -rep "Press [Enter] to return to the main menu."
		return 1
	fi
	
	# Process the custom firmware
	process_and_flash_custom_firmware "/tmp/custom-firmware.rom"
}

function flash_firmware_from_usb()
{
	read -rep "Connect the USB/SD device which contains the custom firmware and press [Enter] to continue. "
	list_usb_devices || { exit_red "No USB devices available to read firmware from."; return 1; }
	usb_dev_index=""
	while [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; do
		read -rep "Enter the number for the device which contains the custom firmware: " usb_dev_index
		if [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; then
			echo -e "Error: Invalid option selected; enter a number from the list above."
		fi
	done

	usb_device="${usb_devs[${usb_dev_index}-1]}"
	mkdir /tmp/usb > /dev/null 2>&1
	mount "${usb_device}" /tmp/usb > /dev/null 2>&1
	if [ $? -ne 0 ]; then
		mount "${usb_device}1" /tmp/usb
	fi
	if [ $? -ne 0 ]; then
		echo_red "USB device failed to mount; cannot proceed."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	
	# Select file from USB device
	echo_yellow "\nFirmware Files on USB:"
	if ! ls /tmp/usb/*.{rom,ROM,bin,BIN} 2>/dev/null | xargs -n 1 basename 2>/dev/null; then
		echo_red "No firmware files found on USB device."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	echo -e ""
	read -rep "Enter the firmware filename: " firmware_file
	firmware_file="/tmp/usb/${firmware_file}"
	if [ ! -f "$firmware_file" ]; then
		echo_red "Invalid filename entered; unable to flash custom firmware."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	
	# Copy firmware to /tmp for processing
	cp "$firmware_file" /tmp/custom-firmware.rom || {
		echo_red "Failed to copy firmware file to /tmp"
		umount /tmp/usb > /dev/null 2>&1
		return 1
	}
	
	# Cleanup USB mount
	umount /tmp/usb > /dev/null 2>&1
	rmdir /tmp/usb
	
	# Process the custom firmware
	process_and_flash_custom_firmware "/tmp/custom-firmware.rom"
}

function process_and_flash_custom_firmware()
{
	local custom_firmware_file="$1"
	
	# Check if we can read the custom firmware
	if ! ${cbfstoolcmd} "${custom_firmware_file}" print > /dev/null 2>&1; then
		echo_yellow "Warning: Unable to read custom firmware with cbfstool. Proceeding anyway."
	fi
	
	# Extract current firmware data to preserve
	echo_yellow "Extracting device-specific data from current firmware"
	
	# Extract serial number if present
	${cbfstoolcmd} /tmp/bios.bin extract -n serial_number -f /tmp/serial.txt >/dev/null 2>&1
	
	# Extract HWID if present
	${cbfstoolcmd} /tmp/bios.bin extract -n hwid -f /tmp/hwid.txt >/dev/null 2>&1
	
	# Extract VPD if possible
	extract_vpd /tmp/bios.bin >/dev/null 2>&1
	
	# Extract RW_MRC_CACHE if present
	${cbfstoolcmd} /tmp/bios.bin read -r RW_MRC_CACHE -f /tmp/mrc.cache > /dev/null 2>&1
	
	# Extract SMMSTORE if present
	${cbfstoolcmd} /tmp/bios.bin read -r SMMSTORE -f /tmp/smmstore > /dev/null 2>&1
	
	# Persist serial number if extracted
	if [ -f /tmp/serial.txt ]; then
		echo_yellow "Persisting device serial number"
		${cbfstoolcmd} "${custom_firmware_file}" add -n serial_number -f /tmp/serial.txt -t raw > /dev/null 2>&1
	fi
	
	# Persist HWID if extracted
	if [ -f /tmp/hwid.txt ]; then
		echo_yellow "Persisting device HWID"
		hwid_content=$(cat /tmp/hwid.txt)
		if [[ "$hwid_content" =~ ^[A-Z0-9]+ ]]; then
			# Use gbb_utility if it's a proper HWID format
			${gbbutilitycmd} "${custom_firmware_file}" --set --hwid="$hwid_content" > /dev/null 2>&1
		else
			# Use cbfstool for raw HWID data
			${cbfstoolcmd} "${custom_firmware_file}" add -n hwid -f /tmp/hwid.txt -t raw > /dev/null 2>&1
		fi
	fi
	
	# Persist VPD if extracted
	if [ -f /tmp/vpd.bin ]; then
		echo_yellow "Persisting VPD data"
		if ! ${cbfstoolcmd} "${custom_firmware_file}" write -r RO_VPD -f /tmp/vpd.bin > /dev/null 2>&1; then
			${cbfstoolcmd} "${custom_firmware_file}" add -n vpd.bin -f /tmp/vpd.bin -t raw > /dev/null 2>&1
		fi
	fi
	
	# Persist RW_MRC_CACHE if extracted
	if [ -f /tmp/mrc.cache ]; then
		echo_yellow "Persisting RW_MRC_CACHE"
		${cbfstoolcmd} "${custom_firmware_file}" write -r RW_MRC_CACHE -f /tmp/mrc.cache > /dev/null 2>&1
	fi
	
	# Persist SMMSTORE if extracted
	if [ -f /tmp/smmstore ]; then
		echo_yellow "Persisting SMMSTORE"
		${cbfstoolcmd} "${custom_firmware_file}" write -r SMMSTORE -f /tmp/smmstore > /dev/null 2>&1
	fi
	
	# Disable software write-protect
	echo_yellow "Disabling software write-protect and clearing the WP range"
	if ! ${flashromcmd} --wp-disable > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
		exit_red "Error disabling software write-protect; unable to flash firmware."; return 1
	fi
	
	# Clear SW WP range
	if ! ${flashromcmd} --wp-range 0 0 > /dev/null 2>&1; then
		if ! ${flashromcmd} --wp-range 0,0 > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
			exit_red "Error clearing software write-protect range; unable to flash firmware."; return 1
		fi
	fi
	
	# Flash the custom firmware
	echo_yellow "Installing custom firmware (may take up to 90s)"
	rm -f /tmp/flashrom.log
	
	# Check if flashrom supports logging to file
	if ${flashromcmd} -V -o /dev/null > /dev/null 2>&1; then
		output_params=">/dev/null 2>&1 -o /tmp/flashrom.log"
		${flashromcmd} ${flashrom_params} ${noverify} -w "${custom_firmware_file}" >/dev/null 2>&1 -o /tmp/flashrom.log
	else
		output_params=">/tmp/flashrom.log 2>&1"
		${flashromcmd} ${flashrom_params} ${noverify} -w "${custom_firmware_file}" >/tmp/flashrom.log 2>&1
	fi
	
	if [ $? -ne 0 ]; then
		echo_red "Error running cmd: ${flashromcmd} ${flashrom_params} ${noverify} -w ${custom_firmware_file} ${output_params}"
		if [ -f /tmp/flashrom.log ]; then
			read -rp "Press enter to view the flashrom log file, then space for next page, q to quit"
			more /tmp/flashrom.log
		fi
		exit_red "An error occurred flashing the custom firmware. DO NOT REBOOT!"; return 1
	else
		echo_green "Custom firmware successfully installed/updated."
		
		# Update firmware type
		firmwareType="Custom Firmware (pending reboot)"
		isStock=false
		isFullRom=true
		isUEFI=true
		
		echo_yellow "IMPORTANT:
The first boot after flashing may take substantially
longer than subsequent boots -- up to 30s or more.
Be patient and eventually your device will boot :)"
	fi
		
	read -rep "Press [Enter] to return to the main menu."
}

########################
# Restore Stock Firmware #
##########################
function restore_stock_firmware()
{
		echo_green "\nRestore Stock Firmware"
		echo_yellow "Standard disclaimer: flashing the firmware has the potential to
brick your device, requiring relatively inexpensive hardware and some
technical knowledge to recover.  You have been warned."
		read -rep "Do you wish to continue? [y/N] "
		[[ "$REPLY" = "Y" || "$REPLY" = "y" ]] || return
		# check if EOL
		if [ "$isEOL" = true ]; then
			echo_yellow "
VERY IMPORTANT:
Your device has reached end of life (EOL) and is no longer supported by Google.
Returning the to stock firmware **IS NOT RECOMMENDED**.
MrChromebox will not provide any support for EOL devices running anything
other than the latest UEFI Full ROM firmware release."
			read -rep "Do you wish to continue? [y/N] "
			[[ "$REPLY" = "Y" || "$REPLY" = "y" ]] || return
		fi
		#spacing
		echo -e ""
		# ensure hardware write protect disabled
		[[ "$wpEnabled" = true ]] && 
			{ exit_red "\nHardware write-protect enabled, cannot restore stock firmware."; return 1; }
		# default file to download to
		firmware_file="/tmp/stock-firmware.rom"
		echo -e ""
		echo_yellow "Please select an option below for restoring the stock firmware:"
		echo -e "1) Restore using a firmware backup on USB"
		echo -e "2) Restore using a ChromeOS Recovery USB"
		echo -e "Q) Quit and return to main menu"
		echo -e ""
		restore_option=-1
		while :
		do
			read -rep "? " restore_option
			case $restore_option in
			1)  restore_fw_from_usb || return 1;
				break;
				;;
			2)  restore_fw_from_recovery || return 1;
				break;
				;;
			Q|q) restore_option="Q";
				break;
				;;
			esac
		done
		[[ "$restore_option" = "Q" ]] && return
		if [[ $restore_option -eq 2 ]]; then
			#extract VPD from current firmware if present
			if extract_vpd /tmp/bios.bin ; then
			#merge with recovery image firmware
				if [ -f /tmp/vpd.bin ]; then
					echo_yellow "Merging VPD into recovery image firmware"
					${cbfstoolcmd} ${firmware_file} write -r RO_VPD -f /tmp/vpd.bin > /dev/null 2>&1
				fi
			fi
			#extract hwid from current firmware if present
			if ${cbfstoolcmd} /tmp/bios.bin extract -n hwid -f /tmp/hwid.txt > /dev/null 2>&1; then
				#merge with recovery image firmware
				hwid="$(sed 's/^hardware_id: //' /tmp/hwid.txt 2>/dev/null)"
				if [[ "$hwid" != "" ]]; then
					echo_yellow "Injecting HWID into recovery image firmware"
					${gbbutilitycmd} ${firmware_file} --set --hwid="$hwid" > /dev/null 2>&1
				fi
			fi
		fi
		#clear GBB flags before flashing
		${gbbutilitycmd} ${firmware_file} --set --flags=0x0 > /dev/null 2>&1
		#flash stock firmware
		echo_yellow "Restoring stock firmware"
		# only verify part of flash we write
		if ! ${flashromcmd} ${flashrom_params} ${noverify} -w "${firmware_file}" -o /tmp/flashrom.log >/dev/null 2>&1; then
			cat /tmp/flashrom.log
			exit_red "An error occurred restoring the stock firmware. DO NOT REBOOT!"; return 1
		fi
		#re-enable software WP to prevent recovery issues
		echo_yellow "Re-enabling software write-protect"
		${flashromcmd} --wp-region WP_RO --fmap > /dev/null 2>&1
		if ! ${flashromcmd} --wp-enable > /dev/null 2>&1; then
			echo_red "Warning: unable to re-enable software write-protect;"
			echo_red "you may need to perform ChromeOS recovery with the battery disconnected."
		fi
		#all good
		echo_green "Stock firmware successfully restored."
		
		# Optionally reset CR50 NVRAM data if device has CR50
		if [[ "$hasCR50" = true ]]; then
			reset_cr50_nvram "$firmware_file"
		fi
		
		echo_green "After rebooting, you need to restore ChromeOS using ChromeOS Recovery media."
		echo_green "See: https://google.com/chromeos/recovery for more info."
		read -rep "Press [Enter] to return to the main menu."
		#set vars to indicate new firmware type
		isStock=true
		isFullRom=false
		isUEFI=false
		firmwareType="Stock ChromeOS (pending reboot)"
}

function restore_fw_from_usb()
{
	read -rep "
Connect the USB/SD device which contains the backed-up stock firmware and press [Enter] to continue. "
	list_usb_devices || { exit_red "No USB devices available to read firmware backup."; return 1; }
	usb_dev_index=""
	while [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; do
		read -rep "Enter the number for the device which contains the stock firmware backup: " usb_dev_index
		if [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; then
			echo -e "Error: Invalid option selected; enter a number from the list above."
		fi
	done

	usb_device="${usb_devs[${usb_dev_index}-1]}"
	mkdir /tmp/usb > /dev/null 2>&1
	mount "${usb_device}" /tmp/usb > /dev/null 2>&1
	if [ $? -ne 0 ]; then
		mount "${usb_device}1" /tmp/usb
	fi
	if [ $? -ne 0 ]; then
		echo_red "USB device failed to mount; cannot proceed."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	#select file from USB device
	echo_yellow "\n(Potential) Firmware Files on USB:"
	if ! ls  /tmp/usb/*.{rom,ROM,bin,BIN} 2>/dev/null | xargs -n 1 basename 2>/dev/null; then
		echo_red "No firmware files found on USB device."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	echo -e ""
	read -rep "Enter the firmware filename:  " firmware_file
	firmware_file=/tmp/usb/${firmware_file}
	if [ ! -f ${firmware_file} ]; then
		echo_red "Invalid filename entered; unable to restore stock firmware."
		read -rep "Press [Enter] to return to the main menu."
		umount /tmp/usb > /dev/null 2>&1
		return 1
	fi
	#text spacing
	echo -e ""
}

function restore_fw_from_recovery()
{
	if ! command -v 7z >/dev/null 2>&1; then
		exit_red "Error: 7z (7zip) is required but not found. Please install it via the 7zip package.";
		return 1
	fi
	echo -e "\nConnect a USB which contains a ChromeOS Recovery Image"
	read -rep "and press [Enter] to continue. "
	list_usb_devices || { exit_red "No USB devices available to read from."; return 1; }
	usb_dev_index=""
	while [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && ! $usb_dev_index -gt $num_usb_devs) ]]; do
		read -rep "Enter the number which corresponds your ChromeOS Recovery USB: " usb_dev_index
		if [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; then
			echo -e "Error: Invalid option selected; enter a number from the list above."
		fi
	done
	usb_device="${usb_devs[${usb_dev_index}-1]}"
	echo -e ""
	echo_yellow "Using USB device: $usb_device"
	if ! extract_firmware_from_recovery_usb ${boardName,,} $usb_device ; then
		exit_red "Error: failed to extract firmware for ${boardName^^} from this ChromeOS recovery USB"
		return 1
	fi
	mv coreboot-Google_* ${firmware_file}
	# set a semi-legit HWID in case we don't have a backup below
	${gbbutilitycmd} --set --hwid="${boardName^^} ABC-123-XYZ-456" ${firmware_file} > /dev/null
	echo_yellow "Stock firmware successfully extracted from ChromeOS recovery image"
}

######################################
# Extract firmware from recovery usb #
######################################
function extract_firmware_from_recovery_usb()
{
	_board=$1
	_debugfs=${2}3
	_firmware=chromeos-firmwareupdate-$_board
	_unpacked=$(mktemp -d)
	if [[ "$1" = "" || "$2" = "" ]]; then
		echo_red "Invalid or missing function parameters: [$*]"
		return 1
	fi
	echo_yellow "Extracting firmware from recovery USB"
	printf "cd /usr/sbin\ndump chromeos-firmwareupdate $_firmware\nquit" | debugfs $_debugfs >/dev/null 2>&1
	if [ ! -f $_firmware ]; then
		echo_red "Failed to copy file 'chromeos-firmwareupdate' from Recovery USB"
		return 1
	fi
	if ! sh $_firmware --unpack $_unpacked >/dev/null 2>&1; then
		if ! sh $_firmware --sb_extract $_unpacked >/dev/null 2>&1; then
			echo_red "Failed to extract shellball from  'chromeos-firmwareupdate'"
			return 1
		fi
	fi
	if [ -d $_unpacked/models/ ]; then
		_version=$(cat $_unpacked/VERSION | grep -m 1 -e Model.*$_board -A5 | grep "BIOS (RW) version:" | cut -f2 -d: | tr -d \ )
		if [ "$_version" = "" ]; then
			_version=$(cat $_unpacked/VERSION | grep -m 1 -e Model.*$_board -A5 | grep "BIOS version:" | cut -f2 -d: | tr -d \ )
		fi
		if [ -f $_unpacked/models/$_board/setvars.sh ]; then
			_bios_image=$(grep "IMAGE_MAIN" $_unpacked/models/$_board/setvars.sh | cut -f2 -d'"')
		else
			# special case for REEF, others?
			_version=$(grep -m1 "host" "$_unpacked/manifest.json" | cut -f12 -d'"')
			_bios_image=$(grep -m1 "image" "$_unpacked/manifest.json" | cut -f4 -d'"')
		fi
	elif [ -f "$_unpacked/manifest.json" ]; then
		_version=$(grep -m1 -A4 "$_board\":" "$_unpacked/manifest.json" | grep -m1 "rw" | sed 's/.*\(rw.*\)/\1/' | sed 's/.*\("Google.*\)/\1/' | cut -f2 -d'"')
		_bios_image=$(grep -m1 -A10 "$_board\":" "$_unpacked/manifest.json" | grep -m1 "image" | sed 's/.*"image": //' | cut -f2 -d'"')
	else
		if [ -f $_unpacked/VERSION ]; then
			_version=$(cat $_unpacked/VERSION | grep BIOS\ version: | cut -f2 -d: | tr -d \ )
			_bios_image=bios.bin
		else
			echo_red "Recovery image missing VERSION file. Shellball directory Contents:"
			ls -lart $_unpacked
			return 1
		fi
	fi
	if ! cp $_unpacked/$_bios_image coreboot-$_version.bin; then
		return 1
	fi
	rm -rf "$_unpacked"
	rm $_firmware
}


########################
# Extract firmware VPD #
########################
function extract_vpd()
{
	#check params
	[[ -z "$1" ]] && { exit_red "Error: extract_vpd(): missing function parameter"; return 1; }

	local firmware_file="$1"

	#try FMAP extraction
	if ! ${cbfstoolcmd} ${firmware_file} read -r RO_VPD -f /tmp/vpd.bin >/dev/null 2>&1 ; then
		#try CBFS extraction
		if ! ${cbfstoolcmd} ${firmware_file} extract -n vpd.bin -f /tmp/vpd.bin >/dev/null 2>&1 ; then
			echo_yellow "No VPD found in current firmware"
			return 1
		fi
	fi
	echo_yellow "VPD extracted from current firmware"
}

#########################
# Backup Current Firmware #
#########################
function backup_current_firmware()
{
	echo_green "\nBackup Current Firmware"
	echo_yellow "This function allows you to backup the current firmware to either
a local file or a USB device. This is useful for creating backups
before flashing custom firmware."

	read -rep "Do you wish to continue? [y/N] "
	[[ "$REPLY" = "Y" || "$REPLY" = "y" ]] || return

	echo -e ""
	echo_yellow "Please select backup destination:"
	echo -e "1) Backup to local filesystem"
	echo -e "2) Backup to USB device"
	echo -e "Q) Quit and return to main menu"
	echo -e ""
	backup_option=-1
	while :
	do
		read -rep "? " backup_option
		case $backup_option in
		1)  backup_firmware_to_local || return 1;
			break;
			;;
		2)  backup_firmware_to_usb || return 1;
			break;
			;;
		Q|q) backup_option="Q";
			break;
			;;
		esac
	done
	[[ "$backup_option" = "Q" ]] && return
}

function backup_firmware_to_local()
{
	echo_yellow "\nBacking up firmware to local filesystem"
	echo -e "Enter the directory path for the backup (e.g., /home/user/backups/)"
	echo -e "Or just press [Enter] to use the current directory."
	read -rep "" backup_dir
	
	# Default to current directory if empty
	if [ -z "$backup_dir" ]; then
		backup_dir=$(pwd)
	fi
	
	# Remove trailing slash if present and normalize path
	backup_dir="${backup_dir%/}"
	
	# Create directory if it doesn't exist
	if [[ ! -d "$backup_dir" ]]; then
		echo_yellow "Creating directory: $backup_dir"
		mkdir -p "$backup_dir" || {
			echo_red "Failed to create directory: $backup_dir"
			return 1
		}
	fi
	
	# Generate backup filename
	backupname="BACKUP-${boardName}-$fwVer-$(date +%Y.%m.%d).rom"
	backup_path="$backup_dir/$backupname"

	echo_yellow "Saving firmware backup:\n$backup_path"
	if ! cp /tmp/bios.bin "$backup_path"; then
		echo_red "Failure copying firmware to $backup_path"
		return 1
	fi
	echo_green "Firmware backup complete."
	read -rep "Press [Enter] to return to the main menu."
}

function backup_firmware_to_usb()
{
	echo -e ""
	read -rep "Connect the USB/SD device to store the firmware backup and press [Enter]
to continue.  This is non-destructive, but it is best to ensure no other
USB/SD devices are connected. "
	if ! list_usb_devices; then
		backup_fail "No USB devices available to store firmware backup."
		return 1
	fi
	usb_dev_index=""
	while [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; do
		read -rep "Enter the number for the device to be used for firmware backup: " usb_dev_index
		if [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; then
			echo -e "Error: Invalid option selected; enter a number from the list above."
		fi
	done
	usb_device="${usb_devs[${usb_dev_index}-1]}"
	mkdir /tmp/usb > /dev/null 2>&1
	if ! mount -o rw "${usb_device}" /tmp/usb > /dev/null 2>&1; then
		if ! mount -o rw "${usb_device}1" /tmp/usb > /dev/null 2>&1; then
			backup_fail "USB backup device failed to mount; cannot proceed. Ensure your USB is FAT32-formatted and try again."
			return 1
		fi
	fi
	backupname="BACKUP-${boardName}-$fwVer-$(date +%Y.%m.%d).rom"
	echo_yellow "\nSaving firmware backup: ${backupname}"
	if ! cp /tmp/bios.bin /tmp/usb/${backupname}; then
		backup_fail "Failure copying firmware to USB; cannot proceed."
		return 1
	fi
	sync
	umount /tmp/usb > /dev/null 2>&1
	rmdir /tmp/usb
	echo_green "Firmware backup complete.\n\nRemove the USB stick and press [Enter] to continue."
	read -rep ""
}

#########################
# Backup stock firmware #
#########################
function backup_firmware()
{
	echo -e ""
	read -rep "Connect the USB/SD device to store the firmware backup and press [Enter]
to continue.  This is non-destructive, but it is best to ensure no other
USB/SD devices are connected. "
	if ! list_usb_devices; then
		backup_fail "No USB devices available to store firmware backup."
		return 1
	fi
	usb_dev_index=""
	while [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; do
		read -rep "Enter the number for the device to be used for firmware backup: " usb_dev_index
		if [[ "$usb_dev_index" = "" || ($usb_dev_index -le 0 && $usb_dev_index -gt $num_usb_devs) ]]; then
			echo -e "Error: Invalid option selected; enter a number from the list above."
		fi
	done
	usb_device="${usb_devs[${usb_dev_index}-1]}"
	mkdir /tmp/usb > /dev/null 2>&1
	if ! mount -o rw "${usb_device}" /tmp/usb > /dev/null 2>&1; then
		if ! mount -o rw "${usb_device}1" /tmp/usb > /dev/null 2>&1; then
			backup_fail "USB backup device failed to mount; cannot proceed. Ensure your USB is FAT32-formatted and try again."
			return 1
		fi
	fi
	backupname="stock-firmware-${boardName}-$(date +%Y%m%d).rom"
	echo_yellow "\nSaving firmware backup as ${backupname}"
	if ! cp /tmp/bios.bin /tmp/usb/${backupname}; then
		backup_fail "Failure copying stock firmware to USB; cannot proceed."
		return 1
	fi
	sync
	umount /tmp/usb > /dev/null 2>&1
	rmdir /tmp/usb
	echo_green "Firmware backup complete. Remove the USB stick and press [Enter] to continue."
	read -rep ""
}

function backup_fail()
{
	umount /tmp/usb > /dev/null 2>&1
	rmdir /tmp/usb > /dev/null 2>&1
	exit_red "\n$*"
}

####################
# Set Boot Options #
####################
function set_boot_options()
{
	# set boot options via firmware boot flags

	# ensure hardware write protect disabled
	[[ "$wpEnabled" = true ]] && { exit_red  "\nHardware write-protect enabled, cannot set Boot Options / GBB Flags."; return 1; }

	echo_green "\nSet Firmware Boot Options (GBB Flags)"
	echo_yellow "Select your preferred boot delay and default boot option.
You can always override the default using [CTRL+D] or
[CTRL+L] on the Developer Mode boot screen"

	echo -e "1) Short boot delay (1s) + Legacy Boot/AltFw default
2) Long boot delay (30s) + Legacy Boot/AltFw default
3) Short boot delay (1s) + ChromeOS default
4) Long boot delay (30s) + ChromeOS default
5) Reset to factory default
6) Cancel/exit
"
	local _flags=0x0
	while :
	do
		read -rep "? " n
		case $n in
		1) _flags=0x4A9; break;;
		2) _flags=0x4A8; break;;
		3) _flags=0xA9; break;;
		4) _flags=0xA8; break;;
		5) _flags=0x0; break;;
		6) read -rep "Press [Enter] to return to the main menu."; break;;
		*) echo -e "invalid option";;
		esac
	done
	[[ $n -eq 6 ]] && return
	echo_yellow "\nSetting boot options..."

	#disable software write-protect
	if ! ${flashromcmd} --wp-disable > /dev/null 2>&1; then
		exit_red "Error disabling software write-protect; unable to set GBB flags."; return 1
	fi
	[[ "$isChromeOS" = false ]] && FMAP="--fmap"
	if ! ${flashromcmd} -r $FMAP -i GBB:/tmp/gbb.temp > /dev/null 2>&1; then
		exit_red "\nError reading firmware (non-stock?); unable to set boot options."; return 1
	fi
	if ! ${gbbutilitycmd} --set --flags="${_flags}" /tmp/gbb.temp > /dev/null; then
		exit_red "\nError setting boot options."; return 1
	fi
	if ! ${flashromcmd} -w $FMAP -i GBB:/tmp/gbb.temp ${noverify} -o /tmp/flashrom.log > /dev/null 2>&1; then
		cat /tmp/flashrom.log
		exit_red "\nError writing back firmware; unable to set boot options."; return 1
	fi
	echo_green "\nFirmware Boot options successfully set."
	read -rep "Press [Enter] to return to the main menu."
}

###################
# Set Hardware ID #
###################
function set_hwid()
{
	# set HWID using gbb_utility
	# ensure hardware write protect disabled
	[[ "$wpEnabled" = true ]] && { exit_red  "\nHardware write-protect enabled, cannot set HWID."; return 1; }

	echo_green "Set Hardware ID (HWID) using gbb_utility"

	#get current HWID
	_hwid="$(crossystem hwid)" >/dev/null 2>&1
	if [[ "$_hwid" != "" ]]; then
		echo_yellow "Current HWID is $_hwid"
	fi

	echo_yellow "Are you sure you know what you're doing here?
Changing this is not normally needed, and if you mess it up,
MrChromebox is not going to help you fix it. This won't let
you run a different/newer version of ChromeOS.
Proceed at your own risk."

	read -rep "Really change your HWID? [y/N] " confirm
	[[ "$confirm" = "Y" || "$confirm" = "y" ]] || return

	read -rep "This is serious. Are you really sure? [y/N] " confirm
	[[ "$confirm" = "Y" || "$confirm" = "y" ]] || return

	read -rep "Enter a new HWID (use all caps): " hwid
	echo -e ""
	read -rep "Confirm changing HWID to $hwid [y/N] " confirm
	if [[ "$confirm" = "Y" || "$confirm" = "y" ]]; then
		echo_yellow "\nSetting hardware ID..."
		#disable software write-protect
		if ! ${flashromcmd} --wp-disable > /dev/null 2>&1; then
			exit_red "Error disabling software write-protect; unable to set HWID."; return 1
		fi
		[[ "$isChromeOS" = false ]] && FMAP="--fmap"
		if ! ${flashromcmd} -r $FMAP -i GBB:/tmp/gbb.temp > /dev/null 2>&1; then
			exit_red "\nError reading firmware (non-stock?); unable to set HWID."; return 1
		fi
		if ! ${gbbutilitycmd} --set --hwid="${hwid}" /tmp/gbb.temp > /dev/null 2>&1; then
			exit_red "\nError setting HWID."; return 1
		fi
		if ! ${flashromcmd} -w $FMAP -i GBB:/tmp/gbb.temp ${noverify} -o /tmp/flashrom.log > /dev/null 2>&1; then
			cat /tmp/flashrom.log
			exit_red "\nError writing back firmware; unable to set HWID."; return 1
		fi
		echo_green "Hardware ID successfully set."
	fi
	read -rep "Press [Enter] to return to the main menu."
}

##########################
# Set HWID for UEFI ROM #
##########################
function set_hwid_uefi()
{
	# set HWID using cbfstool for UEFI firmware
	# ensure hardware write protect disabled
	[[ "$wpEnabled" = true ]] && { exit_red "\nHardware write-protect enabled, cannot set HWID."; return 1; }

	echo_green "\nSet Hardware ID (HWID) for UEFI Firmware"

	# Get current HWID if present
	if ${cbfstoolcmd} /tmp/bios.bin extract -n hwid -f /tmp/hwid_current.txt >/dev/null 2>&1; then
		_current_hwid=$(cat /tmp/hwid_current.txt 2>/dev/null)
		if [[ -n "$_current_hwid" ]]; then
			echo_yellow "Current HWID is: $_current_hwid"
		fi
		rm -f /tmp/hwid_current.txt
	else
		echo_yellow "No current HWID found in firmware"
	fi

	echo_yellow "
WARNING: Changing HWID is not normally needed, and if you mess it up,
it could result in the wrong firmware being flashed, which will almost
certainly result in your device being bricked.
Proceed at your own risk."

	read -rep "Really change your HWID? [y/N] " confirm
	[[ "$confirm" = "Y" || "$confirm" = "y" ]] || return

	read -rep "This is serious. Are you really sure? [y/N] " confirm
	[[ "$confirm" = "Y" || "$confirm" = "y" ]] || return

	local hwid=""
	read -rep "Enter a new HWID: " hwid
	if [[ -z "$hwid" ]]; then
		exit_red "No HWID entered; operation cancelled."; return 1
	fi

	echo -e ""
	read -rep "Confirm changing HWID to '$hwid' [y/N] " confirm
	if [[ "$confirm" = "Y" || "$confirm" = "y" ]]; then
		echo_yellow "\nReading current firmware..."
		# Read current firmware to ensure we have the latest
		if ! ${flashromcmd} --fmap -i COREBOOT -r /tmp/bios_mod.bin > /tmp/flashrom.log 2>&1; then
			cat /tmp/flashrom.log
			exit_red "\nError reading firmware; unable to set HWID."; return 1
		fi

		echo_yellow "Modifying firmware..."
		# Remove old HWID if it exists
		${cbfstoolcmd} /tmp/bios_mod.bin remove -n hwid > /dev/null 2>&1

		# Create new HWID file
		echo -n "$hwid" > /tmp/hwid_new.txt

		# Add new HWID to CBFS
		if ! ${cbfstoolcmd} /tmp/bios_mod.bin add -n hwid -f /tmp/hwid_new.txt -t raw > /dev/null 2>&1; then
			exit_red "\nError adding HWID to firmware."; return 1
		fi

		# Disable software write-protect
		if ! ${flashromcmd} --wp-disable > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
			exit_red "Error disabling software write-protect; unable to set HWID."; return 1
		fi

		# Clear SW WP range
		if ! ${flashromcmd} --wp-range 0 0 > /dev/null 2>&1; then
			if ! ${flashromcmd} --wp-range 0,0 > /dev/null 2>&1 && [[ "$swWp" = "enabled" ]]; then
				exit_red "Error clearing software write-protect range; unable to set HWID."; return 1
			fi
		fi

		# Write firmware back
		echo_yellow "Writing firmware with new HWID..."
		if ! ${flashromcmd} --fmap -i COREBOOT -w /tmp/bios_mod.bin -N > /tmp/flashrom.log 2>&1; then
			if [ -f /tmp/flashrom.log ]; then
				cat /tmp/flashrom.log
			fi
			exit_red "\nError writing firmware; HWID not set. DO NOT REBOOT!"; return 1
		fi

		# Cleanup
		rm -f /tmp/hwid_new.txt /tmp/bios_mod.bin

		echo_green "Hardware ID successfully set. Reboot for the change to take effect."
	fi
	read -rep "Press [Enter] to return to the main menu."
}

###############
# Clear NVRAM #
###############
function clear_nvram() {
	echo_green "\nClear UEFI NVRAM"
	echo_yellow "Clearing the NVRAM will remove all EFI variables\nand reset the boot order to the default."

	read -rep "Would you like to continue? [y/N] "
	[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] || return

	echo_yellow "\nClearing NVRAM..."
	if ! ${flashromcmd} -E -i SMMSTORE --fmap > /tmp/flashrom.log 2>&1; then
		cat /tmp/flashrom.log
		exit_red "\nFailed to erase SMMSTORE firmware region; NVRAM not cleared."
		return 1;
	fi
	#all done
	echo_green "NVRAM has been cleared."
	read -rep "Press Enter to continue"
}

#############################
# Reset CR50 TPM NVRAM Data #
#############################
function reset_cr50_nvram() {
	local firmware_file="$1"
	
	if [[ "$hasCR50" != true ]]; then
		return 0
	fi

	echo_yellow "\nResetting CR50 TPM and kernel version data..."

	# Download tpmc tool if needed
	if ! get_tpmc; then
		echo_red "Unable to download tpmc utility; cannot reset CR50 NVRAM."
		return 1
	fi

	# Reset TPM data in CR50 NVRAM
	# First verify we can read from 0x1007 before writing
	if ! ${tpmccmd} read 0x1007 0xa >/dev/null 2>&1; then
		echo_red "Error: Failed to read from CR50 NVRAM index 0x1007."
		return 1
	fi
	
	# Write TPM reset command
	if ! ${tpmccmd} write 0x1007 02 02 01 00 01 00 00 00 00 69 >/dev/null 2>&1; then
		echo_red "Error: Failed to reset CR50 TPM data."
		return 1
	fi
	
	# Reset kernel version data in CR50 NVRAM
	# First verify we can read from 0x1008 before writing
	if ! ${tpmccmd} read 0x1008 0xa >/dev/null 2>&1; then
		echo_red "Error: Failed to read from CR50 NVRAM index 0x1008."
		return 1
	fi
	
	# Determine which command string to use based on FWID from config file
	if [[ -n "$firmware_file" && -f "$firmware_file" ]]; then
		# Extract config file from COREBOOT region
		if ${cbfstoolcmd} "${firmware_file}" extract -n config -f /tmp/config.txt >/dev/null 2>&1; then
			# Try to find FWID in the config
			fwid_line=$(grep -i "FWID" /tmp/config.txt 2>/dev/null | head -1)
			if [[ -n "$fwid_line" ]]; then
				# Extract major version (field 2 using . as delimiter)
				fwid_major=$(echo "$fwid_line" | cut -d'.' -f2)
				# Clean up any non-numeric characters
				fwid_major=$(echo "$fwid_major" | tr -cd '0-9')
				
				if [[ -n "$fwid_major" ]] && [[ "$fwid_major" -lt 12953 ]] 2>/dev/null; then
					# v0 secdata_kernel (< 12953)
					echo_yellow "Using v0 secdata_kernel format (FWID $fwid_major.$fwid_minor)"
					if ! ${tpmccmd} write 0x1008 02 4c 57 52 47 01 00 01 00 00 00 00 55 >/dev/null 2>&1; then
						echo_red "Error: Failed to reset CR50 kernel version data."
						return 1
					fi
				else
					# v1 secdata kernel (>= 12953)
					echo_yellow "Using v1 secdata_kernel format (FWID $fwid_major.$fwid_minor)"
					if ! ${tpmccmd} write 0x1008 10 28 0c 00 01 00 01 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 >/dev/null 2>&1; then
						echo_red "Error: Failed to reset CR50 kernel version data."
						return 1
					fi
				fi
			else
				echo_red "Error: Could not find FWID in config file."
				return 1
			fi
		else
			echo_red "Error: Failed to extract config file from COREBOOT region."
			return 1
		fi
	else
		echo_red "Error: Firmware file not available for FWID check."
		return 1
	fi

	echo_green "\nCR50 NVRAM reset completed."
	return 0
}

########################
# Firmware Update Menu #
########################
function menu_fwupdate() {
	if [[ "$isFullRom" = true ]]; then
		uefi_menu
	else
		stock_menu
	fi
}

function show_header() {
	printf "\ec"
	echo -e "${NORMAL}\n ChromeOS Device Firmware Utility Script ${script_date} ${NORMAL}"
	echo -e "${NORMAL} (c) Mr Chromebox <mrchromebox@gmail.com> ${NORMAL}"
	echo -e "${MENU}*********************************************************${NORMAL}"
	echo -e "${MENU}**${NUMBER}     Device: ${NORMAL}${deviceDesc}"
	echo -e "${MENU}**${NUMBER} Board Name: ${NORMAL}${boardName^^}"
	echo -e "${MENU}**${NUMBER}   Platform: ${NORMAL}$deviceCpuTypeName"
	echo -e "${MENU}**${NUMBER}    Fw Type: ${NORMAL}$firmwareType"
	echo -e "${MENU}**${NUMBER}     Fw Ver: ${NORMAL}$fwVer ($fwDate)"
	if [[ $isUEFI = true && $hasUEFIoption = true ]]; then
		# check if update available
		curr_yy=$(echo $fwDate | cut -f 3 -d '/')
		curr_mm=$(echo $fwDate | cut -f 1 -d '/')
		curr_dd=$(echo $fwDate | cut -f 2 -d '/')
		eval coreboot_file=$`echo "coreboot_uefi_${device}"`
		date=$(echo $coreboot_file | grep -o "mrchromebox.*" | cut -f 2 -d '_' | cut -f 1 -d '.')
		uefi_yy=$(echo $date | cut -c1-4)
		uefi_mm=$(echo $date | cut -c5-6)
		uefi_dd=$(echo $date | cut -c7-8)
		if [[ ("$firmwareType" != *"pending"*) && (($uefi_yy > $curr_yy) || \
			("$uefi_yy" = "$curr_yy" && "$uefi_mm" > "$curr_mm") || \
			("$uefi_yy" = "$curr_yy" && "$uefi_mm" = "$curr_mm" && "$uefi_dd" > "$curr_dd")) ]]; then
			echo -e "${MENU}**${NORMAL}             ${GREEN_TEXT}Update Available ($uefi_mm/$uefi_dd/$uefi_yy)${NORMAL}"
		fi
	fi
	if [ "$wpEnabled" = true ]; then
		echo -e "${MENU}**${NUMBER}      Fw WP: ${RED_TEXT}Enabled${NORMAL}"
		WP_TEXT=${RED_TEXT}
	else
		echo -e "${MENU}**${NUMBER}      Fw WP: ${NORMAL}Disabled"
		WP_TEXT=${GREEN_TEXT}
	fi
	echo -e "${MENU}*********************************************************${NORMAL}"
}

function stock_menu() {

	show_header

	if [[ "$unlockMenu" = true || ( "$isFullRom" = false && "$isStock" = true && "$isUnsupported" = false \
			&& ("$isCmlBook" = false || "$device" == "drallion") && "$isEOL" = false ) ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} 1)${MENU} Install/Update RW_LEGACY Firmware ${NORMAL}"
	else
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 1)${GRAY_TEXT} Install/Update RW_LEGACY Firmware ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || "$hasUEFIoption" = true ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 2)${MENU} Install/Update UEFI (Full ROM) Firmware ${NORMAL}"
	else
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 2)${GRAY_TEXT} Install/Update UEFI (Full ROM) Firmware${NORMAL}"
	fi
	if [[ "${device^^}" = "EVE" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} D)${MENU} Downgrade Touchpad Firmware ${NORMAL}"
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} U)${MENU} Upgrade Touchpad Firmware ${NORMAL}"
	fi
	if [[ "$isJsl" = true &&  "$device" =~ "gal" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} T)${MENU} Set Touchpad Type in SSFC ${NORMAL}"
	fi
	if [[ "${device^^}" = "TAEKO" || "${device^^}" = "TANIKS" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} S)${MENU} Set Storage Type in FW_CONFIG ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || ("$isFullRom" = false && "$isStock" = true) ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 3)${MENU} Set Boot Options (GBB flags) ${NORMAL}"
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 4)${MENU} Set Hardware ID (HWID) ${NORMAL}"
	else
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 3)${GRAY_TEXT} Set Boot Options (GBB flags)${NORMAL}"
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 4)${GRAY_TEXT} Set Hardware ID (HWID) ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || "$isStock" = true ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} 5)${MENU} Backup Current Firmware ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || ( "$isChromeOS" = false  && "$isFullRom" = true ) ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 6)${MENU} Restore Stock Firmware (full) ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || "$isUEFI" = true ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} C)${MENU} Clear UEFI NVRAM ${NORMAL}"
	fi
	echo -e "${MENU}*********************************************************${NORMAL}"
	echo -e "${ENTER_LINE}Select a numeric menu option or${NORMAL}"
	echo -e "${nvram}${RED_TEXT}R${NORMAL} to reboot ${NORMAL} ${RED_TEXT}P${NORMAL} to poweroff ${NORMAL} ${RED_TEXT}Q${NORMAL} to quit ${NORMAL}"

	read -re opt
	case $opt in

	1)	if [[ "$unlockMenu" = true || ( "$isFullRom" = false && "$isStock" = true && "$isUnsupported" = false \
				&& ("$isCmlBook" = false || "$device" == "drallion") && "$isEOL" = false ) ]]; then
			flash_rwlegacy
		elif [[ "$isEOL" = "true" ]]; then
			echo_red "The RW_LEGACY firmware update is not supported for devices which have reached end-of-life"
			read -rep "Press enter to return to the main menu"
		fi
		menu_fwupdate
		;;

	2)	if [[ "$unlockMenu" = true || "$hasUEFIoption" = true ]]; then
			flash_full_rom
		fi
			menu_fwupdate
		;;

	[dD])	if [[ "${device^^}" = "EVE" ]]; then
			downgrade_touchpad_fw
		fi
		menu_fwupdate
		;;

	[uU])	if [[ "${device^^}" = "EVE" ]]; then
			upgrade_touchpad_fw
		fi
		menu_fwupdate
		;;

	[tT])	if [[ "$isJsl" = true &&  "$device" =~ "gal" ]]; then
			set_touchpad_in_ssfc
		fi
		menu_fwupdate
		;;

	[sS])	if [[ "${device^^}" = "TAEKO" || "${device^^}" = "TANIKS" ]]; then
			set_storage_in_fw_config
		fi
		menu_fwupdate
		;;

	3)	if [[ "$unlockMenu" = true || "$isChromeOS" = true || "$isUnsupported" = false \
				&& "$isFullRom" = false && "$isStock" = true ]]; then
			set_boot_options
		fi
		menu_fwupdate
		;;

	4)	if [[ "$unlockMenu" = true || "$isChromeOS" = true || "$isUnsupported" = false \
				&& "$isFullRom" = false && "$isStock" = true ]]; then
			set_hwid
		fi
		menu_fwupdate
		;;

	5)	if [[ "$unlockMenu" = true || "$isStock" = true ]]; then
			backup_current_firmware
		fi
		menu_fwupdate
		;;

	6)	if [[ "$unlockMenu" = true || "$isChromeOS" = false && "$isUnsupported" = false \
				&& "$isFullRom" = true ]]; then
			restore_stock_firmware
		fi
		menu_fwupdate
		;;

	[rR])	echo -e "\nRebooting...\n";
		cleanup
		reboot
		exit
		;;

	[pP])	echo -e "\nPowering off...\n";
		cleanup
		poweroff
		exit
		;;

	[qQ])	cleanup;
		exit;
		;;

	[lL])	if [ "$unlockMenu" = false ]; then
			echo_yellow "\nAre you sure you wish to unlock all menu functions?"
			read -rep "Only do this if you really know what you are doing... [y/N]? "
			[[ "$REPLY" = "y" || "$REPLY" = "Y" ]] && unlockMenu=true
		fi
		menu_fwupdate
		;;

	[cC])	if [[ "$unlockMenu" = true || "$isUEFI" = true ]]; then
			clear_nvram
		fi
		menu_fwupdate
		;;

	*)	clear
		menu_fwupdate;
		;;

	esac
}


unlockMenu=false

function uefi_menu() {

	show_header

	if [[ "$hasUEFIoption" = true && "$isUnsupported" = false ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 1)${MENU} Install/Update UEFI (Full ROM) Firmware ${NORMAL}"
	else
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 1)${GRAY_TEXT} Install/Update UEFI (Full ROM) Firmware${NORMAL}"
	fi
	if [[ "$isChromeOS" = false  && "$isFullRom" = true && "$isUnsupported" = false ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 2)${MENU} Restore Stock Firmware ${NORMAL}"
	else
		echo -e "${GRAY_TEXT}**     ${GRAY_TEXT} 2)${GRAY_TEXT} Restore Stock ChromeOS Firmware ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || ("$isUEFI" = true && "$isUnsupported" = false) ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} 3)${MENU} Backup Current Firmware ${NORMAL}"
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} 4)${MENU} Flash Custom Firmware ${NORMAL}"
	fi
	if [[ "$isUEFI" = true ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} 5)${MENU} Set Hardware ID (HWID) ${NORMAL}"
	fi
	if [[ "${device^^}" = "EVE" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} D)${MENU} Downgrade Touchpad Firmware ${NORMAL}"
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} U)${MENU} Upgrade Touchpad Firmware ${NORMAL}"
	fi
	if [[ "$isJsl" = true &&  "$device" =~ "gal" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} T)${MENU} Set Touchpad Type in SSFC ${NORMAL}"
	fi
	if [[ "${device^^}" = "TAEKO" || "${device^^}" = "TANIKS" ]]; then
		echo -e "${MENU}**${WP_TEXT} [WP]${NUMBER} S)${MENU} Set Storage Type in FW_CONFIG ${NORMAL}"
	fi
	if [[ "$unlockMenu" = true || ("$isUEFI" = true && "$isUnsupported" = false) ]]; then
		echo -e "${MENU}**${WP_TEXT}     ${NUMBER} C)${MENU} Clear UEFI NVRAM ${NORMAL}"
	fi
	echo -e "${MENU}*********************************************************${NORMAL}"
	echo -e "${ENTER_LINE}Select a numeric menu option or${NORMAL}"
	echo -e "${nvram}${RED_TEXT}R${NORMAL} to reboot ${NORMAL} ${RED_TEXT}P${NORMAL} to poweroff ${NORMAL} ${RED_TEXT}Q${NORMAL} to quit ${NORMAL}"

	read -re opt
	case $opt in

	1)	if [[ "$hasUEFIoption" = true && "$isUnsupported" = false ]]; then
			flash_full_rom
		fi
		uefi_menu
		;;

	2)	if [[ "$isChromeOS" = false && "$isUnsupported" = false && "$isFullRom" = true ]]; then
			restore_stock_firmware
			menu_fwupdate
		else
			uefi_menu
		fi
		;;

	3)	if [[ "$unlockMenu" = true || ("$isUEFI" = true && "$isUnsupported" = false) ]]; then
			backup_current_firmware
		fi
		uefi_menu
		;;

	4)	if [[ "$unlockMenu" = true || ("$isUEFI" = true && "$isUnsupported" = false) ]]; then
			flash_custom_firmware
		fi
		uefi_menu
		;;

	5)	if [[ "$isUEFI" = true ]]; then
			set_hwid_uefi
		fi
		uefi_menu
		;;

	[dD])	if [[  "${device^^}" = "EVE" ]]; then
			downgrade_touchpad_fw
		fi
		uefi_menu
		;;

	[uU])	if [[  "${device^^}" = "EVE" ]]; then
			upgrade_touchpad_fw
		fi
		uefi_menu
		;;

	[tT])	if [[ "$isJsl" = true &&  "$device" =~ "gal" ]]; then
			set_touchpad_in_ssfc
		fi
		uefi_menu
		;;

	[sS])	if [[ "${device^^}" = "TAEKO" || "${device^^}" = "TANIKS" ]]; then
			set_storage_in_fw_config
		fi
		uefi_menu
		;;

	[rR])	echo -e "\nRebooting...\n";
		cleanup
		reboot
		exit
		;;

	[pP])	echo -e "\nPowering off...\n";
		cleanup
		poweroff
		exit
		;;

	[qQ])  cleanup;
	exit;
	;;

	[cC])	if [[ "$isUEFI" = true && "$isUnsupported" = false ]]; then
			clear_nvram
		fi
		uefi_menu
		;;

	*)	clear
		uefi_menu;
		;;

	esac
}