Move GetDocument method from WebFrame to WebLocalFrame.

extensions/renderer/script_context.cc

 // Copyright 2014 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 "extensions/renderer/script_context.h"
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/strings/string_split.h"
@@ skipping 282 matching lines @@
     if (data_source &&
         frame->GetSecurityOrigin().CanAccess(blink::WebSecurityOrigin::Create(
             data_source->GetRequest().Url()))) {
       return GURL(data_source->GetRequest().Url());
     }
   }
   return GURL(weburl);
 }
 
 // static
-GURL ScriptContext::GetEffectiveDocumentURL(const blink::WebLocalFrame* frame,
+GURL ScriptContext::GetEffectiveDocumentURL(blink::WebLocalFrame* frame,
                                             const GURL& document_url,
                                             bool match_about_blank) {
   // Common scenario. If |match_about_blank| is false (as is the case in most
   // extensions), or if the frame is not an about:-page, just return
   // |document_url| (supposedly the URL of the frame).
   if (!match_about_blank || !document_url.SchemeIs(url::kAboutScheme))
     return document_url;
 
   // Non-sandboxed about:blank and about:srcdoc pages inherit their security
   // origin from their parent frame/window. So, traverse the frame/window
   // hierarchy to find the closest non-about:-page and return its URL.
-  const blink::WebFrame* parent = frame;
+  blink::WebFrame* parent = frame;
+  blink::WebDocument parent_document;
   do {
     if (parent->Parent())
       parent = parent->Parent();
     else if (parent->Opener() != parent)
       parent = parent->Opener();
     else
       parent = nullptr;
-  } while (parent && !parent->GetDocument().IsNull() &&
-           GURL(parent->GetDocument().Url()).SchemeIs(url::kAboutScheme));
 
-  if (parent && !parent->GetDocument().IsNull()) {
+    parent_document = parent && parent->IsWebLocalFrame()
+                          ? parent->ToWebLocalFrame()->GetDocument()
+                          : blink::WebDocument();
+  } while (!parent_document.IsNull() &&
+           GURL(parent_document.Url()).SchemeIs(url::kAboutScheme));
+
+  if (!parent_document.IsNull()) {
     // Only return the parent URL if the frame can access it.
-    const blink::WebDocument& parent_document = parent->GetDocument();
     if (frame->GetDocument().GetSecurityOrigin().CanAccess(
             parent_document.GetSecurityOrigin())) {
       return parent_document.Url();
     }
   }
   return document_url;
 }
 
 ScriptContext* ScriptContext::GetContext() {
   DCHECK(thread_checker_.CalledOnValidThread());
@@ skipping 182 matching lines @@
 
   v8::Local<v8::Object> global = v8_context()->Global();
   if (!web_frame_)
     return handle_scope.Escape(function->Call(global, argc, argv));
   return handle_scope.Escape(
       v8::Local<v8::Value>(web_frame_->CallFunctionEvenIfScriptDisabled(
           function, global, argc, argv)));
 }
 
 }  // namespace extensions