leveldb::CorruptionTest Class Reference

Collaboration diagram for leveldb::CorruptionTest:

Public Member Functions
	CorruptionTest ()
	~CorruptionTest ()
Status	TryReopen ()
void	Reopen ()
void	RepairDB ()
void	Build (int n)
void	Check (int min_expected, int max_expected)
void	Corrupt (FileType filetype, int offset, int bytes_to_corrupt)
int	Property (const std::string &name)
Slice	Key (int i, std::string *storage)
Slice	Value (int k, std::string *storage)

Public Attributes
test::ErrorEnv	env_
std::string	dbname_
Cache *	tiny_cache_
Options	options_
DB *	db_

Detailed Description

Definition at line 27 of file corruption_test.cc.

Constructor & Destructor Documentation

CorruptionTest()

leveldb::CorruptionTest::CorruptionTest ( )

inline

Definition at line 35 of file corruption_test.cc.

                   {
    tiny_cache_ = NewLRUCache(100);
    options_.env = &env_;
    options_.block_cache = tiny_cache_;
    dbname_ = test::TmpDir() + "/corruption_test";
    DestroyDB(dbname_, options_);

    db_ = NULL;
    options_.create_if_missing = true;
    Reopen();
    options_.create_if_missing = false;
  }

Here is the call graph for this function:

~CorruptionTest()

leveldb::CorruptionTest::~CorruptionTest ( )

inline

Definition at line 48 of file corruption_test.cc.

                    {
     delete db_;
     DestroyDB(dbname_, Options());
     delete tiny_cache_;
  }

Here is the call graph for this function:

Member Function Documentation

Build()

void leveldb::CorruptionTest::Build ( int  n )

inline

