OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ | 5 #ifndef CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ |
6 #define CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ | 6 #define CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ |
7 | 7 |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include <string> | 10 #include <string> |
11 | 11 |
| 12 #include "chromeos/cryptohome/cryptohome_parameters.h" |
12 #include "chromeos/dbus/cryptohome/rpc.pb.h" | 13 #include "chromeos/dbus/cryptohome/rpc.pb.h" |
13 #include "chromeos/dbus/cryptohome_client.h" | 14 #include "chromeos/dbus/cryptohome_client.h" |
14 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
15 | 16 |
16 namespace chromeos { | 17 namespace chromeos { |
17 | 18 |
18 class MockCryptohomeClient : public CryptohomeClient { | 19 class MockCryptohomeClient : public CryptohomeClient { |
19 public: | 20 public: |
20 MockCryptohomeClient(); | 21 MockCryptohomeClient(); |
21 virtual ~MockCryptohomeClient(); | 22 virtual ~MockCryptohomeClient(); |
22 | 23 |
23 MOCK_METHOD1(Init, void(dbus::Bus* bus)); | 24 MOCK_METHOD1(Init, void(dbus::Bus* bus)); |
24 MOCK_METHOD2(SetAsyncCallStatusHandlers, | 25 MOCK_METHOD2(SetAsyncCallStatusHandlers, |
25 void(const AsyncCallStatusHandler& handler, | 26 void(const AsyncCallStatusHandler& handler, |
26 const AsyncCallStatusWithDataHandler& data_handler)); | 27 const AsyncCallStatusWithDataHandler& data_handler)); |
27 MOCK_METHOD0(ResetAsyncCallStatusHandlers, void()); | 28 MOCK_METHOD0(ResetAsyncCallStatusHandlers, void()); |
28 MOCK_METHOD1(WaitForServiceToBeAvailable, | 29 MOCK_METHOD1(WaitForServiceToBeAvailable, |
29 void(const WaitForServiceToBeAvailableCallback& callback)); | 30 void(const WaitForServiceToBeAvailableCallback& callback)); |
30 MOCK_METHOD1(IsMounted, void(const BoolDBusMethodCallback& callback)); | 31 MOCK_METHOD1(IsMounted, void(const BoolDBusMethodCallback& callback)); |
31 MOCK_METHOD1(Unmount, bool(bool* success)); | 32 MOCK_METHOD1(Unmount, bool(bool* success)); |
32 MOCK_METHOD3(AsyncCheckKey, | 33 MOCK_METHOD3(AsyncCheckKey, |
33 void(const std::string& username, | 34 void(const cryptohome::Identification& cryptohome_id, |
34 const std::string& key, | 35 const std::string& key, |
35 const AsyncMethodCallback& callback)); | 36 const AsyncMethodCallback& callback)); |
36 MOCK_METHOD4(AsyncMigrateKey, | 37 MOCK_METHOD4(AsyncMigrateKey, |
37 void(const std::string& username, | 38 void(const cryptohome::Identification& cryptohome_id, |
38 const std::string& from_key, | 39 const std::string& from_key, |
39 const std::string& to_key, | 40 const std::string& to_key, |
40 const AsyncMethodCallback& callback)); | 41 const AsyncMethodCallback& callback)); |
41 MOCK_METHOD2(AsyncRemove, void(const std::string& username, | 42 MOCK_METHOD2(AsyncRemove, |
42 const AsyncMethodCallback& callback)); | 43 void(const cryptohome::Identification& cryptohome_id, |
| 44 const AsyncMethodCallback& callback)); |
| 45 |
43 MOCK_METHOD1(GetSystemSalt, void(const GetSystemSaltCallback& callback)); | 46 MOCK_METHOD1(GetSystemSalt, void(const GetSystemSaltCallback& callback)); |
44 MOCK_METHOD2(GetSanitizedUsername, | 47 MOCK_METHOD2(GetSanitizedUsername, |
45 void(const std::string& username, | 48 void(const cryptohome::Identification& cryptohome_id, |
46 const StringDBusMethodCallback& callback)); | 49 const StringDBusMethodCallback& callback)); |
47 MOCK_METHOD1(BlockingGetSanitizedUsername, | 50 MOCK_METHOD1(BlockingGetSanitizedUsername, |
48 std::string(const std::string& username)); | 51 std::string(const cryptohome::Identification& cryptohome_id)); |
49 MOCK_METHOD4(AsyncMount, void(const std::string& username, | 52 MOCK_METHOD4(AsyncMount, |
50 const std::string& key, | 53 void(const cryptohome::Identification& cryptohome_id, |
51 int flags, | 54 const std::string& key, |
52 const AsyncMethodCallback& callback)); | 55 int flags, |
53 MOCK_METHOD4(AsyncAddKey, void(const std::string& username, | 56 const AsyncMethodCallback& callback)); |
54 const std::string& key, | 57 MOCK_METHOD4(AsyncAddKey, |
55 const std::string& new_key, | 58 void(const cryptohome::Identification& cryptohome_id, |
56 const AsyncMethodCallback& callback)); | 59 const std::string& key, |
| 60 const std::string& new_key, |
| 61 const AsyncMethodCallback& callback)); |
57 MOCK_METHOD1(AsyncMountGuest, | 62 MOCK_METHOD1(AsyncMountGuest, |
58 void(const AsyncMethodCallback& callback)); | 63 void(const AsyncMethodCallback& callback)); |
59 MOCK_METHOD3(AsyncMountPublic, void(const std::string& public_mount_id, | 64 MOCK_METHOD3(AsyncMountPublic, |
60 int flags, | 65 void(const cryptohome::Identification& public_mount_id, |
61 const AsyncMethodCallback& callback)); | 66 int flags, |
| 67 const AsyncMethodCallback& callback)); |
62 MOCK_METHOD1(TpmIsReady, void(const BoolDBusMethodCallback& callback)); | 68 MOCK_METHOD1(TpmIsReady, void(const BoolDBusMethodCallback& callback)); |
63 MOCK_METHOD1(TpmIsEnabled, void(const BoolDBusMethodCallback& callback)); | 69 MOCK_METHOD1(TpmIsEnabled, void(const BoolDBusMethodCallback& callback)); |
64 MOCK_METHOD1(CallTpmIsEnabledAndBlock, bool(bool* enabled)); | 70 MOCK_METHOD1(CallTpmIsEnabledAndBlock, bool(bool* enabled)); |
65 MOCK_METHOD1(TpmGetPassword, void(const StringDBusMethodCallback& callback)); | 71 MOCK_METHOD1(TpmGetPassword, void(const StringDBusMethodCallback& callback)); |
66 MOCK_METHOD1(TpmIsOwned, void(const BoolDBusMethodCallback& callback)); | 72 MOCK_METHOD1(TpmIsOwned, void(const BoolDBusMethodCallback& callback)); |
67 MOCK_METHOD1(CallTpmIsOwnedAndBlock, bool(bool* owned)); | 73 MOCK_METHOD1(CallTpmIsOwnedAndBlock, bool(bool* owned)); |
68 MOCK_METHOD1(TpmIsBeingOwned, void(const BoolDBusMethodCallback& callback)); | 74 MOCK_METHOD1(TpmIsBeingOwned, void(const BoolDBusMethodCallback& callback)); |
69 MOCK_METHOD1(CallTpmIsBeingOwnedAndBlock, bool(bool* owning)); | 75 MOCK_METHOD1(CallTpmIsBeingOwnedAndBlock, bool(bool* owning)); |
70 MOCK_METHOD1(TpmCanAttemptOwnership, | 76 MOCK_METHOD1(TpmCanAttemptOwnership, |
71 void(const VoidDBusMethodCallback& callback)); | 77 void(const VoidDBusMethodCallback& callback)); |
72 MOCK_METHOD1(TpmClearStoredPassword, | 78 MOCK_METHOD1(TpmClearStoredPassword, |
73 void(const VoidDBusMethodCallback& callback)); | 79 void(const VoidDBusMethodCallback& callback)); |
74 MOCK_METHOD0(CallTpmClearStoredPasswordAndBlock, bool()); | 80 MOCK_METHOD0(CallTpmClearStoredPasswordAndBlock, bool()); |
75 MOCK_METHOD1(Pkcs11IsTpmTokenReady, | 81 MOCK_METHOD1(Pkcs11IsTpmTokenReady, |
76 void(const BoolDBusMethodCallback& callback)); | 82 void(const BoolDBusMethodCallback& callback)); |
77 MOCK_METHOD1(Pkcs11GetTpmTokenInfo, | 83 MOCK_METHOD1(Pkcs11GetTpmTokenInfo, |
78 void(const Pkcs11GetTpmTokenInfoCallback& callback)); | 84 void(const Pkcs11GetTpmTokenInfoCallback& callback)); |
79 MOCK_METHOD2(Pkcs11GetTpmTokenInfoForUser, | 85 MOCK_METHOD2(Pkcs11GetTpmTokenInfoForUser, |
80 void(const std::string& username, | 86 void(const cryptohome::Identification& cryptohome_id, |
81 const Pkcs11GetTpmTokenInfoCallback& callback)); | 87 const Pkcs11GetTpmTokenInfoCallback& callback)); |
82 MOCK_METHOD3(InstallAttributesGet, | 88 MOCK_METHOD3(InstallAttributesGet, |
83 bool(const std::string& name, | 89 bool(const std::string& name, |
84 std::vector<uint8_t>* value, | 90 std::vector<uint8_t>* value, |
85 bool* successful)); | 91 bool* successful)); |
86 MOCK_METHOD3(InstallAttributesSet, | 92 MOCK_METHOD3(InstallAttributesSet, |
87 bool(const std::string& name, | 93 bool(const std::string& name, |
88 const std::vector<uint8_t>& value, | 94 const std::vector<uint8_t>& value, |
89 bool* successful)); | 95 bool* successful)); |
90 MOCK_METHOD1(InstallAttributesFinalize, bool(bool* successful)); | 96 MOCK_METHOD1(InstallAttributesFinalize, bool(bool* successful)); |
91 MOCK_METHOD1(InstallAttributesIsReady, | 97 MOCK_METHOD1(InstallAttributesIsReady, |
92 void(const BoolDBusMethodCallback& callback)); | 98 void(const BoolDBusMethodCallback& callback)); |
93 MOCK_METHOD1(InstallAttributesIsInvalid, bool(bool* is_invalid)); | 99 MOCK_METHOD1(InstallAttributesIsInvalid, bool(bool* is_invalid)); |
94 MOCK_METHOD1(InstallAttributesIsFirstInstall, bool(bool* is_first_install)); | 100 MOCK_METHOD1(InstallAttributesIsFirstInstall, bool(bool* is_first_install)); |
95 MOCK_METHOD1(TpmAttestationIsPrepared, | 101 MOCK_METHOD1(TpmAttestationIsPrepared, |
96 void(const BoolDBusMethodCallback& callback)); | 102 void(const BoolDBusMethodCallback& callback)); |
97 MOCK_METHOD1(TpmAttestationIsEnrolled, | 103 MOCK_METHOD1(TpmAttestationIsEnrolled, |
98 void(const BoolDBusMethodCallback& callback)); | 104 void(const BoolDBusMethodCallback& callback)); |
99 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, | 105 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, |
100 void(attestation::PrivacyCAType pca_type, | 106 void(attestation::PrivacyCAType pca_type, |
101 const AsyncMethodCallback& callback)); | 107 const AsyncMethodCallback& callback)); |
102 MOCK_METHOD3(AsyncTpmAttestationEnroll, | 108 MOCK_METHOD3(AsyncTpmAttestationEnroll, |
103 void(attestation::PrivacyCAType pca_type, | 109 void(attestation::PrivacyCAType pca_type, |
104 const std::string& pca_response, | 110 const std::string& pca_response, |
105 const AsyncMethodCallback& callback)); | 111 const AsyncMethodCallback& callback)); |
106 MOCK_METHOD5( | 112 MOCK_METHOD5( |
107 AsyncTpmAttestationCreateCertRequest, | 113 AsyncTpmAttestationCreateCertRequest, |
108 void(attestation::PrivacyCAType pca_type, | 114 void(attestation::PrivacyCAType pca_type, |
109 attestation::AttestationCertificateProfile certificate_profile, | 115 attestation::AttestationCertificateProfile certificate_profile, |
110 const std::string& user_id, | 116 const cryptohome::Identification& cryptohome_id, |
111 const std::string& request_origin, | 117 const std::string& request_origin, |
112 const AsyncMethodCallback& callback)); | 118 const AsyncMethodCallback& callback)); |
113 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, | 119 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, |
114 void(const std::string& pca_response, | 120 void(const std::string& pca_response, |
115 attestation::AttestationKeyType key_type, | 121 attestation::AttestationKeyType key_type, |
116 const std::string& user_id, | 122 const cryptohome::Identification& cryptohome_id, |
117 const std::string& key_name, | 123 const std::string& key_name, |
118 const AsyncMethodCallback& callback)); | 124 const AsyncMethodCallback& callback)); |
119 MOCK_METHOD4(TpmAttestationDoesKeyExist, | 125 MOCK_METHOD4(TpmAttestationDoesKeyExist, |
120 void(attestation::AttestationKeyType key_type, | 126 void(attestation::AttestationKeyType key_type, |
121 const std::string& user_id, | 127 const cryptohome::Identification& cryptohome_id, |
122 const std::string& key_name, | 128 const std::string& key_name, |
123 const BoolDBusMethodCallback& callback)); | 129 const BoolDBusMethodCallback& callback)); |
124 MOCK_METHOD4(TpmAttestationGetCertificate, | 130 MOCK_METHOD4(TpmAttestationGetCertificate, |
125 void(attestation::AttestationKeyType key_type, | 131 void(attestation::AttestationKeyType key_type, |
126 const std::string& user_id, | 132 const cryptohome::Identification& cryptohome_id, |
127 const std::string& key_name, | 133 const std::string& key_name, |
128 const DataMethodCallback& callback)); | 134 const DataMethodCallback& callback)); |
129 MOCK_METHOD4(TpmAttestationGetPublicKey, | 135 MOCK_METHOD4(TpmAttestationGetPublicKey, |
130 void(attestation::AttestationKeyType key_type, | 136 void(attestation::AttestationKeyType key_type, |
131 const std::string& user_id, | 137 const cryptohome::Identification& cryptohome_id, |
132 const std::string& key_name, | 138 const std::string& key_name, |
133 const DataMethodCallback& callback)); | 139 const DataMethodCallback& callback)); |
134 MOCK_METHOD4(TpmAttestationRegisterKey, | 140 MOCK_METHOD4(TpmAttestationRegisterKey, |
135 void(attestation::AttestationKeyType key_type, | 141 void(attestation::AttestationKeyType key_type, |
136 const std::string& user_id, | 142 const cryptohome::Identification& cryptohome_id, |
137 const std::string& key_name, | 143 const std::string& key_name, |
138 const AsyncMethodCallback& callback)); | 144 const AsyncMethodCallback& callback)); |
139 MOCK_METHOD8(TpmAttestationSignEnterpriseChallenge, | 145 MOCK_METHOD8(TpmAttestationSignEnterpriseChallenge, |
140 void(attestation::AttestationKeyType key_type, | 146 void(attestation::AttestationKeyType key_type, |
141 const std::string& user_id, | 147 const cryptohome::Identification& cryptohome_id, |
142 const std::string& key_name, | 148 const std::string& key_name, |
143 const std::string& domain, | 149 const std::string& domain, |
144 const std::string& device_id, | 150 const std::string& device_id, |
145 attestation::AttestationChallengeOptions options, | 151 attestation::AttestationChallengeOptions options, |
146 const std::string& challenge, | 152 const std::string& challenge, |
147 const AsyncMethodCallback& callback)); | 153 const AsyncMethodCallback& callback)); |
148 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, | 154 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, |
149 void(attestation::AttestationKeyType key_type, | 155 void(attestation::AttestationKeyType key_type, |
150 const std::string& user_id, | 156 const cryptohome::Identification& cryptohome_id, |
151 const std::string& key_name, | 157 const std::string& key_name, |
152 const std::string& challenge, | 158 const std::string& challenge, |
153 const AsyncMethodCallback& callback)); | 159 const AsyncMethodCallback& callback)); |
154 MOCK_METHOD4(TpmAttestationGetKeyPayload, | 160 MOCK_METHOD4(TpmAttestationGetKeyPayload, |
155 void(attestation::AttestationKeyType key_type, | 161 void(attestation::AttestationKeyType key_type, |
156 const std::string& user_id, | 162 const cryptohome::Identification& cryptohome_id, |
157 const std::string& key_name, | 163 const std::string& key_name, |
158 const DataMethodCallback& callback)); | 164 const DataMethodCallback& callback)); |
159 MOCK_METHOD5(TpmAttestationSetKeyPayload, | 165 MOCK_METHOD5(TpmAttestationSetKeyPayload, |
160 void(attestation::AttestationKeyType key_type, | 166 void(attestation::AttestationKeyType key_type, |
161 const std::string& user_id, | 167 const cryptohome::Identification& cryptohome_id, |
162 const std::string& key_name, | 168 const std::string& key_name, |
163 const std::string& payload, | 169 const std::string& payload, |
164 const BoolDBusMethodCallback& callback)); | 170 const BoolDBusMethodCallback& callback)); |
165 MOCK_METHOD4(TpmAttestationDeleteKeys, | 171 MOCK_METHOD4(TpmAttestationDeleteKeys, |
166 void(attestation::AttestationKeyType key_type, | 172 void(attestation::AttestationKeyType key_type, |
167 const std::string& user_id, | 173 const cryptohome::Identification& cryptohome_id, |
168 const std::string& key_prefix, | 174 const std::string& key_prefix, |
169 const BoolDBusMethodCallback& callback)); | 175 const BoolDBusMethodCallback& callback)); |
170 MOCK_METHOD4(GetKeyDataEx, | 176 MOCK_METHOD4(GetKeyDataEx, |
171 void(const cryptohome::AccountIdentifier& id, | 177 void(const cryptohome::Identification& cryptohome_id, |
172 const cryptohome::AuthorizationRequest& auth, | 178 const cryptohome::AuthorizationRequest& auth, |
173 const cryptohome::GetKeyDataRequest& request, | 179 const cryptohome::GetKeyDataRequest& request, |
174 const ProtobufMethodCallback& callback)); | 180 const ProtobufMethodCallback& callback)); |
175 MOCK_METHOD4(CheckKeyEx, | 181 MOCK_METHOD4(CheckKeyEx, |
176 void(const cryptohome::AccountIdentifier& id, | 182 void(const cryptohome::Identification& cryptohome_id, |
177 const cryptohome::AuthorizationRequest& auth, | 183 const cryptohome::AuthorizationRequest& auth, |
178 const cryptohome::CheckKeyRequest& request, | 184 const cryptohome::CheckKeyRequest& request, |
179 const ProtobufMethodCallback& callback)); | 185 const ProtobufMethodCallback& callback)); |
180 MOCK_METHOD4(MountEx, | 186 MOCK_METHOD4(MountEx, |
181 void(const cryptohome::AccountIdentifier& id, | 187 void(const cryptohome::Identification& cryptohome_id, |
182 const cryptohome::AuthorizationRequest& auth, | 188 const cryptohome::AuthorizationRequest& auth, |
183 const cryptohome::MountRequest& request, | 189 const cryptohome::MountRequest& request, |
184 const ProtobufMethodCallback& callback)); | 190 const ProtobufMethodCallback& callback)); |
185 MOCK_METHOD4(AddKeyEx, | 191 MOCK_METHOD4(AddKeyEx, |
186 void(const cryptohome::AccountIdentifier& id, | 192 void(const cryptohome::Identification& cryptohome_id, |
187 const cryptohome::AuthorizationRequest& auth, | 193 const cryptohome::AuthorizationRequest& auth, |
188 const cryptohome::AddKeyRequest& request, | 194 const cryptohome::AddKeyRequest& request, |
189 const ProtobufMethodCallback& callback)); | 195 const ProtobufMethodCallback& callback)); |
190 MOCK_METHOD4(UpdateKeyEx, | 196 MOCK_METHOD4(UpdateKeyEx, |
191 void(const cryptohome::AccountIdentifier& id, | 197 void(const cryptohome::Identification& cryptohome_id, |
192 const cryptohome::AuthorizationRequest& auth, | 198 const cryptohome::AuthorizationRequest& auth, |
193 const cryptohome::UpdateKeyRequest& request, | 199 const cryptohome::UpdateKeyRequest& request, |
194 const ProtobufMethodCallback& callback)); | 200 const ProtobufMethodCallback& callback)); |
195 MOCK_METHOD4(RemoveKeyEx, | 201 MOCK_METHOD4(RemoveKeyEx, |
196 void(const cryptohome::AccountIdentifier& id, | 202 void(const cryptohome::Identification& cryptohome_id, |
197 const cryptohome::AuthorizationRequest& auth, | 203 const cryptohome::AuthorizationRequest& auth, |
198 const cryptohome::RemoveKeyRequest& request, | 204 const cryptohome::RemoveKeyRequest& request, |
199 const ProtobufMethodCallback& callback)); | 205 const ProtobufMethodCallback& callback)); |
200 MOCK_METHOD2(GetBootAttribute, | 206 MOCK_METHOD2(GetBootAttribute, |
201 void(const cryptohome::GetBootAttributeRequest& request, | 207 void(const cryptohome::GetBootAttributeRequest& request, |
202 const ProtobufMethodCallback& callback)); | 208 const ProtobufMethodCallback& callback)); |
203 MOCK_METHOD2(SetBootAttribute, | 209 MOCK_METHOD2(SetBootAttribute, |
204 void(const cryptohome::SetBootAttributeRequest& request, | 210 void(const cryptohome::SetBootAttributeRequest& request, |
205 const ProtobufMethodCallback& callback)); | 211 const ProtobufMethodCallback& callback)); |
206 MOCK_METHOD2( | 212 MOCK_METHOD2( |
207 FlushAndSignBootAttributes, | 213 FlushAndSignBootAttributes, |
208 void(const cryptohome::FlushAndSignBootAttributesRequest& request, | 214 void(const cryptohome::FlushAndSignBootAttributesRequest& request, |
209 const ProtobufMethodCallback& callback)); | 215 const ProtobufMethodCallback& callback)); |
210 }; | 216 }; |
211 | 217 |
212 } // namespace chromeos | 218 } // namespace chromeos |
213 | 219 |
214 #endif // CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ | 220 #endif // CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ |
OLD | NEW |