141 lines
2.7 KiB
Go
141 lines
2.7 KiB
Go
|
// +build windows
|
||
|
|
|
||
|
|
package state
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
"os"
|
||
|
|
"syscall"
|
||
|
|
"unsafe"
|
||
|
|
)
|
||
|
|
|
||
|
|
type stateLock struct {
|
||
|
|
handle syscall.Handle
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
modkernel32 = syscall.NewLazyDLL("kernel32.dll")
|
||
|
|
procLockFileEx = modkernel32.NewProc("LockFileEx")
|
||
|
|
procCreateEventW = modkernel32.NewProc("CreateEventW")
|
||
|
|
|
||
|
|
lockedFiles = map[*os.File]syscall.Handle{}
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
_LOCKFILE_FAIL_IMMEDIATELY = 1
|
||
|
|
_LOCKFILE_EXCLUSIVE_LOCK = 2
|
||
|
|
)
|
||
|
|
|
||
|
|
func (s *LocalState) lock() error {
|
||
|
|
name, err := syscall.UTF16PtrFromString(s.PathOut)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
handle, err := syscall.CreateFile(
|
||
|
|
name,
|
||
|
|
syscall.GENERIC_READ|syscall.GENERIC_WRITE,
|
||
|
|
// since this file is already open in out process, we need shared
|
||
|
|
// access here for this call.
|
||
|
|
syscall.FILE_SHARE_READ|syscall.FILE_SHARE_WRITE,
|
||
|
|
nil,
|
||
|
|
syscall.OPEN_EXISTING,
|
||
|
|
syscall.FILE_ATTRIBUTE_NORMAL,
|
||
|
|
0,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
lockedFiles[s.stateFileOut] = handle
|
||
|
|
|
||
|
|
// even though we're failing immediately, an overlapped event structure is
|
||
|
|
// required
|
||
|
|
ol, err := newOverlapped()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer syscall.CloseHandle(ol.HEvent)
|
||
|
|
|
||
|
|
return lockFileEx(
|
||
|
|
handle,
|
||
|
|
_LOCKFILE_EXCLUSIVE_LOCK|_LOCKFILE_FAIL_IMMEDIATELY,
|
||
|
|
0, // reserved
|
||
|
|
0, // bytes low
|
||
|
|
math.MaxUint32, // bytes high
|
||
|
|
ol,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *LocalState) unlock() error {
|
||
|
|
handle, ok := lockedFiles[s.stateFileOut]
|
||
|
|
if !ok {
|
||
|
|
// we allow multiple Unlock calls
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
delete(lockedFiles, s.stateFileOut)
|
||
|
|
return syscall.Close(handle)
|
||
|
|
}
|
||
|
|
|
||
|
|
func lockFileEx(h syscall.Handle, flags, reserved, locklow, lockhigh uint32, ol *syscall.Overlapped) (err error) {
|
||
|
|
r1, _, e1 := syscall.Syscall6(
|
||
|
|
procLockFileEx.Addr(),
|
||
|
|
6,
|
||
|
|
uintptr(h),
|
||
|
|
uintptr(flags),
|
||
|
|
uintptr(reserved),
|
||
|
|
uintptr(locklow),
|
||
|
|
uintptr(lockhigh),
|
||
|
|
uintptr(unsafe.Pointer(ol)),
|
||
|
|
)
|
||
|
|
if r1 == 0 {
|
||
|
|
if e1 != 0 {
|
||
|
|
err = error(e1)
|
||
|
|
} else {
|
||
|
|
err = syscall.EINVAL
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// newOverlapped creates a structure used to track asynchronous
|
||
|
|
// I/O requests that have been issued.
|
||
|
|
func newOverlapped() (*syscall.Overlapped, error) {
|
||
|
|
event, err := createEvent(nil, true, false, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &syscall.Overlapped{HEvent: event}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func createEvent(sa *syscall.SecurityAttributes, manualReset bool, initialState bool, name *uint16) (handle syscall.Handle, err error) {
|
||
|
|
var _p0 uint32
|
||
|
|
if manualReset {
|
||
|
|
_p0 = 1
|
||
|
|
}
|
||
|
|
var _p1 uint32
|
||
|
|
if initialState {
|
||
|
|
_p1 = 1
|
||
|
|
}
|
||
|
|
|
||
|
|
r0, _, e1 := syscall.Syscall6(
|
||
|
|
procCreateEventW.Addr(),
|
||
|
|
4,
|
||
|
|
uintptr(unsafe.Pointer(sa)),
|
||
|
|
uintptr(_p0),
|
||
|
|
uintptr(_p1),
|
||
|
|
uintptr(unsafe.Pointer(name)),
|
||
|
|
0,
|
||
|
|
0,
|
||
|
|
)
|
||
|
|
handle = syscall.Handle(r0)
|
||
|
|
if handle == syscall.InvalidHandle {
|
||
|
|
if e1 != 0 {
|
||
|
|
err = error(e1)
|
||
|
|
} else {
|
||
|
|
err = syscall.EINVAL
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|