Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-06-14 15:52:47 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: fsctl.c

Browse Source
This commit is contained in:
Bill Zissimopoulos
2015-11-22 12:03:25 -08:00
parent d9406873d6
commit ccf76a23c7
7 changed files with 129 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/sys/debug.c
View File

@ -86,7 +86,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_COMPLETE_MDL)
SYMBRC(IRP_MN_COMPLETE_MDL_DPC)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_DIRECTORY_CONTROL:
switch (MinorFunction)
@ -94,7 +94,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_QUERY_DIRECTORY)
SYMBRC(IRP_MN_NOTIFY_CHANGE_DIRECTORY)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_FILE_SYSTEM_CONTROL:
switch (MinorFunction)
@ -105,7 +105,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_LOAD_FILE_SYSTEM)
SYMBRC(IRP_MN_KERNEL_CALL)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_LOCK_CONTROL:
switch (MinorFunction)
@ -115,7 +115,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_UNLOCK_ALL)
SYMBRC(IRP_MN_UNLOCK_ALL_BY_KEY)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_POWER:
switch (MinorFunction)
@ -125,7 +125,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_SET_POWER)
SYMBRC(IRP_MN_QUERY_POWER)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_SYSTEM_CONTROL:
switch (MinorFunction)
@ -142,7 +142,7 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_EXECUTE_METHOD)
SYMBRC(IRP_MN_REGINFO_EX)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
case IRP_MJ_PNP:
switch (MinorFunction)
@ -153,10 +153,22 @@ const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction)
SYMBRC(IRP_MN_CANCEL_REMOVE_DEVICE)
SYMBRC(IRP_MN_SURPRISE_REMOVAL)
default:
return "[Unknown]";
return "[IRP_MN:Unknown]";
}
default:
return "";
}
}
const char *IoctlCodeSym(ULONG ControlCode)
{
switch (ControlCode)
{
SYM(FSP_FSCTL_CREATE)
SYM(FSP_FSCTL_DELETE)
SYM(FSP_FSCTL_TRANSACT)
default:
return "IOCTL:Unknown";
}
}
#endif

5
src/sys/driver.h
View File

@ -8,6 +8,7 @@
#define WINFSP_SYS_DRIVER_H_INCLUDED
#include <ntifs.h>
#include <ntstrsafe.h>
#include <wdmsec.h>
#include <winfsp/fsctl.h>
@ -176,12 +177,16 @@ FAST_IO_RELEASE_FOR_MOD_WRITE FspReleaseForModWrite;
FAST_IO_ACQUIRE_FOR_CCFLUSH FspAcquireForCcFlush;
FAST_IO_RELEASE_FOR_CCFLUSH FspReleaseForCcFlush;
/* misc */
NTSTATUS CreateGuid(GUID *Guid);
/* debug */
#if DBG
BOOLEAN HasDbgBreakPoint(const char *Function);
const char *NtStatusSym(NTSTATUS Status);
const char *IrpMajorFunctionSym(UCHAR MajorFunction);
const char *IrpMinorFunctionSym(UCHAR MajorFunction, UCHAR MinorFunction);
const char *IoctlCodeSym(ULONG ControlCode);
#endif
/* extern */

126
src/sys/fsctl.c
View File

