Index: chrome/browser/history/download_database.cc |
diff --git a/chrome/browser/history/download_database.cc b/chrome/browser/history/download_database.cc |
deleted file mode 100644 |
index 9461e4523f8689253c107e0482f42d289305f7d7..0000000000000000000000000000000000000000 |
--- a/chrome/browser/history/download_database.cc |
+++ /dev/null |
@@ -1,671 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "chrome/browser/history/download_database.h" |
- |
-#include <limits> |
-#include <string> |
-#include <vector> |
- |
-#include "base/debug/alias.h" |
-#include "base/files/file_path.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/metrics/histogram.h" |
-#include "base/stl_util.h" |
-#include "base/strings/stringprintf.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/time/time.h" |
-#include "build/build_config.h" |
-#include "chrome/browser/history/download_row.h" |
-#include "components/history/core/browser/history_types.h" |
-#include "content/public/browser/download_interrupt_reasons.h" |
-#include "content/public/browser/download_item.h" |
-#include "sql/statement.h" |
- |
-using content::DownloadItem; |
- |
-namespace history { |
- |
-namespace { |
- |
-// Reason for dropping a particular record. |
-enum DroppedReason { |
- DROPPED_REASON_BAD_STATE = 0, |
- DROPPED_REASON_BAD_DANGER_TYPE = 1, |
- DROPPED_REASON_BAD_ID = 2, |
- DROPPED_REASON_DUPLICATE_ID = 3, |
- DROPPED_REASON_MAX |
-}; |
- |
-#if defined(OS_POSIX) |
- |
-// Binds/reads the given file path to the given column of the given statement. |
-void BindFilePath(sql::Statement& statement, const base::FilePath& path, |
- int col) { |
- statement.BindString(col, path.value()); |
-} |
-base::FilePath ColumnFilePath(sql::Statement& statement, int col) { |
- return base::FilePath(statement.ColumnString(col)); |
-} |
- |
-#else |
- |
-// See above. |
-void BindFilePath(sql::Statement& statement, const base::FilePath& path, |
- int col) { |
- statement.BindString16(col, path.value()); |
-} |
-base::FilePath ColumnFilePath(sql::Statement& statement, int col) { |
- return base::FilePath(statement.ColumnString16(col)); |
-} |
- |
-#endif |
- |
-} // namespace |
- |
-// These constants and the transformation functions below are used to allow |
-// DownloadItem::DownloadState and DownloadDangerType to change without |
-// breaking the database schema. |
-// They guarantee that the values of the |state| field in the database are one |
-// of the values returned by StateToInt, and that the values of the |state| |
-// field of the DownloadRows returned by QueryDownloads() are one of the values |
-// returned by IntToState(). |
-const int DownloadDatabase::kStateInvalid = -1; |
-const int DownloadDatabase::kStateInProgress = 0; |
-const int DownloadDatabase::kStateComplete = 1; |
-const int DownloadDatabase::kStateCancelled = 2; |
-const int DownloadDatabase::kStateBug140687 = 3; |
-const int DownloadDatabase::kStateInterrupted = 4; |
- |
-const int DownloadDatabase::kDangerTypeInvalid = -1; |
-const int DownloadDatabase::kDangerTypeNotDangerous = 0; |
-const int DownloadDatabase::kDangerTypeDangerousFile = 1; |
-const int DownloadDatabase::kDangerTypeDangerousUrl = 2; |
-const int DownloadDatabase::kDangerTypeDangerousContent = 3; |
-const int DownloadDatabase::kDangerTypeMaybeDangerousContent = 4; |
-const int DownloadDatabase::kDangerTypeUncommonContent = 5; |
-const int DownloadDatabase::kDangerTypeUserValidated = 6; |
-const int DownloadDatabase::kDangerTypeDangerousHost = 7; |
-const int DownloadDatabase::kDangerTypePotentiallyUnwanted = 8; |
- |
-int DownloadDatabase::StateToInt(DownloadItem::DownloadState state) { |
- switch (state) { |
- case DownloadItem::IN_PROGRESS: return DownloadDatabase::kStateInProgress; |
- case DownloadItem::COMPLETE: return DownloadDatabase::kStateComplete; |
- case DownloadItem::CANCELLED: return DownloadDatabase::kStateCancelled; |
- case DownloadItem::INTERRUPTED: return DownloadDatabase::kStateInterrupted; |
- case DownloadItem::MAX_DOWNLOAD_STATE: |
- NOTREACHED(); |
- return DownloadDatabase::kStateInvalid; |
- } |
- NOTREACHED(); |
- return DownloadDatabase::kStateInvalid; |
-} |
- |
-DownloadItem::DownloadState DownloadDatabase::IntToState(int state) { |
- switch (state) { |
- case DownloadDatabase::kStateInProgress: return DownloadItem::IN_PROGRESS; |
- case DownloadDatabase::kStateComplete: return DownloadItem::COMPLETE; |
- case DownloadDatabase::kStateCancelled: return DownloadItem::CANCELLED; |
- // We should not need kStateBug140687 here because MigrateDownloadsState() |
- // is called in HistoryDatabase::Init(). |
- case DownloadDatabase::kStateInterrupted: return DownloadItem::INTERRUPTED; |
- default: return DownloadItem::MAX_DOWNLOAD_STATE; |
- } |
-} |
- |
-int DownloadDatabase::DangerTypeToInt(content::DownloadDangerType danger_type) { |
- switch (danger_type) { |
- case content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS: |
- return DownloadDatabase::kDangerTypeNotDangerous; |
- case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE: |
- return DownloadDatabase::kDangerTypeDangerousFile; |
- case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL: |
- return DownloadDatabase::kDangerTypeDangerousUrl; |
- case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT: |
- return DownloadDatabase::kDangerTypeDangerousContent; |
- case content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT: |
- return DownloadDatabase::kDangerTypeMaybeDangerousContent; |
- case content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT: |
- return DownloadDatabase::kDangerTypeUncommonContent; |
- case content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED: |
- return DownloadDatabase::kDangerTypeUserValidated; |
- case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_HOST: |
- return DownloadDatabase::kDangerTypeDangerousHost; |
- case content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED: |
- return DownloadDatabase::kDangerTypePotentiallyUnwanted; |
- case content::DOWNLOAD_DANGER_TYPE_MAX: |
- NOTREACHED(); |
- return DownloadDatabase::kDangerTypeInvalid; |
- } |
- NOTREACHED(); |
- return DownloadDatabase::kDangerTypeInvalid; |
-} |
- |
-content::DownloadDangerType DownloadDatabase::IntToDangerType(int danger_type) { |
- switch (danger_type) { |
- case DownloadDatabase::kDangerTypeNotDangerous: |
- return content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS; |
- case DownloadDatabase::kDangerTypeDangerousFile: |
- return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE; |
- case DownloadDatabase::kDangerTypeDangerousUrl: |
- return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL; |
- case DownloadDatabase::kDangerTypeDangerousContent: |
- return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT; |
- case DownloadDatabase::kDangerTypeMaybeDangerousContent: |
- return content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT; |
- case DownloadDatabase::kDangerTypeUncommonContent: |
- return content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT; |
- case DownloadDatabase::kDangerTypeUserValidated: |
- return content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED; |
- case DownloadDatabase::kDangerTypeDangerousHost: |
- return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_HOST; |
- case DownloadDatabase::kDangerTypePotentiallyUnwanted: |
- return content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED; |
- default: |
- return content::DOWNLOAD_DANGER_TYPE_MAX; |
- } |
-} |
- |
-DownloadDatabase::DownloadDatabase() |
- : owning_thread_set_(false), |
- owning_thread_(0), |
- in_progress_entry_cleanup_completed_(false) { |
-} |
- |
-DownloadDatabase::~DownloadDatabase() { |
-} |
- |
-bool DownloadDatabase::EnsureColumnExists( |
- const std::string& name, const std::string& type) { |
- std::string add_col = "ALTER TABLE downloads ADD COLUMN " + name + " " + type; |
- return GetDB().DoesColumnExist("downloads", name.c_str()) || |
- GetDB().Execute(add_col.c_str()); |
-} |
- |
-bool DownloadDatabase::MigrateMimeType() { |
- return EnsureColumnExists("mime_type", "VARCHAR(255) NOT NULL" |
- " DEFAULT \"\"") && |
- EnsureColumnExists("original_mime_type", "VARCHAR(255) NOT NULL" |
- " DEFAULT \"\""); |
-} |
- |
-bool DownloadDatabase::MigrateDownloadsState() { |
- sql::Statement statement(GetDB().GetUniqueStatement( |
- "UPDATE downloads SET state=? WHERE state=?")); |
- statement.BindInt(0, kStateInterrupted); |
- statement.BindInt(1, kStateBug140687); |
- return statement.Run(); |
-} |
- |
-bool DownloadDatabase::MigrateDownloadsReasonPathsAndDangerType() { |
- // We need to rename the table and copy back from it because SQLite |
- // provides no way to rename or delete a column. |
- if (!GetDB().Execute("ALTER TABLE downloads RENAME TO downloads_tmp")) |
- return false; |
- |
- const char kReasonPathDangerSchema[] = |
- "CREATE TABLE downloads (" |
- "id INTEGER PRIMARY KEY," |
- "current_path LONGVARCHAR NOT NULL," |
- "target_path LONGVARCHAR NOT NULL," |
- "start_time INTEGER NOT NULL," |
- "received_bytes INTEGER NOT NULL," |
- "total_bytes INTEGER NOT NULL," |
- "state INTEGER NOT NULL," |
- "danger_type INTEGER NOT NULL," |
- "interrupt_reason INTEGER NOT NULL," |
- "end_time INTEGER NOT NULL," |
- "opened INTEGER NOT NULL)"; |
- |
- static const char kReasonPathDangerUrlChainSchema[] = |
- "CREATE TABLE downloads_url_chains (" |
- "id INTEGER NOT NULL," // downloads.id. |
- "chain_index INTEGER NOT NULL," // Index of url in chain |
- // 0 is initial target, |
- // MAX is target after redirects. |
- "url LONGVARCHAR NOT NULL, " // URL. |
- "PRIMARY KEY (id, chain_index) )"; |
- |
- |
- // Recreate main table. |
- if (!GetDB().Execute(kReasonPathDangerSchema)) |
- return false; |
- |
- // Populate it. As we do so, we transform the time values from time_t |
- // (seconds since 1/1/1970 UTC), to our internal measure (microseconds |
- // since the Windows Epoch). Note that this is dependent on the |
- // internal representation of base::Time and needs to change if that changes. |
- sql::Statement statement_populate(GetDB().GetUniqueStatement( |
- "INSERT INTO downloads " |
- "( id, current_path, target_path, start_time, received_bytes, " |
- " total_bytes, state, danger_type, interrupt_reason, end_time, opened ) " |
- "SELECT id, full_path, full_path, " |
- " CASE start_time WHEN 0 THEN 0 ELSE " |
- " (start_time + 11644473600) * 1000000 END, " |
- " received_bytes, total_bytes, " |
- " state, ?, ?, " |
- " CASE end_time WHEN 0 THEN 0 ELSE " |
- " (end_time + 11644473600) * 1000000 END, " |
- " opened " |
- "FROM downloads_tmp")); |
- statement_populate.BindInt(0, content::DOWNLOAD_INTERRUPT_REASON_NONE); |
- statement_populate.BindInt(1, kDangerTypeNotDangerous); |
- if (!statement_populate.Run()) |
- return false; |
- |
- // Create new chain table and populate it. |
- if (!GetDB().Execute(kReasonPathDangerUrlChainSchema)) |
- return false; |
- |
- if (!GetDB().Execute("INSERT INTO downloads_url_chains " |
- " ( id, chain_index, url) " |
- " SELECT id, 0, url from downloads_tmp")) |
- return false; |
- |
- // Get rid of temporary table. |
- if (!GetDB().Execute("DROP TABLE downloads_tmp")) |
- return false; |
- |
- return true; |
-} |
- |
-bool DownloadDatabase::MigrateReferrer() { |
- return EnsureColumnExists("referrer", "VARCHAR NOT NULL DEFAULT \"\""); |
-} |
- |
-bool DownloadDatabase::MigrateDownloadedByExtension() { |
- return EnsureColumnExists("by_ext_id", "VARCHAR NOT NULL DEFAULT \"\"") && |
- EnsureColumnExists("by_ext_name", "VARCHAR NOT NULL DEFAULT \"\""); |
-} |
- |
-bool DownloadDatabase::MigrateDownloadValidators() { |
- return EnsureColumnExists("etag", "VARCHAR NOT NULL DEFAULT \"\"") && |
- EnsureColumnExists("last_modified", "VARCHAR NOT NULL DEFAULT \"\""); |
-} |
- |
-bool DownloadDatabase::InitDownloadTable() { |
- const char kSchema[] = |
- "CREATE TABLE downloads (" |
- "id INTEGER PRIMARY KEY," // Primary key. |
- "current_path LONGVARCHAR NOT NULL," // Current disk location |
- "target_path LONGVARCHAR NOT NULL," // Final disk location |
- "start_time INTEGER NOT NULL," // When the download was started. |
- "received_bytes INTEGER NOT NULL," // Total size downloaded. |
- "total_bytes INTEGER NOT NULL," // Total size of the download. |
- "state INTEGER NOT NULL," // 1=complete, 4=interrupted |
- "danger_type INTEGER NOT NULL," // Danger type, validated. |
- "interrupt_reason INTEGER NOT NULL," // content::DownloadInterruptReason |
- "end_time INTEGER NOT NULL," // When the download completed. |
- "opened INTEGER NOT NULL," // 1 if it has ever been opened |
- // else 0 |
- "referrer VARCHAR NOT NULL," // HTTP Referrer |
- "by_ext_id VARCHAR NOT NULL," // ID of extension that started the |
- // download |
- "by_ext_name VARCHAR NOT NULL," // name of extension |
- "etag VARCHAR NOT NULL," // ETag |
- "last_modified VARCHAR NOT NULL," // Last-Modified header |
- "mime_type VARCHAR(255) NOT NULL," // MIME type. |
- "original_mime_type VARCHAR(255) NOT NULL)"; // Original MIME type. |
- |
- const char kUrlChainSchema[] = |
- "CREATE TABLE downloads_url_chains (" |
- "id INTEGER NOT NULL," // downloads.id. |
- "chain_index INTEGER NOT NULL," // Index of url in chain |
- // 0 is initial target, |
- // MAX is target after redirects. |
- "url LONGVARCHAR NOT NULL, " // URL. |
- "PRIMARY KEY (id, chain_index) )"; |
- |
- if (GetDB().DoesTableExist("downloads")) { |
- return EnsureColumnExists("end_time", "INTEGER NOT NULL DEFAULT 0") && |
- EnsureColumnExists("opened", "INTEGER NOT NULL DEFAULT 0"); |
- } else { |
- // If the "downloads" table doesn't exist, the downloads_url_chain |
- // table better not. |
- return (!GetDB().DoesTableExist("downloads_url_chain") && |
- GetDB().Execute(kSchema) && GetDB().Execute(kUrlChainSchema)); |
- } |
-} |
- |
-uint32 DownloadDatabase::GetNextDownloadId() { |
- sql::Statement select_max_id(GetDB().GetUniqueStatement( |
- "SELECT max(id) FROM downloads")); |
- bool result = select_max_id.Step(); |
- DCHECK(result); |
- // If there are zero records in the downloads table, then max(id) will return |
- // 0 = kInvalidId, so GetNextDownloadId() will set *id = kInvalidId + 1. |
- // If there is at least one record but all of the |id|s are <= kInvalidId, |
- // then max(id) will return <= kInvalidId, so GetNextDownloadId should return |
- // kInvalidId + 1. Note that any records with |id <= kInvalidId| will be |
- // dropped in QueryDownloads() |
- // SQLITE doesn't have unsigned integers. |
- return 1 + static_cast<uint32>(std::max( |
- static_cast<int64>(content::DownloadItem::kInvalidId), |
- select_max_id.ColumnInt64(0))); |
-} |
- |
-bool DownloadDatabase::DropDownloadTable() { |
- return GetDB().Execute("DROP TABLE downloads"); |
-} |
- |
-void DownloadDatabase::QueryDownloads( |
- std::vector<DownloadRow>* results) { |
- EnsureInProgressEntriesCleanedUp(); |
- |
- results->clear(); |
- std::set<uint32> ids; |
- |
- std::map<uint32, DownloadRow*> info_map; |
- |
- sql::Statement statement_main(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "SELECT id, current_path, target_path, " |
- "mime_type, original_mime_type, " |
- "start_time, received_bytes, " |
- "total_bytes, state, danger_type, interrupt_reason, end_time, opened, " |
- "referrer, by_ext_id, by_ext_name, etag, last_modified " |
- "FROM downloads ORDER BY start_time")); |
- |
- while (statement_main.Step()) { |
- scoped_ptr<DownloadRow> info(new DownloadRow()); |
- int column = 0; |
- |
- // SQLITE does not have unsigned integers, so explicitly handle negative |
- // |id|s instead of casting them to very large uint32s, which would break |
- // the max(id) logic in GetNextDownloadId(). |
- int64 signed_id = statement_main.ColumnInt64(column++); |
- info->id = static_cast<uint32>(signed_id); |
- info->current_path = ColumnFilePath(statement_main, column++); |
- info->target_path = ColumnFilePath(statement_main, column++); |
- info->mime_type = statement_main.ColumnString(column++); |
- info->original_mime_type = statement_main.ColumnString(column++); |
- info->start_time = base::Time::FromInternalValue( |
- statement_main.ColumnInt64(column++)); |
- info->received_bytes = statement_main.ColumnInt64(column++); |
- info->total_bytes = statement_main.ColumnInt64(column++); |
- int state = statement_main.ColumnInt(column++); |
- info->state = IntToState(state); |
- if (info->state == DownloadItem::MAX_DOWNLOAD_STATE) |
- UMA_HISTOGRAM_COUNTS("Download.DatabaseInvalidState", state); |
droger
2015/01/06 10:22:18
Where did this code go? I no longer see the histog
sdefresne
2015/01/06 10:51:17
The file has been moved to components/history/core
droger
2015/01/06 11:52:18
I see, but 8 calls to UMA_HISTOGRAM_ are deleted h
|
- info->danger_type = IntToDangerType(statement_main.ColumnInt(column++)); |
- info->interrupt_reason = static_cast<content::DownloadInterruptReason>( |
- statement_main.ColumnInt(column++)); |
- info->end_time = base::Time::FromInternalValue( |
- statement_main.ColumnInt64(column++)); |
- info->opened = statement_main.ColumnInt(column++) != 0; |
- info->referrer_url = GURL(statement_main.ColumnString(column++)); |
- info->by_ext_id = statement_main.ColumnString(column++); |
- info->by_ext_name = statement_main.ColumnString(column++); |
- info->etag = statement_main.ColumnString(column++); |
- info->last_modified = statement_main.ColumnString(column++); |
- |
- // If the record is corrupted, note that and drop it. |
- // http://crbug.com/251269 |
- DroppedReason dropped_reason = DROPPED_REASON_MAX; |
- if (signed_id <= static_cast<int64>(content::DownloadItem::kInvalidId)) { |
- // SQLITE doesn't have unsigned integers. |
- dropped_reason = DROPPED_REASON_BAD_ID; |
- } else if (!ids.insert(info->id).second) { |
- dropped_reason = DROPPED_REASON_DUPLICATE_ID; |
- NOTREACHED() << info->id; |
- } else if (info->state == DownloadItem::MAX_DOWNLOAD_STATE) { |
- dropped_reason = DROPPED_REASON_BAD_STATE; |
- } else if (info->danger_type == content::DOWNLOAD_DANGER_TYPE_MAX) { |
- dropped_reason = DROPPED_REASON_BAD_DANGER_TYPE; |
- } |
- if (dropped_reason != DROPPED_REASON_MAX) { |
- UMA_HISTOGRAM_ENUMERATION("Download.DatabaseRecordDropped", |
- dropped_reason, |
- DROPPED_REASON_MAX + 1); |
- } else { |
- DCHECK(!ContainsKey(info_map, info->id)); |
- uint32 id = info->id; |
- info_map[id] = info.release(); |
- } |
- } |
- |
- sql::Statement statement_chain(GetDB().GetCachedStatement( |
- SQL_FROM_HERE, |
- "SELECT id, chain_index, url FROM downloads_url_chains " |
- "ORDER BY id, chain_index")); |
- |
- while (statement_chain.Step()) { |
- int column = 0; |
- // See the comment above about SQLITE lacking unsigned integers. |
- int64 signed_id = statement_chain.ColumnInt64(column++); |
- int chain_index = statement_chain.ColumnInt(column++); |
- |
- if (signed_id <= static_cast<int64>(content::DownloadItem::kInvalidId)) |
- continue; |
- uint32 id = static_cast<uint32>(signed_id); |
- |
- // Note that these DCHECKs may trip as a result of corrupted databases. |
- // We have them because in debug builds the chances are higher there's |
- // an actual bug than that the database is corrupt, but we handle the |
- // DB corruption case in production code. |
- |
- // Confirm the id has already been seen--if it hasn't, discard the |
- // record. |
- DCHECK(ContainsKey(info_map, id)); |
- if (!ContainsKey(info_map, id)) |
- continue; |
- |
- // Confirm all previous URLs in the chain have already been seen; |
- // if not, fill in with null or discard record. |
- int current_chain_size = info_map[id]->url_chain.size(); |
- std::vector<GURL>* url_chain(&info_map[id]->url_chain); |
- DCHECK_EQ(chain_index, current_chain_size); |
- while (current_chain_size < chain_index) { |
- url_chain->push_back(GURL()); |
- current_chain_size++; |
- } |
- if (current_chain_size > chain_index) |
- continue; |
- |
- // Save the record. |
- url_chain->push_back(GURL(statement_chain.ColumnString(2))); |
- } |
- |
- for (std::map<uint32, DownloadRow*>::iterator |
- it = info_map.begin(); it != info_map.end(); ++it) { |
- DownloadRow* row = it->second; |
- bool empty_url_chain = row->url_chain.empty(); |
- UMA_HISTOGRAM_BOOLEAN("Download.DatabaseEmptyUrlChain", empty_url_chain); |
- if (empty_url_chain) { |
- RemoveDownload(row->id); |
- } else { |
- // Copy the contents of the stored info. |
- results->push_back(*row); |
- } |
- delete row; |
- it->second = NULL; |
- } |
-} |
- |
-bool DownloadDatabase::UpdateDownload(const DownloadRow& data) { |
- EnsureInProgressEntriesCleanedUp(); |
- |
- DCHECK_NE(content::DownloadItem::kInvalidId, data.id); |
- int state = StateToInt(data.state); |
- if (state == kStateInvalid) { |
- NOTREACHED(); |
- return false; |
- } |
- int danger_type = DangerTypeToInt(data.danger_type); |
- if (danger_type == kDangerTypeInvalid) { |
- NOTREACHED(); |
- return false; |
- } |
- |
- sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "UPDATE downloads " |
- "SET current_path=?, target_path=?, " |
- "mime_type=?, original_mime_type=?, " |
- "received_bytes=?, state=?, " |
- "danger_type=?, interrupt_reason=?, end_time=?, total_bytes=?, " |
- "opened=?, by_ext_id=?, by_ext_name=?, etag=?, last_modified=? " |
- "WHERE id=?")); |
- int column = 0; |
- BindFilePath(statement, data.current_path, column++); |
- BindFilePath(statement, data.target_path, column++); |
- statement.BindString(column++, data.mime_type); |
- statement.BindString(column++, data.original_mime_type); |
- statement.BindInt64(column++, data.received_bytes); |
- statement.BindInt(column++, state); |
- statement.BindInt(column++, danger_type); |
- statement.BindInt(column++, static_cast<int>(data.interrupt_reason)); |
- statement.BindInt64(column++, data.end_time.ToInternalValue()); |
- statement.BindInt64(column++, data.total_bytes); |
- statement.BindInt(column++, (data.opened ? 1 : 0)); |
- statement.BindString(column++, data.by_ext_id); |
- statement.BindString(column++, data.by_ext_name); |
- statement.BindString(column++, data.etag); |
- statement.BindString(column++, data.last_modified); |
- statement.BindInt(column++, data.id); |
- |
- return statement.Run(); |
-} |
- |
-void DownloadDatabase::EnsureInProgressEntriesCleanedUp() { |
- if (in_progress_entry_cleanup_completed_) |
- return; |
- |
- sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "UPDATE downloads SET state=?, interrupt_reason=? WHERE state=?")); |
- statement.BindInt(0, kStateInterrupted); |
- statement.BindInt(1, content::DOWNLOAD_INTERRUPT_REASON_CRASH); |
- statement.BindInt(2, kStateInProgress); |
- |
- statement.Run(); |
- in_progress_entry_cleanup_completed_ = true; |
-} |
- |
-bool DownloadDatabase::CreateDownload(const DownloadRow& info) { |
- DCHECK_NE(content::DownloadItem::kInvalidId, info.id); |
- EnsureInProgressEntriesCleanedUp(); |
- |
- if (info.url_chain.empty()) |
- return false; |
- |
- int state = StateToInt(info.state); |
- if (state == kStateInvalid) |
- return false; |
- |
- int danger_type = DangerTypeToInt(info.danger_type); |
- if (danger_type == kDangerTypeInvalid) |
- return false; |
- |
- { |
- sql::Statement statement_insert(GetDB().GetCachedStatement( |
- SQL_FROM_HERE, |
- "INSERT INTO downloads " |
- "(id, current_path, target_path, " |
- " mime_type, original_mime_type, " |
- " start_time, " |
- " received_bytes, total_bytes, state, danger_type, interrupt_reason, " |
- " end_time, opened, referrer, by_ext_id, by_ext_name, etag, " |
- " last_modified) " |
- "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")); |
- |
- int column = 0; |
- statement_insert.BindInt(column++, info.id); |
- BindFilePath(statement_insert, info.current_path, column++); |
- BindFilePath(statement_insert, info.target_path, column++); |
- statement_insert.BindString(column++, info.mime_type); |
- statement_insert.BindString(column++, info.original_mime_type); |
- statement_insert.BindInt64(column++, info.start_time.ToInternalValue()); |
- statement_insert.BindInt64(column++, info.received_bytes); |
- statement_insert.BindInt64(column++, info.total_bytes); |
- statement_insert.BindInt(column++, state); |
- statement_insert.BindInt(column++, danger_type); |
- statement_insert.BindInt(column++, info.interrupt_reason); |
- statement_insert.BindInt64(column++, info.end_time.ToInternalValue()); |
- statement_insert.BindInt(column++, info.opened ? 1 : 0); |
- statement_insert.BindString(column++, info.referrer_url.spec()); |
- statement_insert.BindString(column++, info.by_ext_id); |
- statement_insert.BindString(column++, info.by_ext_name); |
- statement_insert.BindString(column++, info.etag); |
- statement_insert.BindString(column++, info.last_modified); |
- if (!statement_insert.Run()) { |
- // GetErrorCode() returns a bitmask where the lower byte is a more general |
- // code and the upper byte is a more specific code. In order to save |
- // memory, take the general code, of which there are fewer than 50. See |
- // also sql/connection.cc |
- // http://www.sqlite.org/c3ref/c_abort_rollback.html |
- UMA_HISTOGRAM_ENUMERATION("Download.DatabaseMainInsertError", |
- GetDB().GetErrorCode() & 0xff, 50); |
- return false; |
- } |
- } |
- |
- { |
- sql::Statement count_urls(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "SELECT count(*) FROM downloads_url_chains WHERE id=?")); |
- count_urls.BindInt(0, info.id); |
- if (count_urls.Step()) { |
- bool corrupt_urls = count_urls.ColumnInt(0) > 0; |
- UMA_HISTOGRAM_BOOLEAN("Download.DatabaseCorruptUrls", corrupt_urls); |
- if (corrupt_urls) { |
- // There should not be any URLs in downloads_url_chains for this |
- // info.id. If there are, we don't want them to interfere with |
- // inserting the correct URLs, so just remove them. |
- RemoveDownloadURLs(info.id); |
- } |
- } |
- } |
- |
- sql::Statement statement_insert_chain( |
- GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "INSERT INTO downloads_url_chains " |
- "(id, chain_index, url) " |
- "VALUES (?, ?, ?)")); |
- for (size_t i = 0; i < info.url_chain.size(); ++i) { |
- statement_insert_chain.BindInt(0, info.id); |
- statement_insert_chain.BindInt(1, i); |
- statement_insert_chain.BindString(2, info.url_chain[i].spec()); |
- if (!statement_insert_chain.Run()) { |
- UMA_HISTOGRAM_ENUMERATION("Download.DatabaseURLChainInsertError", |
- GetDB().GetErrorCode() & 0xff, 50); |
- RemoveDownload(info.id); |
- return false; |
- } |
- statement_insert_chain.Reset(true); |
- } |
- return true; |
-} |
- |
-void DownloadDatabase::RemoveDownload(uint32 id) { |
- EnsureInProgressEntriesCleanedUp(); |
- |
- sql::Statement downloads_statement(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "DELETE FROM downloads WHERE id=?")); |
- downloads_statement.BindInt(0, id); |
- if (!downloads_statement.Run()) { |
- UMA_HISTOGRAM_ENUMERATION("Download.DatabaseMainDeleteError", |
- GetDB().GetErrorCode() & 0xff, 50); |
- return; |
- } |
- RemoveDownloadURLs(id); |
-} |
- |
-void DownloadDatabase::RemoveDownloadURLs(uint32 id) { |
- sql::Statement urlchain_statement(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "DELETE FROM downloads_url_chains WHERE id=?")); |
- urlchain_statement.BindInt(0, id); |
- if (!urlchain_statement.Run()) { |
- UMA_HISTOGRAM_ENUMERATION("Download.DatabaseURLChainDeleteError", |
- GetDB().GetErrorCode() & 0xff, 50); |
- } |
-} |
- |
-size_t DownloadDatabase::CountDownloads() { |
- EnsureInProgressEntriesCleanedUp(); |
- |
- sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE, |
- "SELECT count(*) from downloads")); |
- statement.Step(); |
- return statement.ColumnInt(0); |
-} |
- |
-} // namespace history |