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 TPM_MANAGER_SERVER_MOCK_TPM_NVRAM_H_ 18 #define TPM_MANAGER_SERVER_MOCK_TPM_NVRAM_H_ 19 20 #include "tpm_manager/server/tpm_nvram.h" 21 22 #include <map> 23 #include <string> 24 25 #include <gmock/gmock.h> 26 27 namespace tpm_manager { 28 29 struct NvSpace { 30 std::string data; 31 bool read_locked; 32 bool write_locked; 33 std::vector<NvramSpaceAttribute> attributes; 34 std::string authorization_value; 35 NvramSpacePolicy policy; 36 }; 37 38 class MockTpmNvram : public TpmNvram { 39 public: 40 MockTpmNvram(); 41 ~MockTpmNvram() override; 42 43 MOCK_METHOD5(DefineSpace, 44 NvramResult(uint32_t, 45 size_t, 46 const std::vector<NvramSpaceAttribute>&, 47 const std::string&, 48 NvramSpacePolicy)); 49 MOCK_METHOD1(DestroySpace, NvramResult(uint32_t)); 50 MOCK_METHOD3(WriteSpace, 51 NvramResult(uint32_t, const std::string&, const std::string&)); 52 MOCK_METHOD3(ReadSpace, 53 NvramResult(uint32_t, std::string*, const std::string&)); 54 MOCK_METHOD4(LockSpace, 55 NvramResult(uint32_t, bool, bool, const std::string&)); 56 MOCK_METHOD1(ListSpaces, NvramResult(std::vector<uint32_t>*)); 57 MOCK_METHOD6(GetSpaceInfo, 58 NvramResult(uint32_t, 59 size_t*, 60 bool*, 61 bool*, 62 std::vector<NvramSpaceAttribute>*, 63 NvramSpacePolicy*)); 64 65 private: 66 NvramResult FakeDefineSpace( 67 uint32_t index, 68 size_t size, 69 const std::vector<NvramSpaceAttribute>& attributes, 70 const std::string& authorization_value, 71 NvramSpacePolicy policy); 72 NvramResult FakeDestroySpace(uint32_t index); 73 NvramResult FakeWriteSpace(uint32_t index, 74 const std::string& data, 75 const std::string& authorization_value); 76 NvramResult FakeReadSpace(uint32_t index, 77 std::string* data, 78 const std::string& authorization_value); 79 NvramResult FakeLockSpace(uint32_t index, 80 bool lock_read, 81 bool lock_write, 82 const std::string& authorization_value); 83 NvramResult FakeListSpaces(std::vector<uint32_t>* index_list); 84 NvramResult FakeGetSpaceInfo(uint32_t index, 85 size_t* size, 86 bool* is_read_locked, 87 bool* is_write_locked, 88 std::vector<NvramSpaceAttribute>* attributes, 89 NvramSpacePolicy* policy); 90 91 std::map<uint32_t, NvSpace> nvram_map_; 92 }; 93 94 } // namespace tpm_manager 95 96 #endif // TPM_MANAGER_SERVER_MOCK_TPM_NVRAM_H_ 97
