OLD | NEW |
| (Empty) |
1 #!/bin/bash | |
2 | |
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 | |
5 # found in the LICENSE file. | |
6 | |
7 # Script to convert the output of build_image.sh to a VMware image and write a | |
8 # corresponding VMware config file. | |
9 | |
10 # Load common constants. This should be the first executable line. | |
11 # The path to common.sh should be relative to your script's location. | |
12 . "/usr/lib/crosutils/common.sh" | |
13 . "/usr/lib/installer/chromeos-common.sh" | |
14 | |
15 get_default_board | |
16 | |
17 DEFAULT_VMDK="ide.vmdk" | |
18 DEFAULT_VMX="chromiumos.vmx" | |
19 DEFAULT_VBOX_DISK="os.vdi" | |
20 DEFAULT_QEMU_IMAGE="chromiumos_qemu_image.bin" | |
21 | |
22 MOD_SCRIPTS_ROOT="/usr/share/chromeos-installer/mod_for_test_scripts" | |
23 | |
24 # Flags | |
25 DEFINE_string board "${DEFAULT_BOARD}" \ | |
26 "Board for which the image was built" | |
27 DEFINE_boolean factory $FLAGS_FALSE \ | |
28 "Modify the image for manufacturing testing" | |
29 DEFINE_boolean factory_install $FLAGS_FALSE \ | |
30 "Modify the image for factory install shim" | |
31 DEFINE_boolean force_copy ${FLAGS_FALSE} "Always rebuild test image" | |
32 DEFINE_string format "qemu" \ | |
33 "Output format, either qemu, vmware or virtualbox" | |
34 DEFINE_string from "" \ | |
35 "Directory containing rootfs.image and mbr.image" | |
36 DEFINE_boolean make_vmx ${FLAGS_TRUE} \ | |
37 "Create a vmx file for use with vmplayer (vmware only)." | |
38 DEFINE_integer mem "${DEFAULT_MEM}" \ | |
39 "Memory size for the vm config in MBs (vmware only)." | |
40 DEFINE_integer rootfs_partition_size 1024 \ | |
41 "rootfs parition size in MBs." | |
42 DEFINE_string state_image "" \ | |
43 "Stateful partition image (defaults to creating new statful partition)" | |
44 DEFINE_integer statefulfs_size -1 \ | |
45 "Stateful partition size in MBs." | |
46 DEFINE_boolean test_image "${FLAGS_FALSE}" \ | |
47 "Copies normal image to chromiumos_test_image.bin, modifies it for test." | |
48 DEFINE_string to "" \ | |
49 "Destination folder for VM output file(s)" | |
50 DEFINE_string vbox_disk "${DEFAULT_VBOX_DISK}" \ | |
51 "Filename for the output disk (virtualbox only)." | |
52 DEFINE_integer vdisk_size 3072 \ | |
53 "virtual disk size in MBs." | |
54 DEFINE_string vmdk "${DEFAULT_VMDK}" \ | |
55 "Filename for the vmware disk image (vmware only)." | |
56 DEFINE_string vmx "${DEFAULT_VMX}" \ | |
57 "Filename for the vmware config (vmware only)." | |
58 | |
59 # Parse command line | |
60 FLAGS "$@" || exit 1 | |
61 eval set -- "${FLAGS_ARGV}" | |
62 | |
63 # Die on any errors. | |
64 set -e | |
65 | |
66 if [ -z "${FLAGS_board}" ] ; then | |
67 die "--board is required." | |
68 fi | |
69 | |
70 IMAGES_DIR="${DEFAULT_BUILD_ROOT}/images/${FLAGS_board}" | |
71 # Default to the most recent image | |
72 if [ -z "${FLAGS_from}" ] ; then | |
73 FLAGS_from="${IMAGES_DIR}/$(ls -t $IMAGES_DIR | head -1)" | |
74 else | |
75 pushd "${FLAGS_from}" && FLAGS_from=`pwd` && popd | |
76 fi | |
77 if [ -z "${FLAGS_to}" ] ; then | |
78 FLAGS_to="${FLAGS_from}" | |
79 fi | |
80 | |
81 # Use this image as the source image to copy | |
82 SRC_IMAGE="${FLAGS_from}/chromiumos_image.bin" | |
83 | |
84 # If we're asked to modify the image for test, then let's make a copy and | |
85 # modify that instead. | |
86 if [ ${FLAGS_test_image} -eq ${FLAGS_TRUE} ] ; then | |
87 if [ ! -f "${FLAGS_from}/chromiumos_test_image.bin" ] || \ | |
88 [ ${FLAGS_force_copy} -eq ${FLAGS_TRUE} ] ; then | |
89 # Copy it. | |
90 echo "Creating test image from original..." | |
91 cp -f "${SRC_IMAGE}" "${FLAGS_from}/chromiumos_test_image.bin" | |
92 | |
93 # Check for manufacturing image. | |
94 if [ ${FLAGS_factory} -eq ${FLAGS_TRUE} ] ; then | |
95 EXTRA_ARGS="--factory" | |
96 fi | |
97 | |
98 # Check for install shim. | |
99 if [ ${FLAGS_factory_install} -eq ${FLAGS_TRUE} ] ; then | |
100 EXTRA_ARGS="--factory_install" | |
101 fi | |
102 | |
103 # Modify it. Pass --yes so that cros_mod_image_for_test.sh won't ask us if | |
104 # we really want to modify the image; the user gave their assent already | |
105 # with --test-image and the original image is going to be preserved. | |
106 "/usr/lib/installer/bin/cros_mod_image_for_test.sh" --image \ | |
107 "${FLAGS_from}/chromiumos_test_image.bin" ${EXTRA_ARGS} --yes | |
108 echo "Done with cros_mod_image_for_test." | |
109 else | |
110 echo "Using cached test image." | |
111 fi | |
112 SRC_IMAGE="${FLAGS_from}/chromiumos_test_image.bin" | |
113 echo "Source test image is: ${SRC_IMAGE}" | |
114 fi | |
115 | |
116 # Memory units are in MBs | |
117 DEFAULT_MEM="1024" | |
118 TEMP_IMG="$(dirname ${SRC_IMAGE})/vm_temp_image.bin" | |
119 | |
120 # If we're not building for VMWare, don't build the vmx | |
121 if [ "${FLAGS_format}" != "vmware" ]; then | |
122 FLAGS_make_vmx="${FLAGS_FALSE}" | |
123 fi | |
124 | |
125 # Convert args to paths. Need eval to un-quote the string so that shell | |
126 # chars like ~ are processed; just doing FOO=`readlink -f $FOO` won't work. | |
127 FLAGS_from=`eval readlink -f $FLAGS_from` | |
128 FLAGS_to=`eval readlink -f $FLAGS_to` | |
129 | |
130 # Split apart the partitions and make some new ones | |
131 TEMP_DIR=$(mktemp -d) | |
132 (cd "${TEMP_DIR}" && | |
133 "${FLAGS_from}/unpack_partitions.sh" "${SRC_IMAGE}") | |
134 | |
135 # Fix the kernel command line | |
136 TEMP_ESP="${TEMP_DIR}"/part_12 | |
137 TEMP_ROOTFS="${TEMP_DIR}"/part_3 | |
138 TEMP_STATE="${TEMP_DIR}"/part_1 | |
139 TEMP_KERN="${TEMP_DIR}"/part_2 | |
140 if [ -n "${FLAGS_state_image}" ]; then | |
141 TEMP_STATE="${FLAGS_state_image}" | |
142 else | |
143 # If we have a stateful fs size specified create a new state partition | |
144 # of the specified size. | |
145 if [ "${FLAGS_statefulfs_size}" -ne -1 ]; then | |
146 STATEFUL_SIZE_BYTES=$((1024 * 1024 * ${FLAGS_statefulfs_size})) | |
147 original_image_size=$(stat -c%s "${TEMP_STATE}") | |
148 if [ "${original_image_size}" -gt "${STATEFUL_SIZE_BYTES}" ]; then | |
149 die "Cannot resize stateful image to smaller than original. Exiting." | |
150 fi | |
151 | |
152 echo "Resizing stateful partition to ${FLAGS_statefulfs_size}MB" | |
153 STATEFUL_LOOP_DEV=$(sudo losetup -f) | |
154 if [ -z "${STATEFUL_LOOP_DEV}" ]; then | |
155 die "No free loop device. Free up a loop device or reboot. Exiting." | |
156 fi | |
157 | |
158 # Extend the original file size to the new size. | |
159 dd if=/dev/zero of="${TEMP_STATE}" bs=1 count=1 \ | |
160 seek=$((STATEFUL_SIZE_BYTES - 1)) | |
161 # Resize the partition. | |
162 sudo losetup "${STATEFUL_LOOP_DEV}" "${TEMP_STATE}" | |
163 sudo e2fsck -f "${STATEFUL_LOOP_DEV}" | |
164 sudo resize2fs "${STATEFUL_LOOP_DEV}" | |
165 sudo losetup -d "${STATEFUL_LOOP_DEV}" | |
166 fi | |
167 fi | |
168 TEMP_PMBR="${TEMP_DIR}"/pmbr | |
169 dd if="${SRC_IMAGE}" of="${TEMP_PMBR}" bs=512 count=1 | |
170 | |
171 TEMP_MNT=$(mktemp -d) | |
172 TEMP_ESP_MNT=$(mktemp -d) | |
173 cleanup() { | |
174 sudo umount -d "${TEMP_MNT}" | |
175 sudo umount -d "${TEMP_ESP_MNT}" | |
176 rmdir "${TEMP_MNT}" "${TEMP_ESP_MNT}" | |
177 } | |
178 trap cleanup INT TERM EXIT | |
179 mkdir -p "${TEMP_MNT}" | |
180 sudo mount -o loop "${TEMP_ROOTFS}" "${TEMP_MNT}" | |
181 mkdir -p "${TEMP_ESP_MNT}" | |
182 sudo mount -o loop "${TEMP_ESP}" "${TEMP_ESP_MNT}" | |
183 | |
184 if [ "${FLAGS_format}" = "qemu" ]; then | |
185 sudo python ./fixup_image_for_qemu.py --mounted_dir="${TEMP_MNT}" \ | |
186 --enable_tablet=true | |
187 else | |
188 sudo python ./fixup_image_for_qemu.py --mounted_dir="${TEMP_MNT}" \ | |
189 --enable_tablet=false | |
190 fi | |
191 | |
192 # Modify the unverified usb template which uses a default usb_disk of sdb3 | |
193 sudo sed -i -e 's/sdb3/sda3/g' "${TEMP_MNT}/boot/syslinux/usb.A.cfg" | |
194 | |
195 # Unmount everything prior to building a final image | |
196 sync | |
197 trap - INT TERM EXIT | |
198 cleanup | |
199 | |
200 # Make 3 GiB output image | |
201 # TOOD(adlr): pick a size that will for sure accomodate the partitions | |
202 dd if=/dev/zero of="${TEMP_IMG}" bs=1 count=1 \ | |
203 seek=$((${FLAGS_vdisk_size} * 1024 * 1024 - 1)) | |
204 | |
205 # Set up the partition table | |
206 install_gpt "${TEMP_IMG}" "$(numsectors $TEMP_ROOTFS)" \ | |
207 "$(numsectors $TEMP_STATE)" "${TEMP_PMBR}" "$(numsectors $TEMP_ESP)" \ | |
208 false ${FLAGS_rootfs_partition_size} | |
209 # Copy into the partition parts of the file | |
210 dd if="${TEMP_ROOTFS}" of="${TEMP_IMG}" conv=notrunc bs=512 \ | |
211 seek="${START_ROOTFS_A}" | |
212 dd if="${TEMP_STATE}" of="${TEMP_IMG}" conv=notrunc bs=512 \ | |
213 seek="${START_STATEFUL}" | |
214 dd if="${TEMP_KERN}" of="${TEMP_IMG}" conv=notrunc bs=512 \ | |
215 seek="${START_KERN_A}" | |
216 dd if="${TEMP_ESP}" of="${TEMP_IMG}" conv=notrunc bs=512 \ | |
217 seek="${START_ESP}" | |
218 | |
219 # Make the built-image bootable and ensure that the legacy default usb boot | |
220 # uses /dev/sda instead of /dev/sdb3. | |
221 # NOTE: The TEMP_IMG must live in the same image dir as the original image | |
222 # to operate automatically below. | |
223 /usr/bin/cros_make_image_bootable $(dirname "${TEMP_IMG}") \ | |
224 $(basename "${TEMP_IMG}") \ | |
225 --usb_disk /dev/sda3 | |
226 | |
227 echo Creating final image | |
228 # Convert image to output format | |
229 if [ "${FLAGS_format}" = "virtualbox" -o "${FLAGS_format}" = "qemu" ]; then | |
230 if [ "${FLAGS_format}" = "virtualbox" ]; then | |
231 VBoxManage convertdd "${TEMP_IMG}" "${FLAGS_to}/${FLAGS_vbox_disk}" | |
232 else | |
233 mv ${TEMP_IMG} ${FLAGS_to}/${DEFAULT_QEMU_IMAGE} | |
234 fi | |
235 elif [ "${FLAGS_format}" = "vmware" ]; then | |
236 qemu-img convert -f raw "${TEMP_IMG}" \ | |
237 -O vmdk "${FLAGS_to}/${FLAGS_vmdk}" | |
238 else | |
239 die "Invalid format: ${FLAGS_format}" | |
240 fi | |
241 | |
242 rm -rf "${TEMP_DIR}" "${TEMP_IMG}" | |
243 if [ -z "${FLAGS_state_image}" ]; then | |
244 rm -f "${STATE_IMAGE}" | |
245 fi | |
246 | |
247 echo "Created image at ${FLAGS_to}" | |
248 | |
249 # Generate the vmware config file | |
250 # A good reference doc: http://www.sanbarrow.com/vmx.html | |
251 VMX_CONFIG="#!/usr/bin/vmware | |
252 .encoding = \"UTF-8\" | |
253 config.version = \"8\" | |
254 virtualHW.version = \"4\" | |
255 memsize = \"${FLAGS_mem}\" | |
256 ide0:0.present = \"TRUE\" | |
257 ide0:0.fileName = \"${FLAGS_vmdk}\" | |
258 ethernet0.present = \"TRUE\" | |
259 usb.present = \"TRUE\" | |
260 sound.present = \"TRUE\" | |
261 sound.virtualDev = \"es1371\" | |
262 displayName = \"Chromium OS\" | |
263 guestOS = \"otherlinux\" | |
264 ethernet0.addressType = \"generated\" | |
265 floppy0.present = \"FALSE\"" | |
266 | |
267 if [[ "${FLAGS_make_vmx}" = "${FLAGS_TRUE}" ]]; then | |
268 echo "${VMX_CONFIG}" > "${FLAGS_to}/${FLAGS_vmx}" | |
269 echo "Wrote the following config to: ${FLAGS_to}/${FLAGS_vmx}" | |
270 echo "${VMX_CONFIG}" | |
271 fi | |
272 | |
OLD | NEW |