/*
 * Copyright (C) 2019 Intel Corporation. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

#include <multiboot.h>
#include <vm.h>
#include <types.h>
#include <pgtable.h>
#include <acpi.h>
#include <vboot.h>
#include <direct_boot.h>
#include <deprivilege_boot.h>
#include <logmsg.h>

#define BOOTLOADER_NUM 4U
#define BOOTLOADER_NAME_SIZE 20U

struct vboot_bootloader_map {
	const char bootloader_name[BOOTLOADER_NAME_SIZE];
	enum vboot_mode mode;
};

static struct vboot_operations *vboot_ops;
static enum vboot_mode sos_boot_mode;

/**
 * @pre: this function is called during detect mode which is very early stage,
 * other exported interfaces should not be called beforehand.
 */
void init_vboot(void)
{

	struct multiboot_info *mbi;
	uint32_t i;

	const struct vboot_bootloader_map vboot_bootloader_maps[BOOTLOADER_NUM] = {
		{"Slim BootLoader", DIRECT_BOOT_MODE},
		{"Intel IOTG/TSD ABL", DIRECT_BOOT_MODE},
		{"ACRN UEFI loader", DEPRI_BOOT_MODE},
		{"GRUB", DIRECT_BOOT_MODE},
	};

	mbi = (struct multiboot_info *)hpa2hva((uint64_t)boot_regs[1]);
	if (mbi == NULL) {
		panic("No multiboot info");
	} else {
		for (i = 0U; i < BOOTLOADER_NUM; i++) {
			if (strncmp(hpa2hva(mbi->mi_loader_name), vboot_bootloader_maps[i].bootloader_name,
				strnlen_s(vboot_bootloader_maps[i].bootloader_name, BOOTLOADER_NAME_SIZE)) == 0) {
				/* Only support two vboot mode */
				if (vboot_bootloader_maps[i].mode == DEPRI_BOOT_MODE) {
					vboot_ops = get_deprivilege_boot_ops();
					sos_boot_mode = DEPRI_BOOT_MODE;
				} else {
					vboot_ops = get_direct_boot_ops();
					sos_boot_mode = DIRECT_BOOT_MODE;
				}
				break;
			}
		}
	}

	/*
	 * vboot_ops is mandatory and it will be initialized correctly.
	 * The vboot_ops->init is called to assure that the boot env is
	 * initialized before calling other vboot_ops interface.
	 */
	vboot_ops->init();
}

/* @pre: vboot_ops != NULL */
enum vboot_mode get_sos_boot_mode(void)
{
	return sos_boot_mode;
}

/* @pre: vboot_ops->get_ap_trampoline != NULL */
uint64_t get_ap_trampoline_buf(void)
{
	return vboot_ops->get_ap_trampoline();
}

/* @pre: vboot_ops->get_rsdp != NULL */
void *get_rsdp_ptr(void)
{
	return vboot_ops->get_rsdp();
}

/* @pre: vboot_ops->init_irq != NULL */
void init_vboot_irq(void)
{
	return vboot_ops->init_irq();
}