1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_ASH_COMPONENTS_DBUS_USERDATAAUTH_FAKE_CRYPTOHOME_MISC_CLIENT_H_
#define CHROMEOS_ASH_COMPONENTS_DBUS_USERDATAAUTH_FAKE_CRYPTOHOME_MISC_CLIENT_H_
#include "chromeos/ash/components/dbus/userdataauth/cryptohome_misc_client.h"
#include "base/component_export.h"
#include "chromeos/ash/components/cryptohome/error_types.h"
#include "chromeos/ash/components/dbus/cryptohome/UserDataAuth.pb.h"
namespace ash {
class COMPONENT_EXPORT(USERDATAAUTH_CLIENT) FakeCryptohomeMiscClient
: public CryptohomeMiscClient {
public:
FakeCryptohomeMiscClient();
~FakeCryptohomeMiscClient() override;
// Not copyable or movable.
FakeCryptohomeMiscClient(const FakeCryptohomeMiscClient&) = delete;
FakeCryptohomeMiscClient& operator=(const FakeCryptohomeMiscClient&) = delete;
// Checks that a FakeCryptohomeMiscClient instance was initialized and returns
// it.
static FakeCryptohomeMiscClient* Get();
// CryptohomeMiscClient override:
void WaitForServiceToBeAvailable(
chromeos::WaitForServiceToBeAvailableCallback callback) override;
void GetSystemSalt(const ::user_data_auth::GetSystemSaltRequest& request,
GetSystemSaltCallback callback) override;
void GetSanitizedUsername(
const ::user_data_auth::GetSanitizedUsernameRequest& request,
GetSanitizedUsernameCallback callback) override;
void GetLoginStatus(const ::user_data_auth::GetLoginStatusRequest& request,
GetLoginStatusCallback callback) override;
void LockToSingleUserMountUntilReboot(
const ::user_data_auth::LockToSingleUserMountUntilRebootRequest& request,
LockToSingleUserMountUntilRebootCallback callback) override;
void GetRsuDeviceId(const ::user_data_auth::GetRsuDeviceIdRequest& request,
GetRsuDeviceIdCallback callback) override;
std::optional<::user_data_auth::GetSanitizedUsernameReply>
BlockingGetSanitizedUsername(
const ::user_data_auth::GetSanitizedUsernameRequest& request) override;
// Various getter/setters:
// Sets the system salt which will be returned from GetSystemSalt(). By
// default, GetSystemSalt() returns the value generated by
// GetStubSystemSalt().
void set_system_salt(const std::vector<uint8_t>& system_salt) {
system_salt_ = system_salt;
}
// Returns the stub system salt as raw bytes. (not as a string encoded in the
// format used by SystemSaltGetter::ConvertRawSaltToHexString()).
static std::vector<uint8_t> GetStubSystemSalt();
// Set whether we say the system needs powerwash the next time CheckHealth()
// is called.
void set_requires_powerwash(bool requires_powerwash) {
requires_powerwash_ = requires_powerwash;
}
// Was LockToSingleUserMountUntilReboot() called?
bool is_device_locked_to_single_user() const {
return is_device_locked_to_single_user_;
}
// Sets the RSU ID that we return for GetRsuDeviceId().
void set_rsu_device_id(const std::string& rsu_device_id) {
rsu_device_id_ = rsu_device_id;
}
// Sets the CryptohomeError value to return.
void set_cryptohome_error(::user_data_auth::CryptohomeErrorCode error) {
cryptohome_error_ = error;
}
// WaitForServiceToBeAvailable() related:
// Changes the behavior of WaitForServiceToBeAvailable(). This method runs
// pending callbacks if is_available is true.
void SetServiceIsAvailable(bool is_available);
// Runs pending availability callbacks reporting that the service is
// unavailable. Expects service not to be available when called.
void ReportServiceIsNotAvailable();
private:
// Helper that returns the protobuf reply.
template <typename ReplyType>
void ReturnProtobufMethodCallback(
const ReplyType& reply,
chromeos::DBusMethodCallback<ReplyType> callback);
// The next error code to return for various functions.
::user_data_auth::CryptohomeErrorCode cryptohome_error_ =
::user_data_auth::CRYPTOHOME_ERROR_NOT_SET;
// The system salt to return.
std::vector<uint8_t> system_salt_{GetStubSystemSalt()};
// Whether we say we need to powerwash the next time CheckHealth() is called.
bool requires_powerwash_ = false;
// Used by LockToSingleUserMountUntilReboot.
bool is_device_locked_to_single_user_ = false;
// Reply for GetRsuDeviceId().
std::string rsu_device_id_;
// WaitForServiceToBeAvailable() related fields:
// If set, we tell callers that service is available.
bool service_is_available_ = true;
// If set, WaitForServiceToBeAvailable will run the callback, even if service
// is not available (instead of adding the callback to pending callback list).
bool service_reported_not_available_ = false;
// The list of callbacks passed to WaitForServiceToBeAvailable when the
// service wasn't available.
std::vector<chromeos::WaitForServiceToBeAvailableCallback>
pending_wait_for_service_to_be_available_callbacks_;
};
} // namespace ash
#endif // CHROMEOS_ASH_COMPONENTS_DBUS_USERDATAAUTH_FAKE_CRYPTOHOME_MISC_CLIENT_H_
|
