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
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/plus_addresses/plus_address_jit_allocator.h"
#include <utility>
#include "base/functional/callback.h"
#include "base/notreached.h"
#include "base/time/time.h"
#include "components/plus_addresses/plus_address_http_client.h"
#include "components/plus_addresses/plus_address_types.h"
#include "net/http/http_status_code.h"
#include "url/origin.h"
namespace plus_addresses {
namespace {
// The cooldown period applied if a user exhausts their refresh quota. Refresh
// UI will not be offered until the cooldown has expired (or the user has
// restarted the browser).
constexpr base::TimeDelta kRefreshLimitReachedCooldown = base::Days(1);
} // namespace
PlusAddressJitAllocator::PlusAddressJitAllocator(
PlusAddressHttpClient* http_client)
: http_client_(*http_client) {}
PlusAddressJitAllocator::~PlusAddressJitAllocator() = default;
void PlusAddressJitAllocator::AllocatePlusAddress(
const url::Origin& origin,
AllocationMode mode,
PlusAddressRequestCallback callback) {
switch (mode) {
case AllocationMode::kAny: {
http_client_->ReservePlusAddress(origin, /*refresh=*/false,
std::move(callback));
return;
}
case AllocationMode::kNewPlusAddress: {
int& attempts_made = refresh_attempts_[origin];
if (attempts_made >= kMaxPlusAddressRefreshesPerOrigin) {
std::move(callback).Run(base::unexpected(PlusAddressRequestError(
PlusAddressRequestErrorType::kMaxRefreshesReached)));
return;
}
++attempts_made;
http_client_->ReservePlusAddress(
origin, /*refresh=*/true,
base::BindOnce(&PlusAddressJitAllocator::HandleRefreshResponse,
weak_ptr_factory_.GetWeakPtr(), std::move(callback)));
return;
}
}
NOTREACHED();
}
std::optional<PlusProfile>
PlusAddressJitAllocator::AllocatePlusAddressSynchronously(
const url::Origin& origin,
AllocationMode mode) {
return std::nullopt;
}
bool PlusAddressJitAllocator::IsRefreshingSupported(
const url::Origin& origin) const {
if (!time_refresh_limit_reached_.is_null() &&
base::TimeTicks::Now() - time_refresh_limit_reached_ <=
kRefreshLimitReachedCooldown) {
return false;
}
if (auto it = refresh_attempts_.find(origin);
it != refresh_attempts_.cend() &&
it->second >= kMaxPlusAddressRefreshesPerOrigin) {
return false;
}
return true;
}
void PlusAddressJitAllocator::RemoveAllocatedPlusAddress(
const PlusAddress& plus_address) {
// This is a no-op for the JIT allocator - if the plus address was created,
// the backend will ensure that it does not show up again.
}
void PlusAddressJitAllocator::HandleRefreshResponse(
PlusAddressRequestCallback callback,
const PlusProfileOrError& profile_or_error) {
if (!profile_or_error.has_value() &&
profile_or_error.error() == PlusAddressRequestError::AsNetworkError(
net::HTTP_TOO_MANY_REQUESTS)) {
// If the server responds with 429, it means that the refresh quota is
// exhausted.
time_refresh_limit_reached_ = base::TimeTicks::Now();
}
std::move(callback).Run(profile_or_error);
}
} // namespace plus_addresses
|
