From 87336bdef6ae1d00492a5f9a5d97623212d020ba Mon Sep 17 00:00:00 2001
From: Kevin Degi <kdegi@codeaurora.org>
Date: Wed, 9 Jun 2010 23:47:59 -0700
Subject: [PATCH] Add scripts to update an existing image

---
 bin/cros_revert_image.sh | 42 +++++++++++++++++++++++
 bin/cros_update_image.sh | 72 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100755 bin/cros_revert_image.sh
 create mode 100755 bin/cros_update_image.sh

diff --git a/bin/cros_revert_image.sh b/bin/cros_revert_image.sh
new file mode 100755
index 0000000000..5810af836c
--- /dev/null
+++ b/bin/cros_revert_image.sh
@@ -0,0 +1,42 @@
+#!/bin/bash
+
+# Usage:
+#   revert_image.sh [image_to_revert]
+#
+# This assumes the image has been updated by update_image.sh.
+usage()
+{
+cat <<EOF
+
+usage:
+   revert_image.sh [image_to_revert]
+EOF
+}
+
+if [[ $# < 1 ]]; then
+   echo "Not enough arguments supplied."
+   usage
+   exit 1
+fi
+
+IMAGE=$( readlink -f ${1} )
+IMAGE_DIR=$( dirname ${IMAGE} )
+
+if [[ -z "${IMAGE}" ]]; then
+   echo "Missing required argument 'image_to_revert'"
+   usage
+   exit 1
+fi
+
+cd ${IMAGE_DIR}
+
+if [[ ! -d "./orig_partitions" ]]; then
+   echo "Could not find original partitions."
+   exit 1
+fi
+
+yes | cp ./orig_partitions/* ./
+
+./pack_partitions.sh ${IMAGE}
+rm -rf ./orig_partitions
+cd -
diff --git a/bin/cros_update_image.sh b/bin/cros_update_image.sh
new file mode 100755
index 0000000000..cf5f2e1570
--- /dev/null
+++ b/bin/cros_update_image.sh
@@ -0,0 +1,72 @@
+#!/bin/bash
+
+# Usage:
+# update_image.sh [image_to_update] [packages...]
+usage()
+{
+cat <<EOF
+
+usage:
+   update_image.sh [image_to_update] [packages...]
+EOF
+}
+
+if [[ $# < 2 ]]; then
+   echo "Not enough arguments supplied."
+   usage
+   exit 1
+fi
+
+if [[ -f /home/${USER}/trunk/src/scripts/.default_board ]]; then
+   BOARD=$( cat /home/${USER}/trunk/src/scripts/.default_board )
+else
+   BOARD=st1q
+fi
+
+IMAGE=$( readlink -f ${1} )
+IMAGE_DIR=$( dirname ${IMAGE} )
+shift
+PKGS=$@
+
+if [[ -z "${IMAGE}" || ! -f ${IMAGE} ]]; then
+   echo "Missing required argument 'image_to_update'"
+   usage
+   exit 1
+fi
+
+cd ${IMAGE_DIR}
+if ! [[ -x ./unpack_partitions.sh && -x ./pack_partitions.sh ]]; then
+   echo "Could not find image manipulation scripts."
+   exit 1
+fi
+
+./unpack_partitions.sh ${IMAGE}
+mkdir -p ./rootfs
+mkdir -p ./stateful_part
+mkdir -p ./orig_partitions
+
+rm -rf ./orig_partitions/*
+cp ./part_* ./orig_partitions
+sudo mount -o loop part_3 rootfs
+sudo mount -o loop part_1 stateful_part
+sudo mount --bind stateful_part/dev_image rootfs/usr/local
+sudo mount --bind stateful_part/var rootfs/var
+
+emerge-${BOARD} --root="./rootfs" \
+   --root-deps=rdeps --nodeps --usepkgonly ${PKGS}
+
+#if the kernel is one of the packages that got updated
+#we need to update the kernel partition as well.
+if [[ ${PKGS/kernel/} != ${PKGS} ]]; then
+   rm -rf part_2
+   sudo dd if="/dev/zero" of=part_2 bs=512 count=8192
+   sudo dd if="./rootfs/boot/vmlinuz" of=part_2 bs=512 count=8192 conv=notrunc
+fi
+
+sudo umount rootfs/usr/local
+sudo umount rootfs/var
+sudo umount rootfs
+sudo umount stateful_part
+./pack_partitions.sh ${IMAGE}
+
+cd -