[subresource_filter] Display dev tools console message for subresource blocking

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

 // Copyright 2017 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 "core/loader/SubresourceFilter.h"
 
 #include <utility>
 
+#include "core/dom/Document.h"
 #include "core/dom/TaskRunnerHelper.h"
+#include "core/frame/LocalFrame.h"
+#include "core/inspector/ConsoleMessage.h"
 #include "platform/WebTaskRunner.h"
 #include "platform/weborigin/KURL.h"
+#include "platform/wtf/text/StringBuilder.h"
 #include "public/platform/WebTraceLocation.h"
 
 namespace blink {
 
+namespace {
+
+String GetErrorStringForDisallowedLoad(const KURL& url) {
+  // TODO(shivanisha): Update the string when finalized.
+  StringBuilder builder;
+  builder.Append("Subresource filtering disallowed loading this resource, ");
+  builder.Append(url.GetString());
+  builder.Append(".");
+  return builder.ToString();
+}
+
+}  // namespace
+
 // static
 SubresourceFilter* SubresourceFilter::Create(
     DocumentLoader* loader,
     std::unique_ptr<WebDocumentSubresourceFilter> filter) {
   return new SubresourceFilter(loader, std::move(filter));
 }
 
 SubresourceFilter::SubresourceFilter(
     DocumentLoader* document_loader,
     std::unique_ptr<WebDocumentSubresourceFilter> subresource_filter)
     : document_loader_(document_loader),
       subresource_filter_(std::move(subresource_filter)) {}
 
 SubresourceFilter::~SubresourceFilter() {}
 
 bool SubresourceFilter::AllowLoad(
     const KURL& resource_url,
     WebURLRequest::RequestContext request_context,
     SecurityViolationReportingPolicy reporting_policy) {
   // TODO(csharrison): Implement a caching layer here which is a HashMap of
   // Pair<url string, context> -> LoadPolicy.
   WebDocumentSubresourceFilter::LoadPolicy load_policy =
       subresource_filter_->GetLoadPolicy(resource_url, request_context);
+
   if (reporting_policy == SecurityViolationReportingPolicy::kReport)
-    ReportLoad(load_policy);
+    ReportLoad(resource_url, load_policy);
   return load_policy != WebDocumentSubresourceFilter::kDisallow;
 }
 
 bool SubresourceFilter::AllowWebSocketConnection(const KURL& url) {
   WebDocumentSubresourceFilter::LoadPolicy load_policy =
       subresource_filter_->GetLoadPolicyForWebSocketConnect(url);
 
   // Post a task to notify this load to avoid unduly blocking the worker
   // thread. Note that this unconditionally calls reportLoad unlike allowLoad,
   // because there aren't developer-invisible connections (like speculative
   // preloads) happening here.
   RefPtr<WebTaskRunner> task_runner = TaskRunnerHelper::Get(
       TaskType::kNetworking, document_loader_->GetFrame());
   DCHECK(task_runner->RunsTasksOnCurrentThread());
   task_runner->PostTask(BLINK_FROM_HERE,
                         WTF::Bind(&SubresourceFilter::ReportLoad,
-                                  WrapPersistent(this), load_policy));
+                                  WrapPersistent(this), url, load_policy));
   return load_policy != WebDocumentSubresourceFilter::kDisallow;
 }
 
 void SubresourceFilter::ReportLoad(
+    const KURL& resource_url,
     WebDocumentSubresourceFilter::LoadPolicy load_policy) {
+  Document* document = document_loader_->GetFrame()
+                           ? document_loader_->GetFrame()->GetDocument()
+                           : nullptr;
   switch (load_policy) {
     case WebDocumentSubresourceFilter::kAllow:
       break;
     case WebDocumentSubresourceFilter::kDisallow:
       subresource_filter_->ReportDisallowedLoad();
+      if (document && subresource_filter_->ShouldLogToConsole()) {
+        document->AddConsoleMessage(ConsoleMessage::Create(
+            kJSMessageSource, kErrorMessageLevel,
+            GetErrorStringForDisallowedLoad(resource_url)));
+      }
     // fall through
     case WebDocumentSubresourceFilter::kWouldDisallow:
-      // TODO(csharrison): log console errors here based on

[Charlie Harrison, 2017/05/23 20:35:51]

Shouldn't we put it here so this will log if enable_logging is set with DRYRUN?

[shivanisha, 2017/05/23 20:40:01]

I think for warning list we don't want to log for each subresource but just for
the document. WDYT?

[Charlie Harrison, 2017/05/23 20:42:04]

SGTM. Can you add a quick comment to that effect?

[shivanisha, 2017/05/24 16:39:54]

Added a comment

-      // subresource_filter_->ShouldLogToConsole().
       document_loader_->DidObserveLoadingBehavior(
           kWebLoadingBehaviorSubresourceFilterMatch);
       break;
   }
 }
 
 }  // namespace blink