forked from hardcoresushi/DroidFS
154 lines
4.4 KiB
Go
154 lines
4.4 KiB
Go
// Package syscallcompat wraps Linux-specific syscalls.
|
|
package syscallcompat
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
const (
|
|
_FALLOC_FL_KEEP_SIZE = 0x01
|
|
|
|
// O_DIRECT means oncached I/O on Linux. No direct equivalent on MacOS and defined
|
|
// to zero there.
|
|
O_DIRECT = syscall.O_DIRECT
|
|
|
|
// O_PATH is only defined on Linux
|
|
O_PATH = unix.O_PATH
|
|
)
|
|
|
|
var preallocWarn sync.Once
|
|
|
|
// EnospcPrealloc preallocates ciphertext space without changing the file
|
|
// size. This guarantees that we don't run out of space while writing a
|
|
// ciphertext block (that would corrupt the block).
|
|
func EnospcPrealloc(fd int, off int64, len int64) (err error) {
|
|
for {
|
|
err = syscall.Fallocate(fd, _FALLOC_FL_KEEP_SIZE, off, len)
|
|
if err == syscall.EINTR {
|
|
// fallocate, like many syscalls, can return EINTR. This is not an
|
|
// error and just signifies that the operation was interrupted by a
|
|
// signal and we should try again.
|
|
continue
|
|
}
|
|
if err == syscall.EOPNOTSUPP {
|
|
// ZFS and ext3 do not support fallocate. Warn but continue anyway.
|
|
// https://github.com/rfjakob/gocryptfs/issues/22
|
|
preallocWarn.Do(func() {})
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Fallocate wraps the Fallocate syscall.
|
|
func Fallocate(fd int, mode uint32, off int64, len int64) (err error) {
|
|
return syscall.Fallocate(fd, mode, off, len)
|
|
}
|
|
|
|
func getSupplementaryGroups(pid uint32) (gids []int) {
|
|
procPath := fmt.Sprintf("/proc/%d/task/%d/status", pid, pid)
|
|
blob, err := ioutil.ReadFile(procPath)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
lines := strings.Split(string(blob), "\n")
|
|
for _, line := range lines {
|
|
if strings.HasPrefix(line, "Groups:") {
|
|
f := strings.Fields(line[7:])
|
|
gids = make([]int, len(f))
|
|
for i := range gids {
|
|
val, err := strconv.ParseInt(f[i], 10, 32)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
gids[i] = int(val)
|
|
}
|
|
return gids
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Mknodat wraps the Mknodat syscall.
|
|
func Mknodat(dirfd int, path string, mode uint32, dev int) (err error) {
|
|
return syscall.Mknodat(dirfd, path, mode, dev)
|
|
}
|
|
|
|
// Dup3 wraps the Dup3 syscall. We want to use Dup3 rather than Dup2 because Dup2
|
|
// is not implemented on arm64.
|
|
func Dup3(oldfd int, newfd int, flags int) (err error) {
|
|
return syscall.Dup3(oldfd, newfd, flags)
|
|
}
|
|
|
|
// FchmodatNofollow is like Fchmodat but never follows symlinks.
|
|
//
|
|
// This should be handled by the AT_SYMLINK_NOFOLLOW flag, but Linux
|
|
// does not implement it, so we have to perform an elaborate dance
|
|
// with O_PATH and /proc/self/fd.
|
|
//
|
|
// See also: Qemu implemented the same logic as fchmodat_nofollow():
|
|
// https://git.qemu.org/?p=qemu.git;a=blob;f=hw/9pfs/9p-local.c#l335
|
|
func FchmodatNofollow(dirfd int, path string, mode uint32) (err error) {
|
|
// Open handle to the filename (but without opening the actual file).
|
|
// This succeeds even when we don't have read permissions to the file.
|
|
fd, err := syscall.Openat(dirfd, path, syscall.O_NOFOLLOW|O_PATH, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer syscall.Close(fd)
|
|
|
|
// Now we can check the type without the risk of race-conditions.
|
|
// Return syscall.ELOOP if it is a symlink.
|
|
var st syscall.Stat_t
|
|
err = syscall.Fstat(fd, &st)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if st.Mode&syscall.S_IFMT == syscall.S_IFLNK {
|
|
return syscall.ELOOP
|
|
}
|
|
|
|
// Change mode of the actual file. Fchmod does not work with O_PATH,
|
|
// but Chmod via /proc/self/fd works.
|
|
procPath := fmt.Sprintf("/proc/self/fd/%d", fd)
|
|
return syscall.Chmod(procPath, mode)
|
|
}
|
|
|
|
func timesToTimespec(a *time.Time, m *time.Time) []unix.Timespec {
|
|
ts := make([]unix.Timespec, 2)
|
|
ta, _ := unix.TimeToTimespec(*a)
|
|
ts[0] = unix.Timespec(ta)
|
|
tm, _ := unix.TimeToTimespec(*m)
|
|
ts[1] = unix.Timespec(tm)
|
|
return ts
|
|
}
|
|
|
|
// FutimesNano syscall.
|
|
func FutimesNano(fd int, a *time.Time, m *time.Time) (err error) {
|
|
ts := timesToTimespec(a, m)
|
|
// To avoid introducing a separate syscall wrapper for futimens()
|
|
// (as done in go-fuse, for example), we instead use the /proc/self/fd trick.
|
|
procPath := fmt.Sprintf("/proc/self/fd/%d", fd)
|
|
return unix.UtimesNanoAt(unix.AT_FDCWD, procPath, ts, 0)
|
|
}
|
|
|
|
// UtimesNanoAtNofollow is like UtimesNanoAt but never follows symlinks.
|
|
func UtimesNanoAtNofollow(dirfd int, path string, a *time.Time, m *time.Time) (err error) {
|
|
ts := timesToTimespec(a, m)
|
|
return unix.UtimesNanoAt(dirfd, path, ts, unix.AT_SYMLINK_NOFOLLOW)
|
|
}
|
|
|
|
func Getdents(fd int) ([]DirEntry, error) {
|
|
return getdents(fd)
|
|
}
|