Move GetDocument method from WebFrame to WebLocalFrame.

components/safe_browsing/renderer/threat_dom_details.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/safe_browsing/renderer/threat_dom_details.h"
 
 #include <algorithm>
 #include <map>
+#include <string>
 #include <unordered_set>
 
 #include "base/compiler_specific.h"
 #include "base/metrics/field_trial_params.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_split.h"
 #include "base/strings/stringprintf.h"
 #include "components/safe_browsing/common/safebrowsing_messages.h"
 #include "components/safe_browsing/common/safebrowsing_types.h"
 #include "content/public/renderer/render_frame.h"
@@ skipping 82 matching lines @@
   std::sort(tag_and_attributes_list->begin(), tag_and_attributes_list->end(),
             [](const TagAndAttributesItem& a, const TagAndAttributesItem& b) {
               return a.tag_name < b.tag_name;
             });
 }
 
 SafeBrowsingHostMsg_ThreatDOMDetails_Node* GetNodeForElement(
     const blink::WebNode& element,
     const safe_browsing::ElementToNodeMap& element_to_node_map,
     std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>* resources) {
-  DCHECK(element_to_node_map.count(element) > 0);
+  DCHECK_GT(element_to_node_map.count(element), 0u);
   size_t resource_index = element_to_node_map.at(element);
   return &(resources->at(resource_index));
 }
 
 std::string TruncateAttributeString(const std::string& input) {
   if (input.length() <= ThreatDOMDetails::kMaxAttributeStringLength) {
     return input;
   }
 
   std::string truncated;
@@ skipping 151 matching lines @@
 
 void ThreatDOMDetails::OnGetThreatDOMDetails() {
   std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node> resources;
   ExtractResources(&resources);
   // Notify the browser.
   Send(new SafeBrowsingHostMsg_ThreatDOMDetails(routing_id(), resources));
 }
 
 void ThreatDOMDetails::ExtractResources(
     std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>* resources) {
-  blink::WebFrame* frame = render_frame()->GetWebFrame();
+  blink::WebLocalFrame* frame = render_frame()->GetWebFrame();
   if (!frame)
     return;
   SafeBrowsingHostMsg_ThreatDOMDetails_Node details_node;
   blink::WebDocument document = frame->GetDocument();
   details_node.url = GURL(document.Url());
   if (document.IsNull()) {
     // Nothing in this frame. Just report its URL.
     resources->push_back(details_node);
     return;
   }
@@ skipping 13 matching lines @@
     }
   }
   resources->push_back(details_node);
 }
 
 void ThreatDOMDetails::OnDestruct() {
   delete this;
 }
 
 }  // namespace safe_browsing