Limit QUIC uncompressed header size per stream to 16K, by default. Protected by FLAGS_quic_limit_un…

net/tools/quic/end_to_end_test.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stddef.h>
 #include <sys/epoll.h>
 
 #include <list>
 #include <memory>
 #include <string>
@@ skipping 1162 matching lines @@
   // Force the client to write with a stream ID belonging to a nonexistent
   // server-side stream.
   QuicSessionPeer::SetNextOutgoingStreamId(client_->client()->session(), 2);
 
   client_->SendCustomSynchronousRequest(request);
   // EXPECT_EQ(QUIC_STREAM_CONNECTION_ERROR, client_->stream_error());
   EXPECT_EQ(QUIC_STREAM_CONNECTION_ERROR, client_->stream_error());
   EXPECT_EQ(QUIC_INVALID_STREAM_ID, client_->connection_error());
 }
 
+// Test that if the the server will close the connection if the client attempts
+// to send a request with overly large headers.
+TEST_P(EndToEndTest, LargeHeaders) {
+  ASSERT_TRUE(Initialize());
+  client_->client()->WaitForCryptoHandshakeConfirmed();
+
+  string body;
+  test::GenerateBody(&body, kMaxPacketSize);
+
+  HTTPMessage request(HttpConstants::HTTP_1_1, HttpConstants::POST, "/foo");
+  request.AddHeader("key1", string(15 * 1024, 'a'));
+  request.AddHeader("key2", string(15 * 1024, 'a'));
+  request.AddHeader("key3", string(15 * 1024, 'a'));
+  request.AddBody(body, true);
+
+  client_->SendCustomSynchronousRequest(request);
+  if (FLAGS_quic_limit_uncompressed_headers) {
+    EXPECT_EQ(QUIC_HEADERS_TOO_LARGE, client_->stream_error());
+  } else {
+    EXPECT_EQ(QUIC_STREAM_NO_ERROR, client_->stream_error());
+    EXPECT_EQ(kFooResponseBody, client_->response_body());
+    EXPECT_EQ(200u, client_->response_headers()->parsed_response_code());
+  }
+  EXPECT_EQ(QUIC_NO_ERROR, client_->connection_error());
+}
+
 TEST_P(EndToEndTest, EarlyResponseWithQuicStreamNoError) {
   ASSERT_TRUE(Initialize());
   client_->client()->WaitForCryptoHandshakeConfirmed();
 
   string large_body;
   GenerateBody(&large_body, 1024 * 1024);
 
   HTTPMessage request(HttpConstants::HTTP_1_1, HttpConstants::POST, "/foo");
   request.AddBody(large_body, false);
 
@@ skipping 1736 matching lines @@
   client_->WaitForResponse();
   EXPECT_EQ(kBarResponseBody, client_->response_body());
   QuicConnectionStats client_stats =
       client_->client()->session()->connection()->GetStats();
   EXPECT_EQ(0u, client_stats.packets_lost);
   EXPECT_EQ(1, client_->client()->GetNumSentClientHellos());
 }
 }  // namespace
 }  // namespace test
 }  // namespace net