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.
55 lines
1.3 KiB
55 lines
1.3 KiB
3 years ago
|
package bin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
// encodeBytes is same as encodeString, but for bytes.
|
||
|
|
func encodeBytes(b, v []byte) []byte {
|
||
|
|
l := len(v)
|
||
|
|
if l <= maxSmallStringLength {
|
||
|
|
b = append(b, byte(l))
|
||
|
|
b = append(b, v...)
|
||
|
|
currentLen := l + 1
|
||
|
|
b = append(b, make([]byte, nearestPaddedValueLength(currentLen)-currentLen)...)
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
b = append(b, firstLongStringByte, byte(l), byte(l>>8), byte(l>>16))
|
||
|
|
b = append(b, v...)
|
||
|
|
currentLen := l + 4
|
||
|
|
b = append(b, make([]byte, nearestPaddedValueLength(currentLen)-currentLen)...)
|
||
|
|
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
// decodeBytes is same as decodeString, but for bytes.
|
||
|
|
//
|
||
|
|
// NB: v is slice of b.
|
||
|
|
func decodeBytes(b []byte) (n int, v []byte, err error) {
|
||
|
|
if len(b) == 0 {
|
||
|
|
return 0, nil, io.ErrUnexpectedEOF
|
||
|
|
}
|
||
|
|
if b[0] == firstLongStringByte {
|
||
|
|
if len(b) < 4 {
|
||
|
|
return 0, nil, io.ErrUnexpectedEOF
|
||
|
|
}
|
||
|
|
strLen := uint32(b[1]) | uint32(b[2])<<8 | uint32(b[3])<<16
|
||
|
|
if len(b) < (int(strLen) + 4) {
|
||
|
|
return 0, nil, io.ErrUnexpectedEOF
|
||
|
|
}
|
||
|
|
return nearestPaddedValueLength(int(strLen) + 4), b[4 : strLen+4], nil
|
||
|
|
}
|
||
|
|
strLen := int(b[0])
|
||
|
|
if len(b) < (strLen + 1) {
|
||
|
|
return 0, nil, io.ErrUnexpectedEOF
|
||
|
|
}
|
||
|
|
if strLen > maxSmallStringLength {
|
||
|
|
return 0, nil, &InvalidLengthError{
|
||
|
|
Length: strLen,
|
||
|
|
Where: "bytes",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nearestPaddedValueLength(strLen + 1), b[1 : strLen+1], nil
|
||
|
|
}
|