chrome/browser/safe_browsing/safe_browsing_store_file_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h"
   6
   7 #include "base/bind.h"
   8 #include "base/files/scoped_temp_dir.h"
   9 #include "base/md5.h"
  10 #include "chrome/browser/safe_browsing/safe_browsing_store_unittest_helper.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "testing/platform_test.h"
  13
  14 namespace {
  15
  16 class SafeBrowsingStoreFileTest : public PlatformTest {
  17  public:
  18   virtual void SetUp() {
  19     PlatformTest::SetUp();
  20
  21     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
  22
  23     filename_ = temp_dir_.path();
  24     filename_ = filename_.AppendASCII("SafeBrowsingTestStore");
  25
  26     store_.reset(new SafeBrowsingStoreFile());
  27     store_->Init(filename_,
  28                  base::Bind(&SafeBrowsingStoreFileTest::OnCorruptionDetected,
  29                             base::Unretained(this)));
  30     corruption_detected_ = false;
  31   }
  32   virtual void TearDown() {
  33     if (store_.get())
  34       store_->Delete();
  35     store_.reset();
  36
  37     PlatformTest::TearDown();
  38   }
  39
  40   void OnCorruptionDetected() {
  41     corruption_detected_ = true;
  42   }
  43
  44   base::ScopedTempDir temp_dir_;
  45   base::FilePath filename_;
  46   scoped_ptr<SafeBrowsingStoreFile> store_;
  47   bool corruption_detected_;
  48 };
  49
  50 TEST_STORE(SafeBrowsingStoreFileTest, store_.get(), filename_);
  51
  52 // Test that Delete() deletes the temporary store, if present.
  53 TEST_F(SafeBrowsingStoreFileTest, DeleteTemp) {
  54   const base::FilePath temp_file =
  55       SafeBrowsingStoreFile::TemporaryFileForFilename(filename_);
  56
  57   EXPECT_FALSE(base::PathExists(filename_));
  58   EXPECT_FALSE(base::PathExists(temp_file));
  59
  60   // Starting a transaction creates a temporary file.
  61   EXPECT_TRUE(store_->BeginUpdate());
  62   EXPECT_TRUE(base::PathExists(temp_file));
  63
  64   // Pull the rug out from under the existing store, simulating a
  65   // crash.
  66   store_.reset(new SafeBrowsingStoreFile());
  67   store_->Init(filename_, base::Closure());
  68   EXPECT_FALSE(base::PathExists(filename_));
  69   EXPECT_TRUE(base::PathExists(temp_file));
  70
  71   // Make sure the temporary file is deleted.
  72   EXPECT_TRUE(store_->Delete());
  73   EXPECT_FALSE(base::PathExists(filename_));
  74   EXPECT_FALSE(base::PathExists(temp_file));
  75 }
  76
  77 // Test basic corruption-handling.
  78 TEST_F(SafeBrowsingStoreFileTest, DetectsCorruption) {
  79   // Load a store with some data.
  80   SafeBrowsingStoreTestStorePrefix(store_.get());
  81
  82   // Can successfully open and read the store.
  83   std::vector<SBAddFullHash> pending_adds;
  84   SBAddPrefixes orig_prefixes;
  85   std::vector<SBAddFullHash> orig_hashes;
  86   EXPECT_TRUE(store_->BeginUpdate());
  87   EXPECT_TRUE(store_->FinishUpdate(pending_adds, &orig_prefixes, &orig_hashes));
  88   EXPECT_GT(orig_prefixes.size(), 0U);
  89   EXPECT_GT(orig_hashes.size(), 0U);
  90   EXPECT_FALSE(corruption_detected_);
  91
  92   // Corrupt the store.
  93   file_util::ScopedFILE file(base::OpenFile(filename_, "rb+"));
  94   const long kOffset = 60;
  95   EXPECT_EQ(fseek(file.get(), kOffset, SEEK_SET), 0);
  96   const int32 kZero = 0;
  97   int32 previous = kZero;
  98   EXPECT_EQ(fread(&previous, sizeof(previous), 1, file.get()), 1U);
  99   EXPECT_NE(previous, kZero);
 100   EXPECT_EQ(fseek(file.get(), kOffset, SEEK_SET), 0);
 101   EXPECT_EQ(fwrite(&kZero, sizeof(kZero), 1, file.get()), 1U);
 102   file.reset();
 103
 104   // Update fails and corruption callback is called.
 105   SBAddPrefixes add_prefixes;
 106   std::vector<SBAddFullHash> add_hashes;
 107   corruption_detected_ = false;
 108   EXPECT_TRUE(store_->BeginUpdate());
 109   EXPECT_FALSE(store_->FinishUpdate(pending_adds, &add_prefixes, &add_hashes));
 110   EXPECT_TRUE(corruption_detected_);
 111   EXPECT_EQ(add_prefixes.size(), 0U);
 112   EXPECT_EQ(add_hashes.size(), 0U);
 113
 114   // Make it look like there is a lot of add-chunks-seen data.
 115   const long kAddChunkCountOffset = 2 * sizeof(int32);
 116   const int32 kLargeCount = 1000 * 1000 * 1000;
 117   file.reset(base::OpenFile(filename_, "rb+"));
 118   EXPECT_EQ(fseek(file.get(), kAddChunkCountOffset, SEEK_SET), 0);
 119   EXPECT_EQ(fwrite(&kLargeCount, sizeof(kLargeCount), 1, file.get()), 1U);
 120   file.reset();
 121
 122   // Detects corruption and fails to even begin the update.
 123   corruption_detected_ = false;
 124   EXPECT_FALSE(store_->BeginUpdate());
 125   EXPECT_TRUE(corruption_detected_);
 126 }
 127
 128 TEST_F(SafeBrowsingStoreFileTest, CheckValidity) {
 129   // Empty store is valid.
 130   EXPECT_FALSE(base::PathExists(filename_));
 131   ASSERT_TRUE(store_->BeginUpdate());
 132   EXPECT_FALSE(corruption_detected_);
 133   EXPECT_TRUE(store_->CheckValidity());
 134   EXPECT_FALSE(corruption_detected_);
 135   EXPECT_TRUE(store_->CancelUpdate());
 136
 137   // A store with some data is valid.
 138   EXPECT_FALSE(base::PathExists(filename_));
 139   SafeBrowsingStoreTestStorePrefix(store_.get());
 140   EXPECT_TRUE(base::PathExists(filename_));
 141   ASSERT_TRUE(store_->BeginUpdate());
 142   EXPECT_FALSE(corruption_detected_);
 143   EXPECT_TRUE(store_->CheckValidity());
 144   EXPECT_FALSE(corruption_detected_);
 145   EXPECT_TRUE(store_->CancelUpdate());
 146 }
 147
 148 // Corrupt the payload.
 149 TEST_F(SafeBrowsingStoreFileTest, CheckValidityPayload) {
 150   SafeBrowsingStoreTestStorePrefix(store_.get());
 151   EXPECT_TRUE(base::PathExists(filename_));
 152
 153   // 37 is the most random prime number.  It's also past the header,
 154   // as corrupting the header would fail BeginUpdate() in which case
 155   // CheckValidity() cannot be called.
 156   const size_t kOffset = 37;
 157
 158   {
 159     file_util::ScopedFILE file(base::OpenFile(filename_, "rb+"));
 160     EXPECT_EQ(0, fseek(file.get(), kOffset, SEEK_SET));
 161     EXPECT_GE(fputs("hello", file.get()), 0);
 162   }
 163   ASSERT_TRUE(store_->BeginUpdate());
 164   EXPECT_FALSE(corruption_detected_);
 165   EXPECT_FALSE(store_->CheckValidity());
 166   EXPECT_TRUE(corruption_detected_);
 167   EXPECT_TRUE(store_->CancelUpdate());
 168 }
 169
 170 // Corrupt the checksum.
 171 TEST_F(SafeBrowsingStoreFileTest, CheckValidityChecksum) {
 172   SafeBrowsingStoreTestStorePrefix(store_.get());
 173   EXPECT_TRUE(base::PathExists(filename_));
 174
 175   // An offset from the end of the file which is in the checksum.
 176   const int kOffset = -static_cast<int>(sizeof(base::MD5Digest));
 177
 178   {
 179     file_util::ScopedFILE file(base::OpenFile(filename_, "rb+"));
 180     EXPECT_EQ(0, fseek(file.get(), kOffset, SEEK_END));
 181     EXPECT_GE(fputs("hello", file.get()), 0);
 182   }
 183   ASSERT_TRUE(store_->BeginUpdate());
 184   EXPECT_FALSE(corruption_detected_);
 185   EXPECT_FALSE(store_->CheckValidity());
 186   EXPECT_TRUE(corruption_detected_);
 187   EXPECT_TRUE(store_->CancelUpdate());
 188 }
 189
 190 }  // namespace