[PATCH 1/7] common: spl: mtd: Add support for loading images from MTD

[Anurag Dutta]

From: Apurva Nandan <a-nandan at ti.com>

Introduce support for using MTD subsystem for loading images from
memory flashes. This will provide the support of using mtd functions
which are abstracted over the type of flash being used, to load the
boot images from the flash.

Currently, add support for only SPI NAND flashes. This can be extended
to all other flash types, when required.

Signed-off-by: Apurva Nandan <a-nandan at ti.com>
Signed-off-by: Anurag Dutta <a-dutta at ti.com>
---
 common/spl/spl_mtd.c      | 90 +++++++++++++++++++++++++++++++++++++++
 common/spl/spl_mtd_nand.c | 31 ++++++++++++++
 include/spl.h             | 18 ++++++++
 3 files changed, 139 insertions(+)
 create mode 100644 common/spl/spl_mtd.c
 create mode 100644 common/spl/spl_mtd_nand.c

diff --git a/common/spl/spl_mtd.c b/common/spl/spl_mtd.c
new file mode 100644
index 00000000000..341e1a73392
--- /dev/null
+++ b/common/spl/spl_mtd.c
@@ -0,0 +1,90 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * FIT image loader functions using MTD read
+ *
+ * Copyright (C) 2026 Texas Instruments Incorporated - http://www.ti.com/
+ *	Apurva Nandan <a-nandan at ti.com>
+ */
+#include <config.h>
+#include <image.h>
+#include <log.h>
+#include <spl.h>
+#include <asm/io.h>
+#include <mtd.h>
+
+uint32_t __weak spl_mtd_get_uboot_offs(void)
+{
+	return CONFIG_SYS_MTD_U_BOOT_OFFS;
+}
+
+static ulong spl_mtd_fit_read(struct spl_load_info *load, ulong offs,
+			      ulong size, void *dst)
+{
+	struct mtd_info *mtd = load->priv;
+	int err;
+	size_t ret_len;
+
+	err = mtd_read(mtd, offs, size, &ret_len, dst);
+	if (!err)
+		return ret_len;
+
+	return 0;
+}
+
+struct mtd_info *spl_prepare_mtd(uint boot_device)
+{
+	struct mtd_info *mtd = NULL;
+
+	mtd_probe_devices();
+
+	switch (boot_device) {
+	case BOOT_DEVICE_SPINAND:
+		mtd = get_mtd_device_nm("spi-nand0");
+		break;
+	default:
+		puts(PHASE_PROMPT "Unsupported MTD Boot Device!\n");
+		break;
+	}
+
+	return mtd;
+}
+
+int spl_mtd_load(struct spl_image_info *spl_image,
+		 struct mtd_info *mtd, struct spl_boot_device *bootdev)
+{
+	int err;
+	struct legacy_img_hdr *header;
+	__maybe_unused struct spl_load_info load;
+	size_t ret_len;
+
+	header = spl_get_load_buffer(0, sizeof(*header));
+
+	err = mtd_read(mtd, spl_mtd_get_uboot_offs(), sizeof(*header),
+		       &ret_len, (void *)header);
+	if (err)
+		return err;
+
+	if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
+	    image_get_magic(header) == FDT_MAGIC) {
+		debug("Found FIT\n");
+		load.priv = mtd;
+		load.bl_len = 1;
+		load.read = spl_mtd_fit_read;
+		return spl_load_simple_fit(spl_image, &load,
+					   spl_mtd_get_uboot_offs(), header);
+	} else if (IS_ENABLED(CONFIG_SPL_LOAD_IMX_CONTAINER)) {
+		load.priv = mtd;
+		load.bl_len = 1;
+		load.read = spl_mtd_fit_read;
+		return spl_load_imx_container(spl_image, &load,
+					      spl_mtd_get_uboot_offs());
+	} else {
+		err = spl_parse_image_header(spl_image, bootdev, header);
+		if (err)
+			return err;
+		return mtd_read(mtd, spl_mtd_get_uboot_offs(), spl_image->size,
+				&ret_len, (void *)(ulong)spl_image->load_addr);
+	}
+
+	return -EINVAL;
+}
diff --git a/common/spl/spl_mtd_nand.c b/common/spl/spl_mtd_nand.c
new file mode 100644
index 00000000000..9bc6ff86ee8
--- /dev/null
+++ b/common/spl/spl_mtd_nand.c
@@ -0,0 +1,31 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * FIT image loader using MTD read
+ *
+ * Copyright (C) 2026 Texas Instruments Incorporated - http://www.ti.com/
+ *	Anurag Dutta <a-dutta at ti.com>
+ */
+#include <config.h>
+#include <image.h>
+#include <log.h>
+#include <spl.h>
+#include <asm/io.h>
+#include <mtd.h>
+
+static int spl_mtd_load_image(struct spl_image_info *spl_image,
+			      struct spl_boot_device *bootdev)
+{
+	struct mtd_info *mtd;
+	int ret;
+
+	mtd = spl_prepare_mtd(BOOT_DEVICE_SPINAND);
+	if (IS_ERR_OR_NULL(mtd)) {
+		printf("MTD device %s not found, ret %ld\n", "spi-nand",
+		       PTR_ERR(mtd));
+		return -EINVAL;
+	}
+
+	return spl_mtd_load(spl_image, mtd, bootdev);
+}
+
+SPL_LOAD_IMAGE_METHOD("SPINAND", 0, BOOT_DEVICE_SPINAND, spl_mtd_load_image);
diff --git a/include/spl.h b/include/spl.h
index 06dc28362d3..c4442be4fdb 100644
--- a/include/spl.h
+++ b/include/spl.h
@@ -1219,4 +1219,22 @@ int spl_reloc_prepare(struct spl_image_info *image, ulong *addrp);
  */
 int spl_reloc_jump(struct spl_image_info *image, spl_jump_to_image_t func);
 
+/* SPL MTD functions */
+/**
+ * spl_prepare_mtd() - Prepares the mtd for the corresponidng boot device
+ *
+ * @boot_device: A number indicating the BOOT_DEVICE type
+ */
+struct mtd_info *spl_prepare_mtd(uint boot_device);
+
+/**
+ * spl_mtd_load() - FIT Image loader
+ *
+ * @spl_image: SPL image to jump to
+ * @mtd: MTD device descriptor with geometry, flags, and read/write/erase function pointers
+ * @bootdev: boot device used by SPL
+ */
+int spl_mtd_load(struct spl_image_info *spl_image, struct mtd_info *mtd,
+		 struct spl_boot_device *bootdev);
+
 #endif
-- 
2.34.1