bitcask-mirror/flock/flock_unix.go

// +build !aix,!windows

package flock

import (
	"os"

	"golang.org/x/sys/unix"
)

func lock_sys(path string, nonBlocking bool) (_ *os.File, err error) {
	var fh *os.File

	fh, err = os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0666)
	if err != nil {
		return nil, err
	}

	defer func() {
		if err != nil {
			fh.Close()
		}
	}()

	flag := unix.LOCK_EX
	if nonBlocking {
		flag |= unix.LOCK_NB
	}

	err = unix.Flock(int(fh.Fd()), flag)
	if err != nil {
		return nil, err
	}

	if !sameInodes(fh, path) {
		return nil, ErrInodeChangedAtPath
	}

	return fh, nil
}

func rm_if_match(fh *os.File, path string) error {
	// Sanity check :
	// before running "rm", check that the file pointed at by the
	// filehandle has the same inode as the path in the filesystem
	//
	// If this sanity check doesn't pass, store a "ErrInodeChangedAtPath" error,
	// if the check passes, run os.Remove, and store the error if any.
	//
	// note : this sanity check is in no way atomic, but :
	//   - as long as only cooperative processes are involved, it will work as intended
	//   - it allows to avoid 99.9% the major pitfall case: "root user forcefully removed the lockfile"

	if !sameInodes(fh, path) {
		return ErrInodeChangedAtPath
	}

	return os.Remove(path)
}

func sameInodes(f *os.File, path string) bool {
	// get inode from opened file f:
	var fstat unix.Stat_t
	err := unix.Fstat(int(f.Fd()), &fstat)
	if err != nil {
		return false
	}
	fileIno := fstat.Ino

	// get inode for path on disk:
	var dstat unix.Stat_t
	err = unix.Stat(path, &dstat)
	if err != nil {
		return false
	}
	pathIno := dstat.Ino

	return pathIno == fileIno
}