From f50438d3371c751f60c858a0617e9ed85783c21b Mon Sep 17 00:00:00 2001
From: Bill Zissimopoulos <billziss@navimatics.com>
Date: Tue, 1 Dec 2015 23:27:30 -0800
Subject: [PATCH] sys: mount testing

---
 src/sys/create.c              |  3 +-
 tst/winfsp-tests/mount-test.c | 71 ++++++++++++++++++++++++++++++++---
 2 files changed, 67 insertions(+), 7 deletions(-)

diff --git a/src/sys/create.c b/src/sys/create.c
index 50f4ef96..a67ba757 100644
--- a/src/sys/create.c
+++ b/src/sys/create.c
@@ -48,7 +48,8 @@ static NTSTATUS FspFsvolCreate(
 {
     PAGED_CODE();
 
-    return STATUS_INVALID_DEVICE_REQUEST;
+    /* !!!: DEVELOPMENT HACK! */
+    return STATUS_CANCELLED;
 }
 
 NTSTATUS FspCreate(
diff --git a/tst/winfsp-tests/mount-test.c b/tst/winfsp-tests/mount-test.c
index 2e477fbe..b3677209 100644
--- a/tst/winfsp-tests/mount-test.c
+++ b/tst/winfsp-tests/mount-test.c
@@ -1,6 +1,8 @@
 #include <winfsp/winfsp.h>
 #include <winfsp/fsctl.h>
 #include <tlib/testsuite.h>
+#include <process.h>
+#include <strsafe.h>
 
 void mount_invalid_test(void)
 {
@@ -18,10 +20,10 @@ void mount_invalid_test(void)
     ASSERT(STATUS_NO_SUCH_DEVICE == Result);
 }
 
-void mount_dotest(PWSTR DeviceName)
+void mount_create_delete_dotest(PWSTR DeviceName)
 {
     NTSTATUS Result;
-    BOOLEAN Success;
+    BOOL Success;
     FSP_FSCTL_VOLUME_PARAMS Params = { 0 };
     WCHAR VolumePath[MAX_PATH];
     HANDLE VolumeHandle;
@@ -41,14 +43,71 @@ void mount_dotest(PWSTR DeviceName)
     ASSERT(Success);
 }
 
-void mount_test(void)
+void mount_create_delete_test(void)
 {
-    mount_dotest(L"WinFsp.Disk");
-    //mount_dotest(L"WinFsp.Net");
+    mount_create_delete_dotest(L"WinFsp.Disk");
+    //mount_create_delete_dotest(L"WinFsp.Net");
+}
+
+static unsigned __stdcall mount_volume_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_dotest(PWSTR DeviceName)
+{
+    NTSTATUS Result;
+    BOOL Success;
+    FSP_FSCTL_VOLUME_PARAMS Params = { 0 };
+    WCHAR VolumePath[MAX_PATH];
+    WCHAR FilePath[MAX_PATH];
+    HANDLE VolumeHandle;
+    HANDLE Thread;
+    DWORD ExitCode;
+
+    Params.SectorSize = 16384;
+    Params.SerialNumber = 0x12345678;
+    Result = FspFsctlCreateVolume(DeviceName, &Params, 0, VolumePath, sizeof VolumePath);
+    ASSERT(STATUS_SUCCESS == Result);
+
+    StringCbPrintfW(FilePath, sizeof FilePath, L"\\\\?\\GLOBALROOT%s\\file0", VolumePath);
+    Thread = (HANDLE)_beginthreadex(0, 0, mount_volume_dotest_thread, FilePath, 0, 0);
+
+    Sleep(1000); /* give some time to the thread to execute */
+
+    Result = FspFsctlOpenVolume(VolumePath, &VolumeHandle);
+    ASSERT(STATUS_SUCCESS == Result);
+
+    Result = FspFsctlDeleteVolume(VolumeHandle);
+    ASSERT(STATUS_SUCCESS == Result);
+
+    Success = CloseHandle(VolumeHandle);
+    ASSERT(Success);
+
+    WaitForSingleObject(Thread, INFINITE);
+    GetExitCodeThread(Thread, &ExitCode);
+    CloseHandle(Thread);
+
+    ASSERT(ERROR_OPERATION_ABORTED == ExitCode);
+}
+
+void mount_volume_test(void)
+{
+    mount_volume_dotest(L"WinFsp.Disk");
+    //mount_volume_dotest(L"WinFsp.Net");
 }
 
 void mount_tests(void)
 {
     TEST(mount_invalid_test);
-    TEST(mount_test);
+    TEST(mount_create_delete_test);
+    TEST(mount_volume_test);
 }