i.MX8MN support loading images with rom api, so we implement
reuse board_return_to_bootrom to let ROM loading images.

Signed-off-by: Peng Fan <peng....@nxp.com>
---
 arch/arm/include/asm/mach-imx/sys_proto.h |  33 ++++
 arch/arm/mach-imx/Kconfig                 |   4 +
 arch/arm/mach-imx/Makefile                |   2 +
 arch/arm/mach-imx/spl_imx_romapi.c        | 292 ++++++++++++++++++++++++++++++
 4 files changed, 331 insertions(+)
 create mode 100644 arch/arm/mach-imx/spl_imx_romapi.c

diff --git a/arch/arm/include/asm/mach-imx/sys_proto.h 
b/arch/arm/include/asm/mach-imx/sys_proto.h
index dde39c63dd..6ceffb676d 100644
--- a/arch/arm/include/asm/mach-imx/sys_proto.h
+++ b/arch/arm/include/asm/mach-imx/sys_proto.h
@@ -110,6 +110,39 @@ void gpr_init(void);
 
 #endif /* CONFIG_MX6 */
 
+#ifdef CONFIG_IMX8M
+struct rom_api {
+       u16 ver;
+       u16 tag;
+       u32 reserved1;
+       u32 (*download_image)(u8 *dest, u32 offset, u32 size,  u32 xor);
+       u32 (*query_boot_infor)(u32 info_type, u32 *info, u32 xor);
+};
+
+enum boot_dev_type_e {
+       BT_DEV_TYPE_SD = 1,
+       BT_DEV_TYPE_MMC = 2,
+       BT_DEV_TYPE_NAND = 3,
+       BT_DEV_TYPE_FLEXSPINOR = 4,
+
+       BT_DEV_TYPE_USB = 0xE,
+       BT_DEV_TYPE_MEM_DEV = 0xF,
+
+       BT_DEV_TYPE_INVALID = 0xFF
+};
+
+#define QUERY_ROM_VER          1
+#define QUERY_BT_DEV           2
+#define QUERY_PAGE_SZ          3
+#define QUERY_IVT_OFF          4
+#define QUERY_BT_STAGE         5
+#define QUERY_IMG_OFF          6
+
+#define ROM_API_OKAY           0xF0
+
+extern struct rom_api *g_rom_api;
+#endif
+
 u32 get_nr_cpus(void);
 u32 get_cpu_rev(void);
 u32 get_cpu_speed_grade_hz(void);
diff --git a/arch/arm/mach-imx/Kconfig b/arch/arm/mach-imx/Kconfig
index aeb5493488..fb1707d3d7 100644
--- a/arch/arm/mach-imx/Kconfig
+++ b/arch/arm/mach-imx/Kconfig
@@ -101,3 +101,7 @@ config DDRMC_VF610_CALIBRATION
          NXP does NOT recommend to perform this calibration at each boot. One
          shall perform it on a new PCB and then use those values to program
          the ddrmc_cr_setting on relevant board file.
+
+config SPL_IMX_ROMAPI_LOADADDR
+       hex "Default load address to load image through ROM API"
+       depends on SPL_BOOTROM_SUPPORT
diff --git a/arch/arm/mach-imx/Makefile b/arch/arm/mach-imx/Makefile
index 17b241ae99..199b2df22b 100644
--- a/arch/arm/mach-imx/Makefile
+++ b/arch/arm/mach-imx/Makefile
@@ -218,3 +218,5 @@ obj-$(CONFIG_MX7) += mx7/
 obj-$(CONFIG_ARCH_MX7ULP) += mx7ulp/
 obj-$(CONFIG_IMX8M) += imx8m/
 obj-$(CONFIG_ARCH_IMX8) += imx8/
