| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 // Implementation of the ThreatDetails class. | 5 // Implementation of the ThreatDetails class. |
| 6 | 6 |
| 7 #include "chrome/browser/safe_browsing/threat_details.h" | 7 #include "chrome/browser/safe_browsing/threat_details.h" |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/lazy_instance.h" | 10 #include "base/lazy_instance.h" |
| 11 #include "base/md5.h" | 11 #include "base/md5.h" |
| 12 #include "base/strings/string_util.h" | 12 #include "base/strings/string_util.h" |
| 13 #include "chrome/browser/safe_browsing/report.pb.h" | 13 #include "chrome/browser/safe_browsing/report.pb.h" |
| 14 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 14 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 15 #include "chrome/browser/safe_browsing/threat_details_cache.h" | 15 #include "chrome/browser/safe_browsing/threat_details_cache.h" |
| 16 #include "content/public/browser/browser_thread.h" | 16 #include "content/public/browser/browser_thread.h" |
| 17 #include "net/base/host_port_pair.h" | 17 #include "net/base/host_port_pair.h" |
| 18 #include "net/base/load_flags.h" | 18 #include "net/base/load_flags.h" |
| 19 #include "net/base/net_errors.h" | 19 #include "net/base/net_errors.h" |
| 20 #include "net/http/http_response_headers.h" | 20 #include "net/http/http_response_headers.h" |
| 21 #include "net/url_request/url_fetcher.h" | 21 #include "net/url_request/url_fetcher.h" |
| 22 #include "net/url_request/url_request_context_getter.h" | 22 #include "net/url_request/url_request_context_getter.h" |
| 23 #include "net/url_request/url_request_status.h" | 23 #include "net/url_request/url_request_status.h" |
| 24 | 24 |
| 25 using content::BrowserThread; | 25 using content::BrowserThread; |
| 26 using safe_browsing::ClientMalwareReportRequest; | 26 |
| 27 namespace safe_browsing { |
| 28 |
| 29 class ClientMalwareReportRequest; |
| 27 | 30 |
| 28 // Only send small files for now, a better strategy would use the size | 31 // Only send small files for now, a better strategy would use the size |
| 29 // of the whole report and the user's bandwidth. | 32 // of the whole report and the user's bandwidth. |
| 30 static const uint32 kMaxBodySizeBytes = 1024; | 33 static const uint32 kMaxBodySizeBytes = 1024; |
| 31 | 34 |
| 32 ThreatDetailsCacheCollector::ThreatDetailsCacheCollector() | 35 ThreatDetailsCacheCollector::ThreatDetailsCacheCollector() |
| 33 : resources_(NULL), result_(NULL), has_started_(false) {} | 36 : resources_(NULL), result_(NULL), has_started_(false) {} |
| 34 | 37 |
| 35 void ThreatDetailsCacheCollector::StartCacheCollection( | 38 void ThreatDetailsCacheCollector::StartCacheCollection( |
| 36 net::URLRequestContextGetter* request_context_getter, | 39 net::URLRequestContextGetter* request_context_getter, |
| 37 safe_browsing::ResourceMap* resources, | 40 ResourceMap* resources, |
| 38 bool* result, | 41 bool* result, |
| 39 const base::Closure& callback) { | 42 const base::Closure& callback) { |
| 40 // Start the data collection from the HTTP cache. We use a URLFetcher | 43 // Start the data collection from the HTTP cache. We use a URLFetcher |
| 41 // and set the right flags so we only hit the cache. | 44 // and set the right flags so we only hit the cache. |
| 42 DVLOG(1) << "Getting cache data for all urls..."; | 45 DVLOG(1) << "Getting cache data for all urls..."; |
| 43 request_context_getter_ = request_context_getter; | 46 request_context_getter_ = request_context_getter; |
| 44 resources_ = resources; | 47 resources_ = resources; |
| 45 resources_it_ = resources_->begin(); | 48 resources_it_ = resources_->begin(); |
| 46 result_ = result; | 49 result_ = result; |
| 47 callback_ = callback; | 50 callback_ = callback; |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 current_fetch_->SetRequestContext(request_context_getter_.get()); | 85 current_fetch_->SetRequestContext(request_context_getter_.get()); |
| 83 // Only from cache, and don't save cookies. | 86 // Only from cache, and don't save cookies. |
| 84 current_fetch_->SetLoadFlags(net::LOAD_ONLY_FROM_CACHE | | 87 current_fetch_->SetLoadFlags(net::LOAD_ONLY_FROM_CACHE | |
| 85 net::LOAD_DO_NOT_SAVE_COOKIES); | 88 net::LOAD_DO_NOT_SAVE_COOKIES); |
| 86 current_fetch_->SetAutomaticallyRetryOn5xx(false); // No retries. | 89 current_fetch_->SetAutomaticallyRetryOn5xx(false); // No retries. |
| 87 current_fetch_->Start(); // OnURLFetchComplete will be called when done. | 90 current_fetch_->Start(); // OnURLFetchComplete will be called when done. |
| 88 } | 91 } |
| 89 | 92 |
| 90 ClientMalwareReportRequest::Resource* ThreatDetailsCacheCollector::GetResource( | 93 ClientMalwareReportRequest::Resource* ThreatDetailsCacheCollector::GetResource( |
| 91 const GURL& url) { | 94 const GURL& url) { |
| 92 safe_browsing::ResourceMap::iterator it = resources_->find(url.spec()); | 95 ResourceMap::iterator it = resources_->find(url.spec()); |
| 93 if (it != resources_->end()) { | 96 if (it != resources_->end()) { |
| 94 return it->second.get(); | 97 return it->second.get(); |
| 95 } | 98 } |
| 96 return NULL; | 99 return NULL; |
| 97 } | 100 } |
| 98 | 101 |
| 99 void ThreatDetailsCacheCollector::OnURLFetchComplete( | 102 void ThreatDetailsCacheCollector::OnURLFetchComplete( |
| 100 const net::URLFetcher* source) { | 103 const net::URLFetcher* source) { |
| 101 DVLOG(1) << "OnUrlFetchComplete"; | 104 DVLOG(1) << "OnUrlFetchComplete"; |
| 102 DCHECK_CURRENTLY_ON(BrowserThread::IO); | 105 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| (...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 194 base::Bind(&ThreatDetailsCacheCollector::OpenEntry, this)); | 197 base::Bind(&ThreatDetailsCacheCollector::OpenEntry, this)); |
| 195 } | 198 } |
| 196 | 199 |
| 197 void ThreatDetailsCacheCollector::AllDone(bool success) { | 200 void ThreatDetailsCacheCollector::AllDone(bool success) { |
| 198 DVLOG(1) << "AllDone"; | 201 DVLOG(1) << "AllDone"; |
| 199 DCHECK_CURRENTLY_ON(BrowserThread::IO); | 202 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 200 *result_ = success; | 203 *result_ = success; |
| 201 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, callback_); | 204 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, callback_); |
| 202 callback_.Reset(); | 205 callback_.Reset(); |
| 203 } | 206 } |
| 207 |
| 208 } // namespace safe_browsing |
| OLD | NEW |