package codec
import (
+ "bytes"
"encoding/binary"
"math"
- "reflect"
- "unsafe"
)
-var order = binary.BigEndian
-
+// Buffer provides buffer for encoding and decoding data on wire.
type Buffer struct {
- pos int
buf []byte
+ pos int
+}
+
+// NewBuffer creates new buffer using b as data.
+func NewBuffer(b []byte) *Buffer {
+ return &Buffer{
+ buf: b,
+ }
}
+// Bytes returns buffer data up to current position.
func (b *Buffer) Bytes() []byte {
return b.buf[:b.pos]
}
+func (b *Buffer) EncodeBytes(v []byte, length int) {
+ if length == 0 {
+ length = len(v)
+ }
+ copy(b.buf[b.pos:b.pos+length], v)
+ b.pos += length
+}
+
+func (b *Buffer) DecodeBytes(length int) []byte {
+ v := b.buf[b.pos : b.pos+length]
+ b.pos += length
+ return v
+}
+
func (b *Buffer) EncodeBool(v bool) {
if v {
b.buf[b.pos] = 1
b.pos += 1
}
+func (b *Buffer) DecodeBool() bool {
+ v := b.buf[b.pos] != 0
+ b.pos += 1
+ return v
+}
+
func (b *Buffer) EncodeUint8(v uint8) {
- b.buf[b.pos] = v
+ b.buf[b.pos] = byte(v)
+ b.pos += 1
+}
+
+func (b *Buffer) DecodeUint8() uint8 {
+ v := uint8(b.buf[b.pos])
b.pos += 1
+ return v
}
func (b *Buffer) EncodeUint16(v uint16) {
- order.PutUint16(b.buf[b.pos:b.pos+2], v)
+ binary.BigEndian.PutUint16(b.buf[b.pos:b.pos+2], v)
+ b.pos += 2
+}
+
+func (b *Buffer) DecodeUint16() uint16 {
+ v := binary.BigEndian.Uint16(b.buf[b.pos : b.pos+2])
b.pos += 2
+ return v
}
func (b *Buffer) EncodeUint32(v uint32) {
- order.PutUint32(b.buf[b.pos:b.pos+4], v)
+ binary.BigEndian.PutUint32(b.buf[b.pos:b.pos+4], v)
b.pos += 4
}
+func (b *Buffer) DecodeUint32() uint32 {
+ v := binary.BigEndian.Uint32(b.buf[b.pos : b.pos+4])
+ b.pos += 4
+ return v
+}
+
func (b *Buffer) EncodeUint64(v uint64) {
- order.PutUint64(b.buf[b.pos:b.pos+8], v)
+ binary.BigEndian.PutUint64(b.buf[b.pos:b.pos+8], v)
+ b.pos += 8
+}
+
+func (b *Buffer) DecodeUint64() uint64 {
+ v := binary.BigEndian.Uint64(b.buf[b.pos : b.pos+8])
+ b.pos += 8
+ return v
+}
+
+func (b *Buffer) EncodeInt8(v int8) {
+ b.buf[b.pos] = byte(v)
+ b.pos += 1
+}
+
+func (b *Buffer) DecodeInt8() int8 {
+ v := int8(b.buf[b.pos])
+ b.pos += 1
+ return v
+}
+
+func (b *Buffer) EncodeInt16(v int16) {
+ binary.BigEndian.PutUint16(b.buf[b.pos:b.pos+2], uint16(v))
+ b.pos += 2
+}
+
+func (b *Buffer) DecodeInt16() int16 {
+ v := int16(binary.BigEndian.Uint16(b.buf[b.pos : b.pos+2]))
+ b.pos += 2
+ return v
+}
+
+func (b *Buffer) EncodeInt32(v int32) {
+ binary.BigEndian.PutUint32(b.buf[b.pos:b.pos+4], uint32(v))
+ b.pos += 4
+}
+
+func (b *Buffer) DecodeInt32() int32 {
+ v := int32(binary.BigEndian.Uint32(b.buf[b.pos : b.pos+4]))
+ b.pos += 4
+ return v
+}
+
+func (b *Buffer) EncodeInt64(v int64) {
+ binary.BigEndian.PutUint64(b.buf[b.pos:b.pos+8], uint64(v))
+ b.pos += 8
+}
+
+func (b *Buffer) DecodeInt64() int64 {
+ v := int64(binary.BigEndian.Uint64(b.buf[b.pos : b.pos+8]))
b.pos += 8
+ return v
}
func (b *Buffer) EncodeFloat64(v float64) {
- order.PutUint64(b.buf[b.pos:b.pos+8], math.Float64bits(v))
+ binary.BigEndian.PutUint64(b.buf[b.pos:b.pos+8], math.Float64bits(v))
b.pos += 8
}
+func (b *Buffer) DecodeFloat64() float64 {
+ v := math.Float64frombits(binary.BigEndian.Uint64(b.buf[b.pos : b.pos+8]))
+ b.pos += 8
+ return v
+}
+
func (b *Buffer) EncodeString(v string, length int) {
+ if length == 0 {
+ b.EncodeUint32(uint32(len(v)))
+ length = len(v)
+ }
copy(b.buf[b.pos:b.pos+length], v)
b.pos += length
}
-func DecodeString(b []byte) string {
- sliceHeader := (*reflect.SliceHeader)(unsafe.Pointer(&b))
- stringHeader := reflect.StringHeader{Data: sliceHeader.Data, Len: sliceHeader.Len}
- return *(*string)(unsafe.Pointer(&stringHeader))
+func (b *Buffer) DecodeString(length int) string {
+ var v []byte
+ if length == 0 {
+ length = int(b.DecodeUint32())
+ v = b.buf[b.pos : b.pos+length]
+ } else {
+ v = b.buf[b.pos : b.pos+length]
+ if nul := bytes.Index(v, []byte{0x00}); nul >= 0 {
+ v = v[:nul]
+ }
+ }
+ b.pos += length
+ return string(v)
}