Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(124)

Side by Side Diff: chrome/browser/safe_browsing/safe_browsing_store_file_unittest.cc

Issue 507653003: Remove safe-browsing code to fix up injected prefixes. (Closed) Base URL: http://chromium.googlesource.com/chromium/src.git@master
Patch Set: Revert original CL more thoroughly. Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
OLDNEW
« no previous file with comments | « chrome/browser/safe_browsing/safe_browsing_store.cc ('k') | chrome/browser/safe_browsing/safe_browsing_store_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698