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