common/spl/spl_ubi.c

   1 // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
   2 /*
   3  * Copyright (C) 2016
   4  * Ladislav Michl <ladis@linux-mips.org>
   5  */
   6
   7 #include <common.h>
   8 #include <config.h>
   9 #include <nand.h>
  10 #include <onenand_uboot.h>
  11 #include <ubispl.h>
  12 #include <spl.h>
  13
  14 int spl_ubi_load_image(struct spl_image_info *spl_image,
  15                        struct spl_boot_device *bootdev)
  16 {
  17         struct image_header *header;
  18         struct ubispl_info info;
  19         struct ubispl_load volumes[2];
  20         int ret = 1;
  21
  22         switch (bootdev->boot_device) {
  23 #ifdef CONFIG_SPL_NAND_SUPPORT
  24         case BOOT_DEVICE_NAND:
  25                 nand_init();
  26                 info.read = nand_spl_read_block;
  27                 info.peb_size = CONFIG_SYS_NAND_BLOCK_SIZE;
  28                 break;
  29 #endif
  30 #ifdef CONFIG_SPL_ONENAND_SUPPORT
  31         case BOOT_DEVICE_ONENAND:
  32                 info.read = onenand_spl_read_block;
  33                 info.peb_size = CONFIG_SYS_ONENAND_BLOCK_SIZE;
  34                 break;
  35 #endif
  36         default:
  37                 goto out;
  38         }
  39         info.ubi = (struct ubi_scan_info *)CONFIG_SPL_UBI_INFO_ADDR;
  40         info.fastmap = IS_ENABLED(CONFIG_MTD_UBI_FASTMAP);
  41
  42         info.peb_offset = CONFIG_SPL_UBI_PEB_OFFSET;
  43         info.vid_offset = CONFIG_SPL_UBI_VID_OFFSET;
  44         info.leb_start = CONFIG_SPL_UBI_LEB_START;
  45         info.peb_count = CONFIG_SPL_UBI_MAX_PEBS - info.peb_offset;
  46
  47 #ifdef CONFIG_SPL_OS_BOOT
  48         if (!spl_start_uboot()) {
  49                 volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_KERNEL_ID;
  50                 volumes[0].load_addr = (void *)CONFIG_SYS_LOAD_ADDR;
  51                 volumes[1].vol_id = CONFIG_SPL_UBI_LOAD_ARGS_ID;
  52                 volumes[1].load_addr = (void *)CONFIG_SYS_SPL_ARGS_ADDR;
  53
  54                 ret = ubispl_load_volumes(&info, volumes, 2);
  55                 if (!ret) {
  56                         header = (struct image_header *)volumes[0].load_addr;
  57                         spl_parse_image_header(spl_image, header);
  58                         puts("Linux loaded.\n");
  59                         goto out;
  60                 }
  61                 puts("Loading Linux failed, falling back to U-Boot.\n");
  62         }
  63 #endif
  64         header = spl_get_load_buffer(-sizeof(*header), sizeof(header));
  65         volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
  66         volumes[0].load_addr = (void *)header;
  67
  68         ret = ubispl_load_volumes(&info, volumes, 1);
  69         if (!ret)
  70                 spl_parse_image_header(spl_image, header);
  71 out:
  72 #ifdef CONFIG_SPL_NAND_SUPPORT
  73         if (bootdev->boot_device == BOOT_DEVICE_NAND)
  74                 nand_deselect();
  75 #endif
  76         return ret;
  77 }
  78 /* Use priorty 0 so that Ubi will override NAND and ONENAND methods */
  79 SPL_LOAD_IMAGE_METHOD("NAND", 0, BOOT_DEVICE_NAND, spl_ubi_load_image);
  80 SPL_LOAD_IMAGE_METHOD("OneNAND", 0, BOOT_DEVICE_ONENAND, spl_ubi_load_image);