@ -6,23 +6,23 @@
#include <sys/driver.h>
static
NTSTATUS
FspFsctlCreateVolume(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp);
static
NTSTATUS
FspFsctlDeleteVolume(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp);
static
NTSTATUS
FspFsvolTransact(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp);
static DRIVER_DISPATCH FspFsctlFileSystemControl;
static DRIVER_DISPATCH FspFsvrtFileSystemControl;
static DRIVER_DISPATCH FspFsvolFileSystemControl;
static NTSTATUS FspFsctlCreateVolume(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
static NTSTATUS FspFsvolDeleteVolume(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
static NTSTATUS FspFsvolTransact(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
static NTSTATUS FspFsctlFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
static NTSTATUS FspFsvrtFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
static NTSTATUS FspFsvolFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
DRIVER_DISPATCH FspFileSystemControl;
#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE, FspFsctlCreateVolume)
#pragma alloc_text(PAGE, FspFsctlDeleteVolume)
#pragma alloc_text(PAGE, FspFsvolDeleteVolume)
#pragma alloc_text(PAGE, FspFsvolTransact)
#pragma alloc_text(PAGE, FspFsctlFileSystemControl)
#pragma alloc_text(PAGE, FspFsvrtFileSystemControl)
@ -30,66 +30,85 @@ DRIVER_DISPATCH FspFileSystemControl;
#pragma alloc_text(PAGE, FspFileSystemControl)
#endif
static
NTSTATUS
FspFsctlCreateVolume(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp)
static NTSTATUS FspFsctlCreateVolume(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
if (FSP_FSCTL_CREATE_BUFFER_SIZEMAX > IrpSp->Parameters.FileSystemControl.OutputBufferLength)
return STATUS_BUFFER_TOO_SMALL;
NTSTATUS Result;
GUID Guid;
Result = CreateGuid(&Guid);
if (!NT_SUCCESS(Result))
return Result;
PDEVICE_OBJECT FsvrtDeviceObject;
UNICODE_STRING DeviceSddl;
UNICODE_STRING DeviceName;
RtlInitUnicodeString(&DeviceSddl, L"" DEVICE_SDDL);
RtlInitEmptyUnicodeString(&DeviceName, Irp->AssociatedIrp.SystemBuffer, FSP_FSCTL_CREATE_BUFFER_SIZEMAX);
Result = RtlUnicodeStringPrintf(&DeviceName,
L"\\Device\\Volume{%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
Guid.Data1, Guid.Data2, Guid.Data3,
Guid.Data4[0], Guid.Data4[1], Guid.Data4[2], Guid.Data4[3],
Guid.Data4[4], Guid.Data4[5], Guid.Data4[6], Guid.Data4[7]);
ASSERT(NT_SUCCESS(Result));
Result = IoCreateDeviceSecure(DeviceObject->DriverObject,
sizeof(FSP_FSVRT_DEVICE_EXTENSION), &DeviceName, DeviceObject->DeviceType, 0, FALSE,
&DeviceSddl, 0,
&FsvrtDeviceObject);
if (!NT_SUCCESS(Result))
return Result;
FspDeviceExtension(FsvrtDeviceObject)->Kind = FspFsvrtDeviceExtensionKind;
Irp->IoStatus.Information = DeviceName.Length + 1;
return Result;
}
static NTSTATUS FspFsvolDeleteVolume(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
return STATUS_INVALID_DEVICE_REQUEST;
}
static
NTSTATUS
FspFsctlDeleteVolume(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp)
static NTSTATUS FspFsvolTransact(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
return STATUS_INVALID_DEVICE_REQUEST;
}
static
NTSTATUS
FspFsvolTransact(PDEVICE_OBJECT DeviceObject, PIO_STACK_LOCATION IrpSp)
{
return STATUS_INVALID_DEVICE_REQUEST;
}
static
NTSTATUS
FspFsctlFileSystemControl(
_In_ PDEVICE_OBJECT DeviceObject,
_In_ PIRP Irp)
static NTSTATUS FspFsctlFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
NTSTATUS Result = STATUS_INVALID_DEVICE_REQUEST;
PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
switch (IrpSp->MinorFunction)
{
case IRP_MN_USER_FS_REQUEST:
switch (IrpSp->Parameters.FileSystemControl.FsControlCode)
{
case FSP_FSCTL_CREATE:
Result = FspFsctlCreateVolume(DeviceObject, IrpSp);
break;
case FSP_FSCTL_DELETE:
Result = FspFsctlDeleteVolume(DeviceObject, IrpSp);
Result = FspFsctlCreateVolume(DeviceObject, Irp, IrpSp);
break;
}
}
return Result;
}
static
NTSTATUS
FspFsvrtFileSystemControl(
_In_ PDEVICE_OBJECT DeviceObject,
_In_ PIRP Irp)
static NTSTATUS FspFsvrtFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
NTSTATUS Result = STATUS_INVALID_DEVICE_REQUEST;
PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
switch (IrpSp->MinorFunction)
{
case IRP_MN_USER_FS_REQUEST:
switch (IrpSp->Parameters.FileSystemControl.FsControlCode)
{
case FSP_FSCTL_DELETE:
Result = FspFsvolDeleteVolume(DeviceObject, Irp, IrpSp);
break;
case FSP_FSCTL_TRANSACT:
Result = FspFsvolTransact(DeviceObject, IrpSp);
Result = FspFsvolTransact(DeviceObject, Irp, IrpSp);
break;
}
break;
@ -97,14 +116,10 @@ FspFsvrtFileSystemControl(
return Result;
}
static
NTSTATUS
FspFsvolFileSystemControl(
_In_ PDEVICE_OBJECT DeviceObject,
_In_ PIRP Irp)
static NTSTATUS FspFsvolFileSystemControl(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
NTSTATUS Result = STATUS_INVALID_DEVICE_REQUEST;
PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
switch (IrpSp->MinorFunction)
{
case IRP_MN_USER_FS_REQUEST:
@ -129,14 +144,19 @@ FspFileSystemControl(
switch (FspDeviceExtension(DeviceObject)->Kind)
{
case FspFsvolDeviceExtensionKind:
FSP_RETURN(Result = FspFsctlFileSystemControl(DeviceObject, Irp));
FSP_RETURN(Result = FspFsctlFileSystemControl(DeviceObject, Irp, IrpSp));
case FspFsvrtDeviceExtensionKind:
FSP_RETURN(Result = FspFsvrtFileSystemControl(DeviceObject, Irp));
FSP_RETURN(Result = FspFsvrtFileSystemControl(DeviceObject, Irp, IrpSp));
case FspFsctlDeviceExtensionKind:
FSP_RETURN(Result = FspFsvolFileSystemControl(DeviceObject, Irp));
FSP_RETURN(Result = FspFsvolFileSystemControl(DeviceObject, Irp, IrpSp));
default:
FSP_RETURN(Result = STATUS_INVALID_DEVICE_REQUEST);
}
FSP_LEAVE_MJ("", 0);
FSP_LEAVE_MJ(
"FileObject=%p%s%s",
IrpSp->FileObject,
IRP_MN_USER_FS_REQUEST == IrpSp->MinorFunction ? ", " : "",
IRP_MN_USER_FS_REQUEST == IrpSp->MinorFunction ?
IoctlCodeSym(IrpSp->Parameters.FileSystemControl.FsControlCode) : "");
}

24
src/sys/misc.c Normal file
View File

@ -0,0 +1,24 @@
/**
* @file sys/misc.c
*
* @copyright 2015 Bill Zissimopoulos
*/
#include <sys/driver.h>
NTSTATUS CreateGuid(GUID *Guid);
#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE, CreateGuid)
#endif
NTSTATUS CreateGuid(GUID *Guid)
{
NTSTATUS Result;
int Retries = 3;
do
{
Result = ExUuidCreate(Guid);
} while (!NT_SUCCESS(Result) && 0 < --Retries);
return Result;
}