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
|
// 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_DOWNLOAD_SAFE_BROWSING_CLIENT_H_
#define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SAFE_BROWSING_CLIENT_H_
#pragma once
#include "base/callback_old.h"
#include "base/memory/ref_counted.h"
#include "base/time.h"
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
// This is a helper class used by DownloadManager to check a download URL with
// SafeBrowsingService. The client is refcounted and will be released once
// there is no reference to it.
// Usage:
// {
// scoped_refptr<DownloadSBClient> client_ = new DownloadSBClient(...);
// client_->CheckDownloadUrl(..., NewCallback(this,
// &DownloadManager::UrlCallBack));
// or
// client_->CheckDownloadHash(..., NewCallback(this,
// &DownloadManager::HashCallBack));
// }
// DownloadManager::UrlCallBack(...) or HashCallCall {
// // After this, the |client_| is gone.
// }
class DownloadSBClient
: public SafeBrowsingService::Client,
public base::RefCountedThreadSafe<DownloadSBClient> {
public:
typedef Callback2<int32, bool>::Type UrlDoneCallback;
typedef Callback2<int32, bool>::Type HashDoneCallback;
DownloadSBClient(int32 download_id,
const std::vector<GURL>& url_chain,
const GURL& referrer_url,
bool safe_browsing_enabled);
// Call safebrowsing service to verify the download.
// For each DownloadSBClient instance, either CheckDownloadUrl or
// CheckDownloadHash can be called, and be called only once.
// DownloadSBClient instance.
void CheckDownloadUrl(UrlDoneCallback* callback);
void CheckDownloadHash(const std::string& hash, HashDoneCallback* callback);
private:
// Call SafeBrowsingService on IO thread to verify the download URL or
// hash of downloaded file.
void CheckDownloadUrlOnIOThread(const std::vector<GURL>& url_chain);
void CheckDownloadHashOnIOThread(const std::string& hash);
// Callback interfaces for SafeBrowsingService::Client.
virtual void OnDownloadUrlCheckResult(
const std::vector<GURL>& url_chain,
SafeBrowsingService::UrlCheckResult result) OVERRIDE;
virtual void OnDownloadHashCheckResult(
const std::string& hash,
SafeBrowsingService::UrlCheckResult result) OVERRIDE;
// Enumerate for histogramming purposes.
// DO NOT CHANGE THE ORDERING OF THESE VALUES (different histogram data will
// be mixed together based on their values).
enum SBStatsType {
DOWNLOAD_URL_CHECKS_TOTAL,
DOWNLOAD_URL_CHECKS_CANCELED,
DOWNLOAD_URL_CHECKS_MALWARE,
DOWNLOAD_HASH_CHECKS_TOTAL,
DOWNLOAD_HASH_CHECKS_MALWARE,
// Memory space for histograms is determined by the max.
// ALWAYS ADD NEW VALUES BEFORE THIS ONE.
DOWNLOAD_CHECKS_MAX
};
friend class base::RefCountedThreadSafe<DownloadSBClient>;
friend class DownloadSBClientTest_UrlHit_Test;
friend class DownloadSBClientTest_DigestHit_Test;
// Set the |sb_service_| manually for testing purposes.
void SetSBService(SafeBrowsingService* service) {
sb_service_ = service;
}
virtual ~DownloadSBClient();
// Call DownloadManager on UI thread for download URL or hash check.
void SafeBrowsingCheckUrlDone(SafeBrowsingService::UrlCheckResult result);
void SafeBrowsingCheckHashDone(SafeBrowsingService::UrlCheckResult result,
const std::string& hash);
// Report malware hits to safebrowsing service. |hash| should be empty if
// this is an URL check result.
void ReportMalware(SafeBrowsingService::UrlCheckResult result,
const std::string& hash);
// Update the UMA stats.
void UpdateDownloadCheckStats(SBStatsType stat_type);
scoped_ptr<UrlDoneCallback> url_done_callback_;
scoped_ptr<HashDoneCallback> hash_done_callback_;
int32 download_id_;
scoped_refptr<SafeBrowsingService> sb_service_;
// These URLs are used to report malware to safe browsing service.
std::vector<GURL> url_chain_;
GURL referrer_url_;
// When a safebrowsing check starts, for stats purpose.
base::TimeTicks start_time_;
// Whether the profile from which this client was created has enabled the
// safe browsing service.
bool safe_browsing_enabled_;
DISALLOW_COPY_AND_ASSIGN(DownloadSBClient);
};
#endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SAFE_BROWSING_CLIENT_H_
|
