blob: 4ff489312d8a6bff7b64ef1de2dec064ceeba580 [file] [log] [blame]
Jason Kusumabe998f42015-09-03 15:53:13 -07001#!/bin/bash
2
Amin Hassani13520932017-07-26 11:26:05 -07003#
4# Copyright (C) 2015 The Android Open Source Project
5#
6# Licensed under the Apache License, Version 2.0 (the "License");
7# you may not use this file except in compliance with the License.
8# You may obtain a copy of the License at
9#
10# http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing, software
13# distributed under the License is distributed on an "AS IS" BASIS,
14# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15# See the License for the specific language governing permissions and
16# limitations under the License.
17#
Jason Kusumabe998f42015-09-03 15:53:13 -070018
19# Script to generate a Brillo update for use by the update engine.
20#
21# usage: brillo_update_payload COMMAND [ARGS]
22# The following commands are supported:
23# generate generate an unsigned payload
24# hash generate a payload or metadata hash
25# sign generate a signed payload
Alex Deymo98e691c2016-02-04 21:05:45 -080026# properties generate a properties file from a payload
Amin Hassani13520932017-07-26 11:26:05 -070027# verify verify a payload by recreating a target image.
Tudor Brindusb432db82018-06-29 13:13:27 -070028# check verify a payload using paycheck (static testing)
Jason Kusumabe998f42015-09-03 15:53:13 -070029#
30# Generate command arguments:
Tianjie Xu72d512c2019-08-21 15:20:35 -070031# --payload generated unsigned payload output file
32# --source_image if defined, generate a delta payload from the
33# specified image to the target_image
34# --target_image the target image that should be sent to clients
35# --metadata_size_file if defined, generate a file containing the size
36# of the ayload metadata in bytes to the specified
37# file
38# --disable_fec_computation Disable the on device fec data computation for
39# incremental update. This feature is enabled by
40# default
Jason Kusumabe998f42015-09-03 15:53:13 -070041#
42# Hash command arguments:
43# --unsigned_payload the input unsigned payload to generate the hash from
44# --signature_size signature sizes in bytes in the following format:
Alex Deymo89ff9e32015-09-15 19:29:01 -070045# "size1:size2[:...]"
Jason Kusumabe998f42015-09-03 15:53:13 -070046# --payload_hash_file if defined, generate a payload hash and output to the
47# specified file
48# --metadata_hash_file if defined, generate a metadata hash and output to the
49# specified file
50#
51# Sign command arguments:
Alex Deymo89ff9e32015-09-15 19:29:01 -070052# --unsigned_payload the input unsigned payload to insert the signatures
53# --payload the output signed payload
54# --signature_size signature sizes in bytes in the following format:
55# "size1:size2[:...]"
56# --payload_signature_file the payload signature files in the following
57# format:
58# "payload_signature1:payload_signature2[:...]"
59# --metadata_signature_file the metadata signature files in the following
60# format:
61# "metadata_signature1:metadata_signature2[:...]"
Jason Kusuma9a4cae22015-10-08 18:17:57 -070062# --metadata_size_file if defined, generate a file containing the size of
63# the signed payload metadata in bytes to the
64# specified file
Jason Kusumabe998f42015-09-03 15:53:13 -070065# Note that the number of signature sizes and payload signatures have to match.
Alex Deymo98e691c2016-02-04 21:05:45 -080066#
67# Properties command arguments:
68# --payload the input signed or unsigned payload
69# --properties_file the output path where to write the properties, or
70# '-' for stdout.
Amin Hassani13520932017-07-26 11:26:05 -070071# Verify command arguments:
72# --payload payload input file
73# --source_image verify payload to the specified source image.
74# --target_image the target image to verify upon.
Tudor Brindusb432db82018-06-29 13:13:27 -070075#
76# Check command arguments:
77# Symmetrical with the verify command.
Alex Deymo98e691c2016-02-04 21:05:45 -080078
Jason Kusumabe998f42015-09-03 15:53:13 -070079
Alex Deymo61e1fa82016-01-19 15:16:34 -080080# Exit codes:
81EX_UNSUPPORTED_DELTA=100
82
Jason Kusumaf514c542015-11-05 18:43:45 -080083warn() {
84 echo "brillo_update_payload: warning: $*" >&2
85}
86
Gilad Arnold957ce122015-10-14 16:02:55 -070087die() {
88 echo "brillo_update_payload: error: $*" >&2
89 exit 1
Jason Kusumabe998f42015-09-03 15:53:13 -070090}
91
Gilad Arnold957ce122015-10-14 16:02:55 -070092# Loads shflags. We first look at the default install location; then look for
Amin Hassani2a14d412018-05-31 13:01:09 -070093# crosutils (chroot); finally check our own directory.
Gilad Arnold957ce122015-10-14 16:02:55 -070094load_shflags() {
95 local my_dir="$(dirname "$(readlink -f "$0")")"
96 local path
Mike Frysinger26bb8652018-10-12 02:15:09 -040097 for path in /usr/share/misc "${my_dir}"/lib/shflags; do
Gilad Arnold957ce122015-10-14 16:02:55 -070098 if [[ -r "${path}/shflags" ]]; then
99 . "${path}/shflags" || die "Could not load ${path}/shflags."
100 return
101 fi
102 done
103 die "Could not find shflags."
104}
105
106load_shflags
Jason Kusumabe998f42015-09-03 15:53:13 -0700107
Alex Deymoc64ffd52015-09-25 18:10:07 -0700108HELP_GENERATE="generate: Generate an unsigned update payload."
109HELP_HASH="hash: Generate the hashes of the unsigned payload and metadata used \
110for signing."
111HELP_SIGN="sign: Insert the signatures into the unsigned payload."
Alex Deymo98e691c2016-02-04 21:05:45 -0800112HELP_PROPERTIES="properties: Extract payload properties to a file."
Tudor Brindusb432db82018-06-29 13:13:27 -0700113HELP_VERIFY="verify: Verify a (signed) update payload using delta_generator."
114HELP_CHECK="check: Check a (signed) update payload using paycheck (static \
115testing)."
Alex Deymoc64ffd52015-09-25 18:10:07 -0700116
117usage() {
118 echo "Supported commands:"
119 echo
120 echo "${HELP_GENERATE}"
121 echo "${HELP_HASH}"
122 echo "${HELP_SIGN}"
Alex Deymo98e691c2016-02-04 21:05:45 -0800123 echo "${HELP_PROPERTIES}"
Amin Hassani13520932017-07-26 11:26:05 -0700124 echo "${HELP_VERIFY}"
Tudor Brindusb432db82018-06-29 13:13:27 -0700125 echo "${HELP_CHECK}"
Alex Deymoc64ffd52015-09-25 18:10:07 -0700126 echo
127 echo "Use: \"$0 <command> --help\" for more options."
128}
129
130# Check that a command is specified.
Jason Kusumabe998f42015-09-03 15:53:13 -0700131if [[ $# -lt 1 ]]; then
Tudor Brindusb432db82018-06-29 13:13:27 -0700132 echo "Please specify a command [generate|hash|sign|properties|verify|check]"
Jason Kusumabe998f42015-09-03 15:53:13 -0700133 exit 1
134fi
135
Alex Deymoc64ffd52015-09-25 18:10:07 -0700136# Parse command.
137COMMAND="${1:-}"
138shift
139
140case "${COMMAND}" in
141 generate)
142 FLAGS_HELP="${HELP_GENERATE}"
143 ;;
144
145 hash)
146 FLAGS_HELP="${HELP_HASH}"
147 ;;
148
149 sign)
150 FLAGS_HELP="${HELP_SIGN}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700151 ;;
Alex Deymo98e691c2016-02-04 21:05:45 -0800152
153 properties)
154 FLAGS_HELP="${HELP_PROPERTIES}"
155 ;;
Amin Hassani13520932017-07-26 11:26:05 -0700156
157 verify)
158 FLAGS_HELP="${HELP_VERIFY}"
159 ;;
160
Tudor Brindusb432db82018-06-29 13:13:27 -0700161 check)
162 FLAGS_HELP="${HELP_CHECK}"
163 ;;
164
Jason Kusumabe998f42015-09-03 15:53:13 -0700165 *)
Alex Deymoc64ffd52015-09-25 18:10:07 -0700166 echo "Unrecognized command: \"${COMMAND}\"" >&2
167 usage >&2
Jason Kusumabe998f42015-09-03 15:53:13 -0700168 exit 1
169 ;;
170esac
171
Jason Kusumabe998f42015-09-03 15:53:13 -0700172# Flags
Alex Deymoc64ffd52015-09-25 18:10:07 -0700173FLAGS_HELP="Usage: $0 ${COMMAND} [flags]
174${FLAGS_HELP}"
175
176if [[ "${COMMAND}" == "generate" ]]; then
177 DEFINE_string payload "" \
178 "Path to output the generated unsigned payload file."
179 DEFINE_string target_image "" \
180 "Path to the target image that should be sent to clients."
181 DEFINE_string source_image "" \
182 "Optional: Path to a source image. If specified, this makes a delta update."
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700183 DEFINE_string metadata_size_file "" \
184 "Optional: Path to output metadata size."
Sen Jiang8e768e92017-06-28 17:13:19 -0700185 DEFINE_string max_timestamp "" \
186 "Optional: The maximum unix timestamp of the OS allowed to apply this \
187payload, should be set to a number higher than the build timestamp of the \
188system running on the device, 0 if not specified."
Kelvin Zhang1f496422020-08-11 17:18:23 -0400189 DEFINE_string partition_timestamps "" \
190 "Optional: Per-partition maximum unix timestamp of the OS allowed to \
191apply this payload. Should be a comma separated key value pairs. e.x.\
192system:1234,vendor:456"
Tianjie Xu72d512c2019-08-21 15:20:35 -0700193 DEFINE_string disable_fec_computation "" \
194 "Optional: Disables the on device fec data computation for incremental \
195update. This feature is enabled by default."
Kelvin Zhang098e79a2020-11-19 17:40:56 -0500196 DEFINE_string disable_verity_computation "" \
197 "Optional: Disables the on device verity computation for incremental \
198update. This feature is enabled by default."
Tianjief5baff42020-07-17 21:43:22 -0700199 DEFINE_string is_partial_update "" \
200 "Optional: True if the payload is for partial update. i.e. it only updates \
201a subset of partitions on device."
Kelvin Zhangdde2ef42020-11-20 12:26:19 -0500202 DEFINE_string full_boot "" "Will include full boot image"
Alex Deymoc64ffd52015-09-25 18:10:07 -0700203fi
204if [[ "${COMMAND}" == "hash" || "${COMMAND}" == "sign" ]]; then
205 DEFINE_string unsigned_payload "" "Path to the input unsigned payload."
206 DEFINE_string signature_size "" \
207 "Signature sizes in bytes in the following format: size1:size2[:...]"
208fi
209if [[ "${COMMAND}" == "hash" ]]; then
210 DEFINE_string metadata_hash_file "" \
211 "Optional: Path to output metadata hash file."
212 DEFINE_string payload_hash_file "" \
213 "Optional: Path to output payload hash file."
214fi
215if [[ "${COMMAND}" == "sign" ]]; then
216 DEFINE_string payload "" \
217 "Path to output the generated unsigned payload file."
218 DEFINE_string metadata_signature_file "" \
219 "The metatada signatures in the following format: \
220metadata_signature1:metadata_signature2[:...]"
221 DEFINE_string payload_signature_file "" \
222 "The payload signatures in the following format: \
223payload_signature1:payload_signature2[:...]"
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700224 DEFINE_string metadata_size_file "" \
225 "Optional: Path to output metadata size."
Alex Deymoc64ffd52015-09-25 18:10:07 -0700226fi
Alex Deymo98e691c2016-02-04 21:05:45 -0800227if [[ "${COMMAND}" == "properties" ]]; then
228 DEFINE_string payload "" \
229 "Path to the input signed or unsigned payload file."
230 DEFINE_string properties_file "-" \
231 "Path to output the extracted property files. If '-' is passed stdout will \
232be used."
233fi
Tudor Brindusb432db82018-06-29 13:13:27 -0700234if [[ "${COMMAND}" == "verify" || "${COMMAND}" == "check" ]]; then
Amin Hassani13520932017-07-26 11:26:05 -0700235 DEFINE_string payload "" \
236 "Path to the input payload file."
237 DEFINE_string target_image "" \
238 "Path to the target image to verify upon."
239 DEFINE_string source_image "" \
240 "Optional: Path to a source image. If specified, the delta update is \
241applied to this."
242fi
Alex Deymo98e691c2016-02-04 21:05:45 -0800243
Alex Deymo5fbb1102017-01-12 13:55:52 -0800244DEFINE_string work_dir "${TMPDIR:-/tmp}" "Where to dump temporary files."
Jason Kusumabe998f42015-09-03 15:53:13 -0700245
246# Parse command line flag arguments
247FLAGS "$@" || exit 1
248eval set -- "${FLAGS_ARGV}"
Alex Deymo89ff9e32015-09-15 19:29:01 -0700249set -e
Jason Kusumabe998f42015-09-03 15:53:13 -0700250
Alex Deymo5fbb1102017-01-12 13:55:52 -0800251# Override the TMPDIR with the passed work_dir flags, which anyway defaults to
252# ${TMPDIR}.
253TMPDIR="${FLAGS_work_dir}"
254export TMPDIR
255
Alex Deymo89ff9e32015-09-15 19:29:01 -0700256# Associative arrays from partition name to file in the source and target
257# images. The size of the updated area must be the size of the file.
258declare -A SRC_PARTITIONS
259declare -A DST_PARTITIONS
260
Alex Deymo20bdc702016-12-07 21:07:11 -0800261# Associative arrays for the .map files associated with each src/dst partition
262# file in SRC_PARTITIONS and DST_PARTITIONS.
263declare -A SRC_PARTITIONS_MAP
264declare -A DST_PARTITIONS_MAP
265
Sen Jiang788c2d92016-03-09 12:48:40 -0800266# List of partition names in order.
267declare -a PARTITIONS_ORDER
268
Tao Bao96489902019-04-02 16:25:03 -0700269# A list of PIDs of the extract_image workers.
270EXTRACT_IMAGE_PIDS=()
271
Alex Deymo89ff9e32015-09-15 19:29:01 -0700272# A list of temporary files to remove during cleanup.
273CLEANUP_FILES=()
274
Alex Deymo48b502a2015-09-17 19:00:18 -0700275# Global options to force the version of the payload.
276FORCE_MAJOR_VERSION=""
277FORCE_MINOR_VERSION=""
278
Sen Jiang6f7b22c2015-11-12 15:50:39 -0800279# Path to the postinstall config file in target image if exists.
280POSTINSTALL_CONFIG_FILE=""
281
Yifan Hong398cb542018-10-18 11:29:40 -0700282# Path to the dynamic partition info file in target image if exists.
283DYNAMIC_PARTITION_INFO_FILE=""
284
Kelvin Zhang4e5da3e2021-01-21 11:54:36 -0500285# Path to the META/apex_info.pb found in target build
286APEX_INFO_FILE=""
287
Alex Deymoc97df432015-09-25 17:23:52 -0700288# read_option_int <file.txt> <option_key> [default_value]
289#
290# Reads the unsigned integer value associated with |option_key| in a key=value
291# file |file.txt|. Prints the read value if found and valid, otherwise prints
292# the |default_value|.
293read_option_uint() {
294 local file_txt="$1"
295 local option_key="$2"
296 local default_value="${3:-}"
297 local value
Tao Baoc288d5b2019-10-03 13:47:06 -0700298 if value=$(grep "^${option_key}=" "${file_txt}" | tail -n 1); then
Alex Deymoc97df432015-09-25 17:23:52 -0700299 if value=$(echo "${value}" | cut -f 2- -d "=" | grep -E "^[0-9]+$"); then
300 echo "${value}"
301 return
302 fi
303 fi
304 echo "${default_value}"
305}
306
Sen Jiangd0e9a892016-07-22 16:28:07 -0700307# truncate_file <file_path> <file_size>
308#
Dan Willemsen32711862018-10-04 21:25:50 -0700309# Truncate the given |file_path| to |file_size| using python.
Sen Jiangd0e9a892016-07-22 16:28:07 -0700310# The truncate binary might not be available.
311truncate_file() {
312 local file_path="$1"
313 local file_size="$2"
Dan Willemsen32711862018-10-04 21:25:50 -0700314 python -c "open(\"${file_path}\", 'a').truncate(${file_size})"
Sen Jiangd0e9a892016-07-22 16:28:07 -0700315}
316
Alex Deymo89ff9e32015-09-15 19:29:01 -0700317# Create a temporary file in the work_dir with an optional pattern name.
318# Prints the name of the newly created file.
319create_tempfile() {
320 local pattern="${1:-tempfile.XXXXXX}"
321 mktemp --tmpdir="${FLAGS_work_dir}" "${pattern}"
322}
Jason Kusumabe998f42015-09-03 15:53:13 -0700323
324cleanup() {
325 local err=""
Alex Deymo89ff9e32015-09-15 19:29:01 -0700326 rm -f "${CLEANUP_FILES[@]}" || err=1
Jason Kusumabe998f42015-09-03 15:53:13 -0700327
328 # If we are cleaning up after an error, or if we got an error during
329 # cleanup (even if we eventually succeeded) return a non-zero exit
330 # code. This triggers additional logging in most environments that call
331 # this script.
332 if [[ -n "${err}" ]]; then
333 die "Cleanup encountered an error."
334 fi
335}
336
337cleanup_on_error() {
338 trap - INT TERM ERR EXIT
339 cleanup
340 die "Cleanup success after an error."
341}
342
343cleanup_on_exit() {
344 trap - INT TERM ERR EXIT
345 cleanup
346}
347
348trap cleanup_on_error INT TERM ERR
349trap cleanup_on_exit EXIT
350
Tianjie Xu14715ce2019-08-06 17:24:43 -0700351# extract_file <zip_file> <entry_name> <destination>
352#
353# Extracts |entry_name| from |zip_file| to |destination|.
354extract_file() {
355 local zip_file="$1"
356 local entry_name="$2"
357 local destination="$3"
358
359 # unzip -p won't report error upon ENOSPC. Therefore, create a temp directory
360 # as the destination of the unzip, and move the file to the intended
361 # destination.
362 local output_directory=$(
363 mktemp --directory --tmpdir="${FLAGS_work_dir}" "TEMP.XXXXXX")
364 unzip "${zip_file}" "${entry_name}" -d "${output_directory}" ||
365 { rm -rf "${output_directory}"; die "Failed to extract ${entry_name}"; }
366
367 mv "${output_directory}/${entry_name}" "${destination}"
368 rm -rf "${output_directory}"
369}
Alex Deymo48b502a2015-09-17 19:00:18 -0700370
Sen Jiang788c2d92016-03-09 12:48:40 -0800371# extract_image <image> <partitions_array> [partitions_order]
Alex Deymo48b502a2015-09-17 19:00:18 -0700372#
373# Detect the format of the |image| file and extract its updatable partitions
374# into new temporary files. Add the list of partition names and its files to the
Sen Jiang788c2d92016-03-09 12:48:40 -0800375# associative array passed in |partitions_array|. If |partitions_order| is
376# passed, set it to list of partition names in order.
Alex Deymo48b502a2015-09-17 19:00:18 -0700377extract_image() {
378 local image="$1"
379
380 # Brillo images are zip files. We detect the 4-byte magic header of the zip
381 # file.
Elliott Hughes5df503c2018-11-27 16:57:34 -0800382 local magic=$(xxd -p -l4 "${image}")
Alex Deymo48b502a2015-09-17 19:00:18 -0700383 if [[ "${magic}" == "504b0304" ]]; then
384 echo "Detected .zip file, extracting Brillo image."
385 extract_image_brillo "$@"
386 return
387 fi
388
389 # Chrome OS images are GPT partitioned disks. We should have the cgpt binary
390 # bundled here and we will use it to extract the partitions, so the GPT
391 # headers must be valid.
392 if cgpt show -q -n "${image}" >/dev/null; then
393 echo "Detected GPT image, extracting Chrome OS image."
394 extract_image_cros "$@"
395 return
396 fi
397
398 die "Couldn't detect the image format of ${image}"
399}
400
Sen Jiang788c2d92016-03-09 12:48:40 -0800401# extract_image_cros <image.bin> <partitions_array> [partitions_order]
Alex Deymo89ff9e32015-09-15 19:29:01 -0700402#
Alex Deymo48b502a2015-09-17 19:00:18 -0700403# Extract Chromium OS recovery images into new temporary files.
Alex Deymo89ff9e32015-09-15 19:29:01 -0700404extract_image_cros() {
405 local image="$1"
406 local partitions_array="$2"
Sen Jiang788c2d92016-03-09 12:48:40 -0800407 local partitions_order="${3:-}"
Alex Deymo89ff9e32015-09-15 19:29:01 -0700408
409 local kernel root
410 kernel=$(create_tempfile "kernel.bin.XXXXXX")
411 CLEANUP_FILES+=("${kernel}")
412 root=$(create_tempfile "root.bin.XXXXXX")
413 CLEANUP_FILES+=("${root}")
414
415 cros_generate_update_payload --extract \
416 --image "${image}" \
Amin Hassani58e01d62018-09-19 14:56:15 -0700417 --kern_path "${kernel}" --root_path "${root}"
Alex Deymo89ff9e32015-09-15 19:29:01 -0700418
Amin Hassani58e01d62018-09-19 14:56:15 -0700419 # Chrome OS now uses major_version 2 payloads for all boards.
420 # See crbug.com/794404 for more information.
421 FORCE_MAJOR_VERSION="2"
Alex Deymo83f2f702015-10-14 14:49:33 -0700422
Tudor Brindusdda79e22018-06-28 18:03:21 -0700423 eval ${partitions_array}[kernel]=\""${kernel}"\"
424 eval ${partitions_array}[root]=\""${root}"\"
Alex Deymo89ff9e32015-09-15 19:29:01 -0700425
Sen Jiang788c2d92016-03-09 12:48:40 -0800426 if [[ -n "${partitions_order}" ]]; then
Tudor Brindusdda79e22018-06-28 18:03:21 -0700427 eval "${partitions_order}=( \"root\" \"kernel\" )"
Sen Jiang788c2d92016-03-09 12:48:40 -0800428 fi
429
Alex Deymo89ff9e32015-09-15 19:29:01 -0700430 local part varname
Tudor Brindusdda79e22018-06-28 18:03:21 -0700431 for part in kernel root; do
Alex Deymo89ff9e32015-09-15 19:29:01 -0700432 varname="${partitions_array}[${part}]"
433 printf "md5sum of %s: " "${varname}"
434 md5sum "${!varname}"
435 done
436}
437
Sen Jiang3e5804d2018-09-06 15:53:00 -0700438# extract_partition_brillo <target_files.zip> <partitions_array> <partition>
439# <part_file> <part_map_file>
440#
441# Extract the <partition> from target_files zip file into <part_file> and its
442# map file into <part_map_file>.
443extract_partition_brillo() {
444 local image="$1"
445 local partitions_array="$2"
446 local part="$3"
447 local part_file="$4"
448 local part_map_file="$5"
449
450 # For each partition, we in turn look for its image file under IMAGES/ and
451 # RADIO/ in the given target_files zip file.
452 local path path_in_zip
453 for path in IMAGES RADIO; do
454 if unzip -l "${image}" "${path}/${part}.img" >/dev/null; then
455 path_in_zip="${path}"
456 break
457 fi
458 done
459 [[ -n "${path_in_zip}" ]] || die "Failed to find ${part}.img"
Tianjie Xu14715ce2019-08-06 17:24:43 -0700460 extract_file "${image}" "${path_in_zip}/${part}.img" "${part_file}"
Sen Jiang3e5804d2018-09-06 15:53:00 -0700461
462 # If the partition is stored as an Android sparse image file, we need to
463 # convert them to a raw image for the update.
Yifan Hong4b821d72018-12-07 17:26:04 -0800464 local magic=$(xxd -p -l4 "${part_file}")
Sen Jiang3e5804d2018-09-06 15:53:00 -0700465 if [[ "${magic}" == "3aff26ed" ]]; then
466 local temp_sparse=$(create_tempfile "${part}.sparse.XXXXXX")
467 echo "Converting Android sparse image ${part}.img to RAW."
468 mv "${part_file}" "${temp_sparse}"
469 simg2img "${temp_sparse}" "${part_file}"
470 rm -f "${temp_sparse}"
471 fi
472
473 # Extract the .map file (if one is available).
Tianjie Xu14715ce2019-08-06 17:24:43 -0700474 if unzip -l "${image}" "${path_in_zip}/${part}.map" > /dev/null; then
475 extract_file "${image}" "${path_in_zip}/${part}.map" "${part_map_file}"
476 fi
Sen Jiang3e5804d2018-09-06 15:53:00 -0700477
478 # delta_generator only supports images multiple of 4 KiB. For target images
479 # we pad the data with zeros if needed, but for source images we truncate
480 # down the data since the last block of the old image could be padded on
481 # disk with unknown data.
482 local filesize=$(stat -c%s "${part_file}")
483 if [[ $(( filesize % 4096 )) -ne 0 ]]; then
484 if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then
485 echo "Rounding DOWN partition ${part}.img to a multiple of 4 KiB."
486 : $(( filesize = filesize & -4096 ))
487 else
488 echo "Rounding UP partition ${part}.img to a multiple of 4 KiB."
489 : $(( filesize = (filesize + 4095) & -4096 ))
490 fi
491 truncate_file "${part_file}" "${filesize}"
492 fi
493
494 echo "Extracted ${partitions_array}[${part}]: ${filesize} bytes"
495}
496
Sen Jiang788c2d92016-03-09 12:48:40 -0800497# extract_image_brillo <target_files.zip> <partitions_array> [partitions_order]
Alex Deymo48b502a2015-09-17 19:00:18 -0700498#
499# Extract the A/B updated partitions from a Brillo target_files zip file into
500# new temporary files.
501extract_image_brillo() {
502 local image="$1"
503 local partitions_array="$2"
Sen Jiang788c2d92016-03-09 12:48:40 -0800504 local partitions_order="${3:-}"
Alex Deymo48b502a2015-09-17 19:00:18 -0700505
Alex Deymo48b502a2015-09-17 19:00:18 -0700506 local partitions=( "boot" "system" )
Alex Deymo168b5352015-11-04 13:51:52 -0800507 local ab_partitions_list
508 ab_partitions_list=$(create_tempfile "ab_partitions_list.XXXXXX")
509 CLEANUP_FILES+=("${ab_partitions_list}")
Tianjie Xu14715ce2019-08-06 17:24:43 -0700510 if unzip -l "${image}" "META/ab_partitions.txt" > /dev/null; then
511 extract_file "${image}" "META/ab_partitions.txt" "${ab_partitions_list}"
Alex Deymo168b5352015-11-04 13:51:52 -0800512 if grep -v -E '^[a-zA-Z0-9_-]*$' "${ab_partitions_list}" >&2; then
513 die "Invalid partition names found in the partition list."
514 fi
Sen Jiang34c711a2017-10-25 17:25:21 -0700515 # Get partition list without duplicates.
516 partitions=($(awk '!seen[$0]++' "${ab_partitions_list}"))
Alex Deymo168b5352015-11-04 13:51:52 -0800517 if [[ ${#partitions[@]} -eq 0 ]]; then
518 die "The list of partitions is empty. Can't generate a payload."
519 fi
520 else
521 warn "No ab_partitions.txt found. Using default."
522 fi
Sen Jiang3e5804d2018-09-06 15:53:00 -0700523 echo "List of A/B partitions for ${partitions_array}: ${partitions[@]}"
Alex Deymo48b502a2015-09-17 19:00:18 -0700524
Sen Jiang788c2d92016-03-09 12:48:40 -0800525 if [[ -n "${partitions_order}" ]]; then
526 eval "${partitions_order}=(${partitions[@]})"
527 fi
528
Alex Deymo83f2f702015-10-14 14:49:33 -0700529 # All Brillo updaters support major version 2.
530 FORCE_MAJOR_VERSION="2"
531
Alex Deymo48b502a2015-09-17 19:00:18 -0700532 if [[ "${partitions_array}" == "SRC_PARTITIONS" ]]; then
Sen Jiang6f7b22c2015-11-12 15:50:39 -0800533 # Source image
534 local ue_config=$(create_tempfile "ue_config.XXXXXX")
Alex Deymoc97df432015-09-25 17:23:52 -0700535 CLEANUP_FILES+=("${ue_config}")
Tianjie Xu14715ce2019-08-06 17:24:43 -0700536 if unzip -l "${image}" "META/update_engine_config.txt" > /dev/null; then
537 extract_file "${image}" "META/update_engine_config.txt" "${ue_config}"
538 else
Alex Deymoc97df432015-09-25 17:23:52 -0700539 warn "No update_engine_config.txt found. Assuming pre-release image, \
540using payload minor version 2"
541 fi
Alex Deymo83f2f702015-10-14 14:49:33 -0700542 # For delta payloads, we use the major and minor version supported by the
543 # old updater.
Alex Deymoc97df432015-09-25 17:23:52 -0700544 FORCE_MINOR_VERSION=$(read_option_uint "${ue_config}" \
545 "PAYLOAD_MINOR_VERSION" 2)
Alex Deymo83f2f702015-10-14 14:49:33 -0700546 FORCE_MAJOR_VERSION=$(read_option_uint "${ue_config}" \
547 "PAYLOAD_MAJOR_VERSION" 2)
Alex Deymo61e1fa82016-01-19 15:16:34 -0800548
549 # Brillo support for deltas started with minor version 3.
550 if [[ "${FORCE_MINOR_VERSION}" -le 2 ]]; then
551 warn "No delta support from minor version ${FORCE_MINOR_VERSION}. \
552Disabling deltas for this source version."
553 exit ${EX_UNSUPPORTED_DELTA}
554 fi
Sen Jiang6f7b22c2015-11-12 15:50:39 -0800555 else
556 # Target image
557 local postinstall_config=$(create_tempfile "postinstall_config.XXXXXX")
558 CLEANUP_FILES+=("${postinstall_config}")
Tianjie Xu14715ce2019-08-06 17:24:43 -0700559 if unzip -l "${image}" "META/postinstall_config.txt" > /dev/null; then
560 extract_file "${image}" "META/postinstall_config.txt" \
561 "${postinstall_config}"
Sen Jiang6f7b22c2015-11-12 15:50:39 -0800562 POSTINSTALL_CONFIG_FILE="${postinstall_config}"
563 fi
Yifan Hong398cb542018-10-18 11:29:40 -0700564 local dynamic_partitions_info=$(create_tempfile "dynamic_partitions_info.XXXXXX")
565 CLEANUP_FILES+=("${dynamic_partitions_info}")
Tianjie Xu14715ce2019-08-06 17:24:43 -0700566 if unzip -l "${image}" "META/dynamic_partitions_info.txt" > /dev/null; then
567 extract_file "${image}" "META/dynamic_partitions_info.txt" \
568 "${dynamic_partitions_info}"
Yifan Hong398cb542018-10-18 11:29:40 -0700569 DYNAMIC_PARTITION_INFO_FILE="${dynamic_partitions_info}"
570 fi
Kelvin Zhang4e5da3e2021-01-21 11:54:36 -0500571 local apex_info=$(create_tempfile "apex_info.XXXXXX")
572 CLEANUP_FILES+=("${apex_info}")
573 if unzip -l "${image}" "META/apex_info.pb" > /dev/null; then
574 extract_file "${image}" "META/apex_info.pb" \
575 "${apex_info}"
576 APEX_INFO_FILE="${apex_info}"
577 fi
Alex Deymo48b502a2015-09-17 19:00:18 -0700578 fi
579
Sen Jiang3e5804d2018-09-06 15:53:00 -0700580 local part
Alex Deymo48b502a2015-09-17 19:00:18 -0700581 for part in "${partitions[@]}"; do
Sen Jiang3e5804d2018-09-06 15:53:00 -0700582 local part_file=$(create_tempfile "${part}.img.XXXXXX")
583 local part_map_file=$(create_tempfile "${part}.map.XXXXXX")
584 CLEANUP_FILES+=("${part_file}" "${part_map_file}")
585 # Extract partitions in background.
586 extract_partition_brillo "${image}" "${partitions_array}" "${part}" \
587 "${part_file}" "${part_map_file}" &
Tao Bao96489902019-04-02 16:25:03 -0700588 EXTRACT_IMAGE_PIDS+=("$!")
Alex Deymo48b502a2015-09-17 19:00:18 -0700589 eval "${partitions_array}[\"${part}\"]=\"${part_file}\""
Alex Deymo20bdc702016-12-07 21:07:11 -0800590 eval "${partitions_array}_MAP[\"${part}\"]=\"${part_map_file}\""
Sen Jiang3e5804d2018-09-06 15:53:00 -0700591 done
592}
593
594# cleanup_partition_array <partitions_array>
595#
596# Remove all empty files in <partitions_array>.
597cleanup_partition_array() {
598 local partitions_array="$1"
599 # Have to use eval to iterate over associative array keys with variable array
600 # names, we should change it to use nameref once bash 4.3 is available
601 # everywhere.
602 for part in $(eval "echo \${!${partitions_array}[@]}"); do
603 local path="${partitions_array}[$part]"
604 if [[ ! -s "${!path}" ]]; then
605 eval "unset ${partitions_array}[${part}]"
606 fi
Alex Deymo48b502a2015-09-17 19:00:18 -0700607 done
608}
609
Tudor Brindusb432db82018-06-29 13:13:27 -0700610extract_payload_images() {
611 local payload_type=$1
612 echo "Extracting images for ${payload_type} update."
613
614 if [[ "${payload_type}" == "delta" ]]; then
615 extract_image "${FLAGS_source_image}" SRC_PARTITIONS
616 fi
617 extract_image "${FLAGS_target_image}" DST_PARTITIONS PARTITIONS_ORDER
Tao Bao96489902019-04-02 16:25:03 -0700618 # Wait for all subprocesses to finish. Not using `wait` since it doesn't die
619 # on non-zero subprocess exit code. Not using `wait ${EXTRACT_IMAGE_PIDS[@]}`
620 # as it gives the status of the last process it has waited for.
621 for pid in ${EXTRACT_IMAGE_PIDS[@]}; do
622 wait ${pid}
623 done
Sen Jiang3e5804d2018-09-06 15:53:00 -0700624 cleanup_partition_array SRC_PARTITIONS
625 cleanup_partition_array SRC_PARTITIONS_MAP
626 cleanup_partition_array DST_PARTITIONS
627 cleanup_partition_array DST_PARTITIONS_MAP
Tudor Brindusb432db82018-06-29 13:13:27 -0700628}
629
630get_payload_type() {
631 if [[ -z "${FLAGS_source_image}" ]]; then
632 echo "full"
633 else
634 echo "delta"
635 fi
636}
637
Jason Kusumabe998f42015-09-03 15:53:13 -0700638validate_generate() {
639 [[ -n "${FLAGS_payload}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700640 die "You must specify an output filename with --payload FILENAME"
Jason Kusumabe998f42015-09-03 15:53:13 -0700641
642 [[ -n "${FLAGS_target_image}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700643 die "You must specify a target image with --target_image FILENAME"
Jason Kusumabe998f42015-09-03 15:53:13 -0700644}
645
646cmd_generate() {
Sen Jiang3e5804d2018-09-06 15:53:00 -0700647 local payload_type=$(get_payload_type)
648 extract_payload_images ${payload_type}
Jason Kusumabe998f42015-09-03 15:53:13 -0700649
Alex Deymo48b502a2015-09-17 19:00:18 -0700650 echo "Generating ${payload_type} update."
Alex Deymo168b5352015-11-04 13:51:52 -0800651 # Common payload args:
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700652 GENERATOR_ARGS=( --out_file="${FLAGS_payload}" )
Alex Deymo168b5352015-11-04 13:51:52 -0800653
654 local part old_partitions="" new_partitions="" partition_names=""
Alex Deymo20bdc702016-12-07 21:07:11 -0800655 local old_mapfiles="" new_mapfiles=""
Sen Jiang788c2d92016-03-09 12:48:40 -0800656 for part in "${PARTITIONS_ORDER[@]}"; do
Alex Deymo168b5352015-11-04 13:51:52 -0800657 if [[ -n "${partition_names}" ]]; then
658 partition_names+=":"
659 new_partitions+=":"
660 old_partitions+=":"
Alex Deymo20bdc702016-12-07 21:07:11 -0800661 new_mapfiles+=":"
662 old_mapfiles+=":"
Alex Deymo168b5352015-11-04 13:51:52 -0800663 fi
664 partition_names+="${part}"
665 new_partitions+="${DST_PARTITIONS[${part}]}"
Kelvin Zhang999705e2020-11-03 10:07:09 -0500666 if [ "${FLAGS_full_boot}" == "true" ] && [ "${part}" == "boot" ]; then
667 # Skip boot partition.
668 old_partitions+=""
669 else
670 old_partitions+="${SRC_PARTITIONS[${part}]:-}"
671 fi
Alex Deymo20bdc702016-12-07 21:07:11 -0800672 new_mapfiles+="${DST_PARTITIONS_MAP[${part}]:-}"
673 old_mapfiles+="${SRC_PARTITIONS_MAP[${part}]:-}"
Alex Deymo168b5352015-11-04 13:51:52 -0800674 done
675
676 # Target image args:
677 GENERATOR_ARGS+=(
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700678 --partition_names="${partition_names}"
679 --new_partitions="${new_partitions}"
680 --new_mapfiles="${new_mapfiles}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700681 )
682
Tianjief5baff42020-07-17 21:43:22 -0700683 if [[ "${FLAGS_is_partial_update}" == "true" ]]; then
684 GENERATOR_ARGS+=( --is_partial_update="true" )
685 # Need at least minor version 7 for partial update, so generate with minor
686 # version 7 if we don't have a source image. Let the delta_generator to fail
687 # the other incompatiable minor versions.
688 if [[ -z "${FORCE_MINOR_VERSION}" ]]; then
689 FORCE_MINOR_VERSION="7"
690 fi
691 fi
692
Alex Deymo89ff9e32015-09-15 19:29:01 -0700693 if [[ "${payload_type}" == "delta" ]]; then
Alex Deymo168b5352015-11-04 13:51:52 -0800694 # Source image args:
Jason Kusumabe998f42015-09-03 15:53:13 -0700695 GENERATOR_ARGS+=(
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700696 --old_partitions="${old_partitions}"
697 --old_mapfiles="${old_mapfiles}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700698 )
Tianjie Xu72d512c2019-08-21 15:20:35 -0700699 if [[ -n "${FLAGS_disable_fec_computation}" ]]; then
700 GENERATOR_ARGS+=(
701 --disable_fec_computation="${FLAGS_disable_fec_computation}" )
702 fi
Kelvin Zhang098e79a2020-11-19 17:40:56 -0500703 if [[ -n "${FLAGS_disable_verity_computation}" ]]; then
704 GENERATOR_ARGS+=(
705 --disable_verity_computation="${FLAGS_disable_verity_computation}" )
706 fi
Alex Deymo48b502a2015-09-17 19:00:18 -0700707 fi
708
Tianjief5baff42020-07-17 21:43:22 -0700709 # minor version is set only for delta or partial payload.
710 if [[ -n "${FORCE_MINOR_VERSION}" ]]; then
711 GENERATOR_ARGS+=( --minor_version="${FORCE_MINOR_VERSION}" )
712 fi
713
Alex Deymo48b502a2015-09-17 19:00:18 -0700714 if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then
715 GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" )
Jason Kusumabe998f42015-09-03 15:53:13 -0700716 fi
717
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700718 if [[ -n "${FLAGS_metadata_size_file}" ]]; then
719 GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" )
720 fi
721
Sen Jiang8e768e92017-06-28 17:13:19 -0700722 if [[ -n "${FLAGS_max_timestamp}" ]]; then
723 GENERATOR_ARGS+=( --max_timestamp="${FLAGS_max_timestamp}" )
724 fi
725
Kelvin Zhang1f496422020-08-11 17:18:23 -0400726 if [[ -n "${FLAGS_partition_timestamps}" ]]; then
727 GENERATOR_ARGS+=( --partition_timestamps="${FLAGS_partition_timestamps}" )
728 fi
729
Sen Jiang6f7b22c2015-11-12 15:50:39 -0800730 if [[ -n "${POSTINSTALL_CONFIG_FILE}" ]]; then
731 GENERATOR_ARGS+=(
732 --new_postinstall_config_file="${POSTINSTALL_CONFIG_FILE}"
733 )
734 fi
735
Yifan Hong398cb542018-10-18 11:29:40 -0700736 if [[ -n "{DYNAMIC_PARTITION_INFO_FILE}" ]]; then
737 GENERATOR_ARGS+=(
738 --dynamic_partition_info_file="${DYNAMIC_PARTITION_INFO_FILE}"
739 )
740 fi
Kelvin Zhang4e5da3e2021-01-21 11:54:36 -0500741 if [[ -n "{APEX_INFO_FILE}" ]]; then
742 GENERATOR_ARGS+=(
743 --apex_info_file="${APEX_INFO_FILE}"
744 )
745 fi
Yifan Hong398cb542018-10-18 11:29:40 -0700746
Jason Kusumabe998f42015-09-03 15:53:13 -0700747 echo "Running delta_generator with args: ${GENERATOR_ARGS[@]}"
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700748 "${GENERATOR}" "${GENERATOR_ARGS[@]}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700749
Alex Deymo89ff9e32015-09-15 19:29:01 -0700750 echo "Done generating ${payload_type} update."
Jason Kusumabe998f42015-09-03 15:53:13 -0700751}
752
753validate_hash() {
754 [[ -n "${FLAGS_signature_size}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700755 die "You must specify signature size with --signature_size SIZES"
Jason Kusumabe998f42015-09-03 15:53:13 -0700756
757 [[ -n "${FLAGS_unsigned_payload}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700758 die "You must specify the input unsigned payload with \
Jason Kusumabe998f42015-09-03 15:53:13 -0700759--unsigned_payload FILENAME"
760
Jason Kusumabe998f42015-09-03 15:53:13 -0700761 [[ -n "${FLAGS_payload_hash_file}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700762 die "You must specify --payload_hash_file FILENAME"
Jason Kusumaf514c542015-11-05 18:43:45 -0800763
764 [[ -n "${FLAGS_metadata_hash_file}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700765 die "You must specify --metadata_hash_file FILENAME"
Jason Kusumabe998f42015-09-03 15:53:13 -0700766}
767
768cmd_hash() {
Sen Jiangbf1266f2015-10-26 11:29:24 -0700769 "${GENERATOR}" \
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700770 --in_file="${FLAGS_unsigned_payload}" \
771 --signature_size="${FLAGS_signature_size}" \
772 --out_hash_file="${FLAGS_payload_hash_file}" \
773 --out_metadata_hash_file="${FLAGS_metadata_hash_file}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700774
Jason Kusumabe998f42015-09-03 15:53:13 -0700775 echo "Done generating hash."
776}
777
778validate_sign() {
779 [[ -n "${FLAGS_signature_size}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700780 die "You must specify signature size with --signature_size SIZES"
Jason Kusumabe998f42015-09-03 15:53:13 -0700781
782 [[ -n "${FLAGS_unsigned_payload}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700783 die "You must specify the input unsigned payload with \
Jason Kusumabe998f42015-09-03 15:53:13 -0700784--unsigned_payload FILENAME"
785
786 [[ -n "${FLAGS_payload}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700787 die "You must specify the output signed payload with --payload FILENAME"
Jason Kusumabe998f42015-09-03 15:53:13 -0700788
789 [[ -n "${FLAGS_payload_signature_file}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700790 die "You must specify the payload signature file with \
Jason Kusumabe998f42015-09-03 15:53:13 -0700791--payload_signature_file SIGNATURES"
Alex Deymo89ff9e32015-09-15 19:29:01 -0700792
793 [[ -n "${FLAGS_metadata_signature_file}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700794 die "You must specify the metadata signature file with \
Alex Deymo89ff9e32015-09-15 19:29:01 -0700795--metadata_signature_file SIGNATURES"
Jason Kusumabe998f42015-09-03 15:53:13 -0700796}
797
798cmd_sign() {
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700799 GENERATOR_ARGS=(
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700800 --in_file="${FLAGS_unsigned_payload}"
801 --signature_size="${FLAGS_signature_size}"
802 --payload_signature_file="${FLAGS_payload_signature_file}"
803 --metadata_signature_file="${FLAGS_metadata_signature_file}"
804 --out_file="${FLAGS_payload}"
Jason Kusuma9a4cae22015-10-08 18:17:57 -0700805 )
806
807 if [[ -n "${FLAGS_metadata_size_file}" ]]; then
808 GENERATOR_ARGS+=( --out_metadata_size_file="${FLAGS_metadata_size_file}" )
809 fi
810
811 "${GENERATOR}" "${GENERATOR_ARGS[@]}"
Jason Kusumabe998f42015-09-03 15:53:13 -0700812 echo "Done signing payload."
813}
814
Alex Deymo98e691c2016-02-04 21:05:45 -0800815validate_properties() {
816 [[ -n "${FLAGS_payload}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700817 die "You must specify the payload file with --payload FILENAME"
Alex Deymo98e691c2016-02-04 21:05:45 -0800818
819 [[ -n "${FLAGS_properties_file}" ]] ||
Sen Jiang53f04d72016-07-13 16:43:39 -0700820 die "You must specify a non empty --properties_file FILENAME"
Alex Deymo98e691c2016-02-04 21:05:45 -0800821}
822
823cmd_properties() {
824 "${GENERATOR}" \
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700825 --in_file="${FLAGS_payload}" \
826 --properties_file="${FLAGS_properties_file}"
Alex Deymo98e691c2016-02-04 21:05:45 -0800827}
828
Tudor Brindusb432db82018-06-29 13:13:27 -0700829validate_verify_and_check() {
Amin Hassani13520932017-07-26 11:26:05 -0700830 [[ -n "${FLAGS_payload}" ]] ||
831 die "Error: you must specify an input filename with --payload FILENAME"
832
833 [[ -n "${FLAGS_target_image}" ]] ||
834 die "Error: you must specify a target image with --target_image FILENAME"
835}
836
837cmd_verify() {
Tudor Brindusb432db82018-06-29 13:13:27 -0700838 local payload_type=$(get_payload_type)
839 extract_payload_images ${payload_type}
Amin Hassani13520932017-07-26 11:26:05 -0700840
841 declare -A TMP_PARTITIONS
842 for part in "${PARTITIONS_ORDER[@]}"; do
843 local tmp_part=$(create_tempfile "tmp_part.bin.XXXXXX")
844 echo "Creating temporary target partition ${tmp_part} for ${part}"
845 CLEANUP_FILES+=("${tmp_part}")
846 TMP_PARTITIONS[${part}]=${tmp_part}
847 local FILESIZE=$(stat -c%s "${DST_PARTITIONS[${part}]}")
848 echo "Truncating ${TMP_PARTITIONS[${part}]} to ${FILESIZE}"
849 truncate_file "${TMP_PARTITIONS[${part}]}" "${FILESIZE}"
850 done
851
852 echo "Verifying ${payload_type} update."
853 # Common payload args:
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700854 GENERATOR_ARGS=( --in_file="${FLAGS_payload}" )
Amin Hassani13520932017-07-26 11:26:05 -0700855
856 local part old_partitions="" new_partitions="" partition_names=""
857 for part in "${PARTITIONS_ORDER[@]}"; do
858 if [[ -n "${partition_names}" ]]; then
859 partition_names+=":"
860 new_partitions+=":"
861 old_partitions+=":"
862 fi
863 partition_names+="${part}"
864 new_partitions+="${TMP_PARTITIONS[${part}]}"
865 old_partitions+="${SRC_PARTITIONS[${part}]:-}"
866 done
867
868 # Target image args:
869 GENERATOR_ARGS+=(
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700870 --partition_names="${partition_names}"
871 --new_partitions="${new_partitions}"
Amin Hassani13520932017-07-26 11:26:05 -0700872 )
873
874 if [[ "${payload_type}" == "delta" ]]; then
875 # Source image args:
876 GENERATOR_ARGS+=(
Tudor Brindus5ec5bd12018-07-11 11:02:44 -0700877 --old_partitions="${old_partitions}"
Amin Hassani13520932017-07-26 11:26:05 -0700878 )
879 fi
880
Amin Hassania566cb62017-08-23 12:36:55 -0700881 if [[ -n "${FORCE_MAJOR_VERSION}" ]]; then
882 GENERATOR_ARGS+=( --major_version="${FORCE_MAJOR_VERSION}" )
883 fi
884
Amin Hassani13520932017-07-26 11:26:05 -0700885 echo "Running delta_generator to verify ${payload_type} payload with args: \
886${GENERATOR_ARGS[@]}"
Sen Jiang6feb15c2018-08-31 15:45:17 -0700887 "${GENERATOR}" "${GENERATOR_ARGS[@]}" || true
Amin Hassani13520932017-07-26 11:26:05 -0700888
Sen Jiang6feb15c2018-08-31 15:45:17 -0700889 echo "Done applying ${payload_type} update."
890 echo "Checking the newly generated partitions against the target partitions"
891 local need_pause=false
892 for part in "${PARTITIONS_ORDER[@]}"; do
893 local not_str=""
894 if ! cmp "${TMP_PARTITIONS[${part}]}" "${DST_PARTITIONS[${part}]}"; then
895 not_str="in"
896 need_pause=true
897 fi
898 echo "The new partition (${part}) is ${not_str}valid."
899 done
900 # All images will be cleaned up when script exits, pause here to give a chance
901 # to inspect the images.
902 if [[ "$need_pause" == true ]]; then
903 read -n1 -r -s -p "Paused to investigate invalid partitions, \
904press any key to exit."
Amin Hassani13520932017-07-26 11:26:05 -0700905 fi
906}
907
Tudor Brindusb432db82018-06-29 13:13:27 -0700908cmd_check() {
909 local payload_type=$(get_payload_type)
910 extract_payload_images ${payload_type}
911
912 local part dst_partitions="" src_partitions=""
913 for part in "${PARTITIONS_ORDER[@]}"; do
914 if [[ -n "${dst_partitions}" ]]; then
915 dst_partitions+=" "
916 src_partitions+=" "
917 fi
918 dst_partitions+="${DST_PARTITIONS[${part}]}"
919 src_partitions+="${SRC_PARTITIONS[${part}]:-}"
920 done
921
922 # Common payload args:
923 PAYCHECK_ARGS=( "${FLAGS_payload}" --type ${payload_type} \
924 --part_names ${PARTITIONS_ORDER[@]} \
925 --dst_part_paths ${dst_partitions} )
926
927 if [[ ! -z "${SRC_PARTITIONS[@]}" ]]; then
928 PAYCHECK_ARGS+=( --src_part_paths ${src_partitions} )
929 fi
930
931 echo "Checking ${payload_type} update."
932 check_update_payload ${PAYCHECK_ARGS[@]} --check
933}
934
Tianjiee283ce42020-07-29 11:37:51 -0700935# Check that the real generator exists:
Yifan Hong3756c3e2020-07-24 20:25:51 -0700936[[ -x "${GENERATOR}" ]] || GENERATOR="$(which delta_generator || true)"
Jason Kusumabe998f42015-09-03 15:53:13 -0700937[[ -x "${GENERATOR}" ]] || die "can't find delta_generator"
938
939case "$COMMAND" in
940 generate) validate_generate
941 cmd_generate
942 ;;
943 hash) validate_hash
944 cmd_hash
945 ;;
946 sign) validate_sign
947 cmd_sign
948 ;;
Alex Deymo98e691c2016-02-04 21:05:45 -0800949 properties) validate_properties
950 cmd_properties
951 ;;
Tudor Brindusb432db82018-06-29 13:13:27 -0700952 verify) validate_verify_and_check
Amin Hassani13520932017-07-26 11:26:05 -0700953 cmd_verify
954 ;;
Tudor Brindusb432db82018-06-29 13:13:27 -0700955 check) validate_verify_and_check
956 cmd_check
957 ;;
Jason Kusumabe998f42015-09-03 15:53:13 -0700958esac