#ifndef _ASM_X86_UNACCEPTED_MEMORY_H
#define _ASM_X86_UNACCEPTED_MEMORY_H

#include <linux/efi.h>
#include <asm/tdx.h>
#include <asm/sev.h>

static inline void arch_accept_memory(phys_addr_t start, phys_addr_t end)
{
        /* Platform-specific memory-acceptance call goes here */
        if (cpu_feature_enabled(X86_FEATURE_TDX_GUEST)) {
                if (!tdx_accept_memory(start, end))
                        panic("TDX: Failed to accept memory\n");
        } else if (cc_platform_has(CC_ATTR_GUEST_SEV_SNP)) {
                snp_accept_memory(start, end);
        } else {
                panic("Cannot accept memory: unknown platform\n");
        }
}

static inline struct efi_unaccepted_memory *efi_get_unaccepted_table(void)
{
        if (efi.unaccepted == EFI_INVALID_TABLE_ADDR)
                return NULL;
        return __va(efi.unaccepted);
}
#endif