mediatek: mt7622: convert unifi6lr-v{1,2,3}-ubootmod to fitblk
[openwrt/staging/jow.git] / target / linux / mediatek / mt7622 / base-files / lib / upgrade / platform.sh
index 4e45c026ab30f179c181131c83fbbbf87fe49309..6417eb1a3a76d982c3b80a109f7a541aaffb9ace 100755 (executable)
@@ -1,43 +1,18 @@
-RAMFS_COPY_BIN='fw_printenv fw_setenv blockdev'
-RAMFS_COPY_DATA='/etc/fw_env.config /var/lock/fw_printenv.lock'
+REQUIRE_IMAGE_METADATA=1
+RAMFS_COPY_BIN='fitblk'
 
-get_cmdline_var() {
-       local var=$1
-       local cmdlinevar tmp
-       local cmdline="$(cat /proc/cmdline)"
-
-       for cmdlinevar in $cmdline; do
-               tmp=${cmdlinevar##${var}}
-               [ "=" = "${tmp:0:1}" ] && echo ${tmp:1}
-       done
-}
-
-get_rootdev() {
-       local rootvol rootdev
-       rootvol=$(get_cmdline_var root)
-       rootvol=$(basename $rootvol)
-       [ -e /sys/class/block/$rootvol ] || {
-               rootvol=${rootvol%%[0-9]}
-               [ -e /sys/class/block/$rootvol ] && echo $rootvol
-               rootvol=${rootvol%%p}
-               [ -e /sys/class/block/$rootvol ] && echo $rootvol
-               return
-       }
-       [ -e /sys/class/block/$rootvol/partition ] || {
-               echo $rootvol
-               return
-       }
-       rootdev=$(busybox readlink -f /sys/class/block/$rootvol)
-       rootdev=$(basename ${rootdev%%/${rootvol}})
-       [ -e /sys/class/block/$rootdev ] && echo $rootdev
-}
-
-get_partition() {
-       for partname in /sys/class/block/$1/*/name; do
-               [ "$(cat ${partname})" = "$2" ] && {
-                       basename ${partname%%/name}
+platform_get_bootdev() {
+       local rootdisk="$(cat /sys/firmware/devicetree/base/chosen/rootdisk)"
+       local handle bootdev
+       for handle in /sys/class/block/*/of_node/phandle /sys/class/block/*/device/of_node/phandle; do
+               [ ! -e "$handle" ] && continue
+               if [ "$rootdisk" = "$(cat $handle)" ]; then
+                       bootdev="${handle%/of_node/phandle}"
+                       bootdev="${bootdev%/device}"
+                       bootdev="${bootdev#/sys/class/block/}"
+                       echo "$bootdev"
                        break
-               }
+               fi
        done
 }
 
@@ -46,17 +21,52 @@ platform_do_upgrade() {
        local file_type=$(identify $1)
 
        case "$board" in
-       bananapi,bpi-r64)
-               local rootdev=$(get_rootdev)
-               local fitpart=$(get_partition $rootdev production)
-               [ "$fitpart" ] || exit 1
-               dd if=/dev/zero of=/dev/$fitpart bs=4096 count=1 2>/dev/null
-               blockdev --rereadpt /dev/$rootdev
-               get_image "$1" | dd of=/dev/$fitpart
-               echo $rootdev > /tmp/sysupgrade.rootdev
+       bananapi,bpi-r64|\
+       ubnt,unifi-6-lr-v1-ubootmod|\
+       ubnt,unifi-6-lr-v2-ubootmod|\
+       ubnt,unifi-6-lr-v3-ubootmod)
+               [ -e /dev/fit0 ] && fitblk /dev/fit0
+               [ -e /dev/fitrw ] && fitblk /dev/fitrw
+               bootdev="$(platform_get_bootdev)"
+               case "$bootdev" in
+               mmcblk*)
+                       EMMC_KERN_DEV="/dev/$bootdev"
+                       emmc_do_upgrade "$1"
+                       ;;
+               mtdblock*)
+                       PART_NAME="/dev/mtd${bootdev:8}"
+                       default_do_upgrade "$1"
+                       ;;
+               ubiblock*)
+                       CI_KERNPART="fit"
+                       nand_do_upgrade "$1"
+                       ;;
+               esac
                ;;
-       linksys,e8450-ubi|\
-       mediatek,mt7622,ubi)
+
+       buffalo,wsr-2533dhp2|\
+       buffalo,wsr-3200ax4s)
+               local magic="$(get_magic_long "$1")"
+
+               # use "mtd write" if the magic is "DHP2 (0x44485032)"
+               # or "DHP3 (0x44485033)"
+               if [ "$magic" = "44485032" -o "$magic" = "44485033" ]; then
+                       buffalo_upgrade_ubinized "$1"
+               else
+                       CI_KERNPART="firmware"
+                       nand_do_upgrade "$1"
+               fi
+               ;;
+       dlink,eagle-pro-ai-m32-a1|\
+       dlink,eagle-pro-ai-r32-a1|\
+       elecom,wrc-x3200gst3|\
+       mediatek,mt7622-rfb1-ubi|\
+       netgear,wax206|\
+       totolink,a8000ru|\
+       xiaomi,redmi-router-ax6s)
+               nand_do_upgrade "$1"
+               ;;
+       linksys,e8450-ubi)
                CI_KERNPART="fit"
                nand_do_upgrade "$1"
                ;;
@@ -83,6 +93,20 @@ platform_check_image() {
        [ "$#" -gt 1 ] && return 1
 
        case "$board" in
+       buffalo,wsr-2533dhp2|\
+       buffalo,wsr-3200ax4s)
+               buffalo_check_image "$board" "$magic" "$1" || return 1
+               ;;
+       dlink,eagle-pro-ai-m32-a1|\
+       dlink,eagle-pro-ai-r32-a1|\
+       elecom,wrc-x3200gst3|\
+       mediatek,mt7622-rfb1-ubi|\
+       netgear,wax206|\
+       totolink,a8000ru|\
+       xiaomi,redmi-router-ax6s)
+               nand_do_platform_check "$board" "$1"
+               return $?
+               ;;
        *)
                [ "$magic" != "d00dfeed" ] && {
                        echo "Invalid image type."
@@ -95,20 +119,12 @@ platform_check_image() {
        return 0
 }
 
-platform_copy_config_mmc() {
-       [ -e "$UPGRADE_BACKUP" ] || return
-       local rootdev=$(cat /tmp/sysupgrade.rootdev)
-       blockdev --rereadpt /dev/$rootdev
-       local datadev=$(get_partition $rootdev rootfs_data)
-       [ "$datadev" ] || echo "no rootfs_data partition, cannot keep configuration." >&2
-       dd if="$UPGRADE_BACKUP" of=/dev/$datadev
-       sync
-}
-
 platform_copy_config() {
        case "$(board_name)" in
        bananapi,bpi-r64)
-               platform_copy_config_mmc
+               if platform_get_bootdev | grep -q mmc; then
+                       emmc_copy_config
+               fi
                ;;
        esac
 }