summaryrefslogtreecommitdiff
path: root/vendor/github.com/vmihailenco/msgpack/v5/ext.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/vmihailenco/msgpack/v5/ext.go')
-rw-r--r--vendor/github.com/vmihailenco/msgpack/v5/ext.go303
1 files changed, 0 insertions, 303 deletions
diff --git a/vendor/github.com/vmihailenco/msgpack/v5/ext.go b/vendor/github.com/vmihailenco/msgpack/v5/ext.go
deleted file mode 100644
index 354b9d92d..000000000
--- a/vendor/github.com/vmihailenco/msgpack/v5/ext.go
+++ /dev/null
@@ -1,303 +0,0 @@
-package msgpack
-
-import (
- "fmt"
- "math"
- "reflect"
-
- "github.com/vmihailenco/msgpack/v5/msgpcode"
-)
-
-type extInfo struct {
- Type reflect.Type
- Decoder func(d *Decoder, v reflect.Value, extLen int) error
-}
-
-var extTypes = make(map[int8]*extInfo)
-
-type MarshalerUnmarshaler interface {
- Marshaler
- Unmarshaler
-}
-
-func RegisterExt(extID int8, value MarshalerUnmarshaler) {
- RegisterExtEncoder(extID, value, func(e *Encoder, v reflect.Value) ([]byte, error) {
- marshaler := v.Interface().(Marshaler)
- return marshaler.MarshalMsgpack()
- })
- RegisterExtDecoder(extID, value, func(d *Decoder, v reflect.Value, extLen int) error {
- b, err := d.readN(extLen)
- if err != nil {
- return err
- }
- return v.Interface().(Unmarshaler).UnmarshalMsgpack(b)
- })
-}
-
-func UnregisterExt(extID int8) {
- unregisterExtEncoder(extID)
- unregisterExtDecoder(extID)
-}
-
-func RegisterExtEncoder(
- extID int8,
- value interface{},
- encoder func(enc *Encoder, v reflect.Value) ([]byte, error),
-) {
- unregisterExtEncoder(extID)
-
- typ := reflect.TypeOf(value)
- extEncoder := makeExtEncoder(extID, typ, encoder)
- typeEncMap.Store(extID, typ)
- typeEncMap.Store(typ, extEncoder)
- if typ.Kind() == reflect.Ptr {
- typeEncMap.Store(typ.Elem(), makeExtEncoderAddr(extEncoder))
- }
-}
-
-func unregisterExtEncoder(extID int8) {
- t, ok := typeEncMap.Load(extID)
- if !ok {
- return
- }
- typeEncMap.Delete(extID)
- typ := t.(reflect.Type)
- typeEncMap.Delete(typ)
- if typ.Kind() == reflect.Ptr {
- typeEncMap.Delete(typ.Elem())
- }
-}
-
-func makeExtEncoder(
- extID int8,
- typ reflect.Type,
- encoder func(enc *Encoder, v reflect.Value) ([]byte, error),
-) encoderFunc {
- nilable := typ.Kind() == reflect.Ptr
-
- return func(e *Encoder, v reflect.Value) error {
- if nilable && v.IsNil() {
- return e.EncodeNil()
- }
-
- b, err := encoder(e, v)
- if err != nil {
- return err
- }
-
- if err := e.EncodeExtHeader(extID, len(b)); err != nil {
- return err
- }
-
- return e.write(b)
- }
-}
-
-func makeExtEncoderAddr(extEncoder encoderFunc) encoderFunc {
- return func(e *Encoder, v reflect.Value) error {
- if !v.CanAddr() {
- return fmt.Errorf("msgpack: EncodeExt(nonaddressable %T)", v.Interface())
- }
- return extEncoder(e, v.Addr())
- }
-}
-
-func RegisterExtDecoder(
- extID int8,
- value interface{},
- decoder func(dec *Decoder, v reflect.Value, extLen int) error,
-) {
- unregisterExtDecoder(extID)
-
- typ := reflect.TypeOf(value)
- extDecoder := makeExtDecoder(extID, typ, decoder)
- extTypes[extID] = &extInfo{
- Type: typ,
- Decoder: decoder,
- }
-
- typeDecMap.Store(extID, typ)
- typeDecMap.Store(typ, extDecoder)
- if typ.Kind() == reflect.Ptr {
- typeDecMap.Store(typ.Elem(), makeExtDecoderAddr(extDecoder))
- }
-}
-
-func unregisterExtDecoder(extID int8) {
- t, ok := typeDecMap.Load(extID)
- if !ok {
- return
- }
- typeDecMap.Delete(extID)
- delete(extTypes, extID)
- typ := t.(reflect.Type)
- typeDecMap.Delete(typ)
- if typ.Kind() == reflect.Ptr {
- typeDecMap.Delete(typ.Elem())
- }
-}
-
-func makeExtDecoder(
- wantedExtID int8,
- typ reflect.Type,
- decoder func(d *Decoder, v reflect.Value, extLen int) error,
-) decoderFunc {
- return nilAwareDecoder(typ, func(d *Decoder, v reflect.Value) error {
- extID, extLen, err := d.DecodeExtHeader()
- if err != nil {
- return err
- }
- if extID != wantedExtID {
- return fmt.Errorf("msgpack: got ext type=%d, wanted %d", extID, wantedExtID)
- }
- return decoder(d, v, extLen)
- })
-}
-
-func makeExtDecoderAddr(extDecoder decoderFunc) decoderFunc {
- return func(d *Decoder, v reflect.Value) error {
- if !v.CanAddr() {
- return fmt.Errorf("msgpack: DecodeExt(nonaddressable %T)", v.Interface())
- }
- return extDecoder(d, v.Addr())
- }
-}
-
-func (e *Encoder) EncodeExtHeader(extID int8, extLen int) error {
- if err := e.encodeExtLen(extLen); err != nil {
- return err
- }
- if err := e.w.WriteByte(byte(extID)); err != nil {
- return err
- }
- return nil
-}
-
-func (e *Encoder) encodeExtLen(l int) error {
- switch l {
- case 1:
- return e.writeCode(msgpcode.FixExt1)
- case 2:
- return e.writeCode(msgpcode.FixExt2)
- case 4:
- return e.writeCode(msgpcode.FixExt4)
- case 8:
- return e.writeCode(msgpcode.FixExt8)
- case 16:
- return e.writeCode(msgpcode.FixExt16)
- }
- if l <= math.MaxUint8 {
- return e.write1(msgpcode.Ext8, uint8(l))
- }
- if l <= math.MaxUint16 {
- return e.write2(msgpcode.Ext16, uint16(l))
- }
- return e.write4(msgpcode.Ext32, uint32(l))
-}
-
-func (d *Decoder) DecodeExtHeader() (extID int8, extLen int, err error) {
- c, err := d.readCode()
- if err != nil {
- return
- }
- return d.extHeader(c)
-}
-
-func (d *Decoder) extHeader(c byte) (int8, int, error) {
- extLen, err := d.parseExtLen(c)
- if err != nil {
- return 0, 0, err
- }
-
- extID, err := d.readCode()
- if err != nil {
- return 0, 0, err
- }
-
- return int8(extID), extLen, nil
-}
-
-func (d *Decoder) parseExtLen(c byte) (int, error) {
- switch c {
- case msgpcode.FixExt1:
- return 1, nil
- case msgpcode.FixExt2:
- return 2, nil
- case msgpcode.FixExt4:
- return 4, nil
- case msgpcode.FixExt8:
- return 8, nil
- case msgpcode.FixExt16:
- return 16, nil
- case msgpcode.Ext8:
- n, err := d.uint8()
- return int(n), err
- case msgpcode.Ext16:
- n, err := d.uint16()
- return int(n), err
- case msgpcode.Ext32:
- n, err := d.uint32()
- return int(n), err
- default:
- return 0, fmt.Errorf("msgpack: invalid code=%x decoding ext len", c)
- }
-}
-
-func (d *Decoder) decodeInterfaceExt(c byte) (interface{}, error) {
- extID, extLen, err := d.extHeader(c)
- if err != nil {
- return nil, err
- }
-
- info, ok := extTypes[extID]
- if !ok {
- return nil, fmt.Errorf("msgpack: unknown ext id=%d", extID)
- }
-
- v := d.newValue(info.Type).Elem()
- if nilable(v.Kind()) && v.IsNil() {
- v.Set(d.newValue(info.Type.Elem()))
- }
-
- if err := info.Decoder(d, v, extLen); err != nil {
- return nil, err
- }
-
- return v.Interface(), nil
-}
-
-func (d *Decoder) skipExt(c byte) error {
- n, err := d.parseExtLen(c)
- if err != nil {
- return err
- }
- return d.skipN(n + 1)
-}
-
-func (d *Decoder) skipExtHeader(c byte) error {
- // Read ext type.
- _, err := d.readCode()
- if err != nil {
- return err
- }
- // Read ext body len.
- for i := 0; i < extHeaderLen(c); i++ {
- _, err := d.readCode()
- if err != nil {
- return err
- }
- }
- return nil
-}
-
-func extHeaderLen(c byte) int {
- switch c {
- case msgpcode.Ext8:
- return 1
- case msgpcode.Ext16:
- return 2
- case msgpcode.Ext32:
- return 4
- }
- return 0
-}