uboot/common/spl/spl_ubi.c
<<
>>
Prefs
   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 <image.h>
  10#include <nand.h>
  11#include <onenand_uboot.h>
  12#include <ubispl.h>
  13#include <spl.h>
  14
  15int spl_ubi_load_image(struct spl_image_info *spl_image,
  16                       struct spl_boot_device *bootdev)
  17{
  18        struct image_header *header;
  19        struct ubispl_info info;
  20        struct ubispl_load volumes[2];
  21        int ret = 1;
  22
  23        switch (bootdev->boot_device) {
  24#ifdef CONFIG_SPL_NAND_SUPPORT
  25        case BOOT_DEVICE_NAND:
  26                nand_init();
  27                info.read = nand_spl_read_block;
  28                info.peb_size = CONFIG_SYS_NAND_BLOCK_SIZE;
  29                break;
  30#endif
  31#ifdef CONFIG_SPL_ONENAND_SUPPORT
  32        case BOOT_DEVICE_ONENAND:
  33                info.read = onenand_spl_read_block;
  34                info.peb_size = CONFIG_SYS_ONENAND_BLOCK_SIZE;
  35                break;
  36#endif
  37        default:
  38                goto out;
  39        }
  40        info.ubi = (struct ubi_scan_info *)CONFIG_SPL_UBI_INFO_ADDR;
  41        info.fastmap = IS_ENABLED(CONFIG_MTD_UBI_FASTMAP);
  42
  43        info.peb_offset = CONFIG_SPL_UBI_PEB_OFFSET;
  44        info.vid_offset = CONFIG_SPL_UBI_VID_OFFSET;
  45        info.leb_start = CONFIG_SPL_UBI_LEB_START;
  46        info.peb_count = CONFIG_SPL_UBI_MAX_PEBS - info.peb_offset;
  47
  48#ifdef CONFIG_SPL_OS_BOOT
  49        if (!spl_start_uboot()) {
  50                volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_KERNEL_ID;
  51                volumes[0].load_addr = (void *)CONFIG_SYS_LOAD_ADDR;
  52                volumes[1].vol_id = CONFIG_SPL_UBI_LOAD_ARGS_ID;
  53                volumes[1].load_addr = (void *)CONFIG_SYS_SPL_ARGS_ADDR;
  54
  55                ret = ubispl_load_volumes(&info, volumes, 2);
  56                if (!ret) {
  57                        header = (struct image_header *)volumes[0].load_addr;
  58                        spl_parse_image_header(spl_image, header);
  59                        puts("Linux loaded.\n");
  60                        goto out;
  61                }
  62                puts("Loading Linux failed, falling back to U-Boot.\n");
  63        }
  64#endif
  65        header = spl_get_load_buffer(-sizeof(*header), sizeof(header));
  66#ifdef CONFIG_SPL_UBI_LOAD_BY_VOLNAME
  67        volumes[0].vol_id = -1;
  68        strncpy(volumes[0].name,
  69                CONFIG_SPL_UBI_LOAD_MONITOR_VOLNAME,
  70                UBI_VOL_NAME_MAX + 1);
  71#else
  72        volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
  73#endif
  74        volumes[0].load_addr = (void *)header;
  75
  76        ret = ubispl_load_volumes(&info, volumes, 1);
  77        if (!ret)
  78                spl_parse_image_header(spl_image, header);
  79out:
  80#ifdef CONFIG_SPL_NAND_SUPPORT
  81        if (bootdev->boot_device == BOOT_DEVICE_NAND)
  82                nand_deselect();
  83#endif
  84        return ret;
  85}
  86/* Use priorty 0 so that Ubi will override NAND and ONENAND methods */
  87SPL_LOAD_IMAGE_METHOD("NAND", 0, BOOT_DEVICE_NAND, spl_ubi_load_image);
  88SPL_LOAD_IMAGE_METHOD("OneNAND", 0, BOOT_DEVICE_ONENAND, spl_ubi_load_image);
  89