// SPDX-License-Identifier: GPL-1.0+
/*
 * Copyright (C) 2018 IBM Corporation
 */
#include <linux/efi.h>
#include <linux/secure_boot.h>
#include <asm/efi.h>

#ifndef arch_efi_boot_mode
#define arch_efi_boot_mode efi_secureboot_mode_unset
#endif

static enum efi_secureboot_mode get_sb_mode(void)
{
	enum efi_secureboot_mode mode;

	if (!efi_rt_services_supported(EFI_RT_SUPPORTED_GET_VARIABLE)) {
		pr_info("integrity: secureboot mode unknown, no efi\n");
		return efi_secureboot_mode_unknown;
	}

	mode = efi_get_secureboot_mode(efi.get_variable);
	if (mode == efi_secureboot_mode_disabled)
		pr_info("integrity: secureboot mode disabled\n");
	else if (mode == efi_secureboot_mode_unknown)
		pr_info("integrity: secureboot mode unknown\n");
	else
		pr_info("integrity: secureboot mode enabled\n");
	return mode;
}

/*
 * Query secure boot status
 *
 * Note don't call this function too early e.g. in __setup hook otherwise the
 * kernel may hang when calling efi_get_secureboot_mode.
 *
 */
bool arch_get_secureboot(void)
{
	static enum efi_secureboot_mode sb_mode;
	static bool initialized;

	if (!initialized && efi_enabled(EFI_BOOT)) {
		sb_mode = arch_efi_boot_mode;

		if (sb_mode == efi_secureboot_mode_unset)
			sb_mode = get_sb_mode();
		initialized = true;
	}

	if (sb_mode == efi_secureboot_mode_enabled)
		return true;
	else
		return false;
}