Merge pull request #301 from Gouthamve/fix-prom-3957

Use the right sized byte array for large indexes
This commit is contained in:
Goutham Veeramachaneni 2018-03-14 18:17:08 +05:30 committed by GitHub
commit 902e1ffad4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions

View File

@ -19,14 +19,16 @@ import (
"unsafe" "unsafe"
) )
func mmap(f *os.File, sz int) ([]byte, error) { const maxMapSize = 0xFFFFFFFFFFFF // 256TB
low, high := uint32(sz), uint32(sz>>32)
func mmap(f *os.File, size int) ([]byte, error) {
low, high := uint32(size), uint32(size>>32)
h, errno := syscall.CreateFileMapping(syscall.Handle(f.Fd()), nil, syscall.PAGE_READONLY, high, low, nil) h, errno := syscall.CreateFileMapping(syscall.Handle(f.Fd()), nil, syscall.PAGE_READONLY, high, low, nil)
if h == 0 { if h == 0 {
return nil, os.NewSyscallError("CreateFileMapping", errno) return nil, os.NewSyscallError("CreateFileMapping", errno)
} }
addr, errno := syscall.MapViewOfFile(h, syscall.FILE_MAP_READ, 0, 0, uintptr(sz)) addr, errno := syscall.MapViewOfFile(h, syscall.FILE_MAP_READ, 0, 0, uintptr(size))
if addr == 0 { if addr == 0 {
return nil, os.NewSyscallError("MapViewOfFile", errno) return nil, os.NewSyscallError("MapViewOfFile", errno)
} }
@ -35,7 +37,7 @@ func mmap(f *os.File, sz int) ([]byte, error) {
return nil, os.NewSyscallError("CloseHandle", err) return nil, os.NewSyscallError("CloseHandle", err)
} }
return (*[1 << 30]byte)(unsafe.Pointer(addr))[:sz], nil return (*[maxMapSize]byte)(unsafe.Pointer(addr))[:size], nil
} }
func munmap(b []byte) error { func munmap(b []byte) error {