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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
// Copyright (c) 2011 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_DOWNLOAD_CHROME_DOWNLOAD_MANAGER_DELEGATE_H_
#define CHROME_BROWSER_DOWNLOAD_CHROME_DOWNLOAD_MANAGER_DELEGATE_H_
#pragma once
#include "base/compiler_specific.h"
#include "base/hash_tables.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/safe_browsing/download_protection_service.h"
#include "content/public/browser/download_manager_delegate.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
class CrxInstaller;
class DownloadHistory;
class DownloadPrefs;
class Profile;
struct DownloadStateInfo;
namespace content {
class DownloadItem;
class DownloadManager;
}
#if defined(COMPILER_GCC)
namespace __gnu_cxx {
template<>
struct hash<CrxInstaller*> {
std::size_t operator()(CrxInstaller* const& p) const {
return reinterpret_cast<std::size_t>(p);
}
};
} // namespace __gnu_cxx
#endif
// This is the Chrome side helper for the download system.
class ChromeDownloadManagerDelegate
: public base::RefCountedThreadSafe<ChromeDownloadManagerDelegate>,
public content::DownloadManagerDelegate,
public content::NotificationObserver {
public:
explicit ChromeDownloadManagerDelegate(Profile* profile);
void SetDownloadManager(content::DownloadManager* dm);
// Returns true if the given item is for an extension download.
static bool IsExtensionDownload(const content::DownloadItem* item);
virtual void Shutdown() OVERRIDE;
virtual content::DownloadId GetNextId() OVERRIDE;
virtual bool ShouldStartDownload(int32 download_id) OVERRIDE;
virtual void ChooseDownloadPath(content::WebContents* web_contents,
const FilePath& suggested_path,
void* data) OVERRIDE;
virtual FilePath GetIntermediatePath(const FilePath& suggested_path) OVERRIDE;
virtual content::WebContents*
GetAlternativeWebContentsToNotifyForDownload() OVERRIDE;
virtual bool ShouldOpenFileBasedOnExtension(const FilePath& path) OVERRIDE;
virtual bool ShouldCompleteDownload(content::DownloadItem* item) OVERRIDE;
virtual bool ShouldOpenDownload(content::DownloadItem* item) OVERRIDE;
virtual bool GenerateFileHash() OVERRIDE;
virtual void AddItemToPersistentStore(content::DownloadItem* item) OVERRIDE;
virtual void UpdateItemInPersistentStore(
content::DownloadItem* item) OVERRIDE;
virtual void UpdatePathForItemInPersistentStore(
content::DownloadItem* item,
const FilePath& new_path) OVERRIDE;
virtual void RemoveItemFromPersistentStore(
content::DownloadItem* item) OVERRIDE;
virtual void RemoveItemsFromPersistentStoreBetween(
base::Time remove_begin,
base::Time remove_end) OVERRIDE;
virtual void GetSaveDir(content::WebContents* web_contents,
FilePath* website_save_dir,
FilePath* download_save_dir) OVERRIDE;
virtual void ChooseSavePath(
content::WebContents* web_contents,
const FilePath& suggested_path,
const FilePath::StringType& default_extension,
bool can_save_as_complete,
content::SaveFilePathPickedCallback callback) OVERRIDE;
virtual void DownloadProgressUpdated() OVERRIDE;
DownloadPrefs* download_prefs() { return download_prefs_.get(); }
DownloadHistory* download_history() { return download_history_.get(); }
protected:
// So that test classes can inherit from this for override purposes.
virtual ~ChromeDownloadManagerDelegate();
// So that test classes that inherit from this for override purposes
// can call back into the DownloadManager.
scoped_refptr<content::DownloadManager> download_manager_;
private:
friend class base::RefCountedThreadSafe<ChromeDownloadManagerDelegate>;
// content::NotificationObserver implementation.
virtual void Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) OVERRIDE;
// Returns the SafeBrowsing download protection service if it's
// enabled. Returns NULL otherwise.
safe_browsing::DownloadProtectionService* GetDownloadProtectionService();
// Callback function after url is checked with safebrowsing service.
void CheckDownloadUrlDone(
int32 download_id,
safe_browsing::DownloadProtectionService::DownloadCheckResult result);
// Callback function after the DownloadProtectionService completes.
void CheckClientDownloadDone(
int32 download_id,
safe_browsing::DownloadProtectionService::DownloadCheckResult result);
// Callback function after we check whether the referrer URL has been visited
// before today.
void CheckVisitedReferrerBeforeDone(int32 download_id,
bool visited_referrer_before);
// Called on the download thread to check whether the suggested file path
// exists. We don't check if the file exists on the UI thread to avoid UI
// stalls from interacting with the file system.
void CheckIfSuggestedPathExists(int32 download_id,
DownloadStateInfo state,
const FilePath& default_path);
// Called on the UI thread once it's determined whether the suggested file
// path exists.
void OnPathExistenceAvailable(int32 download_id,
const DownloadStateInfo& new_state);
// Returns true if this download should show the "dangerous file" warning.
// Various factors are considered, such as the type of the file, whether a
// user action initiated the download, and whether the user has explicitly
// marked the file type as "auto open".
bool IsDangerousFile(const content::DownloadItem& download,
const DownloadStateInfo& state,
bool visited_referrer_before);
// Callback from history system.
void OnItemAddedToPersistentStore(int32 download_id, int64 db_handle);
Profile* profile_;
int next_download_id_;
scoped_ptr<DownloadPrefs> download_prefs_;
scoped_ptr<DownloadHistory> download_history_;
// Maps from pending extension installations to DownloadItem IDs.
typedef base::hash_map<CrxInstaller*, int> CrxInstallerMap;
CrxInstallerMap crx_installers_;
content::NotificationRegistrar registrar_;
DISALLOW_COPY_AND_ASSIGN(ChromeDownloadManagerDelegate);
};
#endif // CHROME_BROWSER_DOWNLOAD_CHROME_DOWNLOAD_MANAGER_DELEGATE_H_
|
