diff --git a/build/VStudio/testing/winfsp-tests.vcxproj b/build/VStudio/testing/winfsp-tests.vcxproj
index 36c2114d..98fcfd3b 100644
--- a/build/VStudio/testing/winfsp-tests.vcxproj
+++ b/build/VStudio/testing/winfsp-tests.vcxproj
@@ -184,6 +184,7 @@
+
diff --git a/build/VStudio/testing/winfsp-tests.vcxproj.filters b/build/VStudio/testing/winfsp-tests.vcxproj.filters
index 3523d99f..417227fc 100644
--- a/build/VStudio/testing/winfsp-tests.vcxproj.filters
+++ b/build/VStudio/testing/winfsp-tests.vcxproj.filters
@@ -49,6 +49,9 @@
Source
+
+ Source
+
diff --git a/tst/winfsp-tests/lock-test.c b/tst/winfsp-tests/lock-test.c
new file mode 100644
index 00000000..7126ebb2
--- /dev/null
+++ b/tst/winfsp-tests/lock-test.c
@@ -0,0 +1,542 @@
+#include
+#include
+#include
+#include
+#include
+#include "memfs.h"
+
+void *memfs_start_ex(ULONG Flags, ULONG FileInfoTimeout);
+void *memfs_start(ULONG Flags);
+void memfs_stop(void *data);
+PWSTR memfs_volumename(void *data);
+
+extern int NtfsTests;
+extern int WinFspDiskTests;
+extern int WinFspNetTests;
+
+static void lock_dotest(ULONG Flags, PWSTR VolPrefix, PWSTR Prefix, ULONG FileInfoTimeout, DWORD CreateFlags)
+{
+ /* this is not a true locking test since you need 2 processes to really test locks */
+
+ void *memfs = memfs_start_ex(Flags, FileInfoTimeout);
+
+ HANDLE Handle;
+ BOOL Success;
+ WCHAR FilePath[MAX_PATH];
+ SYSTEM_INFO SystemInfo;
+ DWORD SectorsPerCluster;
+ DWORD BytesPerSector;
+ DWORD FreeClusters;
+ DWORD TotalClusters;
+ PVOID AllocBuffer[2], Buffer[2];
+ ULONG AllocBufferSize;
+ DWORD BytesTransferred;
+ DWORD FilePointer;
+
+ GetSystemInfo(&SystemInfo);
+
+ StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\",
+ VolPrefix ? L"" : L"\\\\?\\GLOBALROOT", VolPrefix ? VolPrefix : memfs_volumename(memfs));
+
+ Success = GetDiskFreeSpaceW(FilePath, &SectorsPerCluster, &BytesPerSector, &FreeClusters, &TotalClusters);
+ ASSERT(Success);
+ AllocBufferSize = 16 * SystemInfo.dwPageSize;
+
+ AllocBuffer[0] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
+ AllocBuffer[1] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
+ ASSERT(0 != AllocBuffer[0] && 0 != AllocBuffer[1]);
+
+ srand((unsigned)time(0));
+ for (PUINT8 Bgn = AllocBuffer[0], End = Bgn + AllocBufferSize; End > Bgn; Bgn++)
+ *Bgn = rand();
+
+ Buffer[0] = (PVOID)((PUINT8)AllocBuffer[0] + BytesPerSector);
+ Buffer[1] = (PVOID)((PUINT8)AllocBuffer[1] + BytesPerSector);
+
+ StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0",
+ Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ CREATE_NEW, FILE_ATTRIBUTE_NORMAL | CreateFlags, 0);
+ ASSERT(INVALID_HANDLE_VALUE != Handle);
+
+ Success = LockFile(Handle, BytesPerSector / 2, 0, BytesPerSector, 0);
+ ASSERT(Success);
+
+ Success = LockFile(Handle, BytesPerSector + BytesPerSector / 2, 0, BytesPerSector, 0);
+ ASSERT(Success);
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+
+ FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
+ ASSERT(2 * BytesPerSector == FilePointer);
+ Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
+ ASSERT(2 * BytesPerSector == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
+ ASSERT(2 * BytesPerSector == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ FilePointer = SetFilePointer(Handle, 3 * BytesPerSector, 0, FILE_BEGIN);
+ ASSERT(3 * BytesPerSector == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(0 == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Buffer[0] = AllocBuffer[0];
+ Buffer[1] = AllocBuffer[0];
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Success = UnlockFile(Handle, 0, 0, BytesPerSector, 0);
+ ASSERT(!Success && ERROR_NOT_LOCKED == GetLastError());
+
+ Success = UnlockFile(Handle, BytesPerSector / 2, 0, BytesPerSector, 0);
+ ASSERT(Success);
+
+ Success = CloseHandle(Handle);
+ ASSERT(Success);
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | CreateFlags | FILE_FLAG_DELETE_ON_CLOSE, 0);
+ ASSERT(INVALID_HANDLE_VALUE != Handle);
+
+ FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+ ASSERT(0 == FilePointer);
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, 0);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+ ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Success = CloseHandle(Handle);
+ ASSERT(Success);
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ OPEN_EXISTING, 0, 0);
+ ASSERT(INVALID_HANDLE_VALUE == Handle);
+ ASSERT(ERROR_FILE_NOT_FOUND == GetLastError());
+
+ _aligned_free(AllocBuffer[0]);
+ _aligned_free(AllocBuffer[1]);
+
+ memfs_stop(memfs);
+}
+
+static void lock_overlapped_dotest(ULONG Flags, PWSTR VolPrefix, PWSTR Prefix, ULONG FileInfoTimeout, DWORD CreateFlags)
+{
+ /* this is not a true locking test since you need 2 processes to really test locks */
+
+ void *memfs = memfs_start_ex(Flags, FileInfoTimeout);
+
+ HANDLE Handle;
+ BOOL Success;
+ WCHAR FilePath[MAX_PATH];
+ SYSTEM_INFO SystemInfo;
+ DWORD SectorsPerCluster;
+ DWORD BytesPerSector;
+ DWORD FreeClusters;
+ DWORD TotalClusters;
+ PVOID AllocBuffer[2], Buffer[2];
+ ULONG AllocBufferSize;
+ DWORD BytesTransferred;
+ OVERLAPPED Overlapped;
+
+ GetSystemInfo(&SystemInfo);
+
+ StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\",
+ VolPrefix ? L"" : L"\\\\?\\GLOBALROOT", VolPrefix ? VolPrefix : memfs_volumename(memfs));
+
+ Success = GetDiskFreeSpaceW(FilePath, &SectorsPerCluster, &BytesPerSector, &FreeClusters, &TotalClusters);
+ ASSERT(Success);
+ AllocBufferSize = 16 * SystemInfo.dwPageSize;
+
+ AllocBuffer[0] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
+ AllocBuffer[1] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
+ ASSERT(0 != AllocBuffer[0] && 0 != AllocBuffer[1]);
+
+ srand((unsigned)time(0));
+ for (PUINT8 Bgn = AllocBuffer[0], End = Bgn + AllocBufferSize; End > Bgn; Bgn++)
+ *Bgn = rand();
+
+ Buffer[0] = (PVOID)((PUINT8)AllocBuffer[0] + BytesPerSector);
+ Buffer[1] = (PVOID)((PUINT8)AllocBuffer[1] + BytesPerSector);
+
+ StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0",
+ Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+ memset(&Overlapped, 0, sizeof Overlapped);
+ Overlapped.hEvent = CreateEvent(0, TRUE, FALSE, 0);
+ ASSERT(0 != Overlapped.hEvent);
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ CREATE_NEW,
+ FILE_ATTRIBUTE_NORMAL | CreateFlags | FILE_FLAG_OVERLAPPED,
+ 0);
+ ASSERT(INVALID_HANDLE_VALUE != Handle);
+
+ Overlapped.Offset = BytesPerSector / 2;
+ Success = LockFileEx(Handle, 0, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+
+ Overlapped.Offset = BytesPerSector + BytesPerSector / 2;
+ Success = LockFileEx(Handle, 0 | LOCKFILE_FAIL_IMMEDIATELY, 0,
+ BytesPerSector, 0, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+
+ Overlapped.Offset = BytesPerSector + BytesPerSector / 2;
+ Success = LockFileEx(Handle, LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY, 0,
+ BytesPerSector, 0, &Overlapped);
+ ASSERT(!Success && ERROR_LOCK_VIOLATION == GetLastError());
+
+ Overlapped.Offset = 0;
+ Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError() || ERROR_LOCK_VIOLATION == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(!Success && ERROR_LOCK_VIOLATION == GetLastError());
+
+ Overlapped.Offset = BytesPerSector / 2;
+ Success = UnlockFileEx(Handle, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(Success);
+
+ Overlapped.Offset = BytesPerSector + BytesPerSector / 2;
+ Success = UnlockFileEx(Handle, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(Success);
+
+ Overlapped.Offset = BytesPerSector / 2;
+ Success = LockFileEx(Handle, LOCKFILE_EXCLUSIVE_LOCK, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+
+ Overlapped.Offset = BytesPerSector + BytesPerSector / 2;
+ Success = LockFileEx(Handle, LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY, 0,
+ BytesPerSector, 0, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+
+ Overlapped.Offset = 0;
+ Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+
+ Overlapped.Offset = 2 * BytesPerSector;
+ Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+
+ Overlapped.Offset = 0;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 2 * BytesPerSector;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 2 * BytesPerSector;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(BytesPerSector == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 3 * BytesPerSector;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError() || ERROR_HANDLE_EOF == GetLastError());
+ if (ERROR_HANDLE_EOF != GetLastError())
+ {
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(!Success && ERROR_HANDLE_EOF == GetLastError());
+ }
+ ASSERT(0 == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 0;
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+
+ Overlapped.Offset = 0;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Buffer[0] = AllocBuffer[0];
+ Buffer[1] = AllocBuffer[0];
+
+ Overlapped.Offset = 0;
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+
+ Overlapped.Offset = 0;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 0;
+ Success = WriteFile(Handle, Buffer[0], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+
+ Overlapped.Offset = 0;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Overlapped.Offset = 0;
+ Success = UnlockFileEx(Handle, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(!Success && ERROR_NOT_LOCKED == GetLastError());
+
+ Overlapped.Offset = BytesPerSector / 2;
+ Success = UnlockFileEx(Handle, 0, BytesPerSector, 0, &Overlapped);
+ ASSERT(Success);
+
+ Success = CloseHandle(Handle);
+ ASSERT(Success);
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ OPEN_EXISTING,
+ FILE_ATTRIBUTE_NORMAL | CreateFlags | FILE_FLAG_OVERLAPPED | FILE_FLAG_DELETE_ON_CLOSE,
+ 0);
+ ASSERT(INVALID_HANDLE_VALUE != Handle);
+
+ Overlapped.Offset = 0;
+ memset(AllocBuffer[1], 0, AllocBufferSize);
+ Success = ReadFile(Handle, Buffer[1], 2 * SystemInfo.dwPageSize + BytesPerSector, &BytesTransferred, &Overlapped);
+ ASSERT(Success || ERROR_IO_PENDING == GetLastError());
+ Success = GetOverlappedResult(Handle, &Overlapped, &BytesTransferred, TRUE);
+ ASSERT(Success);
+ ASSERT(2 * SystemInfo.dwPageSize + BytesPerSector == BytesTransferred);
+ ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
+
+ Success = CloseHandle(Handle);
+ ASSERT(Success);
+
+ Handle = CreateFileW(FilePath,
+ GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+ OPEN_EXISTING, 0, 0);
+ ASSERT(INVALID_HANDLE_VALUE == Handle);
+ ASSERT(ERROR_FILE_NOT_FOUND == GetLastError());
+
+ Success = CloseHandle(Overlapped.hEvent);
+ ASSERT(Success);
+
+ _aligned_free(AllocBuffer[0]);
+ _aligned_free(AllocBuffer[1]);
+
+ memfs_stop(memfs);
+}
+
+void lock_noncached_test(void)
+{
+ if (NtfsTests)
+ {
+ WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
+ GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
+ lock_dotest(-1, L"C:", DirBuf, 0, FILE_FLAG_NO_BUFFERING);
+ }
+ if (WinFspDiskTests)
+ {
+ lock_dotest(MemfsDisk, 0, 0, 1000, FILE_FLAG_NO_BUFFERING);
+ lock_dotest(MemfsDisk, 0, 0, INFINITE, FILE_FLAG_NO_BUFFERING);
+ }
+ if (WinFspNetTests)
+ {
+ lock_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", 1000, FILE_FLAG_NO_BUFFERING);
+ lock_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", INFINITE, FILE_FLAG_NO_BUFFERING);
+ }
+}
+
+void lock_noncached_overlapped_test(void)
+{
+ if (NtfsTests)
+ {
+ WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
+ GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
+ lock_overlapped_dotest(-1, L"C:", DirBuf, 0, FILE_FLAG_NO_BUFFERING);
+ }
+ if (WinFspDiskTests)
+ {
+ lock_overlapped_dotest(MemfsDisk, 0, 0, 1000, FILE_FLAG_NO_BUFFERING);
+ lock_overlapped_dotest(MemfsDisk, 0, 0, INFINITE, FILE_FLAG_NO_BUFFERING);
+ }
+ if (WinFspNetTests)
+ {
+ lock_overlapped_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", 1000, FILE_FLAG_NO_BUFFERING);
+ lock_overlapped_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", INFINITE, FILE_FLAG_NO_BUFFERING);
+ }
+}
+
+void lock_cached_test(void)
+{
+ if (NtfsTests)
+ {
+ WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
+ GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
+ lock_dotest(-1, L"C:", DirBuf, 0, 0);
+ }
+ if (WinFspDiskTests)
+ {
+ lock_dotest(MemfsDisk, 0, 0, 1000, 0);
+ lock_dotest(MemfsDisk, 0, 0, INFINITE, 0);
+ }
+ if (WinFspNetTests)
+ {
+ lock_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", 1000, 0);
+ lock_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", INFINITE, 0);
+ }
+}
+
+void lock_cached_overlapped_test(void)
+{
+ if (NtfsTests)
+ {
+ WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
+ GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
+ lock_overlapped_dotest(-1, L"C:", DirBuf, 0, 0);
+ }
+ if (WinFspDiskTests)
+ {
+ lock_overlapped_dotest(MemfsDisk, 0, 0, 1000, 0);
+ lock_overlapped_dotest(MemfsDisk, 0, 0, INFINITE, 0);
+ }
+ if (WinFspNetTests)
+ {
+ lock_overlapped_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", 1000, 0);
+ lock_overlapped_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", INFINITE, 0);
+ }
+}
+
+void lock_tests(void)
+{
+ TEST(lock_noncached_test);
+ TEST(lock_noncached_overlapped_test);
+ TEST(lock_cached_test);
+ TEST(lock_cached_overlapped_test);
+}
diff --git a/tst/winfsp-tests/winfsp-tests.c b/tst/winfsp-tests/winfsp-tests.c
index cda0dece..cc367ee5 100644
--- a/tst/winfsp-tests/winfsp-tests.c
+++ b/tst/winfsp-tests/winfsp-tests.c
@@ -15,6 +15,7 @@ int main(int argc, char *argv[])
TESTSUITE(security_tests);
TESTSUITE(rdwr_tests);
TESTSUITE(flush_tests);
+ TESTSUITE(lock_tests);
TESTSUITE(dirctl_tests);
tlib_run_tests(argc, argv);