X-Git-Url: https://git.rohieb.name/openwrt.git/blobdiff_plain/8c1109a28dbb93dc858530d0efc81033ce9193f4..44a7438217323d90cc430627b82fd165fba52250:/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 38dbedbd4..1de907715 100755
--- 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