mirror of
https://github.com/flatcar/scripts.git
synced 2026-05-04 19:56:32 +02:00
Cleanup and add open to push test image to usb
Review URL: http://codereview.chromium.org/668170
This commit is contained in:
Chris Sosa
parent
83b50104dd
commit
8cf12935c5
107
image_to_usb.sh
107
image_to_usb.sh
@ -13,15 +13,17 @@
|
||||
get_default_board
|
||||
|
||||
# Flags
|
||||
DEFINE_string board "$DEFAULT_BOARD" "Board for which the image was built"
|
||||
DEFINE_string board "${DEFAULT_BOARD}" "Board for which the image was built"
|
||||
DEFINE_string from "" \
|
||||
"Directory containing rootfs.image and mbr.image"
|
||||
DEFINE_string to "" "$DEFAULT_TO_HELP"
|
||||
DEFINE_boolean yes $FLAGS_FALSE "Answer yes to all prompts" "y"
|
||||
DEFINE_boolean install_autotest $FLAGS_FALSE \
|
||||
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."
|
||||
DEFINE_boolean copy_kernel $FLAGS_FALSE \
|
||||
DEFINE_boolean copy_kernel ${FLAGS_FALSE} \
|
||||
"Copy the kernel to the fourth partition."
|
||||
DEFINE_boolean test_image "${FLAGS_FALSE}" \
|
||||
"Uses test image if available, otherwise creates one as rootfs_test.image."
|
||||
|
||||
# Parse command line
|
||||
FLAGS "$@" || exit 1
|
||||
@ -29,7 +31,7 @@ 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 ]
|
||||
if [ ${INSIDE_CHROOT} -eq 1 ]
|
||||
then
|
||||
AUTOTEST_SRC="/usr/local/autotest/${FLAGS_board}"
|
||||
else
|
||||
@ -40,26 +42,26 @@ fi
|
||||
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
|
||||
if [ -z ${FLAGS_from} ] && [ -z ${FLAGS_board} ] ; 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
|
||||
if [ -z "${FLAGS_from}" ]; then
|
||||
IMAGES_DIR="${DEFAULT_BUILD_ROOT}/images/${FLAGS_board}"
|
||||
FLAGS_from="${IMAGES_DIR}/$(ls -t $IMAGES_DIR 2>&-| head -1)"
|
||||
FLAGS_from="${IMAGES_DIR}/$(ls -t ${IMAGES_DIR} 2>&-| head -1)"
|
||||
fi
|
||||
|
||||
if [ ! -d "$FLAGS_from" ] ; then
|
||||
echo "Cannot find image directory $FLAGS_from"
|
||||
if [ ! -d "${FLAGS_from}" ] ; then
|
||||
echo "Cannot find image directory ${FLAGS_from}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# If to isn't explicitly set
|
||||
if [ -z "$FLAGS_to" ]; then
|
||||
if [ -z "${FLAGS_to}" ]; then
|
||||
# Script can be run either inside or outside the chroot.
|
||||
if [ $INSIDE_CHROOT -eq 1 ]
|
||||
if [ ${INSIDE_CHROOT} -eq 1 ]
|
||||
then
|
||||
# Inside the chroot, so output to usb.img in the same dir as the other
|
||||
# images.
|
||||
@ -71,12 +73,27 @@ if [ -z "$FLAGS_to" ]; then
|
||||
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`
|
||||
# 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"
|
||||
|
||||
# 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"
|
||||
sudo losetup -d "${LOOP_DEV}"
|
||||
}
|
||||
|
||||
STATEFUL_DIR=${FLAGS_from}/stateful_partition
|
||||
@ -103,15 +120,15 @@ function install_autotest {
|
||||
}
|
||||
|
||||
# Copy MBR and rootfs to output image
|
||||
if [ -b "$FLAGS_to" ]
|
||||
if [ -b "${FLAGS_to}" ]
|
||||
then
|
||||
# Output to a block device (i.e., a real USB key), so need sudo dd
|
||||
echo "Copying USB image ${FLAGS_from} to device ${FLAGS_to}..."
|
||||
|
||||
# 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
|
||||
if echo ${FLAGS_to} | grep -q '[0-9]$'; then
|
||||
local 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 "
|
||||
@ -121,13 +138,13 @@ then
|
||||
fi
|
||||
|
||||
# Make sure this is really what the user wants, before nuking the device
|
||||
if [ $FLAGS_yes -ne $FLAGS_TRUE ]
|
||||
if [ ${FLAGS_yes} -ne ${FLAGS_TRUE} ]
|
||||
then
|
||||
echo "This will erase all data on this device:"
|
||||
sudo fdisk -l "$FLAGS_to" | grep Disk | head -1
|
||||
sudo fdisk -l "${FLAGS_to}" | grep Disk | head -1
|
||||
read -p "Are you sure (y/N)? " SURE
|
||||
SURE="${SURE:0:1}" # Get just the first character
|
||||
if [ "$SURE" != "y" ]
|
||||
if [ "${SURE}" != "y" ]
|
||||
then
|
||||
echo "Ok, better safe than sorry."
|
||||
exit 1
|
||||
@ -135,23 +152,23 @@ then
|
||||
fi
|
||||
|
||||
echo "attempting to unmount any mounts on the USB device"
|
||||
for i in "$FLAGS_to"*
|
||||
for i in "${FLAGS_to}"*
|
||||
do
|
||||
! sudo umount "$i"
|
||||
done
|
||||
sleep 3
|
||||
|
||||
PART_SIZE=$(stat -c%s "${FLAGS_from}/rootfs.image") # Bytes
|
||||
PART_SIZE=$(stat -c%s "${ROOTFS_IMAGE}") # Bytes
|
||||
|
||||
echo "Copying root fs..."
|
||||
sudo "${SCRIPTS_DIR}"/file_copy.py \
|
||||
if="${FLAGS_from}/rootfs.image" \
|
||||
of="$FLAGS_to" bs=4M \
|
||||
seek_bytes=$(( ($PART_SIZE * 2) + 512 ))
|
||||
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" ]
|
||||
if [ -z "${LOOP_DEV}" ]
|
||||
then
|
||||
echo "No free loop device. Free up a loop device or reboot. exiting."
|
||||
exit 1
|
||||
@ -160,44 +177,44 @@ then
|
||||
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 ]
|
||||
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}"
|
||||
install_autotest
|
||||
fi
|
||||
sync
|
||||
sudo losetup -d "$LOOP_DEV"
|
||||
sudo losetup -d "${LOOP_DEV}"
|
||||
sync
|
||||
|
||||
trap - EXIT
|
||||
|
||||
if [ $FLAGS_copy_kernel -eq $FLAGS_TRUE ]
|
||||
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 ))"
|
||||
--offset "$(( (${PART_SIZE} * 3) + 512 ))"
|
||||
fi
|
||||
|
||||
echo "Copying MBR..."
|
||||
sudo "${SCRIPTS_DIR}"/file_copy.py \
|
||||
if="${FLAGS_from}/mbr.image" of="$FLAGS_to"
|
||||
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 "${FLAGS_from}/rootfs.image")
|
||||
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))
|
||||
seek=$((${PART_SIZE} - 1))
|
||||
mkfs.ext3 -F -L C-STATE "${FLAGS_from}/stateful_partition.image"
|
||||
|
||||
if [ $FLAGS_install_autotest -eq $FLAGS_TRUE ]
|
||||
if [ ${FLAGS_install_autotest} -eq ${FLAGS_TRUE} ]
|
||||
then
|
||||
sudo mount -o loop "${FLAGS_from}/stateful_partition.image" \
|
||||
"${STATEFUL_DIR}"
|
||||
@ -206,19 +223,19 @@ else
|
||||
|
||||
# Create a sparse output file
|
||||
dd if=/dev/zero of="${FLAGS_to}" bs=1 count=1 \
|
||||
seek=$(( ($PART_SIZE * 2) + 512 - 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 \
|
||||
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 "${FLAGS_from}/rootfs.image" >> "$FLAGS_to"
|
||||
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 ]
|
||||
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"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user