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 MOCK_METHOD3(RenameCryptohome, |
| 46 void(const cryptohome::Identification& id_from, |
| 47 const cryptohome::Identification& id_to, |
| 48 const ProtobufMethodCallback& callback)); |
| 49 |
43 MOCK_METHOD1(GetSystemSalt, void(const GetSystemSaltCallback& callback)); | 50 MOCK_METHOD1(GetSystemSalt, void(const GetSystemSaltCallback& callback)); |
44 MOCK_METHOD2(GetSanitizedUsername, | 51 MOCK_METHOD2(GetSanitizedUsername, |
45 void(const std::string& username, | 52 void(const cryptohome::Identification& cryptohome_id, |
46 const StringDBusMethodCallback& callback)); | 53 const StringDBusMethodCallback& callback)); |
47 MOCK_METHOD1(BlockingGetSanitizedUsername, | 54 MOCK_METHOD1(BlockingGetSanitizedUsername, |
48 std::string(const std::string& username)); | 55 std::string(const cryptohome::Identification& cryptohome_id)); |
49 MOCK_METHOD4(AsyncMount, void(const std::string& username, | 56 MOCK_METHOD4(AsyncMount, |
50 const std::string& key, | 57 void(const cryptohome::Identification& cryptohome_id, |
51 int flags, | 58 const std::string& key, |
52 const AsyncMethodCallback& callback)); | 59 int flags, |
53 MOCK_METHOD4(AsyncAddKey, void(const std::string& username, | 60 const AsyncMethodCallback& callback)); |
54 const std::string& key, | 61 MOCK_METHOD4(AsyncAddKey, |
55 const std::string& new_key, | 62 void(const cryptohome::Identification& cryptohome_id, |
56 const AsyncMethodCallback& callback)); | 63 const std::string& key, |
| 64 const std::string& new_key, |
| 65 const AsyncMethodCallback& callback)); |
57 MOCK_METHOD1(AsyncMountGuest, | 66 MOCK_METHOD1(AsyncMountGuest, |
58 void(const AsyncMethodCallback& callback)); | 67 void(const AsyncMethodCallback& callback)); |
59 MOCK_METHOD3(AsyncMountPublic, void(const std::string& public_mount_id, | 68 MOCK_METHOD3(AsyncMountPublic, |
60 int flags, | 69 void(const cryptohome::Identification& public_mount_id, |
61 const AsyncMethodCallback& callback)); | 70 int flags, |
| 71 const AsyncMethodCallback& callback)); |
62 MOCK_METHOD1(TpmIsReady, void(const BoolDBusMethodCallback& callback)); | 72 MOCK_METHOD1(TpmIsReady, void(const BoolDBusMethodCallback& callback)); |
63 MOCK_METHOD1(TpmIsEnabled, void(const BoolDBusMethodCallback& callback)); | 73 MOCK_METHOD1(TpmIsEnabled, void(const BoolDBusMethodCallback& callback)); |
64 MOCK_METHOD1(CallTpmIsEnabledAndBlock, bool(bool* enabled)); | 74 MOCK_METHOD1(CallTpmIsEnabledAndBlock, bool(bool* enabled)); |
65 MOCK_METHOD1(TpmGetPassword, void(const StringDBusMethodCallback& callback)); | 75 MOCK_METHOD1(TpmGetPassword, void(const StringDBusMethodCallback& callback)); |
66 MOCK_METHOD1(TpmIsOwned, void(const BoolDBusMethodCallback& callback)); | 76 MOCK_METHOD1(TpmIsOwned, void(const BoolDBusMethodCallback& callback)); |
67 MOCK_METHOD1(CallTpmIsOwnedAndBlock, bool(bool* owned)); | 77 MOCK_METHOD1(CallTpmIsOwnedAndBlock, bool(bool* owned)); |
68 MOCK_METHOD1(TpmIsBeingOwned, void(const BoolDBusMethodCallback& callback)); | 78 MOCK_METHOD1(TpmIsBeingOwned, void(const BoolDBusMethodCallback& callback)); |
69 MOCK_METHOD1(CallTpmIsBeingOwnedAndBlock, bool(bool* owning)); | 79 MOCK_METHOD1(CallTpmIsBeingOwnedAndBlock, bool(bool* owning)); |
70 MOCK_METHOD1(TpmCanAttemptOwnership, | 80 MOCK_METHOD1(TpmCanAttemptOwnership, |
71 void(const VoidDBusMethodCallback& callback)); | 81 void(const VoidDBusMethodCallback& callback)); |
72 MOCK_METHOD1(TpmClearStoredPassword, | 82 MOCK_METHOD1(TpmClearStoredPassword, |
73 void(const VoidDBusMethodCallback& callback)); | 83 void(const VoidDBusMethodCallback& callback)); |
74 MOCK_METHOD0(CallTpmClearStoredPasswordAndBlock, bool()); | 84 MOCK_METHOD0(CallTpmClearStoredPasswordAndBlock, bool()); |
75 MOCK_METHOD1(Pkcs11IsTpmTokenReady, | 85 MOCK_METHOD1(Pkcs11IsTpmTokenReady, |
76 void(const BoolDBusMethodCallback& callback)); | 86 void(const BoolDBusMethodCallback& callback)); |
77 MOCK_METHOD1(Pkcs11GetTpmTokenInfo, | 87 MOCK_METHOD1(Pkcs11GetTpmTokenInfo, |
78 void(const Pkcs11GetTpmTokenInfoCallback& callback)); | 88 void(const Pkcs11GetTpmTokenInfoCallback& callback)); |
79 MOCK_METHOD2(Pkcs11GetTpmTokenInfoForUser, | 89 MOCK_METHOD2(Pkcs11GetTpmTokenInfoForUser, |
80 void(const std::string& username, | 90 void(const cryptohome::Identification& cryptohome_id, |
81 const Pkcs11GetTpmTokenInfoCallback& callback)); | 91 const Pkcs11GetTpmTokenInfoCallback& callback)); |
82 MOCK_METHOD3(InstallAttributesGet, | 92 MOCK_METHOD3(InstallAttributesGet, |
83 bool(const std::string& name, | 93 bool(const std::string& name, |
84 std::vector<uint8_t>* value, | 94 std::vector<uint8_t>* value, |
85 bool* successful)); | 95 bool* successful)); |
86 MOCK_METHOD3(InstallAttributesSet, | 96 MOCK_METHOD3(InstallAttributesSet, |
87 bool(const std::string& name, | 97 bool(const std::string& name, |
88 const std::vector<uint8_t>& value, | 98 const std::vector<uint8_t>& value, |
89 bool* successful)); | 99 bool* successful)); |
90 MOCK_METHOD1(InstallAttributesFinalize, bool(bool* successful)); | 100 MOCK_METHOD1(InstallAttributesFinalize, bool(bool* successful)); |
91 MOCK_METHOD1(InstallAttributesIsReady, | 101 MOCK_METHOD1(InstallAttributesIsReady, |
92 void(const BoolDBusMethodCallback& callback)); | 102 void(const BoolDBusMethodCallback& callback)); |
93 MOCK_METHOD1(InstallAttributesIsInvalid, bool(bool* is_invalid)); | 103 MOCK_METHOD1(InstallAttributesIsInvalid, bool(bool* is_invalid)); |
94 MOCK_METHOD1(InstallAttributesIsFirstInstall, bool(bool* is_first_install)); | 104 MOCK_METHOD1(InstallAttributesIsFirstInstall, bool(bool* is_first_install)); |
95 MOCK_METHOD1(TpmAttestationIsPrepared, | 105 MOCK_METHOD1(TpmAttestationIsPrepared, |
96 void(const BoolDBusMethodCallback& callback)); | 106 void(const BoolDBusMethodCallback& callback)); |
97 MOCK_METHOD1(TpmAttestationIsEnrolled, | 107 MOCK_METHOD1(TpmAttestationIsEnrolled, |
98 void(const BoolDBusMethodCallback& callback)); | 108 void(const BoolDBusMethodCallback& callback)); |
99 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, | 109 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, |
100 void(attestation::PrivacyCAType pca_type, | 110 void(attestation::PrivacyCAType pca_type, |
101 const AsyncMethodCallback& callback)); | 111 const AsyncMethodCallback& callback)); |
102 MOCK_METHOD3(AsyncTpmAttestationEnroll, | 112 MOCK_METHOD3(AsyncTpmAttestationEnroll, |
103 void(attestation::PrivacyCAType pca_type, | 113 void(attestation::PrivacyCAType pca_type, |
104 const std::string& pca_response, | 114 const std::string& pca_response, |
105 const AsyncMethodCallback& callback)); | 115 const AsyncMethodCallback& callback)); |
106 MOCK_METHOD5( | 116 MOCK_METHOD5( |
107 AsyncTpmAttestationCreateCertRequest, | 117 AsyncTpmAttestationCreateCertRequest, |
108 void(attestation::PrivacyCAType pca_type, | 118 void(attestation::PrivacyCAType pca_type, |
109 attestation::AttestationCertificateProfile certificate_profile, | 119 attestation::AttestationCertificateProfile certificate_profile, |
110 const std::string& user_id, | 120 const cryptohome::Identification& cryptohome_id, |
111 const std::string& request_origin, | 121 const std::string& request_origin, |
112 const AsyncMethodCallback& callback)); | 122 const AsyncMethodCallback& callback)); |
113 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, | 123 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, |
114 void(const std::string& pca_response, | 124 void(const std::string& pca_response, |
115 attestation::AttestationKeyType key_type, | 125 attestation::AttestationKeyType key_type, |
116 const std::string& user_id, | 126 const cryptohome::Identification& cryptohome_id, |
117 const std::string& key_name, | 127 const std::string& key_name, |
118 const AsyncMethodCallback& callback)); | 128 const AsyncMethodCallback& callback)); |
119 MOCK_METHOD4(TpmAttestationDoesKeyExist, | 129 MOCK_METHOD4(TpmAttestationDoesKeyExist, |
120 void(attestation::AttestationKeyType key_type, | 130 void(attestation::AttestationKeyType key_type, |
121 const std::string& user_id, | 131 const cryptohome::Identification& cryptohome_id, |
122 const std::string& key_name, | 132 const std::string& key_name, |
123 const BoolDBusMethodCallback& callback)); | 133 const BoolDBusMethodCallback& callback)); |
124 MOCK_METHOD4(TpmAttestationGetCertificate, | 134 MOCK_METHOD4(TpmAttestationGetCertificate, |
125 void(attestation::AttestationKeyType key_type, | 135 void(attestation::AttestationKeyType key_type, |
126 const std::string& user_id, | 136 const cryptohome::Identification& cryptohome_id, |
127 const std::string& key_name, | 137 const std::string& key_name, |
128 const DataMethodCallback& callback)); | 138 const DataMethodCallback& callback)); |
129 MOCK_METHOD4(TpmAttestationGetPublicKey, | 139 MOCK_METHOD4(TpmAttestationGetPublicKey, |
130 void(attestation::AttestationKeyType key_type, | 140 void(attestation::AttestationKeyType key_type, |
131 const std::string& user_id, | 141 const cryptohome::Identification& cryptohome_id, |
132 const std::string& key_name, | 142 const std::string& key_name, |
133 const DataMethodCallback& callback)); | 143 const DataMethodCallback& callback)); |
134 MOCK_METHOD4(TpmAttestationRegisterKey, | 144 MOCK_METHOD4(TpmAttestationRegisterKey, |
135 void(attestation::AttestationKeyType key_type, | 145 void(attestation::AttestationKeyType key_type, |
136 const std::string& user_id, | 146 const cryptohome::Identification& cryptohome_id, |
137 const std::string& key_name, | 147 const std::string& key_name, |
138 const AsyncMethodCallback& callback)); | 148 const AsyncMethodCallback& callback)); |
139 MOCK_METHOD8(TpmAttestationSignEnterpriseChallenge, | 149 MOCK_METHOD8(TpmAttestationSignEnterpriseChallenge, |
140 void(attestation::AttestationKeyType key_type, | 150 void(attestation::AttestationKeyType key_type, |
141 const std::string& user_id, | 151 const cryptohome::Identification& cryptohome_id, |
142 const std::string& key_name, | 152 const std::string& key_name, |
143 const std::string& domain, | 153 const std::string& domain, |
144 const std::string& device_id, | 154 const std::string& device_id, |
145 attestation::AttestationChallengeOptions options, | 155 attestation::AttestationChallengeOptions options, |
146 const std::string& challenge, | 156 const std::string& challenge, |
147 const AsyncMethodCallback& callback)); | 157 const AsyncMethodCallback& callback)); |
148 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, | 158 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, |
149 void(attestation::AttestationKeyType key_type, | 159 void(attestation::AttestationKeyType key_type, |
150 const std::string& user_id, | 160 const cryptohome::Identification& cryptohome_id, |
151 const std::string& key_name, | 161 const std::string& key_name, |
152 const std::string& challenge, | 162 const std::string& challenge, |
153 const AsyncMethodCallback& callback)); | 163 const AsyncMethodCallback& callback)); |
154 MOCK_METHOD4(TpmAttestationGetKeyPayload, | 164 MOCK_METHOD4(TpmAttestationGetKeyPayload, |
155 void(attestation::AttestationKeyType key_type, | 165 void(attestation::AttestationKeyType key_type, |
156 const std::string& user_id, | 166 const cryptohome::Identification& cryptohome_id, |
157 const std::string& key_name, | 167 const std::string& key_name, |
158 const DataMethodCallback& callback)); | 168 const DataMethodCallback& callback)); |
159 MOCK_METHOD5(TpmAttestationSetKeyPayload, | 169 MOCK_METHOD5(TpmAttestationSetKeyPayload, |
160 void(attestation::AttestationKeyType key_type, | 170 void(attestation::AttestationKeyType key_type, |
161 const std::string& user_id, | 171 const cryptohome::Identification& cryptohome_id, |
162 const std::string& key_name, | 172 const std::string& key_name, |
163 const std::string& payload, | 173 const std::string& payload, |
164 const BoolDBusMethodCallback& callback)); | 174 const BoolDBusMethodCallback& callback)); |
165 MOCK_METHOD4(TpmAttestationDeleteKeys, | 175 MOCK_METHOD4(TpmAttestationDeleteKeys, |
166 void(attestation::AttestationKeyType key_type, | 176 void(attestation::AttestationKeyType key_type, |
167 const std::string& user_id, | 177 const cryptohome::Identification& cryptohome_id, |
168 const std::string& key_prefix, | 178 const std::string& key_prefix, |
169 const BoolDBusMethodCallback& callback)); | 179 const BoolDBusMethodCallback& callback)); |
170 MOCK_METHOD4(GetKeyDataEx, | 180 MOCK_METHOD4(GetKeyDataEx, |
171 void(const cryptohome::AccountIdentifier& id, | 181 void(const cryptohome::Identification& cryptohome_id, |
172 const cryptohome::AuthorizationRequest& auth, | 182 const cryptohome::AuthorizationRequest& auth, |
173 const cryptohome::GetKeyDataRequest& request, | 183 const cryptohome::GetKeyDataRequest& request, |
174 const ProtobufMethodCallback& callback)); | 184 const ProtobufMethodCallback& callback)); |
175 MOCK_METHOD4(CheckKeyEx, | 185 MOCK_METHOD4(CheckKeyEx, |
176 void(const cryptohome::AccountIdentifier& id, | 186 void(const cryptohome::Identification& cryptohome_id, |
177 const cryptohome::AuthorizationRequest& auth, | 187 const cryptohome::AuthorizationRequest& auth, |
178 const cryptohome::CheckKeyRequest& request, | 188 const cryptohome::CheckKeyRequest& request, |
179 const ProtobufMethodCallback& callback)); | 189 const ProtobufMethodCallback& callback)); |
180 MOCK_METHOD4(MountEx, | 190 MOCK_METHOD4(MountEx, |
181 void(const cryptohome::AccountIdentifier& id, | 191 void(const cryptohome::Identification& cryptohome_id, |
182 const cryptohome::AuthorizationRequest& auth, | 192 const cryptohome::AuthorizationRequest& auth, |
183 const cryptohome::MountRequest& request, | 193 const cryptohome::MountRequest& request, |
184 const ProtobufMethodCallback& callback)); | 194 const ProtobufMethodCallback& callback)); |
185 MOCK_METHOD4(AddKeyEx, | 195 MOCK_METHOD4(AddKeyEx, |
186 void(const cryptohome::AccountIdentifier& id, | 196 void(const cryptohome::Identification& cryptohome_id, |
187 const cryptohome::AuthorizationRequest& auth, | 197 const cryptohome::AuthorizationRequest& auth, |
188 const cryptohome::AddKeyRequest& request, | 198 const cryptohome::AddKeyRequest& request, |
189 const ProtobufMethodCallback& callback)); | 199 const ProtobufMethodCallback& callback)); |
190 MOCK_METHOD4(UpdateKeyEx, | 200 MOCK_METHOD4(UpdateKeyEx, |
191 void(const cryptohome::AccountIdentifier& id, | 201 void(const cryptohome::Identification& cryptohome_id, |
192 const cryptohome::AuthorizationRequest& auth, | 202 const cryptohome::AuthorizationRequest& auth, |
193 const cryptohome::UpdateKeyRequest& request, | 203 const cryptohome::UpdateKeyRequest& request, |
194 const ProtobufMethodCallback& callback)); | 204 const ProtobufMethodCallback& callback)); |
195 MOCK_METHOD4(RemoveKeyEx, | 205 MOCK_METHOD4(RemoveKeyEx, |
196 void(const cryptohome::AccountIdentifier& id, | 206 void(const cryptohome::Identification& cryptohome_id, |
197 const cryptohome::AuthorizationRequest& auth, | 207 const cryptohome::AuthorizationRequest& auth, |
198 const cryptohome::RemoveKeyRequest& request, | 208 const cryptohome::RemoveKeyRequest& request, |
199 const ProtobufMethodCallback& callback)); | 209 const ProtobufMethodCallback& callback)); |
200 MOCK_METHOD2(GetBootAttribute, | 210 MOCK_METHOD2(GetBootAttribute, |
201 void(const cryptohome::GetBootAttributeRequest& request, | 211 void(const cryptohome::GetBootAttributeRequest& request, |
202 const ProtobufMethodCallback& callback)); | 212 const ProtobufMethodCallback& callback)); |
203 MOCK_METHOD2(SetBootAttribute, | 213 MOCK_METHOD2(SetBootAttribute, |
204 void(const cryptohome::SetBootAttributeRequest& request, | 214 void(const cryptohome::SetBootAttributeRequest& request, |
205 const ProtobufMethodCallback& callback)); | 215 const ProtobufMethodCallback& callback)); |
206 MOCK_METHOD2( | 216 MOCK_METHOD2( |
207 FlushAndSignBootAttributes, | 217 FlushAndSignBootAttributes, |
208 void(const cryptohome::FlushAndSignBootAttributesRequest& request, | 218 void(const cryptohome::FlushAndSignBootAttributesRequest& request, |
209 const ProtobufMethodCallback& callback)); | 219 const ProtobufMethodCallback& callback)); |
210 }; | 220 }; |
211 | 221 |
212 } // namespace chromeos | 222 } // namespace chromeos |
213 | 223 |
214 #endif // CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ | 224 #endif // CHROMEOS_DBUS_MOCK_CRYPTOHOME_CLIENT_H_ |
OLD | NEW |