ar71xx: verify hardware id of firmware images on TP-Link 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 089f5f4..d2780a5 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
 #
 
 . /lib/ar71xx.sh


@@ -61,49 +61,112 @@ platform_do_upgrade_combined() {
        fi
 }
 
        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")"
 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
 
        [ "$ARGC" -gt 1 ] && return 1
 
        case "$board" in

-       ap81 | ap83 | dir-600-a1 | dir-615-c1 | dir-825-b1 | mzk-w04nu | mzk-w300nh | tew-632brp | wrt400n | bullet-m | nanostation-m | rocket-m | wzr-hp-g300nh )
-               [ "$magic" != "2705" ] && {
+       all0258n )
+               platform_check_image_all0258n "$1" && return 0
+               return 1
+               ;;
+       ap121 | \
+       ap121-mini | \
+       ap96 | \
+       db120 | \
+       zcn-1523h-2 | \
+       zcn-1523h-5)
+               [ "$magic_long" != "68737173" -a "$magic_long" != "19852003" ] && {

                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;
                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;

-       tl-mr3420 | tl-wa901nd | tl-wr741nd | tl-wr841n-v1 | tl-wr941nd | tl-wr1043nd)
-               [ "$magic" != "0100" ] && {
+       ap81 | \
+       ap83 | \
+       dir-600-a1 | \
+       dir-615-c1 | \
+       dir-825-b1 | \
+       mzk-w04nu | \
+       mzk-w300nh | \
+       tew-632brp | \
+       wrt400n | \
+       bullet-m | \
+       nanostation-m | \
+       rocket-m | \
+       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
                ;;
                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;

-       wndr3700)
-               [ "$magic" != "3337" ] && {
+       tl-mr3220 | \
+       tl-mr3420 | \
+       tl-wa901nd | \
+       tl-wa901nd-v2 | \
+       tl-wr703n | \
+       tl-wr741nd | \
+       tl-wr741nd-v4 | \
+       tl-wr841n-v1 | \
+       tl-wr941nd | \
+       tl-wr1043nd)
+               [ "$magic" != "0100" ] && {

                        echo "Invalid image type."
                        return 1
                }
                        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
                ;;
                return 0
                ;;

-       wrt160nl)
-               [ "$magic" != "4e4c" ] && {
-                       echo "Invalid image type."
+       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
                ;;
                        return 1
                }
                return 0
                ;;

-       zcn-1523h-2 | zcn-1523h-5)
-               [ "$magic" != "6873" -a "$magic" != "1985" ] && {
+       wrt160nl)
+               [ "$magic" != "4e4c" ] && {

                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;
                        echo "Invalid image type."
                        return 1
                }
                return 0
                ;;

-       routerstation | routerstation-pro | ls-sr71 | pb42 | pb44 | eap7660d )
+       routerstation | \
+       routerstation-pro | \
+       ls-sr71 | \
+       pb42 | \
+       pb44 | \
+       eap7660d | \
+       ja76pf )

                [ "$magic" != "4349" ] && {
                        echo "Invalid image. Use *-sysupgrade.bin files on this board"
                        return 1
                [ "$magic" != "4349" ] && {
                        echo "Invalid image. Use *-sysupgrade.bin files on this board"
                        return 1


@@ -130,9 +193,16 @@ platform_do_upgrade() {
        local board=$(ar71xx_board_name)
 
        case "$board" in
        local board=$(ar71xx_board_name)
 
        case "$board" in

-       routerstation | routerstation-pro | ls-sr71 | eap7660d )
+       routerstation | \
+       routerstation-pro | \
+       ls-sr71 | \
+       eap7660d | \
+       ja76pf)

                platform_do_upgrade_combined "$ARGV"
                ;;
                platform_do_upgrade_combined "$ARGV"
                ;;

+       all0258n )
+               platform_do_upgrade_all0258n "$ARGV"
+               ;;

        *)
                default_do_upgrade "$ARGV"
                ;;
        *)
                default_do_upgrade "$ARGV"
                ;;