Definition at line 70 of file corruption_test.cc.

                    {
    std::string key_space, value_space;
    WriteBatch batch;
    for (int i = 0; i < n; i++) {
      //if ((i % 100) == 0) fprintf(stderr, "@ %d of %d\n", i, n);
      Slice key = Key(i, &key_space);
      batch.Clear();
      batch.Put(key, Value(i, &value_space));
      WriteOptions options;
      // Corrupt() doesn't work without this sync on windows; stat reports 0 for
      // the file size.
      if (i == n - 1) {
        options.sync = true;
      }
      ASSERT_OK(db_->Write(options, &batch));
    }
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Check()

void leveldb::CorruptionTest::Check ( int  min_expected, int  max_expected  )

inline

Definition at line 88 of file corruption_test.cc.

                                                 {
    int next_expected = 0;
    int missed = 0;
    int bad_keys = 0;
    int bad_values = 0;
    int correct = 0;
    std::string value_space;
    Iterator* iter = db_->NewIterator(ReadOptions());
    for (iter->SeekToFirst(); iter->Valid(); iter->Next()) {
      uint64_t key;
      Slice in(iter->key());
      if (in == "" || in == "~") {
        // Ignore boundary keys.
        continue;
      }
      if (!ConsumeDecimalNumber(&in, &key) ||
          !in.empty() ||
          key < next_expected) {
        bad_keys++;
        continue;
      }
      missed += (key - next_expected);
      next_expected = key + 1;
      if (iter->value() != Value(key, &value_space)) {
        bad_values++;
      } else {
        correct++;
      }
    }
    delete iter;

    fprintf(stderr,
            "expected=%d..%d; got=%d; bad_keys=%d; bad_values=%d; missed=%d\n",
            min_expected, max_expected, correct, bad_keys, bad_values, missed);
    ASSERT_LE(min_expected, correct);
    ASSERT_GE(max_expected, correct);
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Corrupt()

void leveldb::CorruptionTest::Corrupt ( FileType  filetype, int  offset, int  bytes_to_corrupt  )

inline

Definition at line 126 of file corruption_test.cc.

                                                                    {
    // Pick file to corrupt
    std::vector<std::string> filenames;
    ASSERT_OK(env_.GetChildren(dbname_, &filenames));
    uint64_t number;
    FileType type;
    std::string fname;
    int picked_number = -1;
    for (size_t i = 0; i < filenames.size(); i++) {
      if (ParseFileName(filenames[i], &number, &type) &&
          type == filetype &&
          int(number) > picked_number) {  // Pick latest file
        fname = dbname_ + "/" + filenames[i];
        picked_number = number;
      }
    }
    ASSERT_TRUE(!fname.empty()) << filetype;

    struct stat sbuf;
    if (stat(fname.c_str(), &sbuf) != 0) {
      const char* msg = strerror(errno);
      ASSERT_TRUE(false) << fname << ": " << msg;
    }

    if (offset < 0) {
      // Relative to end of file; make it absolute
      if (-offset > sbuf.st_size) {
        offset = 0;
      } else {
        offset = sbuf.st_size + offset;
      }
    }
    if (offset > sbuf.st_size) {
      offset = sbuf.st_size;
    }
    if (offset + bytes_to_corrupt > sbuf.st_size) {
      bytes_to_corrupt = sbuf.st_size - offset;
    }

    // Do it
    std::string contents;
    Status s = ReadFileToString(Env::Default(), fname, &contents);
    ASSERT_TRUE(s.ok()) << s.ToString();
    for (int i = 0; i < bytes_to_corrupt; i++) {
      contents[i + offset] ^= 0x80;
    }
    s = WriteStringToFile(Env::Default(), contents, fname);
    ASSERT_TRUE(s.ok()) << s.ToString();
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Key()

Slice leveldb::CorruptionTest::Key ( int  i, std::string *  storage  )

inline

Definition at line 188 of file corruption_test.cc.

                                       {
    char buf[100];
    snprintf(buf, sizeof(buf), "%016d", i);
    storage->assign(buf, strlen(buf));
    return Slice(*storage);
  }

Here is the caller graph for this function:

Property()

int leveldb::CorruptionTest::Property ( const std::string &  name )

inline

Definition at line 176 of file corruption_test.cc.

                                      {
    std::string property;
    int result;
    if (db_->GetProperty(name, &property) &&
        sscanf(property.c_str(), "%d", &result) == 1) {
      return result;
    } else {
      return -1;
    }
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Reopen()

void leveldb::CorruptionTest::Reopen ( )

inline

Definition at line 60 of file corruption_test.cc.

                {
    ASSERT_OK(TryReopen());
  }

Here is the call graph for this function:

Here is the caller graph for this function:

RepairDB()

void leveldb::CorruptionTest::RepairDB ( )

inline

Definition at line 64 of file corruption_test.cc.

                  {
    delete db_;
    db_ = NULL;
    ASSERT_OK(::leveldb::RepairDB(dbname_, options_));
  }

Here is the call graph for this function:

Here is the caller graph for this function:

TryReopen()

Status leveldb::CorruptionTest::TryReopen ( )

inline

Definition at line 54 of file corruption_test.cc.

                     {
    delete db_;
    db_ = NULL;
    return DB::Open(options_, dbname_, &db_);
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Value()

Slice leveldb::CorruptionTest::Value ( int  k, std::string *  storage  )

inline

Definition at line 196 of file corruption_test.cc.

                                         {
    Random r(k);
    return test::RandomString(&r, kValueSize, storage);
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

db_

DB* leveldb::CorruptionTest::db_

Definition at line 33 of file corruption_test.cc.

dbname_

std::string leveldb::CorruptionTest::dbname_

Definition at line 30 of file corruption_test.cc.

env_

test::ErrorEnv leveldb::CorruptionTest::env_

Definition at line 29 of file corruption_test.cc.

options_

Options leveldb::CorruptionTest::options_

Definition at line 32 of file corruption_test.cc.

tiny_cache_

Cache* leveldb::CorruptionTest::tiny_cache_

Definition at line 31 of file corruption_test.cc.

---

The documentation for this class was generated from the following file:

• db/corruption_test.cc