mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-11-04 06:42:26 -06:00
d2f13e7564
- update gruf/go-stroage v0.2.0 -> v0.2.1 - update KimMachineGun/automemlimit v0.7.1 -> v0.7.2 - update miekg/dns v1.1.65 -> v1.1.66 - update ncruces/go-sqlite3 v0.25.1 -> v0.25.2 - update spf13/cast v1.7.1 -> v1.8.0 - update tdewolff/minify/v2 v2.23.1 -> v2.23.5 - update x/crypto v0.37.0 -> v0.38.0 - update x/image v0.26.0 -> v0.27.0 - update x/net v0.39.0 -> v0.40.0 - update x/oauth2 v0.29.0 -> v0.30.0 - update x/sys v0.32.0 -> v0.33.0 - update x/text v0.24.0 -> v0.25.0 Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4162 Co-authored-by: kim <grufwub@gmail.com> Co-committed-by: kim <grufwub@gmail.com>
91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
//go:build unix
|
|
|
|
package parse
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"syscall"
|
|
)
|
|
|
|
type binaryReaderMmap struct {
|
|
data []byte
|
|
}
|
|
|
|
func newBinaryReaderMmap(filename string) (*binaryReaderMmap, error) {
|
|
f, err := os.Open(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
info, err := f.Stat()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
size := info.Size()
|
|
if size == 0 {
|
|
// Treat (size == 0) as a special case, avoiding the syscall, since
|
|
// "man 2 mmap" says "the length... must be greater than 0".
|
|
//
|
|
// As we do not call syscall.Mmap, there is no need to call
|
|
// runtime.SetFinalizer to enforce a balancing syscall.Munmap.
|
|
return &binaryReaderMmap{
|
|
data: make([]byte, 0),
|
|
}, nil
|
|
} else if size < 0 {
|
|
return nil, fmt.Errorf("mmap: file %s has negative size", filename)
|
|
} else if size != int64(int(size)) {
|
|
return nil, fmt.Errorf("mmap: file %s is too large", filename)
|
|
}
|
|
|
|
data, err := syscall.Mmap(int(f.Fd()), 0, int(size), syscall.PROT_READ, syscall.MAP_SHARED)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r := &binaryReaderMmap{data}
|
|
runtime.SetFinalizer(r, (*binaryReaderMmap).Close)
|
|
return r, nil
|
|
}
|
|
|
|
// Close closes the reader.
|
|
func (r *binaryReaderMmap) Close() error {
|
|
if r.data == nil {
|
|
return nil
|
|
} else if len(r.data) == 0 {
|
|
r.data = nil
|
|
return nil
|
|
}
|
|
data := r.data
|
|
r.data = nil
|
|
runtime.SetFinalizer(r, nil)
|
|
return syscall.Munmap(data)
|
|
}
|
|
|
|
// Len returns the length of the underlying memory-mapped file.
|
|
func (r *binaryReaderMmap) Len() int {
|
|
return len(r.data)
|
|
}
|
|
|
|
func (r *binaryReaderMmap) Bytes(n int, off int64) ([]byte, error) {
|
|
if r.data == nil {
|
|
return nil, errors.New("mmap: closed")
|
|
} else if off < 0 || int64(len(r.data)) < off {
|
|
return nil, fmt.Errorf("mmap: invalid offset %d", off)
|
|
} else if int64(len(r.data)-n) < off {
|
|
return r.data[off:len(r.data):len(r.data)], io.EOF
|
|
}
|
|
return r.data[off : off+int64(n) : off+int64(n)], nil
|
|
}
|
|
|
|
func NewBinaryReader2Mmap(filename string) (*BinaryReader2, error) {
|
|
f, err := newBinaryReaderMmap(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewBinaryReader2(f), nil
|
|
}
|