From b20bc30d4c26e8f63dc3da5cd474a6681e9a487a Mon Sep 17 00:00:00 2001 From: Andrew de los Reyes Date: Mon, 9 Aug 2010 15:55:11 -0700 Subject: [PATCH] generate_update_payload script: script to generate full/delta update payloads This script wraps the old memento udpate generator and the new delta updater. It will also generate new-style full updates soon, which are full updates that use the delta-update format, when that code is ready. BUG=5248 TEST=ran local script to do full/delta update Review URL: http://codereview.chromium.org/2873082 --- cros_generate_update_payload | 176 +++++++++++++++++++++++++++++++++++ 1 file changed, 176 insertions(+) create mode 100755 cros_generate_update_payload diff --git a/cros_generate_update_payload b/cros_generate_update_payload new file mode 100755 index 0000000000..5edc756f8c --- /dev/null +++ b/cros_generate_update_payload @@ -0,0 +1,176 @@ +#!/bin/bash + +# Copyright (c) 2010 The Chromium OS Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. + +# Script to generate a Chromium OS update for use by the update engine. +# If a source .bin is specified, the update is assumed to be a delta update. + +# Load common constants. This should be the first executable line. +# The path to common.sh should be relative to your script's location. +. "$(dirname "$0")/common.sh" + +# Load functions and constants for chromeos-install +. "$(dirname "$0")/chromeos-common.sh" + +SRC_MNT="" +DST_MNT="" +SRC_KERNEL="" +SRC_ROOT="" +DST_KERNEL="" +DST_ROOT="" +STATE_MNT="" +STATE_LOOP_DEV="" + +# Pass an arg to not exit 1 at the end +cleanup() { + set +e + echo "Cleaning up" + if [ -n "$SRC_MNT" ]; then + sudo umount -d "$SRC_MNT" + [ -d "$SRC_MNT" ] && rmdir "$SRC_MNT" + SRC_MNT="" + fi + if [ -n "$DST_MNT" ]; then + sudo umount -d "$DST_MNT" + [ -d "$DST_MNT" ] && rmdir "$DST_MNT" + DST_MNT="" + fi + if [ -n "$STATE_MNT" ]; then + sudo umount "$STATE_MNT" + [ -d "$STATE_MNT" ] && rmdir "$STATE_MNT" + STATE_MNT="" + fi + if [ -n "$STATE_LOOP_DEV" ]; then + sudo losetup -d "$STATE_LOOP_DEV" + STATE_LOOP_DEV="" + fi + rm -f "$SRC_KERNEL" + rm -f "$SRC_ROOT" + rm -f "$DST_KERNEL" + rm -f "$DST_ROOT" + [ -n "$1" ] || exit 1 +} + +extract_partition_to_temp_file() { + local filename="$1" + local partition="$2" + local temp_file=$(mktemp /tmp/generate_update_payload.XXXXXX) + + local offset=$(partoffset "${filename}" ${partition}) # 512-byte sectors + local length=$(partsize "${filename}" ${partition}) # 512-byte sectors + local bs=512 + local sectors_per_two_mib=$((2 * 1024 * 1024 / 512)) + if [ $(( $offset % $sectors_per_two_mib )) -eq 0 -a \ + $(( $length % $sectors_per_two_mib )) -eq 0 ]; then + bs=$((2 * 1024 * 1024)) + offset=$(($offset / $sectors_per_two_mib)) + length=$(($length / $sectors_per_two_mib)) + else + warn "partition offset or length not at 2MiB boundary" + fi + dd if="$filename" of="$temp_file" bs=$bs count="$length" skip="$offset" + echo "$temp_file" +} + +patch_kernel() { + local IMAGE="$1" + local KERN_FILE="$2" + + STATE_LOOP_DEV=$(sudo losetup -f) + [ -n "$STATE_LOOP_DEV" ] || die "no free loop device" + local offset=$(partoffset "${IMAGE}" 1) + offset=$(($offset * 512)) + sudo losetup -o "$offset" "$STATE_LOOP_DEV" "$IMAGE" + STATE_MNT=$(mktemp -d /tmp/state.XXXXXX) + sudo mount "$STATE_LOOP_DEV" "$STATE_MNT" + dd if="$STATE_MNT"/vmlinuz_hd.vblock of="$KERN_FILE" conv=notrunc + sudo umount "$STATE_MNT" + STATE_MNT="" + sudo losetup -d "$STATE_LOOP_DEV" + STATE_LOOP_DEV="" +} + +# We should be in the chroot. +assert_inside_chroot + +DEFINE_string image "" "The image that should be sent to clients." +DEFINE_string src_image "" "Optional: a source image. If specified, this makes\ + a delta update." +DEFINE_boolean old_style "$FLAGS_TRUE" "Generate an old-style .gz full update." +DEFINE_string output "" "Output file" + +# Parse command line +FLAGS "$@" || exit 1 +eval set -- "${FLAGS_ARGV}" + +set -e + +locate_gpt + +DELTA=$FLAGS_TRUE +[ -n "$FLAGS_output" ] || die \ + "Error: you must specify an output filename with --output FILENAME" + +if [ -z "$FLAGS_src_image" ]; then + DELTA=$FLAGS_FALSE + if [ "$FLAGS_old_style" = "$FLAGS_TRUE" ]; then + echo "Generating an old-style full update" + else + die "Generating a new-style full update not yet supported" + fi +fi + +if [ "$DELTA" -eq "$FLAGS_TRUE" ]; then + echo "Generating a delta update" + + # Sanity check that the real generator exists: + GENERATOR="$(dirname "$0")/../platform/update_engine/delta_generator" + [ -x "$GENERATOR" ] || die "$GENERATOR doesn't exist, or isn't executable" + + trap cleanup INT TERM EXIT + SRC_KERNEL=$(extract_partition_to_temp_file "$FLAGS_src_image" 2) + patch_kernel "$FLAGS_src_image" "$SRC_KERNEL" + SRC_ROOT=$(extract_partition_to_temp_file "$FLAGS_src_image" 3) + + echo md5sum of src kernel: + md5sum "$SRC_KERNEL" + echo md5sum of src root: + md5sum "$SRC_ROOT" + + DST_KERNEL=$(extract_partition_to_temp_file "$FLAGS_image" 2) + patch_kernel "$FLAGS_image" "$DST_KERNEL" + DST_ROOT=$(extract_partition_to_temp_file "$FLAGS_image" 3) + + SRC_MNT=$(mktemp -d /tmp/src_root.XXXXXX) + sudo mount -o loop,ro "$SRC_ROOT" "$SRC_MNT" + + DST_MNT=$(mktemp -d /tmp/src_root.XXXXXX) + sudo mount -o loop,ro "$DST_ROOT" "$DST_MNT" + + sudo "$GENERATOR" \ + -new_dir "$DST_MNT" -new_image "$DST_ROOT" -new_kernel "$DST_KERNEL" \ + -old_dir "$SRC_MNT" -old_image "$SRC_ROOT" -old_kernel "$SRC_KERNEL" \ + -out_file "$FLAGS_output" + + trap - INT TERM EXIT + cleanup noexit + echo "Done generating delta." +else + echo "Generating full update" + + trap cleanup INT TERM EXIT + DST_KERNEL=$(extract_partition_to_temp_file "$FLAGS_image" 2) + patch_kernel "$FLAGS_image" "$DST_KERNEL" + DST_ROOT=$(extract_partition_to_temp_file "$FLAGS_image" 3) + + GENERATOR="$(dirname "$0")/mk_memento_images.sh" + + "$GENERATOR" "$DST_KERNEL" "$DST_ROOT" + mv "$(dirname "$DST_KERNEL")/update.gz" "$FLAGS_output" + + trap - INT TERM EXIT + cleanup noexit + echo "Done generating full update." +fi