src/webkit/browser/fileapi/sandbox_database_test_helper.cc

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