image_to_usb: Add an option to install to the USB stick.

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 18 matching lines @@
 DEFINE_boolean factory ${FLAGS_FALSE} \
   "Whether to generate a factory runin image. Implies aututest and test"
 DEFINE_boolean copy_kernel ${FLAGS_FALSE} \
   "Copy the kernel to the fourth partition."
 DEFINE_boolean test_image "${FLAGS_FALSE}" \
   "Copies normal image to chromiumos_test_image.bin, modifies it for test."
 DEFINE_string image_name "chromiumos_image.bin" \
   "Base name of the image" i
 DEFINE_string build_root "/build" \
   "The root location for board sysroots."
+DEFINE_boolean install ${FLAGS_FALSE} "Install to the usb device."
+DEFINE_string arch "" "Architecture of the image."
 
 # Parse command line
 FLAGS "$@" || exit 1
 eval set -- "${FLAGS_ARGV}"
 
 if [ ${FLAGS_factory} -eq ${FLAGS_TRUE} ] ; then
   if [ ${FLAGS_factory_install} -eq ${FLAGS_TRUE} ] ; then
     echo "Factory test image is incompatible with factory install shim."
     exit 1
   fi
@@ skipping 14 matching lines @@
 
 # Die on any errors.
 set -e
 
 # No board, no default and no image set then we can't find the image
 if [ -z ${FLAGS_from} ] && [ -z ${FLAGS_board} ] ; then
   setup_board_warning
   exit 1
 fi
 
+# No board set during install
+if [ -z "${FLAGS_board}" ] && [ ${FLAGS_install} -eq ${FLAGS_TRUE} ]; then
+  setup_board_warning
+  exit 1
+fi
+
 # We have a board name but no image set.  Use image at default location
 if [ -z "${FLAGS_from}" ]; then
   IMAGES_DIR="${DEFAULT_BUILD_ROOT}/images/${FLAGS_board}"
   FLAGS_from="${IMAGES_DIR}/$(ls -t ${IMAGES_DIR} 2>&-| head -1)"
 fi
 
 if [ ! -d "${FLAGS_from}" ] ; then
   echo "Cannot find image directory ${FLAGS_from}"
   exit 1
 fi
 
 if [ "${FLAGS_to}" == "/dev/sdX" ]; then
   echo "You must specify a file or device to write to using --to."
   disks=$(list_usb_disks)
   if [ -n "$disks" ]; then
     echo "Available USB disks:"
     for disk in $disks; do
       echo "  /dev/$disk:"
       echo "    Manufacturer: $(get_disk_info $disk manufacturer)"
       echo "         Product: $(get_disk_info $disk product)"
       echo "            Size: $[$(cat /sys/block/$disk/size) * 512] bytes"
     done
   fi
   exit 1
 fi
 
+# Guess ARCH if it's unset
+if [ "${FLAGS_arch}" = "" ]; then
+  if echo "${FLAGS_board}" | grep -qs "x86"; then
+    FLAGS_arch=INTEL
+  else
+    FLAGS_arch=ARM
+  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}`
 
 # One last check to make sure user is not shooting themselves in the foot
 if [ -b "${FLAGS_to}" ]; then
   if list_usb_disks | grep -q '^'${FLAGS_to##*/}'$'; then
     disk_manufacturer=$(get_disk_info ${FLAGS_to##*/} manufacturer)
     disk_product=$(get_disk_info ${FLAGS_to##*/} product)
@@ skipping 53 matching lines @@
   fi
   SRC_IMAGE="${FLAGS_from}/chromiumos_test_image.bin"
   echo "Source test image is: ${SRC_IMAGE}"
 fi
 
 
 # Let's do it.
 if [ -b "${FLAGS_to}" ]
 then
   # Output to a block device (i.e., a real USB key), so need sudo dd
-  echo "Copying USB image ${SRC_IMAGE} to device ${FLAGS_to}..."
+  if [ ${FLAGS_install} -ne ${FLAGS_TRUE} ]; then
+    echo "Copying USB image ${SRC_IMAGE} to device ${FLAGS_to}..."
+  else
+    echo "Installing USB image ${SRC_IMAGE} to device ${FLAGS_to}..."
+  fi
 
   # Warn if it looks like they supplied a partition as the destination.
   if echo "${FLAGS_to}" | grep -q '[0-9]$'; then
     drive=$(echo "${FLAGS_to}" | sed -re 's/[0-9]+$//')
     if [ -b "${drive}" ]; then
       echo
       echo "NOTE: It looks like you may have supplied a partition as the "
       echo "destination.  This script needs to write to the drive's device "
       echo "node instead (i.e. ${drive} rather than ${FLAGS_to})."
       echo
@@ skipping 23 matching lines @@
       echo
       echo "The device you have specified is already mounted at some point "
       echo "that is not visible from inside the chroot.  Please unmount the "
       echo "device manually from outside the chroot and try again."
       echo
       exit 1
     fi
   done
   sleep 3
 
-  echo "Copying ${SRC_IMAGE} to ${FLAGS_to}..."
-  sudo dd if="${SRC_IMAGE}" of="${FLAGS_to}" bs=4M
-  sync
+  if [ ${FLAGS_install} -ne ${FLAGS_TRUE} ]; then
+    echo "Copying ${SRC_IMAGE} to ${FLAGS_to}..."
+    sudo dd if="${SRC_IMAGE}" of="${FLAGS_to}" bs=4M
+    sync
+  else
+    if [ ${INSIDE_CHROOT} -ne 1 ]; then
+      echo "Installation must be done from inside the chroot."
+      exit 1
+    fi
+    #TODO(kwaters): fix when verified root works on ARM
+    [ "${FLAGS_arch}" = "ARM" ] && SKIP_VBLOCK="--skip_vblock"
 
+    echo "Installing ${SRC_IMAGE} to ${FLAGS_to}..."
+    "${FLAGS_build_root}/${FLAGS_board}/usr/sbin/chromeos-install" \
+      --yes \
+      --skip_src_removable \
+      --skip_dst_removable \
+      ${SKIP_VBLOCK:-} \
+      --arch="${FLAGS_arch}" \
+      --payload_image="${SRC_IMAGE}" \
+      --dst="${FLAGS_to}"
+  fi
   echo "Done."
 else
   # Output to a file, so just make a copy.
   echo "Copying ${SRC_IMAGE} to ${FLAGS_to}..."
   cp -f "${SRC_IMAGE}" "${FLAGS_to}"
 
   echo "Done.  To copy to a USB drive, outside the chroot, do something like:"
   echo "   sudo dd if=${FLAGS_to} of=/dev/sdX bs=4M"
   echo "where /dev/sdX is the entire drive."
   if [ ${INSIDE_CHROOT} -eq 1 ]
   then
     example=$(basename "${FLAGS_to}")
     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/$example)."
   fi
 fi