1 // 2 // Copyright (C) 2015 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef ATTESTATION_COMMON_MOCK_TPM_UTILITY_H_ 18 #define ATTESTATION_COMMON_MOCK_TPM_UTILITY_H_ 19 20 #include "attestation/common/tpm_utility.h" 21 22 #include <string> 23 24 #include <gmock/gmock.h> 25 26 namespace attestation { 27 28 class MockTpmUtility : public TpmUtility { 29 public: 30 MockTpmUtility(); 31 ~MockTpmUtility() override; 32 // By default this class will fake seal/unbind/sign operations by passing the 33 // input through Transform(<method>). E.g. The expected output of a fake Sign 34 // operation on "foo" can be computed by calling 35 // MockTpmUtility::Transform("Sign", "foo"). 36 static std::string Transform(const std::string& method, 37 const std::string& input); 38 39 MOCK_METHOD0(IsTpmReady, bool()); 40 MOCK_METHOD6(ActivateIdentity, 41 bool(const std::string&, 42 const std::string&, 43 const std::string&, 44 const std::string&, 45 const std::string&, 46 std::string*)); 47 MOCK_METHOD9(CreateCertifiedKey, 48 bool(KeyType, 49 KeyUsage, 50 const std::string&, 51 const std::string&, 52 std::string*, 53 std::string*, 54 std::string*, 55 std::string*, 56 std::string*)); 57 MOCK_METHOD2(SealToPCR0, bool(const std::string&, std::string*)); 58 MOCK_METHOD2(Unseal, bool(const std::string&, std::string*)); 59 MOCK_METHOD1(GetEndorsementPublicKey, bool(std::string*)); 60 MOCK_METHOD3(Unbind, 61 bool(const std::string&, const std::string&, std::string*)); 62 MOCK_METHOD3(Sign, 63 bool(const std::string&, const std::string&, std::string*)); 64 }; 65 66 } // namespace attestation 67 68 #endif // ATTESTATION_COMMON_MOCK_TPM_UTILITY_H_ 69