时间戳

首页 > 解决方案 > 将指向局部变量的指针分配给在Golang中作为指针返回的结构是否安全

go - 将指向局部变量的指针分配给在Golang中作为指针返回的结构是否安全

问题描述

我发现 Golang 中的以下代码有效：

type user struct {
  name *string
  email *string
}
func foo() (*user, error) {
  var myname string
  var myemail string
  // Some code here to fetch myname and myemail
  u := user{}
  u.name = &myname
  u.email = &myemail
  return &u, nil
}

我知道返回指向局部变量的指针是安全的，因为局部变量将在函数的范围内存活，所以我不担心&u从函数返回foo()。

我关心的是作业：

  u.name = &myname
  u.email = &myemail

u.nameGo 编译器在堆中分配和分配是偶然的，u.email以便我能够在函数之外访问它，还是有一些保证它会一直工作（通过指针转义分析机制）？

如果上面的代码不安全，我会回退到以下内容：

  u.name = new(string)
  *u.name = myname
  ...

标签： gomemory-management

解决方案

我进一步深入挖掘并从这里和这里弄清楚如何进一步调查并满足我的查询，即在结构中设置指向局部变量的指针确实是安全的，该结构本身作为函数的指针返回。

我的程序（打印行号以便更好地理解）：

  1 package main
  2
  3 import "fmt"
  4
  5 type user struct {
  6   Name *string
  7 }
  8
  9 func setName() (*user) {
 10   myname := "soumya"
 11   u := user{}
 12   u.Name = &myname
 13   return &u
 14 }
 15
 16 func main() {
 17   v := setName()
 18   fmt.Println(*v.Name)
 19 }

输出go run -gcflags='-m -m'：

$ go run -gcflags='-m -m' main.go
./main.go:9:6: can inline setName as: func() *user { myname := "soumya"; u := user literal; u.Name = &myname; return &u }
./main.go:16:6: cannot inline main: function too complex: cost 93 exceeds budget 80
./main.go:17:15: inlining call to setName func() *user { myname := "soumya"; u := user literal; u.Name = &myname; return &u }
./main.go:18:14: inlining call to fmt.Println func(...interface {}) (int, error) { return fmt.Fprintln(io.Writer(os.Stdout), fmt.a...) }
./main.go:13:10: &u escapes to heap
./main.go:13:10:        from ~r0 (return) at ./main.go:13:3
./main.go:11:3: moved to heap: u
./main.go:12:12: &myname escapes to heap
./main.go:12:12:        from u (dot-equals) at ./main.go:12:10
./main.go:12:12:        from &u (address-of) at ./main.go:13:10
./main.go:12:12:        from ~r0 (return) at ./main.go:13:3
./main.go:10:3: moved to heap: myname
./main.go:18:15: *v.Name escapes to heap
./main.go:18:15:        from ~arg0 (assign-pair) at ./main.go:18:14
./main.go:18:14: io.Writer(os.Stdout) escapes to heap
./main.go:18:14:        from io.Writer(os.Stdout) (passed to call[argument escapes]) at ./main.go:18:14
./main.go:17:15: main &myname does not escape
./main.go:17:15: main &u does not escape
./main.go:18:14: main []interface {} literal does not escape
<autogenerated>:1: os.(*File).close .this does not escape
soumya

输出状态&myname转义到第 1 行的堆。12，因此它是安全的。此外，它在第号行之后不再转义。17.

推荐阅读