From: Apurva Nandan <[email protected]>
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 <[email protected]>
Signed-off-by: Anurag Dutta <[email protected]>
---
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 <[email protected]>
+ */
+#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);
+ }
