// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2020 Western Digital Corporation or its affiliates.
 */

#include <linux/efi.h>

#include <asm/efi.h>
#include <asm/sections.h>
#include <linux/unaligned.h>

#include "efistub.h"

unsigned long stext_offset(void)
{
        /*
         * When built as part of the kernel, the EFI stub cannot branch to the
         * kernel proper via the image header, as the PE/COFF header is
         * strictly not part of the in-memory presentation of the image, only
         * of the file representation. So instead, we need to jump to the
         * actual entrypoint in the .text region of the image.
         */
        return _start_kernel - _start;
}

efi_status_t handle_kernel_image(unsigned long *image_addr,
                                 unsigned long *image_size,
                                 unsigned long *reserve_addr,
                                 unsigned long *reserve_size,
                                 efi_loaded_image_t *image,
                                 efi_handle_t image_handle)
{
        unsigned long kernel_size, kernel_codesize, kernel_memsize;
        efi_status_t status;

        kernel_size = _edata - _start;
        kernel_codesize = __init_text_end - _start;
        kernel_memsize = kernel_size + (_end - _edata);
        *image_addr = (unsigned long)_start;
        *image_size = kernel_memsize;
        *reserve_size = *image_size;

        status = efi_kaslr_relocate_kernel(image_addr,
                                           reserve_addr, reserve_size,
                                           kernel_size, kernel_codesize, kernel_memsize,
                                           efi_kaslr_get_phys_seed(image_handle));
        if (status != EFI_SUCCESS) {
                efi_err("Failed to relocate kernel\n");
                *image_size = 0;
        }

        return status;
}

void efi_icache_sync(unsigned long start, unsigned long end)
{
        asm volatile ("fence.i" ::: "memory");
}