2018-05-22 21:47:30 +08:00
|
|
|
// Package mmap implements a large block memory allocator using
|
|
|
|
// anonymous memory maps.
|
|
|
|
|
2021-09-09 20:25:25 +08:00
|
|
|
//go:build windows
|
2018-05-22 21:47:30 +08:00
|
|
|
|
|
|
|
package mmap
|
|
|
|
|
|
|
|
import (
|
2021-11-04 18:12:57 +08:00
|
|
|
"fmt"
|
2018-05-22 21:47:30 +08:00
|
|
|
"reflect"
|
|
|
|
"unsafe"
|
|
|
|
|
|
|
|
"golang.org/x/sys/windows"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Alloc allocates size bytes and returns a slice containing them. If
|
|
|
|
// the allocation fails it will return with an error. This is best
|
|
|
|
// used for allocations which are a multiple of the PageSize.
|
|
|
|
func Alloc(size int) ([]byte, error) {
|
|
|
|
p, err := windows.VirtualAlloc(0, uintptr(size), windows.MEM_COMMIT, windows.PAGE_READWRITE)
|
|
|
|
if err != nil {
|
2021-11-04 18:12:57 +08:00
|
|
|
return nil, fmt.Errorf("mmap: failed to allocate memory for buffer: %w", err)
|
2018-05-22 21:47:30 +08:00
|
|
|
}
|
|
|
|
var mem []byte
|
|
|
|
sh := (*reflect.SliceHeader)(unsafe.Pointer(&mem))
|
|
|
|
sh.Data = p
|
|
|
|
sh.Len = size
|
|
|
|
sh.Cap = size
|
|
|
|
return mem, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Free frees buffers allocated by Alloc. Note it should be passed
|
|
|
|
// the same slice (not a derived slice) that Alloc returned. If the
|
|
|
|
// free fails it will return with an error.
|
|
|
|
func Free(mem []byte) error {
|
|
|
|
sh := (*reflect.SliceHeader)(unsafe.Pointer(&mem))
|
|
|
|
err := windows.VirtualFree(sh.Data, 0, windows.MEM_RELEASE)
|
|
|
|
if err != nil {
|
2021-11-04 18:12:57 +08:00
|
|
|
return fmt.Errorf("mmap: failed to unmap memory: %w", err)
|
2018-05-22 21:47:30 +08:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|