package msgpack import ( "fmt" "reflect" "github.com/vmihailenco/msgpack/v4/codes" ) func (d *Decoder) bytesLen(c codes.Code) (int, error) { if c == codes.Nil { return -1, nil } if codes.IsFixedString(c) { return int(c & codes.FixedStrMask), nil } switch c { case codes.Str8, codes.Bin8: n, err := d.uint8() return int(n), err case codes.Str16, codes.Bin16: n, err := d.uint16() return int(n), err case codes.Str32, codes.Bin32: n, err := d.uint32() return int(n), err } return 0, fmt.Errorf("msgpack: invalid code=%x decoding bytes length", c) } func (d *Decoder) DecodeString() (string, error) { c, err := d.readCode() if err != nil { return "", err } return d.string(c) } func (d *Decoder) string(c codes.Code) (string, error) { n, err := d.bytesLen(c) if err != nil { return "", err } return d.stringWithLen(n) } func (d *Decoder) stringWithLen(n int) (string, error) { if n <= 0 { return "", nil } b, err := d.readN(n) return string(b), err } func decodeStringValue(d *Decoder, v reflect.Value) error { if err := mustSet(v); err != nil { return err } s, err := d.DecodeString() if err != nil { return err } v.SetString(s) return nil } func (d *Decoder) DecodeBytesLen() (int, error) { c, err := d.readCode() if err != nil { return 0, err } return d.bytesLen(c) } func (d *Decoder) DecodeBytes() ([]byte, error) { c, err := d.readCode() if err != nil { return nil, err } return d.bytes(c, nil) } func (d *Decoder) bytes(c codes.Code, b []byte) ([]byte, error) { n, err := d.bytesLen(c) if err != nil { return nil, err } if n == -1 { return nil, nil } return readN(d.r, b, n) } func (d *Decoder) bytesNoCopy() ([]byte, error) { c, err := d.readCode() if err != nil { return nil, err } n, err := d.bytesLen(c) if err != nil { return nil, err } if n == -1 { return nil, nil } return d.readN(n) } func (d *Decoder) decodeBytesPtr(ptr *[]byte) error { c, err := d.readCode() if err != nil { return err } return d.bytesPtr(c, ptr) } func (d *Decoder) bytesPtr(c codes.Code, ptr *[]byte) error { n, err := d.bytesLen(c) if err != nil { return err } if n == -1 { *ptr = nil return nil } *ptr, err = readN(d.r, *ptr, n) return err } func (d *Decoder) skipBytes(c codes.Code) error { n, err := d.bytesLen(c) if err != nil { return err } if n <= 0 { return nil } return d.skipN(n) } func decodeBytesValue(d *Decoder, v reflect.Value) error { if err := mustSet(v); err != nil { return err } c, err := d.readCode() if err != nil { return err } b, err := d.bytes(c, v.Bytes()) if err != nil { return err } v.SetBytes(b) return nil } func decodeByteArrayValue(d *Decoder, v reflect.Value) error { c, err := d.readCode() if err != nil { return err } n, err := d.bytesLen(c) if err != nil { return err } if n == -1 { return nil } if n > v.Len() { return fmt.Errorf("%s len is %d, but msgpack has %d elements", v.Type(), v.Len(), n) } b := v.Slice(0, n).Bytes() return d.readFull(b) }