Repatch

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.
 . "$(dirname "$0")/common.sh"
 
-IMAGES_DIR="${DEFAULT_BUILD_ROOT}/images"
-# Default to the most recent image
-DEFAULT_FROM="${IMAGES_DIR}/$(ls -t $IMAGES_DIR 2>&-| head -1)"
-
-# Script can be run either inside or outside the chroot.
-if [ $INSIDE_CHROOT -eq 1 ]
-then
-  # Inside the chroot, so output to usb.img in the same dir as the other
-  # images.
-  DEFAULT_TO="${DEFAULT_FROM}/usb.img"
-  DEFAULT_TO_HELP="Destination file for USB image."
-  AUTOTEST_SRC=/usr/local/autotest
-else
-  # Outside the chroot, so output to the default device for a usb key.
-  DEFAULT_TO="/dev/sdb"
-  DEFAULT_TO_HELP="Destination device for USB keyfob."
-  AUTOTEST_SRC=${DEFAULT_CHROOT_DIR}/usr/local/autotest
-fi
 
 # Flags
-DEFINE_string from "$DEFAULT_FROM" \
+DEFINE_string board "" "Board for which the image was built"
+DEFINE_string from "" \
   "Directory containing rootfs.image and mbr.image"
-DEFINE_string to "$DEFAULT_TO" "$DEFAULT_TO_HELP"
+DEFINE_string to "" "$DEFAULT_TO_HELP"
 DEFINE_boolean yes $FLAGS_FALSE "Answer yes to all prompts" "y"
 DEFINE_boolean install_autotest $FLAGS_FALSE \
   "Whether to install autotest to the stateful partition."
 
 # Parse command line
 FLAGS "$@" || exit 1
 eval set -- "${FLAGS_ARGV}"
 
+# Inside the chroot, so output to usb.img in the same dir as the other
+# Script can be run either inside or outside the chroot.
+if [ $INSIDE_CHROOT -eq 1 ]
+then
+  AUTOTEST_SRC="/usr/local/autotest/${FLAGS_board}"
+else
+  AUTOTEST_SRC="${DEFAULT_CHROOT_DIR}/usr/local/autotest/${FLAGS_board}"
+fi
+
 # Die on any errors.
 set -e
 
+# If from isn't explicitly set
+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 to isn't explicitly set
+if [ -z "$FLAGS_to" ]; then
+  # Script can be run either inside or outside the chroot.
+  if [ $INSIDE_CHROOT -eq 1 ]
+  then
+    # Inside the chroot, so output to usb.img in the same dir as the other
+    # images.
+    FLAGS_to="${FLAGS_from}/usb.img"
+  else
+    # Outside the chroot, so output to the default device for a usb key.
+    FLAGS_to="/dev/sdb"
+  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`
 
 function do_cleanup {
   sudo losetup -d "$LOOP_DEV"
 }
 
 STATEFUL_DIR=${FLAGS_from}/stateful_partition
@@ skipping 126 matching lines @@
   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