components/ownership/mock_owner_key_util.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef COMPONENTS_OWNERSHIP_MOCK_OWNER_KEY_UTIL_H_
   6 #define COMPONENTS_OWNERSHIP_MOCK_OWNER_KEY_UTIL_H_
   7
   8 #include <vector>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/files/file_path.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "components/ownership/owner_key_util.h"
  15 #include "components/ownership/ownership_export.h"
  16
  17 namespace crypto {
  18 class RSAPrivateKey;
  19 }
  20
  21 namespace ownership {
  22
  23 // Implementation of OwnerKeyUtil which should be used only for
  24 // testing.
  25 class OWNERSHIP_EXPORT MockOwnerKeyUtil : public OwnerKeyUtil {
  26  public:
  27   MockOwnerKeyUtil();
  28
  29   // OwnerKeyUtil implementation:
  30   bool ImportPublicKey(std::vector<uint8>* output) override;
  31   crypto::ScopedSECKEYPrivateKey FindPrivateKeyInSlot(
  32       const std::vector<uint8>& key,
  33       PK11SlotInfo* slot) override;
  34   bool IsPublicKeyPresent() override;
  35
  36   // Clears the public and private keys.
  37   void Clear();
  38
  39   // Configures the mock to return the given public key.
  40   void SetPublicKey(const std::vector<uint8>& key);
  41
  42   // Sets the public key to use from the given private key, but doesn't
  43   // configure the private key.
  44   void SetPublicKeyFromPrivateKey(const crypto::RSAPrivateKey& key);
  45
  46   // Sets the private key (also configures the public key).
  47   void SetPrivateKey(scoped_ptr<crypto::RSAPrivateKey> key);
  48
  49  private:
  50   ~MockOwnerKeyUtil() override;
  51
  52   std::vector<uint8> public_key_;
  53   crypto::ScopedSECKEYPrivateKey private_key_;
  54
  55   DISALLOW_COPY_AND_ASSIGN(MockOwnerKeyUtil);
  56 };
  57
  58 }  // namespace ownership
  59
  60 #endif  // COMPONENTS_OWNERSHIP_MOCK_OWNER_KEY_UTIL_H_