Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-07-03 09:22:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys,dll: MountUseMountmgrFromFSD

Browse Source
This commit is contained in:
Bill Zissimopoulos
2022-06-27 17:28:24 +01:00
parent d7450d740e
commit a731f0e5d8
14 changed files with 936 additions and 377 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/sys/device.c
View File

@ -59,6 +59,8 @@ VOID FspFsvolDeviceGetVolumeInfo(PDEVICE_OBJECT DeviceObject, FSP_FSCTL_VOLUME_I
BOOLEAN FspFsvolDeviceTryGetVolumeInfo(PDEVICE_OBJECT DeviceObject, FSP_FSCTL_VOLUME_INFO *VolumeInfo);
VOID FspFsvolDeviceSetVolumeInfo(PDEVICE_OBJECT DeviceObject, const FSP_FSCTL_VOLUME_INFO *VolumeInfo);
VOID FspFsvolDeviceInvalidateVolumeInfo(PDEVICE_OBJECT DeviceObject);
static NTSTATUS FspFsvrtDeviceInit(PDEVICE_OBJECT DeviceObject);
static VOID FspFsvrtDeviceFini(PDEVICE_OBJECT DeviceObject);
static NTSTATUS FspFsmupDeviceInit(PDEVICE_OBJECT DeviceObject);
static VOID FspFsmupDeviceFini(PDEVICE_OBJECT DeviceObject);
NTSTATUS FspDeviceCopyList(
@ -84,6 +86,8 @@ VOID FspDeviceDeleteAll(VOID);
#pragma alloc_text(PAGE, FspFsvolDeviceCompareContextByName)
#pragma alloc_text(PAGE, FspFsvolDeviceAllocateContextByName)
#pragma alloc_text(PAGE, FspFsvolDeviceFreeContextByName)
#pragma alloc_text(PAGE, FspFsvrtDeviceInit)
#pragma alloc_text(PAGE, FspFsvrtDeviceFini)
#pragma alloc_text(PAGE, FspFsmupDeviceInit)
#pragma alloc_text(PAGE, FspFsmupDeviceFini)
#pragma alloc_text(PAGE, FspDeviceCopyList)
@ -171,7 +175,7 @@ NTSTATUS FspDeviceInitialize(PDEVICE_OBJECT DeviceObject)
Result = FspFsvolDeviceInit(DeviceObject);
break;
case FspFsvrtDeviceExtensionKind:
Result = STATUS_SUCCESS;
Result = FspFsvrtDeviceInit(DeviceObject);
break;
case FspFsmupDeviceExtensionKind:
Result = FspFsmupDeviceInit(DeviceObject);
@ -202,6 +206,7 @@ VOID FspDeviceDelete(PDEVICE_OBJECT DeviceObject)
FspFsvolDeviceFini(DeviceObject);
break;
case FspFsvrtDeviceExtensionKind:
FspFsvrtDeviceFini(DeviceObject);
break;
case FspFsmupDeviceExtensionKind:
FspFsmupDeviceFini(DeviceObject);
@ -848,6 +853,27 @@ VOID FspFsvolDeviceInvalidateVolumeInfo(PDEVICE_OBJECT DeviceObject)
KeReleaseSpinLock(&FsvolDeviceExtension->InfoSpinLock, Irql);
}
static NTSTATUS FspFsvrtDeviceInit(PDEVICE_OBJECT DeviceObject)
{
PAGED_CODE();
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(DeviceObject);
ExInitializeFastMutex(&FsvrtDeviceExtension->MountMutex);
return STATUS_SUCCESS;
}
static VOID FspFsvrtDeviceFini(PDEVICE_OBJECT DeviceObject)
{
PAGED_CODE();
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(DeviceObject);
if (0 != FsvrtDeviceExtension->MountPoint.Buffer)
FspFree(FsvrtDeviceExtension->MountPoint.Buffer);
}
static NTSTATUS FspFsmupDeviceInit(PDEVICE_OBJECT DeviceObject)
{
PAGED_CODE();

35
src/sys/driver.h
View File

@ -1232,13 +1232,18 @@ typedef struct
} FSP_FSVOL_DEVICE_EXTENSION;
typedef struct
{
/* read-only after creation (and insertion in the ContextTable) */
FSP_DEVICE_EXTENSION Base;
UINT16 SectorSize;
LONG IsMountdev;
BOOLEAN Persistent;
GUID UniqueId;
UNICODE_STRING VolumeName;
WCHAR VolumeNameBuf[FSP_FSCTL_VOLUME_NAME_SIZE / sizeof(WCHAR)];
FAST_MUTEX MountMutex;
/* interlocked access */
LONG IsMountdev;
/* protected under MountMutex */
BOOLEAN Persistent;
GUID UniqueId;
UNICODE_STRING MountPoint;
} FSP_FSVRT_DEVICE_EXTENSION;
typedef struct
{
@ -1435,6 +1440,18 @@ BOOLEAN FspFsvolDeviceVolumePrefixInString(PDEVICE_OBJECT DeviceObject, PUNICODE
TRUE);
}
static inline
VOID FspFsvrtDeviceLockMount(PDEVICE_OBJECT DeviceObject)
{
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(DeviceObject);
ExAcquireFastMutexUnsafe(&FsvrtDeviceExtension->MountMutex);
}
static inline
VOID FspFsvrtDeviceUnlockMount(PDEVICE_OBJECT DeviceObject)
{
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(DeviceObject);
ExReleaseFastMutexUnsafe(&FsvrtDeviceExtension->MountMutex);
}
static inline
VOID FspFsmupDeviceLockPrefixTable(PDEVICE_OBJECT DeviceObject)
{
FSP_FSMUP_DEVICE_EXTENSION *FsmupDeviceExtension = FspFsmupDeviceExtension(DeviceObject);
@ -1524,6 +1541,16 @@ NTSTATUS FspMountdevMake(
VOID FspMountdevFini(
PDEVICE_OBJECT FsvrtDeviceObject);
/* mountmgr */
NTSTATUS FspMountmgrCreateDrive(
PUNICODE_STRING VolumeName, GUID *UniqueId, PUNICODE_STRING MountPoint);
NTSTATUS FspMountmgrDeleteDrive(
PUNICODE_STRING MountPoint);
NTSTATUS FspMountmgrNotifyCreateDirectory(
PUNICODE_STRING VolumeName, GUID *UniqueId, PUNICODE_STRING MountPoint);
NTSTATUS FspMountmgrNotifyDeleteDirectory(
PUNICODE_STRING VolumeName, PUNICODE_STRING MountPoint);
/* fsmup */
NTSTATUS FspMupRegister(
PDEVICE_OBJECT FsmupDeviceObject, PDEVICE_OBJECT FsvolDeviceObject);
@ -1543,6 +1570,8 @@ NTSTATUS FspVolumeMount(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeMakeMountdev(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeUseMountmgr(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeGetName(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeGetNameList(

4
src/sys/fsctl.c
View File

@ -81,6 +81,10 @@ static NTSTATUS FspFsctlFileSystemControl(
if (0 != IrpSp->FileObject->FsContext2)
Result = FspVolumeMakeMountdev(FsctlDeviceObject, Irp, IrpSp);
break;
case FSP_FSCTL_MOUNTMGR:
if (0 != IrpSp->FileObject->FsContext2)
Result = FspVolumeUseMountmgr(FsctlDeviceObject, Irp, IrpSp);
break;
case FSP_FSCTL_VOLUME_NAME:
if (0 != IrpSp->FileObject->FsContext2)
Result = FspVolumeGetName(FsctlDeviceObject, Irp, IrpSp);

7
src/sys/mountdev.c
View File

@ -138,7 +138,7 @@ NTSTATUS FspMountdevMake(
* be mounted using the MountManager.
*
* This function requires protection against concurrency. In general this
* is achieved by acquiring the GlobalDeviceLock.
* is achieved by the caller acquiring the MountMutex.
*/
PAGED_CODE();
@ -178,11 +178,6 @@ NTSTATUS FspMountdevMake(
/* initialize the fsvrt device extension */
RtlCopyMemory(&FsvrtDeviceExtension->UniqueId, &Guid, sizeof Guid);
RtlInitEmptyUnicodeString(&FsvrtDeviceExtension->VolumeName,
FsvrtDeviceExtension->VolumeNameBuf, sizeof FsvrtDeviceExtension->VolumeNameBuf);
RtlCopyUnicodeString(&FsvrtDeviceExtension->VolumeName, &FsvolDeviceExtension->VolumeName);
/* mark the fsvrt device as initialized */
InterlockedIncrement(&FspFsvrtDeviceExtension(FsvrtDeviceObject)->IsMountdev);
Result = STATUS_SUCCESS;

155
src/sys/volume.c
View File

@ -38,6 +38,8 @@ static NTSTATUS FspVolumeMountNoLock(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeMakeMountdev(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeUseMountmgr(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeGetName(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
NTSTATUS FspVolumeGetNameList(
@ -76,6 +78,7 @@ NTSTATUS FspVolumeWork(
// ! #pragma alloc_text(PAGE, FspVolumeMount)
// ! #pragma alloc_text(PAGE, FspVolumeMountNoLock)
#pragma alloc_text(PAGE, FspVolumeMakeMountdev)
#pragma alloc_text(PAGE, FspVolumeUseMountmgr)
#pragma alloc_text(PAGE, FspVolumeGetName)
#pragma alloc_text(PAGE, FspVolumeGetNameList)
#pragma alloc_text(PAGE, FspVolumeGetNameListNoLock)
@ -316,8 +319,12 @@ static NTSTATUS FspVolumeCreateNoLock(
{
if (0 != FsvrtDeviceObject)
{
FspFsvrtDeviceExtension(FsvrtDeviceObject)->SectorSize =
FsvolDeviceExtension->VolumeParams.SectorSize;
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(FsvrtDeviceObject);
FsvrtDeviceExtension->SectorSize = FsvolDeviceExtension->VolumeParams.SectorSize;
RtlInitEmptyUnicodeString(&FsvrtDeviceExtension->VolumeName,
FsvrtDeviceExtension->VolumeNameBuf, sizeof FsvrtDeviceExtension->VolumeNameBuf);
RtlCopyUnicodeString(&FsvrtDeviceExtension->VolumeName, &FsvolDeviceExtension->VolumeName);
Result = FspDeviceInitialize(FsvrtDeviceObject);
}
}
@ -638,7 +645,7 @@ NTSTATUS FspVolumeMakeMountdev(
if (sizeof(GUID) > OutputBufferLength)
return STATUS_INVALID_PARAMETER;
FspDeviceGlobalLock();
FspFsvrtDeviceLockMount(FsvrtDeviceObject);
Result = FspMountdevMake(FsvrtDeviceObject, FsvolDeviceObject, Persistent);
if (!NT_SUCCESS(Result))
@ -654,7 +661,147 @@ NTSTATUS FspVolumeMakeMountdev(
Result = STATUS_SUCCESS;
exit:
FspDeviceGlobalUnlock();
FspFsvrtDeviceUnlockMount(FsvrtDeviceObject);
return Result;
}
NTSTATUS FspVolumeUseMountmgr(
PDEVICE_OBJECT FsctlDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
PAGED_CODE();
ASSERT(IRP_MJ_FILE_SYSTEM_CONTROL == IrpSp->MajorFunction);
ASSERT(IRP_MN_USER_FS_REQUEST == IrpSp->MinorFunction);
ASSERT(FSP_FSCTL_MOUNTMGR == IrpSp->Parameters.FileSystemControl.FsControlCode);
ASSERT(0 != IrpSp->FileObject->FsContext2);
PDEVICE_OBJECT FsvolDeviceObject = IrpSp->FileObject->FsContext2;
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
PDEVICE_OBJECT FsvrtDeviceObject = FsvolDeviceExtension->FsvrtDeviceObject;
ULONG InputBufferLength = IrpSp->Parameters.FileSystemControl.InputBufferLength;
NTSTATUS Result;
if (0 == FsvrtDeviceObject)
return STATUS_INVALID_PARAMETER; /* cannot only use fsvrt with mount manager */
if (1024 * sizeof(WCHAR) < InputBufferLength)
return STATUS_INVALID_PARAMETER; /* disallow very long paths */
FspFsvrtDeviceLockMount(FsvrtDeviceObject);
if (0 < InputBufferLength)
{
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(FsvrtDeviceObject);
PWCHAR MountPointBuf = Irp->AssociatedIrp.SystemBuffer;
UNICODE_STRING RegPath;
UNICODE_STRING RegName;
union
{
KEY_VALUE_PARTIAL_INFORMATION V;
UINT8 B[FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data) + sizeof(ULONG)];
} RegValue;
ULONG RegLength;
BOOLEAN Persistent = FALSE;
if (!(
2 * sizeof(WCHAR) <= InputBufferLength &&
(
(L'A' <= MountPointBuf[0] && MountPointBuf[0] <= L'Z') ||
(L'a' <= MountPointBuf[0] && MountPointBuf[0] <= L'z')
) &&
L':' == MountPointBuf[1]
))
{
Result = STATUS_INVALID_PARAMETER;
goto exit;
}
if (0 != FsvrtDeviceExtension->MountPoint.Buffer)
{
Result = STATUS_INVALID_PARAMETER;
goto exit;
}
RtlInitUnicodeString(&RegPath, L"" FSP_REGKEY);
RtlInitUnicodeString(&RegName, L"MountUseMountmgrFromFSD");
RegLength = sizeof RegValue;
Result = FspRegistryGetValue(&RegPath, &RegName, &RegValue.V, &RegLength);
if (!NT_SUCCESS(Result) || REG_DWORD != RegValue.V.Type || 1 != *(PULONG)&RegValue.V.Data)
{
Result = STATUS_ACCESS_DENIED;
goto exit;
}
Result = FspMountdevMake(FsvrtDeviceObject, FsvolDeviceObject, Persistent);
if (!NT_SUCCESS(Result))
{
if (STATUS_TOO_LATE != Result)
goto exit;
}
FsvrtDeviceExtension->MountPoint.Buffer = FspAllocNonPaged(InputBufferLength);
if (0 == FsvrtDeviceExtension->MountPoint.Buffer)
{
Result = STATUS_INSUFFICIENT_RESOURCES;
goto exit;
}
FsvrtDeviceExtension->MountPoint.Length =
FsvrtDeviceExtension->MountPoint.MaximumLength = (USHORT)InputBufferLength;
RtlCopyMemory(FsvrtDeviceExtension->MountPoint.Buffer, MountPointBuf, InputBufferLength);
if (2 * sizeof(WCHAR) == FsvrtDeviceExtension->MountPoint.Length)
Result = FspMountmgrCreateDrive(
&FsvrtDeviceExtension->VolumeName,
&FsvrtDeviceExtension->UniqueId,
&FsvrtDeviceExtension->MountPoint);
else
Result = FspMountmgrNotifyCreateDirectory(
&FsvrtDeviceExtension->VolumeName,
&FsvrtDeviceExtension->UniqueId,
&FsvrtDeviceExtension->MountPoint);
if (!NT_SUCCESS(Result))
{
FspFree(FsvrtDeviceExtension->MountPoint.Buffer);
FsvrtDeviceExtension->MountPoint.Buffer = 0;
FsvrtDeviceExtension->MountPoint.Length =
FsvrtDeviceExtension->MountPoint.MaximumLength = 0;
goto exit;
}
Irp->IoStatus.Information = 0;
Result = STATUS_SUCCESS;
}
else
{
FSP_FSVRT_DEVICE_EXTENSION *FsvrtDeviceExtension = FspFsvrtDeviceExtension(FsvrtDeviceObject);
if (0 == FsvrtDeviceExtension->MountPoint.Buffer)
{
Result = STATUS_INVALID_PARAMETER;
goto exit;
}
if (2 * sizeof(WCHAR) == FsvrtDeviceExtension->MountPoint.Length)
Result = FspMountmgrDeleteDrive(
&FsvrtDeviceExtension->MountPoint);
else
Result = FspMountmgrNotifyDeleteDirectory(
&FsvrtDeviceExtension->VolumeName,
&FsvrtDeviceExtension->MountPoint);
/* ignore Result */
FspFree(FsvrtDeviceExtension->MountPoint.Buffer);
FsvrtDeviceExtension->MountPoint.Buffer = 0;
FsvrtDeviceExtension->MountPoint.Length =
FsvrtDeviceExtension->MountPoint.MaximumLength = 0;
Irp->IoStatus.Information = 0;
Result = STATUS_SUCCESS;
}
exit:
FspFsvrtDeviceUnlockMount(FsvrtDeviceObject);
return Result;
}