Certificate Transparency: New component for obtaining fresh STHs.

chrome/browser/component_updater/sth_set_component_installer_unittest.cc

Index: chrome/browser/component_updater/sth_set_component_installer_unittest.cc
diff --git a/chrome/browser/component_updater/sth_set_component_installer_unittest.cc b/chrome/browser/component_updater/sth_set_component_installer_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..94cc21dfa16807055651b9ad101cabb58fd9a972
--- /dev/null
+++ b/chrome/browser/component_updater/sth_set_component_installer_unittest.cc
@@ -0,0 +1,155 @@
+// Copyright 2016 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 "chrome/browser/component_updater/sth_set_component_installer.h"
+
+#include <map>
+#include <string>
+
+#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/scoped_ptr.h"
+#include "base/run_loop.h"
+#include "base/strings/string_util.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/values.h"
+#include "base/version.h"
+#include "components/safe_json/testing_json_parser.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "net/cert/signed_tree_head.h"
+#include "net/cert/sth_observer.h"
+#include "net/test/ct_test_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+
+namespace component_updater {
+
+class StoringSTHObserver : public net::ct::STHObserver {
+ public:
+  void NewSTHObserved(const net::ct::SignedTreeHead& sth) override {
+    sths[sth.log_id] = sth;
+  }
+
+  std::map<std::string, net::ct::SignedTreeHead> sths;
+};
+
+class STHSetComponentInstallerTest : public PlatformTest {
+ public:
+  STHSetComponentInstallerTest() {}
+  void SetUp() override {
+    PlatformTest::SetUp();
+
+    // ScopedTempDir automatically does a recursive delete on the entire
+    // directory in its destructor, so no cleanup is required in TearDown.
+    // Note that all files created by this test case are created within the
+    // directory that is created here, so even though they are not explicitly
+    // created *as temp files*, they will still get cleaned up automagically.
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+
+    scoped_ptr<StoringSTHObserver> observer(new StoringSTHObserver());
+    observer_ = observer.get();
+    traits_.reset(new STHSetComponentInstallerTraits(std::move(observer)));
+  }
+
+  void WriteSTHToFile(const std::string& sth_json,
+                      const base::FilePath& filename) {
+    ASSERT_EQ(static_cast<int32_t>(sth_json.length()),
+              base::WriteFile(filename, sth_json.data(), sth_json.length()));
+  }
+
+  base::FilePath GetSTHsDir() {
+    return temp_dir_.path()
+        .Append(FILE_PATH_LITERAL("_platform_specific"))
+        .Append(FILE_PATH_LITERAL("all"))
+        .Append(FILE_PATH_LITERAL("sths"));
+  }
+
+  void CreateSTHsDir(const base::DictionaryValue& manifest,
+                     const base::FilePath& sths_dir) {
+    ASSERT_FALSE(traits_->VerifyInstallation(manifest, temp_dir_.path()));
+    ASSERT_TRUE(base::CreateDirectory(sths_dir));
+  }
+
+  void LoadSTHs(const base::DictionaryValue& manifest,
+                const base::FilePath& sths_dir) {
+    ASSERT_TRUE(traits_->VerifyInstallation(manifest, temp_dir_.path()));
+
+    const base::Version v("1.0");
+    traits_->LoadSTHsFromDisk(sths_dir, v);
+    // Drain the RunLoop created by the TestBrowserThreadBundle
+    base::RunLoop().RunUntilIdle();
+  }
+
+ protected:
+  content::TestBrowserThreadBundle thread_bundle_;
+
+  base::ScopedTempDir temp_dir_;
+  scoped_ptr<STHSetComponentInstallerTraits> traits_;
+  StoringSTHObserver* observer_;
+  safe_json::TestingJsonParser::ScopedFactoryOverride factory_override_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(STHSetComponentInstallerTest);
+};
+
+// Parses valid STH JSON in a file with valid hex encoding of log id.
+TEST_F(STHSetComponentInstallerTest, CanLoadAllSTHs) {
+  const base::DictionaryValue manifest;
+  const base::FilePath sths_dir(GetSTHsDir());
+  CreateSTHsDir(manifest, sths_dir);
+
+  const std::string good_sth_json = net::ct::GetSampleSTHAsJson();
+  const base::FilePath first_sth_file =
+      sths_dir.Append(FILE_PATH_LITERAL("616263.sth"));
+  WriteSTHToFile(good_sth_json, first_sth_file);
+
+  const base::FilePath second_sth_file =
+      sths_dir.Append(FILE_PATH_LITERAL("610064.sth"));
+  WriteSTHToFile(good_sth_json, second_sth_file);
+
+  LoadSTHs(manifest, sths_dir);
+
+  EXPECT_EQ(2u, observer_->sths.size());
+
+  const std::string first_log_id("abc");
+  ASSERT_TRUE(observer_->sths.find(first_log_id) != observer_->sths.end());
+  const net::ct::SignedTreeHead& first_sth(observer_->sths[first_log_id]);
+  EXPECT_EQ(21u, first_sth.tree_size);
+
+  const std::string second_log_id("a\00d", 3);
+  ASSERT_TRUE(observer_->sths.find(second_log_id) != observer_->sths.end());
+}
+
+// Does not notify of invalid STH JSON.
+TEST_F(STHSetComponentInstallerTest, DoesNotLoadInvalidJSON) {
+  const base::DictionaryValue manifest;
+  const base::FilePath sths_dir(GetSTHsDir());
+  CreateSTHsDir(manifest, sths_dir);
+
+  const base::FilePath invalid_sth =
+      sths_dir.Append(FILE_PATH_LITERAL("010101.sth"));
+  WriteSTHToFile(std::string("{invalid json}"), invalid_sth);
+
+  LoadSTHs(manifest, sths_dir);
+  EXPECT_EQ(0u, observer_->sths.size());
+}
+
+// Does not notify of valid JSON but in a file not hex-encoded log id.
+TEST_F(STHSetComponentInstallerTest,
+       DoesNotLoadValidJSONFromFileNotHexEncoded) {
+  const base::DictionaryValue manifest;
+  const base::FilePath sths_dir(GetSTHsDir());
+  CreateSTHsDir(manifest, sths_dir);
+
+  const base::FilePath not_hex_sth_file =
+      sths_dir.Append(FILE_PATH_LITERAL("nothex.sth"));
+  WriteSTHToFile(net::ct::GetSampleSTHAsJson(), not_hex_sth_file);
+
+  LoadSTHs(manifest, sths_dir);
+  EXPECT_EQ(0u, observer_->sths.size());
+}
+
+}  // namespace component_updater