| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/safe_browsing_db/prefix_set.h" | 5 #include "components/safe_browsing_db/prefix_set.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <iterator> | 8 #include <iterator> |
| 9 #include <set> | 9 #include <set> |
| 10 #include <string> | 10 #include <string> |
| 11 | 11 |
| 12 #include "base/files/file_util.h" | 12 #include "base/files/file_util.h" |
| 13 #include "base/files/scoped_file.h" | 13 #include "base/files/scoped_file.h" |
| 14 #include "base/files/scoped_temp_dir.h" | 14 #include "base/files/scoped_temp_dir.h" |
| 15 #include "base/logging.h" | 15 #include "base/logging.h" |
| 16 #include "base/md5.h" | 16 #include "base/md5.h" |
| 17 #include "base/memory/scoped_ptr.h" | 17 #include "base/memory/scoped_ptr.h" |
| 18 #include "base/path_service.h" | 18 #include "base/path_service.h" |
| 19 #include "base/rand_util.h" | 19 #include "base/rand_util.h" |
| 20 #include "base/strings/string_number_conversions.h" | 20 #include "base/strings/string_number_conversions.h" |
| 21 #include "base/strings/string_util.h" | 21 #include "base/strings/string_util.h" |
| 22 #include "components/safe_browsing_db/util.h" | 22 #include "components/safe_browsing_db/util.h" |
| 23 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
| 24 #include "testing/platform_test.h" | 24 #include "testing/platform_test.h" |
| 25 | 25 |
| 26 namespace safe_browsing { |
| 27 |
| 26 namespace { | 28 namespace { |
| 27 | 29 |
| 28 const SBPrefix kHighBitClear = 1000u * 1000u * 1000u; | 30 const SBPrefix kHighBitClear = 1000u * 1000u * 1000u; |
| 29 const SBPrefix kHighBitSet = 3u * 1000u * 1000u * 1000u; | 31 const SBPrefix kHighBitSet = 3u * 1000u * 1000u * 1000u; |
| 30 | 32 |
| 31 } // namespace | 33 } // namespace |
| 32 | 34 |
| 33 namespace safe_browsing { | |
| 34 | |
| 35 class PrefixSetTest : public PlatformTest { | 35 class PrefixSetTest : public PlatformTest { |
| 36 protected: | 36 protected: |
| 37 // Constants for the v1 format. | 37 // Constants for the v1 format. |
| 38 static const size_t kMagicOffset = 0 * sizeof(uint32); | 38 static const size_t kMagicOffset = 0 * sizeof(uint32); |
| 39 static const size_t kVersionOffset = 1 * sizeof(uint32); | 39 static const size_t kVersionOffset = 1 * sizeof(uint32); |
| 40 static const size_t kIndexSizeOffset = 2 * sizeof(uint32); | 40 static const size_t kIndexSizeOffset = 2 * sizeof(uint32); |
| 41 static const size_t kDeltasSizeOffset = 3 * sizeof(uint32); | 41 static const size_t kDeltasSizeOffset = 3 * sizeof(uint32); |
| 42 static const size_t kFullHashesSizeOffset = 4 * sizeof(uint32); | 42 static const size_t kFullHashesSizeOffset = 4 * sizeof(uint32); |
| 43 static const size_t kPayloadOffset = 5 * sizeof(uint32); | 43 static const size_t kPayloadOffset = 5 * sizeof(uint32); |
| 44 | 44 |
| (...skipping 674 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 719 const SBFullHash kHash2 = SBFullHashForString("www.evil.com/phishing.html"); | 719 const SBFullHash kHash2 = SBFullHashForString("www.evil.com/phishing.html"); |
| 720 | 720 |
| 721 EXPECT_TRUE(prefix_set->Exists(kHash1)); | 721 EXPECT_TRUE(prefix_set->Exists(kHash1)); |
| 722 EXPECT_TRUE(prefix_set->Exists(kHash2)); | 722 EXPECT_TRUE(prefix_set->Exists(kHash2)); |
| 723 EXPECT_FALSE(prefix_set->PrefixExists(kHash1.prefix)); | 723 EXPECT_FALSE(prefix_set->PrefixExists(kHash1.prefix)); |
| 724 EXPECT_FALSE(prefix_set->PrefixExists(kHash2.prefix)); | 724 EXPECT_FALSE(prefix_set->PrefixExists(kHash2.prefix)); |
| 725 } | 725 } |
| 726 #endif | 726 #endif |
| 727 | 727 |
| 728 } // namespace safe_browsing | 728 } // namespace safe_browsing |
| OLD | NEW |