add test

Author: joybestourous

test/end2end_test.go

@@ -6090,6 +6090,49 @@ func testClientMaxHeaderListSizeServerIntentionalViolation(t *testing.T, e env)
 	}
 }
 
+// TestEarlyAbortStreamHeaderListSizeCheck tests that when the server decides to
+// abort the stream early (e.g., due to invalid content-type), it respects the
+// client's MaxHeaderListSize setting. If the response headers would exceed
+// the limit, the server should send a RST_STREAM instead of the headers.
+func (s) TestEarlyAbortStreamHeaderListSizeCheck(t *testing.T) {
+	lis, err := net.Listen("tcp", "localhost:0")
+	if err != nil {
+		t.Fatalf("Failed to listen: %v", err)
+	}
+	s := grpc.NewServer()
+	defer s.Stop()
+	go s.Serve(lis)
+
+	conn, err := net.DialTimeout("tcp", lis.Addr().String(), defaultTestTimeout)
+	if err != nil {
+		t.Fatalf("Failed to dial: %v", err)
+	}
+	defer conn.Close()
+	st := newServerTesterFromConn(t, conn)
+
+	// Set a very small MaxHeaderListSize that any response headers would violate.
+	// The early abort response includes :status, content-type, grpc-status, and grpc-message,
+	// which together will exceed 1 byte.
+	st.greetWithSettings(http2.Setting{ID: http2.SettingMaxHeaderListSize, Val: 1})
+
+	// Send a request with an invalid content-type to trigger early abort.
+	st.writeHeaders(http2.HeadersFrameParam{
+		StreamID: 1,
+		BlockFragment: st.encodeHeader(
+			":method", "POST",
+			":path", "/grpc.testing.TestService/UnaryCall",
+			"content-type", "text/plain", // Invalid content-type to trigger early abort
+			"te", "trailers",
+		),
+		EndStream:  true,
+		EndHeaders: true,
+	})
+
+	// We should receive a RST_STREAM with ErrCodeInternal because the response
+	// headers exceed the MaxHeaderListSize limit.
+	st.wantRSTStream(http2.ErrCodeInternal)
+}
+
 func (s) TestNetPipeConn(t *testing.T) {
 	// This test will block indefinitely if grpc writes both client and server
 	// prefaces without either reading from the Conn.

test/servertester.go

@@ -91,8 +91,19 @@ func (st *serverTester) readFrame() (http2.Frame, error) {
 // greet initiates the client's HTTP/2 connection into a state where
 // frames may be sent.
 func (st *serverTester) greet() {
+	st.greetWithSettings()
+}
+
+// greetWithSettings initiates the client's HTTP/2 connection with custom settings.
+func (st *serverTester) greetWithSettings(settings ...http2.Setting) {
 	st.writePreface()
-	st.writeInitialSettings()
+	if len(settings) > 0 {
+		if err := st.fr.WriteSettings(settings...); err != nil {
+			st.t.Fatalf("Error writing initial SETTINGS frame from client to server: %v", err)
+		}
+	} else {
+		st.writeInitialSettings()
+	}
 	st.wantSettings()
 	st.writeSettingsAck()
 	for {
@@ -132,6 +143,12 @@ func (st *serverTester) writeInitialSettings() {
 	}
 }
 
+func (st *serverTester) writeInitialSettingsWithMaxHeaderListSize(maxHeaderListSize uint32) {
+	if err := st.fr.WriteSettings(http2.Setting{ID: http2.SettingMaxHeaderListSize, Val: maxHeaderListSize}); err != nil {
+		st.t.Fatalf("Error writing initial SETTINGS frame from client to server: %v", err)
+	}
+}
+
 func (st *serverTester) writeSettingsAck() {
 	if err := st.fr.WriteSettingsAck(); err != nil {
 		st.t.Fatalf("Error writing ACK of server's SETTINGS: %v", err)