hpack_round_trip_test.cc revision 5c02ac1a9c1b504631c0a3d2b6e737b5d738bae1
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 EXPECT_TRUE(RoundTrip(headers)); 85 } 86} 87 88TEST_F(HpackRoundTripTest, RequestFixtures) { 89 { 90 map<string, string> headers; 91 headers[":authority"] = "www.example.com"; 92 headers[":method"] = "GET"; 93 headers[":path"] = "/"; 94 headers[":scheme"] = "http"; 95 headers["cookie"] = "baz=bing; foo=bar"; 96 EXPECT_TRUE(RoundTrip(headers)); 97 } 98 { 99 map<string, string> headers; 100 headers[":authority"] = "www.example.com"; 101 headers[":method"] = "GET"; 102 headers[":path"] = "/"; 103 headers[":scheme"] = "http"; 104 headers["cache-control"] = "no-cache"; 105 headers["cookie"] = "fizzle=fazzle; foo=bar"; 106 EXPECT_TRUE(RoundTrip(headers)); 107 } 108 { 109 map<string, string> headers; 110 headers[":authority"] = "www.example.com"; 111 headers[":method"] = "GET"; 112 headers[":path"] = "/index.html"; 113 headers[":scheme"] = "https"; 114 headers["custom-key"] = "custom-value"; 115 headers["cookie"] = "baz=bing; fizzle=fazzle; garbage"; 116 EXPECT_TRUE(RoundTrip(headers)); 117 } 118} 119 120TEST_F(HpackRoundTripTest, RandomizedExamples) { 121 // Grow vectors of names & values, which are seeded with fixtures and then 122 // expanded with dynamically generated data. Samples are taken using the 123 // exponential distribution. 124 vector<string> names; 125 names.push_back(":authority"); 126 names.push_back(":path"); 127 names.push_back(":status"); 128 // TODO(jgraettinger): Enable "cookie" as a name fixture. Crumbs may be 129 // reconstructed in any order, which breaks the simple validation used here. 130 131 vector<string> values; 132 values.push_back("/"); 133 values.push_back("/index.html"); 134 values.push_back("200"); 135 values.push_back("404"); 136 values.push_back(""); 137 values.push_back("baz=bing; foo=bar; garbage"); 138 values.push_back("baz=bing; fizzle=fazzle; garbage"); 139 140 int seed = std::time(NULL); 141 LOG(INFO) << "Seeding with srand(" << seed << ")"; 142 srand(seed); 143 144 for (size_t i = 0; i != 2000; ++i) { 145 map<string, string> headers; 146 147 size_t header_count = 1 + SampleExponential(7, 50); 148 for (size_t j = 0; j != header_count; ++j) { 149 size_t name_index = SampleExponential(20, 200); 150 size_t value_index = SampleExponential(20, 200); 151 152 string name, value; 153 if (name_index >= names.size()) { 154 names.push_back(base::RandBytesAsString(1 + SampleExponential(5, 30))); 155 name = names.back(); 156 } else { 157 name = names[name_index]; 158 } 159 if (value_index >= values.size()) { 160 values.push_back(base::RandBytesAsString( 161 1 + SampleExponential(15, 75))); 162 value = values.back(); 163 } else { 164 value = values[value_index]; 165 } 166 headers[name] = value; 167 } 168 EXPECT_TRUE(RoundTrip(headers)); 169 } 170} 171 172} // namespace 173 174} // namespace net 175