ar71xx: rewrite WNDR3700/3800 handling

[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 38dbedb..1de9077 100755 (executable)
--- a/target/linux/ar71xx/base-files/lib/upgrade/platform.sh
+++ b/target/linux/ar71xx/base-files/lib/upgrade/platform.sh
@@ -99,7 +99,9 @@ platform_check_image() {
        rocket-m | \
        wzr-hp-g300nh | \
        wzr-hp-ag300h | \
+       whr-g301n | \
        whr-hp-g300n | \
+       whr-hp-gn | \
        nbg460n_550n_550nh | \
        unifi )
                [ "$magic" != "2705" ] && {
@@ -125,15 +127,11 @@ platform_check_image() {
                return 0
                ;;
        wndr3700)
-               [ "$magic_long" != "33373030" ] && {
-                       echo "Invalid image type."
-                       return 1
-               }
-               return 0
-               ;;
-       wndr3700v2)
-               [ "$magic_long" != "33373031" ] && {
-                       echo "Invalid image type."
+               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