Expose the initiating origin/URL of a navigation in the Blink public API

third_party/WebKit/Source/web/tests/WebFrameTest.cpp

 /*
  * Copyright (C) 2010 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 10637 matching lines @@
   webViewHelper.initializeAndLoad(m_baseURL + "frameset-repeated-name.html");
   Frame* mainFrame = webViewHelper.webView()->mainFrameImpl()->frame();
   HashSet<AtomicString> names;
   for (Frame* frame = mainFrame->tree().firstChild(); frame;
        frame = frame->tree().traverseNext()) {
     EXPECT_TRUE(names.add(frame->tree().uniqueName()).isNewEntry);
   }
   EXPECT_EQ(10u, names.size());
 }
 
+TEST_F(WebFrameTest, RequestorURL) {
+  registerMockedHttpURLLoad("single_iframe.html");
+  registerMockedHttpURLLoad("visible_iframe.html");
+
+  class RequestorURLTrackingClient
+      : public FrameTestHelpers::TestWebFrameClient {
+   public:
+    // WebFrameClient overrides:
+    void didCommitProvisionalLoad(WebLocalFrame* frame,
+                                  const WebHistoryItem&,
+                                  WebHistoryCommitType) override {
+      m_trackedFrames.add(frame, frame->dataSource()->requestorURL());
+    }
+
+    WebURL requestorURLForFrame(WebLocalFrame* frame) {
+      return m_trackedFrames.get(frame);
+    }
+
+   private:
+    HashMap<WebLocalFrame*, WebURL> m_trackedFrames;
+  };
+
+  RequestorURLTrackingClient trackingClient;
+
+  FrameTestHelpers::WebViewHelper webViewHelper;
+  webViewHelper.initializeAndLoad(m_baseURL + "single_iframe.html", true,
+                                  &trackingClient);
+
+  WebLocalFrameImpl* mainFrame = webViewHelper.webView()->mainFrameImpl();
+  WebLocalFrameImpl* subFrame = toWebLocalFrameImpl(mainFrame->firstChild());
+
+  // The main frame should have no requestor URL.

[Mike West, 2016/11/23 11:11:20]

Should the main frame have a requestor URL during a navigation? Can you add a
test to verify that behavior?

[dcheng, 2016/11/23 14:41:44]

Done.

+  EXPECT_EQ(WebURL(), trackingClient.requestorURLForFrame(mainFrame));
+  // The sub frame should have the main frame's URL as the requestor URL.
+  EXPECT_EQ(WebURL(mainFrame->frame()->document()->url()),
+            trackingClient.requestorURLForFrame(subFrame));
+
+  // Once the load has committed, the requestor URL should no longer be
+  // accessible, to avoid keeping potentially unreachable Documents alive.
+  EXPECT_EQ(WebURL(), subFrame->dataSource()->requestorURL());
+
+  webViewHelper.reset();
+}
+
 }  // namespace blink