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

net/quic/core/quic_header_list_test.cc

Index: net/quic/core/quic_header_list_test.cc
diff --git a/net/quic/core/quic_header_list_test.cc b/net/quic/core/quic_header_list_test.cc
index 4c91c418b2c52ab670153db47cbfee4586b21652..cef1d25e41c232f48153dc93d1240761999cd5e8 100644
--- a/net/quic/core/quic_header_list_test.cc
+++ b/net/quic/core/quic_header_list_test.cc
@@ -4,9 +4,12 @@
 
 #include "net/quic/core/quic_header_list.h"
 
+#include "net/quic/core/quic_flags.h"
 #include "net/test/gtest_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
+using std::string;
+
 namespace net {
 
 // This test verifies that QuicHeaderList accumulates header pairs in order.
@@ -19,6 +22,28 @@ TEST(QuicHeaderListTest, OnHeader) {
   EXPECT_EQ("{ foo=bar, april=fools, beep=, }", headers.DebugString());
 }
 
+TEST(QuicHeaderListTest, TooLarge) {
+  FLAGS_quic_limit_uncompressed_headers = true;
+  QuicHeaderList headers;
+  string key = "key";
+  string value(1 << 18, '1');
+  headers.OnHeader(key, value);
+  headers.OnHeaderBlockEnd(key.size() + value.size());
+  EXPECT_TRUE(headers.empty());
+
+  EXPECT_EQ("{ }", headers.DebugString());
+}
+
+TEST(QuicHeaderListTest, NotTooLarge) {
+  QuicHeaderList headers;
+  headers.set_max_uncompressed_header_bytes(1 << 20);
+  string key = "key";
+  string value(1 << 18, '1');
+  headers.OnHeader(key, value);
+  headers.OnHeaderBlockEnd(key.size() + value.size());
+  EXPECT_FALSE(headers.empty());
+}
+
 // This test verifies that QuicHeaderList is copyable and assignable.
 TEST(QuicHeaderListTest, IsCopyableAndAssignable) {
   QuicHeaderList headers;