mirror of
https://github.com/json-iterator/go.git
synced 2025-02-19 19:59:49 +02:00
Revert "Merge pull request #418 from bbrks/configurable_maxDepth"
This reverts commit 44a7e7340d23d2e5d5d942966eb683436c027da0, reversing changes made to dc11f49689fd1c9a6de20749def70bd889bf0d42.
This commit is contained in:
parent
44a7e7340d
commit
7c9f8c2d20
@ -2,11 +2,9 @@ package test
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
jsoniter "github.com/json-iterator/go"
|
||||
"github.com/json-iterator/go"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
@ -26,44 +24,6 @@ func Test_customize_float_marshal(t *testing.T) {
|
||||
should.Equal("1.234568", str)
|
||||
}
|
||||
|
||||
func Test_max_depth(t *testing.T) {
|
||||
deepJSON := func(depth int) []byte {
|
||||
return []byte(strings.Repeat(`[`, depth) + strings.Repeat(`]`, depth))
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
jsonDepth int
|
||||
cfgMaxDepth int
|
||||
expectedErr string
|
||||
}{
|
||||
// Test the default depth
|
||||
{jsonDepth: 10000, cfgMaxDepth: 0},
|
||||
{jsonDepth: 10001, cfgMaxDepth: 0, expectedErr: "max depth"},
|
||||
// Test max depth logic
|
||||
{jsonDepth: 5, cfgMaxDepth: 6},
|
||||
{jsonDepth: 5, cfgMaxDepth: 5},
|
||||
{jsonDepth: 5, cfgMaxDepth: 4, expectedErr: "max depth"},
|
||||
// Try a large depth without a limit
|
||||
{jsonDepth: 128000, cfgMaxDepth: -1},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
t.Run(fmt.Sprintf("jsonDepth:%v_cfgMaxDepth:%v", test.jsonDepth, test.cfgMaxDepth), func(t *testing.T) {
|
||||
should := require.New(t)
|
||||
cfg := jsoniter.Config{MaxDepth: test.cfgMaxDepth}.Froze()
|
||||
|
||||
var val interface{}
|
||||
err := cfg.Unmarshal(deepJSON(test.jsonDepth), &val)
|
||||
if test.expectedErr != "" {
|
||||
should.Error(err)
|
||||
should.Contains(err.Error(), test.expectedErr)
|
||||
} else {
|
||||
should.NoError(err)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_customize_tag_key(t *testing.T) {
|
||||
|
||||
type TestObject struct {
|
||||
|
10
config.go
10
config.go
@ -11,9 +11,6 @@ import (
|
||||
"github.com/modern-go/reflect2"
|
||||
)
|
||||
|
||||
// limit maximum depth of nesting, as allowed by https://tools.ietf.org/html/rfc7159#section-9
|
||||
const defaultMaxDepth = 10000
|
||||
|
||||
// Config customize how the API should behave.
|
||||
// The API is created from Config by Froze.
|
||||
type Config struct {
|
||||
@ -28,7 +25,6 @@ type Config struct {
|
||||
ValidateJsonRawMessage bool
|
||||
ObjectFieldMustBeSimpleString bool
|
||||
CaseSensitive bool
|
||||
MaxDepth int
|
||||
}
|
||||
|
||||
// API the public interface of this package.
|
||||
@ -60,7 +56,6 @@ var ConfigCompatibleWithStandardLibrary = Config{
|
||||
EscapeHTML: true,
|
||||
SortMapKeys: true,
|
||||
ValidateJsonRawMessage: true,
|
||||
MaxDepth: -1, // encoding/json has no max depth (stack overflow at 2581101)
|
||||
}.Froze()
|
||||
|
||||
// ConfigFastest marshals float with only 6 digits precision
|
||||
@ -85,7 +80,6 @@ type frozenConfig struct {
|
||||
streamPool *sync.Pool
|
||||
iteratorPool *sync.Pool
|
||||
caseSensitive bool
|
||||
maxDepth int
|
||||
}
|
||||
|
||||
func (cfg *frozenConfig) initCache() {
|
||||
@ -133,9 +127,6 @@ func addFrozenConfigToCache(cfg Config, frozenConfig *frozenConfig) {
|
||||
|
||||
// Froze forge API from config
|
||||
func (cfg Config) Froze() API {
|
||||
if cfg.MaxDepth == 0 {
|
||||
cfg.MaxDepth = defaultMaxDepth
|
||||
}
|
||||
api := &frozenConfig{
|
||||
sortMapKeys: cfg.SortMapKeys,
|
||||
indentionStep: cfg.IndentionStep,
|
||||
@ -143,7 +134,6 @@ func (cfg Config) Froze() API {
|
||||
onlyTaggedField: cfg.OnlyTaggedField,
|
||||
disallowUnknownFields: cfg.DisallowUnknownFields,
|
||||
caseSensitive: cfg.CaseSensitive,
|
||||
maxDepth: cfg.MaxDepth,
|
||||
}
|
||||
api.streamPool = &sync.Pool{
|
||||
New: func() interface{} {
|
||||
|
5
iter.go
5
iter.go
@ -327,9 +327,12 @@ func (iter *Iterator) Read() interface{} {
|
||||
}
|
||||
}
|
||||
|
||||
// limit maximum depth of nesting, as allowed by https://tools.ietf.org/html/rfc7159#section-9
|
||||
const maxDepth = 10000
|
||||
|
||||
func (iter *Iterator) incrementDepth() (success bool) {
|
||||
iter.depth++
|
||||
if iter.depth <= iter.cfg.maxDepth || iter.cfg.maxDepth < 0 {
|
||||
if iter.depth <= maxDepth {
|
||||
return true
|
||||
}
|
||||
iter.ReportError("incrementDepth", "exceeded max depth")
|
||||
|
Loading…
x
Reference in New Issue
Block a user