target/linux/ar71xx/base-files/lib/upgrade/merakinand.sh

   1 #!/bin/sh
   2 #
   3 # Copyright (C) 2015 Chris Blake <chrisrblake93@gmail.com>
   4 #
   5 # Custom upgrade script for Meraki NAND devices (ex. MR18)
   6 # Based on dir825.sh and stock nand functions
   7 #
   8 . /lib/ar71xx.sh
   9 . /lib/functions.sh
  10
  11 get_magic_at() {
  12         local mtddev=$1
  13         local pos=$2
  14         dd bs=1 count=2 skip=$pos if=$mtddev 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
  15 }
  16
  17 mr18_is_caldata_valid() {
  18         local mtddev=$1
  19         local magic
  20
  21         magic=$(get_magic_at $mtddev 4096)
  22         [ "$magic" != "0202" ] && return 0
  23
  24         magic=$(get_magic_at $mtddev 20480)
  25         [ "$magic" != "0202" ] && return 0
  26
  27         magic=$(get_magic_at $mtddev 36864)
  28         [ "$magic" != "0202" ] && return 0
  29
  30         return 1
  31 }
  32
  33 merakinand_copy_caldata() {
  34         local cal_src=$1
  35         local cal_dst=$2
  36         local ubidev=$( nand_find_ubi $CI_UBIPART )
  37         local board_name="$(cat /tmp/sysinfo/board_name)"
  38         local rootfs_size="$(ubinfo /dev/ubi0 -N rootfs_data | grep "Size" | awk '{ print $6 }')"
  39
  40         # Setup partitions using board name, in case of future platforms
  41         case "$board_name" in
  42         "mr18")
  43                 # Src is MTD
  44                 mtd_src=$(find_mtd_chardev $cal_src)
  45                 [ -n "$mtd_src" ] || {
  46                         echo "no mtd device found for partition $cal_src"
  47                         exit 1
  48                 }
  49
  50                 # Dest is UBI
  51                 # TODO: possibly add create (hard to do when rootfs_data is expanded & mounted)
  52                 # Would need to be done from ramdisk
  53                 mtd_dst="$(nand_find_volume $ubidev $cal_dst)"
  54                 [ -n "$mtd_dst" ] || {
  55                         echo "no ubi device found for partition $cal_dst"
  56                         exit 1
  57                 }
  58
  59                 mr18_is_caldata_valid "$mtd_src" && {
  60                         echo "no valid calibration data found in $cal_src"
  61                         exit 1
  62                 }
  63
  64                 mr18_is_caldata_valid "/dev/$mtd_dst" && {
  65                         echo "Copying calibration data from $cal_src to $cal_dst..."
  66                         dd if="$mtd_src" of=/tmp/caldata.tmp 2>/dev/null
  67                         ubiupdatevol "/dev/$mtd_dst" /tmp/caldata.tmp
  68                         rm /tmp/caldata.tmp
  69                         sync
  70                 }
  71                 return 0
  72                 ;;
  73         *)
  74                 echo "Unsupported device $board_name";
  75                 return 1
  76                 ;;
  77         esac
  78 }
  79
  80 merakinand_do_kernel_check() {
  81         local board_name="$1"
  82         local tar_file="$2"
  83         local image_magic_word=`(tar xf $tar_file sysupgrade-$board_name/kernel -O 2>/dev/null | dd bs=1 count=4 skip=0 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"')`
  84
  85         # What is our kernel magic string?
  86         case "$board_name" in
  87         "mr18")
  88                 [ "$image_magic_word" == "8e73ed8a" ] && {
  89                         echo "pass" && return 0
  90                 }
  91                 ;;
  92         esac
  93
  94         exit 1
  95 }
  96
  97 merakinand_do_platform_check() {
  98         local board_name="$1"
  99         local tar_file="$2"
 100         local control_length=`(tar xf $tar_file sysupgrade-$board_name/CONTROL -O | wc -c) 2> /dev/null`
 101         local file_type="$(identify_tar $2 sysupgrade-$board_name/root)"
 102         local kernel_magic="$(merakinand_do_kernel_check $1 $2)"
 103
 104         case "$board_name" in
 105         "mr18")
 106                 [ "$control_length" = 0 -o "$file_type" != "squashfs" -o "$kernel_magic" != "pass" ] && {
 107                         echo "Invalid sysupgrade file for $board_name"
 108                         return 1
 109                 }
 110                 ;;
 111         *)
 112                 echo "Unsupported device $board_name";
 113                 return 1
 114                 ;;
 115         esac
 116
 117         return 0
 118 }
 119
 120 merakinand_do_upgrade() {
 121         local tar_file="$1"
 122         local board_name="$(cat /tmp/sysinfo/board_name)"
 123
 124         # Do we need to do any platform tweaks?
 125         case "$board_name" in
 126         "mr18")
 127                 # Check and create UBI caldata if it's invalid
 128                 merakinand_copy_caldata "odm-caldata" "caldata"
 129                 nand_do_upgrade $1
 130                 ;;
 131         *)
 132                 echo "Unsupported device $board_name";
 133                 exit 1
 134                 ;;
 135         esac
 136 }