Changes to enable dev in stateful partition for usb and startup

src/scripts/image_to_usb.sh

 #!/bin/bash
 
 # Copyright (c) 2009 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 usb image.
 
 # Load common constants.  This should be the first executable line.
 # The path to common.sh should be relative to your script's location.
@@ skipping 61 matching lines @@
   fi
 fi
 
 # Convert args to paths.  Need eval to un-quote the string so that shell
 # chars like ~ are processed; just doing FOO=`readlink -f ${FOO}` won't work.
 FLAGS_from=`eval readlink -f ${FLAGS_from}`
 FLAGS_to=`eval readlink -f ${FLAGS_to}`
 
 # Uses this rootfs image as the source image to copy
 ROOTFS_IMAGE="${FLAGS_from}/rootfs.image"
+PART_SIZE=$(stat -c%s "${ROOTFS_IMAGE}")  # Bytes
+
+# Setup stateful partition variables
+STATEFUL_IMG="${FLAGS_from}/stateful_partition.image"
+STATEFUL_DIR="${FLAGS_from}/stateful_partition"
+
+# TODO(sosa@chromium.org) - Remove legacy support.
+if [ ! -f "${STATEFUL_IMG}" ] ; then
+  echo "WARNING!  Stateful partition not found.  Creating clean stateful"
+  STATEFUL_LOOP_DEV=$(sudo losetup -f)
+  if [ -z "${STATEFUL_LOOP_DEV}" ] ; then
+    echo "No free loop device.  Free up a loop device or reboot.  exiting. "
+    exit 1
+  fi
+  set -x
+  dd if=/dev/zero of="${STATEFUL_IMG}" bs=1 count=1 \
+      seek=$(( (${PART_SIZE} - 1) ))
+  set +x
+  sudo losetup "$STATEFUL_LOOP_DEV" "$STATEFUL_IMG"
+  sudo mkfs.ext3 "$STATEFUL_LOOP_DEV"
+  sudo tune2fs -L "C-STATE" -c 0 -i 0 "$STATEFUL_LOOP_DEV"
+  sudo losetup -d "${STATEFUL_LOOP_DEV}"
+fi
 
 # Modifies image for test if requested
 if [ ${FLAGS_test_image} -eq ${FLAGS_TRUE} ] ; then
   if [ ! -f "${FLAGS_from}/rootfs_test.image" ] ; then
     echo "Test image not found, creating test image from original ... "
     cp "${FLAGS_from}/rootfs.image" "${FLAGS_from}/rootfs_test.image"
     "${SCRIPTS_DIR}/mod_image_for_test.sh" \
       --image "${FLAGS_from}/rootfs_test.image"
   fi
   # Use the test image instead
   ROOTFS_IMAGE="${FLAGS_from}/rootfs_test.image"
 fi
 
 function do_cleanup {
   sudo losetup -d "${LOOP_DEV}"
 }
 
