Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(138)

Side by Side Diff: mod_image_for_dev_recovery.sh

Issue 3432017: clean up excessive output printed by mod_image_for_dev_recovery (Closed) Base URL: http://git.chromium.org/git/crosutils.git
Patch Set: fix trap statements Created 10 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 #!/bin/bash 1 #!/bin/bash
2 2
3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
4 # Use of this source code is governed by a BSD-style license that can be 4 # Use of this source code is governed by a BSD-style license that can be
5 # found in the LICENSE file. 5 # found in the LICENSE file.
6 6
7 # Script to create a Chrome OS dev recovery image using a dev install shim 7 # Script to create a Chrome OS dev recovery image using a dev install shim
8 8
9 # Source constants and utility functions 9 # Source constants and utility functions
10 . "$(dirname "$0")/common.sh" 10 . "$(dirname "$0")/common.sh"
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after
69 info "Using a recent dev install shim at ${FLAGS_dev_install_shim}" 69 info "Using a recent dev install shim at ${FLAGS_dev_install_shim}"
70 fi 70 fi
71 71
72 # Constants 72 # Constants
73 INSTALL_SHIM_DIR="$(dirname "$FLAGS_dev_install_shim")" 73 INSTALL_SHIM_DIR="$(dirname "$FLAGS_dev_install_shim")"
74 DEV_RECOVERY_IMAGE="dev_recovery_image.bin" 74 DEV_RECOVERY_IMAGE="dev_recovery_image.bin"
75 75
76 umount_from_loop_dev() { 76 umount_from_loop_dev() {
77 local mnt_pt=$1 77 local mnt_pt=$1
78 mount | grep -q " on ${mnt_pt}" && sudo umount ${mnt_pt} 78 mount | grep -q " on ${mnt_pt}" && sudo umount ${mnt_pt}
79 rmdir ${mnt_pt}
79 } 80 }
80 81
81 cleanup_loop_dev() { 82 cleanup_loop_dev() {
82 sudo losetup -d ${1} || /bin/true 83 sudo losetup -d ${1} || /bin/true
83 } 84 }
84 85
85 get_loop_dev() { 86 get_loop_dev() {
86 local loop_dev=$(sudo losetup -f) 87 local loop_dev=$(sudo losetup -f)
87 if [ -z "${loop_dev}" ]; then 88 if [ -z "${loop_dev}" ]; then
88 die "No free loop device. Free up a loop device or reboot. Exiting." 89 die "No free loop device. Free up a loop device or reboot. Exiting."
(...skipping 11 matching lines...) Expand all
100 local source_sectors=$(roundup $(numsectors ${source_part})) 101 local source_sectors=$(roundup $(numsectors ${source_part}))
101 info "source partition has ${source_sectors} 512-byte sectors." 102 info "source partition has ${source_sectors} 512-byte sectors."
102 local resized_sectors=$(roundup $(expr $source_sectors + $add_num_sectors)) 103 local resized_sectors=$(roundup $(expr $source_sectors + $add_num_sectors))
103 info "resized partition has ${resized_sectors} 512-byte sectors." 104 info "resized partition has ${resized_sectors} 512-byte sectors."
104 105
105 local loop_dev=$(get_loop_dev) 106 local loop_dev=$(get_loop_dev)
106 trap "cleanup_loop_dev ${loop_dev}" EXIT 107 trap "cleanup_loop_dev ${loop_dev}" EXIT
107 108
108 # Extend the source file size to the new size. 109 # Extend the source file size to the new size.
109 dd if=/dev/zero of="${source_part}" bs=1 count=1 \ 110 dd if=/dev/zero of="${source_part}" bs=1 count=1 \
110 seek=$((512 * ${resized_sectors} - 1)) 111 seek=$((512 * ${resized_sectors} - 1)) &>/dev/null
111 112
112 # Resize the partition. 113 # Resize the partition.
113 sudo losetup "${loop_dev}" "${source_part}" 114 sudo losetup "${loop_dev}" "${source_part}"
114 sudo e2fsck -fp "${loop_dev}" &> /dev/null 115 sudo e2fsck -fp "${loop_dev}" &> /dev/null
115 sudo resize2fs "${loop_dev}" &> /dev/null 116 sudo resize2fs "${loop_dev}" &> /dev/null
116 cleanup_loop_dev "${loop_dev}" 117 # trap handler will clean up the loop device
117
118 echo "${resized_sectors}" 118 echo "${resized_sectors}"
119 } 119 }
120 120
121 # Update partition table with resized stateful partition and create the final 121 # Update partition table with resized stateful partition and create the final
122 # dev recovery image 122 # dev recovery image
123 update_partition_table() { 123 update_partition_table() {
124 local temp_state=$1 # stateful partition image 124 local temp_state=$1 # stateful partition image
125 local resized_sectors=$2 # number of sectors in resized stateful partition 125 local resized_sectors=$2 # number of sectors in resized stateful partition
126 local temp_img=$(mktemp) 126 local temp_img=$(mktemp "/tmp/temp_img.XXXXXX")
127 127
128 local kernel_offset=$(partoffset ${FLAGS_dev_install_shim} 2) 128 local kernel_offset=$(partoffset ${FLAGS_dev_install_shim} 2)
129 local kernel_count=$(partsize ${FLAGS_dev_install_shim} 2) 129 local kernel_count=$(partsize ${FLAGS_dev_install_shim} 2)
130 local rootfs_offset=$(partoffset ${FLAGS_dev_install_shim} 3) 130 local rootfs_offset=$(partoffset ${FLAGS_dev_install_shim} 3)
131 local rootfs_count=$(partsize ${FLAGS_dev_install_shim} 3) 131 local rootfs_count=$(partsize ${FLAGS_dev_install_shim} 3)
132 local oem_offset=$(partoffset ${FLAGS_dev_install_shim} 8) 132 local oem_offset=$(partoffset ${FLAGS_dev_install_shim} 8)
133 local oem_count=$(partsize ${FLAGS_dev_install_shim} 8) 133 local oem_count=$(partsize ${FLAGS_dev_install_shim} 8)
134 local esp_offset=$(partoffset ${FLAGS_dev_install_shim} 12) 134 local esp_offset=$(partoffset ${FLAGS_dev_install_shim} 12)
135 local esp_count=$(partsize ${FLAGS_dev_install_shim} 12) 135 local esp_count=$(partsize ${FLAGS_dev_install_shim} 12)
136 136
137 local temp_pmbr=$(mktemp) 137 local temp_pmbr=$(mktemp "/tmp/pmbr.XXXXXX")
138 dd if="${FLAGS_dev_install_shim}" of="${temp_pmbr}" bs=512 count=1 138 dd if="${FLAGS_dev_install_shim}" of="${temp_pmbr}" bs=512 count=1 &>/dev/null
139 139
140 # Set up a new partition table 140 # Set up a new partition table
141 install_gpt "${temp_img}" "${rootfs_count}" "${resized_sectors}" \ 141 install_gpt "${temp_img}" "${rootfs_count}" "${resized_sectors}" \
142 "${temp_pmbr}" "${esp_count}" false $(roundup ${rootfs_count}) &>/dev/null 142 "${temp_pmbr}" "${esp_count}" false $(roundup ${rootfs_count}) &>/dev/null
143 143
144 rm -rf "${temp_pmbr}" 144 rm -rf "${temp_pmbr}"
145 145
146 # Copy into the partition parts of the file 146 # Copy into the partition parts of the file
147 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \ 147 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \
148 seek="${START_ROOTFS_A}" skip=${rootfs_offset} count=${rootfs_count} 148 seek="${START_ROOTFS_A}" skip=${rootfs_offset} count=${rootfs_count} \
149 &>/dev/null
149 dd if="${temp_state}" of="${temp_img}" conv=notrunc bs=512 \ 150 dd if="${temp_state}" of="${temp_img}" conv=notrunc bs=512 \
150 seek="${START_STATEFUL}" 151 seek="${START_STATEFUL}" &>/dev/null
151 # Copy the full kernel (i.e. with vboot sections) 152 # Copy the full kernel (i.e. with vboot sections)
152 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \ 153 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \
153 seek="${START_KERN_A}" skip=${kernel_offset} count=${kernel_count} 154 seek="${START_KERN_A}" skip=${kernel_offset} count=${kernel_count} \
155 &>/dev/null
154 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \ 156 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \
155 seek="${START_OEM}" skip=${oem_offset} count=${oem_count} 157 seek="${START_OEM}" skip=${oem_offset} count=${oem_count} &>/dev/null
156 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \ 158 dd if="${FLAGS_dev_install_shim}" of="${temp_img}" conv=notrunc bs=512 \
157 seek="${START_ESP}" skip=${esp_offset} count=${esp_count} 159 seek="${START_ESP}" skip=${esp_offset} count=${esp_count} &>/dev/null
158 160
159 echo ${temp_img} 161 echo ${temp_img}
160 } 162 }
161 163
162 # Creates a dev recovery image using an existing dev install shim 164 # Creates a dev recovery image using an existing dev install shim
163 # If successful, content of --payload_dir is copied to a directory named 165 # If successful, content of --payload_dir is copied to a directory named
164 # "dev_payload" under the root of stateful partition. 166 # "dev_payload" under the root of stateful partition.
165 create_dev_recovery_image() { 167 create_dev_recovery_image() {
166 local temp_state=$(mktemp) 168 local temp_state=$(mktemp "/tmp/temp_state.XXXXXX")
167 local stateful_offset=$(partoffset ${FLAGS_dev_install_shim} 1) 169 local stateful_offset=$(partoffset ${FLAGS_dev_install_shim} 1)
168 local stateful_count=$(partsize ${FLAGS_dev_install_shim} 1) 170 local stateful_count=$(partsize ${FLAGS_dev_install_shim} 1)
169 dd if="${FLAGS_dev_install_shim}" of="${temp_state}" conv=notrunc bs=512 \ 171 dd if="${FLAGS_dev_install_shim}" of="${temp_state}" conv=notrunc bs=512 \
170 skip=${stateful_offset} count=${stateful_count} 172 skip=${stateful_offset} count=${stateful_count} &>/dev/null
171 173
172 local resized_sectors=$(resize_partition $temp_state $PAYLOAD_DIR_SIZE) 174 local resized_sectors=$(resize_partition $temp_state $PAYLOAD_DIR_SIZE)
173 175
174 # Mount resized stateful FS and copy payload content to its root directory 176 # Mount resized stateful FS and copy payload content to its root directory
175 local temp_mnt=$(mktemp -d) 177 local temp_mnt=$(mktemp -d "/tmp/temp_mnt.XXXXXX")
176 local loop_dev=$(get_loop_dev) 178 local loop_dev=$(get_loop_dev)
177 trap "umount_from_loop_dev ${temp_mnt} && cleanup_loop_dev ${loop_dev}" EXIT 179 trap "umount_from_loop_dev ${temp_mnt} && cleanup_loop_dev ${loop_dev}" EXIT
178 mkdir -p "${temp_mnt}" 180 mkdir -p "${temp_mnt}"
179 sudo mount -o loop=${loop_dev} "${temp_state}" "${temp_mnt}" 181 sudo mount -o loop=${loop_dev} "${temp_state}" "${temp_mnt}"
180 sudo cp -R "${FLAGS_payload_dir}" "${temp_mnt}" 182 sudo cp -R "${FLAGS_payload_dir}" "${temp_mnt}"
181 sudo mv "${temp_mnt}/$(basename ${FLAGS_payload_dir})" \ 183 sudo mv "${temp_mnt}/$(basename ${FLAGS_payload_dir})" \
182 "${temp_mnt}/dev_payload" 184 "${temp_mnt}/dev_payload"
183 # Mark image as dev recovery 185 # Mark image as dev recovery
184 sudo touch "${temp_mnt}/.recovery" 186 sudo touch "${temp_mnt}/.recovery"
185 sudo touch "${temp_mnt}/.dev_recovery" 187 sudo touch "${temp_mnt}/.dev_recovery"
186 188
187 # TODO(tgao): handle install script (for default and custom cases) 189 # TODO(tgao): handle install script (for default and custom cases)
188 local temp_img=$(update_partition_table $temp_state $resized_sectors) 190 local temp_img=$(update_partition_table $temp_state $resized_sectors)
189 191
190 umount_from_loop_dev "${temp_mnt}"
191 cleanup_loop_dev ${loop_dev}
192 rm -f "${temp_state}" 192 rm -f "${temp_state}"
193 # trap handler will clean up loop device and temp mount point
193 echo ${temp_img} 194 echo ${temp_img}
194 } 195 }
195 196
196 # Main 197 # Main
197 DST_PATH="${INSTALL_SHIM_DIR}/${DEV_RECOVERY_IMAGE}" 198 DST_PATH="${INSTALL_SHIM_DIR}/${DEV_RECOVERY_IMAGE}"
198 info "Attempting to create dev recovery image using dev install shim \ 199 info "Attempting to create dev recovery image using dev install shim \
199 ${FLAGS_dev_install_shim}" 200 ${FLAGS_dev_install_shim}"
200 TEMP_IMG=$(create_dev_recovery_image) 201 TEMP_IMG=$(create_dev_recovery_image)
201 202
202 mv -f $TEMP_IMG $DST_PATH 203 mv -f $TEMP_IMG $DST_PATH
203 info "Dev recovery image created at ${DST_PATH}" 204 info "Dev recovery image created at ${DST_PATH}"
OLDNEW
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698