439 lines
12 KiB
Go
439 lines
12 KiB
Go
package cbor
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math"
|
|
)
|
|
|
|
// AppendNull inserts a 'Nil' object into the dst byte array.
|
|
func AppendNull(dst []byte) []byte {
|
|
return append(dst, byte(majorTypeSimpleAndFloat|additionalTypeNull))
|
|
}
|
|
|
|
// AppendBeginMarker inserts a map start into the dst byte array.
|
|
func AppendBeginMarker(dst []byte) []byte {
|
|
return append(dst, byte(majorTypeMap|additionalTypeInfiniteCount))
|
|
}
|
|
|
|
// AppendEndMarker inserts a map end into the dst byte array.
|
|
func AppendEndMarker(dst []byte) []byte {
|
|
return append(dst, byte(majorTypeSimpleAndFloat|additionalTypeBreak))
|
|
}
|
|
|
|
// AppendBool encodes and inserts a boolean value into the dst byte array.
|
|
func AppendBool(dst []byte, val bool) []byte {
|
|
b := additionalTypeBoolFalse
|
|
if val {
|
|
b = additionalTypeBoolTrue
|
|
}
|
|
return append(dst, byte(majorTypeSimpleAndFloat|b))
|
|
}
|
|
|
|
// AppendBools encodes and inserts an array of boolean values into the dst byte array.
|
|
func AppendBools(dst []byte, vals []bool) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendBool(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInt encodes and inserts an integer value into the dst byte array.
|
|
func AppendInt(dst []byte, val int) []byte {
|
|
major := majorTypeUnsignedInt
|
|
contentVal := val
|
|
if val < 0 {
|
|
major = majorTypeNegativeInt
|
|
contentVal = -val - 1
|
|
}
|
|
if contentVal <= additionalMax {
|
|
lb := byte(contentVal)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(contentVal))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInts encodes and inserts an array of integer values into the dst byte array.
|
|
func AppendInts(dst []byte, vals []int) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendInt(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInt8 encodes and inserts an int8 value into the dst byte array.
|
|
func AppendInt8(dst []byte, val int8) []byte {
|
|
return AppendInt(dst, int(val))
|
|
}
|
|
|
|
// AppendInts8 encodes and inserts an array of integer values into the dst byte array.
|
|
func AppendInts8(dst []byte, vals []int8) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendInt(dst, int(v))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInt16 encodes and inserts a int16 value into the dst byte array.
|
|
func AppendInt16(dst []byte, val int16) []byte {
|
|
return AppendInt(dst, int(val))
|
|
}
|
|
|
|
// AppendInts16 encodes and inserts an array of int16 values into the dst byte array.
|
|
func AppendInts16(dst []byte, vals []int16) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendInt(dst, int(v))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInt32 encodes and inserts a int32 value into the dst byte array.
|
|
func AppendInt32(dst []byte, val int32) []byte {
|
|
return AppendInt(dst, int(val))
|
|
}
|
|
|
|
// AppendInts32 encodes and inserts an array of int32 values into the dst byte array.
|
|
func AppendInts32(dst []byte, vals []int32) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendInt(dst, int(v))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInt64 encodes and inserts a int64 value into the dst byte array.
|
|
func AppendInt64(dst []byte, val int64) []byte {
|
|
major := majorTypeUnsignedInt
|
|
contentVal := val
|
|
if val < 0 {
|
|
major = majorTypeNegativeInt
|
|
contentVal = -val - 1
|
|
}
|
|
if contentVal <= additionalMax {
|
|
lb := byte(contentVal)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(contentVal))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInts64 encodes and inserts an array of int64 values into the dst byte array.
|
|
func AppendInts64(dst []byte, vals []int64) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendInt64(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUint encodes and inserts an unsigned integer value into the dst byte array.
|
|
func AppendUint(dst []byte, val uint) []byte {
|
|
return AppendInt64(dst, int64(val))
|
|
}
|
|
|
|
// AppendUints encodes and inserts an array of unsigned integer values into the dst byte array.
|
|
func AppendUints(dst []byte, vals []uint) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendUint(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUint8 encodes and inserts a unsigned int8 value into the dst byte array.
|
|
func AppendUint8(dst []byte, val uint8) []byte {
|
|
return AppendUint(dst, uint(val))
|
|
}
|
|
|
|
// AppendUints8 encodes and inserts an array of uint8 values into the dst byte array.
|
|
func AppendUints8(dst []byte, vals []uint8) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendUint8(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUint16 encodes and inserts a uint16 value into the dst byte array.
|
|
func AppendUint16(dst []byte, val uint16) []byte {
|
|
return AppendUint(dst, uint(val))
|
|
}
|
|
|
|
// AppendUints16 encodes and inserts an array of uint16 values into the dst byte array.
|
|
func AppendUints16(dst []byte, vals []uint16) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendUint16(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUint32 encodes and inserts a uint32 value into the dst byte array.
|
|
func AppendUint32(dst []byte, val uint32) []byte {
|
|
return AppendUint(dst, uint(val))
|
|
}
|
|
|
|
// AppendUints32 encodes and inserts an array of uint32 values into the dst byte array.
|
|
func AppendUints32(dst []byte, vals []uint32) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendUint32(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUint64 encodes and inserts a uint64 value into the dst byte array.
|
|
func AppendUint64(dst []byte, val uint64) []byte {
|
|
major := majorTypeUnsignedInt
|
|
contentVal := val
|
|
if contentVal <= additionalMax {
|
|
lb := byte(contentVal)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(contentVal))
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendUints64 encodes and inserts an array of uint64 values into the dst byte array.
|
|
func AppendUints64(dst []byte, vals []uint64) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendUint64(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendFloat32 encodes and inserts a single precision float value into the dst byte array.
|
|
func AppendFloat32(dst []byte, val float32) []byte {
|
|
switch {
|
|
case math.IsNaN(float64(val)):
|
|
return append(dst, "\xfa\x7f\xc0\x00\x00"...)
|
|
case math.IsInf(float64(val), 1):
|
|
return append(dst, "\xfa\x7f\x80\x00\x00"...)
|
|
case math.IsInf(float64(val), -1):
|
|
return append(dst, "\xfa\xff\x80\x00\x00"...)
|
|
}
|
|
major := majorTypeSimpleAndFloat
|
|
subType := additionalTypeFloat32
|
|
n := math.Float32bits(val)
|
|
var buf [4]byte
|
|
for i := uint(0); i < 4; i++ {
|
|
buf[i] = byte(n >> ((3 - i) * 8))
|
|
}
|
|
return append(append(dst, byte(major|subType)), buf[0], buf[1], buf[2], buf[3])
|
|
}
|
|
|
|
// AppendFloats32 encodes and inserts an array of single precision float value into the dst byte array.
|
|
func AppendFloats32(dst []byte, vals []float32) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendFloat32(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendFloat64 encodes and inserts a double precision float value into the dst byte array.
|
|
func AppendFloat64(dst []byte, val float64) []byte {
|
|
switch {
|
|
case math.IsNaN(val):
|
|
return append(dst, "\xfb\x7f\xf8\x00\x00\x00\x00\x00\x00"...)
|
|
case math.IsInf(val, 1):
|
|
return append(dst, "\xfb\x7f\xf0\x00\x00\x00\x00\x00\x00"...)
|
|
case math.IsInf(val, -1):
|
|
return append(dst, "\xfb\xff\xf0\x00\x00\x00\x00\x00\x00"...)
|
|
}
|
|
major := majorTypeSimpleAndFloat
|
|
subType := additionalTypeFloat64
|
|
n := math.Float64bits(val)
|
|
dst = append(dst, byte(major|subType))
|
|
for i := uint(1); i <= 8; i++ {
|
|
b := byte(n >> ((8 - i) * 8))
|
|
dst = append(dst, b)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendFloats64 encodes and inserts an array of double precision float values into the dst byte array.
|
|
func AppendFloats64(dst []byte, vals []float64) []byte {
|
|
major := majorTypeArray
|
|
l := len(vals)
|
|
if l == 0 {
|
|
return AppendArrayEnd(AppendArrayStart(dst))
|
|
}
|
|
if l <= additionalMax {
|
|
lb := byte(l)
|
|
dst = append(dst, byte(major|lb))
|
|
} else {
|
|
dst = appendCborTypePrefix(dst, major, uint64(l))
|
|
}
|
|
for _, v := range vals {
|
|
dst = AppendFloat64(dst, v)
|
|
}
|
|
return dst
|
|
}
|
|
|
|
// AppendInterface takes an arbitrary object and converts it to JSON and embeds it dst.
|
|
func AppendInterface(dst []byte, i interface{}) []byte {
|
|
marshaled, err := json.Marshal(i)
|
|
if err != nil {
|
|
return AppendString(dst, fmt.Sprintf("marshaling error: %v", err))
|
|
}
|
|
return AppendEmbeddedJSON(dst, marshaled)
|
|
}
|
|
|
|
// AppendObjectData takes an object in form of a byte array and appends to dst.
|
|
func AppendObjectData(dst []byte, o []byte) []byte {
|
|
return append(dst, o...)
|
|
}
|
|
|
|
// AppendArrayStart adds markers to indicate the start of an array.
|
|
func AppendArrayStart(dst []byte) []byte {
|
|
return append(dst, byte(majorTypeArray|additionalTypeInfiniteCount))
|
|
}
|
|
|
|
// AppendArrayEnd adds markers to indicate the end of an array.
|
|
func AppendArrayEnd(dst []byte) []byte {
|
|
return append(dst, byte(majorTypeSimpleAndFloat|additionalTypeBreak))
|
|
}
|
|
|
|
// AppendArrayDelim adds markers to indicate end of a particular array element.
|
|
func AppendArrayDelim(dst []byte) []byte {
|
|
//No delimiters needed in cbor
|
|
return dst
|
|
}
|
|
|
|
func AppendHex (dst []byte, val []byte) []byte {
|
|
dst = append(dst, byte(majorTypeTags|additionalTypeIntUint16))
|
|
dst = append(dst, byte(additionalTypeTagHexString>>8))
|
|
dst = append(dst, byte(additionalTypeTagHexString&0xff))
|
|
return AppendBytes(dst, val)
|
|
}
|