From 58e417ee41ec6ba4ab2f3ffdcf648d4e57dc781c Mon Sep 17 00:00:00 2001
From: Nitin Yadav <n-yadav@ti.com>
Date: Fri, 30 Jun 2023 13:55:04 +0530
Subject: [PATCH] board: ti: am62x: Setup UBIFS environment for Linux boot

Add UBIFS related environment variable to enable kernel
boot from UBIFS on GPMC NAND.

Signed-off-by: Nitin Yadav <n-yadav@ti.com>
---
 board/ti/am62x/am62x.env        |  1 +
 include/environment/ti/nand.env | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/board/ti/am62x/am62x.env b/board/ti/am62x/am62x.env
index fd3fd76788..7b5d506686 100644
--- a/board/ti/am62x/am62x.env
+++ b/board/ti/am62x/am62x.env
@@ -2,6 +2,7 @@
 #include <environment/ti/mmc.env>
 #include <environment/ti/ospi_nand.env>
 #include <environment/ti/k3_dfu.env>
+#include <environment/ti/nand.env>
 #if CONFIG_CMD_REMOTEPROC
 #include <environment/ti/k3_rproc.env>
 #endif
diff --git a/include/environment/ti/nand.env b/include/environment/ti/nand.env
index 9b32d55386..1908096c39 100644
--- a/include/environment/ti/nand.env
+++ b/include/environment/ti/nand.env
@@ -12,3 +12,21 @@ nandboot=echo Booting from nand ...;
 	nand read ${loadaddr} NAND.kernel;
 	bootz ${loadaddr} - ${fdtaddr}
 
+nbootpart=NAND.file-system
+nbootvolume=ubi0:rootfs
+ubi_init=ubi part ${nbootpart}; ubifsmount ${nbootvolume};
+args_nand=setenv bootargs console=${console}
+	${optargs} ubi.mtd=${nbootpart}
+	root=${nbootvolume} rootfstype=ubifs
+init_nand=run args_all args_nand ubi_init
+get_fdt_nand=ubifsload ${fdtaddr} ${bootdir}/${fdtfile};
+get_overlay_nand=
+	fdt address ${fdtaddr};
+	fdt resize 0x100000;
+	for overlay in $name_overlays;
+	do;
+	ubifsload ${dtboaddr} ${bootdir}/${overlay} &&
+	fdt apply ${dtboaddr};
+	done;
+get_kern_nand=ubifsload ${loadaddr} ${bootdir}/${name_kern}
+get_fit_nand=ubifsload ${addr_fit} ${bootdir}/${name_fit}