Distinguish between subresource filtering and dryrun matching.

third_party/WebKit/Source/core/loader/FrameFetchContextTest.cpp

 /*
  * Copyright (c) 2015, 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 12 matching lines @@
  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "core/loader/FrameFetchContext.h"
 
+#include <memory>
 #include "core/dom/Document.h"
 #include "core/frame/FrameHost.h"
 #include "core/frame/FrameOwner.h"
 #include "core/frame/FrameTypes.h"
 #include "core/frame/FrameView.h"
 #include "core/frame/Settings.h"
 #include "core/html/HTMLIFrameElement.h"
 #include "core/loader/DocumentLoader.h"
 #include "core/loader/EmptyClients.h"
 #include "core/page/Page.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/loader/fetch/FetchInitiatorInfo.h"
 #include "platform/loader/fetch/MockResource.h"
 #include "platform/loader/fetch/UniqueIdentifier.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/WebAddressSpace.h"
 #include "public/platform/WebCachePolicy.h"
+#include "public/platform/WebDocumentSubresourceFilter.h"
 #include "public/platform/WebInsecureRequestPolicy.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include <memory>
 
 namespace blink {
 
 class StubFrameLoaderClientWithParent final : public EmptyFrameLoaderClient {
  public:
   static StubFrameLoaderClientWithParent* create(Frame* parent) {
     return new StubFrameLoaderClientWithParent(parent);
   }
 
   DEFINE_INLINE_VIRTUAL_TRACE() {
@@ skipping 10 matching lines @@
 };
 
 class MockFrameLoaderClient : public EmptyFrameLoaderClient {
  public:
   MockFrameLoaderClient() : EmptyFrameLoaderClient() {}
   MOCK_METHOD1(didDisplayContentWithCertificateErrors, void(const KURL&));
   MOCK_METHOD2(dispatchDidLoadResourceFromMemoryCache,
                void(const ResourceRequest&, const ResourceResponse&));
 };
 
+class FixedPolicySubresourceFilter : public WebDocumentSubresourceFilter {
+ public:
+  FixedPolicySubresourceFilter(LoadPolicy policy, int* filteredLoadCounter)
+      : m_policy(policy), m_filteredLoadCounter(filteredLoadCounter) {}
+
+  LoadPolicy getLoadPolicy(const WebURL& resourceUrl,
+                           WebURLRequest::RequestContext) override {
+    return m_policy;
+  }
+
+  void reportDisallowedLoad() override { ++*m_filteredLoadCounter; }
+
+ private:
+  const LoadPolicy m_policy;
+  int* m_filteredLoadCounter;
+};
+
 class FrameFetchContextTest : public ::testing::Test {
  protected:
   void SetUp() override {
     dummyPageHolder = DummyPageHolder::create(IntSize(500, 500));
     dummyPageHolder->page().setDeviceScaleFactor(1.0);
     document = &dummyPageHolder->document();
     fetchContext =
         static_cast<FrameFetchContext*>(&document->fetcher()->context());
     owner = DummyFrameOwner::create();
     FrameFetchContext::provideDocumentToContext(*fetchContext, document.get());
@@ skipping 24 matching lines @@
   // usage.
   Persistent<Document> document;
   Persistent<FrameFetchContext> fetchContext;
 
   Persistent<StubFrameLoaderClientWithParent> childClient;
   Persistent<LocalFrame> childFrame;
   Persistent<Document> childDocument;
   Persistent<DummyFrameOwner> owner;
 };
 
+class FrameFetchContextSubresourceFilterTest : public FrameFetchContextTest {
+ protected:
+  void SetUp() override {
+    FrameFetchContextTest::SetUp();
+    m_filteredLoadCallbackCounter = 0;
+  }
+
+  void TearDown() override {
+    document->loader()->setSubresourceFilter(nullptr);
+    FrameFetchContextTest::TearDown();
+  }
+
+  int getFilteredLoadCallCount() const { return m_filteredLoadCallbackCounter; }
+
+  void setFilterPolicy(WebDocumentSubresourceFilter::LoadPolicy policy) {
+    document->loader()->setSubresourceFilter(
+        WTF::makeUnique<FixedPolicySubresourceFilter>(
+            policy, &m_filteredLoadCallbackCounter));
+  }
+
+  ResourceRequestBlockedReason canRequest() {
+    return canRequestInternal(
+        FetchContext::SecurityViolationReportingPolicy::Report);
+  }
+
+  ResourceRequestBlockedReason canRequestPreload() {
+    return canRequestInternal(
+        FetchContext::SecurityViolationReportingPolicy::SuppressReporting);
+  }
+
+ private:
+  ResourceRequestBlockedReason canRequestInternal(
+      FetchContext::SecurityViolationReportingPolicy reportingPolicy) {
+    KURL inputURL(ParsedURLString, "http://example.com/");
+    ResourceRequest resourceRequest(inputURL);
+    return fetchContext->canRequest(
+        Resource::Image, resourceRequest, inputURL, ResourceLoaderOptions(),
+        reportingPolicy, FetchRequest::UseDefaultOriginRestrictionForType);
+  }
+
+  int m_filteredLoadCallbackCounter;
+};
+
 // This test class sets up a mock frame loader client.
 class FrameFetchContextMockedFrameLoaderClientTest
     : public FrameFetchContextTest {
  protected:
   void SetUp() override {
     url = KURL(KURL(), "https://example.test/foo");
     mainResourceUrl = KURL(KURL(), "https://www.example.test");
     client = new testing::NiceMock<MockFrameLoaderClient>();
     dummyPageHolder =
         DummyPageHolder::create(IntSize(500, 500), nullptr, client);
@@ skipping 663 matching lines @@
     ResourceRequest mainRequest(test.url);
     fetchContext->addAdditionalRequestHeaders(mainRequest, FetchMainResource);
     EXPECT_EQ(test.isExternalExpectation, mainRequest.isExternalRequest());
 
     ResourceRequest subRequest(test.url);
     fetchContext->addAdditionalRequestHeaders(subRequest, FetchSubresource);
     EXPECT_EQ(test.isExternalExpectation, subRequest.isExternalRequest());
   }
 }
 
+TEST_F(FrameFetchContextSubresourceFilterTest, Filter) {
+  setFilterPolicy(WebDocumentSubresourceFilter::Disallow);
+
+  EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
+  EXPECT_EQ(1, getFilteredLoadCallCount());
+
+  EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
+  EXPECT_EQ(2, getFilteredLoadCallCount());
+
+  EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter,
+            canRequestPreload());
+  EXPECT_EQ(2, getFilteredLoadCallCount());
+
+  EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
+  EXPECT_EQ(3, getFilteredLoadCallCount());
+}
+
+TEST_F(FrameFetchContextSubresourceFilterTest, Allow) {
+  setFilterPolicy(WebDocumentSubresourceFilter::Allow);
+
+  EXPECT_EQ(ResourceRequestBlockedReason::None, canRequest());
+  EXPECT_EQ(0, getFilteredLoadCallCount());
+
+  EXPECT_EQ(ResourceRequestBlockedReason::None, canRequestPreload());
+  EXPECT_EQ(0, getFilteredLoadCallCount());
+}
+
+TEST_F(FrameFetchContextSubresourceFilterTest, WouldDisallow) {
+  setFilterPolicy(WebDocumentSubresourceFilter::WouldDisallow);
+
+  EXPECT_EQ(ResourceRequestBlockedReason::None, canRequest());
+  EXPECT_EQ(0, getFilteredLoadCallCount());
+
+  EXPECT_EQ(ResourceRequestBlockedReason::None, canRequestPreload());
+  EXPECT_EQ(0, getFilteredLoadCallCount());
+}
+
 }  // namespace blink