158 lines
7.8 KiB
Go
158 lines
7.8 KiB
Go
|
// Copyright (c) Faye Amacker. All rights reserved.
|
||
|
|
// Licensed under the MIT License. See LICENSE in the project root for license information.
|
||
|
|
|
||
|
|
package cbor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestValid1(t *testing.T) {
|
||
|
|
for _, mt := range marshalTests {
|
||
|
|
if err := Valid(mt.cborData); err != nil {
|
||
|
|
t.Errorf("Valid() returned error %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValid2(t *testing.T) {
|
||
|
|
for _, mt := range marshalTests {
|
||
|
|
dm, _ := DecOptions{DupMapKey: DupMapKeyEnforcedAPF}.DecMode()
|
||
|
|
if err := dm.Valid(mt.cborData); err != nil {
|
||
|
|
t.Errorf("Valid() returned error %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValidOnStreamingData(t *testing.T) {
|
||
|
|
var buf bytes.Buffer
|
||
|
|
for _, t := range marshalTests {
|
||
|
|
buf.Write(t.cborData)
|
||
|
|
}
|
||
|
|
d := decoder{data: buf.Bytes(), dm: defaultDecMode}
|
||
|
|
for i := 0; i < len(marshalTests); i++ {
|
||
|
|
if err := d.valid(); err != nil {
|
||
|
|
t.Errorf("valid() returned error %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDepth(t *testing.T) {
|
||
|
|
testCases := []struct {
|
||
|
|
name string
|
||
|
|
cborData []byte
|
||
|
|
wantDepth int
|
||
|
|
}{
|
||
|
|
{"uint", hexDecode("00"), 0}, // 0
|
||
|
|
{"int", hexDecode("20"), 0}, // -1
|
||
|
|
{"bool", hexDecode("f4"), 0}, // false
|
||
|
|
{"nil", hexDecode("f6"), 0}, // nil
|
||
|
|
{"float", hexDecode("fa47c35000"), 0}, // 100000.0
|
||
|
|
{"byte string", hexDecode("40"), 0}, // []byte{}
|
||
|
|
{"indefinite length byte string", hexDecode("5f42010243030405ff"), 0}, // []byte{1, 2, 3, 4, 5}
|
||
|
|
{"text string", hexDecode("60"), 0}, // ""
|
||
|
|
{"indefinite length text string", hexDecode("7f657374726561646d696e67ff"), 0}, // "streaming"
|
||
|
|
{"empty array", hexDecode("80"), 1}, // []
|
||
|
|
{"indefinite length empty array", hexDecode("9fff"), 1}, // []
|
||
|
|
{"array", hexDecode("98190102030405060708090a0b0c0d0e0f101112131415161718181819"), 1}, // [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]
|
||
|
|
{"indefinite length array", hexDecode("9f0102030405060708090a0b0c0d0e0f101112131415161718181819ff"), 1}, // [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]
|
||
|
|
{"nested array", hexDecode("8301820203820405"), 2}, // [1,[2,3],[4,5]]
|
||
|
|
{"indefinite length nested array", hexDecode("83018202039f0405ff"), 2}, // [1,[2,3],[4,5]]
|
||
|
|
{"array and map", hexDecode("826161a161626163"), 2}, // [a", {"b": "c"}]
|
||
|
|
{"indefinite length array and map", hexDecode("826161bf61626163ff"), 2}, // [a", {"b": "c"}]
|
||
|
|
{"empty map", hexDecode("a0"), 1}, // {}
|
||
|
|
{"indefinite length empty map", hexDecode("bfff"), 1}, // {}
|
||
|
|
{"map", hexDecode("a201020304"), 1}, // {1:2, 3:4}
|
||
|
|
{"nested map", hexDecode("a26161016162820203"), 2}, // {"a": 1, "b": [2, 3]}
|
||
|
|
{"indefinite length nested map", hexDecode("bf61610161629f0203ffff"), 2}, // {"a": 1, "b": [2, 3]}
|
||
|
|
{"tag", hexDecode("c074323031332d30332d32315432303a30343a30305a"), 0}, // 0("2013-03-21T20:04:00Z")
|
||
|
|
{"tagged map", hexDecode("d864a26161016162820203"), 2}, // 100({"a": 1, "b": [2, 3]})
|
||
|
|
{"tagged map and array", hexDecode("d864a26161016162d865820203"), 2}, // 100({"a": 1, "b": 101([2, 3])})
|
||
|
|
{"tagged map and array", hexDecode("d864a26161016162d865d866820203"), 3}, // 100({"a": 1, "b": 101(102([2, 3]))})
|
||
|
|
{"nested tag", hexDecode("d864d865d86674323031332d30332d32315432303a30343a30305a"), 2}, // 100(101(102("2013-03-21T20:04:00Z")))
|
||
|
|
{"32-level array", hexDecode("82018181818181818181818181818181818181818181818181818181818181818101"), 32},
|
||
|
|
{"32-level indefinite length array", hexDecode("9f018181818181818181818181818181818181818181818181818181818181818101ff"), 32},
|
||
|
|
{"32-level map", hexDecode("a1018181818181818181818181818181818181818181818181818181818181818101"), 32},
|
||
|
|
{"32-level indefinite length map", hexDecode("bf018181818181818181818181818181818181818181818181818181818181818101ff"), 32},
|
||
|
|
{"32-level tag", hexDecode("d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d86474323031332d30332d32315432303a30343a30305a"), 32}, // 100(100(...("2013-03-21T20:04:00Z")))
|
||
|
|
}
|
||
|
|
for _, tc := range testCases {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
d := decoder{data: tc.cborData, dm: defaultDecMode}
|
||
|
|
depth, err := d.validInternal(0)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("valid(0x%x) returned error %v", tc.cborData, err)
|
||
|
|
}
|
||
|
|
if depth != tc.wantDepth {
|
||
|
|
t.Errorf("valid(0x%x) returned depth %d, want %d", tc.cborData, depth, tc.wantDepth)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDepthError(t *testing.T) {
|
||
|
|
testCases := []struct {
|
||
|
|
name string
|
||
|
|
cborData []byte
|
||
|
|
opts DecOptions
|
||
|
|
wantErrorMsg string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "33-level array",
|
||
|
|
cborData: hexDecode("82018181818181818181818181818181818181818181818181818181818181818101"),
|
||
|
|
opts: DecOptions{MaxNestedLevels: 4},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 4",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level array",
|
||
|
|
cborData: hexDecode("82018181818181818181818181818181818181818181818181818181818181818101"),
|
||
|
|
opts: DecOptions{MaxNestedLevels: 10},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 10",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level array",
|
||
|
|
cborData: hexDecode("8201818181818181818181818181818181818181818181818181818181818181818101"),
|
||
|
|
opts: DecOptions{},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 32",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level indefinite length array",
|
||
|
|
cborData: hexDecode("9f01818181818181818181818181818181818181818181818181818181818181818101ff"),
|
||
|
|
opts: DecOptions{},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 32",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level map",
|
||
|
|
cborData: hexDecode("a101818181818181818181818181818181818181818181818181818181818181818101"),
|
||
|
|
opts: DecOptions{},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 32",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level indefinite length map",
|
||
|
|
cborData: hexDecode("bf01818181818181818181818181818181818181818181818181818181818181818101ff"),
|
||
|
|
opts: DecOptions{},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 32",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "33-level tag",
|
||
|
|
cborData: hexDecode("d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d864d86474323031332d30332d32315432303a30343a30305a"),
|
||
|
|
opts: DecOptions{},
|
||
|
|
wantErrorMsg: "cbor: exceeded max nested level 32",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tc := range testCases {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
dm, _ := tc.opts.decMode()
|
||
|
|
d := decoder{data: tc.cborData, dm: dm}
|
||
|
|
if _, err := d.validInternal(0); err == nil {
|
||
|
|
t.Errorf("valid(0x%x) didn't return an error", tc.cborData)
|
||
|
|
} else if _, ok := err.(*MaxNestedLevelError); !ok {
|
||
|
|
t.Errorf("valid(0x%x) returned wrong error type %T, want (*MaxNestedLevelError)", tc.cborData, err)
|
||
|
|
} else if err.Error() != tc.wantErrorMsg {
|
||
|
|
t.Errorf("valid(0x%x) returned error %q, want error %q", tc.cborData, err.Error(), tc.wantErrorMsg)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|