1 #!/bin/bash 2 3 # Copyright (c) 2011 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 # Script to convert a recovery image into an SSD image. Changes are made in- 8 # place. 9 10 usage() { 11 cat <<EOF 12 Usage: $PROG <image> [--force] [--cgpt=/path/to/cgpt] 13 14 In-place converts recovery <image> into an SSD image. With --force, does not 15 ask for confirmation from the user. Use --cgpt= to specify cgpt binary location. 16 17 EOF 18 exit 1 19 } 20 21 if [ $# -lt 1 ] || [ $# -gt 3 ]; then 22 usage 23 else 24 IMAGE=$1 25 shift 26 fi 27 28 for arg in $*; do 29 case "$arg" in 30 --force) 31 IS_FORCE=$arg 32 ;; 33 --cgpt=*) 34 GPT=${arg#--cgpt=} 35 ;; 36 *) 37 usage 38 ;; 39 esac 40 done 41 42 # Load common constants (and use GPT if set above) and variables. 43 . "$(dirname "$0")/common_minimal.sh" 44 45 type -P $GPT &>/dev/null || 46 { echo "cgpt tool must be in the path or specified via --cgpt"; exit 1; } 47 48 # Abort on errors. 49 set -e 50 51 if [ "${IS_FORCE}" != "--force" ]; then 52 echo "This will modify ${IMAGE} in-place and convert it into an SSD image." 53 read -p "Are you sure you want to continue (y/N)?" SURE 54 SURE="${SURE:0:1}" 55 [ "${SURE}" != "y" ] && exit 1 56 fi 57 58 kerna_offset=$(partoffset ${IMAGE} 2) 59 kernb_offset=$(partoffset ${IMAGE} 4) 60 # Kernel partition sizes should be the same. 61 kern_size=$(partsize ${IMAGE} 2) 62 63 # Move Kernel B to Kernel A. 64 kernb=$(make_temp_file) 65 echo "Replacing Kernel partition A with Kernel partition B" 66 extract_image_partition ${IMAGE} 4 ${kernb} 67 replace_image_partition ${IMAGE} 2 ${kernb} 68 69 # Overwrite the vblock. 70 stateful_dir=$(make_temp_dir) 71 tmp_vblock=$(make_temp_file) 72 mount_image_partition_ro ${IMAGE} 1 ${stateful_dir} 73 sudo cp ${stateful_dir}/vmlinuz_hd.vblock ${tmp_vblock} 74 # Unmount before overwriting image to avoid sync issues. 75 sudo umount ${stateful_dir} 76 echo "Overwriting kernel partition A vblock with SSD vblock" 77 sudo dd if=${tmp_vblock} of=${IMAGE} seek=${kerna_offset} bs=512 conv=notrunc 78 79 # Zero out Kernel B partition. 80 echo "Zeroing out Kernel partition B" 81 sudo dd if=/dev/zero of=${IMAGE} seek=${kernb_offset} bs=512 count=${kern_size} conv=notrunc 82 echo "${IMAGE} was converted to an SSD image." 83