target/linux/ath79/generic/base-files/lib/upgrade/dualboot_datachk.sh

   1 # The U-Boot loader with the datachk patchset for dualbooting requires image
   2 # sizes and checksums to be provided in the U-Boot environment.
   3 # The devices come with 2 main partitions - while one is active
   4 # sysupgrade will flash the other. The boot order is changed to boot the
   5 # newly flashed partition. If the new partition can't be booted due to
   6 # upgrade failures the previously used partition is loaded.
   7
   8 platform_do_upgrade_dualboot_datachk() {
   9         local tar_file="$1"
  10         local restore_backup
  11         local primary_kernel_mtd
  12
  13         local setenv_script="/tmp/fw_env_upgrade"
  14
  15         local inactive_mtd="$(find_mtd_index $PART_NAME)"
  16         local inactive_offset="$(cat /sys/class/mtd/mtd${inactive_mtd}/offset)"
  17         local total_size="$(cat /sys/class/mtd/mtd${inactive_mtd}/size)"
  18         local flash_start_mem=0x9f000000
  19
  20         # detect to which flash region the new image is written to.
  21         #
  22         # 1. check what is the mtd index for the first flash region on this
  23         #    device
  24         # 2. check if the target partition ("inactive") has the mtd index of
  25         #    the first flash region
  26         #
  27         #    - when it is: the new bootseq will be 1,2 and the first region is
  28         #      modified
  29         #    - when it isnt: bootseq will be 2,1 and the second region is
  30         #      modified
  31         #
  32         # The detection has to be done via the hardcoded mtd partition because
  33         # the current boot might be done with the fallback region. Let us
  34         # assume that the current bootseq is 1,2. The bootloader detected that
  35         # the image in flash region 1 is corrupt and thus switches to flash
  36         # region 2. The bootseq in the u-boot-env is now still the same and
  37         # the sysupgrade code can now only rely on the actual mtd indexes and
  38         # not the bootseq variable to detect the currently booted flash
  39         # region/image.
  40         #
  41         # In the above example, an implementation which uses bootseq ("1,2") to
  42         # detect the currently booted image would assume that region 1 is booted
  43         # and then overwrite the variables for the wrong flash region (aka the
  44         # one which isn't modified). This could result in a device which doesn't
  45         # boot anymore to Linux until it was reflashed with ap51-flash.
  46         local next_boot_part="1"
  47         case "$(board_name)" in
  48         plasmacloud,pa300)
  49                 primary_kernel_mtd=3
  50                 ;;
  51         *)
  52                 echo "failed to detect primary kernel mtd partition for board"
  53                 return 1
  54                 ;;
  55         esac
  56         [ "$inactive_mtd" = "$primary_kernel_mtd" ] || next_boot_part="2"
  57
  58         local board_dir=$(tar tf $tar_file | grep -m 1 '^sysupgrade-.*/$')
  59         board_dir=${board_dir%/}
  60
  61         local kernel_length=$(tar xf $tar_file ${board_dir}/kernel -O | wc -c)
  62         local rootfs_length=$(tar xf $tar_file ${board_dir}/root -O | wc -c)
  63         # rootfs without EOF marker
  64         rootfs_length=$((rootfs_length-4))
  65
  66         local kernel_md5=$(tar xf $tar_file ${board_dir}/kernel -O | md5sum); kernel_md5="${kernel_md5%% *}"
  67         # md5 checksum of rootfs with EOF marker
  68         local rootfs_md5=$(tar xf $tar_file ${board_dir}/root -O | dd bs=1 count=$rootfs_length | md5sum); rootfs_md5="${rootfs_md5%% *}"
  69
  70         #
  71         # add tar support to get_image() to use default_do_upgrade() instead?
  72         #
  73
  74         # take care of restoring a saved config
  75         [ -n "$UPGRADE_BACKUP" ] && restore_backup="${MTD_CONFIG_ARGS} -j ${UPGRADE_BACKUP}"
  76
  77         mtd -q erase inactive
  78         tar xf $tar_file ${board_dir}/root -O | mtd -n -p $kernel_length $restore_backup write - $PART_NAME
  79         tar xf $tar_file ${board_dir}/kernel -O | mtd -n write - $PART_NAME
  80
  81         # prepare new u-boot env
  82         if [ "$next_boot_part" = "1" ]; then
  83                 echo "bootseq 1,2" > $setenv_script
  84         else
  85                 echo "bootseq 2,1" > $setenv_script
  86         fi
  87
  88         printf "kernel_size_%i %i\n" $next_boot_part $((kernel_length / 1024)) >> $setenv_script
  89         printf "vmlinux_start_addr 0x%08x\n" $((flash_start_mem + inactive_offset)) >> $setenv_script
  90         printf "vmlinux_size 0x%08x\n" ${kernel_length} >> $setenv_script
  91         printf "vmlinux_checksum %s\n" ${kernel_md5} >> $setenv_script
  92
  93         printf "rootfs_size_%i %i\n" $next_boot_part $(((total_size-kernel_length) / 1024)) >> $setenv_script
  94         printf "rootfs_start_addr 0x%08x\n" $((flash_start_mem+inactive_offset+kernel_length)) >> $setenv_script
  95         printf "rootfs_size 0x%08x\n" ${rootfs_length} >> $setenv_script
  96         printf "rootfs_checksum %s\n" ${rootfs_md5} >> $setenv_script
  97
  98         # store u-boot env changes
  99         mkdir -p /var/lock
 100         fw_setenv -s $setenv_script || {
 101                 echo "failed to update U-Boot environment"
 102                 return 1
 103         }
 104 }