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
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_UI_WEBUI_DOWNLOADS_DOWNLOADS_DOM_HANDLER_H_
#define CHROME_BROWSER_UI_WEBUI_DOWNLOADS_DOWNLOADS_DOM_HANDLER_H_
#include <stdint.h>
#include <set>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "chrome/browser/download/download_danger_prompt.h"
#include "chrome/browser/download/download_warning_desktop_hats_utils.h"
#include "chrome/browser/ui/webui/downloads/downloads.mojom-forward.h"
#include "chrome/browser/ui/webui/downloads/downloads_list_tracker.h"
#include "content/public/browser/web_contents_observer.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
namespace content {
class DownloadManager;
class WebContents;
class WebUI;
} // namespace content
namespace download {
class DownloadItem;
}
// Represents the possible outcomes of showing a ESB download row promotion.
// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
//
// LINT.IfChange(SafeBrowsingEsbDownloadRowPromoOutcome)
enum class SafeBrowsingEsbDownloadRowPromoOutcome {
// The kShown and kClicked values are not meant to be mutually exclusive,
// the same promo row can be shown AND clicked.
kShown = 0,
kClicked = 1,
kMaxValue = kClicked,
};
// LINT.ThenChange(//tools/metrics/histograms/metadata/safe_browsing/enums.xml:SafeBrowsingEsbDownloadRowPromoOutcome)
// The handler for Javascript messages related to the "downloads" view,
// also observes changes to the download manager.
// TODO(calamity): Remove WebUIMessageHandler.
class DownloadsDOMHandler : public content::WebContentsObserver,
public downloads::mojom::PageHandler {
public:
DownloadsDOMHandler(
mojo::PendingReceiver<downloads::mojom::PageHandler> receiver,
mojo::PendingRemote<downloads::mojom::Page> page,
content::DownloadManager* download_manager,
content::WebUI* web_ui);
DownloadsDOMHandler(const DownloadsDOMHandler&) = delete;
DownloadsDOMHandler& operator=(const DownloadsDOMHandler&) = delete;
~DownloadsDOMHandler() override;
// WebContentsObserver implementation.
void PrimaryMainFrameRenderProcessGone(
base::TerminationStatus status) override;
// downloads::mojom::PageHandler:
void GetDownloads(const std::vector<std::string>& search_terms) override;
void OpenFileRequiringGesture(const std::string& id) override;
void Drag(const std::string& id) override;
void SaveSuspiciousRequiringGesture(const std::string& id) override;
void RecordOpenBypassWarningDialog(const std::string& id) override;
void SaveDangerousFromDialogRequiringGesture(const std::string& id) override;
void RecordCancelBypassWarningDialog(const std::string& id) override;
void DiscardDangerous(const std::string& id) override;
void RetryDownload(const std::string& id) override;
void Show(const std::string& id) override;
void Pause(const std::string& id) override;
void Resume(const std::string& id) override;
void Remove(const std::string& id) override;
void Undo() override;
void Cancel(const std::string& id) override;
void ClearAll() override;
void OpenDownloadsFolderRequiringGesture() override;
void OpenDuringScanningRequiringGesture(const std::string& id) override;
void ReviewDangerousRequiringGesture(const std::string& id) override;
void DeepScan(const std::string& id) override;
void BypassDeepScanRequiringGesture(const std::string& id) override;
void OpenEsbSettings() override;
void IsEligibleForEsbPromo(IsEligibleForEsbPromoCallback callback) override;
void LogEsbPromotionRowViewed() override;
protected:
// These methods are for mocking so that most of this class does not actually
// depend on WebUI. The other methods that depend on WebUI are
// RegisterMessages() and HandleDrag().
virtual content::WebContents* GetWebUIWebContents();
// Actually remove downloads with an ID in |removals_|. This cannot be undone.
void FinalizeRemovals();
using DownloadVector =
std::vector<raw_ptr<download::DownloadItem, VectorExperimental>>;
// Remove all downloads in |to_remove|. Safe downloads can be revived,
// dangerous ones are immediately removed. Protected for testing.
void RemoveDownloads(const DownloadVector& to_remove);
private:
using IdSet = std::set<uint32_t>;
// Convenience method to call |main_notifier_->GetManager()| while
// null-checking |main_notifier_|.
content::DownloadManager* GetMainNotifierManager() const;
// Convenience method to call |original_notifier_->GetManager()| while
// null-checking |original_notifier_|.
content::DownloadManager* GetOriginalNotifierManager() const;
// Launches a HaTS survey for a download warning that is heeded, bypassed, or
// ignored (if all preconditions are met).
void MaybeTriggerDownloadWarningHatsSurvey(
download::DownloadItem* item,
DownloadWarningHatsType survey_type);
// Called when the downloads page is dismissed by closing the tab, or
// navigating the tab to another page.
void OnDownloadsPageDismissed();
// Returns true if the records of any downloaded items are allowed (and able)
// to be deleted.
bool IsDeletingHistoryAllowed();
// Returns the download that is referred to by a given string |id|.
download::DownloadItem* GetDownloadByStringId(const std::string& id);
// Returns the download with |id| or NULL if it doesn't exist.
download::DownloadItem* GetDownloadById(uint32_t id);
// Removes the download specified by an ID from JavaScript in |args|.
void RemoveDownloadInArgs(const std::string& id);
// Checks whether a download's file was removed from its original location.
void CheckForRemovedFiles();
DownloadsListTracker list_tracker_;
// IDs of downloads to remove when this handler gets deleted.
std::vector<IdSet> removals_;
// Whether the render process has gone.
bool render_process_gone_ = false;
raw_ptr<content::WebUI> web_ui_;
mojo::Receiver<downloads::mojom::PageHandler> receiver_;
base::WeakPtrFactory<DownloadsDOMHandler> weak_ptr_factory_{this};
};
#endif // CHROME_BROWSER_UI_WEBUI_DOWNLOADS_DOWNLOADS_DOM_HANDLER_H_
|
