/*
 * (C) Copyright 2008 Semihalf
 *
 * (C) Copyright 2000-2006
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */


#include <common.h>
#include <watchdog.h>
#include <command.h>
#include <image.h>
#include <malloc.h>
#include <u-boot/zlib.h>
#include <bzlib.h>
#include <environment.h>
#include <asm/byteorder.h>
#include <asm/mp.h>

#if defined(CONFIG_OF_LIBFDT)
#include <fdt.h>
#include <libfdt.h>
#include <fdt_support.h>

#endif

#ifdef CONFIG_SYS_INIT_RAM_LOCK
#include <asm/cache.h>
#endif

DECLARE_GLOBAL_DATA_PTR;

extern ulong get_effective_memsize(void);
static ulong get_sp (void);
static void set_clocks_in_mhz (bd_t *kbd);

#ifndef CONFIG_SYS_LINUX_LOWMEM_MAX_SIZE
#define CONFIG_SYS_LINUX_LOWMEM_MAX_SIZE        (768*1024*1024)
#endif

static void boot_jump_linux(bootm_headers_t *images)
{
        void    (*kernel)(bd_t *, ulong r4, ulong r5, ulong r6,
                          ulong r7, ulong r8, ulong r9);
#ifdef CONFIG_OF_LIBFDT
        char *of_flat_tree = images->ft_addr;
#endif

        kernel = (void (*)(bd_t *, ulong, ulong, ulong,
                           ulong, ulong, ulong))images->ep;
        debug ("## Transferring control to Linux (at address %08lx) ...\n",
                (ulong)kernel);

        show_boot_progress (15);

#if defined(CONFIG_SYS_INIT_RAM_LOCK) && !defined(CONFIG_E500)
        unlock_ram_in_cache();
#endif

#if defined(CONFIG_OF_LIBFDT)
        if (of_flat_tree) {     /* device tree; boot new style */
                /*
                 * Linux Kernel Parameters (passing device tree):
                 *   r3: pointer to the fdt
                 *   r4: 0
                 *   r5: 0
                 *   r6: epapr magic
                 *   r7: size of IMA in bytes
                 *   r8: 0
                 *   r9: 0
                 */
#if defined(CONFIG_MPC85xx) || defined(CONFIG_440)
 #define EPAPR_MAGIC    (0x45504150)
#else
 #define EPAPR_MAGIC    (0x65504150)
#endif

                debug ("   Booting using OF flat tree...\n");
                WATCHDOG_RESET ();
                (*kernel) ((bd_t *)of_flat_tree, 0, 0, EPAPR_MAGIC,
                           getenv_bootm_mapsize(), 0, 0);
                /* does not return */
        } else
#endif
        {
                /*
                 * Linux Kernel Parameters (passing board info data):
                 *   r3: ptr to board info data
                 *   r4: initrd_start or 0 if no initrd
                 *   r5: initrd_end - unused if r4 is 0
                 *   r6: Start of command line string
                 *   r7: End   of command line string
                 *   r8: 0
                 *   r9: 0
                 */
                ulong cmd_start = images->cmdline_start;
                ulong cmd_end = images->cmdline_end;
                ulong initrd_start = images->initrd_start;
                ulong initrd_end = images->initrd_end;
                bd_t *kbd = images->kbd;

                debug ("   Booting using board info...\n");
                WATCHDOG_RESET ();
                (*kernel) (kbd, initrd_start, initrd_end,
                           cmd_start, cmd_end, 0, 0);
                /* does not return */
        }
        return ;
}

