| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "extensions/browser/value_store/leveldb_value_store.h" | 5 #include "extensions/browser/value_store/leveldb_value_store.h" |
| 6 | 6 |
| 7 #include "base/files/file_util.h" | 7 #include "base/files/file_util.h" |
| 8 #include "base/json/json_reader.h" | 8 #include "base/json/json_reader.h" |
| 9 #include "base/json/json_writer.h" | 9 #include "base/json/json_writer.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| (...skipping 128 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 leveldb::ReadOptions options = leveldb::ReadOptions(); | 139 leveldb::ReadOptions options = leveldb::ReadOptions(); |
| 140 // All interaction with the db is done on the same thread, so snapshotting | 140 // All interaction with the db is done on the same thread, so snapshotting |
| 141 // isn't strictly necessary. This is just defensive. | 141 // isn't strictly necessary. This is just defensive. |
| 142 scoped_ptr<base::DictionaryValue> settings(new base::DictionaryValue()); | 142 scoped_ptr<base::DictionaryValue> settings(new base::DictionaryValue()); |
| 143 | 143 |
| 144 ScopedSnapshot snapshot(db_.get()); | 144 ScopedSnapshot snapshot(db_.get()); |
| 145 options.snapshot = snapshot.get(); | 145 options.snapshot = snapshot.get(); |
| 146 scoped_ptr<leveldb::Iterator> it(db_->NewIterator(options)); | 146 scoped_ptr<leveldb::Iterator> it(db_->NewIterator(options)); |
| 147 for (it->SeekToFirst(); it->Valid(); it->Next()) { | 147 for (it->SeekToFirst(); it->Valid(); it->Next()) { |
| 148 std::string key = it->key().ToString(); | 148 std::string key = it->key().ToString(); |
| 149 base::Value* value = json_reader.ReadToValue(it->value().ToString()); | 149 scoped_ptr<base::Value> value = |
| 150 json_reader.ReadToValue(it->value().ToString()); |
| 150 if (!value) { | 151 if (!value) { |
| 151 return MakeReadResult( | 152 return MakeReadResult( |
| 152 Error::Create(CORRUPTION, kInvalidJson, util::NewKey(key))); | 153 Error::Create(CORRUPTION, kInvalidJson, util::NewKey(key))); |
| 153 } | 154 } |
| 154 settings->SetWithoutPathExpansion(key, value); | 155 settings->SetWithoutPathExpansion(key, value.Pass()); |
| 155 } | 156 } |
| 156 | 157 |
| 157 if (it->status().IsNotFound()) { | 158 if (it->status().IsNotFound()) { |
| 158 NOTREACHED() << "IsNotFound() but iterating over all keys?!"; | 159 NOTREACHED() << "IsNotFound() but iterating over all keys?!"; |
| 159 return MakeReadResult(settings.Pass()); | 160 return MakeReadResult(settings.Pass()); |
| 160 } | 161 } |
| 161 | 162 |
| 162 if (!it->status().ok()) | 163 if (!it->status().ok()) |
| 163 return MakeReadResult(ToValueStoreError(it->status(), util::NoKey())); | 164 return MakeReadResult(ToValueStoreError(it->status(), util::NoKey())); |
| 164 | 165 |
| (...skipping 185 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 350 | 351 |
| 351 if (s.IsNotFound()) { | 352 if (s.IsNotFound()) { |
| 352 // Despite there being no value, it was still a success. Check this first | 353 // Despite there being no value, it was still a success. Check this first |
| 353 // because ok() is false on IsNotFound. | 354 // because ok() is false on IsNotFound. |
| 354 return util::NoError(); | 355 return util::NoError(); |
| 355 } | 356 } |
| 356 | 357 |
| 357 if (!s.ok()) | 358 if (!s.ok()) |
| 358 return ToValueStoreError(s, util::NewKey(key)); | 359 return ToValueStoreError(s, util::NewKey(key)); |
| 359 | 360 |
| 360 base::Value* value = base::JSONReader().ReadToValue(value_as_json); | 361 scoped_ptr<base::Value> value = base::JSONReader().ReadToValue(value_as_json); |
| 361 if (!value) | 362 if (!value) |
| 362 return Error::Create(CORRUPTION, kInvalidJson, util::NewKey(key)); | 363 return Error::Create(CORRUPTION, kInvalidJson, util::NewKey(key)); |
| 363 | 364 |
| 364 setting->reset(value); | 365 *setting = value.Pass(); |
| 365 return util::NoError(); | 366 return util::NoError(); |
| 366 } | 367 } |
| 367 | 368 |
| 368 scoped_ptr<ValueStore::Error> LeveldbValueStore::AddToBatch( | 369 scoped_ptr<ValueStore::Error> LeveldbValueStore::AddToBatch( |
| 369 ValueStore::WriteOptions options, | 370 ValueStore::WriteOptions options, |
| 370 const std::string& key, | 371 const std::string& key, |
| 371 const base::Value& value, | 372 const base::Value& value, |
| 372 leveldb::WriteBatch* batch, | 373 leveldb::WriteBatch* batch, |
| 373 ValueStoreChangeList* changes) { | 374 ValueStoreChangeList* changes) { |
| 374 bool write_new_value = true; | 375 bool write_new_value = true; |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 431 CHECK(!status.ok()); | 432 CHECK(!status.ok()); |
| 432 CHECK(!status.IsNotFound()); // not an error | 433 CHECK(!status.IsNotFound()); // not an error |
| 433 | 434 |
| 434 std::string message = status.ToString(); | 435 std::string message = status.ToString(); |
| 435 // The message may contain |db_path_|, which may be considered sensitive | 436 // The message may contain |db_path_|, which may be considered sensitive |
| 436 // data, and those strings are passed to the extension, so strip it out. | 437 // data, and those strings are passed to the extension, so strip it out. |
| 437 ReplaceSubstringsAfterOffset(&message, 0u, db_path_.AsUTF8Unsafe(), "..."); | 438 ReplaceSubstringsAfterOffset(&message, 0u, db_path_.AsUTF8Unsafe(), "..."); |
| 438 | 439 |
| 439 return Error::Create(CORRUPTION, message, key.Pass()); | 440 return Error::Create(CORRUPTION, message, key.Pass()); |
| 440 } | 441 } |
| OLD | NEW |