filters/diag/logbody: use correct calculation of bytes left to log (#…

…2837) Followup up on #2827 Signed-off-by: Alexander Yastrebov <[email protected]>
zalando · Jan 9, 2024 · b2bb841 · b2bb841
1 parent cec514c
commit b2bb841
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/filters/diag/logbody.go b/filters/diag/logbody.go
@@ -112,16 +112,12 @@ func newLogBodyStream(left int, f func([]byte), rc io.ReadCloser) io.ReadCloser
 }
 
 func (lb *logBodyStream) Read(p []byte) (n int, err error) {
-	if lb.left <= 0 {
-		return lb.input.Read(p)
-	}
-
 	n, err = lb.input.Read(p)
-	if n > 0 {
-		lb.f(p[:min(n, lb.left)])
+	if lb.left > 0 && n > 0 {
+		m := min(n, lb.left)
+		lb.f(p[:m])
+		lb.left -= m
 	}
-	lb.left -= n
-
 	return n, err
 }