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
|
// 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/safe_browsing/content/browser/content_unsafe_resource_util.h"
#include "components/safe_browsing/content/browser/async_check_tracker.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
namespace safe_browsing::unsafe_resource_util {
content::NavigationEntry* GetNavigationEntryForResource(
const security_interstitials::UnsafeResource& resource) {
return GetNavigationEntryForLocator(
resource.rfh_locator, resource.navigation_id, resource.threat_type);
}
content::NavigationEntry* GetNavigationEntryForLocator(
const security_interstitials::UnsafeResourceLocator& rfh_locator,
const std::optional<int64_t>& navigation_id,
safe_browsing::SBThreatType threat_type) {
content::WebContents* web_contents =
unsafe_resource_util::GetWebContentsForLocator(rfh_locator);
if (!web_contents) {
return nullptr;
}
// If a safebrowsing hit occurs during main frame navigation, the navigation
// will not be committed, and the pending navigation entry refers to the hit.
if (AsyncCheckTracker::IsMainPageLoadPending(rfh_locator, navigation_id,
threat_type)) {
return web_contents->GetController().GetPendingEntry();
}
// If a safebrowsing hit occurs on a subresource load, or on a main frame
// after the navigation is committed, the last committed navigation entry
// refers to the page with the hit. Note that there may concurrently be an
// unrelated pending navigation to another site, so GetActiveEntry() would be
// wrong.
return web_contents->GetController().GetLastCommittedEntry();
}
content::WebContents* GetWebContentsForResource(
const security_interstitials::UnsafeResource& resource) {
return GetWebContentsForLocator(resource.rfh_locator);
}
content::WebContents* GetWebContentsForLocator(
const security_interstitials::UnsafeResourceLocator& rfh_locator) {
if (rfh_locator.render_frame_token) {
content::RenderFrameHost* rfh = content::RenderFrameHost::FromFrameToken(
content::GlobalRenderFrameHostToken(
rfh_locator.render_process_id,
blink::LocalFrameToken(rfh_locator.render_frame_token.value())));
if (rfh) {
return content::WebContents::FromRenderFrameHost(rfh);
}
}
return content::WebContents::FromFrameTreeNodeId(
content::FrameTreeNodeId(rfh_locator.frame_tree_node_id));
}
} // namespace safe_browsing::unsafe_resource_util
|
