diff options
author | 2024-12-16 11:37:53 +0000 | |
---|---|---|
committer | 2024-12-16 11:37:53 +0000 | |
commit | 9697df0955095cb9be7b5fa0c4408ae4523d8b08 (patch) | |
tree | 737d826b4ea8bc22c0645cbbea7ab143428e0bda /vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go | |
parent | bump gruf/go-ffmpreg to v0.6.3 (#3624) (diff) | |
download | gotosocial-9697df0955095cb9be7b5fa0c4408ae4523d8b08.tar.xz |
bump ncruces/go-sqlite3 to v0.21.1 (#3625)
Diffstat (limited to 'vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go')
-rw-r--r-- | vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go | 96 |
1 files changed, 53 insertions, 43 deletions
diff --git a/vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go b/vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go index 2872201b0..0398f4760 100644 --- a/vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go +++ b/vendor/github.com/ncruces/go-sqlite3/vfs/os_windows.go @@ -45,6 +45,7 @@ func osGetExclusiveLock(file *os.File, state *LockLevel) _ErrorCode { osUnlock(file, _SHARED_FIRST, _SHARED_SIZE) // Acquire the EXCLUSIVE lock. + // Can't wait here, because the file is not OVERLAPPED. rc := osWriteLock(file, _SHARED_FIRST, _SHARED_SIZE, 0) if rc != _OK { @@ -106,64 +107,73 @@ func osCheckReservedLock(file *os.File) (bool, _ErrorCode) { return false, rc } -func osUnlock(file *os.File, start, len uint32) _ErrorCode { - err := windows.UnlockFileEx(windows.Handle(file.Fd()), - 0, len, 0, &windows.Overlapped{Offset: start}) - if err == windows.ERROR_NOT_LOCKED { - return _OK - } - if err != nil { - return _IOERR_UNLOCK - } - return _OK +func osReadLock(file *os.File, start, len uint32, timeout time.Duration) _ErrorCode { + return osLock(file, 0, start, len, timeout, _IOERR_RDLOCK) +} + +func osWriteLock(file *os.File, start, len uint32, timeout time.Duration) _ErrorCode { + return osLock(file, windows.LOCKFILE_EXCLUSIVE_LOCK, start, len, timeout, _IOERR_LOCK) } func osLock(file *os.File, flags, start, len uint32, timeout time.Duration, def _ErrorCode) _ErrorCode { var err error switch { case timeout == 0: - err = osLockEx(file, flags|windows.LOCKFILE_FAIL_IMMEDIATELY, start, len, 0) + err = osLockEx(file, flags|windows.LOCKFILE_FAIL_IMMEDIATELY, start, len) case timeout < 0: - err = osLockEx(file, flags, start, len, 0) + err = osLockEx(file, flags, start, len) default: - var event windows.Handle - event, err = windows.CreateEvent(nil, 1, 0, nil) - if err != nil { - break - } - defer windows.CloseHandle(event) - - err = osLockEx(file, flags, start, len, event) - if err == windows.ERROR_IO_PENDING { - rc, serr := windows.WaitForSingleObject(event, uint32(timeout/time.Millisecond)) - if rc == windows.WAIT_OBJECT_0 { - return _OK - } - if serr != nil { - err = serr - } else { - err = windows.Errno(rc) - } - windows.CancelIo(windows.Handle(file.Fd())) - } + err = osLockExTimeout(file, flags, start, len, timeout) } return osLockErrorCode(err, def) } -func osLockEx(file *os.File, flags, start, len uint32, event windows.Handle) error { - return windows.LockFileEx(windows.Handle(file.Fd()), flags, - 0, len, 0, &windows.Overlapped{ - Offset: start, - HEvent: event, - }) +func osUnlock(file *os.File, start, len uint32) _ErrorCode { + err := windows.UnlockFileEx(windows.Handle(file.Fd()), + 0, len, 0, &windows.Overlapped{Offset: start}) + if err == windows.ERROR_NOT_LOCKED { + return _OK + } + if err != nil { + return _IOERR_UNLOCK + } + return _OK } -func osReadLock(file *os.File, start, len uint32, timeout time.Duration) _ErrorCode { - return osLock(file, 0, start, len, timeout, _IOERR_RDLOCK) +func osLockEx(file *os.File, flags, start, len uint32) error { + return windows.LockFileEx(windows.Handle(file.Fd()), flags, + 0, len, 0, &windows.Overlapped{Offset: start}) } -func osWriteLock(file *os.File, start, len uint32, timeout time.Duration) _ErrorCode { - return osLock(file, windows.LOCKFILE_EXCLUSIVE_LOCK, start, len, timeout, _IOERR_LOCK) +func osLockExTimeout(file *os.File, flags, start, len uint32, timeout time.Duration) error { + event, err := windows.CreateEvent(nil, 1, 0, nil) + if err != nil { + return err + } + defer windows.CloseHandle(event) + + fd := windows.Handle(file.Fd()) + overlapped := &windows.Overlapped{ + Offset: start, + HEvent: event, + } + + err = windows.LockFileEx(fd, flags, 0, len, 0, overlapped) + if err != windows.ERROR_IO_PENDING { + return err + } + + ms := (timeout + time.Millisecond - 1) / time.Millisecond + rc, err := windows.WaitForSingleObject(event, uint32(ms)) + if rc == windows.WAIT_OBJECT_0 { + return nil + } + defer windows.CancelIoEx(fd, overlapped) + + if err != nil { + return err + } + return windows.Errno(rc) } func osLockErrorCode(err error, def _ErrorCode) _ErrorCode { @@ -175,8 +185,8 @@ func osLockErrorCode(err error, def _ErrorCode) _ErrorCode { switch errno { case windows.ERROR_LOCK_VIOLATION, - windows.ERROR_IO_PENDING, windows.ERROR_OPERATION_ABORTED, + windows.ERROR_IO_PENDING, windows.WAIT_TIMEOUT: return _BUSY } |