1be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma#!/bin/bash 2be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 31352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# 41352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# Copyright (C) 2015 The Android Open Source Project 51352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# 61352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# Licensed under the Apache License, Version 2.0 (the "License"); 71352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# you may not use this file except in compliance with the License. 81352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# You may obtain a copy of the License at 91352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# 101352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# http://www.apache.org/licenses/LICENSE-2.0 111352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# 121352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# Unless required by applicable law or agreed to in writing, software 131352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# distributed under the License is distributed on an "AS IS" BASIS, 141352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 151352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# See the License for the specific language governing permissions and 161352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# limitations under the License. 171352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# 18be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 19be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Script to generate a Brillo update for use by the update engine. 20be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 21be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# usage: brillo_update_payload COMMAND [ARGS] 22be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# The following commands are supported: 23be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# generate generate an unsigned payload 24be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# hash generate a payload or metadata hash 25be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# sign generate a signed payload 2698e691cc6debda3ef671acba001bebd45da36377Alex Deymo# properties generate a properties file from a payload 271352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# verify verify a payload by recreating a target image. 28be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 29be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Generate command arguments: 309a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --payload generated unsigned payload output file 319a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --source_image if defined, generate a delta payload from the specified 329a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# image to the target_image 339a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --target_image the target image that should be sent to clients 341352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# --metadata_size_file if defined, generate a file containing the size of the 351352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# payload metadata in bytes to the specified file 36be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 37be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Hash command arguments: 38be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --unsigned_payload the input unsigned payload to generate the hash from 39be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --signature_size signature sizes in bytes in the following format: 4089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 41be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --payload_hash_file if defined, generate a payload hash and output to the 42be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 43be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# --metadata_hash_file if defined, generate a metadata hash and output to the 44be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# specified file 45be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# 46be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sign command arguments: 4789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --unsigned_payload the input unsigned payload to insert the signatures 4889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload the output signed payload 4989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --signature_size signature sizes in bytes in the following format: 5089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "size1:size2[:...]" 5189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --payload_signature_file the payload signature files in the following 5289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 5389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "payload_signature1:payload_signature2[:...]" 5489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# --metadata_signature_file the metadata signature files in the following 5589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# format: 5689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# "metadata_signature1:metadata_signature2[:...]" 579a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# --metadata_size_file if defined, generate a file containing the size of 589a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# the signed payload metadata in bytes to the 599a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma# specified file 60be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Note that the number of signature sizes and payload signatures have to match. 6198e691cc6debda3ef671acba001bebd45da36377Alex Deymo# 6298e691cc6debda3ef671acba001bebd45da36377Alex Deymo# Properties command arguments: 6398e691cc6debda3ef671acba001bebd45da36377Alex Deymo# --payload the input signed or unsigned payload 6498e691cc6debda3ef671acba001bebd45da36377Alex Deymo# --properties_file the output path where to write the properties, or 6598e691cc6debda3ef671acba001bebd45da36377Alex Deymo# '-' for stdout. 661352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# Verify command arguments: 671352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# --payload payload input file 681352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# --source_image verify payload to the specified source image. 691352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani# --target_image the target image to verify upon. 7098e691cc6debda3ef671acba001bebd45da36377Alex Deymo 71be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 7261e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo# Exit codes: 7361e1fa89e01952d772f4aca843fa817a7a30fe66Alex DeymoEX_UNSUPPORTED_DELTA=100 7461e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo 75f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusumawarn() { 76f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma echo "brillo_update_payload: warning: $*" >&2 77f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma} 78f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 79957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnolddie() { 80957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold echo "brillo_update_payload: error: $*" >&2 81957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold exit 1 82957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold} 83be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 84957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# Loads shflags. We first look at the default install location; then look for 85957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold# crosutils (chroot); finally check our own directory (au-generator zipfile). 86957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags() { 87957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local my_dir="$(dirname "$(readlink -f "$0")")" 88957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold local path 89957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold for path in /usr/share/misc {/usr/lib/crosutils,"${my_dir}"}/lib/shflags; do 90957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold if [[ -r "${path}/shflags" ]]; then 91957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold . "${path}/shflags" || die "Could not load ${path}/shflags." 92957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold return 93be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 94be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma done 95957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnold die "Could not find shflags." 96be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 97be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 98957ce12832240c1e8a66af425d8524a3cbfdfd97Gilad Arnoldload_shflags 99be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 100c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_GENERATE="generate: Generate an unsigned update payload." 101c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_HASH="hash: Generate the hashes of the unsigned payload and metadata used \ 102c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofor signing." 103c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoHELP_SIGN="sign: Insert the signatures into the unsigned payload." 10498e691cc6debda3ef671acba001bebd45da36377Alex DeymoHELP_PROPERTIES="properties: Extract payload properties to a file." 1051352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin HassaniHELP_VERIFY="verify: Verify a (signed) update payload." 106c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 107c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymousage() { 108c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Supported commands:" 109c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 110c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_GENERATE}" 111c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_HASH}" 112c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "${HELP_SIGN}" 11398e691cc6debda3ef671acba001bebd45da36377Alex Deymo echo "${HELP_PROPERTIES}" 1141352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "${HELP_VERIFY}" 115c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo 116c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Use: \"$0 <command> --help\" for more options." 117c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo} 118c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 119c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Check that a command is specified. 120be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaif [[ $# -lt 1 ]]; then 12198e691cc6debda3ef671acba001bebd45da36377Alex Deymo echo "Please specify a command [generate|hash|sign|properties]" 122be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 123be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumafi 124be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 125c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo# Parse command. 126c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoCOMMAND="${1:-}" 127c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoshift 128c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 129c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymocase "${COMMAND}" in 130c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo generate) 131c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_GENERATE}" 132c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 133c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 134c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo hash) 135c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_HASH}" 136c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo ;; 137c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 138c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo sign) 139c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo FLAGS_HELP="${HELP_SIGN}" 140be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 14198e691cc6debda3ef671acba001bebd45da36377Alex Deymo 14298e691cc6debda3ef671acba001bebd45da36377Alex Deymo properties) 14398e691cc6debda3ef671acba001bebd45da36377Alex Deymo FLAGS_HELP="${HELP_PROPERTIES}" 14498e691cc6debda3ef671acba001bebd45da36377Alex Deymo ;; 1451352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 1461352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani verify) 1471352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani FLAGS_HELP="${HELP_VERIFY}" 1481352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani ;; 1491352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 150be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma *) 151c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo echo "Unrecognized command: \"${COMMAND}\"" >&2 152c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo usage >&2 153be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma exit 1 154be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 155be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 156be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 157be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Flags 158c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex DeymoFLAGS_HELP="Usage: $0 ${COMMAND} [flags] 159c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo${FLAGS_HELP}" 160c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo 161c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "generate" ]]; then 162c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 163c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 164c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string target_image "" \ 165c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to the target image that should be sent to clients." 166c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string source_image "" \ 167c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to a source image. If specified, this makes a delta update." 1689a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1699a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 1708e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang DEFINE_string max_timestamp "" \ 1718e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang "Optional: The maximum unix timestamp of the OS allowed to apply this \ 1728e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiangpayload, should be set to a number higher than the build timestamp of the \ 1738e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiangsystem running on the device, 0 if not specified." 174c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 175c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" || "${COMMAND}" == "sign" ]]; then 176c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string unsigned_payload "" "Path to the input unsigned payload." 177c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string signature_size "" \ 178c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Signature sizes in bytes in the following format: size1:size2[:...]" 179c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 180c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "hash" ]]; then 181c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_hash_file "" \ 182c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output metadata hash file." 183c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_hash_file "" \ 184c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Optional: Path to output payload hash file." 185c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 186c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymoif [[ "${COMMAND}" == "sign" ]]; then 187c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload "" \ 188c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "Path to output the generated unsigned payload file." 189c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string metadata_signature_file "" \ 190c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The metatada signatures in the following format: \ 191c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymometadata_signature1:metadata_signature2[:...]" 192c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo DEFINE_string payload_signature_file "" \ 193c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymo "The payload signatures in the following format: \ 194c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymopayload_signature1:payload_signature2[:...]" 1959a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma DEFINE_string metadata_size_file "" \ 1969a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "Optional: Path to output metadata size." 197c64ffd591d1f4f4f7bacce0670bc0fce70830f36Alex Deymofi 19898e691cc6debda3ef671acba001bebd45da36377Alex Deymoif [[ "${COMMAND}" == "properties" ]]; then 19998e691cc6debda3ef671acba001bebd45da36377Alex Deymo DEFINE_string payload "" \ 20098e691cc6debda3ef671acba001bebd45da36377Alex Deymo "Path to the input signed or unsigned payload file." 20198e691cc6debda3ef671acba001bebd45da36377Alex Deymo DEFINE_string properties_file "-" \ 20298e691cc6debda3ef671acba001bebd45da36377Alex Deymo "Path to output the extracted property files. If '-' is passed stdout will \ 20398e691cc6debda3ef671acba001bebd45da36377Alex Deymobe used." 20498e691cc6debda3ef671acba001bebd45da36377Alex Deymofi 2051352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassaniif [[ "${COMMAND}" == "verify" ]]; then 2061352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani DEFINE_string payload "" \ 2071352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani "Path to the input payload file." 2081352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani DEFINE_string target_image "" \ 2091352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani "Path to the target image to verify upon." 2101352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani DEFINE_string source_image "" \ 2111352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani "Optional: Path to a source image. If specified, the delta update is \ 2121352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassaniapplied to this." 2131352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassanifi 21498e691cc6debda3ef671acba001bebd45da36377Alex Deymo 2155fbb110c73158e443efe1fcac87e6648c631b4c0Alex DeymoDEFINE_string work_dir "${TMPDIR:-/tmp}" "Where to dump temporary files." 216be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 217be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Parse command line flag arguments 218be998f47ccb02c5f118f73e79729e9187c39e919Jason KusumaFLAGS "$@" || exit 1 219be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaeval set -- "${FLAGS_ARGV}" 22089ff9e3221c358977f9c3124930ee6b559853740Alex Deymoset -e 221be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 2225fbb110c73158e443efe1fcac87e6648c631b4c0Alex Deymo# Override the TMPDIR with the passed work_dir flags, which anyway defaults to 2235fbb110c73158e443efe1fcac87e6648c631b4c0Alex Deymo# ${TMPDIR}. 2245fbb110c73158e443efe1fcac87e6648c631b4c0Alex DeymoTMPDIR="${FLAGS_work_dir}" 2255fbb110c73158e443efe1fcac87e6648c631b4c0Alex Deymoexport TMPDIR 2265fbb110c73158e443efe1fcac87e6648c631b4c0Alex Deymo 22789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Associative arrays from partition name to file in the source and target 22889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# images. The size of the updated area must be the size of the file. 22989ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A SRC_PARTITIONS 23089ff9e3221c358977f9c3124930ee6b559853740Alex Deymodeclare -A DST_PARTITIONS 23189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 23220bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo# Associative arrays for the .map files associated with each src/dst partition 23320bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo# file in SRC_PARTITIONS and DST_PARTITIONS. 23420bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymodeclare -A SRC_PARTITIONS_MAP 23520bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymodeclare -A DST_PARTITIONS_MAP 23620bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo 237788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# List of partition names in order. 238788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiangdeclare -a PARTITIONS_ORDER 239788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 24089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# A list of temporary files to remove during cleanup. 24189ff9e3221c358977f9c3124930ee6b559853740Alex DeymoCLEANUP_FILES=() 24289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 24348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Global options to force the version of the payload. 24448b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MAJOR_VERSION="" 24548b502ab0cd56bf948602a45b990448f51e3e6b5Alex DeymoFORCE_MINOR_VERSION="" 24648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 2476f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang# Path to the postinstall config file in target image if exists. 2486f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen JiangPOSTINSTALL_CONFIG_FILE="" 2496f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 250c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# read_option_int <file.txt> <option_key> [default_value] 251c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# 252c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# Reads the unsigned integer value associated with |option_key| in a key=value 253c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# file |file.txt|. Prints the read value if found and valid, otherwise prints 254c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo# the |default_value|. 255c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymoread_option_uint() { 256c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local file_txt="$1" 257c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local option_key="$2" 258c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local default_value="${3:-}" 259c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo local value 260c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(look "${option_key}=" "${file_txt}" | tail -n 1); then 261c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if value=$(echo "${value}" | cut -f 2- -d "=" | grep -E "^[0-9]+$"); then 262c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${value}" 263c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo return 264c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 265c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 266c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo echo "${default_value}" 267c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo} 268c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo 269d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# truncate_file <file_path> <file_size> 270d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# 271d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# Truncate the given |file_path| to |file_size| using perl. 272d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang# The truncate binary might not be available. 273d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiangtruncate_file() { 274d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang local file_path="$1" 275d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang local file_size="$2" 276d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang perl -e "open(FILE, \"+<\", \$ARGV[0]); \ 277d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang truncate(FILE, ${file_size}); \ 278d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang close(FILE);" "${file_path}" 279d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang} 280d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang 28189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Create a temporary file in the work_dir with an optional pattern name. 28289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# Prints the name of the newly created file. 28389ff9e3221c358977f9c3124930ee6b559853740Alex Deymocreate_tempfile() { 28489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local pattern="${1:-tempfile.XXXXXX}" 28589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo mktemp --tmpdir="${FLAGS_work_dir}" "${pattern}" 28689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 287be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 288be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup() { 289be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma local err="" 29089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo rm -f "${CLEANUP_FILES[@]}" || err=1 291be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 292be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # If we are cleaning up after an error, or if we got an error during 293be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # cleanup (even if we eventually succeeded) return a non-zero exit 294be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # code. This triggers additional logging in most environments that call 295be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma # this script. 296be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -n "${err}" ]]; then 297be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup encountered an error." 298be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 299be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 300be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 301be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_error() { 302be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 303be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 304be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma die "Cleanup success after an error." 305be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 306be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 307be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacleanup_on_exit() { 308be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma trap - INT TERM ERR EXIT 309be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cleanup 310be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 311be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 312be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_error INT TERM ERR 313be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumatrap cleanup_on_exit EXIT 314be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 31548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 316788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image <image> <partitions_array> [partitions_order] 31748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 31848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Detect the format of the |image| file and extract its updatable partitions 31948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# into new temporary files. Add the list of partition names and its files to the 320788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# associative array passed in |partitions_array|. If |partitions_order| is 321788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# passed, set it to list of partition names in order. 32248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image() { 32348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 32448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 32548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Brillo images are zip files. We detect the 4-byte magic header of the zip 32648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # file. 32748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${image}" | hexdump -e '1/1 "%.2x"') 32848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "504b0304" ]]; then 32948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected .zip file, extracting Brillo image." 33048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_brillo "$@" 33148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 33248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 33348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 33448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images are GPT partitioned disks. We should have the cgpt binary 33548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # bundled here and we will use it to extract the partitions, so the GPT 33648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # headers must be valid. 33748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if cgpt show -q -n "${image}" >/dev/null; then 33848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Detected GPT image, extracting Chrome OS image." 33948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image_cros "$@" 34048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo return 34148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 34248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 34348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo die "Couldn't detect the image format of ${image}" 34448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 34548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 346788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image_cros <image.bin> <partitions_array> [partitions_order] 34789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo# 34848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract Chromium OS recovery images into new temporary files. 34989ff9e3221c358977f9c3124930ee6b559853740Alex Deymoextract_image_cros() { 35089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local image="$1" 35189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local partitions_array="$2" 352788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang local partitions_order="${3:-}" 35389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 35489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local kernel root 35589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo kernel=$(create_tempfile "kernel.bin.XXXXXX") 35689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${kernel}") 35789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo root=$(create_tempfile "root.bin.XXXXXX") 35889ff9e3221c358977f9c3124930ee6b559853740Alex Deymo CLEANUP_FILES+=("${root}") 35989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 36089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo cros_generate_update_payload --extract \ 36189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --image "${image}" \ 36289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --kern_path "${kernel}" --root_path "${root}" \ 36389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo --work_dir "${FLAGS_work_dir}" --outside_chroot 36489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 36583f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # Chrome OS uses major_version 1 payloads for all versions, even if the 36683f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # updater supports a newer major version. 36783f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="1" 36883f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 36948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # When generating legacy Chrome OS images, we need to use "boot" and "system" 37048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # for the partition names to be compatible with updating Brillo devices with 37148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # Chrome OS images. 37248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[boot]=\""${kernel}"\" 37348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval ${partitions_array}[system]=\""${root}"\" 37489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 375788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang if [[ -n "${partitions_order}" ]]; then 376788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang eval "${partitions_order}=( \"system\" \"boot\" )" 377788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang fi 378788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 37989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local part varname 38048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in boot system; do 38189ff9e3221c358977f9c3124930ee6b559853740Alex Deymo varname="${partitions_array}[${part}]" 38289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo printf "md5sum of %s: " "${varname}" 38389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo md5sum "${!varname}" 38489ff9e3221c358977f9c3124930ee6b559853740Alex Deymo done 38589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo} 38689ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 387788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang# extract_image_brillo <target_files.zip> <partitions_array> [partitions_order] 38848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# 38948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# Extract the A/B updated partitions from a Brillo target_files zip file into 39048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo# new temporary files. 39148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymoextract_image_brillo() { 39248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local image="$1" 39348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions_array="$2" 394788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang local partitions_order="${3:-}" 39548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 39648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local partitions=( "boot" "system" ) 397168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local ab_partitions_list 398168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo ab_partitions_list=$(create_tempfile "ab_partitions_list.XXXXXX") 399168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo CLEANUP_FILES+=("${ab_partitions_list}") 400168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if unzip -p "${image}" "META/ab_partitions.txt" >"${ab_partitions_list}"; then 401168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if grep -v -E '^[a-zA-Z0-9_-]*$' "${ab_partitions_list}" >&2; then 402168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "Invalid partition names found in the partition list." 403168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 40434c711a0bd9d8ff0650ce18cbcaa436c6720fd6aSen Jiang # Get partition list without duplicates. 40534c711a0bd9d8ff0650ce18cbcaa436c6720fd6aSen Jiang partitions=($(awk '!seen[$0]++' "${ab_partitions_list}")) 406168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ ${#partitions[@]} -eq 0 ]]; then 407168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo die "The list of partitions is empty. Can't generate a payload." 408168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 409168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo else 410168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo warn "No ab_partitions.txt found. Using default." 411168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 412168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo echo "List of A/B partitions: ${partitions[@]}" 41348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 414788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang if [[ -n "${partitions_order}" ]]; then 415788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang eval "${partitions_order}=(${partitions[@]})" 416788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang fi 417788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang 41883f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # All Brillo updaters support major version 2. 41983f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION="2" 42083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo 42148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 4226f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Source image 4236f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local ue_config=$(create_tempfile "ue_config.XXXXXX") 424c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo CLEANUP_FILES+=("${ue_config}") 425c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo if ! unzip -p "${image}" "META/update_engine_config.txt" \ 426c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo >"${ue_config}"; then 427c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo warn "No update_engine_config.txt found. Assuming pre-release image, \ 428c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymousing payload minor version 2" 429c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo fi 43083f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # For delta payloads, we use the major and minor version supported by the 43183f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo # old updater. 432c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo FORCE_MINOR_VERSION=$(read_option_uint "${ue_config}" \ 433c97df43fdc7c4fcfe06a76274f43211e6aa9255cAlex Deymo "PAYLOAD_MINOR_VERSION" 2) 43483f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo FORCE_MAJOR_VERSION=$(read_option_uint "${ue_config}" \ 43583f2f70dd3aa3e4c8f16273bdd026b1f452b33a1Alex Deymo "PAYLOAD_MAJOR_VERSION" 2) 43661e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo 43761e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo # Brillo support for deltas started with minor version 3. 43861e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo if [[ "${FORCE_MINOR_VERSION}" -le 2 ]]; then 43961e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo warn "No delta support from minor version ${FORCE_MINOR_VERSION}. \ 44061e1fa89e01952d772f4aca843fa817a7a30fe66Alex DeymoDisabling deltas for this source version." 44161e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo exit ${EX_UNSUPPORTED_DELTA} 44261e1fa89e01952d772f4aca843fa817a7a30fe66Alex Deymo fi 4436f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang else 4446f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang # Target image 4456f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang local postinstall_config=$(create_tempfile "postinstall_config.XXXXXX") 4466f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang CLEANUP_FILES+=("${postinstall_config}") 4476f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if unzip -p "${image}" "META/postinstall_config.txt" \ 4486f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang >"${postinstall_config}"; then 4496f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang POSTINSTALL_CONFIG_FILE="${postinstall_config}" 4506f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 45148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 45248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 45348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local part part_file temp_raw filesize 45448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo for part in "${partitions[@]}"; do 45548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file=$(create_tempfile "${part}.img.XXXXXX") 45648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${part_file}") 45748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo unzip -p "${image}" "IMAGES/${part}.img" >"${part_file}" 45848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 45948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # If the partition is stored as an Android sparse image file, we need to 46048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # convert them to a raw image for the update. 46148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo local magic=$(head --bytes=4 "${part_file}" | hexdump -e '1/1 "%.2x"') 46248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ "${magic}" == "3aff26ed" ]]; then 46348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo temp_raw=$(create_tempfile "${part}.raw.XXXXXX") 46448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo CLEANUP_FILES+=("${temp_raw}") 46548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Converting Android sparse image ${part}.img to RAW." 46648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo simg2img "${part_file}" "${temp_raw}" 46748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # At this point, we can drop the contents of the old part_file file, but 46848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo # we can't delete the file because it will be deleted in cleanup. 46948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo true >"${part_file}" 47048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo part_file="${temp_raw}" 47148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 47248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 47320bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo # Extract the .map file (if one is available). 47420bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo part_map_file=$(create_tempfile "${part}.map.XXXXXX") 47520bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo CLEANUP_FILES+=("${part_map_file}") 47620bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo unzip -p "${image}" "IMAGES/${part}.map" >"${part_map_file}" || \ 47720bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo part_map_file="" 47820bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo 479a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # delta_generator only supports images multiple of 4 KiB. For target images 480a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # we pad the data with zeros if needed, but for source images we truncate 481a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # down the data since the last block of the old image could be padded on 482a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo # disk with unknown data. 48348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo filesize=$(stat -c%s "${part_file}") 48448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ $(( filesize % 4096 )) -ne 0 ]]; then 485a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then 486a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo echo "Rounding DOWN partition ${part}.img to a multiple of 4 KiB." 487a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo : $(( filesize = filesize & -4096 )) 48804c2305d79b65a3013e3d5b329f1fcc128874479Alex Deymo if [[ ${filesize} == 0 ]]; then 48904c2305d79b65a3013e3d5b329f1fcc128874479Alex Deymo echo "Source partition ${part}.img is empty after rounding down," \ 49004c2305d79b65a3013e3d5b329f1fcc128874479Alex Deymo "skipping." 49104c2305d79b65a3013e3d5b329f1fcc128874479Alex Deymo continue 49204c2305d79b65a3013e3d5b329f1fcc128874479Alex Deymo fi 493a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo else 494a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo echo "Rounding UP partition ${part}.img to a multiple of 4 KiB." 495a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo : $(( filesize = (filesize + 4095) & -4096 )) 496a479a4d0039308fcfdda21f2a8ec8d040fd716f2Alex Deymo fi 497d0e9a893163bb6f70b2bb003cda238d7e1a532edSen Jiang truncate_file "${part_file}" "${filesize}" 49848b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 49948b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 50048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo eval "${partitions_array}[\"${part}\"]=\"${part_file}\"" 50120bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo eval "${partitions_array}_MAP[\"${part}\"]=\"${part_map_file}\"" 50248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracted ${partitions_array}[${part}]: ${filesize} bytes" 50348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo done 50448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo} 50548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 506be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_generate() { 507be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 50853f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify an output filename with --payload FILENAME" 509be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 510be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_target_image}" ]] || 51153f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify a target image with --target_image FILENAME" 512be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 513be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 514be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_generate() { 51589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo local payload_type="delta" 516be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma if [[ -z "${FLAGS_source_image}" ]]; then 51789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo payload_type="full" 518be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 519be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 52048b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Extracting images for ${payload_type} update." 521be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 522788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang extract_image "${FLAGS_target_image}" DST_PARTITIONS PARTITIONS_ORDER 52389ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 52448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo extract_image "${FLAGS_source_image}" SRC_PARTITIONS 525be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 526be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 52748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo echo "Generating ${payload_type} update." 528168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Common payload args: 529168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS=( -out_file="${FLAGS_payload}" ) 530168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 531168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo local part old_partitions="" new_partitions="" partition_names="" 53220bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo local old_mapfiles="" new_mapfiles="" 533788c2d9b4b4da52f0874c391bda59610625e66aaSen Jiang for part in "${PARTITIONS_ORDER[@]}"; do 534168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo if [[ -n "${partition_names}" ]]; then 535168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+=":" 536168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+=":" 537168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+=":" 53820bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo new_mapfiles+=":" 53920bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo old_mapfiles+=":" 540168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo fi 541168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo partition_names+="${part}" 542168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo new_partitions+="${DST_PARTITIONS[${part}]}" 543168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo old_partitions+="${SRC_PARTITIONS[${part}]:-}" 54420bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo new_mapfiles+="${DST_PARTITIONS_MAP[${part}]:-}" 54520bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo old_mapfiles+="${SRC_PARTITIONS_MAP[${part}]:-}" 546168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo done 547168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo 548168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Target image args: 549168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo GENERATOR_ARGS+=( 550168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -partition_names="${partition_names}" 551168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -new_partitions="${new_partitions}" 55220bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo -new_mapfiles="${new_mapfiles}" 553be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 554be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 55589ff9e3221c358977f9c3124930ee6b559853740Alex Deymo if [[ "${payload_type}" == "delta" ]]; then 556168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo # Source image args: 557be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma GENERATOR_ARGS+=( 558168b535ceaeef22d501b5c14a167d323758e4ed6Alex Deymo -old_partitions="${old_partitions}" 55920bdc70e540e021c4a26553dfdbfd2fabe8ea432Alex Deymo -old_mapfiles="${old_mapfiles}" 560be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ) 56148b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MINOR_VERSION}" ]]; then 56248b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --minor_version="${FORCE_MINOR_VERSION}" ) 56348b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 56448b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo fi 56548b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo 56648b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then 56748b502ab0cd56bf948602a45b990448f51e3e6b5Alex Deymo GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" ) 568be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma fi 569be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 5709a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 5719a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 5729a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 5739a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 5748e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang if [[ -n "${FLAGS_max_timestamp}" ]]; then 5758e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang GENERATOR_ARGS+=( --max_timestamp="${FLAGS_max_timestamp}" ) 5768e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang fi 5778e768e9359d01008b286db1f6c54a6c3972a56b9Sen Jiang 5786f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang if [[ -n "${POSTINSTALL_CONFIG_FILE}" ]]; then 5796f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang GENERATOR_ARGS+=( 5806f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang --new_postinstall_config_file="${POSTINSTALL_CONFIG_FILE}" 5816f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang ) 5826f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang fi 5836f7b22c69bef53d33ae2bbb80b6682ff7ea8006eSen Jiang 584be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Running delta_generator with args: ${GENERATOR_ARGS[@]}" 5859a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 586be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 58789ff9e3221c358977f9c3124930ee6b559853740Alex Deymo echo "Done generating ${payload_type} update." 588be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 589be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 590be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_hash() { 591be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 59253f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify signature size with --signature_size SIZES" 593be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 594be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 59553f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the input unsigned payload with \ 596be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 597be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 598be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_hash_file}" ]] || 59953f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify --payload_hash_file FILENAME" 600f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma 601f514c54ce3b91a00abcfb2eededa5d38af8008ceJason Kusuma [[ -n "${FLAGS_metadata_hash_file}" ]] || 60253f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify --metadata_hash_file FILENAME" 603be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 604be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 605be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_hash() { 606bf1266fec223275ff19ef9624651946be9b9112aSen Jiang "${GENERATOR}" \ 607bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -in_file="${FLAGS_unsigned_payload}" \ 608bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -signature_size="${FLAGS_signature_size}" \ 609bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_hash_file="${FLAGS_payload_hash_file}" \ 610bf1266fec223275ff19ef9624651946be9b9112aSen Jiang -out_metadata_hash_file="${FLAGS_metadata_hash_file}" 611be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 612be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done generating hash." 613be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 614be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 615be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumavalidate_sign() { 616be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_signature_size}" ]] || 61753f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify signature size with --signature_size SIZES" 618be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 619be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_unsigned_payload}" ]] || 62053f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the input unsigned payload with \ 621be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--unsigned_payload FILENAME" 622be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 623be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload}" ]] || 62453f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the output signed payload with --payload FILENAME" 625be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 626be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma [[ -n "${FLAGS_payload_signature_file}" ]] || 62753f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the payload signature file with \ 628be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma--payload_signature_file SIGNATURES" 62989ff9e3221c358977f9c3124930ee6b559853740Alex Deymo 63089ff9e3221c358977f9c3124930ee6b559853740Alex Deymo [[ -n "${FLAGS_metadata_signature_file}" ]] || 63153f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the metadata signature file with \ 63289ff9e3221c358977f9c3124930ee6b559853740Alex Deymo--metadata_signature_file SIGNATURES" 633be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 634be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 635be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacmd_sign() { 6369a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS=( 6379a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -in_file="${FLAGS_unsigned_payload}" 6389a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_size="${FLAGS_signature_size}" 6399a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -signature_file="${FLAGS_payload_signature_file}" 6409a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -metadata_signature_file="${FLAGS_metadata_signature_file}" 6419a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma -out_file="${FLAGS_payload}" 6429a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma ) 6439a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 6449a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma if [[ -n "${FLAGS_metadata_size_file}" ]]; then 6459a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" ) 6469a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma fi 6479a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma 6489a4cae24e671face0c8786ff87e9d3de405cf4c8Jason Kusuma "${GENERATOR}" "${GENERATOR_ARGS[@]}" 649be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma echo "Done signing payload." 650be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma} 651be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 65298e691cc6debda3ef671acba001bebd45da36377Alex Deymovalidate_properties() { 65398e691cc6debda3ef671acba001bebd45da36377Alex Deymo [[ -n "${FLAGS_payload}" ]] || 65453f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify the payload file with --payload FILENAME" 65598e691cc6debda3ef671acba001bebd45da36377Alex Deymo 65698e691cc6debda3ef671acba001bebd45da36377Alex Deymo [[ -n "${FLAGS_properties_file}" ]] || 65753f04d74f32397d8bffea9f020aaac2cafaab049Sen Jiang die "You must specify a non empty --properties_file FILENAME" 65898e691cc6debda3ef671acba001bebd45da36377Alex Deymo} 65998e691cc6debda3ef671acba001bebd45da36377Alex Deymo 66098e691cc6debda3ef671acba001bebd45da36377Alex Deymocmd_properties() { 66198e691cc6debda3ef671acba001bebd45da36377Alex Deymo "${GENERATOR}" \ 66298e691cc6debda3ef671acba001bebd45da36377Alex Deymo -in_file="${FLAGS_payload}" \ 66398e691cc6debda3ef671acba001bebd45da36377Alex Deymo -properties_file="${FLAGS_properties_file}" 66498e691cc6debda3ef671acba001bebd45da36377Alex Deymo} 66598e691cc6debda3ef671acba001bebd45da36377Alex Deymo 6661352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassanivalidate_verify() { 6671352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani [[ -n "${FLAGS_payload}" ]] || 6681352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani die "Error: you must specify an input filename with --payload FILENAME" 6691352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6701352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani [[ -n "${FLAGS_target_image}" ]] || 6711352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani die "Error: you must specify a target image with --target_image FILENAME" 6721352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani} 6731352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6741352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassanicmd_verify() { 6751352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani local payload_type="delta" 6761352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ -z "${FLAGS_source_image}" ]]; then 6771352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani payload_type="full" 6781352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 6791352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6801352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Extracting images for ${payload_type} update." 6811352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6821352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ "${payload_type}" == "delta" ]]; then 6831352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani extract_image "${FLAGS_source_image}" SRC_PARTITIONS 6841352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 6851352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani extract_image "${FLAGS_target_image}" DST_PARTITIONS PARTITIONS_ORDER 6861352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6871352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani declare -A TMP_PARTITIONS 6881352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani for part in "${PARTITIONS_ORDER[@]}"; do 6891352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani local tmp_part=$(create_tempfile "tmp_part.bin.XXXXXX") 6901352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Creating temporary target partition ${tmp_part} for ${part}" 6911352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani CLEANUP_FILES+=("${tmp_part}") 6921352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani TMP_PARTITIONS[${part}]=${tmp_part} 6931352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani local FILESIZE=$(stat -c%s "${DST_PARTITIONS[${part}]}") 6941352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Truncating ${TMP_PARTITIONS[${part}]} to ${FILESIZE}" 6951352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani truncate_file "${TMP_PARTITIONS[${part}]}" "${FILESIZE}" 6961352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani done 6971352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 6981352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Verifying ${payload_type} update." 6991352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani # Common payload args: 7001352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani GENERATOR_ARGS=( -in_file="${FLAGS_payload}" ) 7011352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 7021352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani local part old_partitions="" new_partitions="" partition_names="" 7031352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani for part in "${PARTITIONS_ORDER[@]}"; do 7041352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ -n "${partition_names}" ]]; then 7051352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani partition_names+=":" 7061352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani new_partitions+=":" 7071352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani old_partitions+=":" 7081352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 7091352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani partition_names+="${part}" 7101352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani new_partitions+="${TMP_PARTITIONS[${part}]}" 7111352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani old_partitions+="${SRC_PARTITIONS[${part}]:-}" 7121352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani done 7131352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 7141352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani # Target image args: 7151352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani GENERATOR_ARGS+=( 7161352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani -partition_names="${partition_names}" 7171352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani -new_partitions="${new_partitions}" 7181352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani ) 7191352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 7201352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ "${payload_type}" == "delta" ]]; then 7211352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani # Source image args: 7221352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani GENERATOR_ARGS+=( 7231352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani -old_partitions="${old_partitions}" 7241352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani ) 7251352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 7261352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 727a566cb638023f12c9a64df31d83c464c13cf4d92Amin Hassani if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then 728a566cb638023f12c9a64df31d83c464c13cf4d92Amin Hassani GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" ) 729a566cb638023f12c9a64df31d83c464c13cf4d92Amin Hassani fi 730a566cb638023f12c9a64df31d83c464c13cf4d92Amin Hassani 7311352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Running delta_generator to verify ${payload_type} payload with args: \ 7321352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani${GENERATOR_ARGS[@]}" 7331352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani "${GENERATOR}" "${GENERATOR_ARGS[@]}" 7341352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 7351352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ $? -eq 0 ]]; then 7361352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Done applying ${payload_type} update." 7371352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Checking the newly generated partitions against the target partitions" 7381352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani for part in "${PARTITIONS_ORDER[@]}"; do 7391352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani cmp "${TMP_PARTITIONS[${part}]}" "${DST_PARTITIONS[${part}]}" 7401352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani local not_str="" 7411352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani if [[ $? -ne 0 ]]; then 7421352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani not_str="in" 7431352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 7441352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "The new partition (${part}) is ${not_str}valid." 7451352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani done 7461352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani else 7471352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani echo "Failed to apply ${payload_type} update." 7481352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani fi 7491352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani} 7501352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani 751be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma# Sanity check that the real generator exists: 752135197526b9c0f5a9acac57a05129339f311bbcaSen JiangGENERATOR="$(which delta_generator || true)" 753be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma[[ -x "${GENERATOR}" ]] || die "can't find delta_generator" 754be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma 755be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumacase "$COMMAND" in 756be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma generate) validate_generate 757be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_generate 758be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 759be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma hash) validate_hash 760be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_hash 761be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 762be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma sign) validate_sign 763be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma cmd_sign 764be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusuma ;; 76598e691cc6debda3ef671acba001bebd45da36377Alex Deymo properties) validate_properties 76698e691cc6debda3ef671acba001bebd45da36377Alex Deymo cmd_properties 76798e691cc6debda3ef671acba001bebd45da36377Alex Deymo ;; 7681352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani verify) validate_verify 7691352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani cmd_verify 7701352093b7e5f1134ebfa99ad1991753ff4f3f33aAmin Hassani ;; 771be998f47ccb02c5f118f73e79729e9187c39e919Jason Kusumaesac 772