// SPDX-License-Identifier: GPL-2.0+
/*
 * NAND boot for Freescale Enhanced Local Bus Controller, Flash Control Machine
 *
 * (C) Copyright 2006-2008
 * Stefan Roese, DENX Software Engineering, sr@denx.de.
 *
 * Copyright (c) 2008 Freescale Semiconductor, Inc.
 * Author: Scott Wood <scottwood@freescale.com>
 */

#include <common.h>
#include <asm/io.h>
#include <asm/fsl_lbc.h>
#include <nand.h>

#define WINDOW_SIZE 8192

static void nand_wait(void)
{
        fsl_lbc_t *regs = LBC_BASE_ADDR;

        for (;;) {
                uint32_t status = in_be32(&regs->ltesr);

                if (status == 1)
                        return;

                if (status & 1) {
                        puts("read failed (ltesr)\n");
                        for (;;);
                }
        }
}

#ifdef CONFIG_TPL_BUILD
int nand_spl_load_image(uint32_t offs, unsigned int uboot_size, void *vdst)
#else
static int nand_load_image(uint32_t offs, unsigned int uboot_size, void *vdst)
#endif
{
        fsl_lbc_t *regs = LBC_BASE_ADDR;
        uchar *buf = (uchar *)CONFIG_SYS_NAND_BASE;
        const int large = CONFIG_SYS_NAND_OR_PRELIM & OR_FCM_PGS;
        const int block_shift = large ? 17 : 14;
        const int block_size = 1 << block_shift;
        const int page_size = large ? 2048 : 512;
        const int bad_marker = large ? page_size + 0 : page_size + 5;
        int fmr = (15 << FMR_CWTO_SHIFT) | (2 << FMR_AL_SHIFT) | 2;
        int pos = 0;
        char *dst = vdst;

        if (offs & (block_size - 1)) {
                puts("bad offset\n");
                for (;;);
        }

        if (large) {
                fmr |= FMR_ECCM;
                out_be32(&regs->fcr, (NAND_CMD_READ0 << FCR_CMD0_SHIFT) |
                                     (NAND_CMD_READSTART << FCR_CMD1_SHIFT));
                out_be32(&regs->fir,
                         (FIR_OP_CW0 << FIR_OP0_SHIFT) |
                         (FIR_OP_CA  << FIR_OP1_SHIFT) |
                         (FIR_OP_PA  << FIR_OP2_SHIFT) |
                         (FIR_OP_CW1 << FIR_OP3_SHIFT) |
                         (FIR_OP_RBW << FIR_OP4_SHIFT));
        } else {
                out_be32(&regs->fcr, NAND_CMD_READ0 << FCR_CMD0_SHIFT);
                out_be32(&regs->fir,
                         (FIR_OP_CW0 << FIR_OP0_SHIFT) |
                         (FIR_OP_CA  << FIR_OP1_SHIFT) |
                         (FIR_OP_PA  << FIR_OP2_SHIFT) |
                         (FIR_OP_RBW << FIR_OP3_SHIFT));
        }

        out_be32(&regs->fbcr, 0);
        clrsetbits_be32(&regs->bank[0].br, BR_DECC, BR_DECC_CHK_GEN);

        while (pos < uboot_size) {
                int i = 0;
                out_be32(&regs->fbar, offs >> block_shift);

                do {
                        int j;
                        unsigned int page_offs = (offs & (block_size - 1)) << 1;

                        out_be32(&regs->ltesr, ~0);
                        out_be32(&regs->lteatr, 0);
                        out_be32(&regs->fpar, page_offs);
                        out_be32(&regs->fmr, fmr);
                        out_be32(&regs->lsor, 0);
                        nand_wait();

                        page_offs %= WINDOW_SIZE;

                        /*
                         * If either of the first two pages are marked bad,
                         * continue to the next block.
                         */
                        if (i++ < 2 && buf[page_offs + bad_marker] != 0xff) {
                                puts("skipping\n");
                                offs = (offs + block_size) & ~(block_size - 1);
                                pos &= ~(block_size - 1);
                                break;
                        }

                        for (j = 0; j < page_size; j++)
                                dst[pos + j] = buf[page_offs + j];

                        pos += page_size;
                        offs += page_size;
                } while ((offs & (block_size - 1)) && (pos < uboot_size));
        }

        return 0;
}

/*
 * Defines a static function nand_load_image() here, because non-static makes
 * the code too large for certain SPLs(minimal SPL, maximum size <= 4Kbytes)
 */
#ifndef CONFIG_TPL_BUILD
#define nand_spl_load_image(offs, uboot_size, vdst) \
        nand_load_image(offs, uboot_size, vdst)
#endif

/*
 * The main entry for NAND booting. It's necessary that SDRAM is already
 * configured and available since this code loads the main U-Boot image
 * from NAND into SDRAM and starts it from there.
 */
void nand_boot(void)
{
        __attribute__((noreturn)) void (*uboot)(void);
        /*
         * Load U-Boot image from NAND into RAM
         */
        nand_spl_load_image(CONFIG_SYS_NAND_U_BOOT_OFFS,
                            CONFIG_SYS_NAND_U_BOOT_SIZE,
                            (void *)CONFIG_SYS_NAND_U_BOOT_DST);

#ifdef CONFIG_NAND_ENV_DST
        nand_spl_load_image(CONFIG_ENV_OFFSET, CONFIG_ENV_SIZE,
                            (void *)CONFIG_NAND_ENV_DST);

#ifdef CONFIG_ENV_OFFSET_REDUND
        nand_spl_load_image(CONFIG_ENV_OFFSET_REDUND, CONFIG_ENV_SIZE,
                            (void *)CONFIG_NAND_ENV_DST + CONFIG_ENV_SIZE);
#endif
#endif

#ifdef CONFIG_SPL_FLUSH_IMAGE
        /*
         * Clean d-cache and invalidate i-cache, to
         * make sure that no stale data is executed.
         */
        flush_cache(CONFIG_SYS_NAND_U_BOOT_DST, CONFIG_SYS_NAND_U_BOOT_SIZE);
#endif

        puts("transfering control\n");
        /*
         * Jump to U-Boot image
         */
        uboot = (void *)CONFIG_SYS_NAND_U_BOOT_START;
        (*uboot)();
}