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 #include "chrome/browser/safe_browsing/ping_manager.h" | 5 #include "chrome/browser/safe_browsing/ping_manager.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
11 #include "base/stl_util.h" | |
12 #include "base/strings/string_util.h" | 11 #include "base/strings/string_util.h" |
13 #include "base/strings/stringprintf.h" | 12 #include "base/strings/stringprintf.h" |
14 #include "chrome/browser/safe_browsing/permission_reporter.h" | 13 #include "chrome/browser/safe_browsing/permission_reporter.h" |
15 #include "components/certificate_reporting/error_reporter.h" | 14 #include "components/certificate_reporting/error_reporter.h" |
16 #include "content/public/browser/browser_thread.h" | 15 #include "content/public/browser/browser_thread.h" |
17 #include "google_apis/google_api_keys.h" | 16 #include "google_apis/google_api_keys.h" |
18 #include "net/base/escape.h" | 17 #include "net/base/escape.h" |
19 #include "net/base/load_flags.h" | 18 #include "net/base/load_flags.h" |
20 #include "net/ssl/ssl_info.h" | 19 #include "net/ssl/ssl_info.h" |
21 #include "net/url_request/report_sender.h" | 20 #include "net/url_request/report_sender.h" |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
67 cookies_preference)); | 66 cookies_preference)); |
68 | 67 |
69 permission_reporter_.reset( | 68 permission_reporter_.reset( |
70 new PermissionReporter(request_context_getter->GetURLRequestContext())); | 69 new PermissionReporter(request_context_getter->GetURLRequestContext())); |
71 } | 70 } |
72 | 71 |
73 version_ = SafeBrowsingProtocolManagerHelper::Version(); | 72 version_ = SafeBrowsingProtocolManagerHelper::Version(); |
74 } | 73 } |
75 | 74 |
76 SafeBrowsingPingManager::~SafeBrowsingPingManager() { | 75 SafeBrowsingPingManager::~SafeBrowsingPingManager() { |
77 // Delete in-progress safebrowsing reports (hits and details). | |
78 base::STLDeleteContainerPointers(safebrowsing_reports_.begin(), | |
79 safebrowsing_reports_.end()); | |
80 } | 76 } |
81 | 77 |
82 // net::URLFetcherDelegate implementation ---------------------------------- | 78 // net::URLFetcherDelegate implementation ---------------------------------- |
83 | 79 |
84 // All SafeBrowsing request responses are handled here. | 80 // All SafeBrowsing request responses are handled here. |
85 void SafeBrowsingPingManager::OnURLFetchComplete( | 81 void SafeBrowsingPingManager::OnURLFetchComplete( |
86 const net::URLFetcher* source) { | 82 const net::URLFetcher* source) { |
87 Reports::iterator sit = safebrowsing_reports_.find(source); | 83 auto it = |
88 DCHECK(sit != safebrowsing_reports_.end()); | 84 std::find_if(safebrowsing_reports_.begin(), safebrowsing_reports_.end(), |
89 delete *sit; | 85 [source](const std::unique_ptr<net::URLFetcher>& ptr) { |
90 safebrowsing_reports_.erase(sit); | 86 return ptr.get() == source; |
87 }); | |
88 DCHECK(it != safebrowsing_reports_.end()); | |
89 safebrowsing_reports_.erase(it); | |
91 } | 90 } |
92 | 91 |
93 // Sends a SafeBrowsing "hit" report. | 92 // Sends a SafeBrowsing "hit" report. |
94 void SafeBrowsingPingManager::ReportSafeBrowsingHit( | 93 void SafeBrowsingPingManager::ReportSafeBrowsingHit( |
95 const safe_browsing::HitReport& hit_report) { | 94 const safe_browsing::HitReport& hit_report) { |
96 GURL report_url = SafeBrowsingHitUrl(hit_report); | 95 GURL report_url = SafeBrowsingHitUrl(hit_report); |
97 net::URLFetcher* report = | 96 std::unique_ptr<net::URLFetcher> report_ptr = net::URLFetcher::Create( |
98 net::URLFetcher::Create(report_url, hit_report.post_data.empty() | 97 report_url, hit_report.post_data.empty() ? net::URLFetcher::GET |
99 ? net::URLFetcher::GET | 98 : net::URLFetcher::POST, |
100 : net::URLFetcher::POST, | 99 this); |
101 this) | 100 net::URLFetcher* report = report_ptr.get(); |
102 .release(); | 101 report_ptr->SetLoadFlags(net::LOAD_DISABLE_CACHE); |
103 report->SetLoadFlags(net::LOAD_DISABLE_CACHE); | 102 report_ptr->SetRequestContext(request_context_getter_.get()); |
104 report->SetRequestContext(request_context_getter_.get()); | |
105 if (!hit_report.post_data.empty()) | 103 if (!hit_report.post_data.empty()) |
106 report->SetUploadData("text/plain", hit_report.post_data); | 104 report_ptr->SetUploadData("text/plain", hit_report.post_data); |
107 safebrowsing_reports_.insert(report); | 105 safebrowsing_reports_.insert(std::move(report_ptr)); |
Nico
2016/09/15 16:07:16
nit: If you do
(*safebrowsing_reports_.insert(
Avi (use Gerrit)
2016/09/15 19:53:45
I suppose so, but I'll pass for simplicity/clarity
| |
108 report->Start(); | 106 report->Start(); |
109 } | 107 } |
110 | 108 |
111 // Sends threat details for users who opt-in. | 109 // Sends threat details for users who opt-in. |
112 void SafeBrowsingPingManager::ReportThreatDetails(const std::string& report) { | 110 void SafeBrowsingPingManager::ReportThreatDetails(const std::string& report) { |
113 GURL report_url = ThreatDetailsUrl(); | 111 GURL report_url = ThreatDetailsUrl(); |
114 net::URLFetcher* fetcher = | 112 std::unique_ptr<net::URLFetcher> fetcher = |
115 net::URLFetcher::Create(report_url, net::URLFetcher::POST, this) | 113 net::URLFetcher::Create(report_url, net::URLFetcher::POST, this); |
116 .release(); | |
117 fetcher->SetLoadFlags(net::LOAD_DISABLE_CACHE); | 114 fetcher->SetLoadFlags(net::LOAD_DISABLE_CACHE); |
118 fetcher->SetRequestContext(request_context_getter_.get()); | 115 fetcher->SetRequestContext(request_context_getter_.get()); |
119 fetcher->SetUploadData("application/octet-stream", report); | 116 fetcher->SetUploadData("application/octet-stream", report); |
120 // Don't try too hard to send reports on failures. | 117 // Don't try too hard to send reports on failures. |
121 fetcher->SetAutomaticallyRetryOn5xx(false); | 118 fetcher->SetAutomaticallyRetryOn5xx(false); |
122 fetcher->Start(); | 119 fetcher->Start(); |
123 safebrowsing_reports_.insert(fetcher); | 120 safebrowsing_reports_.insert(std::move(fetcher)); |
124 } | 121 } |
125 | 122 |
126 void SafeBrowsingPingManager::ReportInvalidCertificateChain( | 123 void SafeBrowsingPingManager::ReportInvalidCertificateChain( |
127 const std::string& serialized_report) { | 124 const std::string& serialized_report) { |
128 DCHECK(certificate_error_reporter_); | 125 DCHECK(certificate_error_reporter_); |
129 certificate_error_reporter_->SendExtendedReportingReport(serialized_report); | 126 certificate_error_reporter_->SendExtendedReportingReport(serialized_report); |
130 } | 127 } |
131 | 128 |
132 void SafeBrowsingPingManager::SetCertificateErrorReporterForTesting( | 129 void SafeBrowsingPingManager::SetCertificateErrorReporterForTesting( |
133 std::unique_ptr<certificate_reporting::ErrorReporter> | 130 std::unique_ptr<certificate_reporting::ErrorReporter> |
(...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
230 version_.c_str()); | 227 version_.c_str()); |
231 std::string api_key = google_apis::GetAPIKey(); | 228 std::string api_key = google_apis::GetAPIKey(); |
232 if (!api_key.empty()) { | 229 if (!api_key.empty()) { |
233 base::StringAppendF(&url, "&key=%s", | 230 base::StringAppendF(&url, "&key=%s", |
234 net::EscapeQueryParamValue(api_key, true).c_str()); | 231 net::EscapeQueryParamValue(api_key, true).c_str()); |
235 } | 232 } |
236 return GURL(url); | 233 return GURL(url); |
237 } | 234 } |
238 | 235 |
239 } // namespace safe_browsing | 236 } // namespace safe_browsing |
OLD | NEW |