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
135
136
137
138
139
140
141
142
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/ambient/ambient_access_token_controller.h"
#include "ash/login/ui/lock_screen.h"
#include "ash/public/cpp/ambient/ambient_client.h"
#include "base/functional/bind.h"
#include "base/logging.h"
#include "base/rand_util.h"
#include "base/time/time.h"
namespace ash {
namespace {
constexpr int kMaxRetries = 3;
constexpr net::BackoffEntry::Policy kRetryBackoffPolicy = {
0, // Number of initial errors to ignore.
1000, // Initial delay in ms.
2.0, // Factor by which the waiting time will be multiplied.
0.2, // Fuzzing percentage.
60 * 1000, // Maximum delay in ms.
-1, // Never discard the entry.
true, // Use initial delay.
};
} // namespace
AmbientAccessTokenController::AmbientAccessTokenController()
: refresh_token_retry_backoff_(&kRetryBackoffPolicy) {}
AmbientAccessTokenController::~AmbientAccessTokenController() = default;
void AmbientAccessTokenController::RequestAccessToken(
AccessTokenCallback callback,
bool may_refresh_token_on_lock) {
// |token_refresh_timer_| may become stale during sleeping.
if (token_refresh_timer_.IsRunning())
token_refresh_timer_.Stop();
if (!access_token_.empty()) {
DCHECK(!has_pending_request_);
// Return the token if there is enough time to use the access token when
// requested.
if (expiration_time_ - base::Time::Now() > token_usage_time_buffer_) {
RunCallback(std::move(callback));
return;
}
access_token_ = std::string();
expiration_time_ = base::Time::Now();
}
if (!may_refresh_token_on_lock && LockScreen::HasInstance()) {
RunCallback(std::move(callback));
return;
}
callbacks_.emplace_back(std::move(callback));
if (has_pending_request_)
return;
RefreshAccessToken();
}
base::WeakPtr<AmbientAccessTokenController>
AmbientAccessTokenController::AsWeakPtr() {
return weak_factory_.GetWeakPtr();
}
void AmbientAccessTokenController::RefreshAccessToken() {
DCHECK(!token_refresh_timer_.IsRunning());
has_pending_request_ = true;
AmbientClient::Get()->RequestAccessToken(
base::BindOnce(&AmbientAccessTokenController::AccessTokenRefreshed,
weak_factory_.GetWeakPtr()));
}
void AmbientAccessTokenController::AccessTokenRefreshed(
const GaiaId& gaia_id,
const std::string& access_token,
const base::Time& expiration_time) {
has_pending_request_ = false;
if (gaia_id.empty() || access_token.empty()) {
refresh_token_retry_backoff_.InformOfRequest(/*succeeded=*/false);
if (refresh_token_retry_backoff_.failure_count() <= kMaxRetries) {
LOG(WARNING) << "Unable to refresh access token. Retrying..";
RetryRefreshAccessToken();
} else {
LOG(ERROR) << "Unable to refresh access token. All retries attempted.";
NotifyAccessTokenRefreshed();
}
return;
}
DVLOG(1) << "Access token fetched.";
DCHECK(gaia_id_.empty() || gaia_id_ == gaia_id);
refresh_token_retry_backoff_.Reset();
gaia_id_ = gaia_id;
access_token_ = access_token;
expiration_time_ = expiration_time;
NotifyAccessTokenRefreshed();
}
void AmbientAccessTokenController::RetryRefreshAccessToken() {
base::TimeDelta backoff_delay =
refresh_token_retry_backoff_.GetTimeUntilRelease();
token_refresh_timer_.Start(
FROM_HERE, backoff_delay,
base::BindOnce(&AmbientAccessTokenController::RefreshAccessToken,
base::Unretained(this)));
}
void AmbientAccessTokenController::NotifyAccessTokenRefreshed() {
for (auto& callback : callbacks_)
RunCallback(std::move(callback));
callbacks_.clear();
}
void AmbientAccessTokenController::RunCallback(AccessTokenCallback callback) {
std::move(callback).Run(gaia_id_, access_token_);
}
void AmbientAccessTokenController::SetTokenUsageBufferForTesting(
base::TimeDelta time) {
token_usage_time_buffer_ = time;
}
base::TimeDelta AmbientAccessTokenController::GetTimeUntilReleaseForTesting() {
return refresh_token_retry_backoff_.GetTimeUntilRelease();
}
} // namespace ash
|
