Limit binary data to 21MB

This commit is contained in:
Jae Kwon
2015-07-07 18:35:21 -07:00
parent 5734496819
commit 1f34236948
5 changed files with 44 additions and 10 deletions

View File

@ -14,6 +14,11 @@ func ReadString(r io.Reader, n *int64, err *error) string {
if *err != nil {
return ""
}
if MaxBinaryReadSize < *n+int64(length) {
*err = ErrMaxBinaryReadSizeReached
return ""
}
buf := make([]byte, length)
ReadFull(buf, r, n, err)
return string(buf)