// SPDX-License-Identifier: GPL-2.0
/*
 * Machine keyring routines.
 *
 * Copyright (c) 2021, Oracle and/or its affiliates.
 */

#include <linux/efi.h>
#include "../integrity.h"

static __init int machine_keyring_init(void)
{
 int rc;

 rc = integrity_init_keyring(INTEGRITY_KEYRING_MACHINE);
 if (rc)
  return rc;

 pr_notice("Machine keyring initialized\n");
 return 0;
}
device_initcall(machine_keyring_init);

void __init add_to_machine_keyring(const char *source, const void *data, size_t len)
{
 key_perm_t perm;
 int rc;

 perm = (KEY_POS_ALL & ~KEY_POS_SETATTR) | KEY_USR_VIEW;
 rc = integrity_load_cert(INTEGRITY_KEYRING_MACHINE, source, data, len, perm);

 /*
  * Some MOKList keys may not pass the machine keyring restrictions.
  * If the restriction check does not pass and the platform keyring
  * is configured, try to add it into that keyring instead.
 */
 if (rc && efi_enabled(EFI_BOOT) &&
     IS_ENABLED(CONFIG_INTEGRITY_PLATFORM_KEYRING))
  rc = integrity_load_cert(INTEGRITY_KEYRING_PLATFORM, source,
      data, len, perm);

 if (rc)
  pr_info("Error adding keys to machine keyring %s\n", source);
}

/*
 * Try to load the MokListTrustedRT MOK variable to see if we should trust
 * the MOK keys within the kernel. It is not an error if this variable
 * does not exist.  If it does not exist, MOK keys should not be trusted
 * within the machine keyring.
 */
static __init bool uefi_check_trust_mok_keys(void)
{
 struct efi_mokvar_table_entry *mokvar_entry;

 mokvar_entry = efi_mokvar_entry_find("MokListTrustedRT");

 if (mokvar_entry)
  return true;

 return false;
}

static bool __init trust_moklist(void)
{
 static bool initialized;
 static bool trust_mok;

 if (!initialized) {
  initialized = true;
  trust_mok = false;

  if (uefi_check_trust_mok_keys())
   trust_mok = true;
 }

 return trust_mok;
}

/*
 * Provides platform specific check for trusting imputed keys before loading
 * on .machine keyring. UEFI systems enable this trust based on a variable,
 * and for other platforms, it is always enabled.
 */
bool __init imputed_trust_enabled(void)
{
 if (efi_enabled(EFI_BOOT))
  return trust_moklist();

 return true;
}