2014-06-16 16:39:25 -07:00
|
|
|
package binary
|
|
|
|
|
2014-09-03 19:21:19 -07:00
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
)
|
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) {
|
2014-12-22 17:38:16 -08:00
|
|
|
WriteUvarint(uint(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 {
|
2014-12-22 17:38:16 -08:00
|
|
|
length := ReadUvarint(r, n, err)
|
2014-09-03 19:21:19 -07:00
|
|
|
if *err != nil {
|
|
|
|
return nil
|
2014-07-01 14:50:24 -07:00
|
|
|
}
|
2014-09-03 19:21:19 -07:00
|
|
|
buf := make([]byte, int(length))
|
2014-12-09 18:49:04 -08:00
|
|
|
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) {
|
2014-12-22 17:38:16 -08:00
|
|
|
WriteUvarint(uint(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 {
|
2014-12-22 17:38:16 -08:00
|
|
|
length := ReadUvarint(r, n, err)
|
2014-09-14 15:37:32 -07:00
|
|
|
if *err != nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
bzz := make([][]byte, length)
|
2014-12-09 18:49:04 -08:00
|
|
|
for i := uint(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
|
|
|
|
}
|