Introduce AsyncDocumentSubresourceFilter.

components/subresource_filter/content/browser/async_document_subresource_filter.cc

Index: components/subresource_filter/content/browser/async_document_subresource_filter.cc
diff --git a/components/subresource_filter/content/browser/async_document_subresource_filter.cc b/components/subresource_filter/content/browser/async_document_subresource_filter.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2716f890169f5e50d6e51efc6763ca15c5e5391b
--- /dev/null
+++ b/components/subresource_filter/content/browser/async_document_subresource_filter.cc
@@ -0,0 +1,137 @@
+// 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 "components/subresource_filter/content/browser/async_document_subresource_filter.h"
+
+#include "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/callback.h"
+#include "base/location.h"
+#include "base/logging.h"
+#include "base/task_runner_util.h"
+#include "base/threading/sequenced_task_runner_handle.h"
+#include "components/subresource_filter/core/common/memory_mapped_ruleset.h"
+
+namespace subresource_filter {
+
+// AsyncDocumentSubresourceFilter::InitializationParams ------------------------
+
+using InitializationParams =
+    AsyncDocumentSubresourceFilter::InitializationParams;
+
+InitializationParams::InitializationParams() = default;
+
+InitializationParams::InitializationParams(GURL document_url,
+                                           ActivationLevel activation_level,
+                                           bool measure_performance)
+    : document_url(std::move(document_url)),
+      parent_activation_state(activation_level) {
+  parent_activation_state.measure_performance = measure_performance;
+}
+
+InitializationParams::InitializationParams(
+    GURL document_url,
+    url::Origin parent_document_origin,
+    ActivationState parent_activation_state)
+    : document_url(std::move(document_url)),
+      parent_document_origin(std::move(parent_document_origin)),
+      parent_activation_state(parent_activation_state) {}
+
+InitializationParams::~InitializationParams() = default;
+InitializationParams::InitializationParams(InitializationParams&&) = default;
+InitializationParams& InitializationParams::operator=(InitializationParams&&) =
+    default;
+
+// AsyncDocumentSubresourceFilter ----------------------------------------------
+
+AsyncDocumentSubresourceFilter::AsyncDocumentSubresourceFilter(
+    VerifiedRuleset::Handle* ruleset_handle,
+    InitializationParams params,
+    base::Callback<void(ActivationState)> activation_state_callback,
+    base::OnceClosure first_disallowed_load_callback)
+    : task_runner_(ruleset_handle->task_runner()),
+      core_(new Core(), base::OnTaskRunnerDeleter(task_runner_)),
+      first_disallowed_load_callback_(
+          std::move(first_disallowed_load_callback)) {
+  // Note: It is safe to post |ruleset_handle|'s VerifiedRuleset pointer,
+  // because a task to delete it can only be posted to (and, therefore,
+  // processed by) |task_runner| after this method returns, hence after the
+  // below task is posted.
+  base::PostTaskAndReplyWithResult(
+      task_runner_, FROM_HERE,
+      base::Bind(&Core::Initialize, base::Unretained(core_.get()),
+                 base::Passed(&params), ruleset_handle->ruleset_.get()),
+      std::move(activation_state_callback));
+}
+
+AsyncDocumentSubresourceFilter::~AsyncDocumentSubresourceFilter() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+}
+
+void AsyncDocumentSubresourceFilter::GetLoadPolicyForSubdocument(
+    const GURL& subdocument_url,
+    LoadPolicyCallback result_callback) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  auto compute_load_policy = [](AsyncDocumentSubresourceFilter::Core* core,
+                                const GURL& subdocument_url) {
+    DCHECK(core);
+    DocumentSubresourceFilter* filter = core->filter();
+    return filter ? filter->GetLoadPolicyForSubdocument(subdocument_url)
+                  : LoadPolicy::Allow;
+  };
+
+  // TODO(pkalinnikov): Think about avoiding copy of |subdocument_url| if it is
+  // too big and won't be allowed anyway (e.g., it's a data: URI).
+  base::PostTaskAndReplyWithResult(
+      task_runner_, FROM_HERE,
+      base::Bind(compute_load_policy, core_.get(), subdocument_url),
+      std::move(result_callback));
+}
+
+void AsyncDocumentSubresourceFilter::ReportDisallowedLoad() {
+  if (!first_disallowed_load_callback_.is_null())
+    std::move(first_disallowed_load_callback_).Run();
+
+  auto report_disallowed_load = [](AsyncDocumentSubresourceFilter::Core* core) {
+    DCHECK(core);
+    DocumentSubresourceFilter* filter = core->filter();
+    if (filter)
+      filter->reportDisallowedLoad();
+  };
+  task_runner_->PostTask(FROM_HERE,
+                         base::Bind(report_disallowed_load, core_.get()));
+}
+
+// AsyncDocumentSubresourceFilter::Core ----------------------------------------
+
+AsyncDocumentSubresourceFilter::Core::Core() {
+  thread_checker_.DetachFromThread();
+}
+
+AsyncDocumentSubresourceFilter::Core::~Core() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+}
+
+ActivationState AsyncDocumentSubresourceFilter::Core::Initialize(
+    InitializationParams params,
+    VerifiedRuleset* verified_ruleset) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(verified_ruleset);
+
+  const MemoryMappedRuleset* ruleset = verified_ruleset->Get();
+  if (!ruleset)
+    return ActivationState(ActivationLevel::DISABLED);
+
+  ActivationState activation_state =
+      ComputeActivationState(params.document_url, params.parent_document_origin,
+                             params.parent_activation_state, ruleset);
+
+  filter_.emplace(url::Origin(params.document_url), activation_state, ruleset,

[Charlie Harrison, 2017/02/13 20:07:50]

Can you wrap this in a if (activation_state.activation_level !=
ActivationLevel::DISABLED)?


Otherwise, if we try to initialize an ADSF with a main frame whose page isn't
activated, we'll DCHECK in DocumentSubresourceFilter's constructor.

If we don't want to allow creating a main frame ADSF with page-level activation
disabled, let's not include the activation level bool in the "main frame"
initialization param constructor.

Also, a test that goes through this path would be appreciated.

[pkalinnikov, 2017/02/14 09:31:07]

Just like with DocumentSubresourceFilter, let's not try creating an ADSF with
DISABLED activation level in the first place. For this purpose I added DCHECKs
here and in couple places above.

As for tests, there is no need to exercise this path any more, because it's
forbidden.

We can't completely eliminate ActivationLevel from initialization param
constructor, because there should still be an option between ENABLED and DRYRUN.
I could to an |is_dry_run| bool instead, but this seems less future-proof.

Does this sound good?

[Charlie Harrison, 2017/02/14 13:07:40]

I prefer just DCHECKing the activation state is not DISABLED, rather than using
an opaque bool. What you have now lgtm

+                  base::OnceClosure());
+
+  return activation_state;
+}
+
+}  // namespace subresource_filter