wjqserver/touka
1
0
Fork 0
mirror of https://github.com/infinite-iroha/touka.git synced 2026-06-13 15:47:38 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

fix(maxreader): treat non-positive limits as unlimited

Browse source
This commit is contained in:
wjqserver 2026-04-01 00:03:23 +08:00
parent 91c50536c4
commit e6ff0fa6b9
2 changed files with 10 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

27
context_bodylimit_test.go
View file

@ -27,16 +27,6 @@ func (r *zeroNilThenEOFReader) Close() error {
return nil
}
type zeroNilForeverReader struct{}
func (r *zeroNilForeverReader) Read(_ []byte) (int, error) {
return 0, nil
}
func (r *zeroNilForeverReader) Close() error {
return nil
}
func TestFileTextUsesProvidedStatusCode(t *testing.T) {
t.Helper()
@ -107,21 +97,18 @@ func TestMaxBytesReaderAllowsZeroNilThenEOFAtExactLimit(t *testing.T) {
}
}
func TestMaxBytesReaderRejectsOverLimitWithoutProbeLoop(t *testing.T) {
func TestMaxBytesReaderTreatsZeroLimitAsUnlimited(t *testing.T) {
t.Helper()
reader := NewMaxBytesReader(&zeroNilForeverReader{}, 0)
reader := NewMaxBytesReader(io.NopCloser(strings.NewReader("abc")), 0)
defer reader.Close()
buf := make([]byte, 1)
n, err := reader.Read(buf)
if n != 0 || err != nil {
t.Fatalf("expected initial zero,nil read result, got n=%d err=%v", n, err)
data, err := io.ReadAll(reader)
if err != nil {
t.Fatalf("expected zero limit to leave body unlimited, got %v", err)
}
n, err = reader.Read(buf)
if n != 0 || !errors.Is(err, ErrBodyTooLarge) {
t.Fatalf("expected ErrBodyTooLarge after repeated zero,nil reads, got n=%d err=%v", n, err)
if string(data) != "abc" {
t.Fatalf("unexpected data: %q", string(data))
}
}