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 |