[subresource_filter] Add matching perftest

Bug: 810203
Change-Id: I69036af7dab173d84afaf39b52653461225d3e3b
Reviewed-on: https://chromium-review.googlesource.com/909413Reviewed-by: Cait Phillips <caitkp@chromium.org>
Reviewed-by: Josh Karlin <jkarlin@chromium.org>
Commit-Queue: Charlie Harrison <csharrison@chromium.org>
Cr-Commit-Position: refs/heads/master@{#544029}

components/BUILD.gn

@@ -543,6 +543,7 @@ if (!is_ios) {
     sources = [
       "discardable_memory/common/discardable_shared_memory_heap_perftest.cc",
       "omnibox/browser/history_quick_provider_performance_unittest.cc",
+      "subresource_filter/core/common/perftests/indexed_ruleset_perftest.cc",
       "test/run_all_perftests.cc",
       "visitedlink/test/visitedlink_perftest.cc",
     ]
@@ -553,6 +554,8 @@ if (!is_ios) {
       "//components/history/core/test",
       "//components/omnibox/browser",
       "//components/omnibox/browser:test_support",
+      "//components/subresource_filter/core/common",
+      "//components/subresource_filter/core/common:tools_lib",
       "//components/test:test_support",
       "//components/visitedlink/browser",
       "//testing/perf",
@@ -567,6 +570,7 @@ if (!is_ios) {
     data = [
       "$root_out_dir/components_tests_resources.pak",
       "$root_out_dir/ui_test.pak",
+      "//components/subresource_filter/core/common/perftests/data",
     ]
 
     if (is_android) {

components/subresource_filter/core/common/perftests/indexed_ruleset_perftest.cc

+// Copyright 2018 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 <algorithm>
+#include <fstream>
+#include <memory>
+#include <sstream>
+#include <string>
+#include <utility>
+
+#include "base/files/file.h"
+#include "base/files/file_path.h"
+#include "base/files/file_util.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_refptr.h"
+#include "base/path_service.h"
+#include "base/time/time.h"
+#include "base/timer/elapsed_timer.h"
+#include "components/subresource_filter/core/common/memory_mapped_ruleset.h"
+#include "components/subresource_filter/core/common/tools/filter_tool.h"
+#include "components/subresource_filter/core/common/tools/indexing_tool.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/perf/perf_test.h"
+
+namespace subresource_filter {
+
+class IndexedRulesetPerftest : public testing::Test {
+ public:
+  IndexedRulesetPerftest() {}
+
+  ~IndexedRulesetPerftest() override {}
+
+  void SetUp() override {
+    base::FilePath dir_path;
+    PathService::Get(base::DIR_SOURCE_ROOT, &dir_path);
+    base::FilePath request_path = dir_path.AppendASCII(
+        "components/subresource_filter/core/common/perftests/"
+        "/data/httparchive_request_corpus.csv");
+    base::ReadFileToString(request_path, &requests_);
+
+    unindexed_path_ = dir_path.AppendASCII(
+        "components/subresource_filter/core/common/perftests/"
+        "data/UnindexedRules_7.54");
+
+    ASSERT_TRUE(scoped_dir_.CreateUniqueTempDir());
+    base::FilePath indexed_path =
+        scoped_dir_.GetPath().AppendASCII("IndexedRuleset");
+    ASSERT_TRUE(base::PathExists(unindexed_path_));
+    ASSERT_TRUE(base::DirectoryExists(indexed_path.DirName()));
+    ASSERT_TRUE(IndexAndWriteRuleset(unindexed_path_, indexed_path));
+
+    base::File indexed_file =
+        base::File(indexed_path, base::File::FLAG_OPEN | base::File::FLAG_READ);
+    ASSERT_TRUE(indexed_file.IsValid());
+    auto ruleset =
+        base::MakeRefCounted<subresource_filter::MemoryMappedRuleset>(
+            std::move(indexed_file));
+    filter_tool_ = std::make_unique<FilterTool>(std::move(ruleset), &output_);
+  }
+
+  FilterTool* filter_tool() { return filter_tool_.get(); }
+
+  const std::string& requests() const { return requests_; }
+
+  const base::FilePath& unindexed_path() const { return unindexed_path_; }
+
+ private:
+  base::ScopedTempDir scoped_dir_;
+  base::FilePath unindexed_path_;
+
+  std::string requests_;
+
+  // Use an unopened output stream as a sort of null stream. All writes will
+  // fail so things should be a bit faster than writing to a string.
+  std::ofstream output_;
+
+  std::unique_ptr<FilterTool> filter_tool_;
+  DISALLOW_COPY_AND_ASSIGN(IndexedRulesetPerftest);
+};
+
+TEST_F(IndexedRulesetPerftest, IndexRuleset) {
+  base::ScopedTempDir scoped_dir;
+  ASSERT_TRUE(scoped_dir.CreateUniqueTempDir());
+  base::FilePath indexed_path =
+      scoped_dir.GetPath().AppendASCII("IndexedRuleset");
+
+  base::ElapsedTimer timer;
+  ASSERT_TRUE(IndexAndWriteRuleset(unindexed_path(), indexed_path));
+  perf_test::PrintResult("index_and_write_time", "", "",
+                         static_cast<size_t>(timer.Elapsed().InMicroseconds()),
+                         "microseconds", true /* important */);
+}
+
+TEST_F(IndexedRulesetPerftest, MatchAll) {
+  std::vector<int64_t> results;
+  for (int i = 0; i < 5; ++i) {
+    base::ElapsedTimer timer;
+    std::istringstream request_stream(requests());
+    filter_tool()->MatchBatch(&request_stream);
+    results.push_back(timer.Elapsed().InMicroseconds());
+  }
+  std::sort(results.begin(), results.end());
+  perf_test::PrintResult("median_match_time", "", "",
+                         static_cast<size_t>(results[2]), "microseconds",
+                         true /* important */);
+}
+
+}  // namespace subresource_filter