1 files changed, 94 insertions, 0 deletions
diff --git a/common/spl/spl_mtd.c b/common/spl/spl_mtd.c
new file mode 100644
index 0000000000..8af3329f91
--- /dev/null
+++ b/common/spl/spl_mtd.c
@@ -0,0 +1,94 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * FIT image loader using MTD read
+ *
+ * Copyright (C) 2018 Texas Instruments Incorporated - http://www.ti.com/
+ *	Apurva Nandan <a-nandan@ti.com>
+ */
+#include <common.h>
+#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->dev;
+	int err;
+	size_t ret_len;
+
+	err = mtd_read(mtd, offs, size, &ret_len, dst);
+	if (!err)
+		return ret_len;
+
+	return 0;
+}
+
+static int spl_mtd_load_image(struct spl_image_info *spl_image,
+			      struct spl_boot_device *bootdev)
+{
+	int err;
+	struct legacy_img_hdr *header;
+	__maybe_unused struct spl_load_info load;
+	struct mtd_info *mtd;
+	size_t ret_len;
+
+	mtd_probe_devices();
+
+	switch (bootdev->boot_device) {
+	case BOOT_DEVICE_SPINAND:
+		mtd = get_mtd_device_nm("spi-nand0");
+		if (IS_ERR_OR_NULL(mtd))
+			printf("MTD device %s not found, ret %ld\n", "spi-nand",
+			       PTR_ERR(mtd));
+		break;
+	default:
+		puts(SPL_TPL_PROMPT "Unsupported MTD Boot Device!\n");
+		return -EINVAL;
+	}
+
+	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.dev = mtd;
+		load.priv = NULL;
+		load.filename = NULL;
+		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.dev = mtd;
+		load.priv = NULL;
+		load.filename = NULL;
+		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;
+}
+
+SPL_LOAD_IMAGE_METHOD("SPINAND", 0, BOOT_DEVICE_SPINAND, spl_mtd_load_image);