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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/341324165): Fix and remove.
#pragma allow_unsafe_buffers
#endif
#include "content/web_test/browser/web_test_origin_trial_throttle.h"
#include <string>
#include "base/containers/flat_set.h"
#include "base/containers/span.h"
#include "base/strings/string_util.h"
#include "base/time/time.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/origin_trials_controller_delegate.h"
#include "content/public/browser/render_frame_host.h"
#include "url/gurl.h"
#include "url/origin.h"
namespace content {
namespace {
const char kThrottleName[] = "WebTestOriginTrialThrottle";
const char kWebTestOriginTrialHeaderName[] = "X-Web-Test-Enabled-Origin-Trials";
} // namespace
WebTestOriginTrialThrottle::WebTestOriginTrialThrottle(
NavigationThrottleRegistry& registry,
OriginTrialsControllerDelegate* delegate)
: NavigationThrottle(registry),
origin_trials_controller_delegate_(delegate) {}
NavigationThrottle::ThrottleCheckResult
WebTestOriginTrialThrottle::WillStartRequest() {
SetHeaderForRequest();
return NavigationThrottle::ThrottleAction::PROCEED;
}
NavigationThrottle::ThrottleCheckResult
WebTestOriginTrialThrottle::WillRedirectRequest() {
SetHeaderForRequest();
return NavigationThrottle::ThrottleAction::PROCEED;
}
void WebTestOriginTrialThrottle::SetHeaderForRequest() {
GURL request_url = navigation_handle()->GetURL();
url::Origin origin = url::Origin::CreateFromNormalizedTuple(
request_url.scheme(), request_url.host(), request_url.EffectiveIntPort());
base::flat_set<std::string> trials;
if (!origin.opaque()) {
url::Origin partition_origin = origin;
if (navigation_handle()->GetParentFrameOrOuterDocument()) {
partition_origin = navigation_handle()
->GetParentFrameOrOuterDocument()
->GetOutermostMainFrame()
->GetLastCommittedOrigin();
}
trials = origin_trials_controller_delegate_->GetPersistedTrialsForOrigin(
origin, partition_origin, base::Time::Now());
}
std::string header_value = base::JoinString(
base::span<std::string>(trials.begin(), trials.end()), ", ");
if (!header_value.empty()) {
navigation_handle()->SetRequestHeader(kWebTestOriginTrialHeaderName,
header_value);
}
}
const char* WebTestOriginTrialThrottle::GetNameForLogging() {
return kThrottleName;
}
} // namespace content
|
