blob: ef85279ca0861115bc94b4a2bb98820f9dc8facf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
// 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.
#ifndef CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_
#define CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_
#include <string>
#include <vector>
#include "base/gtest_prod_util.h"
#include "base/threading/platform_thread.h"
#include "content/public/browser/download_item.h"
#include "sql/meta_table.h"
namespace sql {
class Connection;
}
namespace history {
struct DownloadRow;
// Maintains a table of downloads.
class DownloadDatabase {
public:
// The value of |db_handle| indicating that the associated DownloadItem is not
// yet persisted.
static const int64 kUninitializedHandle;
// Must call InitDownloadTable before using any other functions.
DownloadDatabase();
virtual ~DownloadDatabase();
int next_download_id() const { return next_id_; }
// Get all the downloads from the database.
void QueryDownloads(
std::vector<DownloadRow>* results);
// Update the state of one download. Returns true if successful.
// Does not update |url|, |start_time|; uses |db_handle| only
// to select the row in the database table to update.
bool UpdateDownload(const DownloadRow& data);
// Create a new database entry for one download and return its primary db id.
int64 CreateDownload(const DownloadRow& info);
// Remove |handle| from the database.
void RemoveDownload(int64 handle);
int CountDownloads();
protected:
// Returns the database for the functions in this interface.
virtual sql::Connection& GetDB() = 0;
// Returns the meta-table object for the functions in this interface.
virtual sql::MetaTable& GetMetaTable() = 0;
// Returns true if able to successfully rewrite the invalid values for the
// |state| field from 3 to 4. Returns false if there was an error fixing the
// database. See http://crbug.com/140687
bool MigrateDownloadsState();
// Returns true if able to successfully add the last interrupt reason and the
// two target paths to downloads.
bool MigrateDownloadsReasonPathsAndDangerType();
// Returns true if able to successfully add the referrer column to the
// downloads table.
bool MigrateReferrer();
// Creates the downloads table if needed.
bool InitDownloadTable();
// Used to quickly clear the downloads. First you would drop it, then you
// would re-initialize it.
bool DropDownloadTable();
private:
FRIEND_TEST_ALL_PREFIXES(
HistoryBackendDBTest, ConfirmDownloadInProgressCleanup);
// Values used in the database for DownloadItem::State.
static const int kStateInvalid;
static const int kStateInProgress;
static const int kStateComplete;
static const int kStateCancelled;
static const int kStateBug140687;
static const int kStateInterrupted;
// Values used in the database for DownloadItem::DangerType
static const int kDangerTypeInvalid;
static const int kDangerTypeNotDangerous;
static const int kDangerTypeDangerousFile;
static const int kDangerTypeDangerousUrl;
static const int kDangerTypeDangerousContent;
static const int kDangerTypeMaybeDangerousContent;
static const int kDangerTypeUncommonContent;
static const int kDangerTypeUserValidated;
static const int kDangerTypeDangerousHost;
// Fixes state of the download entries. Sometimes entries with IN_PROGRESS
// state are not updated during browser shutdown (particularly when crashing).
// On the next start such entries are considered interrupted with
// interrupt reason |DOWNLOAD_INTERRUPT_REASON_CRASH|. This function
// fixes such entries.
void EnsureInProgressEntriesCleanedUp();
bool EnsureColumnExists(const std::string& name, const std::string& type);
void RemoveDownloadURLs(int64 handle);
// Utility functions for conversion between DownloadItem types
// and DownloadDatabase constants.
static int StateToInt(content::DownloadItem::DownloadState state);
static content::DownloadItem::DownloadState IntToState(int state);
static int DangerTypeToInt(content::DownloadDangerType danger_type);
static content::DownloadDangerType IntToDangerType(int danger_type);
bool owning_thread_set_;
base::PlatformThreadId owning_thread_;
int next_id_;
int next_db_handle_;
// Initialized to false on construction, and checked in all functional
// routines post-migration in the database for a possible call to
// CleanUpInProgressEntries(). This allows us to avoid
// doing the cleanup until after any DB migration and unless we are
// actually use the downloads database.
bool in_progress_entry_cleanup_completed_;
DISALLOW_COPY_AND_ASSIGN(DownloadDatabase);
};
} // namespace history
#endif // CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_
|
