实现golang中memset函数

原文链接：实现golang中memset函数
发布时间：2018-12-16 12:11:11

C语言中的memset

在C/C++中，有一个memset函数，常常用来清空一段内存。

1
2
3
4
5
6
7
8
9
```cpp
#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中一种实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
```go
// 可以填充数组、结构体
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，再转为对应类型的指针。