Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(186)

Unified Diff: components/safe_browsing_db/v4_database.cc

Issue 2384893002: PVer4: Test checksum on startup outside the hotpath of DB load (Closed)
Patch Set: Verify that the checksum check happens async Created 4 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « components/safe_browsing_db/v4_database.h ('k') | components/safe_browsing_db/v4_database_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: components/safe_browsing_db/v4_database.cc
diff --git a/components/safe_browsing_db/v4_database.cc b/components/safe_browsing_db/v4_database.cc
index 6e14e5f099dfeee935a8f5265357fddf71719192..30a3b0f538d5756e5314329a88cdfb9a5f0191dd 100644
--- a/components/safe_browsing_db/v4_database.cc
+++ b/components/safe_browsing_db/v4_database.cc
@@ -69,7 +69,7 @@ void V4Database::CreateOnTaskRunner(
new V4Database(db_task_runner, std::move(store_map)));
// Database is done loading, pass it to the new_db_callback on the caller's
- // thread.
+ // thread. This would unblock resource loads.
callback_task_runner->PostTask(
FROM_HERE, base::Bind(new_db_callback, base::Passed(&v4_database)));
}
@@ -81,11 +81,11 @@ V4Database::V4Database(
store_map_(std::move(store_map)),
pending_store_updates_(0) {
DCHECK(db_task_runner->RunsTasksOnCurrentThread());
- // TODO(vakh): Implement skeleton
}
// static
void V4Database::Destroy(std::unique_ptr<V4Database> v4_database) {
+ DCHECK_CURRENTLY_ON(BrowserThread::IO);
V4Database* v4_database_raw = v4_database.release();
if (v4_database_raw) {
v4_database_raw->db_task_runner_->DeleteSoon(FROM_HERE, v4_database_raw);
@@ -152,17 +152,6 @@ void V4Database::UpdatedStoreReady(ListIdentifier identifier,
}
}
-bool V4Database::ResetDatabase() {
- DCHECK(db_task_runner_->RunsTasksOnCurrentThread());
- bool reset_success = true;
- for (const auto& store_map_iter : *store_map_) {
- if (!store_map_iter.second->Reset()) {
- reset_success = false;
- }
- }
- return reset_success;
-}
-
std::unique_ptr<StoreStateMap> V4Database::GetStoreStateMap() {
std::unique_ptr<StoreStateMap> store_state_map =
base::MakeUnique<StoreStateMap>();
@@ -189,6 +178,40 @@ void V4Database::GetStoresMatchingFullHash(
}
}
+void V4Database::ResetStores(
+ const std::vector<ListIdentifier>& stores_to_reset) {
+ DCHECK_CURRENTLY_ON(BrowserThread::IO);
+ for (const ListIdentifier& identifier : stores_to_reset) {
+ store_map_->at(identifier)->Reset();
+ }
+}
+
+void V4Database::VerifyChecksum(
+ DatabaseReadyForUpdatesCallback db_ready_for_updates_callback) {
+ DCHECK_CURRENTLY_ON(BrowserThread::IO);
+ // TODO(vakh): Consider using PostTaskAndReply instead.
+ const scoped_refptr<base::SingleThreadTaskRunner> callback_task_runner =
+ base::ThreadTaskRunnerHandle::Get();
+ db_task_runner_->PostTask(
+ FROM_HERE, base::Bind(&V4Database::VerifyChecksumOnTaskRunner,
+ base::Unretained(this), callback_task_runner,
+ db_ready_for_updates_callback));
+}
+
+void V4Database::VerifyChecksumOnTaskRunner(
+ const scoped_refptr<base::SingleThreadTaskRunner>& callback_task_runner,
+ DatabaseReadyForUpdatesCallback db_ready_for_updates_callback) {
+ std::vector<ListIdentifier> stores_to_reset;
+ for (const auto& store_map_iter : *store_map_) {
+ if (!store_map_iter.second->VerifyChecksum()) {
+ stores_to_reset.push_back(store_map_iter.first);
+ }
+ }
+
+ callback_task_runner->PostTask(
+ FROM_HERE, base::Bind(db_ready_for_updates_callback, stores_to_reset));
+}
+
ListInfo::ListInfo(const bool fetch_updates,
const std::string& filename,
const ListIdentifier& list_id,
« no previous file with comments | « components/safe_browsing_db/v4_database.h ('k') | components/safe_browsing_db/v4_database_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698