File lock (#4734)

* add RWLock implementation for file [posix] * add implementation FileStreamRWLock for windows replace FileStreamRWLock to the Process package * add files FileStreamRWLock* into makefile and vcproj * remove unnecessary file from makefile * use absolute path to the TesApp with ProcessRunner * fix vc*.proj * add new test files into vc.proj.filters * fix comments * fix spelling fo PR #4723 add atomic_bool _locked and check if FileStreamRWLock is locked on destruction for force unlock * add atomic header * File lock (#4740) * throw error on any errno not only on EDEADLK * fix function naming * fix windows build * fix windows build --------- Co-authored-by: Alexander B <ale.bychuk@gmail.com> Co-authored-by: bas524 <bas524@ya.ru>
2025-10-26 18:42:41 +01:00 · 2024-10-16 16:58:00 -05:00
parent 6f34ec89ce
commit e5752a5c6d
18 changed files with 1141 additions and 8 deletions
--- a/Foundation/include/Poco/FileStreamRWLock_WIN32.h
+++ b/Foundation/include/Poco/FileStreamRWLock_WIN32.h
@@ -0,0 +1,134 @@
+//
+// FileStreamRWLock_WIN32.h
+//
+// Library: Foundation
+// Package: Processes
+// Module:  FileStreamRWLock
+//
+// Definition of the FileStreamRWLockImpl class for WIN32 FileStream.
+//
+// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
+// and Contributors.
+//
+// SPDX-License-Identifier:	BSL-1.0
+//
+
+
+#ifndef Foundation_FileStreamRWLock_WIN32_INCLUDED
+#define Foundation_FileStreamRWLock_WIN32_INCLUDED
+
+
+#include "Poco/Foundation.h"
+#include "Poco/Exception.h"
+#include "Poco/FileStream.h"
+#include <Windows.h>
+
+namespace Poco {
+
+struct LockMode
+{
+	static constexpr DWORD READ = 0;
+	static constexpr DWORD WRITE = LOCKFILE_EXCLUSIVE_LOCK;
+	static constexpr DWORD TRY_READ = LOCKFILE_FAIL_IMMEDIATELY;
+	static constexpr DWORD TRY_WRITE = (LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY);
+};
+
+class Foundation_API FileStreamRWLockImpl
+{
+protected:
+	FileStreamRWLockImpl(const FileStream::NativeHandle &fd, Poco::UInt64 offset, Poco::UInt64 size);
+	~FileStreamRWLockImpl();
+	void readLockImpl();
+	bool tryReadLockImpl();
+	void writeLockImpl();
+	bool tryWriteLockImpl();
+	void unlockImpl();
+
+private:
+	FileStream::NativeHandle _fd;
+	OVERLAPPED _overlapped;
+	LARGE_INTEGER _size;
+};
+
+
+//
+// inlines
+//
+inline void FileStreamRWLockImpl::readLockImpl()
+{
+	BOOL fSuccess = LockFileEx(_fd, LockMode::READ, 0, _size.LowPart, _size.HighPart, &_overlapped);
+	if (!fSuccess)
+	{
+		throw SystemException("cannot lock reader lock", GetLastError());
+	}
+}
+
+
+inline bool FileStreamRWLockImpl::tryReadLockImpl()
+{
+	BOOL fSuccess = LockFileEx(_fd, LockMode::TRY_READ, 0, _size.LowPart, _size.HighPart, &_overlapped);
+	if (fSuccess)
+	{
+		return true;
+	}
+	else
+	{
+		auto lastError = GetLastError();
+		if (lastError == ERROR_IO_PENDING || lastError == ERROR_LOCK_VIOLATION)
+		{
+			return false;
+		}
+		else
+		{
+			throw SystemException("cannot lock try-reader lock", lastError);
+		}
+	}
+}
+
+
+inline void FileStreamRWLockImpl::writeLockImpl()
+{
+	BOOL fSuccess = LockFileEx(_fd, LockMode::WRITE, 0, _size.LowPart, _size.HighPart, &_overlapped);
+	if (!fSuccess)
+	{
+		throw SystemException("cannot lock writer lock", GetLastError());
+	}
+}
+
+
+inline bool FileStreamRWLockImpl::tryWriteLockImpl()
+{
+	BOOL fSuccess = LockFileEx(_fd, LockMode::TRY_WRITE, 0, _size.LowPart, _size.HighPart, &_overlapped);
+	if (fSuccess)
+	{
+		return true;
+	}
+	else 
+	{
+		auto lastError = GetLastError();
+		if (lastError == ERROR_IO_PENDING || lastError == ERROR_LOCK_VIOLATION)
+		{
+			return false;
+		}
+		else
+		{
+			throw SystemException("cannot lock try-writer lock", lastError);
+		}
+	}
+}
+
+
+inline void FileStreamRWLockImpl::unlockImpl()
+{
+	BOOL fSuccess = UnlockFileEx(_fd, 0, _size.LowPart, _size.HighPart, &_overlapped);
+	if (!fSuccess)
+	{
+		throw SystemException("cannot unlock ", GetLastError());
+	}
+}
+
+
+} // namespace Poco
+
+
+#endif // Foundation_FileStreamRWLock_WIN32_INCLUDED