Pass any extra command line arguments

[people/ms/firmware-update.git] / src / firmware-update.in

#!/bin/bash
###############################################################################
#                                                                             #
# firmware-update - IPFire Firmware Update Tool                               #
# Copyright (C) 2019 IPFire Development Team                                  #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

shopt -s nullglob

PACKAGE_NAME="@PACKAGE_NAME@"
PACKAGE_VERSION="@PACKAGE_VERSION@"

FIRMWARE_DIR="@firmwaredir@"

get_consent() {
	local firmware="${1}"

	cat <<EOF

You are going to flash the firmware on device. This is a dangerous operation
that might potentially damage your device.

Please do NOT TURN OFF the system while the update is in progress and do
not abort the process.

  Firmware file: ${firmware}

EOF

	local answer
	while [ -t 0 -a -t 1 -a -t 2 ]; do
		# Wait for user to type y
		printf "Do you want to proceed? [y/n] "
		read -r answer

		case "${answer}" in
			y|Y)
				return 0
				;;
			n|N)
				echo "Aborting."
				return 1
				;;
		esac
	done
}

read_dmi() {
	local what="${1}"

	# Return an error when what is empty
	[ -z "${what}" ] && return 2

	local file="/sys/class/dmi/id/${what}"

	# Read file
	if [ -r "${file}" ]; then
		printf "%s" "$(<${file})"
		return 0
	fi

	# File could not be read
	return 1
}

board_vendor() {
	read_dmi "board_vendor"
}

board_name() {
	read_dmi "board_name"
}

board_version() {
	read_dmi "board_version"
}

board_serial() {
	read_dmi "board_serial"
}

bios_version() {
	read_dmi "bios_version"
}

bios_date() {
	read_dmi "bios_date"
}

do_flashrom() {
	echo "Flashing process is starting now..."
	echo "PLEASE DO NOT TURN OFF THE SYSTEM WHILE FLASHING!"

	if ! flashrom "$@"; then
		echo "Error: Flashing the firmware was not successful" >&2
		return 1
	fi

	echo "Flashing was successful! Please reboot."
	return 0
}

find_firmware() {
	local path="${FIRMWARE_DIR}/${1}"

	# Expand globbing
	local files=( ${path} )

	# Return file with highest order
	local file="${files[-1]}"

	# Check if we can read the file
	if [ ! -r "${file}" ]; then
		echo "Error: Cannot read firmware file: ${file}" >&2
		return 1
	fi

	echo "${file}"
	return 0
}

# Returns true if a is greater than b
version_is_gt() {
	local a="${1}"
	local b="${2}"

	# Check if a == b
	[ "${a}" = "${b}" ] && return 1

	# Sort by version
	local sorted=( $(printf "${a}\n${b}\n" | sort -Vr) )

	# If a comes first in the sorted order, it is largest
	[ "${a}" = "${sorted[0]}" ] && return 0

	# Otherwise b is greater than a
	return 1
}

update_pcengines_apu() {
	local board="${1}"
	local running_version="${2}"
	shift 2

	# Find a new firmware file
	local firmware="$(find_firmware "pcengines/apu/${board}_*.rom")"
	if [ -z "${firmware}" ]; then
		echo "Error: No firmware found" >&2
		return 1
	fi

	# Get basename and remove suffix
	local new_version="$(basename "${firmware%.rom}")"
	new_version="${new_version#*_}"

	# Check if we actually have a new version
	if ! version_is_gt "${new_version}" "${running_version}"; then
		echo "We have ${new_version} and this board is already on ${running_version}"
		echo "No new firmware available. Aborting."
		return 1
	fi

	echo "New firmware version available: ${new_version}"

	# Perform update
	if get_consent "${firmware}"; then
		do_flashrom -p "internal" -w "${firmware}"
	fi
}

do_update() {
	local vendor="$(board_vendor)"
	local board="$(board_name)"
	local bios_version="$(bios_version)"

	# Check if we could read the BIOS version
	if [ -z "${bios_version}" ]; then
		echo "Error: Could not detect BIOS version" >&2
		return 1
	fi

	echo "Detected ${vendor} ${board} running BIOS version ${bios_version}"

	case "${vendor},${board}" in
		"PC Engines",apu*)
			update_pcengines_apu "${board}" "${bios_version}" "$@"
			;;

		*)
			echo "Unkown vendor: ${vendor}" >&2
			echo "Support for this vendor might not have been implement, yet" >&2
			return 1
			;;
	esac
}

main() {
	local action="${1}"
	shift

	case "${action}" in
		info)
			printf "%-12s: %s %s\n" "Board" \
				"$(board_vendor)" "$(board_name)"
			printf "%-12s: %s\n" "HW Version" "$(board_version)"
			printf "%-12s: %s\n" "Serial" "$(board_serial)"
			printf "%-12s: %s (%s)\n" "BIOS Version" \
				"$(bios_version)" "$(bios_date)"
			return 0
			;;

		update)
			do_update "$@"
			;;

		version)
			echo "${PACKAGE_NAME}: Version ${PACKAGE_VERSION}"
			return 0
			;;

		"")
			echo "No command given" >&2
			return 2
			;;

		*)
			echo "Invalid command: ${action}" >&2
			return 2
			;;
	esac
}

# Call main
main "$@" || exit ${?}