void arch_lmb_reserve(struct lmb *lmb)
{
        phys_size_t bootm_size;
        ulong size, sp, bootmap_base;

        bootmap_base = getenv_bootm_low();
        bootm_size = getenv_bootm_size();

#ifdef DEBUG
        if (((u64)bootmap_base + bootm_size) >
            (CONFIG_SYS_SDRAM_BASE + (u64)gd->ram_size))
                puts("WARNING: bootm_low + bootm_size exceed total memory\n");
        if ((bootmap_base + bootm_size) > get_effective_memsize())
                puts("WARNING: bootm_low + bootm_size exceed eff. memory\n");
#endif

        size = min(bootm_size, get_effective_memsize());
        size = min(size, CONFIG_SYS_LINUX_LOWMEM_MAX_SIZE);

        if (size < bootm_size) {
                ulong base = bootmap_base + size;
                printf("WARNING: adjusting available memory to %lx\n", size);
                lmb_reserve(lmb, base, bootm_size - size);
        }

        /*
         * Booting a (Linux) kernel image
         *
         * Allocate space for command line and board info - the
         * address should be as high as possible within the reach of
         * the kernel (see CONFIG_SYS_BOOTMAPSZ settings), but in unused
         * memory, which means far enough below the current stack
         * pointer.
         */
        sp = get_sp();
        debug ("## Current stack ends at 0x%08lx\n", sp);

        /* adjust sp by 4K to be safe */
        sp -= 4096;
        lmb_reserve(lmb, sp, (CONFIG_SYS_SDRAM_BASE + get_effective_memsize() - sp));

#ifdef CONFIG_MP
        cpu_mp_lmb_reserve(lmb);
#endif

        return ;
}

static int boot_cmdline_linux(bootm_headers_t *images)
{
        ulong of_size = images->ft_len;
        struct lmb *lmb = &images->lmb;
        ulong *cmd_start = &images->cmdline_start;
        ulong *cmd_end = &images->cmdline_end;

        int ret = 0;

        if (!of_size) {
                /* allocate space and init command line */
                ret = boot_get_cmdline (lmb, cmd_start, cmd_end);
                if (ret) {
                        puts("ERROR with allocation of cmdline\n");
                        return ret;
                }
        }

        return ret;
}

static int boot_bd_t_linux(bootm_headers_t *images)
{
        ulong of_size = images->ft_len;
        struct lmb *lmb = &images->lmb;
        bd_t **kbd = &images->kbd;

        int ret = 0;

        if (!of_size) {
                /* allocate space for kernel copy of board info */
                ret = boot_get_kbd (lmb, kbd);
                if (ret) {
                        puts("ERROR with allocation of kernel bd\n");
                        return ret;
                }
                set_clocks_in_mhz(*kbd);
        }

        return ret;
}

/*
 * Verify the device tree.
 *
 * This function is called after all device tree fix-ups have been enacted,
 * so that the final device tree can be verified.  The definition of "verified"
 * is up to the specific implementation.  However, it generally means that the
 * addresses of some of the devices in the device tree are compared with the
 * actual addresses at which U-Boot has placed them.
 *
 * Returns 1 on success, 0 on failure.  If 0 is returned, U-boot will halt the
 * boot process.
 */
static int __ft_verify_fdt(void *fdt)
{
        return 1;
}
__attribute__((weak, alias("__ft_verify_fdt"))) int ft_verify_fdt(void *fdt);

