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
|
// 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 "components/embedder_support/android/util/cdn_utils.h"
#include "base/command_line.h"
#include "base/feature_list.h"
#include "base/no_destructor.h"
#include "content/public/browser/page.h"
#include "content/public/browser/render_frame_host.h"
#include "net/http/http_response_headers.h"
#include "services/network/public/mojom/url_response_head.mojom.h"
#include "url/gurl.h"
namespace embedder_support {
namespace {
constexpr char kDefaultTrustedCDNBaseURL[] = "https://cdn.ampproject.org";
// Specifies a base URL for the trusted CDN for tests.
const char kTrustedCDNBaseURLForTests[] = "trusted-cdn-base-url-for-tests";
// Returns whether the given URL is hosted by a trusted CDN. The base URL to
// trust can be overridden via a command line flag for testing.
bool IsTrustedCDN(const GURL& url) {
// Use a static local (without destructor) to construct the base URL only
// once. |trusted_cdn_base_url| is initialized with the result of an
// immediately evaluated lambda, which allows wrapping the code in a single
// expression.
static const base::NoDestructor<GURL> trusted_cdn_base_url([]() {
const base::CommandLine* command_line =
base::CommandLine::ForCurrentProcess();
if (command_line->HasSwitch(kTrustedCDNBaseURLForTests)) {
GURL base_url(
command_line->GetSwitchValueASCII(kTrustedCDNBaseURLForTests));
LOG_IF(WARNING, !base_url.is_valid()) << "Invalid trusted CDN base URL: "
<< base_url.possibly_invalid_spec();
return base_url;
}
return GURL(kDefaultTrustedCDNBaseURL);
}());
// Allow any subdomain of the base URL.
return url.DomainIs(trusted_cdn_base_url->host_piece()) &&
(url.scheme_piece() == trusted_cdn_base_url->scheme_piece()) &&
(url.EffectiveIntPort() == trusted_cdn_base_url->EffectiveIntPort());
}
} // namespace
GURL GetPublisherURL(content::RenderFrameHost* rfh) {
if (!rfh || !rfh->IsInPrimaryMainFrame() ||
!IsTrustedCDN(rfh->GetLastCommittedURL())) {
return GURL();
}
const network::mojom::URLResponseHead* response_head =
rfh->GetLastResponseHead();
if (!response_head || !response_head->headers) {
// TODO(crbug.com/41381000): In some cases other than offline pages
// we don't have headers.
LOG(WARNING) << "No headers for navigation to "
<< rfh->GetLastCommittedURL();
return GURL();
}
std::optional<std::string> publisher_url =
response_head->headers->GetNormalizedHeader("x-amp-cache");
if (!publisher_url) {
return GURL();
}
return GURL(*publisher_url);
}
} // namespace embedder_support
|
