mirror of
https://github.com/winfsp/winfsp.git
synced 2025-06-09 05:22:12 -05:00
346 lines
12 KiB
C
346 lines
12 KiB
C
/**
|
|
* @file sys/volinfo.c
|
|
*
|
|
* @copyright 2015 Bill Zissimopoulos
|
|
*/
|
|
|
|
#include <sys/driver.h>
|
|
|
|
static NTSTATUS FspFsvolQueryFsAttributeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd);
|
|
static NTSTATUS FspFsvolQueryFsDeviceInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd);
|
|
static NTSTATUS FspFsvolQueryFsFullSizeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo);
|
|
static NTSTATUS FspFsvolQueryFsSizeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo);
|
|
static NTSTATUS FspFsvolQueryFsVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo);
|
|
static NTSTATUS FspFsvolQueryVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
|
|
FSP_IOCMPL_DISPATCH FspFsvolQueryVolumeInformationComplete;
|
|
static NTSTATUS FspFsvolSetVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
|
|
FSP_IOCMPL_DISPATCH FspFsvolSetVolumeInformationComplete;
|
|
FSP_DRIVER_DISPATCH FspQueryVolumeInformation;
|
|
FSP_DRIVER_DISPATCH FspSetVolumeInformation;
|
|
|
|
#ifdef ALLOC_PRAGMA
|
|
#pragma alloc_text(PAGE, FspFsvolQueryFsAttributeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryFsDeviceInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryFsFullSizeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryFsSizeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryFsVolumeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryVolumeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolQueryVolumeInformationComplete)
|
|
#pragma alloc_text(PAGE, FspFsvolSetVolumeInformation)
|
|
#pragma alloc_text(PAGE, FspFsvolSetVolumeInformationComplete)
|
|
#pragma alloc_text(PAGE, FspQueryVolumeInformation)
|
|
#pragma alloc_text(PAGE, FspSetVolumeInformation)
|
|
#endif
|
|
|
|
#define GETVOLUMEINFO() \
|
|
FSP_FSCTL_VOLUME_INFO VolumeInfoBuf;\
|
|
if (0 == VolumeInfo) \
|
|
{ \
|
|
if (!FspFsvolTryGetVolumeInfo(FsvolDeviceObject, &VolumeInfoBuf))\
|
|
return FSP_STATUS_IOQ_POST; \
|
|
VolumeInfo = &VolumeInfoBuf; \
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryFsAttributeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
if (*PBuffer + sizeof(FILE_FS_ATTRIBUTE_INFORMATION) > BufferEnd)
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
NTSTATUS Result = STATUS_SUCCESS;
|
|
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
|
|
PFILE_FS_ATTRIBUTE_INFORMATION Info = (PFILE_FS_ATTRIBUTE_INFORMATION)*PBuffer;
|
|
PUINT8 Buffer = (PUINT8)Info->FileSystemName;
|
|
ULONG CopyLength;
|
|
|
|
Info->FileSystemAttributes =
|
|
(FsvolDeviceExtension->VolumeParams.CaseSensitiveSearch ? FILE_CASE_SENSITIVE_SEARCH : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.CasePreservedNames ? FILE_CASE_PRESERVED_NAMES : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.UnicodeOnDisk ? FILE_UNICODE_ON_DISK : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.PersistentAcls ? FILE_PERSISTENT_ACLS : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.ReparsePoints ? FILE_SUPPORTS_REPARSE_POINTS : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.NamedStreams ? FILE_NAMED_STREAMS : 0) |
|
|
//(FsvolDeviceExtension->VolumeParams.HardLinks ? FILE_SUPPORTS_HARD_LINKS : 0) |
|
|
//(FsvolDeviceExtension->VolumeParams.ExtendedAttributes ? FILE_SUPPORTS_EXTENDED_ATTRIBUTES : 0) |
|
|
(FsvolDeviceExtension->VolumeParams.ReadOnlyVolume ? FILE_READ_ONLY_VOLUME : 0);
|
|
Info->MaximumComponentNameLength = FsvolDeviceExtension->VolumeParams.MaxComponentLength;
|
|
Info->FileSystemNameLength = sizeof L"" DRIVER_NAME - sizeof(WCHAR);
|
|
|
|
CopyLength = Info->FileSystemNameLength;
|
|
if (Buffer + CopyLength > BufferEnd)
|
|
{
|
|
CopyLength = (ULONG)(BufferEnd - Buffer);
|
|
Result = STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
RtlCopyMemory(Buffer, L"" DRIVER_NAME, CopyLength);
|
|
Buffer += CopyLength;
|
|
|
|
*PBuffer = Buffer;
|
|
|
|
return Result;
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryFsDeviceInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
if (*PBuffer + sizeof(FILE_FS_DEVICE_INFORMATION) > BufferEnd)
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
PFILE_FS_DEVICE_INFORMATION Info = (PFILE_FS_DEVICE_INFORMATION)*PBuffer;
|
|
|
|
Info->DeviceType = FsvolDeviceObject->DeviceType;
|
|
Info->Characteristics = FsvolDeviceObject->Characteristics;
|
|
|
|
*PBuffer += sizeof(FILE_FS_DEVICE_INFORMATION);
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryFsFullSizeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
if (*PBuffer + sizeof(FILE_FS_FULL_SIZE_INFORMATION) > BufferEnd)
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
GETVOLUMEINFO();
|
|
|
|
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
|
|
PFILE_FS_FULL_SIZE_INFORMATION Info = (PFILE_FS_FULL_SIZE_INFORMATION)*PBuffer;
|
|
UINT64 AllocationUnit = FsvolDeviceExtension->VolumeParams.SectorSize *
|
|
FsvolDeviceExtension->VolumeParams.SectorsPerAllocationUnit;
|
|
|
|
Info->TotalAllocationUnits.QuadPart = VolumeInfo->TotalSize / AllocationUnit;
|
|
Info->CallerAvailableAllocationUnits.QuadPart =
|
|
Info->ActualAvailableAllocationUnits.QuadPart = VolumeInfo->FreeSize / AllocationUnit;
|
|
Info->SectorsPerAllocationUnit = FsvolDeviceExtension->VolumeParams.SectorsPerAllocationUnit;
|
|
Info->BytesPerSector = FsvolDeviceExtension->VolumeParams.SectorSize;
|
|
|
|
*PBuffer += sizeof(FILE_FS_FULL_SIZE_INFORMATION);
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryFsSizeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
if (*PBuffer + sizeof(FILE_FS_SIZE_INFORMATION) > BufferEnd)
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
GETVOLUMEINFO();
|
|
|
|
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
|
|
PFILE_FS_SIZE_INFORMATION Info = (PFILE_FS_SIZE_INFORMATION)*PBuffer;
|
|
UINT64 AllocationUnit = FsvolDeviceExtension->VolumeParams.SectorSize *
|
|
FsvolDeviceExtension->VolumeParams.SectorsPerAllocationUnit;
|
|
|
|
Info->TotalAllocationUnits.QuadPart = VolumeInfo->TotalSize / AllocationUnit;
|
|
Info->AvailableAllocationUnits.QuadPart = VolumeInfo->FreeSize / AllocationUnit;
|
|
Info->SectorsPerAllocationUnit = FsvolDeviceExtension->VolumeParams.SectorsPerAllocationUnit;
|
|
Info->BytesPerSector = FsvolDeviceExtension->VolumeParams.SectorSize;
|
|
|
|
*PBuffer += sizeof(FILE_FS_SIZE_INFORMATION);
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryFsVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PUINT8 *PBuffer, PUINT8 BufferEnd,
|
|
const FSP_FSCTL_VOLUME_INFO *VolumeInfo)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
if (*PBuffer + sizeof(FILE_FS_VOLUME_INFORMATION) > BufferEnd)
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
GETVOLUMEINFO();
|
|
|
|
NTSTATUS Result = STATUS_SUCCESS;
|
|
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
|
|
PFILE_FS_VOLUME_INFORMATION Info = (PFILE_FS_VOLUME_INFORMATION)*PBuffer;
|
|
PUINT8 Buffer = (PUINT8)Info->VolumeLabel;
|
|
ULONG CopyLength;
|
|
|
|
Info->VolumeCreationTime.QuadPart = VolumeInfo->VolumeCreationTime;
|
|
Info->VolumeSerialNumber = FsvolDeviceExtension->VolumeParams.VolumeSerialNumber;
|
|
Info->VolumeLabelLength = VolumeInfo->VolumeLabelLength;
|
|
Info->SupportsObjects = FALSE;
|
|
|
|
CopyLength = VolumeInfo->VolumeLabelLength;
|
|
if (Buffer + CopyLength > BufferEnd)
|
|
{
|
|
CopyLength = (ULONG)(BufferEnd - Buffer);
|
|
Result = STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
RtlCopyMemory(Buffer, VolumeInfo->VolumeLabel, CopyLength);
|
|
Buffer += CopyLength;
|
|
|
|
*PBuffer = Buffer;
|
|
|
|
return Result;
|
|
}
|
|
|
|
static NTSTATUS FspFsvolQueryVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
NTSTATUS Result;
|
|
PUINT8 Buffer = Irp->AssociatedIrp.SystemBuffer;
|
|
PUINT8 BufferEnd = Buffer + IrpSp->Parameters.QueryFile.Length;
|
|
|
|
switch (IrpSp->Parameters.QueryVolume.FsInformationClass)
|
|
{
|
|
case FileFsAttributeInformation:
|
|
Result = FspFsvolQueryFsAttributeInformation(FsvolDeviceObject, &Buffer, BufferEnd);
|
|
break;
|
|
case FileFsDeviceInformation:
|
|
Result = FspFsvolQueryFsDeviceInformation(FsvolDeviceObject, &Buffer, BufferEnd);
|
|
break;
|
|
case FileFsFullSizeInformation:
|
|
Result = FspFsvolQueryFsFullSizeInformation(FsvolDeviceObject, &Buffer, BufferEnd, 0);
|
|
break;
|
|
case FileFsSizeInformation:
|
|
Result = FspFsvolQueryFsSizeInformation(FsvolDeviceObject, &Buffer, BufferEnd, 0);
|
|
break;
|
|
case FileFsVolumeInformation:
|
|
Result = FspFsvolQueryFsVolumeInformation(FsvolDeviceObject, &Buffer, BufferEnd, 0);
|
|
break;
|
|
default:
|
|
Result = STATUS_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
if (FSP_STATUS_IOQ_POST != Result)
|
|
{
|
|
Irp->IoStatus.Information = (UINT_PTR)(Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer);
|
|
return Result;
|
|
}
|
|
|
|
FSP_FSCTL_TRANSACT_REQ *Request;
|
|
|
|
Result = FspIopCreateRequest(Irp, 0, 0, &Request);
|
|
if (!NT_SUCCESS(Result))
|
|
return Result;
|
|
|
|
Request->Kind = FspFsctlTransactQueryVolumeInformationKind;
|
|
|
|
return FSP_STATUS_IOQ_POST;
|
|
}
|
|
|
|
NTSTATUS FspFsvolQueryVolumeInformationComplete(
|
|
PIRP Irp, const FSP_FSCTL_TRANSACT_RSP *Response)
|
|
{
|
|
FSP_ENTER_IOC(PAGED_CODE());
|
|
|
|
if (!NT_SUCCESS(Response->IoStatus.Status))
|
|
{
|
|
Irp->IoStatus.Information = Response->IoStatus.Information;
|
|
Result = Response->IoStatus.Status;
|
|
FSP_RETURN();
|
|
}
|
|
|
|
PDEVICE_OBJECT FsvolDeviceObject = IrpSp->DeviceObject;
|
|
PUINT8 Buffer = Irp->AssociatedIrp.SystemBuffer;
|
|
PUINT8 BufferEnd = Buffer + IrpSp->Parameters.QueryFile.Length;
|
|
|
|
FspFsvolSetVolumeInfo(FsvolDeviceObject, &Response->Rsp.QueryVolumeInformation.VolumeInfo);
|
|
|
|
switch (IrpSp->Parameters.QueryVolume.FsInformationClass)
|
|
{
|
|
case FileFsFullSizeInformation:
|
|
Result = FspFsvolQueryFsFullSizeInformation(FsvolDeviceObject, &Buffer, BufferEnd,
|
|
&Response->Rsp.QueryVolumeInformation.VolumeInfo);
|
|
break;
|
|
case FileFsSizeInformation:
|
|
Result = FspFsvolQueryFsSizeInformation(FsvolDeviceObject, &Buffer, BufferEnd,
|
|
&Response->Rsp.QueryVolumeInformation.VolumeInfo);
|
|
break;
|
|
case FileFsVolumeInformation:
|
|
Result = FspFsvolQueryFsVolumeInformation(FsvolDeviceObject, &Buffer, BufferEnd,
|
|
&Response->Rsp.QueryVolumeInformation.VolumeInfo);
|
|
break;
|
|
default:
|
|
ASSERT(0);
|
|
Result = STATUS_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
ASSERT(FSP_STATUS_IOQ_POST != Result);
|
|
|
|
Irp->IoStatus.Information = (UINT_PTR)(Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer);
|
|
|
|
FSP_LEAVE_IOC("%s",
|
|
FsInformationClassSym(IrpSp->Parameters.QueryVolume.FsInformationClass));
|
|
}
|
|
|
|
static NTSTATUS FspFsvolSetVolumeInformation(
|
|
PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
|
|
{
|
|
PAGED_CODE();
|
|
|
|
return STATUS_INVALID_DEVICE_REQUEST;
|
|
}
|
|
|
|
NTSTATUS FspFsvolSetVolumeInformationComplete(
|
|
PIRP Irp, const FSP_FSCTL_TRANSACT_RSP *Response)
|
|
{
|
|
FSP_ENTER_IOC(PAGED_CODE());
|
|
|
|
FSP_LEAVE_IOC("%s",
|
|
FsInformationClassSym(IrpSp->Parameters.SetVolume.FsInformationClass));
|
|
}
|
|
|
|
NTSTATUS FspQueryVolumeInformation(
|
|
PDEVICE_OBJECT DeviceObject, PIRP Irp)
|
|
{
|
|
FSP_ENTER_MJ(PAGED_CODE());
|
|
|
|
switch (FspDeviceExtension(DeviceObject)->Kind)
|
|
{
|
|
case FspFsvolDeviceExtensionKind:
|
|
FSP_RETURN(Result = FspFsvolQueryVolumeInformation(DeviceObject, Irp, IrpSp));
|
|
default:
|
|
FSP_RETURN(Result = STATUS_INVALID_DEVICE_REQUEST);
|
|
}
|
|
|
|
FSP_LEAVE_MJ("%s",
|
|
FsInformationClassSym(IrpSp->Parameters.QueryVolume.FsInformationClass));
|
|
}
|
|
|
|
NTSTATUS FspSetVolumeInformation(
|
|
PDEVICE_OBJECT DeviceObject, PIRP Irp)
|
|
{
|
|
FSP_ENTER_MJ(PAGED_CODE());
|
|
|
|
switch (FspDeviceExtension(DeviceObject)->Kind)
|
|
{
|
|
case FspFsvolDeviceExtensionKind:
|
|
FSP_RETURN(Result = FspFsvolSetVolumeInformation(DeviceObject, Irp, IrpSp));
|
|
default:
|
|
FSP_RETURN(Result = STATUS_INVALID_DEVICE_REQUEST);
|
|
}
|
|
|
|
FSP_LEAVE_MJ("%s",
|
|
FsInformationClassSym(IrpSp->Parameters.SetVolume.FsInformationClass));
|
|
}
|