github.com/goshafaq/sonic@v0.0.0-20231026082336-871835fb94c6/internal/decoder/norace_test.go (about)

     1  //go:build !race
     2  // +build !race
     3  
     4  /*
     5   * Copyright 2021 ByteDance Inc.
     6   *
     7   * Licensed under the Apache License, Version 2.0 (the "License");
     8   * you may not use this file except in compliance with the License.
     9   * You may obtain a copy of the License at
    10   *
    11   *     http://www.apache.org/licenses/LICENSE-2.0
    12   *
    13   * Unless required by applicable law or agreed to in writing, software
    14   * distributed under the License is distributed on an "AS IS" BASIS,
    15   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    16   * See the License for the specific language governing permissions and
    17   * limitations under the License.
    18   */
    19  
    20  package decoder
    21  
    22  import (
    23  	"runtime"
    24  	"testing"
    25  	"time"
    26  	"unsafe"
    27  
    28  	"github.com/goshafaq/sonic/internal/rt"
    29  )
    30  
    31  var referred = false
    32  
    33  func TestStringReferring(t *testing.T) {
    34  	str := []byte(`{"A":"0","B":"1"}`)
    35  	sp := *(**byte)(unsafe.Pointer(&str))
    36  	println("malloc *byte ", sp)
    37  	runtime.SetFinalizer(sp, func(sp *byte) {
    38  		referred = false
    39  		println("*byte ", sp, " got free 1")
    40  	})
    41  	runtime.GC()
    42  	println("first GC")
    43  	var obj struct {
    44  		A string
    45  		B string
    46  	}
    47  	dc := NewDecoder(rt.Mem2Str(str))
    48  	dc.CopyString()
    49  	referred = true
    50  	if err := dc.Decode(&obj); err != nil {
    51  		t.Fatal(err)
    52  	}
    53  	runtime.GC()
    54  	println("second GC")
    55  	time.Sleep(time.Millisecond)
    56  	if referred {
    57  		t.Fatal("*byte is being referred")
    58  	}
    59  
    60  	str2 := []byte(`{"A":"0","B":"1"}`)
    61  	sp2 := *(**byte)(unsafe.Pointer(&str2))
    62  	println("malloc *byte ", sp2)
    63  	runtime.SetFinalizer(sp2, func(sp *byte) {
    64  		referred = false
    65  		println("*byte ", sp, " got free")
    66  	})
    67  	runtime.GC()
    68  	println("first GC")
    69  	var obj2 interface{}
    70  	dc2 := NewDecoder(rt.Mem2Str(str2))
    71  	dc2.UseNumber()
    72  	dc2.CopyString()
    73  	referred = true
    74  	if err := dc2.Decode(&obj2); err != nil {
    75  		t.Fatal(err)
    76  	}
    77  	runtime.GC()
    78  	println("second GC")
    79  	time.Sleep(time.Millisecond)
    80  	if referred {
    81  		t.Fatal("*byte is being referred")
    82  	}
    83  
    84  	runtime.KeepAlive(&obj)
    85  	runtime.KeepAlive(&obj2)
    86  }
    87  
    88  func TestDecoderErrorStackOverflower(t *testing.T) {
    89  	src := `{"a":[]}`
    90  	N := _MaxStack
    91  	for i := 0; i < N; i++ {
    92  		var obj map[string]string
    93  		err := NewDecoder(src).Decode(&obj)
    94  		if err == nil {
    95  			t.Fatal(err)
    96  		}
    97  	}
    98  }