| 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 // Make sure stdint.h includes SIZE_MAX. (See C89, p259, footnote 221.) | 5 // Make sure stdint.h includes SIZE_MAX. (See C89, p259, footnote 221.) |
| 6 #ifndef __STDC_LIMIT_MACROS | 6 #ifndef __STDC_LIMIT_MACROS |
| 7 #define __STDC_LIMIT_MACROS 1 | 7 #define __STDC_LIMIT_MACROS 1 |
| 8 #endif | 8 #endif |
| 9 | 9 |
| 10 #include "components/domain_reliability/config.h" | 10 #include "components/domain_reliability/config.h" |
| 11 | 11 |
| 12 #include <stdint.h> | 12 #include <stdint.h> |
| 13 | 13 |
| 14 #include "base/json/json_reader.h" | 14 #include "base/json/json_reader.h" |
| 15 #include "base/json/json_value_converter.h" | 15 #include "base/json/json_value_converter.h" |
| 16 #include "base/profiler/scoped_tracker.h" | 16 #include "base/profiler/scoped_tracker.h" |
| 17 #include "base/rand_util.h" | 17 #include "base/rand_util.h" |
| 18 #include "base/strings/pattern.h" | 18 #include "base/strings/pattern.h" |
| 19 #include "base/strings/string_util.h" | 19 #include "base/strings/string_util.h" |
| 20 | 20 |
| 21 namespace { | 21 namespace { |
| 22 | 22 |
| 23 bool ConvertURL(const base::StringPiece& string_piece, GURL* url) { | 23 bool ConvertURL(const base::Value* value, GURL* url) { |
| 24 *url = GURL(string_piece.as_string()); | 24 std::string url_string; |
| 25 if (!value->GetAsString(&url_string)) |
| 26 return false; |
| 27 *url = GURL(url_string); |
| 25 return url->is_valid(); | 28 return url->is_valid(); |
| 26 } | 29 } |
| 27 | 30 |
| 28 bool IsValidSampleRate(double p) { return p >= 0.0 && p <= 1.0; } | 31 bool ConvertOrigin(const base::Value* value, GURL* url) { |
| 32 return ConvertURL(value, url) && !url->has_username() && |
| 33 !url->has_password() && url->SchemeIs("https") && |
| 34 url->path_piece() == "/" && !url->has_query() && !url->has_ref(); |
| 35 } |
| 36 |
| 37 bool IsValidSampleRate(double p) { |
| 38 return p >= 0.0 && p <= 1.0; |
| 39 } |
| 29 | 40 |
| 30 } // namespace | 41 } // namespace |
| 31 | 42 |
| 32 namespace domain_reliability { | 43 namespace domain_reliability { |
| 33 | 44 |
| 34 // static | 45 DomainReliabilityConfig::DomainReliabilityConfig() |
| 35 const size_t DomainReliabilityConfig::kInvalidResourceIndex = SIZE_MAX; | 46 : include_subdomains(false), |
| 36 | 47 success_sample_rate(-1.0), |
| 37 DomainReliabilityConfig::Resource::Resource() { | 48 failure_sample_rate(-1.0) { |
| 38 } | 49 } |
| 39 DomainReliabilityConfig::Resource::~Resource() {} | |
| 40 | |
| 41 bool DomainReliabilityConfig::Resource::MatchesUrl(const GURL& url) const { | |
| 42 const std::string& spec = url.spec(); | |
| 43 | |
| 44 for (const auto& url_pattern : url_patterns) { | |
| 45 if (base::MatchPattern(spec, *url_pattern)) | |
| 46 return true; | |
| 47 } | |
| 48 | |
| 49 return false; | |
| 50 } | |
| 51 | |
| 52 bool DomainReliabilityConfig::Resource::DecideIfShouldReportRequest( | |
| 53 bool success) const { | |
| 54 double sample_rate = success ? success_sample_rate : failure_sample_rate; | |
| 55 DCHECK(IsValidSampleRate(sample_rate)); | |
| 56 return base::RandDouble() < sample_rate; | |
| 57 } | |
| 58 | |
| 59 // static | |
| 60 void DomainReliabilityConfig::Resource::RegisterJSONConverter( | |
| 61 base::JSONValueConverter<DomainReliabilityConfig::Resource>* converter) { | |
| 62 converter->RegisterStringField("resource_name", &Resource::name); | |
| 63 converter->RegisterRepeatedString("url_patterns", &Resource::url_patterns); | |
| 64 converter->RegisterDoubleField("success_sample_rate", | |
| 65 &Resource::success_sample_rate); | |
| 66 converter->RegisterDoubleField("failure_sample_rate", | |
| 67 &Resource::failure_sample_rate); | |
| 68 } | |
| 69 | |
| 70 bool DomainReliabilityConfig::Resource::IsValid() const { | |
| 71 return !name.empty() && !url_patterns.empty() && | |
| 72 IsValidSampleRate(success_sample_rate) && | |
| 73 IsValidSampleRate(failure_sample_rate); | |
| 74 } | |
| 75 | |
| 76 DomainReliabilityConfig::Collector::Collector() {} | |
| 77 DomainReliabilityConfig::Collector::~Collector() {} | |
| 78 | |
| 79 // static | |
| 80 void DomainReliabilityConfig::Collector::RegisterJSONConverter( | |
| 81 base::JSONValueConverter<DomainReliabilityConfig::Collector>* converter) { | |
| 82 converter->RegisterCustomField<GURL>("upload_url", &Collector::upload_url, | |
| 83 &ConvertURL); | |
| 84 } | |
| 85 | |
| 86 bool DomainReliabilityConfig::Collector::IsValid() const { | |
| 87 return upload_url.is_valid(); | |
| 88 } | |
| 89 | |
| 90 DomainReliabilityConfig::DomainReliabilityConfig() {} | |
| 91 DomainReliabilityConfig::~DomainReliabilityConfig() {} | 50 DomainReliabilityConfig::~DomainReliabilityConfig() {} |
| 92 | 51 |
| 93 // static | 52 // static |
| 94 scoped_ptr<const DomainReliabilityConfig> DomainReliabilityConfig::FromJSON( | 53 scoped_ptr<const DomainReliabilityConfig> DomainReliabilityConfig::FromJSON( |
| 95 const base::StringPiece& json) { | 54 const base::StringPiece& json) { |
| 96 scoped_ptr<base::Value> value = base::JSONReader::Read(json); | 55 scoped_ptr<base::Value> value = base::JSONReader::Read(json); |
| 97 base::JSONValueConverter<DomainReliabilityConfig> converter; | 56 base::JSONValueConverter<DomainReliabilityConfig> converter; |
| 98 scoped_ptr<DomainReliabilityConfig> config(new DomainReliabilityConfig()); | 57 scoped_ptr<DomainReliabilityConfig> config(new DomainReliabilityConfig()); |
| 99 | 58 |
| 100 // If we can parse and convert the JSON into a valid config, return that. | 59 // If we can parse and convert the JSON into a valid config, return that. |
| 101 if (value && converter.Convert(*value, config.get()) && config->IsValid()) | 60 if (value && converter.Convert(*value, config.get()) && config->IsValid()) |
| 102 return config.Pass(); | 61 return config.Pass(); |
| 103 return scoped_ptr<const DomainReliabilityConfig>(); | 62 return scoped_ptr<const DomainReliabilityConfig>(); |
| 104 } | 63 } |
| 105 | 64 |
| 106 bool DomainReliabilityConfig::IsValid() const { | 65 bool DomainReliabilityConfig::IsValid() const { |
| 107 if (domain.empty() || resources.empty() || collectors.empty()) | 66 if (!origin.is_valid() || collectors.empty() || |
| 67 !IsValidSampleRate(success_sample_rate) || |
| 68 !IsValidSampleRate(failure_sample_rate)) { |
| 108 return false; | 69 return false; |
| 109 | |
| 110 for (auto& resource : resources) { | |
| 111 if (!resource->IsValid()) | |
| 112 return false; | |
| 113 } | 70 } |
| 114 | 71 |
| 115 for (auto& collector : collectors) { | 72 for (const auto& url : collectors) { |
| 116 if (!collector->IsValid()) | 73 if (!url->is_valid()) |
| 117 return false; | 74 return false; |
| 118 } | 75 } |
| 119 | 76 |
| 120 return true; | 77 return true; |
| 121 } | 78 } |
| 122 | 79 |
| 123 size_t DomainReliabilityConfig::GetResourceIndexForUrl(const GURL& url) const { | 80 bool DomainReliabilityConfig::DecideIfShouldReportRequest(bool success) const { |
| 124 // Removes username, password, and fragment. | 81 double sample_rate = success ? success_sample_rate : failure_sample_rate; |
| 125 GURL sanitized_url = url.GetAsReferrer(); | 82 return base::RandDouble() < sample_rate; |
| 126 | |
| 127 for (size_t i = 0; i < resources.size(); ++i) { | |
| 128 if (resources[i]->MatchesUrl(sanitized_url)) | |
| 129 return i; | |
| 130 } | |
| 131 | |
| 132 return kInvalidResourceIndex; | |
| 133 } | 83 } |
| 134 | 84 |
| 135 // static | 85 // static |
| 136 void DomainReliabilityConfig::RegisterJSONConverter( | 86 void DomainReliabilityConfig::RegisterJSONConverter( |
| 137 base::JSONValueConverter<DomainReliabilityConfig>* converter) { | 87 base::JSONValueConverter<DomainReliabilityConfig>* converter) { |
| 138 converter->RegisterStringField("config_version", | 88 converter->RegisterCustomValueField<GURL>( |
| 139 &DomainReliabilityConfig::version); | 89 "origin", &DomainReliabilityConfig::origin, &ConvertOrigin); |
| 140 converter->RegisterStringField("monitored_domain", | 90 converter->RegisterBoolField("include_subdomains", |
| 141 &DomainReliabilityConfig::domain); | 91 &DomainReliabilityConfig::include_subdomains); |
| 142 converter->RegisterRepeatedMessage("monitored_resources", | 92 converter->RegisterRepeatedCustomValue( |
| 143 &DomainReliabilityConfig::resources); | 93 "collectors", &DomainReliabilityConfig::collectors, &ConvertURL); |
| 144 converter->RegisterRepeatedMessage("collectors", | 94 converter->RegisterRepeatedString("path_prefixes", |
| 145 &DomainReliabilityConfig::collectors); | 95 &DomainReliabilityConfig::path_prefixes); |
| 96 converter->RegisterDoubleField("success_sample_rate", |
| 97 &DomainReliabilityConfig::success_sample_rate); |
| 98 converter->RegisterDoubleField("failure_sample_rate", |
| 99 &DomainReliabilityConfig::failure_sample_rate); |
| 146 } | 100 } |
| 147 | 101 |
| 148 } // namespace domain_reliability | 102 } // namespace domain_reliability |
| OLD | NEW |