From 81b8d95f2c66a1a80039bdd5bab2c4bb795ca024 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafa=C5=82=20Mi=C5=82ecki?= <zajec5@gmail.com>
Date: Tue, 30 Dec 2014 17:19:36 +0000
Subject: [PATCH] brcm47xx: support for flashing CHK and CyberTAN images
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This will allow simple sysupgrade-s including simpler reverting to the
original firmware.

Signed-off-by: Rafał Miłecki <zajec5@gmail.com>

SVN-Revision: 43799
---
 .../base-files/lib/upgrade/platform.sh        | 33 ++++++++++++++++---
 1 file changed, 28 insertions(+), 5 deletions(-)

diff --git a/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh b/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
index 0a09da51a4..f96935bf59 100644
--- a/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
+++ b/target/linux/brcm47xx/base-files/lib/upgrade/platform.sh
@@ -49,8 +49,7 @@ platform_check_image() {
 				return 1
 			}
 
-			echo "Flashing CHK images in unsupported. Please use only .trx files."
-			return 1
+			return 0
 		;;
 		"cybertan")
 			magic=$(dd if="$1" bs=1 count=4 2>/dev/null | hexdump -v -e '1/1 "%c"')
@@ -62,8 +61,7 @@ platform_check_image() {
 				return 1
 			}
 
-			echo "Flashing CyberTAN images in unsupported. Please use only .trx files."
-			return 1
+			return 0
 		;;
 		"trx")
 			return 0
@@ -75,4 +73,29 @@ platform_check_image() {
 	esac
 }
 
-# use default for platform_do_upgrade()
+platform_do_upgrade_chk() {
+	local header_len=$((0x$(get_magic_long_at "$1" 4)))
+	local trx="/tmp/$1.trx"
+
+	dd if="$1" of="$trx" bs=$header_len skip=1
+	shift
+	default_do_upgrade "$trx" "$@"
+}
+
+platform_do_upgrade_cybertan() {
+	local trx="/tmp/$1.trx"
+
+	dd if="$1" of="$trx" bs=32 skip=1
+	shift
+	default_do_upgrade "$trx" "$@"
+}
+
+platform_do_upgrade() {
+	local file_type=$(brcm47xx_identify "$1")
+
+	case "$file_type" in
+		"chk")		platform_do_upgrade_chk "$ARGV";;
+		"cybertan")	platform_do_upgrade_cybertan "$ARGV";;
+		*)		default_do_upgrade "$ARGV";;
+	esac
+}