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 "chrome/browser/safe_browsing/safe_browsing_store_file.h" | 5 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/files/file_util.h" | 8 #include "base/files/file_util.h" |
9 #include "base/files/scoped_file.h" | 9 #include "base/files/scoped_file.h" |
10 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
(...skipping 800 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
811 EXPECT_EQ(kAddChunk1, chunks[0]); | 811 EXPECT_EQ(kAddChunk1, chunks[0]); |
812 | 812 |
813 store_->GetSubChunks(&chunks); | 813 store_->GetSubChunks(&chunks); |
814 ASSERT_EQ(1U, chunks.size()); | 814 ASSERT_EQ(1U, chunks.size()); |
815 EXPECT_EQ(kSubChunk1, chunks[0]); | 815 EXPECT_EQ(kSubChunk1, chunks[0]); |
816 | 816 |
817 EXPECT_TRUE(store_->CheckAddChunk(kAddChunk1)); | 817 EXPECT_TRUE(store_->CheckAddChunk(kAddChunk1)); |
818 EXPECT_TRUE(store_->CheckSubChunk(kSubChunk1)); | 818 EXPECT_TRUE(store_->CheckSubChunk(kSubChunk1)); |
819 | 819 |
820 // Sub chunk kAddChunk1 hash kHash2. | 820 // Sub chunk kAddChunk1 hash kHash2. |
| 821 // NOTE(shess): Having full hashes and prefixes in the same chunk is no longer |
| 822 // supported, though it was when this code was written. |
821 store_->SetSubChunk(kSubChunk2); | 823 store_->SetSubChunk(kSubChunk2); |
822 EXPECT_TRUE(store_->CheckSubChunk(kSubChunk1)); | 824 EXPECT_TRUE(store_->CheckSubChunk(kSubChunk1)); |
823 EXPECT_TRUE(store_->WriteSubPrefix(kSubChunk1, kAddChunk1, kHash2.prefix)); | 825 EXPECT_TRUE(store_->WriteSubPrefix(kSubChunk1, kAddChunk1, kHash2.prefix)); |
824 EXPECT_TRUE(store_->WriteSubHash(kSubChunk1, kAddChunk1, kHash2)); | 826 EXPECT_TRUE(store_->WriteSubHash(kSubChunk1, kAddChunk1, kHash2)); |
825 EXPECT_TRUE(store_->FinishChunk()); | 827 EXPECT_TRUE(store_->FinishChunk()); |
826 | 828 |
827 { | 829 { |
828 safe_browsing::PrefixSetBuilder builder; | 830 safe_browsing::PrefixSetBuilder builder; |
829 std::vector<SBAddFullHash> add_full_hashes_result; | 831 std::vector<SBAddFullHash> add_full_hashes_result; |
830 EXPECT_TRUE(store_->FinishUpdate(&builder, &add_full_hashes_result)); | 832 EXPECT_TRUE(store_->FinishUpdate(&builder, &add_full_hashes_result)); |
831 | 833 |
832 // The sub'ed prefix and hash are gone. | 834 // The sub'ed prefix and hash are gone. |
833 std::vector<SBPrefix> prefixes_result; | 835 std::vector<SBPrefix> prefixes_result; |
834 builder.GetPrefixSetNoHashes()->GetPrefixes(&prefixes_result); | 836 builder.GetPrefixSetNoHashes()->GetPrefixes(&prefixes_result); |
835 ASSERT_EQ(1U, prefixes_result.size()); | 837 ASSERT_EQ(1U, prefixes_result.size()); |
836 EXPECT_EQ(kHash1.prefix, prefixes_result[0]); | 838 EXPECT_EQ(kHash1.prefix, prefixes_result[0]); |
837 EXPECT_TRUE(add_full_hashes_result.empty()); | 839 EXPECT_TRUE(add_full_hashes_result.empty()); |
838 } | 840 } |
839 } | 841 } |
840 #endif | 842 #endif |
841 | 843 |
842 // Test that when the v8 golden file is updated, the add prefix injected from | |
843 // the full hash is removed. All platforms generating v8 files are | |
844 // little-endian, so there is no point to testing this transition if/when a | |
845 // big-endian port is added. | |
846 #if defined(ARCH_CPU_LITTLE_ENDIAN) | |
847 TEST_F(SafeBrowsingStoreFileTest, KnockoutPrefixVolunteers) { | |
848 store_.reset(); | |
849 | |
850 // Copy the golden file into temporary storage. The golden file contains: | |
851 // - Add chunk kAddChunk1 containing kHash1.prefix and kHash2. | |
852 // - Sub chunk kSubChunk1 containing kHash3. | |
853 const char kBasename[] = "FileStoreVersion8"; | |
854 base::FilePath golden_path; | |
855 ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &golden_path)); | |
856 golden_path = golden_path.AppendASCII("SafeBrowsing"); | |
857 golden_path = golden_path.AppendASCII(kBasename); | |
858 ASSERT_TRUE(base::CopyFile(golden_path, filename_)); | |
859 | |
860 // Reset the store to make sure it re-reads the file. | |
861 store_.reset(new SafeBrowsingStoreFile()); | |
862 store_->Init(filename_, | |
863 base::Bind(&SafeBrowsingStoreFileTest::OnCorruptionDetected, | |
864 base::Unretained(this))); | |
865 | |
866 // Check that the expected prefixes and hashes are in place. | |
867 { | |
868 SBAddPrefixes add_prefixes; | |
869 EXPECT_TRUE(store_->GetAddPrefixes(&add_prefixes)); | |
870 ASSERT_EQ(2U, add_prefixes.size()); | |
871 EXPECT_EQ(kAddChunk1, add_prefixes[0].chunk_id); | |
872 EXPECT_EQ(kHash1.prefix, add_prefixes[0].prefix); | |
873 EXPECT_EQ(kAddChunk1, add_prefixes[1].chunk_id); | |
874 EXPECT_EQ(kHash2.prefix, add_prefixes[1].prefix); | |
875 | |
876 std::vector<SBAddFullHash> add_hashes; | |
877 EXPECT_TRUE(store_->GetAddFullHashes(&add_hashes)); | |
878 ASSERT_EQ(1U, add_hashes.size()); | |
879 EXPECT_EQ(kAddChunk1, add_hashes[0].chunk_id); | |
880 EXPECT_TRUE(SBFullHashEqual(kHash2, add_hashes[0].full_hash)); | |
881 } | |
882 | |
883 // Update the store. | |
884 { | |
885 EXPECT_TRUE(store_->BeginUpdate()); | |
886 | |
887 safe_browsing::PrefixSetBuilder builder; | |
888 std::vector<SBAddFullHash> add_full_hashes_result; | |
889 ASSERT_TRUE(store_->FinishUpdate(&builder, &add_full_hashes_result)); | |
890 } | |
891 | |
892 // Reset the store to make sure it re-reads the file. | |
893 store_.reset(new SafeBrowsingStoreFile()); | |
894 store_->Init(filename_, | |
895 base::Bind(&SafeBrowsingStoreFileTest::OnCorruptionDetected, | |
896 base::Unretained(this))); | |
897 | |
898 // |kHash2.prefix| should have dropped. | |
899 { | |
900 SBAddPrefixes add_prefixes; | |
901 EXPECT_TRUE(store_->GetAddPrefixes(&add_prefixes)); | |
902 ASSERT_EQ(1U, add_prefixes.size()); | |
903 EXPECT_EQ(kAddChunk1, add_prefixes[0].chunk_id); | |
904 EXPECT_EQ(kHash1.prefix, add_prefixes[0].prefix); | |
905 | |
906 std::vector<SBAddFullHash> add_hashes; | |
907 EXPECT_TRUE(store_->GetAddFullHashes(&add_hashes)); | |
908 ASSERT_EQ(1U, add_hashes.size()); | |
909 EXPECT_EQ(kAddChunk1, add_hashes[0].chunk_id); | |
910 EXPECT_TRUE(SBFullHashEqual(kHash2, add_hashes[0].full_hash)); | |
911 } | |
912 } | |
913 #endif | |
914 | |
915 } // namespace safe_browsing | 844 } // namespace safe_browsing |
OLD | NEW |