mox/webapi/limitreader.go

package webapi

// similar between ../moxio/limitreader.go and ../webapi/limitreader.go

import (
	"errors"
	"io"
)

var errLimit = errors.New("input exceeds maximum size") // Returned by limitReader.

// limitReader reads up to Limit bytes, returning an error if more bytes are
// read. LimitReader can be used to enforce a maximum input length.
type limitReader struct {
	R     io.Reader
	Limit int64
}

// Read reads bytes from the underlying reader.
func (r *limitReader) Read(buf []byte) (int, error) {
	n, err := r.R.Read(buf)
	if n > 0 {
		r.Limit -= int64(n)
		if r.Limit < 0 {
			return 0, errLimit
		}
	}
	return n, err
}