scripts/upgrade-support/upgrade-trigger.sh

   1 #!/bin/bash
   2 # Master script for upgrade. Called by update-manager/manually to start upgrade procedure.
   3 set -o errexit
   4 trap 'echo "Aborting due to errexit on ${0##*/}:$LINENO. Exit code: $?" >&2' ERR
   5 set -o errtrace -e -o pipefail
   6
   7 if [ ! -z "${UPGRADE_DEBUG}" ]; then
   8         set -x
   9 fi
  10
  11 PATH=/bin:/usr/bin:/sbin:/usr/sbin
  12 FOTA_DIR="/opt/usr/data/fota"
  13 STATUS_DIR="/opt/data/update"
  14 DOWNLOAD_DELTA="$1"
  15 SCRIPT_NAME="upgrade-trigger.sh"
  16 LOG_FILE="/tmp/upgrade-trigger.log"
  17 SCRIPT_UPGRADE_PREPARE_PARTITIONS="upgrade-prepare-partitions.sh"
  18 SCRIPT_UPGRADE_PARTIAL="upgrade-partial.sh"
  19 SCRIPT_UPGRADE_FOTA="upgrade-fota.sh"
  20 FLOCK_PATH="/var/lock/clone_partitions.lock"
  21 DELTA_VERIFIER="/usr/bin/delta-verifier"
  22
  23 prepare_fota_dir() {
  24         if [ -d "$FOTA_DIR" ]; then
  25                 # Cleanup FOTA_DIR
  26                 if [ "$(dirname $DOWNLOAD_DELTA)" = "$FOTA_DIR" ]; then
  27                         # If provided delta is from inside the FOTA_DIR, delete everything else
  28                         DELTA_FILE_NAME=$(basename $DOWNLOAD_DELTA)
  29                         rm -rf -- "$FOTA_DIR/!($DELTA_FILE_NAME)"
  30                 else
  31                         # If provided delta is from outside the FOTA_DIR delete everything
  32                         rm -rf -- "$FOTA_DIR/*"
  33                 fi
  34         else
  35                 log "[Info] Create fota dir..." "$LOG_FILE"
  36                 mkdir -p "$FOTA_DIR"
  37         fi
  38
  39         if [ ! -d "$STATUS_DIR" ]; then
  40                 log "[Info] Create status dir..." "$LOG_FILE"
  41                 mkdir -p "$STATUS_DIR"
  42         fi
  43 }
  44
  45 log() {
  46         # log format: [script_name][tag]actual_log
  47         LOG="[${SCRIPT_NAME}]$1"
  48         if [ "$2" != "" ]; then
  49                 echo "$LOG" >> "$2"
  50         fi
  51         echo "$LOG"
  52
  53         return 0
  54 }
  55
  56 is_ab_upgrade() {
  57         if grep -q partition_ab= /proc/cmdline; then
  58                 return $TRUE
  59         fi
  60         return $FALSE
  61 }
  62
  63 do_update() {
  64         if is_ab_upgrade; then
  65                 set_upgrade_status 5
  66
  67                 if [ "$(device_board_get_partition_ab_cloned)" -eq 0 ]; then
  68                         unpack_file "${DOWNLOAD_DELTA}" "${SCRIPT_UPGRADE_PREPARE_PARTITIONS}"
  69                         if ! "${FOTA_DIR}/${SCRIPT_UPGRADE_PREPARE_PARTITIONS}"; then
  70                                 exit_error
  71                         fi
  72                 else
  73                         echo "[Info] Partitions already cloned"
  74                 fi
  75
  76                 set_upgrade_status 20
  77
  78                 # In the next steps the content of the partitions of the second slot will
  79                 # be modified, so they will not be a clone of the current partitions
  80                 device_board_clear_partition_ab_cloned
  81
  82                 unpack_file "${DOWNLOAD_DELTA}" "${SCRIPT_UPGRADE_PARTIAL}"
  83                 unpack_file "${DOWNLOAD_DELTA}" "${SCRIPT_UPGRADE_FOTA}"
  84
  85                 if ! "${FOTA_DIR}/${SCRIPT_UPGRADE_PARTIAL}" "${DOWNLOAD_DELTA}"; then
  86                         exit_error
  87                 fi
  88
  89                 set_upgrade_status 40
  90
  91                 if ! "${FOTA_DIR}/${SCRIPT_UPGRADE_FOTA}"; then
  92                         exit_error
  93                 fi
  94
  95                 set_upgrade_status 80
  96         else
  97                 echo "[Error] Non-A/B upgrade is unsupported"
  98                 device_board_set_upgrade_status -1
  99                 exit 1
 100         fi
 101 }
 102
 103 verify_file() {
 104         FILE_PATH="$1"
 105         FILE_NAME=$(basename "$FILE_PATH")
 106
 107         VALID_CHECKSUM=$(awk "\$2 ~ /^$FILE_NAME\$/ {print \$1}" "$FOTA_DIR/checksum.SHA1")
 108         if [ "$VALID_CHECKSUM" == "" ]; then
 109                 echo "[Error] No $FILE_NAME in checksum.SHA1"
 110                 device_board_set_upgrade_status -1
 111                 exit 1
 112         fi
 113
 114         if ! echo "$VALID_CHECKSUM  $FILE_PATH" | sha1sum --check --status; then
 115                 echo "[Error] Checksum for file $FILE_NAME is invalid"
 116                 device_board_set_upgrade_status -1
 117                 exit 1
 118         fi
 119
 120         echo "[Info] Checksum of $FILE_NAME is OK"
 121 }
 122
 123 (
 124 flock 9
 125
 126 rm -f "$LOG_FILE" && touch "$LOG_FILE"
 127
 128 if [ "$#" != "1" ] || [ ! -f "$1" ]; then
 129         log "[Error] Usage: $0 path_to_delta.tar[.gz]" "$LOG_FILE"
 130         exit 1
 131 fi
 132
 133 log "[Info] Using <$DOWNLOAD_DELTA> delta file." "$LOG_FILE"
 134
 135 prepare_fota_dir
 136 tar xfp "$DOWNLOAD_DELTA" -C "$FOTA_DIR" checksum.SHA1
 137 verify_file "$0"
 138 tar xfp "$DOWNLOAD_DELTA" -C "$FOTA_DIR" upgrade-common.inc
 139 verify_file "$FOTA_DIR/upgrade-common.inc"
 140 tar xfp "$DOWNLOAD_DELTA" -C "$FOTA_DIR" update-info.ini
 141 verify_file "$FOTA_DIR/update-info.ini"
 142 tar xfp "$DOWNLOAD_DELTA" -C "$FOTA_DIR" delta-verifier
 143 verify_file "$FOTA_DIR/delta-verifier"
 144
 145 log "[Info] Begin delta verification"
 146 RET=0
 147 # '||'' used to stop script form exiting (errexit is set)
 148 "${FOTA_DIR}/delta-verifier" --update_info_path "$FOTA_DIR/update-info.ini" || RET=$?
 149
 150 if [ ${RET} -ne 0 ]; then
 151         log "[Error] Delta verification unsuccessful"
 152         device_board_set_upgrade_status -1
 153         exit 1
 154 fi
 155 log "[Info] Delta verification success"
 156
 157 . "$FOTA_DIR"/upgrade-common.inc
 158
 159 set_upgrade_status 1
 160 verify_img "${DOWNLOAD_DELTA}"
 161 do_update
 162
 163 critical_log "RO update: $SCRIPT_NAME success" "$LOG_FILE"
 164 ) 9> "$FLOCK_PATH"