Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-24 01:13:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: IRP_MJ_QUERY_INFORMATION: FileAllInformation, FileNameInformation: protect with FileRenameResource

Browse Source
This commit is contained in:
Bill Zissimopoulos 2016-02-07 12:54:19 -08:00
parent 9c200f2a62
commit 2a303827a2
1 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/sys/fileinfo.c
View File

@ -82,6 +82,8 @@ enum
{ {
RequestFileNode = 0, RequestFileNode = 0,
RequestInfoChangeNumber = 1, RequestInfoChangeNumber = 1,
RequestAllInformationResult = 2,
RequestAllInformationBuffer = 3,
}; };
static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject,
@ -98,7 +100,8 @@ static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject,
if ((PVOID)(Info + 1) > BufferEnd) if ((PVOID)(Info + 1) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
return STATUS_SUCCESS; *PBuffer = (PVOID)&Info->NameInformation;
return FspFsvolQueryNameInformation(FileObject, PBuffer, BufferEnd);
} }
Info->BasicInformation.CreationTime.QuadPart = FileInfo->CreationTime; Info->BasicInformation.CreationTime.QuadPart = FileInfo->CreationTime;
@ -120,8 +123,7 @@ static NTSTATUS FspFsvolQueryAllInformation(PFILE_OBJECT FileObject,
Info->PositionInformation.CurrentByteOffset = FileObject->CurrentByteOffset; Info->PositionInformation.CurrentByteOffset = FileObject->CurrentByteOffset;
*PBuffer = (PVOID)&Info->NameInformation; return STATUS_SUCCESS;
return FspFsvolQueryNameInformation(FileObject, PBuffer, BufferEnd);
} }
static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject, static NTSTATUS FspFsvolQueryAttributeTagInformation(PFILE_OBJECT FileObject,
@ -205,12 +207,14 @@ static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject,
PUINT8 Buffer = (PUINT8)Info->FileName; PUINT8 Buffer = (PUINT8)Info->FileName;
ULONG CopyLength; ULONG CopyLength;
FSP_FILE_NODE *FileNode = FileObject->FsContext; FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = PDEVICE_OBJECT FsvolDeviceObject = FileNode->FsvolDeviceObject;
FspFsvolDeviceExtension(FileNode->FsvolDeviceObject); FSP_FSVOL_DEVICE_EXTENSION *FsvolDeviceExtension = FspFsvolDeviceExtension(FsvolDeviceObject);
if ((PVOID)(Info + 1) > BufferEnd) if ((PVOID)(Info + 1) > BufferEnd)
return STATUS_BUFFER_TOO_SMALL; return STATUS_BUFFER_TOO_SMALL;
FspFsvolDeviceFileRenameAcquireShared(FsvolDeviceObject);
Info->FileNameLength = FsvolDeviceExtension->VolumePrefix.Length + FileNode->FileName.Length; Info->FileNameLength = FsvolDeviceExtension->VolumePrefix.Length + FileNode->FileName.Length;
CopyLength = FsvolDeviceExtension->VolumePrefix.Length; CopyLength = FsvolDeviceExtension->VolumePrefix.Length;
@ -231,6 +235,8 @@ static NTSTATUS FspFsvolQueryNameInformation(PFILE_OBJECT FileObject,
RtlCopyMemory(Buffer, FileNode->FileName.Buffer, CopyLength); RtlCopyMemory(Buffer, FileNode->FileName.Buffer, CopyLength);
Buffer += CopyLength; Buffer += CopyLength;
FspFsvolDeviceFileRenameRelease(FsvolDeviceObject);
*PBuffer = Buffer; *PBuffer = Buffer;
return Result; return Result;
@ -333,6 +339,8 @@ static NTSTATUS FspFsvolQueryInformation(
PVOID Buffer = Irp->AssociatedIrp.SystemBuffer; PVOID Buffer = Irp->AssociatedIrp.SystemBuffer;
PVOID BufferEnd = (PUINT8)Buffer + IrpSp->Parameters.QueryFile.Length; PVOID BufferEnd = (PUINT8)Buffer + IrpSp->Parameters.QueryFile.Length;
FSP_FSCTL_FILE_INFO FileInfoBuf; FSP_FSCTL_FILE_INFO FileInfoBuf;
NTSTATUS AllInformationResult = STATUS_INVALID_PARAMETER;
PVOID AllInformationBuffer = 0;
ASSERT(FileNode == FileDesc->FileNode); ASSERT(FileNode == FileDesc->FileNode);
@ -340,6 +348,11 @@ static NTSTATUS FspFsvolQueryInformation(
{ {
case FileAllInformation: case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, 0);
AllInformationResult = Result;
AllInformationBuffer = Buffer;
if (STATUS_BUFFER_OVERFLOW == Result)
Result = STATUS_SUCCESS;
Buffer = Irp->AssociatedIrp.SystemBuffer;
break; break;
case FileAttributeTagInformation: case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, 0); Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, 0);
@ -393,6 +406,9 @@ static NTSTATUS FspFsvolQueryInformation(
{ {
case FileAllInformation: case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, &FileInfoBuf); Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, &FileInfoBuf);
ASSERT(NT_SUCCESS(Result));
Result = AllInformationResult;
Buffer = AllInformationBuffer;
break; break;
case FileAttributeTagInformation: case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, &FileInfoBuf); Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, &FileInfoBuf);
@ -436,6 +452,8 @@ static NTSTATUS FspFsvolQueryInformation(
FspFileNodeSetOwner(FileNode, Full, Request); FspFileNodeSetOwner(FileNode, Full, Request);
FspIopRequestContext(Request, RequestFileNode) = FileNode; FspIopRequestContext(Request, RequestFileNode) = FileNode;
FspIopRequestContext(Request, RequestAllInformationResult) = (PVOID)(ULONG)AllInformationResult;
FspIopRequestContext(Request, RequestAllInformationBuffer) = AllInformationBuffer;
return FSP_STATUS_IOQ_POST; return FSP_STATUS_IOQ_POST;
} }
@ -487,6 +505,9 @@ NTSTATUS FspFsvolQueryInformationComplete(
{ {
case FileAllInformation: case FileAllInformation:
Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, FileInfo); Result = FspFsvolQueryAllInformation(FileObject, &Buffer, BufferEnd, FileInfo);
ASSERT(NT_SUCCESS(Result));
Result = (NTSTATUS)(UINT_PTR)FspIopRequestContext(Request, RequestAllInformationResult);
Buffer = FspIopRequestContext(Request, RequestAllInformationBuffer);
break; break;
case FileAttributeTagInformation: case FileAttributeTagInformation:
Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, FileInfo); Result = FspFsvolQueryAttributeTagInformation(FileObject, &Buffer, BufferEnd, FileInfo);