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
5#include <cmath>
6#include <ctime>
7#include <map>
8#include <string>
9#include <vector>
10
11#include "base/rand_util.h"
12#include "net/spdy/hpack_constants.h"
13#include "net/spdy/hpack_decoder.h"
14#include "net/spdy/hpack_encoder.h"
15#include "testing/gtest/include/gtest/gtest.h"
16
17namespace net {
18
19using std::map;
20using std::string;
21using std::vector;
22
23namespace {
24
25class HpackRoundTripTest : public ::testing::Test {
26 protected:
27  HpackRoundTripTest()
28      : encoder_(ObtainHpackHuffmanTable()),
29        decoder_(ObtainHpackHuffmanTable()) {}
30
31  virtual void SetUp() {
32    // Use a small table size to tickle eviction handling.
33    encoder_.ApplyHeaderTableSizeSetting(256);
34    decoder_.ApplyHeaderTableSizeSetting(256);
35  }
36
37  bool RoundTrip(const map<string, string>& header_set) {
38    string encoded;
39    encoder_.EncodeHeaderSet(header_set, &encoded);
40
41    bool success = decoder_.HandleControlFrameHeadersData(
42        1, encoded.data(), encoded.size());
43    success &= decoder_.HandleControlFrameHeadersComplete(1);
44
45    EXPECT_EQ(header_set, decoder_.decoded_block());
46    return success;
47  }
48
49  size_t SampleExponential(size_t mean, size_t sanity_bound) {
50    return std::min<size_t>(-std::log(base::RandDouble()) * mean,
51                            sanity_bound);
52  }
53
54  HpackEncoder encoder_;
55  HpackDecoder decoder_;
56};
57
58TEST_F(HpackRoundTripTest, ResponseFixtures) {
59  {
60    map<string, string> headers;
61    headers[":status"] = "302";
62    headers["cache-control"] = "private";
63    headers["date"] = "Mon, 21 Oct 2013 20:13:21 GMT";
64    headers["location"] = "https://www.example.com";
65    EXPECT_TRUE(RoundTrip(headers));
66  }
67  {
68    map<string, string> headers;
69    headers[":status"] = "200";
70    headers["cache-control"] = "private";
71    headers["date"] = "Mon, 21 Oct 2013 20:13:21 GMT";
72    headers["location"] = "https://www.example.com";
73    EXPECT_TRUE(RoundTrip(headers));
74  }
75  {
76    map<string, string> headers;
77    headers[":status"] = "200";
78    headers["cache-control"] = "private";
79    headers["content-encoding"] = "gzip";
80    headers["date"] = "Mon, 21 Oct 2013 20:13:22 GMT";
81    headers["location"] = "https://www.example.com";
82    headers["set-cookie"] = "foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU;"
83        " max-age=3600; version=1";
84    headers["multivalue"] = string("foo\0bar", 7);
85    EXPECT_TRUE(RoundTrip(headers));
86  }
87}
88
89TEST_F(HpackRoundTripTest, RequestFixtures) {
90  {
91    map<string, string> headers;
92    headers[":authority"] = "www.example.com";
93    headers[":method"] = "GET";
94    headers[":path"] = "/";
95    headers[":scheme"] = "http";
96    headers["cookie"] = "baz=bing; foo=bar";
97    EXPECT_TRUE(RoundTrip(headers));
98  }
99  {
100    map<string, string> headers;
101    headers[":authority"] = "www.example.com";
102    headers[":method"] = "GET";
103    headers[":path"] = "/";
104    headers[":scheme"] = "http";
105    headers["cache-control"] = "no-cache";
106    headers["cookie"] = "foo=bar; fizzle=fazzle";
107    EXPECT_TRUE(RoundTrip(headers));
108  }
109  {
110    map<string, string> headers;
111    headers[":authority"] = "www.example.com";
112    headers[":method"] = "GET";
113    headers[":path"] = "/index.html";
114    headers[":scheme"] = "https";
115    headers["custom-key"] = "custom-value";
116    headers["cookie"] = "baz=bing; fizzle=fazzle; garbage";
117    headers["multivalue"] = string("foo\0bar", 7);
118    EXPECT_TRUE(RoundTrip(headers));
119  }
120}
121
122TEST_F(HpackRoundTripTest, RandomizedExamples) {
123  // Grow vectors of names & values, which are seeded with fixtures and then
124  // expanded with dynamically generated data. Samples are taken using the
125  // exponential distribution.
126  vector<string> names;
127  names.push_back(":authority");
128  names.push_back(":path");
129  names.push_back(":status");
130  // TODO(jgraettinger): Enable "cookie" as a name fixture. Crumbs may be
131  // reconstructed in any order, which breaks the simple validation used here.
132
133  vector<string> values;
134  values.push_back("/");
135  values.push_back("/index.html");
136  values.push_back("200");
137  values.push_back("404");
138  values.push_back("");
139  values.push_back("baz=bing; foo=bar; garbage");
140  values.push_back("baz=bing; fizzle=fazzle; garbage");
141
142  int seed = std::time(NULL);
143  LOG(INFO) << "Seeding with srand(" << seed << ")";
144  srand(seed);
145
146  for (size_t i = 0; i != 2000; ++i) {
147    map<string, string> headers;
148
149    size_t header_count = 1 + SampleExponential(7, 50);
150    for (size_t j = 0; j != header_count; ++j) {
151      size_t name_index = SampleExponential(20, 200);
152      size_t value_index = SampleExponential(20, 200);
153
154      string name, value;
155      if (name_index >= names.size()) {
156        names.push_back(base::RandBytesAsString(1 + SampleExponential(5, 30)));
157        name = names.back();
158      } else {
159        name = names[name_index];
160      }
161      if (value_index >= values.size()) {
162        string newvalue =
163            base::RandBytesAsString(1 + SampleExponential(15, 75));
164        // Currently order is not preserved in the encoder.  In particular,
165        // when a value is decomposed at \0 delimiters, its parts might get
166        // encoded out of order if some but not all of them already exist in
167        // the header table.  For now, avoid \0 bytes in values.
168        std::replace(newvalue.begin(), newvalue.end(), '\x00', '\x01');
169        values.push_back(newvalue);
170        value = values.back();
171      } else {
172        value = values[value_index];
173      }
174      headers[name] = value;
175    }
176    EXPECT_TRUE(RoundTrip(headers));
177  }
178}
179
180}  // namespace
181
182}  // namespace net
183