Go unsafe 包的使用详解

作者: 自从看了段子
来源: 51数据库
2021-08-05

unsafe包

golang是一种静态的强类型的语言，所有的类型都是不能随意转换的，go语言是不允许两个指针类型进行转换的。go官方是不推荐使用unsafe的操作因为它是不安全的，它绕过了golang的内存安全原则，容易使你的程序出现莫名其妙的问题，不利于程序的扩展与维护。但是在很多地方却是很实用。在一些go底层的包中unsafe包被很频繁的使用。

unsafe 定义

package unsafe
//arbitrarytype仅用于文档目的，实际上并不是unsafe包的一部分,它表示任意go表达式的类型。
type arbitrarytype int
//任意类型的指针，类似于c的*void
type pointer *arbitrarytype
//确定结构在内存中占用的确切大小
func sizeof(x arbitrarytype) uintptr
//返回结构体中某个field的偏移量
func offsetof(x arbitrarytype) uintptr
//返回结构体中某个field的对其值（字节对齐的原因）
func alignof(x arbitrarytype) uintptr

官方中定义了四个描述：

任何类型的指针都可以被转化为pointer
pointer可以被转化为任何类型的指针
uintptr可以被转化为pointer
pointer可以被转化为uintptr

unsafe的使用

类型转换

使用unsafe可以实现类型的转换，下面的例子可以看到i是一个int类型，使用unsafe.pointer转换成float64并且还修改了指针对应的值。

func main() {
  i := 10
  ip := &i

  fp := (*float64)(unsafe.pointer(ip))

  *fp = *fp * 3

  fmt.println(i)
}

// 结果： 30

但是使用起来要十分的小心，如果使用不当会引发错误。可以举一个例子：

func main() {
  i := 10
  ip := &i

  fp := (*string)(unsafe.pointer(ip))

  *fp = "a"

  fmt.println(i)
  // 结果：19678090
}

上面的误操作就是把int类型转成了string，并且修改了值导致结果出现了错误，并且这种错误

根据位移获取、修改对象的字段

利用unsafe的pointer和offsetof函数，可以获取对象的属性，并且可以修改对象的属性

type student struct {
  name string
  age int
}

func main() {
  s := student{}
  s.name = "peter"
  s.age = 33

  pstudent := unsafe.pointer(&s)
  // 整个对象转换成指针，默认是获取第一个属性
  name := (*string)(unsafe.pointer(pstudent))
  fmt.println("name:", *name)
  // 利用offsetof获取age属性的偏移量获取属性
  age := (*int)(unsafe.pointer(uintptr(pstudent) + unsafe.offsetof(s.age)))
  fmt.println("age:", *age)
  
  // 修改指针的值
  *name = "mary"
  *age = 20
  fmt.println(s)
}

获取私有变量

可以通过unsafe获取私有变量的值，也可以修改值。这个操作跟上面的获取值是一样的简单的例子如下：

type teacher struct {
  name string
  age int
}

func main() {
  t := teacher{"ttt", 20}

  pt := unsafe.pointer(&t)
  name := (*string)(unsafe.pointer(pt))
  fmt.println("name:", *name)
}

根据sizeof函数获取、修改

利用unsafe中的sizeof函数获取数组的值

func main() {
  array := []int{0, 1, -2, 3, 4}
  pointer := &array[0]
  fmt.print(*pointer, " ")
  memoryaddress := uintptr(unsafe.pointer(pointer)) + unsafe.sizeof(array[0])
  for i := 0; i < len(array)-1; i++ {
    pointer = (*int)(unsafe.pointer(memoryaddress))
    fmt.print(*pointer, " ")
    memoryaddress = uintptr(unsafe.pointer(pointer)) + unsafe.sizeof(array[0])
  }
}

结果：0 1 -2 3 4

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。