From 735ea0d11ed91fc310730b8c76430f22b6cc5371 Mon Sep 17 00:00:00 2001
From: Tony91590 <91764010+Tony91590@users.noreply.github.com>
Date: Fri, 16 Feb 2024 15:40:15 +0100
Subject: [PATCH] mediatek: filogic: set special upgrade for Asus AX4200
 (#11893)

---
 .../base-files/lib/upgrade/platform.sh        | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh b/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
index 0a6f892a1..6642318f8 100755
--- a/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
@@ -1,5 +1,16 @@
 REQUIRE_IMAGE_METADATA=1
 
+asus_initial_setup()
+{
+	# initialize UBI if it's running on initramfs
+	[ "$(rootfs_type)" = "tmpfs" ] || return 0
+
+	ubirmvol /dev/ubi0 -N rootfs
+	ubirmvol /dev/ubi0 -N rootfs_data
+	ubirmvol /dev/ubi0 -N jffs2
+	ubimkvol /dev/ubi0 -N jffs2 -s 0x3e000
+}
+
 platform_do_upgrade() {
 	local board=$(board_name)
 
@@ -84,4 +95,14 @@ platform_copy_config() {
 		emmc_copy_config
 		;;
 	esac
+ }
+ 
+platform_pre_upgrade() {
+	local board=$(board_name)
+
+	case "$board" in
+	asus,tuf-ax4200)
+		asus_initial_setup
+		;;
+	esac
 }