MM-43716: enforce size limits in slack importer (#20144)

Automatic Merge
Этот коммит содержится в:
Ashish Bhate
2022-05-12 18:39:55 +05:30
коммит произвёл GitHub
родитель 87a719b4b4
Коммит 3c625743e5
3 изменённых файлов: 119 добавлений и 11 удалений

Просмотреть файл

@@ -4,6 +4,7 @@
package utils
import (
"errors"
"fmt"
"io"
"io/ioutil"
@@ -112,3 +113,23 @@ func CopyDir(src string, dst string) (err error) {
return
}
var SizeLimitExceeded = errors.New("Size limit exceeded")
type LimitedReaderWithError struct {
limitedReader *io.LimitedReader
}
func NewLimitedReaderWithError(reader io.Reader, maxBytes int64) *LimitedReaderWithError {
return &LimitedReaderWithError{
limitedReader: &io.LimitedReader{R: reader, N: maxBytes + 1},
}
}
func (l *LimitedReaderWithError) Read(p []byte) (int, error) {
n, err := l.limitedReader.Read(p)
if l.limitedReader.N <= 0 && err == io.EOF {
return n, SizeLimitExceeded
}
return n, err
}