C语言中的memset
在C/C++中,有一个memset函数,常常用来清空一段内存。
#include <string.h>
// 将s的前n个字节用c填充
void *memset(void *s, int c, size_t n);
// 用法举例
int arr[100];
// 将数组内存全部置为0
memset(&arr, 0, sizeof(arr));
由于是s是void* 类型的指针,那么任何类型的内存都可以清空。但是在golang中,类型检查特别严格,并没有提供类似的函数。
Golang中一种实现
// 可以填充数组、结构体
func MemSet(s unsafe.Pointer, c byte, n uintptr) {
ptr := uintptr(s)
var i uintptr
for i = 0; i < n; i++ {
pByte := (*byte)(unsafe.Pointer(ptr + i))
*pByte = c
}
}
func main() {
var arr [10]int32
MemSet(unsafe.Pointer(&arr), 1, unsafe.Sizeof(arr))
fmt.Printf("%+v\n", arr)
MemSet(unsafe.Pointer(&arr), 0, unsafe.Sizeof(arr))
fmt.Printf("%+v\n", arr)
}
// 输出:
[KentZhang@LOCAL-192-168-97-2]$ go run memset.go
[16843009 16843009 16843009 16843009 16843009 16843009 16843009 16843009 16843009 16843009]
[0 0 0 0 0 0 0 0 0 0]
三点说明:
- unsafe.Pointer 是一种通用的指针类型,可以将任何数据类型的指针转化为 unsafe.Pointer,相当于C语言中的void*指针。
- unsafe.Pointer不能进行计算,即指针不能加减偏移计算,所以还必须转化为uintptr类型,uintptr本质上是一个整数,于是就可以计算了。
- uintptr是整数,不是指针,无法进行取值赋值运算,还要转化为unsafe.Pointer,再转为对应类型的指针。