You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.1 KiB
58 lines
1.1 KiB
package binary
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
func WriteByteSlice(bz []byte, w io.Writer, n *int64, err *error) {
|
|
WriteVarint(len(bz), w, n, err)
|
|
WriteTo(bz, w, n, err)
|
|
}
|
|
|
|
func ReadByteSlice(r io.Reader, n *int64, err *error) []byte {
|
|
length := ReadVarint(r, n, err)
|
|
if *err != nil {
|
|
return nil
|
|
}
|
|
if MaxBinaryReadSize < *n+int64(length) {
|
|
*err = ErrMaxBinaryReadSizeReached
|
|
return nil
|
|
}
|
|
|
|
buf := make([]byte, length)
|
|
ReadFull(buf, r, n, err)
|
|
return buf
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
func WriteByteSlices(bzz [][]byte, w io.Writer, n *int64, err *error) {
|
|
WriteVarint(len(bzz), w, n, err)
|
|
for _, bz := range bzz {
|
|
WriteByteSlice(bz, w, n, err)
|
|
if *err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func ReadByteSlices(r io.Reader, n *int64, err *error) [][]byte {
|
|
length := ReadVarint(r, n, err)
|
|
if *err != nil {
|
|
return nil
|
|
}
|
|
if MaxBinaryReadSize < *n+int64(length) {
|
|
*err = ErrMaxBinaryReadSizeReached
|
|
return nil
|
|
}
|
|
|
|
bzz := make([][]byte, length)
|
|
for i := 0; i < length; i++ {
|
|
bz := ReadByteSlice(r, n, err)
|
|
if *err != nil {
|
|
return nil
|
|
}
|
|
bzz[i] = bz
|
|
}
|
|
return bzz
|
|
}
|