Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 08:53:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

winfsp-tests: rdwr-test

Browse Source
This commit is contained in:
Bill Zissimopoulos 2016-03-07 21:15:08 -08:00
parent 1ad09f9b99
commit 6efb5cc113
3 changed files with 82 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/sys/util.c
View File

@ -504,6 +504,7 @@ NTSTATUS FspSafeMdlCreate(PMDL UserMdl, LOCK_OPERATION Operation, FSP_SAFE_MDL *
ULONG BufferPageCount; ULONG BufferPageCount;
ASSERT(0 != PageCount); ASSERT(0 != PageCount);
ASSERT(FlagOn(UserMdl->MdlFlags, MDL_PAGES_LOCKED));
*PSafeMdl = 0; *PSafeMdl = 0;

2
tst/winfsp-tests/memfs-test.c
View File

@ -14,7 +14,7 @@ void *memfs_start_ex(ULONG Flags, ULONG FileInfoTimeout)
MEMFS *Memfs; MEMFS *Memfs;
NTSTATUS Result; NTSTATUS Result;
Result = MemfsCreate(Flags, FileInfoTimeout, 1000, 65500, &Memfs); Result = MemfsCreate(Flags, FileInfoTimeout, 1024, 1024 * 1024, &Memfs);
ASSERT(NT_SUCCESS(Result)); ASSERT(NT_SUCCESS(Result));
ASSERT(0 != Memfs); ASSERT(0 != Memfs);

99
tst/winfsp-tests/rdwr-test.c
View File

@ -20,28 +20,36 @@ void rdwr_noncached_dotest(ULONG Flags, PWSTR VolPrefix, PWSTR Prefix, ULONG Fil
HANDLE Handle; HANDLE Handle;
BOOL Success; BOOL Success;
WCHAR FilePath[MAX_PATH]; WCHAR FilePath[MAX_PATH];
SYSTEM_INFO SystemInfo;
DWORD SectorsPerCluster; DWORD SectorsPerCluster;
DWORD BytesPerSector; DWORD BytesPerSector;
DWORD FreeClusters; DWORD FreeClusters;
DWORD TotalClusters; DWORD TotalClusters;
PVOID Buffer[2]; PVOID AllocBuffer[2], Buffer[2];
ULONG AllocBufferSize;
DWORD BytesTransferred; DWORD BytesTransferred;
DWORD FilePointer; DWORD FilePointer;
GetSystemInfo(&SystemInfo);
StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\", StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\",
VolPrefix ? L"" : L"\\\\?\\GLOBALROOT", VolPrefix ? VolPrefix : memfs_volumename(memfs)); VolPrefix ? L"" : L"\\\\?\\GLOBALROOT", VolPrefix ? VolPrefix : memfs_volumename(memfs));
Success = GetDiskFreeSpaceW(FilePath, &SectorsPerCluster, &BytesPerSector, &FreeClusters, &TotalClusters); Success = GetDiskFreeSpaceW(FilePath, &SectorsPerCluster, &BytesPerSector, &FreeClusters, &TotalClusters);
ASSERT(Success); ASSERT(Success);
AllocBufferSize = 16 * SystemInfo.dwPageSize;
Buffer[0] = _aligned_malloc(BytesPerSector, BytesPerSector); AllocBuffer[0] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
Buffer[1] = _aligned_malloc(BytesPerSector, BytesPerSector); AllocBuffer[1] = _aligned_malloc(AllocBufferSize, SystemInfo.dwPageSize);
ASSERT(0 != Buffer[0] && 0 != Buffer[1]); ASSERT(0 != AllocBuffer[0] && 0 != AllocBuffer[1]);
srand((unsigned)time(0)); srand((unsigned)time(0));
for (PUINT8 Bgn = Buffer[0], End = Bgn + BytesPerSector; End > Bgn; Bgn++) for (PUINT8 Bgn = AllocBuffer[0], End = Bgn + AllocBufferSize; End > Bgn; Bgn++)
*Bgn = rand(); *Bgn = rand();
Buffer[0] = (PVOID)((PUINT8)AllocBuffer[0] + BytesPerSector);
Buffer[1] = (PVOID)((PUINT8)AllocBuffer[1] + BytesPerSector);
StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0", StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0",
Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs)); Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
@ -55,54 +63,107 @@ void rdwr_noncached_dotest(ULONG Flags, PWSTR VolPrefix, PWSTR Prefix, ULONG Fil
Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0); Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred); ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN); FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(2 * BytesPerSector == FilePointer); ASSERT(2 * BytesPerSector == FilePointer);
Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0); Success = WriteFile(Handle, Buffer[0], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred); ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN); FilePointer = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
ASSERT(0 == FilePointer); ASSERT(0 == FilePointer);
memset(AllocBuffer[1], 0, AllocBufferSize);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0); Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred); ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(BytesTransferred == FilePointer); ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN); FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(2 * BytesPerSector == FilePointer); ASSERT(2 * BytesPerSector == FilePointer);
memset(AllocBuffer[1], 0, AllocBufferSize);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0); Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred); ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer); ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN); FilePointer = SetFilePointer(Handle, 2 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(2 * BytesPerSector == FilePointer); ASSERT(2 * BytesPerSector == FilePointer);
memset(AllocBuffer[1], 0, AllocBufferSize);
Success = ReadFile(Handle, Buffer[1], 2 * BytesPerSector, &BytesTransferred, 0); Success = ReadFile(Handle, Buffer[1], 2 * BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(BytesPerSector == BytesTransferred); ASSERT(BytesPerSector == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(2 * BytesPerSector + BytesTransferred == FilePointer); ASSERT(0 == memcmp(Buffer[0], Buffer[1], BytesTransferred));
FilePointer = SetFilePointer(Handle, 3 * BytesPerSector, 0, FILE_BEGIN); FilePointer = SetFilePointer(Handle, 3 * BytesPerSector, 0, FILE_BEGIN);
ASSERT(3 * BytesPerSector == FilePointer); ASSERT(3 * BytesPerSector == FilePointer);
memset(AllocBuffer[1], 0, AllocBufferSize);
Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0); Success = ReadFile(Handle, Buffer[1], BytesPerSector, &BytesTransferred, 0);
ASSERT(Success); ASSERT(Success);
ASSERT(0 == BytesTransferred); ASSERT(0 == BytesTransferred);
FilePointer = SetFilePointer(Handle, 0, 0, FILE_CURRENT); ASSERT(FilePointer + BytesTransferred == SetFilePointer(Handle, 0, 0, FILE_CURRENT));
ASSERT(3 * BytesPerSector + BytesTransferred == FilePointer); 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 = CloseHandle(Handle); Success = CloseHandle(Handle);
ASSERT(Success); ASSERT(Success);
free(Buffer[0]); _aligned_free(AllocBuffer[0]);
free(Buffer[1]); _aligned_free(AllocBuffer[1]);
memfs_stop(memfs); memfs_stop(memfs);
} }