rclone/lib/mmap/mmap_windows.go

52 lines
1.5 KiB
Go
Raw Normal View History

// Package mmap implements a large block memory allocator using
// anonymous memory maps.
2021-09-09 20:25:25 +08:00
//go:build windows
package mmap
import (
"fmt"
"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 {
return nil, fmt.Errorf("mmap: failed to allocate memory for buffer: %w", err)
}
// SliceHeader is deprecated...
var mem []byte
sh := (*reflect.SliceHeader)(unsafe.Pointer(&mem)) // nolint:staticcheck
sh.Data = p
sh.Len = size
sh.Cap = size
return mem, nil
// ...However the correct code gives a go vet warning
// "possible misuse of unsafe.Pointer"
//
// Maybe there is a different way of writing this, but none of
// the allowed uses of unsafe.Pointer seemed to cover it other
// than using SliceHeader (use 6 of unsafe.Pointer).
//
// return unsafe.Slice((*byte)(unsafe.Pointer(p)), size), 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 {
p := unsafe.SliceData(mem)
err := windows.VirtualFree(uintptr(unsafe.Pointer(p)), 0, windows.MEM_RELEASE)
if err != nil {
return fmt.Errorf("mmap: failed to unmap memory: %w", err)
}
return nil
}