githubmirror/client

go/msgpack/msgpack.go
56 строк1 KB
heronhaye
blindmerkle tree types
24 июл 2019, 20:11
24 июл 2019, 20:11524bfdb
// Copyright 2018 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.

package msgpack

import (
	"fmt"

	"github.com/keybase/client/go/kbcrypto"
	"github.com/keybase/go-codec/codec"
)

func EncodeCanonical(src interface{}) (dst []byte, err error) {
	ch := kbcrypto.CodecHandle()
	ch.Canonical = true
	err = codec.NewEncoderBytes(&dst, ch).Encode(src)
	return dst, err
}

func Decode(dst interface{}, src []byte) (err error) {
	ch := kbcrypto.CodecHandle()
	return codec.NewDecoderBytes(src, ch).Decode(dst)
}

func Encode(src interface{}) (dst []byte, err error) {
	ch := kbcrypto.CodecHandle()
	err = codec.NewEncoderBytes(&dst, ch).Encode(src)
	return dst, err
}

// Decode data into out, but make sure that all bytes in data are
// used.
func DecodeAll(data []byte, handle *codec.MsgpackHandle, out interface{}) error {
	decoder := codec.NewDecoderBytes(data, handle)
	err := decoder.Decode(out)
	if err != nil {
		return err
	}

	if decoder.NumBytesRead() != len(data) {
		return fmt.Errorf("Did not consume entire buffer: %d byte(s) left", len(data)-decoder.NumBytesRead())
	}
	return nil
}

func IsEncodedMsgpackArray(data []byte) bool {
	if len(data) == 0 {
		return false
	}
	b := data[0]
	return (b >= 0x90 && b <= 0x9f) || b == 0xdc || b == 0xdd
}

func IsJSONObject(data []byte) bool {
	return len(data) > 0 && data[0] == '{'
}