344 lines
10 KiB
Bash
344 lines
10 KiB
Bash
#!/bin/sh
|
|
#===============================================================================
|
|
#
|
|
# Copyright (C) 2020-2024 by Digi International Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License version 2 as published by
|
|
# the Free Software Foundation.
|
|
#
|
|
#
|
|
# Description:
|
|
# Script to flash Yocto build artifacts over USB to the target.
|
|
#===============================================================================
|
|
# set -x
|
|
|
|
#
|
|
# U-Boot script for installing Linux images created by Yocto
|
|
#
|
|
|
|
# Exit on any error
|
|
set -e
|
|
|
|
# Parse uuu cmd output
|
|
getenv()
|
|
{
|
|
uuu -v fb: ucmd printenv "${1}" | sed -ne "s,^${1}=,,g;T;p"
|
|
}
|
|
|
|
show_usage()
|
|
{
|
|
echo "Usage: $0 [options]"
|
|
echo ""
|
|
echo " Options:"
|
|
echo " -b Activate bootcount mechanism (3 boot attempts)."
|
|
echo " -d Install firmware on dualboot partitions (system A and system B)."
|
|
echo " (Implies -b)."
|
|
echo " -h Show this help."
|
|
echo " -i <dey-image-name> Image name that prefixes the image filenames, such as 'dey-image-qt', "
|
|
echo " 'dey-image-webkit', 'core-image-base'..."
|
|
echo " Defaults to '##DEFAULT_IMAGE_NAME##' if not provided."
|
|
echo " -k <dek-blob-file> Update includes dek blob file."
|
|
echo " (requires -t)."
|
|
echo " -n No wait. Skips 10 seconds delay to stop script."
|
|
echo " -t Install Trustfence artifacts."
|
|
echo " -u <u-boot-filename> U-Boot filename."
|
|
echo " Auto-determined by variant if not provided."
|
|
exit 2
|
|
}
|
|
|
|
# Update a partition
|
|
# Params:
|
|
# 1. partition
|
|
# 2. file
|
|
# 3. dek blob file when updating an encrypted bootloader
|
|
part_update()
|
|
{
|
|
echo "\033[36m"
|
|
echo "====================================================================================="
|
|
echo "Updating '${1}' partition with file: ${2}"
|
|
echo "====================================================================================="
|
|
echo "\033[0m"
|
|
|
|
if [ "${TRUSTFENCE}" = "true" ] && [ "${1}" = "bootloader" ]; then
|
|
uuu fb: download -f "${2}"
|
|
if [ -n "${DEK_BLOB_KEY}" ]; then
|
|
uuu fb: ucmd setenv uboot_size $filesize
|
|
uuu fb: ucmd setenv fastboot_buffer $initrd_addr
|
|
uuu fb: download -f "${3}"
|
|
uuu fb: ucmd setenv dek_size $filesize
|
|
uuu fb: ucmd trustfence update ram \${loadaddr} \${uboot_size} \${initrd_addr} \${dek_size}
|
|
else
|
|
uuu fb: ucmd trustfence update ram \${fastboot_buffer} \${fastboot_bytes}
|
|
fi
|
|
else
|
|
if [ "${1}" = "bootloader" ]; then
|
|
uuu fb: flash "${1}" "${2}"
|
|
else
|
|
uuu fb: flash -raw2sparse "${1}" "${2}"
|
|
fi
|
|
fi
|
|
}
|
|
|
|
clear
|
|
echo "############################################################"
|
|
echo "# Linux firmware install through USB OTG #"
|
|
echo "############################################################"
|
|
|
|
# Command line admits the following parameters:
|
|
# -b, -d, -n (booleans)
|
|
# -i <image-name>
|
|
# -u <u-boot-filename>
|
|
# -k <dek-blob-name>
|
|
while getopts 'bdhi:k:ntu:' c
|
|
do
|
|
case $c in
|
|
b) BOOTCOUNT=true ;;
|
|
d) INSTALL_DUALBOOT=true && BOOTCOUNT=true ;;
|
|
h) show_usage ;;
|
|
i) IMAGE_NAME=${OPTARG} ;;
|
|
k) DEK_BLOB_KEY=${OPTARG} ;;
|
|
n) NOWAIT=true ;;
|
|
t) TRUSTFENCE=true ;;
|
|
u) INSTALL_UBOOT_FILENAME=${OPTARG} ;;
|
|
esac
|
|
done
|
|
|
|
# Enable the redirect support to get u-boot variables values
|
|
uuu fb: ucmd setenv stdout serial,fastboot
|
|
|
|
# Check if dualboot variable is active
|
|
dualboot=$(getenv "dualboot")
|
|
if [ "${dualboot}" = "yes" ]; then
|
|
DUALBOOT=true;
|
|
fi
|
|
|
|
echo ""
|
|
echo "Determining image files to use..."
|
|
|
|
# Determine U-Boot file to program basing on SOM's SOC revision
|
|
if [ -z ${INSTALL_UBOOT_FILENAME} ]; then
|
|
# Since SOMs with the B0 SOC might have an older U-Boot that doesn't export the
|
|
# SOC revision to the environment, use B0 by default
|
|
soc_rev=$(getenv "soc_rev")
|
|
if [ -z "${soc_rev}" ]; then
|
|
soc_rev="B0"
|
|
fi
|
|
|
|
INSTALL_UBOOT_FILENAME="imx-boot-##SIGNED##-##MACHINE##-${soc_rev}.bin"
|
|
fi
|
|
|
|
# remove redirect
|
|
uuu fb: ucmd setenv stdout serial
|
|
|
|
# Determine linux, recovery, and rootfs image filenames to update
|
|
if [ -z "${IMAGE_NAME}" ]; then
|
|
IMAGE_NAME="##DEFAULT_IMAGE_NAME##"
|
|
fi
|
|
GRAPHICAL_IMAGES="##GRAPHICAL_IMAGES##"
|
|
for g in ${GRAPHICAL_IMAGES}; do
|
|
if [ "${IMAGE_NAME}" = "${g}" ]; then
|
|
BASEFILENAME="${IMAGE_NAME}-##GRAPHICAL_BACKEND##"
|
|
fi
|
|
done
|
|
if [ -z "${BASEFILENAME}" ]; then
|
|
BASEFILENAME="${IMAGE_NAME}"
|
|
fi
|
|
INSTALL_LINUX_FILENAME="${BASEFILENAME}-##MACHINE##.boot.vfat"
|
|
INSTALL_RECOVERY_FILENAME="${BASEFILENAME}-##MACHINE##.recovery.vfat"
|
|
INSTALL_ROOTFS_FILENAME="${BASEFILENAME}-##MACHINE##.ext4"
|
|
|
|
COMPRESSED_ROOTFS_IMAGE="${INSTALL_ROOTFS_FILENAME}.gz"
|
|
|
|
# If the rootfs image is compressed, make sure to decompress it before the update
|
|
if [ -f ${COMPRESSED_ROOTFS_IMAGE} ] && [ ! -f ${INSTALL_ROOTFS_FILENAME} ]; then
|
|
echo "\033[36m"
|
|
echo "====================================================================================="
|
|
echo "Decompressing rootfs image '${COMPRESSED_ROOTFS_IMAGE}'"
|
|
echo "====================================================================================="
|
|
echo "\033[0m"
|
|
gzip -d -k -f "${COMPRESSED_ROOTFS_IMAGE}"
|
|
fi
|
|
|
|
# Verify existence of files before starting the update
|
|
FILES="${INSTALL_UBOOT_FILENAME} ${INSTALL_LINUX_FILENAME}"
|
|
if [ "${DUALBOOT}" != true ]; then
|
|
FILES="${FILES} ${INSTALL_RECOVERY_FILENAME}"
|
|
fi
|
|
for f in ${FILES}; do
|
|
if [ ! -f ${f} ]; then
|
|
echo "\033[31m[ERROR] Could not find file '${f}'\033[0m"
|
|
ABORT=true
|
|
fi
|
|
done;
|
|
|
|
# Verify what kind of rootfs is going to be programmed
|
|
if [ ! -f ${INSTALL_ROOTFS_FILENAME} ]; then
|
|
echo "\033[31m[ERROR] Could not find file '${INSTALL_ROOTFS_FILENAME}'\033[0m"
|
|
INSTALL_ROOTFS_FILENAME="${BASEFILENAME}-##MACHINE##.squashfs"
|
|
echo "\033[32m[INFO] Trying with file '${INSTALL_ROOTFS_FILENAME}'\033[0m"
|
|
if [ ! -f "${INSTALL_ROOTFS_FILENAME}" ]; then
|
|
echo "\033[31m[ERROR] Could not find file '${INSTALL_ROOTFS_FILENAME}'\033[0m"
|
|
ABORT=true
|
|
fi
|
|
fi
|
|
|
|
[ "${ABORT}" = true ] && exit 1
|
|
|
|
# Enable bootcount mechanism by setting a bootlimit
|
|
if [ "${BOOTCOUNT}" = true ]; then
|
|
bootlimit_cmd="setenv bootlimit 3"
|
|
fi
|
|
|
|
# parts names
|
|
LINUX_NAME="linux"
|
|
RECOVERY_NAME="recovery"
|
|
ROOTFS_NAME="rootfs"
|
|
# Print warning about storage media being deleted
|
|
if [ "${NOWAIT}" != true ]; then
|
|
WAIT=10
|
|
printf "\n"
|
|
printf " ====================\n"
|
|
printf " = IMPORTANT! =\n"
|
|
printf " ====================\n"
|
|
printf " This process will erase your eMMC and will install the following files\n"
|
|
printf " on the partitions of the eMMC.\n"
|
|
printf "\n"
|
|
printf " PARTITION\tFILENAME\n"
|
|
printf " ---------\t--------\n"
|
|
printf " bootloader\t${INSTALL_UBOOT_FILENAME}\n"
|
|
if [ "${DUALBOOT}" = true ]; then
|
|
printf " ${LINUX_NAME}_a\t${INSTALL_LINUX_FILENAME}\n"
|
|
if [ "${INSTALL_DUALBOOT}" = true ]; then
|
|
printf " ${LINUX_NAME}_b\t${INSTALL_LINUX_FILENAME}\n"
|
|
fi
|
|
printf " ${ROOTFS_NAME}_a\t${INSTALL_ROOTFS_FILENAME}\n"
|
|
if [ "${INSTALL_DUALBOOT}" = true ]; then
|
|
printf " ${ROOTFS_NAME}_b\t${INSTALL_ROOTFS_FILENAME}\n"
|
|
fi
|
|
else
|
|
printf " ${LINUX_NAME}\t${INSTALL_LINUX_FILENAME}\n"
|
|
printf " ${RECOVERY_NAME}\t${INSTALL_RECOVERY_FILENAME}\n"
|
|
printf " ${ROOTFS_NAME}\t${INSTALL_ROOTFS_FILENAME}\n"
|
|
fi
|
|
printf "\n"
|
|
printf " Press CTRL+C now if you wish to abort.\n"
|
|
printf "\n"
|
|
while [ ${WAIT} -gt 0 ]; do
|
|
printf "\r Update process starts in %d " ${WAIT}
|
|
sleep 1
|
|
WAIT=$(( ${WAIT} - 1 ))
|
|
done
|
|
printf "\r \n"
|
|
printf " Starting update process\n"
|
|
fi
|
|
|
|
# Set fastboot buffer address to $loadaddr, just in case
|
|
uuu fb: ucmd setenv fastboot_buffer \${loadaddr}
|
|
|
|
# Skip user confirmation for U-Boot update
|
|
uuu fb: ucmd setenv forced_update 1
|
|
|
|
# Update U-Boot
|
|
part_update "bootloader" "${INSTALL_UBOOT_FILENAME}" "${DEK_BLOB_KEY}"
|
|
|
|
# Set MMC to boot from BOOT1 partition
|
|
uuu fb: ucmd mmc partconf 0 1 1 1
|
|
|
|
# Set 'bootcmd' for the second part of the script that will
|
|
# - Reset environment to defaults
|
|
# - Reset the bootcount
|
|
# - Set bootlimit (if required)
|
|
# - Save the environment
|
|
# - Partition the eMMC user data area for Linux
|
|
# - Update the 'linux' partition
|
|
# - Update the 'recovery' partition
|
|
# - Update the 'rootfs' partition
|
|
uuu fb: ucmd setenv bootcmd "
|
|
env default -a;
|
|
setenv dualboot \${dualboot};
|
|
bootcount reset;
|
|
${bootlimit_cmd};
|
|
saveenv;
|
|
echo \"\";
|
|
echo \"\";
|
|
echo \">> Creating Linux partition table on the eMMC\";
|
|
echo \"\";
|
|
echo \"\";
|
|
run partition_mmc_linux;
|
|
if test \$? -eq 1; then
|
|
echo \"[ERROR] Failed to create Linux partition table!\";
|
|
echo \"\";
|
|
echo \"Aborted.\";
|
|
exit;
|
|
fi;
|
|
echo \"\";
|
|
echo \"\";
|
|
echo \">> Start installation Linux firmware files\";
|
|
echo \"\";
|
|
echo \"\";
|
|
saveenv;
|
|
fastboot 1;
|
|
"
|
|
|
|
uuu fb: ucmd saveenv
|
|
uuu fb: acmd reset
|
|
|
|
# Wait for the target to reset
|
|
sleep 3
|
|
|
|
# Restart fastboot with the latest MMC partition configuration
|
|
uuu fb: ucmd setenv fastboot_dev sata
|
|
uuu fb: ucmd setenv fastboot_dev mmc
|
|
|
|
# Set fastboot buffer address to $loadaddr, just in case
|
|
uuu fb: ucmd setenv fastboot_buffer \${loadaddr}
|
|
|
|
if [ "${DUALBOOT}" = true ]; then
|
|
# Update Linux A
|
|
part_update "${LINUX_NAME}_a" "${INSTALL_LINUX_FILENAME}"
|
|
# Update Linux B
|
|
if [ "${INSTALL_DUALBOOT}" = true ]; then
|
|
part_update "${LINUX_NAME}_b" "${INSTALL_LINUX_FILENAME}"
|
|
fi
|
|
# Update Rootfs A
|
|
part_update "${ROOTFS_NAME}_a" "${INSTALL_ROOTFS_FILENAME}"
|
|
# Update Rootfs B
|
|
if [ "${INSTALL_DUALBOOT}" = true ]; then
|
|
part_update "${ROOTFS_NAME}_b" "${INSTALL_ROOTFS_FILENAME}"
|
|
fi
|
|
else
|
|
# Update Linux
|
|
part_update "${LINUX_NAME}" "${INSTALL_LINUX_FILENAME}"
|
|
# Update Recovery
|
|
part_update "${RECOVERY_NAME}" "${INSTALL_RECOVERY_FILENAME}"
|
|
# Update Rootfs
|
|
part_update "${ROOTFS_NAME}" "${INSTALL_ROOTFS_FILENAME}"
|
|
fi
|
|
|
|
# If the rootfs image was originally compressed, remove the uncompressed image
|
|
if [ -f ${COMPRESSED_ROOTFS_IMAGE} ] && [ -f ${INSTALL_ROOTFS_FILENAME} ]; then
|
|
rm -f "${INSTALL_ROOTFS_FILENAME}"
|
|
fi
|
|
|
|
if [ "${DUALBOOT}" != true ]; then
|
|
# Configure u-boot to boot into recovery mode
|
|
uuu fb: ucmd setenv boot_recovery yes
|
|
uuu fb: ucmd setenv recovery_command wipe_update
|
|
fi
|
|
uuu fb: ucmd saveenv
|
|
|
|
# Reset the bootcount
|
|
uuu fb: ucmd bootcount reset
|
|
# Reset the target
|
|
uuu fb: acmd reset
|
|
|
|
echo "\033[32m"
|
|
echo "============================================================="
|
|
echo "Done! Wait for the target to complete first boot process."
|
|
echo "============================================================="
|
|
echo "\033[0m"
|
|
|
|
exit
|