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