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
|
// 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/task.h"
#include "content/browser/download/download_manager_delegate.h"
#include "content/common/notification_observer.h"
#include "content/common/notification_registrar.h"
class CrxInstaller;
class DownloadHistory;
class DownloadItem;
class DownloadManager;
class DownloadPrefs;
class Profile;
struct DownloadStateInfo;
#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 DownloadManagerDelegate,
public NotificationObserver {
public:
explicit ChromeDownloadManagerDelegate(Profile* profile);
// Returns true if the given item is for an extension download.
static bool IsExtensionDownload(const DownloadItem* item);
void SetDownloadManager(DownloadManager* dm);
virtual void Shutdown() OVERRIDE;
virtual bool ShouldStartDownload(int32 download_id) OVERRIDE;
virtual void ChooseDownloadPath(TabContents* tab_contents,
const FilePath& suggested_path,
void* data) OVERRIDE;
virtual bool OverrideIntermediatePath(DownloadItem* item,
FilePath* intermediate_path) OVERRIDE;
virtual TabContents* GetAlternativeTabContentsToNotifyForDownload() OVERRIDE;
virtual bool ShouldOpenFileBasedOnExtension(const FilePath& path) OVERRIDE;
virtual bool ShouldOpenDownload(DownloadItem* item) OVERRIDE;
virtual bool ShouldCompleteDownload(DownloadItem* item) OVERRIDE;
virtual bool GenerateFileHash() OVERRIDE;
virtual void OnResponseCompleted(DownloadItem* item,
const std::string& hash) OVERRIDE;
virtual void AddItemToPersistentStore(DownloadItem* item) OVERRIDE;
virtual void UpdateItemInPersistentStore(DownloadItem* item) OVERRIDE;
virtual void UpdatePathForItemInPersistentStore(
DownloadItem* item,
const FilePath& new_path) OVERRIDE;
virtual void RemoveItemFromPersistentStore(DownloadItem* item) OVERRIDE;
virtual void RemoveItemsFromPersistentStoreBetween(
const base::Time remove_begin,
const base::Time remove_end) OVERRIDE;
virtual void GetSaveDir(TabContents* tab_contents,
FilePath* website_save_dir,
FilePath* download_save_dir) OVERRIDE;
virtual void ChooseSavePath(const base::WeakPtr<SavePackage>& save_package,
const FilePath& suggested_path,
bool can_save_as_complete) 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<DownloadManager> download_manager_;
private:
friend class base::RefCountedThreadSafe<ChromeDownloadManagerDelegate>;
// NotificationObserver implementation.
virtual void Observe(int type,
const NotificationSource& source,
const NotificationDetails& details) OVERRIDE;
// Callback function after url is checked with safebrowsing service.
void CheckDownloadUrlDone(int32 download_id, bool is_dangerous_url);
// 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 DownloadItem& download,
const DownloadStateInfo& state,
bool visited_referrer_before);
// Callback from history system.
void OnItemAddedToPersistentStore(int32 download_id, int64 db_handle);
// Callback function after download file hash is checked with safebrowsing
// service.
void CheckDownloadHashDone(int32 download_id, bool is_dangerous_hash);
Profile* profile_;
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_;
NotificationRegistrar registrar_;
DISALLOW_COPY_AND_ASSIGN(ChromeDownloadManagerDelegate);
};
#endif // CHROME_BROWSER_DOWNLOAD_CHROME_DOWNLOAD_MANAGER_DELEGATE_H_
|
