// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "crypto/unexportable_key.h"

#include "base/check.h"
#include "base/functional/bind.h"
#if BUILDFLAG(IS_WIN)
#include "crypto/unexportable_key_win.h"
#elif BUILDFLAG(IS_APPLE) && !BUILDFLAG(IS_IOS_TVOS)
#include "crypto/apple/unexportable_key_apple.h"
#endif

namespace crypto {

namespace {
std::unique_ptr<UnexportableKeyProvider> (*g_mock_provider)() = nullptr;
}  // namespace

UnexportableKeyProvider::~UnexportableKeyProvider() = default;

std::unique_ptr<UnexportableAttestationKey>
UnexportableKeyProvider::GenerateAttestationKeySlowly(
    base::span<const SignatureVerifier::SignatureAlgorithm>
        acceptable_algorithms) {
  return nullptr;
}

std::unique_ptr<UnexportableAttestationKey>
UnexportableKeyProvider::FromWrappedAttestationKeySlowly(
    base::span<const uint8_t> wrapped_key) {
  return nullptr;
}

VirtualUnexportableSigningKey::~VirtualUnexportableSigningKey() = default;
VirtualUnexportableKeyProvider::~VirtualUnexportableKeyProvider() = default;

bool UnexportableSigningKey::IsHardwareBacked() const {
  return false;
}

const StatefulKey* UnexportableSigningKey::AsStatefulKey() const {
  return nullptr;
}

std::unique_ptr<UnexportableKeyProvider> GetUnexportableKeyProvider(
    UnexportableKeyProvider::Config config) {
  if (g_mock_provider) {
    return g_mock_provider();
  }

#if BUILDFLAG(IS_WIN)
  return GetUnexportableKeyProviderWin();
#elif BUILDFLAG(IS_APPLE) && !BUILDFLAG(IS_IOS_TVOS)
  return apple::GetUnexportableKeyProviderApple(std::move(config));
#else
  return nullptr;
#endif
}

std::unique_ptr<UnexportableKeyProvider>
GetMicrosoftSoftwareUnexportableKeyProvider() {
  if (g_mock_provider) {
    return g_mock_provider();
  }
#if BUILDFLAG(IS_WIN)
  return GetMicrosoftSoftwareUnexportableKeyProviderWin();
#else
  return nullptr;
#endif
}

std::unique_ptr<VirtualUnexportableKeyProvider>
GetVirtualUnexportableKeyProvider_DO_NOT_USE_METRICS_ONLY() {
#if BUILDFLAG(IS_WIN)
  return GetVirtualUnexportableKeyProviderWin();
#else
  return nullptr;
#endif
}

namespace internal {

bool HasScopedUnexportableKeyProvider() {
  return g_mock_provider != nullptr;
}

void SetUnexportableKeyProviderForTesting(
    std::unique_ptr<UnexportableKeyProvider> (*func)()) {
  if (g_mock_provider) {
    // Nesting ScopedMockUnexportableSigningKeyForTesting is not supported.
    CHECK(!func);
    g_mock_provider = nullptr;
  } else {
    g_mock_provider = func;
  }
}

}  // namespace internal
}  // namespace crypto