From 5b917c4566c41b4bc128bff1e9d1a4bd54f69eac Mon Sep 17 00:00:00 2001
From: Bill Zissimopoulos <billziss@navimatics.com>
Date: Fri, 8 Apr 2016 14:12:56 -0700
Subject: [PATCH] dll: FspFileSystemSetMountPoint

---
 inc/winfsp/winfsp.h |  3 ++
 src/dll/dispatch.c  | 77 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+)

diff --git a/inc/winfsp/winfsp.h b/inc/winfsp/winfsp.h
index c2e4c7b7..092d0a72 100644
--- a/inc/winfsp/winfsp.h
+++ b/inc/winfsp/winfsp.h
@@ -124,12 +124,15 @@ typedef struct _FSP_FILE_SYSTEM
     HANDLE DispatcherThread;
     ULONG DispatcherThreadCount;
     NTSTATUS DispatcherResult;
+    PWSTR MountPoint;
 } FSP_FILE_SYSTEM;
 FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath,
     const FSP_FSCTL_VOLUME_PARAMS *VolumeParams,
     const FSP_FILE_SYSTEM_INTERFACE *Interface,
     FSP_FILE_SYSTEM **PFileSystem);
 FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem);
+FSP_API NTSTATUS FspFileSystemSetMountPoint(FSP_FILE_SYSTEM *FileSystem, PWSTR MountPoint);
+FSP_API VOID FspFileSystemRemoveMountPoint(FSP_FILE_SYSTEM *FileSystem);
 FSP_API NTSTATUS FspFileSystemStartDispatcher(FSP_FILE_SYSTEM *FileSystem, ULONG ThreadCount);
 FSP_API VOID FspFileSystemStopDispatcher(FSP_FILE_SYSTEM *FileSystem);
 FSP_API VOID FspFileSystemSendResponse(FSP_FILE_SYSTEM *FileSystem,
diff --git a/src/dll/dispatch.c b/src/dll/dispatch.c
index 3179dbe3..58a899d0 100644
--- a/src/dll/dispatch.c
+++ b/src/dll/dispatch.c
@@ -64,10 +64,87 @@ FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath,
 
 FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem)
 {
+    FspFileSystemRemoveMountPoint(FileSystem);
     CloseHandle(FileSystem->VolumeHandle);
     MemFree(FileSystem);
 }
 
+FSP_API NTSTATUS FspFileSystemSetMountPoint(FSP_FILE_SYSTEM *FileSystem, PWSTR MountPoint)
+{
+    if (0 != FileSystem->MountPoint)
+        return STATUS_INVALID_PARAMETER;
+
+    if (0 == MountPoint)
+    {
+        DWORD Drives;
+        WCHAR Drive;
+
+        MountPoint = MemAlloc(3 * sizeof(WCHAR));
+        if (0 == MountPoint)
+            return STATUS_INSUFFICIENT_RESOURCES;
+        MountPoint[1] = L':';
+        MountPoint[1] = L'\0';
+
+        Drives = GetLogicalDrives();
+        if (0 != Drives)
+        {
+            for (Drive = 'Z'; 'D' <= Drive; Drive--)
+                if (0 == (Drives & (1 << (Drive - 'A'))))
+                {
+                    MountPoint[0] = Drive;
+                    if (DefineDosDeviceW(DDD_RAW_TARGET_PATH, MountPoint, FileSystem->VolumeName))
+                    {
+                        FileSystem->MountPoint = MountPoint;
+                        return STATUS_SUCCESS;
+                    }
+                }
+
+            SetLastError(ERROR_FILE_NOT_FOUND);
+        }
+
+        MemFree(MountPoint);
+
+        return FspNtStatusFromWin32(GetLastError());
+    }
+    else
+    {
+        PWSTR P;
+        ULONG L;
+
+        for (P = MountPoint; *P; P++)
+            ;
+        L = (ULONG)((P - MountPoint + 1) * sizeof(WCHAR));
+
+        P = MemAlloc(L);
+        if (0 == P)
+            return STATUS_INSUFFICIENT_RESOURCES;
+        memcpy(P, MountPoint, L);
+        MountPoint = P;
+
+        if (DefineDosDeviceW(DDD_RAW_TARGET_PATH, MountPoint, FileSystem->VolumeName))
+        {
+            FileSystem->MountPoint = MountPoint;
+            return STATUS_SUCCESS;
+        }
+
+        MemFree(MountPoint);
+
+        return FspNtStatusFromWin32(GetLastError());
+    }
+}
+
+FSP_API VOID FspFileSystemRemoveMountPoint(FSP_FILE_SYSTEM *FileSystem)
+{
+    if (0 == FileSystem->MountPoint)
+        return;
+
+    DefineDosDeviceW(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE,
+        FileSystem->MountPoint, FileSystem->VolumeName);
+
+    MemFree(FileSystem->MountPoint);
+    FileSystem->MountPoint = 0;
+}
+
 static DWORD WINAPI FspFileSystemDispatcherThread(PVOID FileSystem0)
 {
     FSP_FILE_SYSTEM *FileSystem = FileSystem0;