// SPDX-License-Identifier: GPL-2.0

#include <linux/kernel.h>
#include <linux/console.h>
#include <linux/kexec.h>
#include <linux/delay.h>
#include <linux/reboot.h>

#include <asm/cacheflush.h>
#include <asm/sections.h>

extern void relocate_new_kernel(unsigned long head,
                                unsigned long start,
                                unsigned long phys);

extern const unsigned int relocate_new_kernel_size;
extern unsigned int kexec_initrd_start_offset;
extern unsigned int kexec_initrd_end_offset;
extern unsigned int kexec_cmdline_offset;
extern unsigned int kexec_free_mem_offset;

static void kexec_show_segment_info(const struct kimage *kimage,
                                    unsigned long n)
{
        pr_debug("    segment[%lu]: %016lx - %016lx, 0x%lx bytes, %lu pages\n",
                        n,
                        kimage->segment[n].mem,
                        kimage->segment[n].mem + kimage->segment[n].memsz,
                        (unsigned long)kimage->segment[n].memsz,
                        (unsigned long)kimage->segment[n].memsz /  PAGE_SIZE);
}

static void kexec_image_info(const struct kimage *kimage)
{
        unsigned long i;

        pr_debug("kexec kimage info:\n");
        pr_debug("  type:        %d\n", kimage->type);
        pr_debug("  start:       %lx\n", kimage->start);
        pr_debug("  head:        %lx\n", kimage->head);
        pr_debug("  nr_segments: %lu\n", kimage->nr_segments);

        for (i = 0; i < kimage->nr_segments; i++)
                kexec_show_segment_info(kimage, i);

#ifdef CONFIG_KEXEC_FILE
        if (kimage->file_mode) {
                pr_debug("cmdline: %.*s\n", (int)kimage->cmdline_buf_len,
                         kimage->cmdline_buf);
        }
#endif
}

void machine_kexec_cleanup(struct kimage *kimage)
{
}

void machine_crash_shutdown(struct pt_regs *regs)
{
}

void machine_shutdown(void)
{
        smp_send_stop();
        while (num_online_cpus() > 1) {
                cpu_relax();
                mdelay(1);
        }
}

void machine_kexec(struct kimage *image)
{
#ifdef CONFIG_64BIT
        Elf64_Fdesc desc;
#endif
        void (*reloc)(unsigned long head,
                      unsigned long start,
                      unsigned long phys);

        unsigned long phys = page_to_phys(image->control_code_page);
        void *virt = (void *)__fix_to_virt(FIX_TEXT_KEXEC);
        struct kimage_arch *arch = &image->arch;

        set_fixmap(FIX_TEXT_KEXEC, phys);

        flush_cache_all();

#ifdef CONFIG_64BIT
        reloc = (void *)&desc;
        desc.addr = (long long)virt;
#else
        reloc = (void *)virt;
#endif

        memcpy(virt, dereference_function_descriptor(relocate_new_kernel),
                relocate_new_kernel_size);

        *(unsigned long *)(virt + kexec_cmdline_offset) = arch->cmdline;
        *(unsigned long *)(virt + kexec_initrd_start_offset) = arch->initrd_start;
        *(unsigned long *)(virt + kexec_initrd_end_offset) = arch->initrd_end;
        *(unsigned long *)(virt + kexec_free_mem_offset) = PAGE0->mem_free;

        flush_cache_all();
        flush_tlb_all();
        local_irq_disable();

        reloc(image->head & PAGE_MASK, image->start, phys);
}

int machine_kexec_prepare(struct kimage *image)
{
        kexec_image_info(image);
        return 0;
}