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#include "attestation/common/mock_tpm_utility.h" 18 19using ::testing::_; 20using ::testing::Invoke; 21using ::testing::Return; 22using ::testing::WithArgs; 23 24namespace { 25 26class TransformString { 27 public: 28 explicit TransformString(std::string method) : method_(method) {} 29 bool operator()(const std::string& in, std::string* out) { 30 *out = attestation::MockTpmUtility::Transform(method_, in); 31 return true; 32 } 33 34 private: 35 std::string method_; 36}; 37 38class UntransformString { 39 public: 40 explicit UntransformString(std::string method) : method_(method) {} 41 bool operator()(const std::string& in, std::string* out) { 42 std::string suffix = "_fake_transform_" + method_; 43 auto position = in.find(suffix); 44 if (position == std::string::npos) { 45 return false; 46 } 47 *out = in.substr(0, position); 48 return true; 49 } 50 51 private: 52 std::string method_; 53}; 54 55} // namespace 56 57namespace attestation { 58 59MockTpmUtility::MockTpmUtility() { 60 ON_CALL(*this, IsTpmReady()).WillByDefault(Return(true)); 61 ON_CALL(*this, ActivateIdentity(_, _, _, _, _, _)) 62 .WillByDefault(Return(true)); 63 ON_CALL(*this, CreateCertifiedKey(_, _, _, _, _, _, _, _, _)) 64 .WillByDefault(Return(true)); 65 ON_CALL(*this, SealToPCR0(_, _)) 66 .WillByDefault(Invoke(TransformString("SealToPCR0"))); 67 ON_CALL(*this, Unseal(_, _)) 68 .WillByDefault(Invoke(UntransformString("SealToPCR0"))); 69 ON_CALL(*this, Unbind(_, _, _)) 70 .WillByDefault(WithArgs<1, 2>(Invoke(TransformString("Unbind")))); 71 ON_CALL(*this, Sign(_, _, _)) 72 .WillByDefault(WithArgs<1, 2>(Invoke(TransformString("Sign")))); 73} 74 75MockTpmUtility::~MockTpmUtility() {} 76 77// static 78std::string MockTpmUtility::Transform(const std::string& method, 79 const std::string& input) { 80 return input + "_fake_transform_" + method; 81} 82 83} // namespace attestation 84