| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/net/safe_search_util.h" | 5 #include "chrome/browser/net/safe_search_util.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 #include <utility> | 8 #include <utility> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/logging.h" | 11 #include "base/logging.h" |
| 12 #include "base/macros.h" | 12 #include "base/macros.h" |
| 13 #include "base/strings/string_number_conversions.h" | 13 #include "base/strings/string_number_conversions.h" |
| 14 #include "base/strings/string_piece.h" | 14 #include "base/strings/string_piece.h" |
| 15 #include "base/strings/string_split.h" | 15 #include "base/strings/string_split.h" |
| 16 #include "base/strings/string_util.h" | 16 #include "base/strings/string_util.h" |
| 17 #include "base/strings/stringprintf.h" | 17 #include "base/strings/stringprintf.h" |
| 18 #include "chrome/common/url_constants.h" | 18 #include "chrome/common/url_constants.h" |
| 19 #include "components/google/core/browser/google_util.h" | 19 #include "components/google/core/browser/google_util.h" |
| 20 #include "net/cookies/cookie_util.h" | 20 #include "net/cookies/cookie_util.h" |
| 21 #include "net/http/http_request_headers.h" | 21 #include "net/http/http_request_headers.h" |
| 22 #include "net/url_request/url_request.h" | 22 #include "net/url_request/url_request.h" |
| 23 #include "url/gurl.h" | 23 #include "url/gurl.h" |
| 24 | 24 |
| 25 namespace { | 25 namespace { |
| 26 | 26 |
| 27 int g_force_google_safe_search_count_for_test = 0; | 27 int g_force_google_safe_search_count_for_test = 0; |
| 28 int g_force_youtube_restrict_count_for_test = 0; | 28 int g_force_youtube_safety_mode_count_for_test = 0; |
| 29 | 29 |
| 30 const char kYouTubeRestrictHeaderName[] = "YouTube-Restrict"; | 30 const char kYouTubeSafetyModeHeaderName[] = "YouTube-Safety-Mode"; |
| 31 const char kYouTubeRestrictHeaderValueModerate[] = "Moderate"; | 31 const char kYouTubeSafetyModeHeaderValue[] = "Active"; |
| 32 const char kYouTubeRestrictHeaderValueStrict[] = "Strict"; | |
| 33 | 32 |
| 34 // Returns whether a URL parameter, |first_parameter| (e.g. foo=bar), has the | 33 // Returns whether a URL parameter, |first_parameter| (e.g. foo=bar), has the |
| 35 // same key as the the |second_parameter| (e.g. foo=baz). Both parameters | 34 // same key as the the |second_parameter| (e.g. foo=baz). Both parameters |
| 36 // must be in key=value form. | 35 // must be in key=value form. |
| 37 bool HasSameParameterKey(const std::string& first_parameter, | 36 bool HasSameParameterKey(const std::string& first_parameter, |
| 38 const std::string& second_parameter) { | 37 const std::string& second_parameter) { |
| 39 DCHECK(second_parameter.find("=") != std::string::npos); | 38 DCHECK(second_parameter.find("=") != std::string::npos); |
| 40 // Prefix for "foo=bar" is "foo=". | 39 // Prefix for "foo=bar" is "foo=". |
| 41 std::string parameter_prefix = second_parameter.substr( | 40 std::string parameter_prefix = second_parameter.substr( |
| 42 0, second_parameter.find("=") + 1); | 41 0, second_parameter.find("=") + 1); |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 std::string query = request->url().query(); | 81 std::string query = request->url().query(); |
| 83 std::string new_query = AddSafeSearchParameters(query); | 82 std::string new_query = AddSafeSearchParameters(query); |
| 84 if (query == new_query) | 83 if (query == new_query) |
| 85 return; | 84 return; |
| 86 | 85 |
| 87 GURL::Replacements replacements; | 86 GURL::Replacements replacements; |
| 88 replacements.SetQueryStr(new_query); | 87 replacements.SetQueryStr(new_query); |
| 89 *new_url = request->url().ReplaceComponents(replacements); | 88 *new_url = request->url().ReplaceComponents(replacements); |
| 90 } | 89 } |
| 91 | 90 |
| 92 void ForceYouTubeRestrict(const net::URLRequest* request, | 91 // If |request| is a request to YouTube, enforces YouTube's Safety Mode by |
| 93 net::HttpRequestHeaders* headers, | 92 // setting YouTube's Safety Mode header. |
| 94 YouTubeRestrictMode mode) { | 93 void ForceYouTubeSafetyMode(const net::URLRequest* request, |
| 95 ++g_force_youtube_restrict_count_for_test; | 94 net::HttpRequestHeaders* headers) { |
| 95 ++g_force_youtube_safety_mode_count_for_test; |
| 96 | 96 |
| 97 if (!google_util::IsYoutubeDomainUrl( | 97 if (!google_util::IsYoutubeDomainUrl( |
| 98 request->url(), | 98 request->url(), |
| 99 google_util::ALLOW_SUBDOMAIN, | 99 google_util::ALLOW_SUBDOMAIN, |
| 100 google_util::DISALLOW_NON_STANDARD_PORTS)) | 100 google_util::DISALLOW_NON_STANDARD_PORTS)) |
| 101 return; | 101 return; |
| 102 | 102 |
| 103 switch (mode) { | 103 headers->SetHeader(kYouTubeSafetyModeHeaderName, |
| 104 case YOUTUBE_RESTRICT_OFF: | 104 kYouTubeSafetyModeHeaderValue); |
| 105 case YOUTUBE_RESTRICT_COUNT: | |
| 106 NOTREACHED(); | |
| 107 break; | |
| 108 | |
| 109 case YOUTUBE_RESTRICT_MODERATE: | |
| 110 headers->SetHeader(kYouTubeRestrictHeaderName, | |
| 111 kYouTubeRestrictHeaderValueModerate); | |
| 112 break; | |
| 113 | |
| 114 case YOUTUBE_RESTRICT_STRICT: | |
| 115 headers->SetHeader(kYouTubeRestrictHeaderName, | |
| 116 kYouTubeRestrictHeaderValueStrict); | |
| 117 break; | |
| 118 } | |
| 119 } | 105 } |
| 120 | 106 |
| 121 int GetForceGoogleSafeSearchCountForTesting() { | 107 int GetForceGoogleSafeSearchCountForTesting() { |
| 122 return g_force_google_safe_search_count_for_test; | 108 return g_force_google_safe_search_count_for_test; |
| 123 } | 109 } |
| 124 | 110 |
| 125 int GetForceYouTubeRestrictCountForTesting() { | 111 int GetForceYouTubeSafetyModeCountForTesting() { |
| 126 return g_force_youtube_restrict_count_for_test; | 112 return g_force_youtube_safety_mode_count_for_test; |
| 127 } | 113 } |
| 128 | 114 |
| 129 void ClearForceGoogleSafeSearchCountForTesting() { | 115 void ClearForceGoogleSafeSearchCountForTesting() { |
| 130 g_force_google_safe_search_count_for_test = 0; | 116 g_force_google_safe_search_count_for_test = 0; |
| 131 } | 117 } |
| 132 | 118 |
| 133 void ClearForceYouTubeRestrictCountForTesting() { | 119 void ClearForceYouTubeSafetyModeCountForTesting() { |
| 134 g_force_youtube_restrict_count_for_test = 0; | 120 g_force_youtube_safety_mode_count_for_test = 0; |
| 135 } | 121 } |
| 136 | 122 |
| 137 } // namespace safe_search_util | 123 } // namespace safe_search_util |
| OLD | NEW |