From 54faeadb0769e6858367f28ea8d1ebf6e9259966 Mon Sep 17 00:00:00 2001 From: John Crispin Date: Thu, 19 Jun 2014 14:13:41 +0000 Subject: [PATCH] procd: various improvements to nand.sh Signed-off-by: John Crispin SVN-Revision: 41278 --- package/system/procd/files/nand.sh | 40 ++++++++++++++++++++++++------ 1 file changed, 33 insertions(+), 7 deletions(-) diff --git a/package/system/procd/files/nand.sh b/package/system/procd/files/nand.sh index 9131ddaa7b..09cc091d4f 100644 --- a/package/system/procd/files/nand.sh +++ b/package/system/procd/files/nand.sh @@ -97,9 +97,9 @@ nand_restore_config() { nand_upgrade_prepare_ubi() { local rootfs_length="$1" - local rootfs_type="$1" - local has_kernel="${2:-0}" - local has_env="${3:-0}" + local rootfs_type="$2" + local has_kernel="${3:-0}" + local has_env="${4:-0}" local mtdnum="$( find_mtd_index "$CI_UBIPART" )" if [ ! "$mtdnum" ]; then @@ -188,6 +188,11 @@ nand_upgrade_ubinized() { local ubi_file="$1" local mtdnum="$(find_mtd_index "$CI_UBIPART")" + [ ! "$mtdnum" ] && { + CI_UBIPART="rootfs" + mtdnum="$(find_mtd_index "$CI_UBIPART")" + } + if [ ! "$mtdnum" ]; then echo "cannot find mtd device $CI_UBIPART" reboot -f @@ -201,9 +206,19 @@ nand_upgrade_ubinized() { nand_do_upgrade_success } -nand_do_upgrade_stage2() { - [ "$(identify $1)" == "ubi" ] && nand_upgrade_ubinized $1 +nand_upgrade_ubifs() { + local rootfs_length=`(cat $1 | wc -c) 2> /dev/null` + nand_upgrade_prepare_ubi "$rootfs_length" "ubifs" "0" "0" + + local ubidev="$( nand_find_ubi "$CI_UBIPART" )" + local root_ubivol="$(nand_find_volume $ubidev rootfs)" + ubiupdatevol /dev/$root_ubivol -s $rootfs_length $1 + + nand_do_upgrade_success +} + +nand_upgrade_tar() { local tar_file="$1" local board_name="$(cat /tmp/sysinfo/board_name)" local kernel_mtd="$(find_mtd_index kernel)" @@ -237,6 +252,16 @@ nand_do_upgrade_stage2() { nand_do_upgrade_success } +nand_do_upgrade_stage2() { + local file_type=$(identify $1) + + [ ! "$(find_mtd_index "$CI_UBIPART")" ] && CI_UBIPART="rootfs" + + [ "$file_type" == "ubi" ] && nand_upgrade_ubinized $1 + [ "$file_type" == "ubifs" ] && nand_upgrade_ubifs $1 + nand_upgrade_tar $1 +} + nand_upgrade_stage2() { [ $1 = "nand" ] && { [ -f "$2" ] && { @@ -281,8 +306,9 @@ nand_do_platform_check() { local board_name="$1" local tar_file="$2" local control_length=`(tar xf $tar_file sysupgrade-$board_name/CONTROL -O | wc -c) 2> /dev/null` - - [ "$control_length" = 0 -a "$(identify $2)" != "ubi" ] && { + local file_type="$(identify $2)" + + [ "$control_length" = 0 -a "$file_type" != "ubi" -a "$file_type" != "ubifs" ] && { echo "Invalid sysupgrade file." return 1 }