2014-06-16 16:39:25 -07:00
|
|
|
package binary
|
|
|
|
|
2014-09-03 19:21:19 -07:00
|
|
|
import (
|
|
|
|
"io"
|
2015-07-08 18:27:16 -07:00
|
|
|
|
|
|
|
. "github.com/tendermint/tendermint/common"
|
2014-09-03 19:21:19 -07:00
|
|
|
)
|
2014-06-16 16:39:25 -07:00
|
|
|
|
2014-12-09 18:49:04 -08:00
|
|
|
func WriteByteSlice(bz []byte, w io.Writer, n *int64, err *error) {
|
2015-06-25 20:28:34 -07:00
|
|
|
WriteVarint(len(bz), w, n, err)
|
2014-12-09 18:49:04 -08:00
|
|
|
WriteTo(bz, w, n, err)
|
2014-06-30 16:53:04 -07:00
|
|
|
}
|
|
|
|
|
2014-09-03 19:21:19 -07:00
|
|
|
func ReadByteSlice(r io.Reader, n *int64, err *error) []byte {
|
2015-06-25 20:28:34 -07:00
|
|
|
length := ReadVarint(r, n, err)
|
2014-09-03 19:21:19 -07:00
|
|
|
if *err != nil {
|
|
|
|
return nil
|
2014-07-01 14:50:24 -07:00
|
|
|
}
|
2015-07-08 18:27:16 -07:00
|
|
|
if length < 0 {
|
|
|
|
*err = ErrBinaryReadSizeUnderflow
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if MaxBinaryReadSize < MaxInt64(int64(length), *n+int64(length)) {
|
|
|
|
*err = ErrBinaryReadSizeOverflow
|
2015-07-07 18:35:21 -07:00
|
|
|
return nil
|
|
|
|
}
|
2015-04-25 13:25:44 -07:00
|
|
|
|
2015-07-08 12:43:05 -07:00
|
|
|
buf := make([]byte, length)
|
|
|
|
ReadFull(buf, r, n, err)
|
2014-09-03 19:21:19 -07:00
|
|
|
return buf
|
2014-06-24 17:28:40 -07:00
|
|
|
}
|
2014-09-14 15:37:32 -07:00
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
2014-12-09 18:49:04 -08:00
|
|
|
func WriteByteSlices(bzz [][]byte, w io.Writer, n *int64, err *error) {
|
2015-06-25 20:28:34 -07:00
|
|
|
WriteVarint(len(bzz), w, n, err)
|
2014-09-14 15:37:32 -07:00
|
|
|
for _, bz := range bzz {
|
2014-12-09 18:49:04 -08:00
|
|
|
WriteByteSlice(bz, w, n, err)
|
2014-09-14 15:37:32 -07:00
|
|
|
if *err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func ReadByteSlices(r io.Reader, n *int64, err *error) [][]byte {
|
2015-06-25 20:28:34 -07:00
|
|
|
length := ReadVarint(r, n, err)
|
2014-09-14 15:37:32 -07:00
|
|
|
if *err != nil {
|
|
|
|
return nil
|
|
|
|
}
|
2015-07-08 18:27:16 -07:00
|
|
|
if length < 0 {
|
|
|
|
*err = ErrBinaryReadSizeUnderflow
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if MaxBinaryReadSize < MaxInt64(int64(length), *n+int64(length)) {
|
|
|
|
*err = ErrBinaryReadSizeOverflow
|
2015-07-07 18:35:21 -07:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-09-14 15:37:32 -07:00
|
|
|
bzz := make([][]byte, length)
|
2015-05-23 14:45:36 -07:00
|
|
|
for i := 0; i < length; i++ {
|
2014-09-14 15:37:32 -07:00
|
|
|
bz := ReadByteSlice(r, n, err)
|
|
|
|
if *err != nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
bzz[i] = bz
|
|
|
|
}
|
|
|
|
return bzz
|
|
|
|
}
|