winfsp-tests: rdwr-test

This commit is contained in:
Bill Zissimopoulos 2016-03-06 21:22:17 -08:00
parent ad9dafbaaa
commit 26b5d8a458

View File

@ -1,5 +1,7 @@
#include <winfsp/winfsp.h> #include <winfsp/winfsp.h>
#include <tlib/testsuite.h> #include <tlib/testsuite.h>
#include <strsafe.h>
#include <time.h>
#include "memfs.h" #include "memfs.h"
void *memfs_start_ex(ULONG Flags, ULONG FileInfoTimeout); void *memfs_start_ex(ULONG Flags, ULONG FileInfoTimeout);
@ -11,10 +13,97 @@ extern int NtfsTests;
extern int WinFspDiskTests; extern int WinFspDiskTests;
extern int WinFspNetTests; extern int WinFspNetTests;
void rdwr_noncached_dotest(ULONG Flags, PWSTR Prefix, ULONG FileInfoTimeout) void rdwr_noncached_dotest(ULONG Flags, PWSTR VolPrefix, PWSTR Prefix, ULONG FileInfoTimeout)
{ {
void *memfs = memfs_start_ex(Flags, FileInfoTimeout); void *memfs = memfs_start_ex(Flags, FileInfoTimeout);
HANDLE Handle;
BOOL Success;
WCHAR FilePath[MAX_PATH];
DWORD SectorsPerCluster;
DWORD BytesPerSector;
DWORD FreeClusters;
DWORD TotalClusters;
PVOID Buffer[2];
DWORD BytesTransferred;
DWORD FilePointer;
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);
Buffer[0] = malloc(BytesPerSector);
Buffer[1] = malloc(BytesPerSector);
ASSERT(0 != Buffer[0] && 0 != Buffer[1]);
srand((unsigned)time(0));
for (PUINT8 Bgn = Buffer[0], End = Bgn + BytesPerSector; End > Bgn; Bgn++)
*Bgn = rand();
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 | FILE_FLAG_NO_BUFFERING | FILE_FLAG_DELETE_ON_CLOSE, 0);
ASSERT(INVALID_HANDLE_VALUE != Handle);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
ASSERT(0 == FilePointer);
Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(BytesTransferred == FilePointer);
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);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
ASSERT(0 == FilePointer);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(2 * BytesPerSector == FilePointer);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(2 * BytesPerSector == FilePointer);
Success = ReadFile(Handle, Buffer[1], 2 * BytesPerSector, &BytesTransferred, 0);
ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 3 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(3 * BytesPerSector == FilePointer);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success);
ASSERT(0 == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT);
ASSERT(3 * BytesPerSector + BytesTransferred == FilePointer);
Success = CloseHandle(Handle);
ASSERT(Success);
free(Buffer[0]);
free(Buffer[1]);
memfs_stop(memfs); memfs_stop(memfs);
} }
@ -24,17 +113,17 @@ void rdwr_noncached_test(void)
{ {
WCHAR DirBuf[MAX_PATH] = L"\\\\?\\"; WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4); GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
rdwr_noncached_dotest(-1, DirBuf, 0); rdwr_noncached_dotest(-1, L"C:", DirBuf, 0);
} }
if (WinFspDiskTests) if (WinFspDiskTests)
{ {
rdwr_noncached_dotest(MemfsDisk, 0, 0); rdwr_noncached_dotest(MemfsDisk, 0, 0, 1000);
rdwr_noncached_dotest(MemfsDisk, 0, 1000); rdwr_noncached_dotest(MemfsDisk, 0, 0, INFINITE);
} }
if (WinFspNetTests) if (WinFspNetTests)
{ {
rdwr_noncached_dotest(MemfsNet, L"\\\\memfs\\share", 0); rdwr_noncached_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", 1000);
rdwr_noncached_dotest(MemfsNet, L"\\\\memfs\\share", 1000); rdwr_noncached_dotest(MemfsNet, L"\\\\memfs\\share", L"\\\\memfs\\share", INFINITE);
} }
} }