blob: 30a04bf1b31653fd5c5f84574743a226fbeca8e1 (
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:
// Must call InitDownloadTable before using any other functions.
DownloadDatabase();
virtual ~DownloadDatabase();
uint32 GetNextDownloadId();
// 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 |id| 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 true if the
// creation succeeded, false otherwise.
bool CreateDownload(const DownloadRow& info);
// Remove |id| from the database.
void RemoveDownload(uint32 id);
size_t CountDownloads();
protected:
// Returns the database for the functions in this interface.
virtual sql::Connection& GetDB() = 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();
// Returns true if able to successfully add the by_ext_id and by_ext_name
// columns to the downloads table.
bool MigrateDownloadedByExtension();
// Returns true if able to successfully add the etag and last-modified columns
// to the downloads table.
bool MigrateDownloadValidators();
// 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;
static const int kDangerTypePotentiallyUnwanted;
// 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(uint32 id);
// 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_;
// 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_
|