+
+obj-$(CONFIG_SPL_BOOTROM_SUPPORT) += spl_imx_romapi.o
diff --git a/arch/arm/mach-imx/spl_imx_romapi.c 
b/arch/arm/mach-imx/spl_imx_romapi.c
new file mode 100644
index 0000000000..5dc0f7174e
--- /dev/null
+++ b/arch/arm/mach-imx/spl_imx_romapi.c
@@ -0,0 +1,292 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Copyright 2019 NXP
+ */
+
+#include <common.h>
+#include <errno.h>
+#include <image.h>
+#include <linux/libfdt.h>
+#include <spl.h>
+
+#include <asm/arch/sys_proto.h>
+
+DECLARE_GLOBAL_DATA_PTR;
+
+static int is_boot_from_stream_device(u32 boot)
+{
+       u32 interface;
+
+       interface = boot >> 16;
+       if (interface >= BT_DEV_TYPE_USB)
+               return 1;
+
+       if (interface == BT_DEV_TYPE_MMC && (boot & 1))
+               return 1;
+
+       return 0;
+}
+
+static ulong spl_romapi_read_seekable(struct spl_load_info *load,
+                                     ulong sector, ulong count,
+                                     void *buf)
+{
+       u32 pagesize = *(u32 *)load->priv;
+       volatile gd_t *pgd = gd;
+       ulong byte = count * pagesize;
+       int ret;
+       u32 offset;
+
+       offset = sector * pagesize;
+
+       debug("ROM API load from 0x%x, size 0x%x\n", offset, (u32)byte);
+
+       ret = g_rom_api->download_image(buf, offset, byte,
+                                       ((uintptr_t)buf) ^ offset ^ byte);
+       gd = pgd;
+
+       if (ret == ROM_API_OKAY)
+               return count;
+
+       printf("ROM API Failure when load 0x%x\n", offset);
+
+       return 0;
+}
+
+static int spl_romapi_load_image_seekable(struct spl_image_info *spl_image,
+                                         struct spl_boot_device *bootdev,
+                                         u32 rom_bt_dev)
+{
+       volatile gd_t *pgd = gd;
+       int ret;
+       u32 offset;
+       u32 pagesize, size;
+       struct image_header *header;
+       u32 image_offset;
+
+       ret = g_rom_api->query_boot_infor(QUERY_IVT_OFF, &offset,
+                                         ((uintptr_t)&offset) ^ QUERY_IVT_OFF);
+       ret |= g_rom_api->query_boot_infor(QUERY_PAGE_SZ, &pagesize,
+                                          ((uintptr_t)&pagesize) ^ 
QUERY_PAGE_SZ);
+       ret |= g_rom_api->query_boot_infor(QUERY_IMG_OFF, &image_offset,
+                                          ((uintptr_t)&image_offset) ^ 
QUERY_IMG_OFF);
+
+       gd = pgd;
+
+       if (ret != ROM_API_OKAY) {
+               puts("ROMAPI: Failure query boot infor pagesize/offset\n");
+               return -1;
+       }
+
+       header = (struct image_header *)(CONFIG_SPL_IMX_ROMAPI_LOADADDR);
+
+       printf("image offset 0x%x, pagesize 0x%x, ivt offset 0x%x\n",
+              image_offset, pagesize, offset);
+
+       if (((rom_bt_dev >> 16) & 0xff) ==  BT_DEV_TYPE_FLEXSPINOR)
+               offset = CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_SECTOR * 512;
+       else
+               offset = image_offset +
+                       CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_SECTOR * 512 - 0x8000;
+
+       size = ALIGN(sizeof(struct image_header), pagesize);
+       ret = g_rom_api->download_image((u8 *)header, offset, size,
+                                       ((uintptr_t)header) ^ offset ^ size);
+       gd = pgd;
+
+       if (ret != ROM_API_OKAY) {
+               printf("ROMAPI: download failure offset 0x%x size 0x%x\n",
+                      offset, size);
+               return -1;
+       }
+
+       if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
+           image_get_magic(header) == FDT_MAGIC) {
+               struct spl_load_info load;
+
+               memset(&load, 0, sizeof(load));
+               load.bl_len = pagesize;
+               load.read = spl_romapi_read_seekable;
+               load.priv = &pagesize;
+               return spl_load_simple_fit(spl_image, &load,
+                                          offset / pagesize, header);
+       } else {
+               /* TODO */
+               puts("Can't support legacy image\n");
+               return -1;
+       }
+
+       return 0;
+}
+
+static ulong spl_ram_load_read(struct spl_load_info *load, ulong sector,
+                              ulong count, void *buf)
+{
+       memcpy(buf, (void *)(sector), count);
+
+       if (load->priv) {
+               ulong *p = (ulong *)load->priv;
+               ulong total = sector + count;
+
+               if (total > *p)
+                       *p = total;
+       }
+
+       return count;
+}
+
+static ulong get_fit_image_size(void *fit)
+{
+       struct spl_image_info spl_image;
+       struct spl_load_info spl_load_info;
+       ulong last = (ulong)fit;
+
+       memset(&spl_load_info, 0, sizeof(spl_load_info));
+       spl_load_info.bl_len = 1;
+       spl_load_info.read = spl_ram_load_read;
+       spl_load_info.priv = &last;
+
+       spl_load_simple_fit(&spl_image, &spl_load_info,
+                           (uintptr_t)fit, fit);
+
+       return last - (ulong)fit;
+}
+
+u8 *search_fit_header(u8 *p, int size)
+{
+       int i;
+
+       for (i = 0; i < size; i += 4)
+               if (genimg_get_format(p + i) == IMAGE_FORMAT_FIT)
+                       return p + i;
+
+       return NULL;
+}
+
+static int spl_romapi_load_image_stream(struct spl_image_info *spl_image,
+                                       struct spl_boot_device *bootdev)
+{
+       struct spl_load_info load;
+       volatile gd_t *pgd = gd;
+       u32 pagesize, pg;
+       int ret;
+       int i = 0;
+       u8 *p = (u8 *)CONFIG_SPL_IMX_ROMAPI_LOADADDR;
+       u8 *pfit = NULL;
+       int imagesize;
+       int total;
+
+       ret = g_rom_api->query_boot_infor(QUERY_PAGE_SZ, &pagesize,
+                                         ((uintptr_t)&pagesize) ^ 
QUERY_PAGE_SZ);
+       gd = pgd;
+
+       if (ret != ROM_API_OKAY)
+               puts("failure at query_boot_info\n");
+
+       pg = pagesize;
+       if (pg < 1024)
+               pg = 1024;
+
+       for (i = 0; i < 640; i++) {
+               ret = g_rom_api->download_image(p, 0, pg,
+                                               ((uintptr_t)p) ^ pg);
+               gd = pgd;
+
+               if (ret != ROM_API_OKAY) {
+                       puts("Steam(USB) download failure\n");
+                       return -1;
+               }
+
+               pfit = search_fit_header(p, pg);
+               p += pg;
+
+               if (pfit)
+                       break;
+       }
+
+       if (!pfit) {
+               puts("Can't found uboot FIT image in 640K range \n");
+               return -1;
+       }
+
+       if (p - pfit < sizeof(struct fdt_header)) {
+               ret = g_rom_api->download_image(p, 0, pg,  ((uintptr_t)p) ^ pg);
+               gd = pgd;
+
+               if (ret != ROM_API_OKAY) {
+                       puts("Steam(USB) download failure\n");
+                       return -1;
+               }
+
+               p += pg;
+       }
+
+       imagesize = fit_get_size(pfit);
+       printf("Find FIT header 0x&%p, size %d\n", pfit, imagesize);
+
+       if (p - pfit < imagesize) {
+               imagesize -= p - pfit;
+               /*need pagesize hear after ROM fix USB problme*/
+               imagesize += pg - 1;
+               imagesize /= pg;
+               imagesize *= pg;
+
+               printf("Need continue download %d\n", imagesize);
+
+               ret = g_rom_api->download_image(p, 0, imagesize,
+                                               ((uintptr_t)p) ^ imagesize);
+               gd = pgd;
+
+               p += imagesize;
+
+               if (ret != ROM_API_OKAY) {
+                       printf("Failure download %d\n", imagesize);
+                       return -1;
+               }
+       }
+
+       total = get_fit_image_size(pfit);
+       total += 3;
+       total &= ~0x3;
+
+       imagesize = total - (p - pfit);
+
+       imagesize += pagesize - 1;
+       imagesize /= pagesize;
+       imagesize *= pagesize;
+
+       printf("Download %d, total fit %d\n", imagesize, total);
+
+       ret = g_rom_api->download_image(p, 0, imagesize,
+                                       ((uintptr_t)p) ^ imagesize);
+       if (ret != ROM_API_OKAY)
+               printf("ROM download failure %d\n", imagesize);
+
+       memset(&load, 0, sizeof(load));
+       load.bl_len = 1;
+       load.read = spl_ram_load_read;
+
+       return spl_load_simple_fit(spl_image, &load, (ulong)pfit, pfit);
+}
+
+int board_return_to_bootrom(struct spl_image_info *spl_image,
+                           struct spl_boot_device *bootdev)
+{
+       volatile gd_t *pgd = gd;
+       int ret;
+       u32 boot;
+
+       ret = g_rom_api->query_boot_infor(QUERY_BT_DEV, &boot,
+                                         ((uintptr_t)&boot) ^ QUERY_BT_DEV);
+       gd =  pgd;
+
+       if (ret != ROM_API_OKAY) {
+               puts("ROMAPI: failure at query_boot_info\n");
+               return -1;
+       }
+
+       if (is_boot_from_stream_device(boot))
+               return spl_romapi_load_image_stream(spl_image, bootdev);
+
+       return spl_romapi_load_image_seekable(spl_image, bootdev, boot);
+}
-- 
2.16.4

_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
https://lists.denx.de/listinfo/u-boot

Reply via email to