ar71xx: verify hardware id of firmware images on TP-Link boards

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

diff --git a/target/linux/ar71xx/base-files/lib/ar71xx.sh b/target/linux/ar71xx/base-files/lib/ar71xx.sh
index 02286e9..ca0799d 100755 (executable)
--- a/target/linux/ar71xx/base-files/lib/ar71xx.sh
+++ b/target/linux/ar71xx/base-files/lib/ar71xx.sh
@@ -6,6 +6,58 @@
 AR71XX_BOARD_NAME=
 AR71XX_MODEL=
 
+ar71xx_get_mem_total() {
+       $(awk '/MemTotal:/ {print($2)}' /proc/meminfo)
+}
+
+ar71xx_get_mtd_part_magic() {
+       local part="$1"
+       local mtd
+
+       mtd=$(find_mtd_part $part)
+       [ -z "$mtd" ] && return
+
+       dd if=$mtd bs=4 count=1 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
+}
+
+wndr3700_board_detect() {
+       local machine="$1"
+       local magic
+       local name
+
+       name="wndr3700"
+
+       magic="$(ar71xx_get_mtd_part_magic firmware)"
+       case $magic in
+       "33373030")
+               machine="NETGEAR WNDR3700"
+               ;;
+       "33373031")
+               local mt
+
+               mt=$(ar71xx_get_mem_total)
+               if [ "$mt" -lt "65536" ]; then
+                       machine="NETGEAR WNDR3700v2"
+               else
+                       machine="NETGEAR WNDR3800"
+               fi
+               ;;
+       esac
+
+       AR71XX_BOARD_NAME="$name"
+       AR71XX_MODEL="$machine"
+}
+
+tplink_get_hwid() {
+       local part
+       local hwid
+
+       part=$(find_mtd_part firmware)
+       [ -z "$part" ] && return 1
+
+       dd if=$part bs=4 count=1 skip=16 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
+}
+
 ar71xx_board_detect() {
        local machine
        local name
@@ -169,14 +221,8 @@ ar71xx_board_detect() {
        *WP543)
                name="wp543"
                ;;
-       *WNDR3700)
-               name="wndr3700"
-               ;;
-       *WNDR3700v2)
-               name="wndr3700v2"
-               ;;
-       *WNDR3800)
-               name="wndr3800"
+       *"WNDR3700/WNDR3800")
+               wndr3700_board_detect "$machine"
                ;;
        *WNR2000)
                name="wnr2000"
@@ -193,6 +239,9 @@ ar71xx_board_detect() {
        *WZR-HP-G300NH)
                name="wzr-hp-g300nh"
                ;;
+       *WZR-HP-G450H)
+               name="wzr-hp-g450h"
+               ;;
        *WHR-HP-G300N)
                name="whr-hp-g300n"
                ;;