| // 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/security_interstitials/content/unsafe_resource_util.h" |
| |
| #include "content/public/browser/navigation_entry.h" |
| #include "content/public/browser/render_frame_host.h" |
| #include "content/public/browser/web_contents.h" |
| |
| namespace security_interstitials { |
| |
| content::NavigationEntry* GetNavigationEntryForResource( |
| const UnsafeResource& resource) { |
| content::WebContents* web_contents = GetWebContentsForResource(resource); |
| 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 (resource.IsMainPageLoadBlocked()) |
| 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 UnsafeResource& resource) { |
| content::RenderFrameHost* rfh = content::RenderFrameHost::FromID( |
| resource.render_process_id, resource.render_frame_id); |
| if (rfh) { |
| return content::WebContents::FromRenderFrameHost(rfh); |
| } |
| return content::WebContents::FromFrameTreeNodeId(resource.frame_tree_node_id); |
| } |
| |
| } // namespace security_interstitials |
