Chromium Code Reviews| Index: net/disk_cache/entry_unittest.cc |
| diff --git a/net/disk_cache/entry_unittest.cc b/net/disk_cache/entry_unittest.cc |
| index 3612fab63bfd632254448f65b20c6e242dbc77af..4260f81d403eb71943ae2e193bb3dc81d9b576ba 100644 |
| --- a/net/disk_cache/entry_unittest.cc |
| +++ b/net/disk_cache/entry_unittest.cc |
| @@ -9,6 +9,7 @@ |
| #include "base/files/file.h" |
| #include "base/files/file_util.h" |
| #include "base/macros.h" |
| +#include "base/run_loop.h" |
| #include "base/strings/string_number_conversions.h" |
| #include "base/strings/string_util.h" |
| #include "base/threading/platform_thread.h" |
| @@ -21,6 +22,7 @@ |
| #include "net/disk_cache/disk_cache_test_base.h" |
| #include "net/disk_cache/disk_cache_test_util.h" |
| #include "net/disk_cache/memory/mem_entry_impl.h" |
| +#include "net/disk_cache/simple/simple_backend_impl.h" |
| #include "net/disk_cache/simple/simple_entry_format.h" |
| #include "net/disk_cache/simple/simple_entry_impl.h" |
| #include "net/disk_cache/simple/simple_synchronous_entry.h" |
| @@ -2708,7 +2710,7 @@ TEST_F(DiskCacheEntryTest, SimpleCacheNoEOF) { |
| SetSimpleCacheMode(); |
| InitCache(); |
| - const char key[] = "the first key"; |
| + const std::string key("the first key"); |
| disk_cache::Entry* entry = NULL; |
| ASSERT_EQ(net::OK, CreateEntry(key, &entry)); |
| @@ -2728,9 +2730,8 @@ TEST_F(DiskCacheEntryTest, SimpleCacheNoEOF) { |
| int kTruncationBytes = -static_cast<int>(sizeof(disk_cache::SimpleFileEOF)); |
| const base::FilePath entry_path = cache_path_.AppendASCII( |
| disk_cache::simple_util::GetFilenameFromKeyAndFileIndex(key, 0)); |
| - const int64_t invalid_size = |
| - disk_cache::simple_util::GetFileSizeFromKeyAndDataSize(key, |
| - kTruncationBytes); |
| + const int64_t invalid_size = disk_cache::simple_util::GetFileSizeFromDataSize( |
| + key.size(), kTruncationBytes); |
| EXPECT_TRUE(TruncatePath(entry_path, invalid_size)); |
| EXPECT_EQ(net::ERR_FAILED, OpenEntry(key, &entry)); |
| DisableIntegrityCheck(); |
| @@ -3688,7 +3689,7 @@ TEST_F(DiskCacheEntryTest, SimpleCacheStream1SizeChanges) { |
| SetSimpleCacheMode(); |
| InitCache(); |
| disk_cache::Entry* entry = NULL; |
| - const char key[] = "the key"; |
| + const std::string key("the key"); |
| const int kSize = 100; |
| scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kSize)); |
| scoped_refptr<net::IOBuffer> buffer_read(new net::IOBuffer(kSize)); |
| @@ -3726,7 +3727,7 @@ TEST_F(DiskCacheEntryTest, SimpleCacheStream1SizeChanges) { |
| int sparse_data_size = 0; |
| disk_cache::SimpleEntryStat entry_stat( |
| base::Time::Now(), base::Time::Now(), data_size, sparse_data_size); |
| - int eof_offset = entry_stat.GetEOFOffsetInFile(key, 0); |
| + int eof_offset = entry_stat.GetEOFOffsetInFile(key.size(), 0); |
| disk_cache::SimpleFileEOF eof_record; |
| ASSERT_EQ(static_cast<int>(sizeof(eof_record)), |
| entry_file0.Read(eof_offset, reinterpret_cast<char*>(&eof_record), |
| @@ -4165,3 +4166,39 @@ TEST_F(DiskCacheEntryTest, SimpleCacheTruncateLargeSparseFile) { |
| entry->Close(); |
| } |
| + |
| +TEST_F(DiskCacheEntryTest, SimpleCacheReadWithoutKeySHA256) { |
| + // This test runs as APP_CACHE to make operations more synchronous. |
| + SetCacheType(net::APP_CACHE); |
| + SetSimpleCacheMode(); |
| + InitCache(); |
| + disk_cache::Entry* entry; |
| + std::string key("a key"); |
| + ASSERT_EQ(net::OK, CreateEntry(key, &entry)); |
| + entry->Close(); |
| + |
| + disk_cache::SimpleBackendImpl::FlushWorkerPoolForTesting(); |
| + base::RunLoop().RunUntilIdle(); |
| + |
| + EXPECT_TRUE( |
| + disk_cache::simple_util::RemoveKeySHA256FromEntry(key, cache_path_)); |
| + EXPECT_EQ(net::OK, OpenEntry(key, &entry)); |
|
Randy Smith (Not in Mondays)
2016/05/19 19:38:50
Confirm contents ok? (Probably need to write some
gavinp
2016/05/20 01:42:32
Done.
|
| +} |
| + |
| +TEST_F(DiskCacheEntryTest, SimpleCacheReadCorruptKeySHA256) { |
| + // This test runs as APP_CACHE to make operations more synchronous. |
| + SetCacheType(net::APP_CACHE); |
| + SetSimpleCacheMode(); |
| + InitCache(); |
| + disk_cache::Entry* entry; |
| + std::string key("a key"); |
| + ASSERT_EQ(net::OK, CreateEntry(key, &entry)); |
| + entry->Close(); |
| + |
| + disk_cache::SimpleBackendImpl::FlushWorkerPoolForTesting(); |
| + base::RunLoop().RunUntilIdle(); |
| + |
| + EXPECT_TRUE( |
| + disk_cache::simple_util::CorruptKeySHA256FromEntry(key, cache_path_)); |
| + EXPECT_NE(net::OK, OpenEntry(key, &entry)); |
| +} |