#!/bin/bash
#
set -e -o pipefail
. /etc/functions
. /etc/gui_functions
. /tmp/config

TRACE_FUNC

ROOT_HASH_FILE="/boot/kexec_root_hashes.txt"

param=$1

# Read the current ROM; if it fails display an error and exit.
read_rom() {
	/bin/flash.sh -r "$1"
	if [ ! -s "$1" ]; then
		whiptail_error --title 'ERROR: BIOS Read Failed!' \
			--msgbox "Unable to read BIOS" 0 80
		exit 1
	fi
}

while true; do
	if [ ! -z "$param" ]; then
		# use first char from parameter
		menu_choice=${param::1}
		unset param
	else
		# Re-source config because we change it when an option is toggled
		. /tmp/config

		dynamic_config_options=(
			'b' ' Change the /boot device'
		)

		# Options that don't apply to basic mode
		[ "$CONFIG_BASIC" != "y" ] && dynamic_config_options+=(
			'r' ' Clear GPG key(s) and reset all user settings'
			'R' ' Change the root device for hashing'
			'D' ' Change the root directories to hash'
			'B' " $(get_config_display_action "$CONFIG_ROOT_CHECK_AT_BOOT") root check at boot"
			'L' " $(get_config_display_action "$CONFIG_RESTRICTED_BOOT") Restricted Boot"
		)

		# Basic itself is always available (though RB will refuse to enable it)
		dynamic_config_options+=(
			'P' " $(get_config_display_action "$CONFIG_BASIC") $CONFIG_BRAND_NAME Basic Mode"
		)

		# Blob jail is only offered if this is a configuration with the blobs in
		# firmware
		[ "$CONFIG_SUPPORT_BLOB_JAIL" = "y" ] && dynamic_config_options+=(
			'J' " $(get_config_display_action "$CONFIG_USE_BLOB_JAIL") Firmware Blob Jail"
		)

		# Automatic boot
		dynamic_config_options+=(
			'M' " Configure automatic boot"
		)

		# Basic-only options for automatic boot
		[ "$CONFIG_BASIC" = "y" ] && dynamic_config_options+=(
			'A' " $(get_inverted_config_display_action "$CONFIG_BASIC_NO_AUTOMATIC_DEFAULT") automatic default boot option"
			'U' " $(get_config_display_action "$CONFIG_BASIC_USB_AUTOBOOT") USB automatic boot"
		)

		# Automatic power on - requires board support
		[ "$CONFIG_SUPPORT_AUTOMATIC_POWERON" = "y" ] && dynamic_config_options+=(
			'N' " $(get_config_display_action "$CONFIG_AUTOMATIC_POWERON") automatic power-on"
		)

		# Boards with built-in keyboards can support optional USB keyboards as well.
		# Boards that do not have a built-in keyboard/internal keyboard is USB connected export
		# CONFIG_USB_KEYBOARD_REQUIRED=y; this hides the config option and ensures
		# USB keyboard support always loads.
		[ "$CONFIG_USB_KEYBOARD_REQUIRED" != y ] && dynamic_config_options+=(
			'K' " $(get_config_display_action "$CONFIG_USER_USB_KEYBOARD") USB keyboard"
		)

		# Debugging option always available
		dynamic_config_options+=(
			'Z' " Configure $CONFIG_BRAND_NAME informational / debug output"
		)

		[ "$CONFIG_FINALIZE_PLATFORM_LOCKING" = "y" ] && dynamic_config_options+=(
			't' ' Deactivate Platform Locking to permit OS write access to firmware'
		)

		dynamic_config_options+=(
			's' ' Save the current configuration to the running BIOS'
			'x' ' Return to Main Menu'
		)

		unset menu_choice
		whiptail_type $BG_COLOR_MAIN_MENU --title "Config Management Menu" \
			--menu "This menu lets you change settings for the current BIOS session.\n\nAll changes will revert after a reboot,\n\nunless you also save them to the running BIOS." 0 80 10 \
			"${dynamic_config_options[@]}" \
			2>/tmp/whiptail || recovery "GUI menu failed"

		menu_choice=$(cat /tmp/whiptail)
	fi

	case "$menu_choice" in
	"t")
		unset CONFIG_FINALIZE_PLATFORM_LOCKING
		replace_config /etc/config.user "CONFIG_FINALIZE_PLATFORM_LOCKING" "n"
		combine_configs
		. /tmp/config
		;;
	"x")
		exit 0
		;;
	"b")
		CURRENT_OPTION="$(load_config_value CONFIG_BOOT_DEV)"
		if ! fdisk -l 2>/dev/null | grep "Disk /dev/" | cut -f2 -d " " | cut -f1 -d ":" >/tmp/disklist.txt; then
			whiptail_error --title 'ERROR: No bootable devices found' \
				--msgbox "    $ERROR\n\n" 0 80
			exit 1
		fi
		# filter out extraneous options
		>/tmp/boot_device_list.txt
		for i in $(cat /tmp/disklist.txt); do
			# remove block device from list if numeric partitions exist, since not bootable
			DEV_NUM_PARTITIONS=$(($(ls -1 $i* | wc -l) - 1))
			if [ ${DEV_NUM_PARTITIONS} -eq 0 ]; then
				echo $i >>/tmp/boot_device_list.txt
			else
				ls $i* | tail -${DEV_NUM_PARTITIONS} >>/tmp/boot_device_list.txt
			fi
		done
		file_selector "/tmp/boot_device_list.txt" \
			"Choose the default /boot device.\n\n${CURRENT_OPTION:+\n\nCurrently set to }$CURRENT_OPTION." \
			"Boot Device Selection"
		if [ "$FILE" == "" ]; then
			return
		else
			SELECTED_FILE=$FILE
		fi

		# unmount /boot if needed
		if grep -q /boot /proc/mounts; then
			umount /boot 2>/dev/null
		fi
		# mount newly selected /boot device
		if ! mount -o ro $SELECTED_FILE /boot 2>/tmp/error; then
			ERROR=$(cat /tmp/error)
			whiptail_error --title 'ERROR: unable to mount /boot' \
				--msgbox "    $ERROR\n\n" 0 80
			exit 1
		fi

		set_config /etc/config.user "CONFIG_BOOT_DEV" "$SELECTED_FILE"
		combine_configs

		whiptail --title 'Config change successful' \
			--msgbox "The /boot device was successfully changed to $SELECTED_FILE" 0 80
		;;
	"s")
		read_rom /tmp/config-gui.rom

		replace_rom_file /tmp/config-gui.rom "heads/initrd/etc/config.user" /etc/config.user

		if (whiptail --title 'Update ROM?' \
			--yesno "This will reflash your BIOS with the updated version\n\nDo you want to proceed?" 0 80); then
			/bin/flash.sh /tmp/config-gui.rom
			whiptail --title 'BIOS Updated Successfully' \
				--msgbox "BIOS updated successfully.\n\nIf your keys have changed, be sure to re-sign all files in /boot\nafter you reboot.\n\nPress Enter to reboot" 0 80
			/bin/reboot
		else
			exit 0
		fi
		;;
	"r")
		# prompt for confirmation
		if (whiptail_warning --title 'Reset Configuration?' \
			--yesno "This will clear all GPG keys, clear boot signatures and checksums,
                  \nreset the /boot device, clear/reset the TPM (if present),
                  \nand reflash your BIOS with the cleaned configuration.
                  \n\nDo you want to proceed?" 0 80); then
			read_rom /tmp/config-gui.rom
			# clear local keyring
			rm -rf /.gnupg/* || true

			# clear /boot signatures/checksums
			detect_boot_device
			mount -o remount,rw /boot
			rm -f /boot/kexec* || true
			mount -o remount,ro /boot

			# clear GPG keys and user settings
			for i in $(cbfs.sh -o /tmp/config-gui.rom -l | grep -e "heads/"); do
				cbfs.sh -o /tmp/config-gui.rom -d $i
			done
			# flash cleared ROM
			/bin/flash.sh -c /tmp/config-gui.rom

			# reset TPM if present
			if [ "$CONFIG_TPM" = "y" ]; then
				/bin/tpm-reset
			fi
			whiptail --title 'Configuration Reset Updated Successfully' \
				--msgbox "Configuration reset and BIOS updated successfully.\n\nPress Enter to reboot" 0 80
			/bin/reboot
		else
			exit 0
		fi
		;;
	"R")
		CURRENT_OPTION="$(load_config_value CONFIG_ROOT_DEV)"
		fdisk -l 2>/dev/null | grep "Disk /dev/" | cut -f2 -d " " | cut -f1 -d ":" >/tmp/disklist.txt
		# filter out extraneous options
		>/tmp/root_device_list.txt
		for i in $(cat /tmp/disklist.txt); do
			# remove block device from list if numeric partitions exist, since not bootable
			DEV_NUM_PARTITIONS=$(($(ls -1 $i* | wc -l) - 1))
			if [ ${DEV_NUM_PARTITIONS} -eq 0 ]; then
				echo $i >>/tmp/root_device_list.txt
			else
				ls $i* | tail -${DEV_NUM_PARTITIONS} >>/tmp/root_device_list.txt
			fi
		done
		file_selector "/tmp/root_device_list.txt" \
			"Choose the default root device.${CURRENT_OPTION:+\n\nCurrently set to }$CURRENT_OPTION." \
			"Root Device Selection"
		if [ "$FILE" == "" ]; then
			break
		else
			SELECTED_FILE=$FILE
		fi

		set_config /etc/config.user "CONFIG_ROOT_DEV" "$SELECTED_FILE"
		combine_configs

		whiptail --title 'Config change successful' \
			--msgbox "The root device was successfully changed to $SELECTED_FILE" 0 80
		;;
	"D")
		CURRENT_OPTION="$(load_config_value CONFIG_ROOT_DIRLIST)"

		# Separate from prior prompt history on the terminal with two blanks
		echo -e "\n"

		if [ -n "$CURRENT_OPTION" ]; then
			echo -e "The current list of directories to hash is $CURRENT_OPTION"
		fi
		echo -e "Enter the new list of directories separated by spaces:"
		echo -e "(Press enter with the list empty to cancel)"
		read -r NEW_CONFIG_ROOT_DIRLIST

		# strip any leading forward slashes
		NEW_CONFIG_ROOT_DIRLIST=$(echo $NEW_CONFIG_ROOT_DIRLIST | sed -e 's/^\///;s/ \// /g')

		#check if list empty
		if [ -z "$NEW_CONFIG_ROOT_DIRLIST" ]; then
			whiptail --title 'Config change canceled' \
				--msgbox "Root device directory change canceled by user" 0 80
			break
		fi

		set_config /etc/config.user "CONFIG_ROOT_DIRLIST" "$NEW_CONFIG_ROOT_DIRLIST"
		combine_configs

		whiptail --title 'Config change successful' \
			--msgbox "The root directories to hash was successfully changed to:\n$NEW_CONFIG_ROOT_DIRLIST" 0 80
		;;
	"B")
		if [ "$CONFIG_ROOT_CHECK_AT_BOOT" != "y" ]; then
			# Root device and directories must be set to enable this
			if [ -z "$CONFIG_ROOT_DEV" ] || [ -z "$CONFIG_ROOT_DIRLIST" ]; then
				whiptail_error --title 'Root Check Not Configured' \
					--msgbox "Set the root device and directories to hash before enabling this feature." 0 80
			elif (whiptail --title 'Enable Root Hash Check at Boot?' \
				--yesno "This will enable checking root hashes each time you boot.
                    \nDepending on the directories you are checking, this might add
                    \na minute or more to the boot time.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_ROOT_CHECK_AT_BOOT" "y"

				# check that root hash file exists
				if [ ! -f ${ROOT_HASH_FILE} ]; then
					if (whiptail --title 'Generate Root Hash File' \
						--yesno "\nNo root hash file exists.
                        \nWould you like to create the initial hash file now?" 0 80); then
						root-hashes-gui.sh -n
					fi
				fi

				whiptail --title 'Config change successful' \
					--msgbox "The root device will be checked at each boot." 0 80

			fi
		else
			if (whiptail --title 'Disable Root Hash Check at Boot?' \
				--yesno "This will disable checking root hashes each time you boot.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_ROOT_CHECK_AT_BOOT" "n"

				whiptail --title 'Config change successful' \
					--msgbox "The root device will not be checked at each boot." 0 80
			fi
		fi
		;;
	"P")
		if [ "$CONFIG_RESTRICTED_BOOT" = "y" ]; then
			whiptail_error --title 'Restricted Boot Active' \
				--msgbox "Disable Restricted Boot to enable Basic Mode." 0 80
		elif [ "$CONFIG_BASIC" != "y" ]; then
			if (whiptail --title "Enable $CONFIG_BRAND_NAME Basic Mode?" \
				--yesno "This will remove all signature checking on the firmware
                    \nand boot files, and disable use of the Librem Key.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC" "y"

				whiptail --title 'Config change successful' \
					--msgbox "$CONFIG_BRAND_NAME Basic mode enabled;\nsave the config change and reboot for it to go into effect." 0 80

			fi
		else
			if (whiptail --title "Disable $CONFIG_BRAND_NAME Basic Mode?" \
				--yesno "This will enable all signature checking on the firmware
                    \nand boot files, and enable use of the Librem Key.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC" "n"

				whiptail --title 'Config change successful' \
					--msgbox "$CONFIG_BRAND_NAME Basic mode has been disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"L")
		if [ "$CONFIG_RESTRICTED_BOOT" != "y" ]; then
			if (whiptail --title 'Enable Restricted Boot Mode?' \
				--yesno "Restricted Boot allows booting:
                    \n* Signed installed OS
                    \n* Signed ISOs from USB
                    \nAll other boot methods are blocked.  Recovery console and firmware updates
                    \nwill be blocked.
                    \nRestricted boot can be disabled at any time.  This resets TOTP/HOTP so it
                    \nis evident that Restricted Boot was disabled.
                    \n
                    \nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_RESTRICTED_BOOT" "y"

				whiptail --title 'Config change successful' \
					--msgbox "Restricted Boot mode enabled;\nsave the config change and reboot for it to go into effect." 0 80

			fi
		else
			if (whiptail --title 'Disable Restricted Boot Mode?' \
				--yesno "This will re-enable all boot methods, the recovery console, and firmware
                    \nupdates.
                    \nThis will also erase the TOTP/HOTP secret.
                    \nProceeding will automatically update the boot firmware and reboot!
                    \n\nDo you want to proceed?" 0 80); then

				# Wipe the TPM TOTP/HOTP secret before flashing.  Otherwise, enabling
				# Restricted Boot again might restore the firmware to an identical
				# state, and there would be no evidence that it had been temporarily
				# disabled.
				if ! wipe-totp >/dev/null 2>/tmp/error; then
					ERROR=$(tail -n 1 /tmp/error | fold -s)
					whiptail_error --title 'ERROR: erasing TOTP secret' \
						--msgbox "Erasing TOTP Secret Failed\n\n${ERROR}" 0 80
					exit 1
				fi

				# We can't allow Restricted Boot to be disabled without flashing the
				# firmware - this would allow the use of unrestricted mode without
				# leaving evidence in the firmware.  Disable it by flashing the new
				# config directly.
				FLASH_USER_CONFIG=/tmp/config-gui-config-user
				cp /etc/config.user "$FLASH_USER_CONFIG"
				set_config "$FLASH_USER_CONFIG" "CONFIG_RESTRICTED_BOOT" "n"

				read_rom /tmp/config-gui.rom

				replace_rom_file /tmp/config-gui.rom "heads/initrd/etc/config.user" "$FLASH_USER_CONFIG"

				/bin/flash.sh /tmp/config-gui.rom
				whiptail --title 'BIOS Updated Successfully' \
					--msgbox "BIOS updated successfully.\n\nIf your keys have changed, be sure to re-sign all files in /boot\nafter you reboot.\n\nPress Enter to reboot" 0 80
				/bin/reboot
			fi
		fi
		;;
	"J")
		if [ "$CONFIG_USE_BLOB_JAIL" != "y" ]; then
			if (whiptail --title 'Enable Firmware Blob Jail?' \
				--yesno "This will enable loading of firmware from flash on each boot
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_USE_BLOB_JAIL" "y"

				whiptail --title 'Config change successful' \
					--msgbox "Firmware Blob Jail use has been enabled;\nsave the config change and reboot for it to go into effect." 0 80

			fi
		else
			if (whiptail --title 'Disable Firmware Blob Jail?' \
				--yesno "This will disable loading of firmware from flash on each boot.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_USE_BLOB_JAIL" "n"

				whiptail --title 'Config change successful' \
					--msgbox "Firmware Blob Jail use has been disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"M")
		if [ -z "$CONFIG_AUTO_BOOT_TIMEOUT" ]; then
			current_msg="Automatic boot is currently disabled."
		elif [ "$CONFIG_AUTO_BOOT_TIMEOUT" = 1 ]; then
			current_msg="Currently boots automatically after 1 second."
		else
			current_msg="Currently boots automatically after $CONFIG_AUTO_BOOT_TIMEOUT seconds."
		fi
		whiptail --title "Automatic Boot" \
			--menu "$CONFIG_BRAND_NAME can boot automatically.  Select the amount of time to wait\nbefore booting.\n\n$current_msg" 0 80 10 \
			"0" "Don't boot automatically" \
			"1" "1 second" \
			"5" "5 seconds" \
			"10" "10 seconds" \
			"C" "Cancel" \
			2>/tmp/whiptail
		new_setting="$(cat /tmp/whiptail)"
		if ! [ "$new_setting" = "C" ]; then
			if [ "$new_setting" = "0" ]; then
				new_setting= # Empty disables automatic boot
				current_msg="$CONFIG_BRAND_NAME will not boot automatically."
			elif [ "$new_setting" = "1" ]; then
				current_msg="$CONFIG_BRAND_NAME will boot automatically after 1 second."
			else
				current_msg="$CONFIG_BRAND_NAME will boot automatically after $new_setting seconds."
			fi
			set_user_config "CONFIG_AUTO_BOOT_TIMEOUT" "$new_setting"
			whiptail --title 'Config change successful' \
				--msgbox "$current_msg\nSave the config change and reboot for it to go into effect." 0 80
		fi
		;;
	"A")
		if [ "$CONFIG_BASIC_NO_AUTOMATIC_DEFAULT" != "y" ]; then
			if (whiptail --title 'Disable automatic default boot?' \
				--yesno "You will need to select a default boot option.
                    \nIf the boot options are changed, such as for an OS update,
                    \nyou will be prompted to select a new default.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC_NO_AUTOMATIC_DEFAULT" "y"

				whiptail --title 'Config change successful' \
					--msgbox "Automatic default boot disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		else
			if (whiptail --title 'Enable automatic default boot?' \
				--yesno "The first boot option will be used automatically.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC_NO_AUTOMATIC_DEFAULT" "n"

				whiptail --title 'Config change successful' \
					--msgbox "Automatic default boot enabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"U")
		if [ "$CONFIG_BASIC_USB_AUTOBOOT" != "y" ]; then
			if (whiptail --title 'Enable USB automatic boot?' \
				--yesno "During boot, an attached bootable USB disk will be booted
                    \nby default instead of the installed operating system.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC_USB_AUTOBOOT" "y"

				whiptail --title 'Config change successful' \
					--msgbox "USB automatic boot enabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		else
			if (whiptail --title 'Disable USB automatic boot?' \
				--yesno "USB disks will no longer be booted by default.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_BASIC_USB_AUTOBOOT" "n"

				whiptail --title 'Config change successful' \
					--msgbox "USB automatic boot disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"N")
		if [ "$CONFIG_AUTOMATIC_POWERON" != "y" ]; then
			if (whiptail --title 'Enable automatic power-on?' \
				--yesno "The system will boot automatically when power is applied.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_AUTOMATIC_POWERON" "y"

				whiptail --title 'Config change successful' \
					--msgbox "Automatic power-on enabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		else
			if (whiptail --title 'Disable automatic power-on?' \
				--yesno "The system will stay off when power is applied.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_AUTOMATIC_POWERON" "n"

				# Disable the EC BRAM setting too, otherwise it persists until
				# manually disabled.  On the off chance the user does not actually
				# flash this change, we'll enable it again during boot.
				set_ec_poweron.sh n

				whiptail --title 'Config change successful' \
					--msgbox "Automatic power-on disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"K")
		if [ "$CONFIG_USER_USB_KEYBOARD" != "y" ]; then
			if (whiptail --title 'Enable USB Keyboard?' \
				--yesno "USB keyboards will be usable in $CONFIG_BRAND_NAME.
                    \n\nEnabling USB keyboards could allow a compromised USB device to control
                    \n$CONFIG_BRAND_NAME.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_USER_USB_KEYBOARD" "y"

				whiptail --title 'Config change successful' \
					--msgbox "USB Keyboard support has been enabled;\nsave the config change and reboot for it to go into effect." 0 80

			fi
		else
			if (whiptail --title 'Disable USB Keyboard?' \
				--yesno "Only the built-in keyboard will be usable in $CONFIG_BRAND_NAME.
                    \n\nDo you want to proceed?" 0 80); then

				set_user_config "CONFIG_USER_USB_KEYBOARD" "n"

				whiptail --title 'Config change successful' \
					--msgbox "USB Keyboard support has been disabled;\nsave the config change and reboot for it to go into effect." 0 80
			fi
		fi
		;;
	"Z")
		unset output_choice
		whiptail_type $BG_COLOR_MAIN_MENU --title "Informational / Debug Output" \
			--menu "$CONFIG_BRAND_NAME can display informational or debug output.\n\nChoose the output level:" 0 80 10 \
			0 'None - Show no extra output' \
			1 "Info - Show information about operations in $CONFIG_BRAND_NAME" \
			2 "Debug - Show detailed information suitable for debugging $CONFIG_BRAND_NAME" \
			2>/tmp/whiptail || recovery "GUI menu failed"

		output_choice=$(cat /tmp/whiptail)
		case "$output_choice" in
			0)
				set_user_config "CONFIG_DEBUG_OUTPUT" "n"
				set_user_config "CONFIG_ENABLE_FUNCTION_TRACING_OUTPUT" "n"
				set_user_config "CONFIG_QUIET_MODE" "y"
				;;
			1)
				set_user_config "CONFIG_DEBUG_OUTPUT" "n"
				set_user_config "CONFIG_ENABLE_FUNCTION_TRACING_OUTPUT" "n"
				set_user_config "CONFIG_QUIET_MODE" "n"
				;;
			2)
				set_user_config "CONFIG_DEBUG_OUTPUT" "y"
				set_user_config "CONFIG_ENABLE_FUNCTION_TRACING_OUTPUT" "y"
				set_user_config "CONFIG_QUIET_MODE" "n"
				;;
		esac
		whiptail --title 'Config change successful' \
			--msgbox "Output level changed.\nSave the config change and reboot for it to go into effect." 0 80
		;;
	esac
done
exit 0