| 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 |