Skip to content

Commit 556b3f5

Browse files
committed
test: add regress tests for unsafe.Pointer rules
Updates #23109. Change-Id: I55f7860c868acc948a6397ab6a9295e177724a56 Reviewed-on: https://go-review.googlesource.com/c/go/+/170450 Run-TryBot: Matthew Dempsky <[email protected]> TryBot-Result: Gobot Gobot <[email protected]> Reviewed-by: Cherry Zhang <[email protected]>
1 parent 17436af commit 556b3f5

File tree

1 file changed

+69
-0
lines changed

1 file changed

+69
-0
lines changed

test/escape_unsafe.go

+69
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
1+
// errorcheck -0 -m -l
2+
3+
// Copyright 2019 The Go Authors. All rights reserved.
4+
// Use of this source code is governed by a BSD-style
5+
// license that can be found in the LICENSE file.
6+
7+
// Test escape analysis for unsafe.Pointer rules.
8+
9+
package escape
10+
11+
import (
12+
"reflect"
13+
"unsafe"
14+
)
15+
16+
// (1) Conversion of a *T1 to Pointer to *T2.
17+
18+
func convert(p *float64) *uint64 { // ERROR "leaking param: p to result ~r1 level=0$"
19+
return (*uint64)(unsafe.Pointer(p))
20+
}
21+
22+
// (3) Conversion of a Pointer to a uintptr and back, with arithmetic.
23+
24+
func arithAdd() unsafe.Pointer {
25+
var x [2]byte // ERROR "moved to heap: x"
26+
return unsafe.Pointer(uintptr(unsafe.Pointer(&x[0])) + 1)
27+
}
28+
29+
func arithSub() unsafe.Pointer {
30+
var x [2]byte // ERROR "moved to heap: x"
31+
return unsafe.Pointer(uintptr(unsafe.Pointer(&x[1])) - 1)
32+
}
33+
34+
func arithMask() unsafe.Pointer {
35+
var x [2]byte // ERROR "moved to heap: x"
36+
return unsafe.Pointer(uintptr(unsafe.Pointer(&x[1])) &^ 1)
37+
}
38+
39+
// (5) Conversion of the result of reflect.Value.Pointer or
40+
// reflect.Value.UnsafeAddr from uintptr to Pointer.
41+
42+
// BAD: should be "leaking param: p to result ~r1 level=0$"
43+
func valuePointer(p *int) unsafe.Pointer { // ERROR "leaking param: p$"
44+
return unsafe.Pointer(reflect.ValueOf(p).Pointer()) // ERROR "p escapes to heap"
45+
}
46+
47+
// BAD: should be "leaking param: p to result ~r1 level=0$"
48+
func valueUnsafeAddr(p *int) unsafe.Pointer { // ERROR "leaking param: p$"
49+
return unsafe.Pointer(reflect.ValueOf(p).Elem().UnsafeAddr()) // ERROR "p escapes to heap"
50+
}
51+
52+
// (6) Conversion of a reflect.SliceHeader or reflect.StringHeader
53+
// Data field to or from Pointer.
54+
55+
func fromSliceData(s []int) unsafe.Pointer { // ERROR "leaking param: s to result ~r1 level=0$"
56+
return unsafe.Pointer((*reflect.SliceHeader)(unsafe.Pointer(&s)).Data)
57+
}
58+
59+
func fromStringData(s string) unsafe.Pointer { // ERROR "leaking param: s to result ~r1 level=0$"
60+
return unsafe.Pointer((*reflect.StringHeader)(unsafe.Pointer(&s)).Data)
61+
}
62+
63+
func toSliceData(s *[]int, p unsafe.Pointer) { // ERROR "s does not escape" "leaking param: p$"
64+
(*reflect.SliceHeader)(unsafe.Pointer(s)).Data = uintptr(p)
65+
}
66+
67+
func toStringData(s *string, p unsafe.Pointer) { // ERROR "s does not escape" "leaking param: p$"
68+
(*reflect.SliceHeader)(unsafe.Pointer(s)).Data = uintptr(p)
69+
}

0 commit comments

Comments
 (0)