crypto/unexportable_key.cc

   1 // Copyright 2021 The Chromium Authors
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "crypto/unexportable_key.h"
   6
   7 #include "base/check.h"
   8 #include "base/functional/bind.h"
   9 #include "build/build_config.h"
  10
  11 namespace crypto {
  12
  13 namespace {
  14 std::unique_ptr<UnexportableKeyProvider> (*g_mock_provider)() = nullptr;
  15 }  // namespace
  16
  17 UnexportableSigningKey::~UnexportableSigningKey() = default;
  18 UnexportableKeyProvider::~UnexportableKeyProvider() = default;
  19
  20 VirtualUnexportableSigningKey::~VirtualUnexportableSigningKey() = default;
  21 VirtualUnexportableKeyProvider::~VirtualUnexportableKeyProvider() = default;
  22
  23 #if BUILDFLAG(IS_WIN)
  24 std::unique_ptr<UnexportableKeyProvider> GetUnexportableKeyProviderWin();
  25 std::unique_ptr<VirtualUnexportableKeyProvider>
  26 GetVirtualUnexportableKeyProviderWin();
  27 #endif
  28
  29 std::unique_ptr<UnexportableKeyProvider> GetUnexportableKeyProvider() {
  30   if (g_mock_provider) {
  31     return g_mock_provider();
  32   }
  33
  34 #if BUILDFLAG(IS_WIN)
  35   return GetUnexportableKeyProviderWin();
  36 #else
  37   return nullptr;
  38 #endif
  39 }
  40
  41 std::unique_ptr<VirtualUnexportableKeyProvider>
  42 GetVirtualUnexportableKeyProvider_DO_NOT_USE_METRICS_ONLY() {
  43 #if BUILDFLAG(IS_WIN)
  44   return GetVirtualUnexportableKeyProviderWin();
  45 #else
  46   return nullptr;
  47 #endif
  48 }
  49
  50 namespace internal {
  51
  52 void SetUnexportableKeyProviderForTesting(
  53     std::unique_ptr<UnexportableKeyProvider> (*func)()) {
  54   if (g_mock_provider) {
  55     // Nesting ScopedMockUnexportableSigningKeyForTesting is not supported.
  56     CHECK(!func);
  57     g_mock_provider = nullptr;
  58   } else {
  59     g_mock_provider = func;
  60   }
  61 }
  62
  63 }  // namespace internal
  64 }  // namespace crypto