| // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CHAPS_CHAPS_PROXY_MOCK_H_ |
| #define CHAPS_CHAPS_PROXY_MOCK_H_ |
| |
| #include <string> |
| #include <vector> |
| |
| #include <brillo/secure_blob.h> |
| #include <chaps/proto_bindings/ck_structs.pb.h> |
| #include <gmock/gmock.h> |
| #include <gtest/gtest.h> |
| |
| #include "chaps/chaps_interface.h" |
| #include "chaps/isolate.h" |
| |
| namespace chaps { |
| |
| // Defined in chaps.cc. |
| extern void EnableMockProxy(ChapsInterface* proxy, |
| brillo::SecureBlob* isolate_credential, |
| bool is_initialized); |
| extern void DisableMockProxy(); |
| extern void SetRetryTimeParameters(uint32_t timeout_ms, uint32_t delay_ms); |
| |
| // ChapsProxyMock is a mock of ChapsInterface. |
| class ChapsProxyMock : public ChapsInterface { |
| public: |
| explicit ChapsProxyMock(bool is_initialized) |
| : isolate_credential_( |
| IsolateCredentialManager::GetDefaultIsolateCredential()) { |
| EnableMockProxy(this, &isolate_credential_, is_initialized); |
| } |
| ChapsProxyMock(const ChapsProxyMock&) = delete; |
| ChapsProxyMock& operator=(const ChapsProxyMock&) = delete; |
| |
| ~ChapsProxyMock() override { DisableMockProxy(); } |
| |
| MOCK_METHOD3(GetSlotList, |
| uint32_t(const brillo::SecureBlob&, |
| bool, |
| std::vector<uint64_t>*)); |
| MOCK_METHOD3(GetSlotInfo, |
| uint32_t(const brillo::SecureBlob&, uint64_t, SlotInfo*)); |
| MOCK_METHOD3(GetTokenInfo, |
| uint32_t(const brillo::SecureBlob&, uint64_t, TokenInfo*)); |
| MOCK_METHOD3(GetMechanismList, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| std::vector<uint64_t>*)); |
| MOCK_METHOD4( |
| GetMechanismInfo, |
| uint32_t(const brillo::SecureBlob&, uint64_t, uint64_t, MechanismInfo*)); |
| MOCK_METHOD4(InitToken, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::string*, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD3(InitPIN, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::string*)); |
| MOCK_METHOD4(SetPIN, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::string*, |
| const std::string*)); |
| MOCK_METHOD4( |
| OpenSession, |
| uint32_t(const brillo::SecureBlob&, uint64_t, uint64_t, uint64_t*)); |
| MOCK_METHOD2(CloseSession, uint32_t(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD2(CloseAllSessions, uint32_t(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD3(GetSessionInfo, |
| uint32_t(const brillo::SecureBlob&, uint64_t, SessionInfo*)); |
| MOCK_METHOD3(GetOperationState, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD5(SetOperationState, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t)); |
| MOCK_METHOD4(Login, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::string*)); |
| MOCK_METHOD2(Logout, uint32_t(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD4(CreateObject, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t*)); |
| MOCK_METHOD5(CopyObject, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t*)); |
| MOCK_METHOD3(DestroyObject, |
| uint32_t(const brillo::SecureBlob&, uint64_t, uint64_t)); |
| MOCK_METHOD4( |
| GetObjectSize, |
| uint32_t(const brillo::SecureBlob&, uint64_t, uint64_t, uint64_t*)); |
| MOCK_METHOD5(GetAttributeValue, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD4(SetAttributeValue, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD3(FindObjectsInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD4(FindObjects, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| std::vector<uint64_t>*)); |
| MOCK_METHOD2(FindObjectsFinal, uint32_t(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD5(EncryptInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t key_handle)); |
| MOCK_METHOD6(Encrypt, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(EncryptUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD5(EncryptFinal, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD2(EncryptCancel, void(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD5(DecryptInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t)); |
| MOCK_METHOD6(Decrypt, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(DecryptUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD5(DecryptFinal, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD2(DecryptCancel, void(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD4(DigestInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD6(Digest, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD3(DigestUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD3(DigestKey, |
| uint32_t(const brillo::SecureBlob&, uint64_t, uint64_t)); |
| MOCK_METHOD5(DigestFinal, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD2(DigestCancel, void(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD5(SignInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t)); |
| MOCK_METHOD6(Sign, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD3(SignUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD5(SignFinal, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD2(SignCancel, void(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD5(SignRecoverInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t)); |
| MOCK_METHOD6(SignRecover, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD5(VerifyInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t)); |
| MOCK_METHOD4(Verify, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD3(VerifyUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD3(VerifyFinal, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD2(VerifyCancel, void(const brillo::SecureBlob&, uint64_t)); |
| MOCK_METHOD5(VerifyRecoverInit, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t)); |
| MOCK_METHOD6(VerifyRecover, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(DigestEncryptUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(DecryptDigestUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(SignEncryptUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(DecryptVerifyUpdate, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD6(GenerateKey, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| const std::vector<uint8_t>&, |
| uint64_t*)); |
| MOCK_METHOD8(GenerateKeyPair, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| const std::vector<uint8_t>&, |
| const std::vector<uint8_t>&, |
| uint64_t*, |
| uint64_t*)); |
| MOCK_METHOD9(WrapKey, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| uint64_t, |
| uint64_t, |
| uint64_t*, |
| std::vector<uint8_t>*)); |
| MOCK_METHOD8(UnwrapKey, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| const std::vector<uint8_t>&, |
| uint64_t*)); |
| MOCK_METHOD7(DeriveKey, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t, |
| const std::vector<uint8_t>&, |
| uint64_t*)); |
| MOCK_METHOD3(SeedRandom, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| const std::vector<uint8_t>&)); |
| MOCK_METHOD4(GenerateRandom, |
| uint32_t(const brillo::SecureBlob&, |
| uint64_t, |
| uint64_t, |
| std::vector<uint8_t>*)); |
| |
| private: |
| brillo::SecureBlob isolate_credential_; |
| }; |
| |
| } // namespace chaps |
| |
| #endif // CHAPS_CHAPS_PROXY_MOCK_H_ |
