brillo_update_payload revision 6f7b22c69bef53d33ae2bbb80b6682ff7ea8006e
1be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma#!/bin/bash 2be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 3be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Copyright 2015 The Chromium OS Authors. All rights reserved. 4be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Use of this source code is governed by a BSD-style license that can be 5be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# found in the LICENSE file. 6be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 7be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Script to generate a Brillo update for use by the update engine. 8be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 9be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# usage: brillo_update_payload COMMAND [ARGS] 10be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# The following commands are supported: 11be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# generate generate an unsigned payload 12be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# hash generate a payload or metadata hash 13be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# sign generate a signed payload 14be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 15be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Generate command arguments: 169a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --payload generated unsigned payload output file 179a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --source_image if defined, generate a delta payload from the specified 189a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# image to the target_image 199a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --target_image the target image that should be sent to clients 209a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --metadata_size_file if defined, generate a file containing the size of the payload 219a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# metadata in bytes to the specified file 22be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 23be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Hash command arguments: 24be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --unsigned_payload the input unsigned payload to generate the hash from 25be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --signature_size signature sizes in bytes in the following format: 2689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 27be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --payload_hash_file if defined, generate a payload hash and output to the 28be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 29be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --metadata_hash_file if defined, generate a metadata hash and output to the 30be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 31be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 32be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sign command arguments: 3389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --unsigned_payload the input unsigned payload to insert the signatures 3489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload the output signed payload 3589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --signature_size signature sizes in bytes in the following format: 3689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 3789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload_signature_file the payload signature files in the following 3889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 3989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "payload_signature1:payload_signature2[:...]" 4089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --metadata_signature_file the metadata signature files in the following 4189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 4289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "metadata_signature1:metadata_signature2[:...]" 439a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --metadata_size_file if defined, generate a file containing the size of 449a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# the signed payload metadata in bytes to the 459a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# specified file 46be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Note that the number of signature sizes and payload signatures have to match. 47be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 48f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusumawarn() { 49f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma echo "brillo_update_payload: warning: $*" >&2 50f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma} 51f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 52957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnolddie() { 53957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold echo "brillo_update_payload: error: $*" >&2 54957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold exit 1 55957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold} 56be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 57957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# Loads shflags. We first look at the default install location; then look for 58957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# crosutils (chroot); finally check our own directory (au-generator zipfile). 59957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags() { 60957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local my_dir="$(dirname "$(readlink -f "$0")")" 61957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local path 62957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold for path in /usr/share/misc {/usr/lib/crosutils,"${my_dir}"}/lib/shflags; do 63957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold if [[ -r "${path}/shflags" ]]; then 64957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold . "${path}/shflags" || die "Could not load ${path}/shflags." 65957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold return 66be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 67be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma done 68957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold die "Could not find shflags." 69be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 70be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 71957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags 72be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 73c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_GENERATE="generate: Generate an unsigned update payload." 74c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_HASH="hash: Generate the hashes of the unsigned payload and metadata used \ 75c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofor signing." 76c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_SIGN="sign: Insert the signatures into the unsigned payload." 77c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 78c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymousage() { 79c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Supported commands:" 80c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 81c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_GENERATE}" 82c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_HASH}" 83c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_SIGN}" 84c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 85c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Use: \"$0 <command> --help\" for more options." 86c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo} 87c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 88c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Check that a command is specified. 89be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaif [[ $# -lt 1 ]]; then 90be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Please specify a command [generate|hash|sign]" 91be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 92be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumafi 93be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 94c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Parse command. 95c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoCOMMAND="${1:-}" 96c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoshift 97c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 98c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymocase "${COMMAND}" in 99c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo generate) 100c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_GENERATE}" 101c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 102c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 103c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo hash) 104c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_HASH}" 105c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 106c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 107c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo sign) 108c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_SIGN}" 109be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 110be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma *) 111c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Unrecognized command: \"${COMMAND}\"" >&2 112c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo usage >&2 113be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 114be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 115be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 116be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 117be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Flags 118c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoFLAGS_HELP="Usage: $0 ${COMMAND} [flags] 119c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo${FLAGS_HELP}" 120c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 121c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "generate" ]]; then 122c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 123c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 124c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string target_image "" \ 125c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to the target image that should be sent to clients." 126c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string source_image "" \ 127c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to a source image. If specified, this makes a delta update." 1289a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1299a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 130c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 131c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" || "${COMMAND}" == "sign" ]]; then 132c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string unsigned_payload "" "Path to the input unsigned payload." 133c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string signature_size "" \ 134c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Signature sizes in bytes in the following format: size1:size2[:...]" 135c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 136c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" ]]; then 137c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_hash_file "" \ 138c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output metadata hash file." 139c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_hash_file "" \ 140c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output payload hash file." 141c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 142c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "sign" ]]; then 143c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 144c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 145c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_signature_file "" \ 146c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The metatada signatures in the following format: \ 147c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymometadata_signature1:metadata_signature2[:...]" 148c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_signature_file "" \ 149c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The payload signatures in the following format: \ 150c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymopayload_signature1:payload_signature2[:...]" 1519a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1529a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 153c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 154be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaDEFINE_string work_dir "/tmp" "Where to dump temporary files." 155be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 156be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Parse command line flag arguments 157be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaFLAGS "$@" || exit 1 158be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaeval set -- "${FLAGS_ARGV}" 15989ff9e3221c358977f9c3124930ee6b559853740Alex Deymoset -e 160be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 16189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Associative arrays from partition name to file in the source and target 16289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# images. The size of the updated area must be the size of the file. 16389ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A SRC_PARTITIONS 16489ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A DST_PARTITIONS 16589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 16689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# A list of temporary files to remove during cleanup. 16789ff9e3221c358977f9c3124930ee6b559853740Alex DeymoCLEANUP_FILES=() 16889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 16948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Global options to force the version of the payload. 17048b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MAJOR_VERSION="" 17148b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MINOR_VERSION="" 17248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 1736f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang# Path to the postinstall config file in target image if exists. 1746f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen JiangPOSTINSTALL_CONFIG_FILE="" 1756f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 176c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# read_option_int <file.txt> <option_key> [default_value] 177c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# 178c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# Reads the unsigned integer value associated with |option_key| in a key=value 179c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# file |file.txt|. Prints the read value if found and valid, otherwise prints 180c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# the |default_value|. 181c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymoread_option_uint() { 182c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local file_txt="$1" 183c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local option_key="$2" 184c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local default_value="${3:-}" 185c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local value 186c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(look "${option_key}=" "${file_txt}" | tail -n 1); then 187c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(echo "${value}" | cut -f 2- -d "=" | grep -E "^[0-9]+$"); then 188c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${value}" 189c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo return 190c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 191c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 192c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${default_value}" 193c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo} 194c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo 19589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Create a temporary file in the work_dir with an optional pattern name. 19689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Prints the name of the newly created file. 19789ff9e3221c358977f9c3124930ee6b559853740Alex Deymocreate_tempfile() { 19889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local pattern="${1:-tempfile.XXXXXX}" 19989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo mktemp --tmpdir="${FLAGS_work_dir}" "${pattern}" 20089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 201be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 202be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup() { 203be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma local err="" 20489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo rm -f "${CLEANUP_FILES[@]}" || err=1 205be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 206be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # If we are cleaning up after an error, or if we got an error during 207be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # cleanup (even if we eventually succeeded) return a non-zero exit 208be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # code. This triggers additional logging in most environments that call 209be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # this script. 210be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -n "${err}" ]]; then 211be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup encountered an error." 212be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 213be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 214be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 215be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_error() { 216be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 217be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 218be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup success after an error." 219be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 220be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 221be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_exit() { 222be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 223be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 224be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 225be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 226be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_error INT TERM ERR 227be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_exit EXIT 228be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 22948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 23048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# extract_image <image> <partitions_array> 23148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 23248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Detect the format of the |image| file and extract its updatable partitions 23348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# into new temporary files. Add the list of partition names and its files to the 23448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# associative array passed in |partitions_array|. 23548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image() { 23648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 23748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 23848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Brillo images are zip files. We detect the 4-byte magic header of the zip 23948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # file. 24048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${image}" | hexdump -e '1/1 "%.2x"') 24148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "504b0304" ]]; then 24248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected .zip file, extracting Brillo image." 24348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_brillo "$@" 24448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 24548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 24648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 24748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images are GPT partitioned disks. We should have the cgpt binary 24848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # bundled here and we will use it to extract the partitions, so the GPT 24948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # headers must be valid. 25048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if cgpt show -q -n "${image}" >/dev/null; then 25148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected GPT image, extracting Chrome OS image." 25248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_cros "$@" 25348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 25448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 25548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 25648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo die "Couldn't detect the image format of ${image}" 25748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 25848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 25989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# extract_image_cros <image.bin> <partitions_array> 26089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# 26148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract Chromium OS recovery images into new temporary files. 26289ff9e3221c358977f9c3124930ee6b559853740Alex Deymoextract_image_cros() { 26389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local image="$1" 26489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local partitions_array="$2" 26589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 26689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local kernel root 26789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo kernel=$(create_tempfile "kernel.bin.XXXXXX") 26889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${kernel}") 26989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo root=$(create_tempfile "root.bin.XXXXXX") 27089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${root}") 27189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 27289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo cros_generate_update_payload --extract \ 27389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --image "${image}" \ 27489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --kern_path "${kernel}" --root_path "${root}" \ 27589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --work_dir "${FLAGS_work_dir}" --outside_chroot 27689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 27783f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # Chrome OS uses major_version 1 payloads for all versions, even if the 27883f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # updater supports a newer major version. 27983f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="1" 28083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 28148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # When generating legacy Chrome OS images, we need to use "boot" and "system" 28248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # for the partition names to be compatible with updating Brillo devices with 28348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images. 28448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[boot]=\""${kernel}"\" 28548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[system]=\""${root}"\" 28689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 28789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local part varname 28848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in boot system; do 28989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo varname="${partitions_array}[${part}]" 29089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo printf "md5sum of %s: " "${varname}" 29189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo md5sum "${!varname}" 29289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo done 29389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 29489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 29548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# extract_image_brillo <target_files.zip> <partitions_array> 29648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 29748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract the A/B updated partitions from a Brillo target_files zip file into 29848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# new temporary files. 29948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image_brillo() { 30048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 30148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions_array="$2" 30248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 30348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions=( "boot" "system" ) 304168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local ab_partitions_list 305168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo ab_partitions_list=$(create_tempfile "ab_partitions_list.XXXXXX") 306168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo CLEANUP_FILES+=("${ab_partitions_list}") 307168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if unzip -p "${image}" "META/ab_partitions.txt" >"${ab_partitions_list}"; then 308168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if grep -v -E '^[a-zA-Z0-9_-]*$' "${ab_partitions_list}" >&2; then 309168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "Invalid partition names found in the partition list." 310168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 311168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partitions=($(cat "${ab_partitions_list}")) 312168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ ${#partitions[@]} -eq 0 ]]; then 313168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "The list of partitions is empty. Can't generate a payload." 314168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 315168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo else 316168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo warn "No ab_partitions.txt found. Using default." 317168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 318168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo echo "List of A/B partitions: ${partitions[@]}" 31948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 32083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # All Brillo updaters support major version 2. 32183f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="2" 32283f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 32348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 3246f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Source image 3256f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local ue_config=$(create_tempfile "ue_config.XXXXXX") 326c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo CLEANUP_FILES+=("${ue_config}") 327c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if ! unzip -p "${image}" "META/update_engine_config.txt" \ 328c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo >"${ue_config}"; then 329c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo warn "No update_engine_config.txt found. Assuming pre-release image, \ 330c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymousing payload minor version 2" 331c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 33283f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # For delta payloads, we use the major and minor version supported by the 33383f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # old updater. 334c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo FORCE_MINOR_VERSION=$(read_option_uint "${ue_config}" \ 335c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo "PAYLOAD_MINOR_VERSION" 2) 33683f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION=$(read_option_uint "${ue_config}" \ 33783f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo "PAYLOAD_MAJOR_VERSION" 2) 3386f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang else 3396f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Target image 3406f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local postinstall_config=$(create_tempfile "postinstall_config.XXXXXX") 3416f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang CLEANUP_FILES+=("${postinstall_config}") 3426f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if unzip -p "${image}" "META/postinstall_config.txt" \ 3436f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang >"${postinstall_config}"; then 3446f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang POSTINSTALL_CONFIG_FILE="${postinstall_config}" 3456f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 34648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 34748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 34848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local part part_file temp_raw filesize 34948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in "${partitions[@]}"; do 35048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file=$(create_tempfile "${part}.img.XXXXXX") 35148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${part_file}") 35248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo unzip -p "${image}" "IMAGES/${part}.img" >"${part_file}" 35348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 35448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # If the partition is stored as an Android sparse image file, we need to 35548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # convert them to a raw image for the update. 35648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${part_file}" | hexdump -e '1/1 "%.2x"') 35748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "3aff26ed" ]]; then 35848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo temp_raw=$(create_tempfile "${part}.raw.XXXXXX") 35948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${temp_raw}") 36048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Converting Android sparse image ${part}.img to RAW." 36148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo simg2img "${part_file}" "${temp_raw}" 36248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # At this point, we can drop the contents of the old part_file file, but 36348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # we can't delete the file because it will be deleted in cleanup. 36448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo true >"${part_file}" 36548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file="${temp_raw}" 36648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 36748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 36848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # delta_generator only supports images multiple of 4 KiB, so we pad with 36948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # zeros if needed. 37048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo filesize=$(stat -c%s "${part_file}") 37148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ $(( filesize % 4096 )) -ne 0 ]]; then 37248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Rounding up partition ${part}.img to multiple of 4 KiB." 37348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo : $(( filesize = (filesize + 4095) & -4096 )) 37448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo truncate --size="${filesize}" "${part_file}" 37548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 37648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 37748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval "${partitions_array}[\"${part}\"]=\"${part_file}\"" 37848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracted ${partitions_array}[${part}]: ${filesize} bytes" 37948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo done 38048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 38148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 382be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_generate() { 383be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 384be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify an output filename with --payload FILENAME" 385be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 386be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_target_image}" ]] || 387be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify a target image with --target_image FILENAME" 388be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 389be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 390be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_generate() { 39189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local payload_type="delta" 392be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -z "${FLAGS_source_image}" ]]; then 39389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo payload_type="full" 394be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 395be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 39648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracting images for ${payload_type} update." 397be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 39848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image "${FLAGS_target_image}" DST_PARTITIONS 39989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 40048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image "${FLAGS_source_image}" SRC_PARTITIONS 401be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 402be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 40348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Generating ${payload_type} update." 404168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Common payload args: 405168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS=( -out_file="${FLAGS_payload}" ) 406168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 407168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local part old_partitions="" new_partitions="" partition_names="" 408168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo for part in "${!DST_PARTITIONS[@]}"; do 409168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ -n "${partition_names}" ]]; then 410168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+=":" 411168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+=":" 412168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+=":" 413168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 414168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+="${part}" 415168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+="${DST_PARTITIONS[${part}]}" 416168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+="${SRC_PARTITIONS[${part}]:-}" 417168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo done 418168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 419168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Target image args: 420168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS+=( 421168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -partition_names="${partition_names}" 422168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -new_partitions="${new_partitions}" 423be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 424be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 42589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 426168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Source image args: 427be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma GENERATOR_ARGS+=( 428168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -old_partitions="${old_partitions}" 429be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 43048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MINOR_VERSION}" ]]; then 43148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --minor_version="${FORCE_MINOR_VERSION}" ) 43248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 43348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 43448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 43548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then 43648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" ) 437be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 438be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 4399a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 4409a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 4419a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 4429a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 4436f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if [[ -n "${POSTINSTALL_CONFIG_FILE}" ]]; then 4446f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang GENERATOR_ARGS+=( 4456f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang --new_postinstall_config_file="${POSTINSTALL_CONFIG_FILE}" 4466f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang ) 4476f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 4486f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 449be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Running delta_generator with args: ${GENERATOR_ARGS[@]}" 4509a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 451be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 45289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo echo "Done generating ${payload_type} update." 453be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 454be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 455be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_hash() { 456be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 457be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify signature size with --signature_size SIZES" 458be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 459be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 460be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify the input unsigned payload with \ 461be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 462be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 463be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_hash_file}" ]] || 464bf1266fec223275ff19ef9624651946be9b9112aSen Jiang die "Error: you must specify --payload_hash_file FILENAME" 465f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 466f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma [[ -n "${FLAGS_metadata_hash_file}" ]] || 467f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma die "Error: you must specify --metadata_hash_file FILENAME" 468be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 469be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 470be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_hash() { 471bf1266fec223275ff19ef9624651946be9b9112aSen Jiang "${GENERATOR}" \ 472bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -in_file="${FLAGS_unsigned_payload}" \ 473bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -signature_size="${FLAGS_signature_size}" \ 474bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_hash_file="${FLAGS_payload_hash_file}" \ 475bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_metadata_hash_file="${FLAGS_metadata_hash_file}" 476be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 477be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done generating hash." 478be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 479be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 480be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_sign() { 481be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 482be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify signature size with --signature_size SIZES" 483be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 484be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 485be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify the input unsigned payload with \ 486be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 487be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 488be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 489be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify the output signed payload with \ 490be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--payload FILENAME" 491be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 492be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_signature_file}" ]] || 493be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Error: you must specify the payload signature file with \ 494be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--payload_signature_file SIGNATURES" 49589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 49689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo [[ -n "${FLAGS_metadata_signature_file}" ]] || 49789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo die "Error: you must specify the metadata signature file with \ 49889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo--metadata_signature_file SIGNATURES" 499be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 500be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 501be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_sign() { 5029a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS=( 5039a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -in_file="${FLAGS_unsigned_payload}" 5049a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_size="${FLAGS_signature_size}" 5059a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_file="${FLAGS_payload_signature_file}" 5069a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -metadata_signature_file="${FLAGS_metadata_signature_file}" 5079a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -out_file="${FLAGS_payload}" 5089a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma ) 5099a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5109a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 5119a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 5129a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 5139a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5149a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 515be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done signing payload." 516be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 517be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 518be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sanity check that the real generator exists: 519be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaGENERATOR="$(which delta_generator)" 520be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma[[ -x "${GENERATOR}" ]] || die "can't find delta_generator" 521be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 522be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacase "$COMMAND" in 523be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma generate) validate_generate 524be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_generate 525be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 526be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma hash) validate_hash 527be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_hash 528be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 529be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma sign) validate_sign 530be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_sign 531be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 532be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 533