fix types

bytedance · Sep 27, 2022 · 1ea952b · 1ea952b
1 parent c7e36bc
commit 1ea952b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/decode_test.go b/decode_test.go
@@ -38,6 +38,7 @@ import (
     `unicode/utf8`
 
     `github.com/bytedance/sonic/decoder`
+    `github.com/bytedance/sonic/internal/native/types`
     `github.com/davecgh/go-spew/spew`
 )
 
@@ -2361,19 +2362,24 @@ func TestUnmarshalRescanLiteralMangledUnquote(t *testing.T) {
 }
 
 func TestUnmarshalMaxDepth(t *testing.T) {
+    const (
+        _MaxDepth = types.MAX_RECURSE
+        _OverMaxDepth = types.MAX_RECURSE + 1
+        _UnderMaxDepth = types.MAX_RECURSE - 2
+    )
     testcases := []struct {
         name        string
         data        string
         errMaxDepth bool
     }{
         {
             name:        "ArrayUnderMaxNestingDepth",
-            data:        `{"a":` + strings.Repeat(`[`, 65534) + `0` + strings.Repeat(`]`, 65534) + `}`,
+            data:        `{"a":` + strings.Repeat(`[`, _UnderMaxDepth) + `0` + strings.Repeat(`]`, _UnderMaxDepth) + `}`,
             errMaxDepth: false,
         },
         {
             name:        "ArrayOverMaxNestingDepth",
-            data:        `{"a":` + strings.Repeat(`[`, 65537) + `0` + strings.Repeat(`]`, 65537) + `}`,
+            data:        `{"a":` + strings.Repeat(`[`, _OverMaxDepth) + `0` + strings.Repeat(`]`, _OverMaxDepth) + `}`,
             errMaxDepth: true,
         },
         {
@@ -2383,12 +2389,12 @@ func TestUnmarshalMaxDepth(t *testing.T) {
         },
         {
             name:        "ObjectUnderMaxNestingDepth",
-            data:        `{"a":` + strings.Repeat(`{"a":`, 65534) + `0` + strings.Repeat(`}`, 65534) + `}`,
+            data:        `{"a":` + strings.Repeat(`{"a":`, _UnderMaxDepth) + `0` + strings.Repeat(`}`, _UnderMaxDepth) + `}`,
             errMaxDepth: false,
         },
         {
             name:        "ObjectOverMaxNestingDepth",
-            data:        `{"a":` + strings.Repeat(`{"a":`, 65537) + `0` + strings.Repeat(`}`, 65537) + `}`,
+            data:        `{"a":` + strings.Repeat(`{"a":`, _OverMaxDepth) + `0` + strings.Repeat(`}`, _OverMaxDepth) + `}`,
             errMaxDepth: true,
         },
         {

diff --git a/internal/native/types/types.go b/internal/native/types/types.go
@@ -58,7 +58,7 @@ const (
 )
 
 const (
-    MAX_RECURSE = 65536
+    MAX_RECURSE = 4096
 )
 
 const (