| OLD | NEW |
| 1 #!/bin/sh | 1 #!/bin/sh |
| 2 | 2 |
| 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
| 6 | 6 |
| 7 # This scripts performs update of stateful partition directories useful for | 7 # This scripts performs update of stateful partition directories useful for |
| 8 # dev_mode. | 8 # dev_mode. |
| 9 | 9 |
| 10 . /usr/lib/shflags | 10 . /usr/lib/shflags |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 # Die on error. | 21 # Die on error. |
| 22 set -e | 22 set -e |
| 23 | 23 |
| 24 remove_quotes() { | 24 remove_quotes() { |
| 25 echo "$1" | sed -e "s/^'//; s/'$//" | 25 echo "$1" | sed -e "s/^'//; s/'$//" |
| 26 } | 26 } |
| 27 | 27 |
| 28 update_dev_image () { | 28 update_dev_image () { |
| 29 LSB_RELEASE="/etc/lsb-release" | 29 LSB_RELEASE="/etc/lsb-release" |
| 30 STATEFUL_DIR="/mnt/stateful_partition" | 30 STATEFUL_DIR="/mnt/stateful_partition" |
| 31 STATEFUL_IMAGE="$STATEFUL_DIR/stateful.image" | 31 WORK_DIR="${STATEFUL_DIR}/stateful_update" |
| 32 STATEFUL_MOUNT_POINT="/tmp/s" | |
| 33 | 32 |
| 34 if [ -n "${FLAGS_ARGV}" ]; then | 33 if [ -n "${FLAGS_ARGV}" ]; then |
| 35 BASE_UPDATE_URL=$(remove_quotes "${FLAGS_ARGV}") | 34 BASE_UPDATE_URL=$(remove_quotes "${FLAGS_ARGV}") |
| 36 else | 35 else |
| 37 if [ -f "$STATEFUL_DIR$LSB_RELEASE" ]; then | 36 if [ -f "${STATEFUL_DIR}${LSB_RELEASE}" ]; then |
| 38 DEVSERVER_URL=$(grep CHROMEOS_DEVSERVER $STATEFUL_DIR$LSB_RELEASE | \ | 37 DEVSERVER_URL=$(grep CHROMEOS_DEVSERVER ${STATEFUL_DIR}${LSB_RELEASE} | \ |
| 39 cut -f 2 -d '=') | 38 cut -f 2 -d '=') |
| 40 fi | 39 fi |
| 41 if [ -z "$DEVSERVER_URL" ]; then | 40 if [ -z "${DEVSERVER_URL}" ]; then |
| 42 DEVSERVER_URL=$(grep CHROMEOS_DEVSERVER $LSB_RELEASE | cut -f 2 -d '=') | 41 DEVSERVER_URL=$(grep CHROMEOS_DEVSERVER ${LSB_RELEASE} | cut -f 2 -d '=') |
| 43 fi | 42 fi |
| 44 # Sanity check. | 43 # Sanity check. |
| 45 if [ -z "$DEVSERVER_URL" ]; then | 44 if [ -z "${DEVSERVER_URL}" ]; then |
| 46 echo >&2 "No CHROMEOS_DEVSERVER URL found in lsb-release file" | 45 echo >&2 "No CHROMEOS_DEVSERVER URL found in lsb-release file" |
| 47 exit 1 | 46 exit 1 |
| 48 fi | 47 fi |
| 49 # Devserver URL should never contain "/update" | 48 # Devserver URL should never contain "/update" |
| 50 DEVSERVER_URL=$(echo $DEVSERVER_URL | sed -e 's#/update##') | 49 DEVSERVER_URL=$(echo ${DEVSERVER_URL} | sed -e 's#/update##') |
| 51 BASE_UPDATE_URL="$DEVSERVER_URL/static" | 50 BASE_UPDATE_URL="${DEVSERVER_URL}/static" |
| 52 fi | 51 fi |
| 53 | 52 |
| 54 STATEFUL_UPDATE_URL="$BASE_UPDATE_URL/stateful.image.gz" | 53 STATEFUL_UPDATE_URL="${BASE_UPDATE_URL}/stateful.tgz" |
| 55 | 54 |
| 56 # Prepare directories for update. | 55 # Prepare directories for update. |
| 57 rm -rf "$STATEFUL_DIR/var_new" "$STATEFUL_DIR/dev_image_new" | 56 rm -rf "${WORK_DIR}" |
| 57 mkdir "${WORK_DIR}" |
| 58 | 58 |
| 59 # Prepare and mount new stateful partition. | 59 echo "Download stateful image from ${STATEFUL_UPDATE_URL}" |
| 60 rm -rf "$STATEFUL_MOUNT_POINT" | |
| 61 mkdir -p "$STATEFUL_MOUNT_POINT" | |
| 62 | |
| 63 echo "Download stateful partition image from $STATEFUL_UPDATE_URL" | |
| 64 # Unzip mount and copy the relevant directories. | 60 # Unzip mount and copy the relevant directories. |
| 65 # Get the update. | 61 # Get the update. |
| 66 eval "wget -qS -T 300 -O - \"$STATEFUL_UPDATE_URL\"" | \ | 62 eval "wget -qS -T 300 -O - \"${STATEFUL_UPDATE_URL}\"" | \ |
| 67 gzip -d > $STATEFUL_IMAGE | 63 tar -C ${WORK_DIR} -xz |
| 68 echo >&2 "Successfully downloaded update" | 64 echo >&2 "Successfully downloaded update" |
| 69 trap "rm -f \"$STATEFUL_IMAGE\"" EXIT | 65 |
| 70 mount -n -o loop "$STATEFUL_IMAGE" "$STATEFUL_MOUNT_POINT" | 66 if [ -d "${WORK_DIR}/var" ] && [ -d "${WORK_DIR}/dev_image" ] ; then |
| 71 if [ -d "$STATEFUL_MOUNT_POINT/var" ] && \ | |
| 72 [ -d "$STATEFUL_MOUNT_POINT/dev_image" ] ; then | |
| 73 echo >&2 "Copying new /var from stateful update" | |
| 74 cp -rf "$STATEFUL_MOUNT_POINT/var" "$STATEFUL_DIR/var_new" | |
| 75 echo >&2 "Copying new /usr/local from stateful update" | |
| 76 cp -rf "$STATEFUL_MOUNT_POINT/dev_image" "$STATEFUL_DIR/dev_image_new" | |
| 77 echo >&2 "Notifying startup that an update is available" | 67 echo >&2 "Notifying startup that an update is available" |
| 78 touch "$STATEFUL_DIR/.update_available" | 68 touch "${STATEFUL_DIR}/.update_available" |
| 79 else | 69 else |
| 80 echo >&2 "No update available" | 70 echo >&2 "No update available" |
| 81 fi | 71 fi |
| 82 | |
| 83 umount -n -d "$STATEFUL_MOUNT_POINT" | |
| 84 | |
| 85 rm -f "$STATEFUL_IMAGE" | |
| 86 | |
| 87 trap - EXIT | |
| 88 } | 72 } |
| 89 | 73 |
| 90 update_old_state () { | 74 update_old_state () { |
| 91 echo >&2 "Not updating state except for dev_image." | 75 echo >&2 "Performing standard stateful update." |
| 92 } | 76 } |
| 93 | 77 |
| 94 update_clean_state () { | 78 update_clean_state () { |
| 95 echo >&2 "Restoring state to factory_install with dev_image." | 79 echo >&2 "Restoring state to factory_install with dev_image." |
| 96 echo "fast test" > "/mnt/stateful_partition/factory_install_reset" | 80 echo "fast test" > "/mnt/stateful_partition/factory_install_reset" |
| 97 } | 81 } |
| 98 | 82 |
| 99 main () { | 83 main () { |
| 100 update_dev_image | 84 update_dev_image |
| 101 if [ "${FLAGS_stateful_change}" = "${OLD_STATE}" ]; then | 85 if [ "${FLAGS_stateful_change}" = "${OLD_STATE}" ]; then |
| 102 update_old_state | 86 update_old_state |
| 103 elif [ "${FLAGS_stateful_change}" = "${CLEAN_STATE}" ]; then | 87 elif [ "${FLAGS_stateful_change}" = "${CLEAN_STATE}" ]; then |
| 104 update_clean_state | 88 update_clean_state |
| 105 else | 89 else |
| 106 echo >&2 "Invalid state given to stateful update. Aborting..." | 90 echo >&2 "Invalid state given to stateful update. Aborting..." |
| 107 exit 1 | 91 exit 1 |
| 108 fi | 92 fi |
| 109 } | 93 } |
| 110 | 94 |
| 111 main $@ | 95 main $@ |
| OLD | NEW |