Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 17:03:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: IRP_MJ_QUERY_INFORMATION: refactoring

Browse Source
This commit is contained in:
Bill Zissimopoulos 2016-01-26 13:57:09 -08:00
parent ba0896d3c8
commit a9838939e4
1 changed files with 238 additions and 120 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

346
src/sys/fileinfo.c
View File

@ -7,21 +7,26 @@
#include <sys/driver.h> #include <sys/driver.h>
static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo); PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response);
static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo); PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response);
static NTSTATUS FspFsvolQueryBasicInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryBasicInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo); PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response);
static NTSTATUS FspFsvolQueryInternalInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryInternalInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd); PVOID *PBuffer, PVOID BufferEnd);
static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd); PVOID *PBuffer, PVOID BufferEnd);
static NTSTATUS FspFsvolQueryNetworkOpenInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryNetworkOpenInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo); PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response);
static NTSTATUS FspFsvolQueryPositionInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryPositionInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd); PVOID *PBuffer, PVOID BufferEnd);
static NTSTATUS FspFsvolQueryStandardInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryStandardInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd); PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response);
static NTSTATUS FspFsvolQueryInformation( static NTSTATUS FspFsvolQueryInformation(
PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp); PDEVICE_OBJECT FsvolDeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
FSP_IOCMPL_DISPATCH FspFsvolQueryInformationComplete; FSP_IOCMPL_DISPATCH FspFsvolQueryInformationComplete;
@ -74,86 +79,135 @@ FSP_DRIVER_DISPATCH FspSetInformation;
#pragma alloc_text(PAGE, FspSetInformation) #pragma alloc_text(PAGE, FspSetInformation)
#endif #endif
#define GETFILEINFO() \ enum
FSP_FSCTL_FILE_INFO FileInfoBuf; \ {
if (0 == FileInfo) \ RequestFileNode = 0,
{ \ RequestAcquireFlags = 1,
if (!FspFileNodeTryGetFileInfo((FSP_FILE_NODE *)FileObject->FsContext, &FileInfoBuf))\ };
return FSP_STATUS_IOQ_POST; \
FileInfo = &FileInfoBuf; \
}
static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo) PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_ALL_INFORMATION) > BufferEnd) PFILE_ALL_INFORMATION Info = (PFILE_ALL_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfoBuf;
const FSP_FSCTL_FILE_INFO *FileInfo;
if (0 == Request)
{
if ((PVOID)(Info + 1) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
PFILE_ALL_INFORMATION Info = (PFILE_ALL_INFORMATION)*PBuffer; if (!FspFileNodeTryGetFileInfo(FileNode, &FileInfoBuf))
return FSP_STATUS_IOQ_POST;
FileInfo = &FileInfoBuf;
if (0 == FileInfo) FspFileNodeAcquireShared(FileNode, Both);
}
else if (0 == Response)
{ {
FSP_FILE_NODE *FileNode = FileObject->FsContext; FspFileNodeAcquireShared(FileNode, Both);
FspIopRequestContext(Request, RequestFileNode) = FileNode;
FspIopRequestContext(Request, RequestAcquireFlags) = (PVOID)FspFileNodeAcquireBoth;
FspFileNodeAcquireShared(FileNode, Main); return FSP_STATUS_IOQ_POST;
}
else
{
FspIopRequestContext(Request, RequestFileNode) = 0;
*PBuffer = (PVOID)&Info->PositionInformation; FileInfo = &Response->Rsp.QueryInformation.FileInfo;
FspFsvolQueryPositionInformation(FileObject, PBuffer, BufferEnd); }
*PBuffer = (PVOID)&Info->StandardInformation; Info->StandardInformation.AllocationSize = FileNode->Header.AllocationSize;
FspFsvolQueryStandardInformation(FileObject, PBuffer, BufferEnd); Info->StandardInformation.EndOfFile = FileNode->Header.FileSize;
FspFileNodeRelease(FileNode, Pgio);
Info->StandardInformation.NumberOfLinks = 1;
Info->StandardInformation.DeletePending = FileObject->DeletePending;
Info->StandardInformation.Directory = FileNode->IsDirectory;
Info->PositionInformation.CurrentByteOffset = FileObject->CurrentByteOffset;
FspFileNodeRelease(FileNode, Main); FspFileNodeRelease(FileNode, Main);
Info->BasicInformation.CreationTime.QuadPart = FileInfo->CreationTime;
Info->BasicInformation.LastAccessTime.QuadPart = FileInfo->LastAccessTime;
Info->BasicInformation.LastWriteTime.QuadPart = FileInfo->LastWriteTime;
Info->BasicInformation.ChangeTime.QuadPart = FileInfo->ChangeTime;
Info->BasicInformation.FileAttributes = 0 != FileInfo->FileAttributes ?
FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL;
Info->EaInformation.EaSize = 0; Info->EaInformation.EaSize = 0;
*PBuffer = (PVOID)&Info->InternalInformation; Info->InternalInformation.IndexNumber.QuadPart = FileNode->IndexNumber;
FspFsvolQueryInternalInformation(FileObject, PBuffer, BufferEnd);
}
GETFILEINFO();
*PBuffer = (PVOID)&Info->BasicInformation;
FspFsvolQueryBasicInformation(FileObject, PBuffer, BufferEnd, FileInfo);
*PBuffer = (PVOID)&Info->NameInformation; *PBuffer = (PVOID)&Info->NameInformation;
return FspFsvolQueryNameInformation(FileObject, PBuffer, BufferEnd); return FspFsvolQueryNameInformation(FileObject, PBuffer, BufferEnd);
} }
static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo) PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_ATTRIBUTE_TAG_INFORMATION) > BufferEnd) PFILE_ATTRIBUTE_TAG_INFORMATION Info = (PFILE_ATTRIBUTE_TAG_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfoBuf;
const FSP_FSCTL_FILE_INFO *FileInfo;
if (0 == Request)
{
if ((PVOID)(Info + 1) > (PVOID)BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
GETFILEINFO(); if (!FspFileNodeTryGetFileInfo(FileNode, &FileInfoBuf))
return FSP_STATUS_IOQ_POST;
PFILE_ATTRIBUTE_TAG_INFORMATION Info = (PFILE_ATTRIBUTE_TAG_INFORMATION)*PBuffer; FileInfo = &FileInfoBuf;
}
else if (0 == Response)
return FSP_STATUS_IOQ_POST;
else
FileInfo = &Response->Rsp.QueryInformation.FileInfo;
Info->FileAttributes = 0 != FileInfo->FileAttributes ? Info->FileAttributes = 0 != FileInfo->FileAttributes ?
FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL; FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL;
Info->ReparseTag = FileInfo->ReparseTag; Info->ReparseTag = FileInfo->ReparseTag;
*PBuffer += sizeof(FILE_ATTRIBUTE_TAG_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static NTSTATUS FspFsvolQueryBasicInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryBasicInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo) PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_BASIC_INFORMATION) > BufferEnd) PFILE_BASIC_INFORMATION Info = (PFILE_BASIC_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfoBuf;
const FSP_FSCTL_FILE_INFO *FileInfo;
if (0 == Request)
{
if ((PVOID)(Info + 1) > (PVOID)BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
GETFILEINFO(); if (!FspFileNodeTryGetFileInfo(FileNode, &FileInfoBuf))
return FSP_STATUS_IOQ_POST;
PFILE_BASIC_INFORMATION Info = (PFILE_BASIC_INFORMATION)*PBuffer; FileInfo = &FileInfoBuf;
}
else if (0 == Response)
return FSP_STATUS_IOQ_POST;
else
FileInfo = &Response->Rsp.QueryInformation.FileInfo;
Info->CreationTime.QuadPart = FileInfo->CreationTime; Info->CreationTime.QuadPart = FileInfo->CreationTime;
Info->LastAccessTime.QuadPart = FileInfo->LastAccessTime; Info->LastAccessTime.QuadPart = FileInfo->LastAccessTime;
@ -162,60 +216,60 @@ static NTSTATUS FspFsvolQueryBasicInformation(PFILE_OBJECT FileObject,
Info->FileAttributes = 0 != FileInfo->FileAttributes ? Info->FileAttributes = 0 != FileInfo->FileAttributes ?
FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL; FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL;
*PBuffer += sizeof(FILE_BASIC_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static NTSTATUS FspFsvolQueryInternalInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryInternalInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd) PVOID *PBuffer, PVOID BufferEnd)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_INTERNAL_INFORMATION) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
PFILE_INTERNAL_INFORMATION Info = (PFILE_INTERNAL_INFORMATION)*PBuffer; PFILE_INTERNAL_INFORMATION Info = (PFILE_INTERNAL_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
if ((PVOID)(Info + 1) > (PVOID)BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
Info->IndexNumber.QuadPart = FileNode->IndexNumber; Info->IndexNumber.QuadPart = FileNode->IndexNumber;
*PBuffer += sizeof(FILE_INTERNAL_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd) PVOID *PBuffer, PVOID BufferEnd)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_NAME_INFORMATION) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
NTSTATUS Result = STATUS_SUCCESS; NTSTATUS Result = STATUS_SUCCESS;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension =
FspFsvolDeviceExtension(FileNode->FsvolDeviceObject);
PFILE_NAME_INFORMATION Info = (PFILE_NAME_INFORMATION)*PBuffer; PFILE_NAME_INFORMATION Info = (PFILE_NAME_INFORMATION)*PBuffer;
PUINT8 Buffer = (PUINT8)Info->FileName; PUINT8 Buffer = (PUINT8)Info->FileName;
ULONG CopyLength; ULONG CopyLength;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension =
FspFsvolDeviceExtension(FileNode->FsvolDeviceObject);
if ((PVOID)(Info + 1) > (PVOID)BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
Info->FileNameLength = FsvolDeviceExtension->VolumePrefix.Length + FileNode->FileName.Length; Info->FileNameLength = FsvolDeviceExtension->VolumePrefix.Length + FileNode->FileName.Length;
CopyLength = FsvolDeviceExtension->VolumePrefix.Length; CopyLength = FsvolDeviceExtension->VolumePrefix.Length;
if (Buffer + CopyLength > BufferEnd) if (Buffer + CopyLength > (PUINT8)BufferEnd)
{ {
CopyLength = (ULONG)(BufferEnd - Buffer); CopyLength = (ULONG)((PUINT8)BufferEnd - Buffer);
Result = STATUS_BUFFER_OVERFLOW; Result = STATUS_BUFFER_OVERFLOW;
} }
RtlCopyMemory(Buffer, FsvolDeviceExtension->VolumePrefix.Buffer, CopyLength); RtlCopyMemory(Buffer, FsvolDeviceExtension->VolumePrefix.Buffer, CopyLength);
Buffer += CopyLength; Buffer += CopyLength;
CopyLength = FileNode->FileName.Length; CopyLength = FileNode->FileName.Length;
if (Buffer + CopyLength > BufferEnd) if (Buffer + CopyLength > (PUINT8)BufferEnd)
{ {
CopyLength = (ULONG)(BufferEnd - Buffer); CopyLength = (ULONG)((PUINT8)BufferEnd - Buffer);
Result = STATUS_BUFFER_OVERFLOW; Result = STATUS_BUFFER_OVERFLOW;
} }
RtlCopyMemory(Buffer, FileNode->FileName.Buffer, CopyLength); RtlCopyMemory(Buffer, FileNode->FileName.Buffer, CopyLength);
@ -227,28 +281,46 @@ static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject,
} }
static NTSTATUS FspFsvolQueryNetworkOpenInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryNetworkOpenInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd, const FSP_FSCTL_FILE_INFO *FileInfo) PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_NETWORK_OPEN_INFORMATION) > BufferEnd) PFILE_NETWORK_OPEN_INFORMATION Info = (PFILE_NETWORK_OPEN_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfoBuf;
const FSP_FSCTL_FILE_INFO *FileInfo;
if (0 == Request)
{
if ((PVOID)(Info + 1) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
PFILE_NETWORK_OPEN_INFORMATION Info = (PFILE_NETWORK_OPEN_INFORMATION)*PBuffer; if (!FspFileNodeTryGetFileInfo(FileNode, &FileInfoBuf))
return FSP_STATUS_IOQ_POST;
FileInfo = &FileInfoBuf;
if (0 == FileInfo) FspFileNodeAcquireShared(FileNode, Both);
}
else if (0 == Response)
{ {
FSP_FILE_NODE *FileNode = FileObject->FsContext; FspFileNodeAcquireShared(FileNode, Both);
FspIopRequestContext(Request, RequestFileNode) = FileNode;
FspIopRequestContext(Request, RequestAcquireFlags) = (PVOID)FspFileNodeAcquireBoth;
FspFileNodeAcquireShared(FileNode, Main); return FSP_STATUS_IOQ_POST;
}
else
{
FspIopRequestContext(Request, RequestFileNode) = 0;
FileInfo = &Response->Rsp.QueryInformation.FileInfo;
}
Info->AllocationSize = FileNode->Header.AllocationSize; Info->AllocationSize = FileNode->Header.AllocationSize;
Info->EndOfFile = FileNode->Header.FileSize; Info->EndOfFile = FileNode->Header.FileSize;
FspFileNodeRelease(FileNode, Main); FspFileNodeRelease(FileNode, Both);
}
GETFILEINFO();
Info->CreationTime.QuadPart = FileInfo->CreationTime; Info->CreationTime.QuadPart = FileInfo->CreationTime;
Info->LastAccessTime.QuadPart = FileInfo->LastAccessTime; Info->LastAccessTime.QuadPart = FileInfo->LastAccessTime;
@ -257,21 +329,21 @@ static NTSTATUS FspFsvolQueryNetworkOpenInformation(PFILE_OBJECT FileObject,
Info->FileAttributes = 0 != FileInfo->FileAttributes ? Info->FileAttributes = 0 != FileInfo->FileAttributes ?
FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL; FileInfo->FileAttributes : FILE_ATTRIBUTE_NORMAL;
*PBuffer += sizeof(FILE_BASIC_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static NTSTATUS FspFsvolQueryPositionInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryPositionInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd) PVOID *PBuffer, PVOID BufferEnd)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_POSITION_INFORMATION) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
PFILE_POSITION_INFORMATION Info = (PFILE_POSITION_INFORMATION)*PBuffer; PFILE_POSITION_INFORMATION Info = (PFILE_POSITION_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
if ((PVOID)(Info + 1) > (PVOID)BufferEnd)
return STATUS_BUFFER_TOO_SMALL;
FspFileNodeAcquireShared(FileNode, Main); FspFileNodeAcquireShared(FileNode, Main);
@ -279,33 +351,60 @@ static NTSTATUS FspFsvolQueryPositionInformation(PFILE_OBJECT FileObject,
FspFileNodeRelease(FileNode, Main); FspFileNodeRelease(FileNode, Main);
*PBuffer += sizeof(FILE_POSITION_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static NTSTATUS FspFsvolQueryStandardInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryStandardInformation(PFILE_OBJECT FileObject,
PUINT8 *PBuffer, PUINT8 BufferEnd) PVOID *PBuffer, PVOID BufferEnd,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)
{ {
PAGED_CODE(); PAGED_CODE();
if (*PBuffer + sizeof(FILE_STANDARD_INFORMATION) > BufferEnd) PFILE_STANDARD_INFORMATION Info = (PFILE_STANDARD_INFORMATION)*PBuffer;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfoBuf;
const FSP_FSCTL_FILE_INFO *FileInfo;
if (0 == Request)
{
if ((PVOID)(Info + 1) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
FSP_FILE_NODE *FileNode = FileObject->FsContext; if (!FspFileNodeTryGetFileInfo(FileNode, &FileInfoBuf))
PFILE_STANDARD_INFORMATION Info = (PFILE_STANDARD_INFORMATION)*PBuffer; return FSP_STATUS_IOQ_POST;
FileInfo = &FileInfoBuf;
FspFileNodeAcquireShared(FileNode, Main); FspFileNodeAcquireShared(FileNode, Both);
}
else if (0 == Response)
{
FspFileNodeAcquireShared(FileNode, Both);
FspIopRequestContext(Request, RequestFileNode) = FileNode;
FspIopRequestContext(Request, RequestAcquireFlags) = (PVOID)FspFileNodeAcquireBoth;
return FSP_STATUS_IOQ_POST;
}
else
{
FspIopRequestContext(Request, RequestFileNode) = 0;
FileInfo = &Response->Rsp.QueryInformation.FileInfo;
}
Info->AllocationSize = FileNode->Header.AllocationSize; Info->AllocationSize = FileNode->Header.AllocationSize;
Info->EndOfFile = FileNode->Header.FileSize; Info->EndOfFile = FileNode->Header.FileSize;
FspFileNodeRelease(FileNode, Pgio);
Info->NumberOfLinks = 1; Info->NumberOfLinks = 1;
Info->DeletePending = FileObject->DeletePending; Info->DeletePending = FileObject->DeletePending;
Info->Directory = FileNode->IsDirectory; Info->Directory = FileNode->IsDirectory;
FspFileNodeRelease(FileNode, Main); FspFileNodeRelease(FileNode, Main);
*PBuffer += sizeof(FILE_STANDARD_INFORMATION); *PBuffer = (PVOID)(Info + 1);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
@ -320,20 +419,21 @@ static NTSTATUS FspFsvolQueryInformation(
return STATUS_INVALID_DEVICE_REQUEST; return STATUS_INVALID_DEVICE_REQUEST;
NTSTATUS Result; NTSTATUS Result;
FILE_INFORMATION_CLASS FileInformationClass = IrpSp->Parameters.QueryFile.FileInformationClass;
PFILE_OBJECT FileObject = IrpSp->FileObject; PFILE_OBJECT FileObject = IrpSp->FileObject;
PUINT8 Buffer = Irp->AssociatedIrp.SystemBuffer; PVOID Buffer = Irp->AssociatedIrp.SystemBuffer;
PUINT8 BufferEnd = Buffer + IrpSp->Parameters.QueryFile.Length; PVOID BufferEnd = (PUINT8)Buffer + IrpSp->Parameters.QueryFile.Length;
switch (IrpSp->Parameters.QueryFile.FileInformationClass) switch (FileInformationClass)
{ {
case FileAllInformation: case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, 0, 0);
break; break;
case FileAttributeTagInformation: case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, 0, 0);
break; break;
case FileBasicInformation: case FileBasicInformation:
Result = FspFsvolQueryBasicInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryBasicInformation(FileObject, &Buffer, BufferEnd, 0, 0);
break; break;
case FileCompressionInformation: case FileCompressionInformation:
Result = STATUS_INVALID_PARAMETER; /* no compression support */ Result = STATUS_INVALID_PARAMETER; /* no compression support */
@ -351,13 +451,13 @@ static NTSTATUS FspFsvolQueryInformation(
Result = FspFsvolQueryNameInformation(FileObject, &Buffer, BufferEnd); Result = FspFsvolQueryNameInformation(FileObject, &Buffer, BufferEnd);
break; break;
case FileNetworkOpenInformation: case FileNetworkOpenInformation:
Result = FspFsvolQueryNetworkOpenInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryNetworkOpenInformation(FileObject, &Buffer, BufferEnd, 0, 0);
break; break;
case FilePositionInformation: case FilePositionInformation:
Result = FspFsvolQueryPositionInformation(FileObject, &Buffer, BufferEnd); Result = FspFsvolQueryPositionInformation(FileObject, &Buffer, BufferEnd);
break; break;
case FileStandardInformation: case FileStandardInformation:
Result = FspFsvolQueryStandardInformation(FileObject, &Buffer, BufferEnd); Result = FspFsvolQueryStandardInformation(FileObject, &Buffer, BufferEnd, 0, 0);
break; break;
case FileStreamInformation: case FileStreamInformation:
Result = STATUS_INVALID_PARAMETER; /* !!!: no stream support yet! */ Result = STATUS_INVALID_PARAMETER; /* !!!: no stream support yet! */
@ -369,7 +469,7 @@ static NTSTATUS FspFsvolQueryInformation(
if (FSP_STATUS_IOQ_POST != Result) if (FSP_STATUS_IOQ_POST != Result)
{ {
Irp->IoStatus.Information = (UINT_PTR)(Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer); Irp->IoStatus.Information = (UINT_PTR)((PUINT8)Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer);
return Result; return Result;
} }
@ -390,7 +490,30 @@ static NTSTATUS FspFsvolQueryInformation(
Request->Req.QueryInformation.UserContext = FileNode->UserContext; Request->Req.QueryInformation.UserContext = FileNode->UserContext;
Request->Req.QueryInformation.UserContext2 = FileDesc->UserContext2; Request->Req.QueryInformation.UserContext2 = FileDesc->UserContext2;
return FSP_STATUS_IOQ_POST; switch (FileInformationClass)
{
case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, Request, 0);
break;
case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, Request, 0);
break;
case FileBasicInformation:
Result = FspFsvolQueryBasicInformation(FileObject, &Buffer, BufferEnd, Request, 0);
break;
case FileNetworkOpenInformation:
Result = FspFsvolQueryNetworkOpenInformation(FileObject, &Buffer, BufferEnd, Request, 0);
break;
case FileStandardInformation:
Result = FspFsvolQueryStandardInformation(FileObject, &Buffer, BufferEnd, Request, 0);
break;
default:
ASSERT(0);
Result = STATUS_INVALID_PARAMETER;
break;
}
return Result;
} }
VOID FspFsvolQueryInformationComplete( VOID FspFsvolQueryInformationComplete(
@ -405,30 +528,31 @@ VOID FspFsvolQueryInformationComplete(
FSP_RETURN(); FSP_RETURN();
} }
FILE_INFORMATION_CLASS FileInformationClass = IrpSp->Parameters.QueryFile.FileInformationClass;
PFILE_OBJECT FileObject = IrpSp->FileObject; PFILE_OBJECT FileObject = IrpSp->FileObject;
FSP_FILE_NODE *FileNode = FileObject->FsContext; FSP_FILE_NODE *FileNode = FileObject->FsContext;
PUINT8 Buffer = Irp->AssociatedIrp.SystemBuffer; PVOID Buffer = Irp->AssociatedIrp.SystemBuffer;
PUINT8 BufferEnd = Buffer + IrpSp->Parameters.QueryFile.Length; PVOID BufferEnd = (PUINT8)Buffer + IrpSp->Parameters.QueryFile.Length;
FSP_FSCTL_TRANSACT_REQ *Request = FspIrpRequest(Irp);
FspFileNodeSetFileInfo(FileNode, &Response->Rsp.QueryInformation.FileInfo); FspFileNodeSetFileInfo(FileNode, &Response->Rsp.QueryInformation.FileInfo);
switch (IrpSp->Parameters.QueryFile.FileInformationClass) switch (FileInformationClass)
{ {
case FileAllInformation: case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, Request, Response);
&Response->Rsp.QueryInformation.FileInfo);
break; break;
case FileAttributeTagInformation: case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, Request, Response);
&Response->Rsp.QueryInformation.FileInfo);
break; break;
case FileBasicInformation: case FileBasicInformation:
Result = FspFsvolQueryBasicInformation(FileObject, &Buffer, BufferEnd, Result = FspFsvolQueryBasicInformation(FileObject, &Buffer, BufferEnd, Request, Response);
&Response->Rsp.QueryInformation.FileInfo);
break; break;
case FileNetworkOpenInformation: case FileNetworkOpenInformation:
Result = FspFsvolQueryNetworkOpenInformation(FileObject, &Buffer, BufferEnd, Result = FspFsvolQueryNetworkOpenInformation(FileObject, &Buffer, BufferEnd, Request, Response);
&Response->Rsp.QueryInformation.FileInfo); break;
case FileStandardInformation:
Result = FspFsvolQueryStandardInformation(FileObject, &Buffer, BufferEnd, Request, Response);
break; break;
default: default:
ASSERT(0); ASSERT(0);
@ -438,19 +562,13 @@ VOID FspFsvolQueryInformationComplete(
ASSERT(FSP_STATUS_IOQ_POST != Result); ASSERT(FSP_STATUS_IOQ_POST != Result);
Irp->IoStatus.Information = (UINT_PTR)(Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer); Irp->IoStatus.Information = (UINT_PTR)((PUINT8)Buffer - (PUINT8)Irp->AssociatedIrp.SystemBuffer);
FSP_LEAVE_IOC("%s, FileObject=%p", FSP_LEAVE_IOC("%s, FileObject=%p",
FileInformationClassSym(IrpSp->Parameters.QueryFile.FileInformationClass), FileInformationClassSym(IrpSp->Parameters.QueryFile.FileInformationClass),
IrpSp->FileObject); IrpSp->FileObject);
} }
enum
{
RequestFileNode = 0,
RequestAcquireFlags = 1,
};
static NTSTATUS FspFsvolSetAllocationInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolSetAllocationInformation(PFILE_OBJECT FileObject,
PVOID Buffer, ULONG Length, PVOID Buffer, ULONG Length,
FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response) FSP_FSCTL_TRANSACT_REQ *Request, const FSP_FSCTL_TRANSACT_RSP *Response)