erigon-pulse/turbo/rlphacks/bytes.go

59 lines
1.4 KiB
Go
Raw Normal View History

package rlphacks
import (
"io"
)
type RlpSerializableBytes []byte
func (b RlpSerializableBytes) ToDoubleRLP(w io.Writer, prefixBuf []byte) error {
return encodeBytesAsRlpToWriter(b, w, generateByteArrayLenDouble, prefixBuf)
}
func (b RlpSerializableBytes) RawBytes() []byte {
return b
}
func (b RlpSerializableBytes) DoubleRLPLen() int {
if len(b) < 1 {
return 0
}
return generateRlpPrefixLenDouble(len(b), b[0]) + len(b)
}
type RlpEncodedBytes []byte
func (b RlpEncodedBytes) ToDoubleRLP(w io.Writer, prefixBuf []byte) error {
return encodeBytesAsRlpToWriter(b, w, generateByteArrayLen, prefixBuf)
}
func (b RlpEncodedBytes) RawBytes() []byte {
return b
}
func (b RlpEncodedBytes) DoubleRLPLen() int {
return generateRlpPrefixLen(len(b)) + len(b)
}
func encodeBytesAsRlpToWriter(source []byte, w io.Writer, prefixGenFunc func([]byte, int, int) int, prefixBuf []byte) error {
// > 1 byte, write a prefix or prefixes first
if len(source) > 1 || (len(source) == 1 && source[0] >= 0x80) {
prefixLen := prefixGenFunc(prefixBuf, 0, len(source))
if _, err := w.Write(prefixBuf[:prefixLen]); err != nil {
return err
}
}
_, err := w.Write(source)
return err
}
func EncodeByteArrayAsRlp(raw []byte, w io.Writer, prefixBuf []byte) (int, error) {
err := encodeBytesAsRlpToWriter(raw, w, generateByteArrayLen, prefixBuf)
if err != nil {
return 0, err
}
return generateRlpPrefixLen(len(raw)) + len(raw), nil
}