OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 #include "net/spdy/hpack_huffman_aggregator.h" | |
5 | |
6 #include "base/metrics/bucket_ranges.h" | |
7 #include "base/metrics/field_trial.h" | |
8 #include "base/metrics/histogram.h" | |
9 #include "base/metrics/sample_vector.h" | |
10 #include "base/stl_util.h" | |
11 #include "base/strings/string_number_conversions.h" | |
12 #include "base/strings/string_util.h" | |
13 #include "net/base/load_flags.h" | |
14 #include "net/http/http_request_headers.h" | |
15 #include "net/http/http_request_info.h" | |
16 #include "net/http/http_response_headers.h" | |
17 #include "net/spdy/hpack_encoder.h" | |
18 #include "net/spdy/spdy_http_utils.h" | |
19 | |
20 namespace net { | |
21 | |
22 namespace { | |
23 | |
24 const char kHistogramName[] = "Net.SpdyHpackEncodedCharacterFrequency"; | |
25 | |
26 const size_t kTotalCountsPublishThreshold = 50000; | |
27 | |
28 // Each encoder uses the default dynamic table size of 4096 total bytes. | |
29 const size_t kMaxEncoders = 20; | |
30 | |
31 } // namespace | |
32 | |
33 HpackHuffmanAggregator::HpackHuffmanAggregator() | |
34 : counts_(256, 0), | |
35 total_counts_(0), | |
36 max_encoders_(kMaxEncoders) { | |
37 } | |
38 | |
39 HpackHuffmanAggregator::~HpackHuffmanAggregator() { | |
40 STLDeleteContainerPairSecondPointers(encoders_.begin(), encoders_.end()); | |
41 encoders_.clear(); | |
42 } | |
43 | |
44 void HpackHuffmanAggregator::AggregateTransactionCharacterCounts( | |
45 const HttpRequestInfo& request, | |
46 const HttpRequestHeaders& request_headers, | |
47 const ProxyServer& proxy, | |
48 const HttpResponseHeaders& response_headers) { | |
49 if (IsCrossOrigin(request)) { | |
50 return; | |
51 } | |
52 HpackEncoder* encoder = ObtainEncoder(SpdySessionKey( | |
53 HostPortPair::FromURL(request.url), proxy, request.privacy_mode)); | |
54 | |
55 // Convert and encode the request and response header sets. | |
56 { | |
57 SpdyHeaderBlock headers; | |
58 CreateSpdyHeadersFromHttpRequest( | |
59 request, request_headers, SPDY4, false, &headers); | |
60 | |
61 std::string tmp_out; | |
62 encoder->EncodeHeaderSet(headers, &tmp_out); | |
63 } | |
64 { | |
65 SpdyHeaderBlock headers; | |
66 CreateSpdyHeadersFromHttpResponse(response_headers, &headers); | |
67 | |
68 std::string tmp_out; | |
69 encoder->EncodeHeaderSet(headers, &tmp_out); | |
70 } | |
71 if (total_counts_ >= kTotalCountsPublishThreshold) { | |
72 PublishCounts(); | |
73 } | |
74 } | |
75 | |
76 // static | |
77 bool HpackHuffmanAggregator::UseAggregator() { | |
78 const std::string group_name = | |
79 base::FieldTrialList::FindFullName("HpackHuffmanAggregator"); | |
80 if (group_name == "Enabled") { | |
81 return true; | |
82 } | |
83 return false; | |
84 } | |
85 | |
86 // static | |
87 void HpackHuffmanAggregator::CreateSpdyHeadersFromHttpResponse( | |
88 const HttpResponseHeaders& headers, | |
89 SpdyHeaderBlock* headers_out) { | |
90 // Lower-case header names, and coalesce multiple values delimited by \0. | |
91 // Also add the fixed status header. | |
92 std::string name, value; | |
93 void* it = NULL; | |
94 while (headers.EnumerateHeaderLines(&it, &name, &value)) { | |
95 base::StringToLowerASCII(&name); | |
96 if (headers_out->find(name) == headers_out->end()) { | |
97 (*headers_out)[name] = value; | |
98 } else { | |
99 (*headers_out)[name] += std::string(1, '\0') + value; | |
100 } | |
101 } | |
102 (*headers_out)[":status"] = base::IntToString(headers.response_code()); | |
103 } | |
104 | |
105 // static | |
106 bool HpackHuffmanAggregator::IsCrossOrigin(const HttpRequestInfo& request) { | |
107 // Require that the request is top-level, or that it shares | |
108 // an origin with its referer. | |
109 if ((request.load_flags & LOAD_MAIN_FRAME) == 0) { | |
110 std::string referer_str; | |
111 if (!request.extra_headers.GetHeader(HttpRequestHeaders::kReferer, | |
112 &referer_str)) { | |
113 // Require a referer. | |
114 return true; | |
115 } | |
116 GURL referer(referer_str); | |
117 if (!HostPortPair::FromURL(request.url).Equals( | |
118 HostPortPair::FromURL(referer))) { | |
119 // Cross-origin request. | |
120 return true; | |
121 } | |
122 } | |
123 return false; | |
124 } | |
125 | |
126 HpackEncoder* HpackHuffmanAggregator::ObtainEncoder(const SpdySessionKey& key) { | |
127 for (OriginEncoders::iterator it = encoders_.begin(); | |
128 it != encoders_.end(); ++it) { | |
129 if (key.Equals(it->first)) { | |
130 // Move to head of list and return. | |
131 OriginEncoder origin_encoder = *it; | |
132 encoders_.erase(it); | |
133 encoders_.push_front(origin_encoder); | |
134 return origin_encoder.second; | |
135 } | |
136 } | |
137 // Not found. Create a new encoder, evicting one if needed. | |
138 encoders_.push_front(std::make_pair( | |
139 key, new HpackEncoder(ObtainHpackHuffmanTable()))); | |
140 if (encoders_.size() > max_encoders_) { | |
141 delete encoders_.back().second; | |
142 encoders_.pop_back(); | |
143 } | |
144 encoders_.front().second->SetCharCountsStorage(&counts_, &total_counts_); | |
145 return encoders_.front().second; | |
146 } | |
147 | |
148 void HpackHuffmanAggregator::PublishCounts() { | |
149 // base::Histogram requires that values be 1-indexed. | |
150 const size_t kRangeMin = 1; | |
151 const size_t kRangeMax = counts_.size() + 1; | |
152 const size_t kBucketCount = kRangeMax + 1; | |
153 | |
154 base::BucketRanges ranges(kBucketCount + 1); | |
155 for (size_t i = 0; i != ranges.size(); ++i) { | |
156 ranges.set_range(i, i); | |
157 } | |
158 ranges.ResetChecksum(); | |
159 | |
160 // Copy |counts_| into a SampleVector. | |
161 base::SampleVector samples(&ranges); | |
162 for (size_t i = 0; i != counts_.size(); ++i) { | |
163 samples.Accumulate(i + 1, counts_[i]); | |
164 } | |
165 | |
166 STATIC_HISTOGRAM_POINTER_BLOCK( | |
167 kHistogramName, | |
168 AddSamples(samples), | |
169 base::LinearHistogram::FactoryGet( | |
170 kHistogramName, kRangeMin, kRangeMax, kBucketCount, | |
171 base::HistogramBase::kUmaTargetedHistogramFlag)); | |
172 | |
173 // Clear counts. | |
174 counts_.assign(counts_.size(), 0); | |
175 total_counts_ = 0; | |
176 } | |
177 | |
178 } // namespace net | |
OLD | NEW |