-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
cbor.go
54 lines (43 loc) · 1.04 KB
/
cbor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package models
import (
"bytes"
"encoding/json"
"github.com/smartcontractkit/chainlink/core/utils"
"github.com/ugorji/go/codec"
)
// ParseCBOR attempts to coerce the input byte array into valid CBOR
// and then coerces it into a JSON object.
func ParseCBOR(b []byte) (JSON, error) {
if len(b) == 0 {
return JSON{}, nil
}
var m map[interface{}]interface{}
cbor := codec.NewDecoderBytes(autoAddMapDelimiters(b), new(codec.CborHandle))
if err := cbor.Decode(&m); err != nil {
return JSON{}, err
}
coerced, err := utils.CoerceInterfaceMapToStringMap(m)
if err != nil {
return JSON{}, err
}
jsb, err := json.Marshal(coerced)
if err != nil {
return JSON{}, err
}
var js JSON
return js, json.Unmarshal(jsb, &js)
}
// Automatically add missing start map and end map to a CBOR encoded buffer
func autoAddMapDelimiters(b []byte) []byte {
if len(b) < 2 {
return b
}
if (b[0] >> 5) != 5 {
var buffer bytes.Buffer
buffer.Write([]byte{0xbf})
buffer.Write(b)
buffer.Write([]byte{0xff})
return buffer.Bytes()
}
return b
}