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
|
// Copyright 2013 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_SAFE_BROWSING_DOWNLOAD_PROTECTION_DOWNLOAD_FEEDBACK_SERVICE_H_
#define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_DOWNLOAD_FEEDBACK_SERVICE_H_
#include <memory>
#include <string>
#include "base/containers/queue.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "chrome/browser/download/download_commands.h"
#include "chrome/browser/safe_browsing/download_protection/download_protection_service.h"
#include "chrome/browser/safe_browsing/download_protection/download_protection_util.h"
#include "components/download/public/common/download_danger_type.h"
namespace base {
class TaskRunner;
}
namespace download {
class DownloadItem;
}
namespace safe_browsing {
class DownloadFeedback;
// Tracks active DownloadFeedback objects, provides interface for storing ping
// data for malicious downloads.
// Lives on the UI thread.
// TODO(crbug.com/397407934): Download feedback is not supported on Android yet,
// but it will be.
class DownloadFeedbackService {
public:
DownloadFeedbackService(
DownloadProtectionService* download_protection_service,
base::TaskRunner* file_task_runner);
DownloadFeedbackService(const DownloadFeedbackService&) = delete;
DownloadFeedbackService& operator=(const DownloadFeedbackService&) = delete;
virtual ~DownloadFeedbackService();
// Begin download feedback for the given |download| in the given
// |profile|. This must only be called if IsEnabledForDownload is
// true for |download|.
virtual void BeginFeedbackForDownload(Profile* profile,
download::DownloadItem* download,
const std::string& ping_request,
const std::string& ping_response);
private:
static void BeginFeedbackOrDeleteFile(
const scoped_refptr<base::TaskRunner>& file_task_runner,
const base::WeakPtr<DownloadFeedbackService>& service,
Profile* profile,
const std::string& ping_request,
const std::string& ping_response,
uint64_t file_size,
const base::FilePath& path);
void StartPendingFeedback();
void BeginFeedback(Profile* profile,
const std::string& ping_request,
const std::string& ping_response,
const base::FilePath& path,
uint64_t file_size);
void FeedbackComplete();
// Safe because the DownloadProtectionService owns this.
raw_ptr<DownloadProtectionService> download_protection_service_;
scoped_refptr<base::TaskRunner> file_task_runner_;
// Currently active & pending uploads. The first item is active, remaining
// items are pending.
base::queue<std::unique_ptr<DownloadFeedback>> active_feedback_;
base::WeakPtrFactory<DownloadFeedbackService> weak_ptr_factory_{this};
};
} // namespace safe_browsing
#endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_DOWNLOAD_FEEDBACK_SERVICE_H_
|
