mirror of
https://github.com/winfsp/winfsp.git
synced 2025-10-30 19:48:38 -05:00
267 lines
9.1 KiB
C
267 lines
9.1 KiB
C
#include <winfsp/winfsp.h>
|
|
#include <tlib/testsuite.h>
|
|
#include <process.h>
|
|
#include <strsafe.h>
|
|
|
|
extern int WinFspDiskTests;
|
|
extern int WinFspNetTests;
|
|
|
|
void mount_invalid_test(void)
|
|
{
|
|
NTSTATUS Result;
|
|
FSP_FSCTL_VOLUME_PARAMS VolumeParams = { 0 };
|
|
WCHAR VolumeName[MAX_PATH] = L"foo";
|
|
HANDLE VolumeHandle;
|
|
|
|
VolumeParams.SectorSize = 16384;
|
|
VolumeParams.VolumeSerialNumber = 0x12345678;
|
|
Result = FspFsctlCreateVolume(L"WinFsp.DoesNotExist", &VolumeParams,
|
|
VolumeName, sizeof VolumeName, &VolumeHandle);
|
|
ASSERT(STATUS_NO_SUCH_DEVICE == Result);
|
|
ASSERT(L'\0' == VolumeName[0]);
|
|
ASSERT(INVALID_HANDLE_VALUE == VolumeHandle);
|
|
}
|
|
|
|
void mount_open_device_dotest(PWSTR DeviceName)
|
|
{
|
|
BOOL Success;
|
|
HANDLE DeviceHandle;
|
|
WCHAR DevicePath[MAX_PATH];
|
|
|
|
StringCbPrintfW(DevicePath, sizeof DevicePath, L"\\\\?\\GLOBALROOT\\Device\\%s", DeviceName);
|
|
DeviceHandle = CreateFileW(DevicePath,
|
|
0, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, OPEN_EXISTING, 0, 0);
|
|
ASSERT(INVALID_HANDLE_VALUE != DeviceHandle);
|
|
|
|
Success = CloseHandle(DeviceHandle);
|
|
ASSERT(Success);
|
|
}
|
|
|
|
void mount_open_device_test(void)
|
|
{
|
|
if (WinFspDiskTests)
|
|
mount_open_device_dotest(L"WinFsp.Disk");
|
|
if (WinFspNetTests)
|
|
mount_open_device_dotest(L"WinFsp.Net");
|
|
}
|
|
|
|
void mount_create_volume_dotest(PWSTR DeviceName)
|
|
{
|
|
NTSTATUS Result;
|
|
BOOL Success;
|
|
FSP_FSCTL_VOLUME_PARAMS VolumeParams = { 0 };
|
|
WCHAR VolumeName[MAX_PATH];
|
|
HANDLE VolumeHandle;
|
|
|
|
VolumeParams.SectorSize = 16384;
|
|
VolumeParams.VolumeSerialNumber = 0x12345678;
|
|
wcscpy_s(VolumeParams.Prefix, sizeof VolumeParams.Prefix / sizeof(WCHAR), L"\\winfsp-tests\\share");
|
|
Result = FspFsctlCreateVolume(DeviceName, &VolumeParams,
|
|
VolumeName, sizeof VolumeName, &VolumeHandle);
|
|
ASSERT(STATUS_SUCCESS == Result);
|
|
ASSERT(0 == wcsncmp(L"\\Device\\Volume{", VolumeName, 15));
|
|
ASSERT(INVALID_HANDLE_VALUE != VolumeHandle);
|
|
|
|
Success = CloseHandle(VolumeHandle);
|
|
ASSERT(Success);
|
|
}
|
|
|
|
void mount_create_volume_test(void)
|
|
{
|
|
if (WinFspDiskTests)
|
|
mount_create_volume_dotest(L"WinFsp.Disk");
|
|
if (WinFspNetTests)
|
|
mount_create_volume_dotest(L"WinFsp.Net");
|
|
}
|
|
|
|
static unsigned __stdcall mount_volume_cancel_dotest_thread(void *FilePath)
|
|
{
|
|
FspDebugLog(__FUNCTION__ ": \"%S\"\n", FilePath);
|
|
|
|
HANDLE Handle;
|
|
Handle = CreateFileW(FilePath,
|
|
GENERIC_READ | GENERIC_WRITE, 0, 0, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0);
|
|
if (INVALID_HANDLE_VALUE == Handle)
|
|
return GetLastError();
|
|
CloseHandle(Handle);
|
|
return 0;
|
|
}
|
|
|
|
void mount_volume_cancel_dotest(PWSTR DeviceName, PWSTR Prefix)
|
|
{
|
|
NTSTATUS Result;
|
|
BOOL Success;
|
|
FSP_FSCTL_VOLUME_PARAMS VolumeParams = { 0 };
|
|
WCHAR VolumeName[MAX_PATH];
|
|
WCHAR FilePath[MAX_PATH];
|
|
HANDLE VolumeHandle;
|
|
HANDLE Thread;
|
|
DWORD ExitCode;
|
|
|
|
VolumeParams.SectorSize = 16384;
|
|
VolumeParams.VolumeSerialNumber = 0x12345678;
|
|
wcscpy_s(VolumeParams.Prefix, sizeof VolumeParams.Prefix / sizeof(WCHAR), L"\\winfsp-tests\\share");
|
|
Result = FspFsctlCreateVolume(DeviceName, &VolumeParams,
|
|
VolumeName, sizeof VolumeName, &VolumeHandle);
|
|
ASSERT(STATUS_SUCCESS == Result);
|
|
ASSERT(0 == wcsncmp(L"\\Device\\Volume{", VolumeName, 15));
|
|
ASSERT(INVALID_HANDLE_VALUE != VolumeHandle);
|
|
|
|
StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0",
|
|
Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : VolumeName);
|
|
Thread = (HANDLE)_beginthreadex(0, 0, mount_volume_cancel_dotest_thread, FilePath, 0, 0);
|
|
ASSERT(0 != Thread);
|
|
|
|
Sleep(0 == Prefix ? 1000 : 5000); /* give some time to the thread to execute */
|
|
|
|
Success = CloseHandle(VolumeHandle);
|
|
ASSERT(Success);
|
|
|
|
WaitForSingleObject(Thread, INFINITE);
|
|
GetExitCodeThread(Thread, &ExitCode);
|
|
CloseHandle(Thread);
|
|
|
|
ASSERT(ERROR_OPERATION_ABORTED == ExitCode);
|
|
}
|
|
|
|
void mount_volume_cancel_test(void)
|
|
{
|
|
if (WinFspDiskTests)
|
|
mount_volume_cancel_dotest(L"WinFsp.Disk", 0);
|
|
if (WinFspNetTests)
|
|
mount_volume_cancel_dotest(L"WinFsp.Net", L"\\\\winfsp-tests\\share");
|
|
}
|
|
|
|
static unsigned __stdcall mount_volume_transact_dotest_thread(void *FilePath)
|
|
{
|
|
FspDebugLog(__FUNCTION__ ": \"%S\"\n", FilePath);
|
|
|
|
HANDLE Handle;
|
|
Handle = CreateFileW(FilePath,
|
|
FILE_GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0);
|
|
if (INVALID_HANDLE_VALUE == Handle)
|
|
return GetLastError();
|
|
CloseHandle(Handle);
|
|
return 0;
|
|
}
|
|
|
|
void mount_volume_transact_dotest(PWSTR DeviceName, PWSTR Prefix)
|
|
{
|
|
NTSTATUS Result;
|
|
BOOL Success;
|
|
FSP_FSCTL_VOLUME_PARAMS VolumeParams = { 0 };
|
|
WCHAR VolumeName[MAX_PATH];
|
|
WCHAR FilePath[MAX_PATH];
|
|
HANDLE VolumeHandle;
|
|
HANDLE Thread;
|
|
DWORD ExitCode;
|
|
|
|
VolumeParams.TransactTimeout = 10000; /* allow for longer transact timeout to handle MUP redir */
|
|
VolumeParams.SectorSize = 16384;
|
|
VolumeParams.VolumeSerialNumber = 0x12345678;
|
|
wcscpy_s(VolumeParams.Prefix, sizeof VolumeParams.Prefix / sizeof(WCHAR), L"\\winfsp-tests\\share");
|
|
Result = FspFsctlCreateVolume(DeviceName, &VolumeParams,
|
|
VolumeName, sizeof VolumeName, &VolumeHandle);
|
|
ASSERT(STATUS_SUCCESS == Result);
|
|
ASSERT(0 == wcsncmp(L"\\Device\\Volume{", VolumeName, 15));
|
|
ASSERT(INVALID_HANDLE_VALUE != VolumeHandle);
|
|
|
|
StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\file0",
|
|
Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : VolumeName);
|
|
Thread = (HANDLE)_beginthreadex(0, 0, mount_volume_transact_dotest_thread, FilePath, 0, 0);
|
|
ASSERT(0 != Thread);
|
|
|
|
Sleep(1000); /* give some time to the thread to execute */
|
|
|
|
FSP_FSCTL_DECLSPEC_ALIGN UINT8 RequestBuf[FSP_FSCTL_TRANSACT_BATCH_BUFFER_SIZEMIN];
|
|
FSP_FSCTL_DECLSPEC_ALIGN UINT8 ResponseBuf[FSP_FSCTL_TRANSACT_RSP_SIZEMAX];
|
|
UINT8 *RequestBufEnd;
|
|
UINT8 *ResponseBufEnd = ResponseBuf + sizeof ResponseBuf;
|
|
SIZE_T RequestBufSize;
|
|
SIZE_T ResponseBufSize;
|
|
FSP_FSCTL_TRANSACT_REQ *Request = (PVOID)RequestBuf, *NextRequest;
|
|
FSP_FSCTL_TRANSACT_RSP *Response = (PVOID)ResponseBuf;
|
|
|
|
ResponseBufSize = 0;
|
|
RequestBufSize = sizeof RequestBuf;
|
|
Result = FspFsctlTransact(VolumeHandle, 0, 0, RequestBuf, &RequestBufSize, TRUE);
|
|
ASSERT(STATUS_SUCCESS == Result);
|
|
|
|
RequestBufEnd = RequestBuf + RequestBufSize;
|
|
|
|
NextRequest = FspFsctlTransactConsumeRequest(Request, RequestBufEnd);
|
|
ASSERT(0 != NextRequest);
|
|
|
|
ASSERT(0 == Request->Version);
|
|
ASSERT(FSP_FSCTL_TRANSACT_REQ_SIZEMAX >= Request->Size);
|
|
ASSERT(0 != Request->Hint);
|
|
ASSERT(FspFsctlTransactCreateKind == Request->Kind ||
|
|
FspFsctlTransactQueryVolumeInformationKind == Request->Kind);
|
|
if (FspFsctlTransactCreateKind == Request->Kind)
|
|
{
|
|
ASSERT(FILE_CREATE == ((Request->Req.Create.CreateOptions >> 24) & 0xff));
|
|
ASSERT(0 == Request->Req.Create.FileAttributes);
|
|
ASSERT(0 == Request->Req.Create.SecurityDescriptor.Offset);
|
|
ASSERT(0 == Request->Req.Create.SecurityDescriptor.Size);
|
|
ASSERT(0 == Request->Req.Create.AllocationSize);
|
|
ASSERT(FILE_GENERIC_READ == Request->Req.Create.DesiredAccess);
|
|
ASSERT((FILE_SHARE_READ | FILE_SHARE_WRITE) == Request->Req.Create.ShareAccess);
|
|
ASSERT(0 == Request->Req.Create.Ea.Offset);
|
|
ASSERT(0 == Request->Req.Create.Ea.Size);
|
|
ASSERT(Request->Req.Create.UserMode);
|
|
ASSERT(Request->Req.Create.HasTraversePrivilege);
|
|
ASSERT(!Request->Req.Create.OpenTargetDirectory);
|
|
ASSERT(!Request->Req.Create.CaseSensitive);
|
|
ASSERT(0 == Request->FileName.Offset);
|
|
ASSERT((wcslen((PVOID)Request->Buffer) + 1) * sizeof(WCHAR) == Request->FileName.Size);
|
|
ASSERT(0 == wcscmp((PVOID)Request->Buffer, L"\\file0"));
|
|
}
|
|
|
|
ASSERT(FspFsctlTransactCanProduceResponse(Response, ResponseBufEnd));
|
|
|
|
RtlZeroMemory(Response, sizeof *Response);
|
|
Response->Size = sizeof *Response;
|
|
Response->Hint = Request->Hint;
|
|
Response->Kind = Request->Kind;
|
|
Response->IoStatus.Status = STATUS_ACCESS_DENIED;
|
|
Response->IoStatus.Information = 0;
|
|
|
|
Response = FspFsctlTransactProduceResponse(Response, Response->Size);
|
|
ASSERT(0 != Response);
|
|
|
|
Request = NextRequest;
|
|
NextRequest = FspFsctlTransactConsumeRequest(Request, RequestBufEnd);
|
|
ASSERT(0 == NextRequest);
|
|
|
|
ResponseBufSize = (PUINT8)Response - ResponseBuf;
|
|
RequestBufSize = 0;
|
|
Result = FspFsctlTransact(VolumeHandle, ResponseBuf, ResponseBufSize, 0, &RequestBufSize, TRUE);
|
|
ASSERT(STATUS_SUCCESS == Result);
|
|
|
|
Success = CloseHandle(VolumeHandle);
|
|
ASSERT(Success);
|
|
|
|
WaitForSingleObject(Thread, INFINITE);
|
|
GetExitCodeThread(Thread, &ExitCode);
|
|
CloseHandle(Thread);
|
|
|
|
ASSERT(ERROR_ACCESS_DENIED == ExitCode || ERROR_OPERATION_ABORTED == ExitCode);
|
|
}
|
|
|
|
void mount_volume_transact_test(void)
|
|
{
|
|
if (WinFspDiskTests)
|
|
mount_volume_transact_dotest(L"WinFsp.Disk", 0);
|
|
if (WinFspNetTests)
|
|
mount_volume_transact_dotest(L"WinFsp.Net", L"\\\\winfsp-tests\\share");
|
|
}
|
|
|
|
void mount_tests(void)
|
|
{
|
|
TEST(mount_invalid_test);
|
|
TEST(mount_open_device_test);
|
|
TEST(mount_create_volume_test);
|
|
TEST(mount_volume_cancel_test);
|
|
TEST(mount_volume_transact_test);
|
|
}
|