brcm47xx: use separated function in upgrade to detect file type

[openwrt/staging/lynxis.git] / target / linux / brcm47xx / base-files / lib / upgrade / platform.sh

diff --git a/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh b/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
index 9921cf571a2796504a46defdd6647fd43de1f74c..d216ca4d04e3f7dd5ff7d5a32d2e6c0cbc60333e 100644 (file)
--- a/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
+++ b/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
@@ -1,11 +1,30 @@
-PART_NAME=linux
+PART_NAME=firmware
+
+brcm47xx_identify() {
+       local magic
+
+       magic=$(get_magic_long "$1")
+       case "$magic" in
+               "48445230")
+                       echo "trx"
+                       return
+                       ;;
+               "2a23245e")
+                       echo "chk"
+                       return
+                       ;;
+       esac
+
+       echo "unknown"
+}
 
 platform_check_image() {
-       [ "$ARGC" -gt 1 ] && return 1
+       [ "$#" -gt 1 ] && return 1
+
+       local file_type=$(brcm47xx_identify "$1")
 
-       case "$(get_magic_word "$1")" in
-               # .trx files
-               4844) return 0;;
+       case "$file_type" in
+               "trx") return 0;;
                *)
                        echo "Invalid image type. Please use only .trx files"
                        return 1