1
0
mirror of https://github.com/golang/go synced 2024-11-23 10:30:03 -07:00
go/test/fixedbugs/issue57823.go
Cuong Manh Le 198074abd7 cmd/compile: fix unsafe.{SliceData,StringData} escape analysis memory corruption
Fixes #57823

Change-Id: I54654d3ecb20b75afa9052c5c9db2072a86188d4
Reviewed-on: https://go-review.googlesource.com/c/go/+/461759
Reviewed-by: Cherry Mui <cherryyz@google.com>
Auto-Submit: Cuong Manh Le <cuong.manhle.vn@gmail.com>
Reviewed-by: Keith Randall <khr@golang.org>
TryBot-Result: Gopher Robot <gobot@golang.org>
Run-TryBot: Cuong Manh Le <cuong.manhle.vn@gmail.com>
Reviewed-by: Keith Randall <khr@google.com>
Reviewed-by: Matthew Dempsky <mdempsky@google.com>
2023-01-18 01:27:21 +00:00

77 lines
1.1 KiB
Go

// run
// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"runtime"
"unsafe"
)
//go:noinline
func g(x *byte) *byte { return x }
func main() {
slice()
str("AAAAAAAA", "BBBBBBBBB")
}
func wait(done <-chan struct{}) bool {
for i := 0; i < 10; i++ {
runtime.GC()
select {
case <-done:
return true
default:
}
}
return false
}
func slice() {
s := make([]byte, 100)
s[0] = 1
one := unsafe.SliceData(s)
done := make(chan struct{})
runtime.SetFinalizer(one, func(*byte) { close(done) })
h := g(one)
if wait(done) {
panic("GC'd early")
}
if *h != 1 {
panic("lost one")
}
if !wait(done) {
panic("never GC'd")
}
}
var strDone = make(chan struct{})
//go:noinline
func str(x, y string) {
s := x + y // put in temporary on stack
p := unsafe.StringData(s)
runtime.SetFinalizer(p, func(*byte) { close(strDone) })
if wait(strDone) {
panic("GC'd early")
}
if *p != 'A' {
panic("lost p")
}
if !wait(strDone) {
panic("never GC'd")
}
}