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