OpenWrt – Rev 1

Subversion Repositories:
Rev:
#!/bin/sh
#
# Copyright (C) 2015-2016 Chris Blake <chrisrblake93@gmail.com>
#
# Custom upgrade script for Meraki NAND devices (ex. MR18)
# Based on dir825.sh and stock nand functions
#
. /lib/ar71xx.sh
. /lib/functions.sh

get_magic_at() {
        local mtddev=$1
        local pos=$2
        dd bs=1 count=2 skip=$pos if=$mtddev 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
}

meraki_is_caldata_valid() {
        local board=$1
        local mtddev=$2
        local magic

        case "$board" in
        "mr18")
                magic=$(get_magic_at $mtddev 4096)
                [ "$magic" != "0202" ] && return 0

                magic=$(get_magic_at $mtddev 20480)
                [ "$magic" != "0202" ] && return 0

                magic=$(get_magic_at $mtddev 36864)
                [ "$magic" != "0202" ] && return 0

                return 1
                ;;
        "z1")
                magic=$(get_magic_at $mtddev 4096)
                [ "$magic" != "0202" ] && return 0

                magic=$(get_magic_at $mtddev 86016)
                [ "$magic" != "a55a" ] && return 0

                return 1
                ;;
        *)
                return 1
                ;;
        esac
}

merakinand_copy_caldata() {
        local cal_src=$1
        local cal_dst=$2
        local ubidev="$(nand_find_ubi $CI_UBIPART)"
        local board_name="$(board_name)"
        local rootfs_size="$(ubinfo /dev/ubi0 -N rootfs_data | grep "Size" | awk '{ print $6 }')"

        # Setup partitions using board name, in case of future platforms
        case "$board_name" in
        "mr18"|\
        "z1")
                # Src is MTD
                mtd_src="$(find_mtd_chardev $cal_src)"
                [ -n "$mtd_src" ] || {
                        echo "no mtd device found for partition $cal_src"
                        exit 1
                }

                # Dest is UBI
                # TODO: possibly add create (hard to do when rootfs_data is expanded & mounted)
                # Would need to be done from ramdisk
                mtd_dst="$(nand_find_volume $ubidev $cal_dst)"
                [ -n "$mtd_dst" ] || {
                        echo "no ubi device found for partition $cal_dst"
                        exit 1
                }

                meraki_is_caldata_valid "$board_name" "$mtd_src" && {
                        echo "no valid calibration data found in $cal_src"
                        exit 1
                }

                meraki_is_caldata_valid "$board_name" "/dev/$mtd_dst" && {
                        echo "Copying calibration data from $cal_src to $cal_dst..."
                        dd if="$mtd_src" of=/tmp/caldata.tmp 2>/dev/null
                        ubiupdatevol "/dev/$mtd_dst" /tmp/caldata.tmp
                        rm /tmp/caldata.tmp
                        sync
                }
                return 0
                ;;
        *)
                echo "Unsupported device $board_name";
                return 1
                ;;
        esac
}

merakinand_do_kernel_check() {
        local board_name="$1"
        local tar_file="$2"
        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"')`

        # What is our kernel magic string?
        case "$board_name" in
        "mr18")
                [ "$image_magic_word" == "8e73ed8a" ] && {
                        echo "pass" && return 0
                }
                ;;
        "z1")
                [ "$image_magic_word" == "4d495053" ] && {
                        echo "pass" && return 0
                }
                ;;
        esac

        exit 1
}

merakinand_do_platform_check() {
        local board_name="$1"
        local tar_file="$2"
        local control_length=`(tar xf $tar_file sysupgrade-$board_name/CONTROL -O | wc -c) 2> /dev/null`
        local file_type="$(identify_tar $2 sysupgrade-$board_name/root)"
        local kernel_magic="$(merakinand_do_kernel_check $1 $2)"

        case "$board_name" in
        "mr18"|\
        "z1")
                [ "$control_length" = 0 -o "$file_type" != "squashfs" -o "$kernel_magic" != "pass" ] && {
                        echo "Invalid sysupgrade file for $board_name"
                        return 1
                }
                ;;
        *)
                echo "Unsupported device $board_name";
                return 1
                ;;
        esac

        return 0
}

merakinand_do_upgrade() {
        local tar_file="$1"
        local board_name="$(board_name)"

        # Do we need to do any platform tweaks?
        case "$board_name" in
        "mr18")
                # Check and create UBI caldata if it's invalid
                merakinand_copy_caldata "odm-caldata" "caldata"
                nand_do_upgrade $1
                ;;
        "z1")
                # Check and create UBI caldata if it's invalid
                merakinand_copy_caldata "origcaldata" "caldata"
                nand_do_upgrade $1
                ;;
        *)
                echo "Unsupported device $board_name";
                exit 1
                ;;
        esac
}