如果你看go的源码,尤其是runtime的部分的源码,你一定经常会发现unsafe.Pointer和uintptr这两个函数,例如下面就是runtime里面的map源码实现里面的一个函数:
func (b *bmap) overflow(t *maptype) *bmap {
return *(**bmap)(add(unsafe.Pointer(b), uintptr(t.bucketsize)-sys.PtrSize))
}
那么这两个方法有什么用呢?下面我们来重点介绍一下。
Go中的指针及与指针对指针的操作主要有以下三种:
一普通的指针类型,例如 var intptr *T,定义一个T类型指针变量。
二内置类型uintptr,本质是一个无符号的整型,它的长度是跟平台相关的,它的长度可以用来保存一个指针地址。
三是unsafe包提供的Pointer,表示可以指向任意类型的指针。
1.普通的指针类型
count := 1
Counter(&count)
fmt.Println(count)
func Counter(count *int) {
*count++
}
普通指针可以通过引用来修改变量的值,这个跟C语言指针有点像。
2.uintptr类型
uintptr用来进行指针计算,因为它是整型,所以很容易计算出下一个指针所指向的位置。uintptr在builtin包中定义,定义如下:
// uintptr is an integer type that is large enough to hold the bit pattern of any pointer.
// uintptr是一个能足够容纳指针位数大小的整数类型
type uintptr uintptr
虽然uintpr保存了一个指针地址,但它只是一个值,不引用任何对象。因此使用的时候要注意以下情况:
1.如果uintptr地址相关联对象移动,则其值也不会更新。例如goroutine的堆栈信息发生变化
2.uintptr地址关联的对象可以被垃圾回收。GC不认为uintptr是活引用,因此unitptr地址指向的对象可以被垃圾收集。
一个uintptr可以被转换成unsafe.Pointer,同时unsafe.Pointer也可以被转换为uintptr。可以使用使用uintptr + offset计算出地址,然后使用unsafe.Pointer进行转换,格式如下:p = unsafe.Pointer(uintptr(p) + offset)
n := 10
b := make([]int, n)
for i:= 0;i< n;i++ {
b[i] = i
}
fmt.Println(b)
// [0 1 2 3 4 5 6 7 8 9]
// 取slice的最后的一个元素
end := unsafe.Pointer(uintptr(unsafe.Pointer(&b[0])) + 9 * unsafe.Sizeof(b[0]))
// 等价于unsafe.Pointer(&b[9])
fmt.Println(*(*int)(end))
// 9
3.unsafe.Pointer
unsafe.Pointer是特别定义的一种指针类型,它可以包含任意类型变量的地址。Pointer在unsafe包中定义,定义如下:
package unsafe
// ArbitraryType is here for the purposes of documentation only and is not actually
// part of the unsafe package. It represents the type of an arbitrary Go expression.
// ArbitraryType在这里不是unsafe包的实际的一部分,仅仅是为了文档记录
type ArbitraryType int
type Pointer *ArbitraryType
func Sizeof(x ArbitraryType) uintptr
func Offsetof(x ArbitraryType) uintptr
func Alignof(x ArbitraryType) uintptr
官方文档unsafe.Pointer的使用有以下说明:
Pointer represents a pointer to an arbitrary type. There are four special operations available for type Pointer that are not available for other types: // Pointer代表了一个任意类型的指针。Pointer类型有四种特殊的操作是其他类型不能使用的: - A pointer value of any type can be converted to a Pointer. // 任意类型的指针可以被转换为Pointer - A Pointer can be converted to a pointer value of any type. // Pointer可以被转换为任务类型的值的指针 - A uintptr can be converted to a Pointer. // uintptr可以被转换为Pointer - A Pointer can be converted to a uintptr. // Pointer可以被转换为uintptr Pointer therefore allows a program to defeat the type system and read and write arbitrary memory. It should be used with extreme care. // 因此Pointer允许程序不按类型系统的要求来读写任意的内存,应该非常小心地使用它。
所以unsafe.Pointer做的主要是用来进行桥接,用于不同类型的指针进行互相转换。
在任何情况下,结果都必须继续指向原分配的对象。
4.unsafe.Pointer,uintptr与普通指针的互相转换
unsafe.Pointer和普通指针的相互转换
var f float64 = 1.0 fmt.Println(Float64bits(f)) // 4607182418800017408 func Float64bits(f float64) uint64 { return *((*uint64)(unsafe.Pointer(&f))) }
借助unsafe.Pointer指针,实现float64转换为uint64类型。当然,我们不可以直接通过
*p
来获取unsafe.Pointer指针指向的真实变量的值,因为我们并不知道变量的具体类型。另外一个重要的要注意的是,在进行普通类型转换的时候,要注意转换的前后的类型要有相同的内存布局,下面两个结构也能完成转换,就因为他们有相同的内存布局type s1 struct { id int name string } type s2 struct { field1 *[5]byte filed2 int } b := s1{name:"123"} var j s2 j = *(*s2)(unsafe.Pointer(&b)) fmt.Println(j)
- unsafe.Pointer和uintrptr的互相转换及配合
uintptr类型的主要是用来与unsafe.Pointer配合使用来访问和操作unsafe的内存。unsafe.Pointer不能执行算术操作。要想对指针进行算术运算必须这样来做:
1.将unsafe.Pointer转换为uintptr
2.对uintptr执行算术运算
3.将uintptr转换回unsafe.Pointer,然后访问uintptr地址指向的对象
需要小心的是,上面的步骤对于垃圾收集器来说应该是原子的,否则可能会导致问题。
例如,在第1步之后,引用的对象可能被收集。如果在步骤3之后发生这种情况,指针将是一个无效的Go指针,并可能导致程序崩溃
unsafe.Pointer和uintrptr的转换
package main
import (
"fmt"
"unsafe"
)
type Person struct {
age int
name string
}
func main() {
p := &Person{age: 30, name: "Bob"}
//获取到struct s中b字段的地址
p := unsafe.Pointer(uintptr(unsafe.Pointer(p)) + unsafe.Offsetof(p.name))
//将其转换为一个string的指针,并且打印该指针的对应的值
fmt.Println(*(*string)(p))
}