OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 // Helper class which handles communication with the SafeBrowsing servers for | 5 // Helper class which handles communication with the SafeBrowsing servers for |
6 // improved binary download protection. | 6 // improved binary download protection. |
7 | 7 |
8 #ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ | 8 #ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ |
9 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ | 9 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ |
10 #pragma once | 10 #pragma once |
11 | 11 |
12 #include <map> | 12 #include <map> |
13 #include <string> | 13 #include <string> |
14 #include <vector> | 14 #include <vector> |
15 | 15 |
16 #include "base/basictypes.h" | 16 #include "base/basictypes.h" |
17 #include "base/callback.h" | 17 #include "base/callback.h" |
18 #include "base/gtest_prod_util.h" | 18 #include "base/gtest_prod_util.h" |
19 #include "base/memory/ref_counted.h" | 19 #include "base/memory/ref_counted.h" |
20 #include "base/memory/scoped_ptr.h" | 20 #include "base/memory/scoped_ptr.h" |
21 #include "base/time.h" | 21 #include "base/time.h" |
22 #include "content/common/net/url_fetcher.h" | 22 #include "content/public/common/url_fetcher_delegate.h" |
23 #include "googleurl/src/gurl.h" | 23 #include "googleurl/src/gurl.h" |
24 | 24 |
25 namespace net { | 25 namespace net { |
26 class URLRequestContextGetter; | 26 class URLRequestContextGetter; |
27 class URLRequestStatus; | 27 class URLRequestStatus; |
28 } // namespace net | 28 } // namespace net |
29 class SafeBrowsingService; | 29 class SafeBrowsingService; |
30 | 30 |
31 namespace safe_browsing { | 31 namespace safe_browsing { |
32 | 32 |
33 // This class provides an asynchronous API to check whether a particular | 33 // This class provides an asynchronous API to check whether a particular |
34 // client download is malicious or not. | 34 // client download is malicious or not. |
35 class DownloadProtectionService | 35 class DownloadProtectionService |
36 : public base::RefCountedThreadSafe<DownloadProtectionService>, | 36 : public base::RefCountedThreadSafe<DownloadProtectionService>, |
37 public URLFetcher::Delegate { | 37 public content::URLFetcherDelegate { |
38 public: | 38 public: |
39 // TODO(noelutz): we're missing some fields here: filename to get | 39 // TODO(noelutz): we're missing some fields here: filename to get |
40 // the signature, server IPs, tab URL redirect chain, ... | 40 // the signature, server IPs, tab URL redirect chain, ... |
41 struct DownloadInfo { | 41 struct DownloadInfo { |
42 std::vector<GURL> download_url_chain; | 42 std::vector<GURL> download_url_chain; |
43 GURL referrer_url; | 43 GURL referrer_url; |
44 std::string sha256_hash; | 44 std::string sha256_hash; |
45 int64 total_bytes; | 45 int64 total_bytes; |
46 bool user_initiated; | 46 bool user_initiated; |
47 DownloadInfo(); | 47 DownloadInfo(); |
(...skipping 10 matching lines...) Expand all Loading... |
58 // Callback type which is invoked once the download request is done. | 58 // Callback type which is invoked once the download request is done. |
59 typedef base::Callback<void(DownloadCheckResult)> CheckDownloadCallback; | 59 typedef base::Callback<void(DownloadCheckResult)> CheckDownloadCallback; |
60 | 60 |
61 // Creates a download service. The service is initially disabled. You need | 61 // Creates a download service. The service is initially disabled. You need |
62 // to call SetEnabled() to start it. We keep scoped references to both of | 62 // to call SetEnabled() to start it. We keep scoped references to both of |
63 // these objects. | 63 // these objects. |
64 DownloadProtectionService( | 64 DownloadProtectionService( |
65 SafeBrowsingService* sb_service, | 65 SafeBrowsingService* sb_service, |
66 net::URLRequestContextGetter* request_context_getter); | 66 net::URLRequestContextGetter* request_context_getter); |
67 | 67 |
68 // From the URLFetcher::Delegate interface. | 68 // From the content::URLFetcherDelegate interface. |
69 virtual void OnURLFetchComplete(const URLFetcher* source, | 69 virtual void OnURLFetchComplete(const URLFetcher* source) OVERRIDE; |
70 const GURL& url, | |
71 const net::URLRequestStatus& status, | |
72 int response_code, | |
73 const net::ResponseCookies& cookies, | |
74 const std::string& data) OVERRIDE; | |
75 | 70 |
76 // Checks whether the given client download is likely to be | 71 // Checks whether the given client download is likely to be |
77 // malicious or not. If this method returns true it means the | 72 // malicious or not. If this method returns true it means the |
78 // download is safe or we're unable to tell whether it is safe or | 73 // download is safe or we're unable to tell whether it is safe or |
79 // not. In this case the callback will never be called. If this | 74 // not. In this case the callback will never be called. If this |
80 // method returns false we will asynchronously check whether the | 75 // method returns false we will asynchronously check whether the |
81 // download is safe and call the callback when we have the response. | 76 // download is safe and call the callback when we have the response. |
82 // This method should be called on the UI thread. The callback will | 77 // This method should be called on the UI thread. The callback will |
83 // be called on the UI thread and will always be called asynchronously. | 78 // be called on the UI thread and will always be called asynchronously. |
84 virtual bool CheckClientDownload(const DownloadInfo& info, | 79 virtual bool CheckClientDownload(const DownloadInfo& info, |
(...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
153 std::map<const URLFetcher*, CheckDownloadCallback> download_requests_; | 148 std::map<const URLFetcher*, CheckDownloadCallback> download_requests_; |
154 | 149 |
155 // Keeps track of the state of the service. | 150 // Keeps track of the state of the service. |
156 bool enabled_; | 151 bool enabled_; |
157 | 152 |
158 DISALLOW_COPY_AND_ASSIGN(DownloadProtectionService); | 153 DISALLOW_COPY_AND_ASSIGN(DownloadProtectionService); |
159 }; | 154 }; |
160 } // namespace safe_browsing | 155 } // namespace safe_browsing |
161 | 156 |
162 #endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ | 157 #endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_ |
OLD | NEW |