-STATEFUL_DIR=${FLAGS_from}/stateful_partition
-mkdir -p "${STATEFUL_DIR}"
-
 function install_autotest {
   if [ -d ${AUTOTEST_SRC} ]
   then
+    local stateful_loop_dev=$(sudo losetup -f)
+    if [ -z "${stateful_loop_dev}" ]
+    then
+      echo "No free loop device. Free up a loop device or reboot. exiting."
+      exit 1
+    fi
+    
+    sudo mount "${stateful_loop_dev}" "${STATEFUL_DIR}"
+
     echo -ne "Install autotest into stateful partition..."
           local autotest_client="/home/autotest-client"
     sudo mkdir -p "${STATEFUL_DIR}${autotest_client}"
     sudo cp -fpru ${AUTOTEST_SRC}/client/* \
             "${STATEFUL_DIR}${autotest_client}"
     sudo chmod 755 "${STATEFUL_DIR}${autotest_client}"
     sudo chown -R 1000:1000 "${STATEFUL_DIR}${autotest_client}"
-    echo "Done."
-    sudo umount "${STATEFUL_DIR}"
+    
+    sudo umount ${STATEFUL_DIR}
+    sudo losetup -d "${stateful_loop_dev}"    
   else
     echo "/usr/local/autotest under ${DEFAULT_CHROOT_DIR} is not installed."
     echo "Please call make_autotest.sh inside chroot first."
     sudo umount "${STATEFUL_DIR}"
     exit -1
   fi
 }
 
 # Copy MBR and rootfs to output image
 if [ -b "${FLAGS_to}" ]
@@ skipping 26 matching lines @@
       exit 1
     fi
   fi
 
   echo "attempting to unmount any mounts on the USB device"
   for i in "${FLAGS_to}"*
   do
     ! sudo umount "$i"
   done
   sleep 3
-
-  PART_SIZE=$(stat -c%s "${ROOTFS_IMAGE}")  # Bytes
-
-  echo "Copying root fs..."
-  sudo "${SCRIPTS_DIR}"/file_copy.py \
-    if="${ROOTFS_IMAGE}" \
-    of="${FLAGS_to}" bs=4M \
-    seek_bytes=$(( (${PART_SIZE} * 2) + 512 ))
-
-  # Set up loop device
-  LOOP_DEV=$(sudo losetup -f)
-  if [ -z "${LOOP_DEV}" ]
-  then
-    echo "No free loop device. Free up a loop device or reboot. exiting."
-    exit 1
-  fi
-
-  trap do_cleanup EXIT
-
-  echo "Creating stateful partition..."
-  sudo losetup -o 512 "${LOOP_DEV}" "${FLAGS_to}"
-  sudo mkfs.ext3 -F -b 4096 -L C-STATE "${LOOP_DEV}" $(( ${PART_SIZE} / 4096 ))
-  if [ ${FLAGS_install_autotest} -eq ${FLAGS_TRUE} ]
-  then
-    sudo mount "${LOOP_DEV}" "${STATEFUL_DIR}"
+ 
+  if [ ${FLAGS_install_autotest} -eq ${FLAGS_TRUE} ] ; then
     install_autotest
   fi
-  sync
-  sudo losetup -d "${LOOP_DEV}"
-  sync
+ 
+  # Write stateful partition to first partition. 
+  echo "Copying stateful partition ..."
+  sudo "${SCRIPTS_DIR}"/file_copy.py \
+      if="${STATEFUL_IMG}" of="${FLAGS_to}" bs=4M \
+      seek_bytes=512
 
+  # Write root fs to third partition.
+  echo "Copying root fs partition ..."
+  sudo "${SCRIPTS_DIR}"/file_copy.py \
+      if="${ROOTFS_IMAGE}" of="${FLAGS_to}" bs=4M \
+      seek_bytes=$(( (${PART_SIZE} * 2) + 512 ))
+  
   trap - EXIT
 
   if [ ${FLAGS_copy_kernel} -eq ${FLAGS_TRUE} ]
   then
     echo "Copying Kernel..."
     "${SCRIPTS_DIR}"/kernel_fetcher.sh \
       --from "${FLAGS_from}" \
       --to "${FLAGS_to}" \
       --offset "$(( (${PART_SIZE} * 3) + 512 ))"
   fi
 
   echo "Copying MBR..."
   sudo "${SCRIPTS_DIR}"/file_copy.py \
     if="${FLAGS_from}/mbr.image" of="${FLAGS_to}"
   sync
   echo "Done."
 else
   # Output to a file, so just cat the source images together
 
   PART_SIZE=$(stat -c%s "${ROOTFS_IMAGE}")
 
-  echo "Creating empty stateful partition"
-  dd if=/dev/zero of="${FLAGS_from}/stateful_partition.image" bs=1 count=1 \
-      seek=$((${PART_SIZE} - 1))
-  mkfs.ext3 -F -L C-STATE "${FLAGS_from}/stateful_partition.image"
-
-  if [ ${FLAGS_install_autotest} -eq ${FLAGS_TRUE} ]
-  then
-    sudo mount -o loop "${FLAGS_from}/stateful_partition.image" \
-      "${STATEFUL_DIR}"
+  if [ ${FLAGS_install_autotest} -eq ${FLAGS_TRUE} ] ; then
     install_autotest
   fi
 
   # Create a sparse output file
   dd if=/dev/zero of="${FLAGS_to}" bs=1 count=1 \
       seek=$(( (${PART_SIZE} * 2) + 512 - 1))
 
   echo "Copying USB image to file ${FLAGS_to}..."
 
   dd if="${FLAGS_from}/mbr.image" of="${FLAGS_to}" conv=notrunc
   dd if="${FLAGS_from}/stateful_partition.image" of="${FLAGS_to}" seek=1 bs=512 \
       conv=notrunc
   cat "${ROOTFS_IMAGE}" >> "${FLAGS_to}"
 
   echo "Done.  To copy to USB keyfob, outside the chroot, do something like:"
   echo "   sudo dd if=${FLAGS_to} of=/dev/sdb bs=4M"
   echo "where /dev/sdb is the entire keyfob."
   if [ ${INSIDE_CHROOT} -eq 1 ]
   then
     echo "NOTE: Since you are currently inside the chroot, and you'll need to"
     echo "run dd outside the chroot, the path to the USB image will be"
     echo "different (ex: ~/chromeos/trunk/src/build/images/SOME_DIR/usb.img)."
   fi
 fi