2017-04-10 11:59:45 -07:00
|
|
|
// Copyright 2017 The Prometheus Authors
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
2017-11-30 06:34:49 -08:00
|
|
|
package fileutil
|
2017-04-04 06:59:52 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
|
|
|
"syscall"
|
|
|
|
"unsafe"
|
|
|
|
)
|
|
|
|
|
2018-03-14 00:53:33 -07:00
|
|
|
func mmap(f *os.File, size int) ([]byte, error) {
|
|
|
|
low, high := uint32(size), uint32(size>>32)
|
2017-10-31 07:37:41 -07:00
|
|
|
h, errno := syscall.CreateFileMapping(syscall.Handle(f.Fd()), nil, syscall.PAGE_READONLY, high, low, nil)
|
2017-04-04 06:59:52 -07:00
|
|
|
if h == 0 {
|
2017-04-05 05:14:30 -07:00
|
|
|
return nil, os.NewSyscallError("CreateFileMapping", errno)
|
2017-04-04 06:59:52 -07:00
|
|
|
}
|
|
|
|
|
2018-03-14 00:53:33 -07:00
|
|
|
addr, errno := syscall.MapViewOfFile(h, syscall.FILE_MAP_READ, 0, 0, uintptr(size))
|
2017-04-04 06:59:52 -07:00
|
|
|
if addr == 0 {
|
2017-04-05 05:14:30 -07:00
|
|
|
return nil, os.NewSyscallError("MapViewOfFile", errno)
|
2017-04-04 06:59:52 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if err := syscall.CloseHandle(syscall.Handle(h)); err != nil {
|
2017-04-05 05:14:30 -07:00
|
|
|
return nil, os.NewSyscallError("CloseHandle", err)
|
2017-04-04 06:59:52 -07:00
|
|
|
}
|
|
|
|
|
2018-03-14 00:53:33 -07:00
|
|
|
return (*[maxMapSize]byte)(unsafe.Pointer(addr))[:size], nil
|
2017-04-04 06:59:52 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func munmap(b []byte) error {
|
|
|
|
if err := syscall.UnmapViewOfFile((uintptr)(unsafe.Pointer(&b[0]))); err != nil {
|
|
|
|
return os.NewSyscallError("UnmapViewOfFile", err)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|