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 |