brillo_update_payload revision 135197526b9c0f5a9acac57a05129339f311bbca
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 1498e691cc6debda3ef671acba001bebd45da36377Alex Deymo# properties generate a properties file from a payload 15be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 16be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Generate command arguments: 179a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --payload generated unsigned payload output file 189a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --source_image if defined, generate a delta payload from the specified 199a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# image to the target_image 209a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --target_image the target image that should be sent to clients 219a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --metadata_size_file if defined, generate a file containing the size of the payload 229a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# metadata in bytes to the specified file 23be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 24be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Hash command arguments: 25be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --unsigned_payload the input unsigned payload to generate the hash from 26be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --signature_size signature sizes in bytes in the following format: 2789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 28be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --payload_hash_file if defined, generate a payload hash and output to the 29be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 30be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --metadata_hash_file if defined, generate a metadata hash and output to the 31be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 32be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 33be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sign command arguments: 3489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --unsigned_payload the input unsigned payload to insert the signatures 3589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload the output signed payload 3689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --signature_size signature sizes in bytes in the following format: 3789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 3889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload_signature_file the payload signature files in the following 3989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 4089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "payload_signature1:payload_signature2[:...]" 4189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --metadata_signature_file the metadata signature files in the following 4289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 4389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "metadata_signature1:metadata_signature2[:...]" 449a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --metadata_size_file if defined, generate a file containing the size of 459a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# the signed payload metadata in bytes to the 469a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# specified file 47be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Note that the number of signature sizes and payload signatures have to match. 4898e691cc6debda3ef671acba001bebd45da36377Alex Deymo# 4998e691cc6debda3ef671acba001bebd45da36377Alex Deymo# Properties command arguments: 5098e691cc6debda3ef671acba001bebd45da36377Alex Deymo# --payload the input signed or unsigned payload 5198e691cc6debda3ef671acba001bebd45da36377Alex Deymo# --properties_file the output path where to write the properties, or 5298e691cc6debda3ef671acba001bebd45da36377Alex Deymo# '-' for stdout. 5398e691cc6debda3ef671acba001bebd45da36377Alex Deymo 54be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 5561e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo# Exit codes: 5661e1fa89e01952d772f4aca843fa817a7a30fe66Alex DeymoEX_UNSUPPORTED_DELTA=100 5761e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo 58f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusumawarn() { 59f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma echo "brillo_update_payload: warning: $*" >&2 60f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma} 61f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 62957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnolddie() { 63957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold echo "brillo_update_payload: error: $*" >&2 64957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold exit 1 65957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold} 66be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 67957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# Loads shflags. We first look at the default install location; then look for 68957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# crosutils (chroot); finally check our own directory (au-generator zipfile). 69957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags() { 70957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local my_dir="$(dirname "$(readlink -f "$0")")" 71957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local path 72957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold for path in /usr/share/misc {/usr/lib/crosutils,"${my_dir}"}/lib/shflags; do 73957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold if [[ -r "${path}/shflags" ]]; then 74957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold . "${path}/shflags" || die "Could not load ${path}/shflags." 75957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold return 76be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 77be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma done 78957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold die "Could not find shflags." 79be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 80be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 81957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags 82be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 83c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_GENERATE="generate: Generate an unsigned update payload." 84c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_HASH="hash: Generate the hashes of the unsigned payload and metadata used \ 85c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofor signing." 86c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_SIGN="sign: Insert the signatures into the unsigned payload." 8798e691cc6debda3ef671acba001bebd45da36377Alex DeymoHELP_PROPERTIES="properties: Extract payload properties to a file." 88c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 89c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymousage() { 90c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Supported commands:" 91c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 92c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_GENERATE}" 93c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_HASH}" 94c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_SIGN}" 9598e691cc6debda3ef671acba001bebd45da36377Alex Deymo echo "${HELP_PROPERTIES}" 96c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 97c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Use: \"$0 <command> --help\" for more options." 98c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo} 99c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 100c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Check that a command is specified. 101be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaif [[ $# -lt 1 ]]; then 10298e691cc6debda3ef671acba001bebd45da36377Alex Deymo echo "Please specify a command [generate|hash|sign|properties]" 103be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 104be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumafi 105be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 106c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Parse command. 107c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoCOMMAND="${1:-}" 108c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoshift 109c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 110c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymocase "${COMMAND}" in 111c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo generate) 112c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_GENERATE}" 113c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 114c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 115c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo hash) 116c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_HASH}" 117c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 118c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 119c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo sign) 120c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_SIGN}" 121be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 12298e691cc6debda3ef671acba001bebd45da36377Alex Deymo 12398e691cc6debda3ef671acba001bebd45da36377Alex Deymo properties) 12498e691cc6debda3ef671acba001bebd45da36377Alex Deymo FLAGS_HELP="${HELP_PROPERTIES}" 12598e691cc6debda3ef671acba001bebd45da36377Alex Deymo ;; 126be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma *) 127c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Unrecognized command: \"${COMMAND}\"" >&2 128c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo usage >&2 129be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 130be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 131be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 132be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 133be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Flags 134c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoFLAGS_HELP="Usage: $0 ${COMMAND} [flags] 135c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo${FLAGS_HELP}" 136c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 137c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "generate" ]]; then 138c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 139c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 140c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string target_image "" \ 141c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to the target image that should be sent to clients." 142c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string source_image "" \ 143c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to a source image. If specified, this makes a delta update." 1449a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1459a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 146c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 147c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" || "${COMMAND}" == "sign" ]]; then 148c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string unsigned_payload "" "Path to the input unsigned payload." 149c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string signature_size "" \ 150c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Signature sizes in bytes in the following format: size1:size2[:...]" 151c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 152c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" ]]; then 153c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_hash_file "" \ 154c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output metadata hash file." 155c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_hash_file "" \ 156c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output payload hash file." 157c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 158c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "sign" ]]; then 159c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 160c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 161c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_signature_file "" \ 162c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The metatada signatures in the following format: \ 163c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymometadata_signature1:metadata_signature2[:...]" 164c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_signature_file "" \ 165c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The payload signatures in the following format: \ 166c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymopayload_signature1:payload_signature2[:...]" 1679a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1689a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 169c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 17098e691cc6debda3ef671acba001bebd45da36377Alex Deymoif [[ "${COMMAND}" == "properties" ]]; then 17198e691cc6debda3ef671acba001bebd45da36377Alex Deymo DEFINE_string payload "" \ 17298e691cc6debda3ef671acba001bebd45da36377Alex Deymo "Path to the input signed or unsigned payload file." 17398e691cc6debda3ef671acba001bebd45da36377Alex Deymo DEFINE_string properties_file "-" \ 17498e691cc6debda3ef671acba001bebd45da36377Alex Deymo "Path to output the extracted property files. If '-' is passed stdout will \ 17598e691cc6debda3ef671acba001bebd45da36377Alex Deymobe used." 17698e691cc6debda3ef671acba001bebd45da36377Alex Deymofi 17798e691cc6debda3ef671acba001bebd45da36377Alex Deymo 178be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaDEFINE_string work_dir "/tmp" "Where to dump temporary files." 179be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 180be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Parse command line flag arguments 181be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaFLAGS "$@" || exit 1 182be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaeval set -- "${FLAGS_ARGV}" 18389ff9e3221c358977f9c3124930ee6b559853740Alex Deymoset -e 184be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 18589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Associative arrays from partition name to file in the source and target 18689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# images. The size of the updated area must be the size of the file. 18789ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A SRC_PARTITIONS 18889ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A DST_PARTITIONS 18989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 190788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# List of partition names in order. 191788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiangdeclare -a PARTITIONS_ORDER 192788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 19389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# A list of temporary files to remove during cleanup. 19489ff9e3221c358977f9c3124930ee6b559853740Alex DeymoCLEANUP_FILES=() 19589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 19648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Global options to force the version of the payload. 19748b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MAJOR_VERSION="" 19848b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MINOR_VERSION="" 19948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 2006f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang# Path to the postinstall config file in target image if exists. 2016f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen JiangPOSTINSTALL_CONFIG_FILE="" 2026f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 203e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang# The fingerprint of zlib in the source image. 204e0d04282910d423ddc950dd38224c001efc41d42Sen JiangZLIB_FINGERPRINT="" 205e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang 206c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# read_option_int <file.txt> <option_key> [default_value] 207c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# 208c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# Reads the unsigned integer value associated with |option_key| in a key=value 209c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# file |file.txt|. Prints the read value if found and valid, otherwise prints 210c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# the |default_value|. 211c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymoread_option_uint() { 212c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local file_txt="$1" 213c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local option_key="$2" 214c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local default_value="${3:-}" 215c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local value 216c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(look "${option_key}=" "${file_txt}" | tail -n 1); then 217c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(echo "${value}" | cut -f 2- -d "=" | grep -E "^[0-9]+$"); then 218c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${value}" 219c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo return 220c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 221c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 222c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${default_value}" 223c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo} 224c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo 225d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# truncate_file <file_path> <file_size> 226d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# 227d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# Truncate the given |file_path| to |file_size| using perl. 228d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# The truncate binary might not be available. 229d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiangtruncate_file() { 230d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang local file_path="$1" 231d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang local file_size="$2" 232d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang perl -e "open(FILE, \"+<\", \$ARGV[0]); \ 233d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang truncate(FILE, ${file_size}); \ 234d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang close(FILE);" "${file_path}" 235d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang} 236d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang 23789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Create a temporary file in the work_dir with an optional pattern name. 23889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Prints the name of the newly created file. 23989ff9e3221c358977f9c3124930ee6b559853740Alex Deymocreate_tempfile() { 24089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local pattern="${1:-tempfile.XXXXXX}" 24189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo mktemp --tmpdir="${FLAGS_work_dir}" "${pattern}" 24289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 243be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 244be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup() { 245be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma local err="" 24689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo rm -f "${CLEANUP_FILES[@]}" || err=1 247be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 248be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # If we are cleaning up after an error, or if we got an error during 249be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # cleanup (even if we eventually succeeded) return a non-zero exit 250be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # code. This triggers additional logging in most environments that call 251be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # this script. 252be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -n "${err}" ]]; then 253be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup encountered an error." 254be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 255be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 256be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 257be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_error() { 258be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 259be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 260be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup success after an error." 261be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 262be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 263be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_exit() { 264be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 265be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 266be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 267be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 268be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_error INT TERM ERR 269be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_exit EXIT 270be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 27148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 272788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image <image> <partitions_array> [partitions_order] 27348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 27448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Detect the format of the |image| file and extract its updatable partitions 27548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# into new temporary files. Add the list of partition names and its files to the 276788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# associative array passed in |partitions_array|. If |partitions_order| is 277788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# passed, set it to list of partition names in order. 27848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image() { 27948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 28048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 28148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Brillo images are zip files. We detect the 4-byte magic header of the zip 28248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # file. 28348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${image}" | hexdump -e '1/1 "%.2x"') 28448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "504b0304" ]]; then 28548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected .zip file, extracting Brillo image." 28648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_brillo "$@" 28748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 28848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 28948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 29048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images are GPT partitioned disks. We should have the cgpt binary 29148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # bundled here and we will use it to extract the partitions, so the GPT 29248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # headers must be valid. 29348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if cgpt show -q -n "${image}" >/dev/null; then 29448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected GPT image, extracting Chrome OS image." 29548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_cros "$@" 29648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 29748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 29848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 29948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo die "Couldn't detect the image format of ${image}" 30048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 30148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 302788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image_cros <image.bin> <partitions_array> [partitions_order] 30389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# 30448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract Chromium OS recovery images into new temporary files. 30589ff9e3221c358977f9c3124930ee6b559853740Alex Deymoextract_image_cros() { 30689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local image="$1" 30789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local partitions_array="$2" 308788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang local partitions_order="${3:-}" 30989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 31089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local kernel root 31189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo kernel=$(create_tempfile "kernel.bin.XXXXXX") 31289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${kernel}") 31389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo root=$(create_tempfile "root.bin.XXXXXX") 31489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${root}") 31589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 31689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo cros_generate_update_payload --extract \ 31789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --image "${image}" \ 31889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --kern_path "${kernel}" --root_path "${root}" \ 31989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --work_dir "${FLAGS_work_dir}" --outside_chroot 32089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 32183f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # Chrome OS uses major_version 1 payloads for all versions, even if the 32283f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # updater supports a newer major version. 32383f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="1" 32483f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 325e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 326e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang # Copy from zlib_fingerprint in source image to stdout. 327e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang ZLIB_FINGERPRINT=$(e2cp "${root}":/etc/zlib_fingerprint -) 328e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang fi 329e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang 33048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # When generating legacy Chrome OS images, we need to use "boot" and "system" 33148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # for the partition names to be compatible with updating Brillo devices with 33248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images. 33348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[boot]=\""${kernel}"\" 33448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[system]=\""${root}"\" 33589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 336788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang if [[ -n "${partitions_order}" ]]; then 337788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang eval "${partitions_order}=( \"system\" \"boot\" )" 338788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang fi 339788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 34089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local part varname 34148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in boot system; do 34289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo varname="${partitions_array}[${part}]" 34389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo printf "md5sum of %s: " "${varname}" 34489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo md5sum "${!varname}" 34589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo done 34689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 34789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 348788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image_brillo <target_files.zip> <partitions_array> [partitions_order] 34948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 35048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract the A/B updated partitions from a Brillo target_files zip file into 35148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# new temporary files. 35248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image_brillo() { 35348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 35448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions_array="$2" 355788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang local partitions_order="${3:-}" 35648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 35748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions=( "boot" "system" ) 358168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local ab_partitions_list 359168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo ab_partitions_list=$(create_tempfile "ab_partitions_list.XXXXXX") 360168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo CLEANUP_FILES+=("${ab_partitions_list}") 361168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if unzip -p "${image}" "META/ab_partitions.txt" >"${ab_partitions_list}"; then 362168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if grep -v -E '^[a-zA-Z0-9_-]*$' "${ab_partitions_list}" >&2; then 363168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "Invalid partition names found in the partition list." 364168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 365168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partitions=($(cat "${ab_partitions_list}")) 366168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ ${#partitions[@]} -eq 0 ]]; then 367168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "The list of partitions is empty. Can't generate a payload." 368168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 369168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo else 370168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo warn "No ab_partitions.txt found. Using default." 371168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 372168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo echo "List of A/B partitions: ${partitions[@]}" 37348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 374788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang if [[ -n "${partitions_order}" ]]; then 375788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang eval "${partitions_order}=(${partitions[@]})" 376788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang fi 377788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 37883f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # All Brillo updaters support major version 2. 37983f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="2" 38083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 38148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 3826f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Source image 3836f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local ue_config=$(create_tempfile "ue_config.XXXXXX") 384c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo CLEANUP_FILES+=("${ue_config}") 385c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if ! unzip -p "${image}" "META/update_engine_config.txt" \ 386c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo >"${ue_config}"; then 387c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo warn "No update_engine_config.txt found. Assuming pre-release image, \ 388c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymousing payload minor version 2" 389c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 39083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # For delta payloads, we use the major and minor version supported by the 39183f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # old updater. 392c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo FORCE_MINOR_VERSION=$(read_option_uint "${ue_config}" \ 393c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo "PAYLOAD_MINOR_VERSION" 2) 39483f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION=$(read_option_uint "${ue_config}" \ 39583f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo "PAYLOAD_MAJOR_VERSION" 2) 39661e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo 39761e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo # Brillo support for deltas started with minor version 3. 39861e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo if [[ "${FORCE_MINOR_VERSION}" -le 2 ]]; then 39961e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo warn "No delta support from minor version ${FORCE_MINOR_VERSION}. \ 40061e1fa89e01952d772f4aca843fa817a7a30fe66Alex DeymoDisabling deltas for this source version." 40161e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo exit ${EX_UNSUPPORTED_DELTA} 40261e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo fi 403e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang 404e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang if [[ "${FORCE_MINOR_VERSION}" -ge 4 ]]; then 405e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang ZLIB_FINGERPRINT=$(unzip -p "${image}" "META/zlib_fingerprint.txt") 406e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang fi 4076f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang else 4086f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Target image 4096f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local postinstall_config=$(create_tempfile "postinstall_config.XXXXXX") 4106f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang CLEANUP_FILES+=("${postinstall_config}") 4116f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if unzip -p "${image}" "META/postinstall_config.txt" \ 4126f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang >"${postinstall_config}"; then 4136f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang POSTINSTALL_CONFIG_FILE="${postinstall_config}" 4146f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 41548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 41648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 41748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local part part_file temp_raw filesize 41848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in "${partitions[@]}"; do 41948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file=$(create_tempfile "${part}.img.XXXXXX") 42048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${part_file}") 42148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo unzip -p "${image}" "IMAGES/${part}.img" >"${part_file}" 42248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 42348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # If the partition is stored as an Android sparse image file, we need to 42448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # convert them to a raw image for the update. 42548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${part_file}" | hexdump -e '1/1 "%.2x"') 42648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "3aff26ed" ]]; then 42748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo temp_raw=$(create_tempfile "${part}.raw.XXXXXX") 42848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${temp_raw}") 42948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Converting Android sparse image ${part}.img to RAW." 43048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo simg2img "${part_file}" "${temp_raw}" 43148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # At this point, we can drop the contents of the old part_file file, but 43248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # we can't delete the file because it will be deleted in cleanup. 43348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo true >"${part_file}" 43448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file="${temp_raw}" 43548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 43648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 437a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # delta_generator only supports images multiple of 4 KiB. For target images 438a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # we pad the data with zeros if needed, but for source images we truncate 439a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # down the data since the last block of the old image could be padded on 440a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # disk with unknown data. 44148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo filesize=$(stat -c%s "${part_file}") 44248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ $(( filesize % 4096 )) -ne 0 ]]; then 443a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 444a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo echo "Rounding DOWN partition ${part}.img to a multiple of 4 KiB." 445a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo : $(( filesize = filesize & -4096 )) 446a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo else 447a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo echo "Rounding UP partition ${part}.img to a multiple of 4 KiB." 448a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo : $(( filesize = (filesize + 4095) & -4096 )) 449a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo fi 450d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang truncate_file "${part_file}" "${filesize}" 45148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 45248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 45348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval "${partitions_array}[\"${part}\"]=\"${part_file}\"" 45448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracted ${partitions_array}[${part}]: ${filesize} bytes" 45548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo done 45648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 45748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 458be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_generate() { 459be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 46053f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify an output filename with --payload FILENAME" 461be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 462be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_target_image}" ]] || 46353f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify a target image with --target_image FILENAME" 464be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 465be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 466be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_generate() { 46789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local payload_type="delta" 468be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -z "${FLAGS_source_image}" ]]; then 46989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo payload_type="full" 470be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 471be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 47248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracting images for ${payload_type} update." 473be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 474788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang extract_image "${FLAGS_target_image}" DST_PARTITIONS PARTITIONS_ORDER 47589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 47648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image "${FLAGS_source_image}" SRC_PARTITIONS 477be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 478be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 47948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Generating ${payload_type} update." 480168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Common payload args: 481168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS=( -out_file="${FLAGS_payload}" ) 482168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 483168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local part old_partitions="" new_partitions="" partition_names="" 484788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang for part in "${PARTITIONS_ORDER[@]}"; do 485168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ -n "${partition_names}" ]]; then 486168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+=":" 487168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+=":" 488168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+=":" 489168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 490168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+="${part}" 491168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+="${DST_PARTITIONS[${part}]}" 492168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+="${SRC_PARTITIONS[${part}]:-}" 493168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo done 494168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 495168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Target image args: 496168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS+=( 497168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -partition_names="${partition_names}" 498168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -new_partitions="${new_partitions}" 499be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 500be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 50189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 502168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Source image args: 503be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma GENERATOR_ARGS+=( 504168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -old_partitions="${old_partitions}" 505be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 50648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MINOR_VERSION}" ]]; then 50748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --minor_version="${FORCE_MINOR_VERSION}" ) 50848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 509e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang if [[ -n "${ZLIB_FINGERPRINT}" ]]; then 510e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang GENERATOR_ARGS+=( --zlib_fingerprint="${ZLIB_FINGERPRINT}" ) 511e0d04282910d423ddc950dd38224c001efc41d42Sen Jiang fi 51248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 51348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 51448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then 51548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" ) 516be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 517be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 5189a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 5199a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 5209a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 5219a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5226f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if [[ -n "${POSTINSTALL_CONFIG_FILE}" ]]; then 5236f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang GENERATOR_ARGS+=( 5246f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang --new_postinstall_config_file="${POSTINSTALL_CONFIG_FILE}" 5256f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang ) 5266f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 5276f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 528be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Running delta_generator with args: ${GENERATOR_ARGS[@]}" 5299a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 530be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 53189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo echo "Done generating ${payload_type} update." 532be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 533be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 534be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_hash() { 535be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 53653f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify signature size with --signature_size SIZES" 537be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 538be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 53953f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the input unsigned payload with \ 540be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 541be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 542be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_hash_file}" ]] || 54353f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify --payload_hash_file FILENAME" 544f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 545f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma [[ -n "${FLAGS_metadata_hash_file}" ]] || 54653f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify --metadata_hash_file FILENAME" 547be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 548be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 549be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_hash() { 550bf1266fec223275ff19ef9624651946be9b9112aSen Jiang "${GENERATOR}" \ 551bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -in_file="${FLAGS_unsigned_payload}" \ 552bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -signature_size="${FLAGS_signature_size}" \ 553bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_hash_file="${FLAGS_payload_hash_file}" \ 554bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_metadata_hash_file="${FLAGS_metadata_hash_file}" 555be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 556be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done generating hash." 557be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 558be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 559be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_sign() { 560be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 56153f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify signature size with --signature_size SIZES" 562be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 563be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 56453f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the input unsigned payload with \ 565be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 566be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 567be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 56853f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the output signed payload with --payload FILENAME" 569be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 570be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_signature_file}" ]] || 57153f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the payload signature file with \ 572be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--payload_signature_file SIGNATURES" 57389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 57489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo [[ -n "${FLAGS_metadata_signature_file}" ]] || 57553f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the metadata signature file with \ 57689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo--metadata_signature_file SIGNATURES" 577be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 578be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 579be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_sign() { 5809a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS=( 5819a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -in_file="${FLAGS_unsigned_payload}" 5829a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_size="${FLAGS_signature_size}" 5839a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_file="${FLAGS_payload_signature_file}" 5849a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -metadata_signature_file="${FLAGS_metadata_signature_file}" 5859a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -out_file="${FLAGS_payload}" 5869a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma ) 5879a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5889a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 5899a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 5909a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 5919a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5929a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 593be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done signing payload." 594be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 595be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 59698e691cc6debda3ef671acba001bebd45da36377Alex Deymovalidate_properties() { 59798e691cc6debda3ef671acba001bebd45da36377Alex Deymo [[ -n "${FLAGS_payload}" ]] || 59853f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the payload file with --payload FILENAME" 59998e691cc6debda3ef671acba001bebd45da36377Alex Deymo 60098e691cc6debda3ef671acba001bebd45da36377Alex Deymo [[ -n "${FLAGS_properties_file}" ]] || 60153f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify a non empty --properties_file FILENAME" 60298e691cc6debda3ef671acba001bebd45da36377Alex Deymo} 60398e691cc6debda3ef671acba001bebd45da36377Alex Deymo 60498e691cc6debda3ef671acba001bebd45da36377Alex Deymocmd_properties() { 60598e691cc6debda3ef671acba001bebd45da36377Alex Deymo "${GENERATOR}" \ 60698e691cc6debda3ef671acba001bebd45da36377Alex Deymo -in_file="${FLAGS_payload}" \ 60798e691cc6debda3ef671acba001bebd45da36377Alex Deymo -properties_file="${FLAGS_properties_file}" 60898e691cc6debda3ef671acba001bebd45da36377Alex Deymo} 60998e691cc6debda3ef671acba001bebd45da36377Alex Deymo 610be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sanity check that the real generator exists: 611135197526b9c0f5a9acac57a05129339f311bbcaSen JiangGENERATOR="$(which delta_generator || true)" 612be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma[[ -x "${GENERATOR}" ]] || die "can't find delta_generator" 613be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 614be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacase "$COMMAND" in 615be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma generate) validate_generate 616be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_generate 617be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 618be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma hash) validate_hash 619be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_hash 620be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 621be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma sign) validate_sign 622be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_sign 623be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 62498e691cc6debda3ef671acba001bebd45da36377Alex Deymo properties) validate_properties 62598e691cc6debda3ef671acba001bebd45da36377Alex Deymo cmd_properties 62698e691cc6debda3ef671acba001bebd45da36377Alex Deymo ;; 627be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 628