caddyhttp: Write header if needed in responseRecorder.WriteResponse (#…

…6380)
caddyserver · Jun 7, 2024 · 9be4f19 · 9be4f19
1 parent a10117f
commit 9be4f19
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/modules/caddyhttp/responsewriter.go b/modules/caddyhttp/responsewriter.go
@@ -219,13 +219,13 @@ func (rr *responseRecorder) Buffered() bool {
 }
 
 func (rr *responseRecorder) WriteResponse() error {
-	if rr.stream {
-		return nil
-	}
 	if rr.statusCode == 0 {
 		// could happen if no handlers actually wrote anything,
 		// and this prevents a panic; status must be > 0
-		rr.statusCode = http.StatusOK
+		rr.WriteHeader(http.StatusOK)
+	}
+	if rr.stream {
+		return nil
 	}
 	rr.ResponseWriterWrapper.WriteHeader(rr.statusCode)
 	_, err := io.Copy(rr.ResponseWriterWrapper, rr.buf)