static int boot_body_linux(bootm_headers_t *images)
{
        ulong rd_len;
        struct lmb *lmb = &images->lmb;
        ulong *initrd_start = &images->initrd_start;
        ulong *initrd_end = &images->initrd_end;
#if defined(CONFIG_OF_LIBFDT)
        ulong of_size = images->ft_len;
        char **of_flat_tree = &images->ft_addr;
#endif

        int ret;

#if defined(CONFIG_OF_LIBFDT)
        boot_fdt_add_mem_rsv_regions(lmb, *of_flat_tree);
#endif

        /* allocate space and init command line */
        ret = boot_cmdline_linux(images);
        if (ret)
                return ret;

        /* allocate space for kernel copy of board info */
        ret = boot_bd_t_linux(images);
        if (ret)
                return ret;

        rd_len = images->rd_end - images->rd_start;
        ret = boot_ramdisk_high (lmb, images->rd_start, rd_len, initrd_start, initrd_end);
        if (ret)
                return ret;

#if defined(CONFIG_OF_LIBFDT)
        ret = boot_relocate_fdt(lmb, of_flat_tree, &of_size);
        if (ret)
                return ret;

        /*
         * Add the chosen node if it doesn't exist, add the env and bd_t
         * if the user wants it (the logic is in the subroutines).
         */
        if (of_size) {
                if (fdt_chosen(*of_flat_tree, 1) < 0) {
                        puts ("ERROR: ");
                        puts ("/chosen node create failed");
                        puts (" - must RESET the board to recover.\n");
                        return -1;
                }
#ifdef CONFIG_OF_BOARD_SETUP
                /* Call the board-specific fixup routine */
                ft_board_setup(*of_flat_tree, gd->bd);
#endif

                /* Delete the old LMB reservation */
                lmb_free(lmb, (phys_addr_t)(u32)*of_flat_tree,
                                (phys_size_t)fdt_totalsize(*of_flat_tree));

                ret = fdt_resize(*of_flat_tree);
                if (ret < 0)
                        return ret;
                of_size = ret;

                if (*initrd_start && *initrd_end) {
                        of_size += FDT_RAMDISK_OVERHEAD;
                        fdt_set_totalsize(*of_flat_tree, of_size);
                }
                /* Create a new LMB reservation */
                lmb_reserve(lmb, (ulong)*of_flat_tree, of_size);

                /* fixup the initrd now that we know where it should be */
                if (*initrd_start && *initrd_end)
                        fdt_initrd(*of_flat_tree, *initrd_start, *initrd_end, 1);

                if (!ft_verify_fdt(*of_flat_tree))
                        return -1;
        }
#endif  /* CONFIG_OF_LIBFDT */
        return 0;
}

__attribute__((noinline))
int do_bootm_linux(int flag, int argc, char * const argv[], bootm_headers_t *images)
{
        int     ret;

        if (flag & BOOTM_STATE_OS_CMDLINE) {
                boot_cmdline_linux(images);
                return 0;
        }

        if (flag & BOOTM_STATE_OS_BD_T) {
                boot_bd_t_linux(images);
                return 0;
        }

        /*
         * We do nothing & report success to retain compatiablity with older
         * versions of u-boot in which this use to flush the dcache on MP
         * systems
         */
        if (flag & BOOTM_STATE_OS_PREP)
                return 0;

        if (flag & BOOTM_STATE_OS_GO) {
                boot_jump_linux(images);
                return 0;
        }

        ret = boot_body_linux(images);
        if (ret)
                return ret;
        boot_jump_linux(images);

        return 0;
}

static ulong get_sp (void)
{
        ulong sp;

        asm( "mr %0,1": "=r"(sp) : );
        return sp;
}

static void set_clocks_in_mhz (bd_t *kbd)
{
        char    *s;

        if ((s = getenv ("clocks_in_mhz")) != NULL) {
                /* convert all clock information to MHz */
                kbd->bi_intfreq /= 1000000L;
                kbd->bi_busfreq /= 1000000L;
#if defined(CONFIG_MPC8220)
                kbd->bi_inpfreq /= 1000000L;
                kbd->bi_pcifreq /= 1000000L;
                kbd->bi_pevfreq /= 1000000L;
                kbd->bi_flbfreq /= 1000000L;
                kbd->bi_vcofreq /= 1000000L;
#endif
#if defined(CONFIG_CPM2)
                kbd->bi_cpmfreq /= 1000000L;
                kbd->bi_brgfreq /= 1000000L;
                kbd->bi_sccfreq /= 1000000L;
                kbd->bi_vco     /= 1000000L;
#endif
#if defined(CONFIG_MPC5xxx)
                kbd->bi_ipbfreq /= 1000000L;
                kbd->bi_pcifreq /= 1000000L;
#endif /* CONFIG_MPC5xxx */
        }
}