Resize stateful partition flag

image_to_vm.sh

 #!/bin/bash
 
 # Copyright (c) 2010 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 the output of build_image.sh to a VMware image and write a
 # corresponding VMware config file.
 
 # Load common constants.  This should be the first executable line.
@@ skipping 23 matching lines @@
 DEFINE_string from "" \
   "Directory containing rootfs.image and mbr.image"
 DEFINE_boolean make_vmx ${FLAGS_TRUE} \
   "Create a vmx file for use with vmplayer (vmware only)."
 DEFINE_integer mem "${DEFAULT_MEM}" \
   "Memory size for the vm config in MBs (vmware only)."
 DEFINE_integer rootfs_partition_size 1024 \
   "rootfs parition size in MBs."
 DEFINE_string state_image "" \
   "Stateful partition image (defaults to creating new statful partition)"
+DEFINE_integer statefulfs_size -1 \
+  "Stateful partition size in MBs."
 DEFINE_boolean test_image "${FLAGS_FALSE}" \
   "Copies normal image to chromiumos_test_image.bin, modifies it for test."
 DEFINE_string to "" \
   "Destination folder for VM output file(s)"
 DEFINE_string vbox_disk "${DEFAULT_VBOX_DISK}" \
   "Filename for the output disk (virtualbox only)."
 DEFINE_integer vdisk_size 3072 \
   "virtual disk size in MBs."
 DEFINE_string vmdk "${DEFAULT_VMDK}" \
   "Filename for the vmware disk image (vmware only)."
@@ skipping 76 matching lines @@
 TEMP_DIR=$(mktemp -d)
 (cd "${TEMP_DIR}" &&
   "${FLAGS_from}/unpack_partitions.sh" "${SRC_IMAGE}")
 
 # Fix the kernel command line
 TEMP_ESP="${TEMP_DIR}"/part_12
 TEMP_ROOTFS="${TEMP_DIR}"/part_3
 TEMP_STATE="${TEMP_DIR}"/part_1
 if [ -n "${FLAGS_state_image}" ]; then
   TEMP_STATE="${FLAGS_state_image}"
+else
+  # If we have a stateful fs size specified create a new state partition
+  # of the specified size.
+  if [ "${FLAGS_statefulfs_size}" -ne -1 ]; then
+    STATEFUL_SIZE_BYTES=$((1024 * 1024 * ${FLAGS_statefulfs_size}))
+    original_image_size=$(stat -c%s "${TEMP_STATE}")
+    if [ "${original_image_size}" -gt "${STATEFUL_SIZE_BYTES}" ]; then
+      die "Cannot resize stateful image to smaller than original. Exiting."
+    fi
+
+    echo "Resizing stateful partition to ${FLAGS_statefulfs_size}MB"
+    STATEFUL_LOOP_DEV=$(sudo losetup -f)
+    if [ -z "${STATEFUL_LOOP_DEV}" ]; then
+      die "No free loop device. Free up a loop device or reboot. Exiting."
+    fi
+
+    # Extend the original file size to the new size.
+    dd if=/dev/zero of="${TEMP_STATE}" bs=1 count=1 \
+        seek=$((STATEFUL_SIZE_BYTES - 1))
+    # Resize the partition.
+    sudo losetup "${STATEFUL_LOOP_DEV}" "${TEMP_STATE}"
+    sudo e2fsck -f "${STATEFUL_LOOP_DEV}"
+    sudo resize2fs "${STATEFUL_LOOP_DEV}"
+    sudo losetup -d "${STATEFUL_LOOP_DEV}"
+  fi
 fi
 TEMP_KERN="${TEMP_DIR}"/part_2
 TEMP_PMBR="${TEMP_DIR}"/pmbr
 dd if="${SRC_IMAGE}" of="${TEMP_PMBR}" bs=512 count=1
 
 TEMP_MNT=$(mktemp -d)
 cleanup() {
   sudo umount -d "${TEMP_MNT}"
   rmdir "${TEMP_MNT}"
 }
@@ skipping 72 matching lines @@
 guestOS = \"otherlinux\"
 ethernet0.addressType = \"generated\"
 floppy0.present = \"FALSE\""
 
 if [[ "${FLAGS_make_vmx}" = "${FLAGS_TRUE}" ]]; then
   echo "${VMX_CONFIG}" > "${FLAGS_to}/${FLAGS_vmx}"
   echo "Wrote the following config to: ${FLAGS_to}/${FLAGS_vmx}"
   echo "${VMX_CONFIG}"
 fi