Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #!/bin/bash | |
| 2 | |
| 3 # Copyright (c) 2009 The Chromium OS Authors. All rights reserved. | |
| 4 # Use of this source code is governed by a BSD-style license that can be | |
| 5 # found in the LICENSE file. | |
| 6 | |
| 7 # This script takes a path to a rootfs.ext2 which was generated by | |
| 8 # build_image.sh and generates an image that can be used for auto | |
| 9 # update. | |
| 10 | |
| 11 set -e | |
| 12 | |
| 13 # --- BEGIN COMMON.SH BOILERPLATE --- | |
| 14 # Load common CrOS utilities. Inside the chroot this file is installed in | |
| 15 # /usr/lib/crosutils. Outside the chroot we find it relative to the script's | |
| 16 # location. | |
| 17 find_common_sh() { | |
| 18 local common_paths=(/usr/lib/crosutils $(dirname "$(readlink -f "$0")")) | |
| 19 local path | |
| 20 | |
| 21 SCRIPT_ROOT= | |
| 22 for path in "${common_paths[@]}"; do | |
| 23 if [ -r "${path}/common.sh" ]; then | |
| 24 SCRIPT_ROOT=${path} | |
| 25 break | |
| 26 fi | |
| 27 done | |
| 28 } | |
| 29 | |
| 30 find_common_sh | |
| 31 . "${SCRIPT_ROOT}/common.sh" || (echo "Unable to load common.sh" && exit 1) | |
| 32 # --- END COMMON.SH BOILERPLATE --- | |
| 33 | |
| 34 . "${SCRIPT_ROOT}/common.sh" || (echo "Unable to load common.sh" && exit 1) | |
|
Nick Sanders
2011/02/04 19:55:13
Why is it here twice?
| |
| 35 | |
| 36 # Load functions designed for image processing | |
| 37 if ! . "${SCRIPT_ROOT}/lib/cros_image_common.sh"; then | |
| 38 echo "ERROR: Cannot load required library: lib/cros_image_common.sh; Abort." | |
| 39 exit 1 | |
| 40 fi | |
| 41 | |
| 42 if [ -z "$2" -o -z "$1" ] || [ "${#@}" -ne 2 -a "${#@}" -ne 3 ]; then | |
| 43 echo "usage: $0 path/to/kernel_partition_img path/to/rootfs_partition_img" | |
| 44 echo " or $0 path/to/chromiumos_img kern_part_no rootfs_part_no" | |
| 45 exit 1 | |
| 46 fi | |
| 47 | |
| 48 if [ "$CROS_GENERATE_UPDATE_PAYLOAD_CALLED" != "1" ]; then | |
| 49 echo "WARNING:" | |
| 50 echo " This script should only be called from cros_generate_update_payload" | |
| 51 echo " Please run that script with --help to see how to use it." | |
| 52 fi | |
| 53 | |
| 54 if ! image_has_command pigz; then | |
| 55 (echo "WARNING:" | |
| 56 echo " Your system does not have pigz (parallel gzip) installed." | |
| 57 echo " COMPRESSING WILL BE VERY SLOW. It is recommended to install pigz" | |
| 58 if image_has_command apt-get; then | |
| 59 echo " by 'sudo apt-get install pigz'." | |
| 60 elif image_has_command emerge; then | |
| 61 echo " by 'sudo emerge pigz'." | |
| 62 fi) >&2 | |
| 63 fi | |
| 64 | |
| 65 if [ $(whoami) = "root" ]; then | |
| 66 echo "running $0 as root which is unneccessary" | |
| 67 fi | |
| 68 | |
| 69 # Determine the offset size, and file name of parameters | |
| 70 if [ -z "$3" ]; then | |
| 71 # kernnel_img rootfs_img | |
| 72 KPART="$1" | |
| 73 ROOT_PART="$2" | |
| 74 KPART_SIZE=$(stat -c%s "$KPART") | |
| 75 ROOT_PART_SIZE=$(stat -c%s "$ROOT_PART") | |
| 76 KPART_OFFSET=0 | |
| 77 KPART_SECTORS=$((KPART_SIZE / 512)) | |
| 78 ROOT_OFFSET=0 | |
| 79 ROOT_SECTORS=$((ROOT_PART_SIZE / 512)) | |
| 80 else | |
| 81 # chromiumos_img kern_part_no rootfs_part_no | |
| 82 KPART="$1" | |
| 83 ROOT_PART="$1" | |
| 84 KPART_OFFSET="$(image_part_offset "$KPART" "$2")" || | |
| 85 image_die "cannot retieve kernel partition offset" | |
| 86 KPART_SECTORS="$(image_part_size "$KPART" "$2")" || | |
| 87 image_die "cannot retieve kernel partition size" | |
| 88 ROOT_OFFSET="$(image_part_offset "$ROOT_PART" "$3")" || | |
| 89 image_die "cannot retieve root partition offset" | |
| 90 ROOT_SECTORS="$(image_part_size "$ROOT_PART" "$3")" || | |
| 91 image_die "cannot retieve root partition size" | |
| 92 KPART_SIZE=$((KPART_SECTORS * 512)) | |
| 93 fi | |
| 94 | |
| 95 # Sanity check size. | |
| 96 if [ "$KPART_SIZE" -gt $((16 * 1024 * 1024)) ]; then | |
| 97 echo "Kernel partition size ($KPART_SIZE bytes) greater than 16 MiB." | |
| 98 echo "That's too big." | |
| 99 exit 1 | |
| 100 fi | |
| 101 | |
| 102 FINAL_OUT_FILE=$(dirname "$1")/update.gz | |
| 103 | |
| 104 # Update payload format: | |
| 105 # [kernel_size: big-endian uint64][kernel_blob][rootfs_blob] | |
| 106 | |
| 107 # Prepare kernel_size by using printf as a number like 00000000003d0900, then | |
| 108 # sed to convert as: \x00\x00\x00\x00\x00\x3d\x09\x00, finally echo -e to | |
| 109 # convert into binary. | |
| 110 KPART_SIZE_SIGNATURE="$(printf "%016x" "$KPART_SIZE" | | |
| 111 sed 's/\([0-9a-f][0-9a-f]\)/\\x\1/g')" | |
| 112 | |
| 113 # Build the blob! | |
| 114 CS_AND_RET_CODES="$( | |
| 115 (echo -en "$KPART_SIZE_SIGNATURE" | |
| 116 echo "Compressing kernel..." >&2 | |
| 117 image_dump_partial_file "$KPART" "$KPART_OFFSET" "$KPART_SECTORS" | |
| 118 echo "Compressing rootfs..." >&2 | |
| 119 image_dump_partial_file "$ROOT_PART" "$ROOT_OFFSET" "$ROOT_SECTORS") | | |
| 120 image_gzip_compress -c -9 | | |
| 121 tee "$FINAL_OUT_FILE" | | |
| 122 openssl sha1 -binary | | |
| 123 openssl base64 | | |
| 124 tr '\n' ' ' | |
| 125 echo ${PIPESTATUS[*]})" | |
| 126 | |
| 127 EXPECTED_RET_CODES="0 0 0 0 0 0" | |
| 128 set -- $CS_AND_RET_CODES | |
| 129 CALC_CS="$1" | |
| 130 shift | |
| 131 RET_CODES="$@" | |
| 132 if [ "$RET_CODES" != "$EXPECTED_RET_CODES" ]; then | |
| 133 echo compression/hash failed. $RET_CODES | |
| 134 exit 1 | |
| 135 fi | |
| 136 | |
| 137 echo Success. hash is "$CALC_CS" | |
| OLD | NEW |