From 1e6897f6304f8a53cbc36a040d63bbb5849b3017 Mon Sep 17 00:00:00 2001
From: Felix Fietkau <nbd@nbd.name>
Date: Mon, 10 Jan 2022 12:35:08 +0100
Subject: [PATCH] mediatek: rework and fix mt7622-rfb1-ubi support

Limit bmt remapping range to cover everything up to and including the kernel image,
use the rest of the flash area for ubi.
Fix partition table and sysupgrade support

Signed-off-by: Felix Fietkau <nbd@nbd.name>
---
 target/linux/mediatek/dts/mt7622-rfb1-ubi.dts  | 18 +++++++++---------
 target/linux/mediatek/image/mt7622.mk          |  1 +
 .../mt7622/base-files/etc/board.d/02_network   |  3 ++-
 .../mt7622/base-files/lib/upgrade/platform.sh  |  5 +++--
 4 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/target/linux/mediatek/dts/mt7622-rfb1-ubi.dts b/target/linux/mediatek/dts/mt7622-rfb1-ubi.dts
index 3816dcb9cc..1a94446c7a 100644
--- a/target/linux/mediatek/dts/mt7622-rfb1-ubi.dts
+++ b/target/linux/mediatek/dts/mt7622-rfb1-ubi.dts
@@ -3,10 +3,13 @@
 #include "mt7622-rfb1.dts"
 / {
 	model = "MT7622_MT7531 RFB (UBI)";
-	compatible = "mediatek,mt7622,ubi";
+	compatible = "mediatek,mt7622-rfb1-ubi";
 };
 
 &snand {
+	mediatek,bmt-v2;
+	mediatek,bmt-remap-range = <0x0 0x6c0000>;
+
 	partitions {
 		compatible = "fixed-partitions";
 		#address-cells = <1>;
@@ -35,22 +38,19 @@
 
 		factory: partition@1c0000 {
 			label = "Factory";
-			reg = <0x1c0000 0x0040000>;
+			reg = <0x1c0000 0x0100000>;
 		};
 
 		partition@200000 {
 			label = "kernel";
-			reg = <0x200000 0x400000>;
+			reg = <0x2c0000 0x400000>;
 		};
 
-		partition@600000 {
+		partition@6c0000 {
 			label = "ubi";
-			reg = <0x600000 0x1C00000>;
+			reg = <0x6c0000 0x6f00000>;
 		};
 
-		partition@2200000 {
-			label = "User_data";
-			reg = <0x2200000 0x4000000>;
-		};
+		/delete-node/ partition@2200000;
 	};
 };
diff --git a/target/linux/mediatek/image/mt7622.mk b/target/linux/mediatek/image/mt7622.mk
index 795bb3fc76..3b0afa98df 100644
--- a/target/linux/mediatek/image/mt7622.mk
+++ b/target/linux/mediatek/image/mt7622.mk
@@ -190,6 +190,7 @@ define Device/mediatek_mt7622-rfb1-ubi
   DEVICE_DTS := mt7622-rfb1-ubi
   DEVICE_DTS_DIR := ../dts
   DEVICE_PACKAGES := kmod-ata-ahci-mtk kmod-btmtkuart kmod-usb3
+  BOARD_NAME := mediatek,mt7622-rfb1-ubi
   UBINIZE_OPTS := -E 5
   BLOCKSIZE := 128k
   PAGESIZE := 2048
diff --git a/target/linux/mediatek/mt7622/base-files/etc/board.d/02_network b/target/linux/mediatek/mt7622/base-files/etc/board.d/02_network
index 81e5c7c650..7abbaed328 100755
--- a/target/linux/mediatek/mt7622/base-files/etc/board.d/02_network
+++ b/target/linux/mediatek/mt7622/base-files/etc/board.d/02_network
@@ -12,7 +12,8 @@ mediatek_setup_interfaces()
 	bananapi,bpi-r64|\
 	linksys,e8450|\
 	linksys,e8450-ubi|\
-	mediatek,mt7622-rfb1)
+	mediatek,mt7622-rfb1|\
+	mediatek,mt7622-rfb1-ubi)
 		ucidef_set_interfaces_lan_wan "lan1 lan2 lan3 lan4" wan
 		;;
 	buffalo,wsr-2533dhp2)
diff --git a/target/linux/mediatek/mt7622/base-files/lib/upgrade/platform.sh b/target/linux/mediatek/mt7622/base-files/lib/upgrade/platform.sh
index ef8dbaaf81..1d480c73ac 100755
--- a/target/linux/mediatek/mt7622/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/mt7622/base-files/lib/upgrade/platform.sh
@@ -42,8 +42,7 @@ platform_do_upgrade() {
 			nand_do_upgrade "$1"
 		fi
 		;;
-	linksys,e8450-ubi|\
-	mediatek,mt7622,ubi)
+	linksys,e8450-ubi)
 		CI_KERNPART="fit"
 		nand_do_upgrade "$1"
 		;;
@@ -55,6 +54,7 @@ platform_do_upgrade() {
 		fi
 		default_do_upgrade "$1"
 		;;
+	mediatek,mt7622-rfb1-ubi|\
 	totolink,a8000ru)
 		nand_do_upgrade "$1"
 		;;
@@ -76,6 +76,7 @@ platform_check_image() {
 	buffalo,wsr-2533dhp2)
 		buffalo_check_image "$board" "$magic" "$1" || return 1
 		;;
+	mediatek,mt7622-rfb1-ubi|\
 	totolink,a8000ru)
 		nand_do_platform_check "$board" "$1"
 		;;