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 
19 using ::testing::_;
20 using ::testing::Invoke;
21 using ::testing::Return;
22 using ::testing::WithArgs;
23 
24 namespace {
25 
26 class 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 
38 class 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 
57 namespace attestation {
58 
59 MockTpmUtility::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 
75 MockTpmUtility::~MockTpmUtility() {}
76 
77 // static
78 std::string MockTpmUtility::Transform(const std::string& method,
79                                       const std::string& input) {
80   return input + "_fake_transform_" + method;
81 }
82 
83 }  // namespace attestation
84