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 "webkit/browser/fileapi/sandbox_database_test_helper.h" 6 7 #include <algorithm> 8 #include <functional> 9 #include <vector> 10 11 #include "base/file_util.h" 12 #include "base/files/file_enumerator.h" 13 #include "base/stl_util.h" 14 #include "testing/gtest/include/gtest/gtest.h" 15 #include "webkit/common/fileapi/file_system_util.h" 16 17 namespace fileapi { 18 19 void CorruptDatabase(const base::FilePath& db_path, 20 leveldb::FileType type, 21 ptrdiff_t offset, 22 size_t size) { 23 base::FileEnumerator file_enum(db_path, false /* not recursive */, 24 base::FileEnumerator::DIRECTORIES | base::FileEnumerator::FILES); 25 base::FilePath file_path; 26 base::FilePath picked_file_path; 27 uint64 picked_file_number = kuint64max; 28 29 while (!(file_path = file_enum.Next()).empty()) { 30 uint64 number = kuint64max; 31 leveldb::FileType file_type; 32 EXPECT_TRUE(leveldb::ParseFileName(FilePathToString(file_path.BaseName()), 33 &number, &file_type)); 34 if (file_type == type && 35 (picked_file_number == kuint64max || picked_file_number < number)) { 36 picked_file_path = file_path; 37 picked_file_number = number; 38 } 39 } 40 41 EXPECT_FALSE(picked_file_path.empty()); 42 EXPECT_NE(kuint64max, picked_file_number); 43 44 bool created = true; 45 base::PlatformFileError error = base::PLATFORM_FILE_ERROR_FAILED; 46 base::PlatformFile file = 47 CreatePlatformFile(picked_file_path, 48 base::PLATFORM_FILE_OPEN | 49 base::PLATFORM_FILE_READ | 50 base::PLATFORM_FILE_WRITE, 51 &created, &error); 52 EXPECT_EQ(base::PLATFORM_FILE_OK, error); 53 EXPECT_FALSE(created); 54 55 base::PlatformFileInfo file_info; 56 EXPECT_TRUE(base::GetPlatformFileInfo(file, &file_info)); 57 if (offset < 0) 58 offset += file_info.size; 59 EXPECT_GE(offset, 0); 60 EXPECT_LE(offset, file_info.size); 61 62 size = std::min(size, static_cast<size_t>(file_info.size - offset)); 63 64 std::vector<char> buf(size); 65 int read_size = base::ReadPlatformFile(file, offset, 66 vector_as_array(&buf), buf.size()); 67 EXPECT_LT(0, read_size); 68 EXPECT_GE(buf.size(), static_cast<size_t>(read_size)); 69 buf.resize(read_size); 70 71 std::transform(buf.begin(), buf.end(), buf.begin(), 72 std::logical_not<char>()); 73 74 int written_size = base::WritePlatformFile(file, offset, 75 vector_as_array(&buf), buf.size()); 76 EXPECT_GT(written_size, 0); 77 EXPECT_EQ(buf.size(), static_cast<size_t>(written_size)); 78 79 base::ClosePlatformFile(file); 80 } 81 82 void DeleteDatabaseFile(const base::FilePath& db_path, 83 leveldb::FileType type) { 84 base::FileEnumerator file_enum(db_path, false /* not recursive */, 85 base::FileEnumerator::DIRECTORIES | base::FileEnumerator::FILES); 86 base::FilePath file_path; 87 while (!(file_path = file_enum.Next()).empty()) { 88 uint64 number = kuint64max; 89 leveldb::FileType file_type; 90 EXPECT_TRUE(leveldb::ParseFileName(FilePathToString(file_path.BaseName()), 91 &number, &file_type)); 92 if (file_type == type) { 93 base::DeleteFile(file_path, false); 94 // We may have multiple files for the same type, so don't break here. 95 } 96 } 97 98 } 99 100 } // namespace fileapi 101
