ar71xx: fix sysupgrade on the TL-WR841N/ND v7 boards

[openwrt.git] / target / linux / ar71xx / base-files / lib / upgrade / platform.sh

diff --git a/target/linux/ar71xx/base-files/lib/upgrade/platform.sh b/target/linux/ar71xx/base-files/lib/upgrade/platform.sh
index ef84e84..db8bdae 100755 (executable)
--- a/target/linux/ar71xx/base-files/lib/upgrade/platform.sh
+++ b/target/linux/ar71xx/base-files/lib/upgrade/platform.sh
@@ -1,5 +1,5 @@
 #
-# Copyright (C) 2009 OpenWrt.org
+# Copyright (C) 2011 OpenWrt.org
 #
 
 . /lib/ar71xx.sh
@@ -15,7 +15,7 @@ platform_find_partitions() {
        while read dev size erasesize name; do
                name=${name#'"'}; name=${name%'"'}
                case "$name" in
-                       vmlinux.bin.l7|kernel|linux|rootfs)
+                       vmlinux.bin.l7|vmlinux|kernel|linux|rootfs)
                                if [ -z "$first" ]; then
                                        first="$name"
                                else
@@ -31,7 +31,7 @@ platform_find_kernelpart() {
        local part
        for part in "${1%:*}" "${1#*:}"; do
                case "$part" in
-                       vmlinux.bin.l7|kernel|linux)
+                       vmlinux.bin.l7|vmlinux|kernel|linux)
                                echo "$part"
                                break
                        ;;
@@ -61,25 +61,104 @@ platform_do_upgrade_combined() {
        fi
 }
 
+tplink_get_image_hwid() {
+       get_image "$@" | dd bs=4 count=1 skip=16 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
+}
+
 platform_check_image() {
        local board=$(ar71xx_board_name)
        local magic="$(get_magic_word "$1")"
+       local magic_long="$(get_magic_long "$1")"
 
        [ "$ARGC" -gt 1 ] && return 1
 
        case "$board" in
-       ap83 | mzk-w04nu | mzk-w300nh | tew-632brp | wrt-400n)
+       all0258n )
+               platform_check_image_all0258n "$1" && return 0
+               return 1
+               ;;
+       alfa-ap96 | \
+       alfa-nx | \
+       ap121 | \
+       ap121-mini | \
+       ap96 | \
+       db120 | \
+       hornet-ub | \
+       zcn-1523h-2 | \
+       zcn-1523h-5)
+               [ "$magic_long" != "68737173" -a "$magic_long" != "19852003" ] && {
+                       echo "Invalid image type."
+                       return 1
+               }
+               return 0
+               ;;
+       ap81 | \
+       ap83 | \
+       dir-600-a1 | \
+       dir-615-c1 | \
+       dir-825-b1 | \
+       mzk-w04nu | \
+       mzk-w300nh | \
+       tew-632brp | \
+       wrt400n | \
+       airrouter | \
+       bullet-m | \
+       nanostation-m | \
+       rocket-m | \
+       rw2458n | \
+       wzr-hp-g300nh2 | \
+       wzr-hp-g300nh | \
+       wzr-hp-g450h | \
+       wzr-hp-ag300h | \
+       whr-g301n | \
+       whr-hp-g300n | \
+       whr-hp-gn | \
+       nbg460n_550n_550nh | \
+       unifi )
                [ "$magic" != "2705" ] && {
                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;
-       tl-wr741nd | tl-wr941nd)
+       tl-mr3020 | \
+       tl-mr3220 | \
+       tl-mr3420 | \
+       tl-wa901nd | \
+       tl-wa901nd-v2 | \
+       tl-wr703n | \
+       tl-wr741nd | \
+       tl-wr741nd-v4 | \
+       tl-wr841n-v1 | \
+       tl-wr841n-v7 | \
+       tl-wr941nd | \
+       tl-wr1043nd)
                [ "$magic" != "0100" ] && {
                        echo "Invalid image type."
                        return 1
                }
+
+               local hwid
+               local imageid
+
+               hwid=$(tplink_get_hwid)
+               imageid=$(tplink_get_image_hwid "$1")
+
+               [ "$hwid" != "$imageid" ] && {
+                       echo "Invalid image, hardware ID mismatch, hw:$hwid image:$imageid."
+                       return 1
+               }
+
+               return 0
+               ;;
+       wndr3700)
+               local hw_magic
+
+               hw_magic="$(ar71xx_get_mtd_part_magic firmware)"
+               [ "$magic_long" != "$hw_magic" ] && {
+                       echo "Invalid image, hardware ID mismatch, hw:$hw_magic image:$magic_long."
+                       return 1
+               }
                return 0
                ;;
        wrt160nl)
@@ -89,7 +168,13 @@ platform_check_image() {
                }
                return 0
                ;;
-       routerstation | routerstation-pro | ls-sr71)
+       routerstation | \
+       routerstation-pro | \
+       ls-sr71 | \
+       pb42 | \
+       pb44 | \
+       eap7660d | \
+       ja76pf )
                [ "$magic" != "4349" ] && {
                        echo "Invalid image. Use *-sysupgrade.bin files on this board"
                        return 1
@@ -116,9 +201,18 @@ platform_do_upgrade() {
        local board=$(ar71xx_board_name)
 
        case "$board" in
-       routerstation | routerstation-pro)
+       routerstation | \
+       routerstation-pro | \
+       ls-sr71 | \
+       eap7660d | \
+       pb42 | \
+       pb44 | \
+       ja76pf)
                platform_do_upgrade_combined "$ARGV"
                ;;
+       all0258n )
+               platform_do_upgrade_all0258n "$ARGV"
+               ;;
        *)
                default_do_upgrade "$ARGV"
                ;;