unexportable_key.cc 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  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. #include "crypto/unexportable_key.h"
  5. #include "base/bind.h"
  6. #include "base/check.h"
  7. #include "build/build_config.h"
  8. namespace crypto {
  9. namespace {
  10. std::unique_ptr<UnexportableKeyProvider> (*g_mock_provider)() = nullptr;
  11. } // namespace
  12. UnexportableSigningKey::~UnexportableSigningKey() = default;
  13. UnexportableKeyProvider::~UnexportableKeyProvider() = default;
  14. #if BUILDFLAG(IS_WIN)
  15. std::unique_ptr<UnexportableKeyProvider> GetUnexportableKeyProviderWin();
  16. #endif
  17. std::unique_ptr<UnexportableKeyProvider> GetUnexportableKeyProvider() {
  18. if (g_mock_provider) {
  19. return g_mock_provider();
  20. }
  21. #if BUILDFLAG(IS_WIN)
  22. return GetUnexportableKeyProviderWin();
  23. #else
  24. return nullptr;
  25. #endif
  26. }
  27. namespace internal {
  28. void SetUnexportableKeyProviderForTesting(
  29. std::unique_ptr<UnexportableKeyProvider> (*func)()) {
  30. if (g_mock_provider) {
  31. // Nesting ScopedMockUnexportableSigningKeyForTesting is not supported.
  32. CHECK(!func);
  33. g_mock_provider = nullptr;
  34. } else {
  35. g_mock_provider = func;
  36. }
  37. }
  38. } // namespace internal
  39. } // namespace crypto