Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-01 04:11:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
reactos/drivers/filesystems/udfs/fileinfo.cpp
Timo Kreuzer c334c17d69 [UDFS] Fix 64 bit issues
2019-04-02 10:53:23 +02:00

2704 lines
106 KiB
C++
Raw Blame History

////////////////////////////////////////////////////////////////////
// Copyright (C) Alexander Telyatnikov, Ivan Keliukh, Yegor Anchishkin, SKIF Software, 1999-2013. Kiev, Ukraine
// All rights reserved
// This file was released under the GPLv2 on June 2015.
////////////////////////////////////////////////////////////////////
/*************************************************************************
*
* File: Fileinfo.cpp
*
* Module: UDF File System Driver (Kernel mode execution only)
*
* Description:
* Contains code to handle the "set/query file information" dispatch
* entry points.
*
*************************************************************************/
#include "udffs.h"
// define the file specific bug-check id
#define UDF_BUG_CHECK_ID UDF_FILE_INFORMATION
#define MEM_USREN_TAG "US_Ren"
#define MEM_USREN2_TAG "US_Ren2"
#define MEM_USFIDC_TAG "US_FIDC"
#define MEM_USHL_TAG "US_HL"
/*************************************************************************
*
* Function: UDFFileInfo()
*
* Description:
* The I/O Manager will invoke this routine to handle a set/query file
* information request
*
* Expected Interrupt Level (for execution) :
*
* IRQL_PASSIVE_LEVEL (invocation at higher IRQL will cause execution
* to be deferred to a worker thread context)
*
* Return Value: STATUS_SUCCESS/Error
*
*************************************************************************/
NTSTATUS
NTAPI
UDFFileInfo(
PDEVICE_OBJECT DeviceObject, // the logical volume device object
PIRP Irp // I/O Request Packet
)
{
NTSTATUS RC = STATUS_SUCCESS;
PtrUDFIrpContext PtrIrpContext = NULL;
BOOLEAN AreWeTopLevel = FALSE;
TmPrint(("UDFFileInfo: \n"));
FsRtlEnterFileSystem();
ASSERT(DeviceObject);
ASSERT(Irp);
// set the top level context
AreWeTopLevel = UDFIsIrpTopLevel(Irp);
ASSERT(!UDFIsFSDevObj(DeviceObject));
_SEH2_TRY {
// get an IRP context structure and issue the request
PtrIrpContext = UDFAllocateIrpContext(Irp, DeviceObject);
if(PtrIrpContext) {
RC = UDFCommonFileInfo(PtrIrpContext, Irp);
} else {
RC = STATUS_INSUFFICIENT_RESOURCES;
Irp->IoStatus.Status = RC;
Irp->IoStatus.Information = 0;
// complete the IRP
IoCompleteRequest(Irp, IO_DISK_INCREMENT);
}
} _SEH2_EXCEPT(UDFExceptionFilter(PtrIrpContext, _SEH2_GetExceptionInformation())) {
RC = UDFExceptionHandler(PtrIrpContext, Irp);
UDFLogEvent(UDF_ERROR_INTERNAL_ERROR, RC);
} _SEH2_END;
if(AreWeTopLevel) {
IoSetTopLevelIrp(NULL);
}
FsRtlExitFileSystem();
return(RC);
} // end UDFFileInfo()
/*************************************************************************
*
* Function: UDFCommonFileInfo()
*
* Description:
* The actual work is performed here. This routine may be invoked in one'
* of the two possible contexts:
* (a) in the context of a system worker thread
* (b) in the context of the original caller
*
* Expected Interrupt Level (for execution) :
*
* IRQL_PASSIVE_LEVEL
*
* Return Value: STATUS_SUCCESS/Error
*
*************************************************************************/
NTSTATUS
UDFCommonFileInfo(
PtrUDFIrpContext PtrIrpContext,
PIRP Irp
)
{
NTSTATUS RC = STATUS_SUCCESS;
PIO_STACK_LOCATION IrpSp = NULL;
PFILE_OBJECT FileObject = NULL;
PtrUDFFCB Fcb = NULL;
PtrUDFCCB Ccb = NULL;
PVCB Vcb = NULL;
PtrUDFNTRequiredFCB NtReqFcb = NULL;
BOOLEAN MainResourceAcquired = FALSE;
BOOLEAN ParentResourceAcquired = FALSE;
BOOLEAN PagingIoResourceAcquired = FALSE;
PVOID PtrSystemBuffer = NULL;
LONG BufferLength = 0;
FILE_INFORMATION_CLASS FunctionalityRequested;
BOOLEAN CanWait = FALSE;
BOOLEAN PostRequest = FALSE;
BOOLEAN AcquiredVcb = FALSE;
PIRP TopIrp;
TmPrint(("UDFCommonFileInfo: irp %x\n", Irp));
TopIrp = IoGetTopLevelIrp();
switch((ULONG_PTR)TopIrp) {
case FSRTL_FSP_TOP_LEVEL_IRP:
UDFPrint((" FSRTL_FSP_TOP_LEVEL_IRP\n"));
break;
case FSRTL_CACHE_TOP_LEVEL_IRP:
UDFPrint((" FSRTL_CACHE_TOP_LEVEL_IRP\n"));
break;
case FSRTL_MOD_WRITE_TOP_LEVEL_IRP:
UDFPrint((" FSRTL_MOD_WRITE_TOP_LEVEL_IRP\n"));
break;
case FSRTL_FAST_IO_TOP_LEVEL_IRP:
UDFPrint((" FSRTL_FAST_IO_TOP_LEVEL_IRP\n"));
BrutePoint()
break;
case NULL:
UDFPrint((" NULL TOP_LEVEL_IRP\n"));
break;
default:
if(TopIrp == Irp) {
UDFPrint((" TOP_LEVEL_IRP\n"));
} else {
UDFPrint((" RECURSIVE_IRP, TOP = %x\n", TopIrp));
}
}
_SEH2_TRY {
// First, get a pointer to the current I/O stack location.
IrpSp = IoGetCurrentIrpStackLocation(Irp);
ASSERT(IrpSp);
FileObject = IrpSp->FileObject;
ASSERT(FileObject);
// Get the FCB and CCB pointers.
Ccb = (PtrUDFCCB)(FileObject->FsContext2);
ASSERT(Ccb);
if(!Ccb) {
// some applications sends us FO without Ccb
// This is not allowed...
RC = STATUS_INVALID_PARAMETER;
try_return(RC);
}
Fcb = Ccb->Fcb;
ASSERT(Fcb);
NtReqFcb = Fcb->NTRequiredFCB;
CanWait = (PtrIrpContext->IrpContextFlags & UDF_IRP_CONTEXT_CAN_BLOCK) ? TRUE : FALSE;
// If the caller has opened a logical volume and is attempting to
// query information for it as a file stream, return an error.
if(Fcb->NodeIdentifier.NodeType == UDF_NODE_TYPE_VCB) {
// This is not allowed. Caller must use get/set volume information instead.
RC = STATUS_INVALID_PARAMETER;
try_return(RC);
}
Vcb = (PVCB)(IrpSp->DeviceObject->DeviceExtension);
ASSERT(Vcb);
ASSERT(Fcb->NodeIdentifier.NodeType == UDF_NODE_TYPE_FCB);
//Vcb->VCBFlags |= UDF_VCB_SKIP_EJECT_CHECK;
// The NT I/O Manager always allocates and supplies a system
// buffer for query and set file information calls.
// Copying information to/from the user buffer and the system
// buffer is performed by the I/O Manager and the FSD need not worry about it.
PtrSystemBuffer = Irp->AssociatedIrp.SystemBuffer;
UDFFlushTryBreak(Vcb);
if(!UDFAcquireResourceShared(&(Vcb->VCBResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
AcquiredVcb = TRUE;
if(IrpSp->MajorFunction == IRP_MJ_QUERY_INFORMATION) {
// Now, obtain some parameters.
BufferLength = IrpSp->Parameters.QueryFile.Length;
FunctionalityRequested = IrpSp->Parameters.QueryFile.FileInformationClass;
#ifdef UDF_ENABLE_SECURITY
RC = IoCheckFunctionAccess(
Ccb->PreviouslyGrantedAccess,
PtrIrpContext->MajorFunction,
PtrIrpContext->MinorFunction,
0,
&FunctionalityRequested,
NULL);
if(!NT_SUCCESS(RC)) {
try_return(RC);
}
#endif //UDF_ENABLE_SECURITY
// Acquire the MainResource shared (NOTE: for paging-IO on a
// page file, we should avoid acquiring any resources and simply
// trust the VMM to do the right thing, else we could possibly
// run into deadlocks).
if(!(Fcb->FCBFlags & UDF_FCB_PAGE_FILE)) {
// Acquire the MainResource shared.
UDF_CHECK_PAGING_IO_RESOURCE(NtReqFcb);
UDFAcquireResourceShared(&(NtReqFcb->MainResource), TRUE);
MainResourceAcquired = TRUE;
}
// Do whatever the caller asked us to do
switch (FunctionalityRequested) {
case FileBasicInformation:
RC = UDFGetBasicInformation(FileObject, Fcb, (PFILE_BASIC_INFORMATION)PtrSystemBuffer, &BufferLength);
break;
case FileStandardInformation:
RC = UDFGetStandardInformation(Fcb, (PFILE_STANDARD_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
#if(_WIN32_WINNT >= 0x0400)
case FileNetworkOpenInformation:
RC = UDFGetNetworkInformation(Fcb, (PFILE_NETWORK_OPEN_INFORMATION)PtrSystemBuffer, &BufferLength);
break;
#endif // _WIN32_WINNT >= 0x0400
case FileInternalInformation:
RC = UDFGetInternalInformation(PtrIrpContext, Fcb, Ccb, (PFILE_INTERNAL_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
case FileEaInformation:
RC = UDFGetEaInformation(PtrIrpContext, Fcb, (PFILE_EA_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
case FileNameInformation:
RC = UDFGetFullNameInformation(FileObject, (PFILE_NAME_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
case FileAlternateNameInformation:
RC = UDFGetAltNameInformation(Fcb, (PFILE_NAME_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
// case FileCompressionInformation:
// // RC = UDFGetCompressionInformation(...);
// break;
case FilePositionInformation:
RC = UDFGetPositionInformation(FileObject, (PFILE_POSITION_INFORMATION)PtrSystemBuffer, &BufferLength);
break;
case FileStreamInformation:
RC = UDFGetFileStreamInformation(Fcb, (PFILE_STREAM_INFORMATION) PtrSystemBuffer, &BufferLength);
break;
case FileAllInformation:
// The I/O Manager supplies the Mode, Access, and Alignment
// information. The rest is up to us to provide.
// Therefore, decrement the BufferLength appropriately (assuming
// that the above 3 types on information are already in the
// buffer)
{
PFILE_ALL_INFORMATION PtrAllInfo = (PFILE_ALL_INFORMATION)PtrSystemBuffer;
BufferLength -= (sizeof(FILE_MODE_INFORMATION) +
sizeof(FILE_ACCESS_INFORMATION) +
sizeof(FILE_ALIGNMENT_INFORMATION));
// Get the remaining stuff.
if(!NT_SUCCESS(RC = UDFGetBasicInformation(FileObject, Fcb, &(PtrAllInfo->BasicInformation), &BufferLength)) ||
!NT_SUCCESS(RC = UDFGetStandardInformation(Fcb, &(PtrAllInfo->StandardInformation), &BufferLength)) ||
!NT_SUCCESS(RC = UDFGetInternalInformation(PtrIrpContext, Fcb, Ccb, &(PtrAllInfo->InternalInformation), &BufferLength)) ||
!NT_SUCCESS(RC = UDFGetEaInformation(PtrIrpContext, Fcb, &(PtrAllInfo->EaInformation), &BufferLength)) ||
!NT_SUCCESS(RC = UDFGetPositionInformation(FileObject, &(PtrAllInfo->PositionInformation), &BufferLength)) ||
!NT_SUCCESS(RC = UDFGetFullNameInformation(FileObject, &(PtrAllInfo->NameInformation), &BufferLength))
)
try_return(RC);
}
break;
default:
RC = STATUS_INVALID_PARAMETER;
try_return(RC);
}
#ifndef UDF_READ_ONLY_BUILD
} else {
// if(IrpSp->MajorFunction == IRP_MJ_SET_INFORMATION) {
Vcb->VCBFlags |= UDF_VCB_SKIP_EJECT_CHECK;
ASSERT(IrpSp->MajorFunction == IRP_MJ_SET_INFORMATION);
// Now, obtain some parameters.
FunctionalityRequested = IrpSp->Parameters.SetFile.FileInformationClass;
if((Vcb->VCBFlags & UDF_VCB_FLAGS_VOLUME_READ_ONLY) &&
(FunctionalityRequested != FilePositionInformation)) {
try_return(RC = STATUS_ACCESS_DENIED);
}
#ifdef UDF_ENABLE_SECURITY
RC = IoCheckFunctionAccess(
Ccb->PreviouslyGrantedAccess,
PtrIrpContext->MajorFunction,
PtrIrpContext->MinorFunction,
0,
&FunctionalityRequested,
NULL);
if(!NT_SUCCESS(RC)) {
try_return(RC);
}
#endif //UDF_ENABLE_SECURITY
// If the FSD supports opportunistic locking,
// then we should check whether the oplock state
// allows the caller to proceed.
// Rename, and link operations require creation of a directory
// entry and possibly deletion of another directory entry.
// Unless this is an operation on a page file, we should go ahead and
// acquire the FCB exclusively at this time. Note that we will pretty
// much block out anything being done to the FCB from this point on.
if(!(Fcb->FCBFlags & UDF_FCB_PAGE_FILE) &&
(FunctionalityRequested != FilePositionInformation) &&
(FunctionalityRequested != FileRenameInformation) &&
(FunctionalityRequested != FileLinkInformation)) {
// Acquire the Parent & Main Resources exclusive.
if(Fcb->FileInfo->ParentFile) {
UDF_CHECK_PAGING_IO_RESOURCE(Fcb->ParentFcb->NTRequiredFCB);
if(!UDFAcquireResourceExclusive(&(Fcb->ParentFcb->NTRequiredFCB->MainResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
ParentResourceAcquired = TRUE;
}
UDF_CHECK_PAGING_IO_RESOURCE(NtReqFcb);
if(!UDFAcquireResourceExclusive(&(NtReqFcb->MainResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
MainResourceAcquired = TRUE;
} else
// The only operations that could conceivably proceed from this point
// on are paging-IO read/write operations. For delete, link (rename),
// set allocation size, and set EOF, should also acquire the paging-IO
// resource, thereby synchronizing with paging-IO requests.
if((Fcb->FCBFlags & UDF_FCB_PAGE_FILE) &&
((FunctionalityRequested == FileDispositionInformation) ||
(FunctionalityRequested == FileAllocationInformation) ||
(FunctionalityRequested == FileEndOfFileInformation)) ) {
// Acquire the MainResource shared.
UDF_CHECK_PAGING_IO_RESOURCE(NtReqFcb);
if(!UDFAcquireResourceShared(&(NtReqFcb->MainResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
MainResourceAcquired = TRUE;
// Acquire the PagingResource exclusive.
if(!UDFAcquireResourceExclusive(&(NtReqFcb->PagingIoResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
PagingIoResourceAcquired = TRUE;
} else if((FunctionalityRequested != FileRenameInformation) &&
(FunctionalityRequested != FileLinkInformation)) {
// Acquire the MainResource shared.
UDF_CHECK_PAGING_IO_RESOURCE(NtReqFcb);
if(!UDFAcquireResourceShared(&(NtReqFcb->MainResource), CanWait)) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
MainResourceAcquired = TRUE;
}
if((Vcb->VCBFlags & UDF_VCB_FLAGS_RAW_DISK) &&
(FunctionalityRequested != FilePositionInformation)) {
AdPrint((" Can't change File Information on blank volume ;)\n"));
try_return(RC = STATUS_ACCESS_DENIED);
}
// Do whatever the caller asked us to do
switch (FunctionalityRequested) {
case FileBasicInformation:
RC = UDFSetBasicInformation(Fcb, Ccb, FileObject, (PFILE_BASIC_INFORMATION)PtrSystemBuffer);
break;
case FilePositionInformation: {
// Check if no intermediate buffering has been specified.
// If it was specified, do not allow non-aligned set file
// position requests to succeed.
PFILE_POSITION_INFORMATION PtrFileInfoBuffer;
PtrFileInfoBuffer = (PFILE_POSITION_INFORMATION)PtrSystemBuffer;
if(FileObject->Flags & FO_NO_INTERMEDIATE_BUFFERING) {
if(PtrFileInfoBuffer->CurrentByteOffset.LowPart & IrpSp->DeviceObject->AlignmentRequirement) {
// Invalid alignment.
try_return(RC = STATUS_INVALID_PARAMETER);
}
}
FileObject->CurrentByteOffset = PtrFileInfoBuffer->CurrentByteOffset;
break;
}
case FileDispositionInformation:
RC = UDFSetDispositionInformation(Fcb, Ccb, Vcb, FileObject,
((PFILE_DISPOSITION_INFORMATION)PtrSystemBuffer)->DeleteFile ? TRUE : FALSE);
break;
case FileRenameInformation:
if(!CanWait) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
RC = UDFRename(IrpSp, Fcb, Ccb, FileObject, (PFILE_RENAME_INFORMATION)PtrSystemBuffer);
if(RC == STATUS_PENDING) {
PostRequest = TRUE;
try_return(RC);
}
break;
#ifdef UDF_ALLOW_HARD_LINKS
case FileLinkInformation:
if(!CanWait) {
PostRequest = TRUE;
try_return(RC = STATUS_PENDING);
}
RC = UDFHardLink(IrpSp, Fcb, Ccb, FileObject, (PFILE_LINK_INFORMATION)PtrSystemBuffer);
break;
#endif //UDF_ALLOW_HARD_LINKS
case FileAllocationInformation:
RC = UDFSetAllocationInformation(Fcb, Ccb, Vcb, FileObject,
PtrIrpContext, Irp,
(PFILE_ALLOCATION_INFORMATION)PtrSystemBuffer);
break;
case FileEndOfFileInformation:
RC = UDFSetEOF(IrpSp, Fcb, Ccb, Vcb, FileObject, Irp, (PFILE_END_OF_FILE_INFORMATION)PtrSystemBuffer);
break;
default:
RC = STATUS_INVALID_PARAMETER;
try_return(RC);
}
#endif //UDF_READ_ONLY_BUILD
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(PagingIoResourceAcquired) {
UDFReleaseResource(&(NtReqFcb->PagingIoResource));
PagingIoResourceAcquired = FALSE;
}
if(MainResourceAcquired) {
UDF_CHECK_PAGING_IO_RESOURCE(NtReqFcb);
UDFReleaseResource(&(NtReqFcb->MainResource));
MainResourceAcquired = FALSE;
}
if(ParentResourceAcquired) {
UDF_CHECK_PAGING_IO_RESOURCE(Fcb->ParentFcb->NTRequiredFCB);
UDFReleaseResource(&(Fcb->ParentFcb->NTRequiredFCB->MainResource));
ParentResourceAcquired = FALSE;
}
// Post IRP if required
if(PostRequest) {
// Since, the I/O Manager gave us a system buffer, we do not
// need to "lock" anything.
// Perform the post operation which will mark the IRP pending
// and will return STATUS_PENDING back to us
RC = UDFPostRequest(PtrIrpContext, Irp);
} else {
if (!_SEH2_AbnormalTermination()) {
Irp->IoStatus.Status = RC;
// Set status for "query" requests
if(IrpSp->MajorFunction == IRP_MJ_QUERY_INFORMATION) {
// Return the amount of information transferred.
Irp->IoStatus.Information = IrpSp->Parameters.QueryFile.Length - BufferLength;
#ifndef UDF_READ_ONLY_BUILD
#ifdef UDF_DELAYED_CLOSE
} else
if(NT_SUCCESS(RC)) {
if(FunctionalityRequested == FileDispositionInformation) {
if(AcquiredVcb) {
AcquiredVcb = FALSE;
UDFReleaseResource(&(Vcb->VCBResource));
}
UDFRemoveFromDelayedQueue(Fcb);
}
#endif //UDF_DELAYED_CLOSE
#endif //UDF_READ_ONLY_BUILD
}
// complete the IRP
IoCompleteRequest(Irp, IO_DISK_INCREMENT);
// Free up the Irp Context
UDFReleaseIrpContext(PtrIrpContext);
} // can we complete the IRP ?
}
if(AcquiredVcb) {
UDFReleaseResource(&(Vcb->VCBResource));
}
} _SEH2_END;// end of "__finally" processing
return(RC);
} // end UDFCommonFileInfo()
/*
Return some time-stamps and file attributes to the caller.
*/
NTSTATUS
UDFGetBasicInformation(
IN PFILE_OBJECT FileObject,
IN PtrUDFFCB Fcb,
IN PFILE_BASIC_INFORMATION PtrBuffer,
IN OUT LONG* PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO FileInfo;
PDIR_INDEX_ITEM DirNdx;
AdPrint(("UDFGetBasicInformation: \n"));
_SEH2_TRY {
if(*PtrReturnedLength < (LONG)sizeof(FILE_BASIC_INFORMATION)) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
// Zero out the supplied buffer.
RtlZeroMemory(PtrBuffer, sizeof(FILE_BASIC_INFORMATION));
// Get information from the FCB and update TimesCache in DirIndex
FileInfo = Fcb->FileInfo;
if(!FileInfo) {
AdPrint(("!!!!!!!! Bu-u-u-u-u-g !!!!!!!!!!!\n"));
AdPrint(("!!!! GetBasicInfo to unopened file !!!!\n"));
try_return(RC = STATUS_INVALID_PARAMETER);
}
DirNdx = UDFDirIndex(UDFGetDirIndexByFileInfo(FileInfo), FileInfo->Index);
PtrBuffer->CreationTime = Fcb->NTRequiredFCB->CreationTime;
DirNdx->CreationTime = PtrBuffer->CreationTime.QuadPart;
PtrBuffer->LastAccessTime = Fcb->NTRequiredFCB->LastAccessTime;
DirNdx->LastAccessTime = PtrBuffer->LastAccessTime.QuadPart;
PtrBuffer->LastWriteTime = Fcb->NTRequiredFCB->LastWriteTime;
DirNdx->LastWriteTime = PtrBuffer->LastWriteTime.QuadPart;
PtrBuffer->ChangeTime = Fcb->NTRequiredFCB->ChangeTime;
DirNdx->ChangeTime = PtrBuffer->ChangeTime.QuadPart;
// Now fill in the attributes.
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY) {
PtrBuffer->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
#ifdef UDF_DBG
if(!FileInfo->Dloc->DirIndex) AdPrint(("*****!!!!! Directory has no DirIndex !!!!!*****\n"));
#endif
}
// Similarly, fill in attributes indicating a hidden file, system
// file, compressed file, temporary file, etc. if the FSD supports
// such file attribute values.
PtrBuffer->FileAttributes |= UDFAttributesToNT(DirNdx,NULL);
if(FileObject->Flags & FO_TEMPORARY_FILE) {
PtrBuffer->FileAttributes |= FILE_ATTRIBUTE_TEMPORARY;
} else {
PtrBuffer->FileAttributes &= ~FILE_ATTRIBUTE_TEMPORARY;
}
if(!PtrBuffer->FileAttributes) {
PtrBuffer->FileAttributes = FILE_ATTRIBUTE_NORMAL;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(NT_SUCCESS(RC)) {
// Return the amount of information filled in.
(*PtrReturnedLength) -= sizeof(FILE_BASIC_INFORMATION);
}
} _SEH2_END;
return(RC);
} // end UDFGetBasicInformation()
/*
Return file sizes to the caller.
*/
NTSTATUS
UDFGetStandardInformation(
IN PtrUDFFCB Fcb,
IN PFILE_STANDARD_INFORMATION PtrBuffer,
IN OUT LONG* PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO FileInfo;
// PVCB Vcb;
AdPrint(("UDFGetStandardInformation: \n"));
_SEH2_TRY {
if(*PtrReturnedLength < (LONG)sizeof(FILE_STANDARD_INFORMATION)) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
// Zero out the supplied buffer.
RtlZeroMemory(PtrBuffer, sizeof(FILE_STANDARD_INFORMATION));
FileInfo = Fcb->FileInfo;
if(!FileInfo) {
AdPrint(("!!!!!!!! Bu-u-u-u-u-g !!!!!!!!!!!\n"));
AdPrint(("!!!! GetStandardInfo to unopened file !!!!\n"));
try_return(RC = STATUS_INVALID_PARAMETER);
}
// Vcb = Fcb->Vcb;
PtrBuffer->NumberOfLinks = UDFGetFileLinkCount(FileInfo);
PtrBuffer->DeletePending = (Fcb->FCBFlags & UDF_FCB_DELETE_ON_CLOSE) ? TRUE : FALSE;
// Case on whether this is a file or a directory, and extract
// the information and fill in the fcb/dcb specific parts
// of the output buffer
if(UDFIsADirectory(Fcb->FileInfo)) {
PtrBuffer->Directory = TRUE;
} else {
if(Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.LowPart == 0xffffffff) {
Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.QuadPart =
UDFSysGetAllocSize(Fcb->Vcb, UDFGetFileSize(FileInfo));
}
PtrBuffer->AllocationSize = Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize;
PtrBuffer->EndOfFile = Fcb->NTRequiredFCB->CommonFCBHeader.FileSize;
PtrBuffer->Directory = FALSE;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(NT_SUCCESS(RC)) {
// Return the amount of information filled in.
*PtrReturnedLength -= sizeof(FILE_STANDARD_INFORMATION);
}
} _SEH2_END;
return(RC);
} // end UDFGetStandardInformation()
/*
Return some time-stamps and file attributes to the caller.
*/
NTSTATUS
UDFGetNetworkInformation(
IN PtrUDFFCB Fcb,
IN PFILE_NETWORK_OPEN_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO FileInfo;
AdPrint(("UDFGetNetworkInformation: \n"));
_SEH2_TRY {
if(*PtrReturnedLength < (LONG)sizeof(FILE_NETWORK_OPEN_INFORMATION)) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
// Zero out the supplied buffer.
RtlZeroMemory(PtrBuffer, sizeof(FILE_NETWORK_OPEN_INFORMATION));
// Get information from the FCB.
PtrBuffer->CreationTime = Fcb->NTRequiredFCB->CreationTime;
PtrBuffer->LastAccessTime = Fcb->NTRequiredFCB->LastAccessTime;
PtrBuffer->LastWriteTime = Fcb->NTRequiredFCB->LastWriteTime;
PtrBuffer->ChangeTime = Fcb->NTRequiredFCB->ChangeTime;
FileInfo = Fcb->FileInfo;
if(!FileInfo) {
AdPrint(("!!!!!!!! Bu-u-u-u-u-g !!!!!!!!!!!\n"));
AdPrint(("!!!! UDFGetNetworkInformation to unopened file !!!!\n"));
try_return(RC = STATUS_INVALID_PARAMETER);
}
// Now fill in the attributes.
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY) {
PtrBuffer->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
#ifdef UDF_DBG
if(!FileInfo->Dloc->DirIndex) AdPrint(("*****!!!!! Directory has no DirIndex !!!!!*****\n"));
#endif
} else {
if(Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.LowPart == 0xffffffff) {
Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.QuadPart =
UDFSysGetAllocSize(Fcb->Vcb, UDFGetFileSize(FileInfo));
}
PtrBuffer->AllocationSize = Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize;
PtrBuffer->EndOfFile = Fcb->NTRequiredFCB->CommonFCBHeader.FileSize;
}
// Similarly, fill in attributes indicating a hidden file, system
// file, compressed file, temporary file, etc. if the FSD supports
// such file attribute values.
PtrBuffer->FileAttributes |= UDFAttributesToNT(UDFDirIndex(UDFGetDirIndexByFileInfo(FileInfo), FileInfo->Index),NULL);
if(!PtrBuffer->FileAttributes) {
PtrBuffer->FileAttributes = FILE_ATTRIBUTE_NORMAL;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(NT_SUCCESS(RC)) {
// Return the amount of information filled in.
(*PtrReturnedLength) -= sizeof(FILE_NETWORK_OPEN_INFORMATION);
}
} _SEH2_END;
return(RC);
} // end UDFGetNetworkInformation()
/*
Return some time-stamps and file attributes to the caller.
*/
NTSTATUS
UDFGetInternalInformation(
PtrUDFIrpContext PtrIrpContext,
IN PtrUDFFCB Fcb,
IN PtrUDFCCB Ccb,
IN PFILE_INTERNAL_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO FileInfo;
PVCB Vcb;
AdPrint(("UDFGetInternalInformation\n"));
_SEH2_TRY {
if(*PtrReturnedLength < (LONG)sizeof(FILE_INTERNAL_INFORMATION)) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
// Zero out the supplied buffer.
RtlZeroMemory(PtrBuffer, sizeof(FILE_INTERNAL_INFORMATION));
FileInfo = Fcb->FileInfo;
if(!FileInfo) {
AdPrint(("!!!!!!!! Bu-u-u-u-u-g !!!!!!!!!!!\n"));
AdPrint(("!!!! UDFGetInternalInformation to unopened file !!!!\n"));
try_return(RC = STATUS_INVALID_PARAMETER);
}
Vcb = Fcb->Vcb;
PtrBuffer->IndexNumber.QuadPart = UDFGetNTFileId(Vcb, FileInfo, &(Fcb->FCBName->ObjectName));
UDFAcquireResourceExclusive(&(Fcb->Vcb->FileIdResource), TRUE);
// remember File Id & full path
UDFStoreFileId(Fcb->Vcb, Ccb, FileInfo, PtrBuffer->IndexNumber.QuadPart);
UDFReleaseResource(&(Fcb->Vcb->FileIdResource));
try_exit: NOTHING;
} _SEH2_FINALLY {
if(NT_SUCCESS(RC)) {
// Return the amount of information filled in.
*PtrReturnedLength -= sizeof(FILE_INTERNAL_INFORMATION);
}
} _SEH2_END;
return(RC);
} // end UDFGetInternalInformation()
/*
Return zero-filled EAs to the caller.
*/
NTSTATUS
UDFGetEaInformation(
PtrUDFIrpContext PtrIrpContext,
IN PtrUDFFCB Fcb,
IN PFILE_EA_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
AdPrint(("UDFGetEaInformation\n"));
_SEH2_TRY {
if(*PtrReturnedLength < (LONG)sizeof(FILE_EA_INFORMATION)) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
// Zero out the supplied buffer.
PtrBuffer->EaSize = 0;
try_exit: NOTHING;
} _SEH2_FINALLY {
if(NT_SUCCESS(RC)) {
// Return the amount of information filled in.
*PtrReturnedLength -= sizeof(FILE_EA_INFORMATION);
}
} _SEH2_END;
return(RC);
} // end UDFGetEaInformation()
/*
Return file's long name to the caller.
*/
NTSTATUS
UDFGetFullNameInformation(
IN PFILE_OBJECT FileObject,
IN PFILE_NAME_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
ULONG BytesToCopy;
NTSTATUS RC = STATUS_SUCCESS;
AdPrint(("UDFGetFullNameInformation\n"));
PtrBuffer->FileNameLength = FileObject->FileName.Length;
BytesToCopy = FileObject->FileName.Length;
if (PtrBuffer->FileNameLength + sizeof( ULONG ) > (ULONG)(*PtrReturnedLength)) {
BytesToCopy = *PtrReturnedLength - sizeof( ULONG );
RC = STATUS_BUFFER_OVERFLOW;
}
RtlCopyMemory( PtrBuffer->FileName, FileObject->FileName.Buffer, BytesToCopy );
// Reduce the available bytes by the amount stored into this buffer.
*PtrReturnedLength -= sizeof( ULONG ) + PtrBuffer->FileNameLength;
return RC;
} // end UDFGetFullNameInformation()
/*
Return file short(8.3) name to the caller.
*/
NTSTATUS
UDFGetAltNameInformation(
IN PtrUDFFCB Fcb,
IN PFILE_NAME_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
PDIR_INDEX_ITEM DirNdx;
ULONG BytesToCopy;
UNICODE_STRING ShortName;
WCHAR ShortNameBuffer[13];
AdPrint(("UDFGetAltNameInformation: \n"));
*PtrReturnedLength -= FIELD_OFFSET(FILE_NAME_INFORMATION, FileName[0]);
DirNdx = UDFDirIndex(UDFGetDirIndexByFileInfo(Fcb->FileInfo), Fcb->FileInfo->Index);
ShortName.MaximumLength = 13 * sizeof(WCHAR);
ShortName.Buffer = (PWCHAR)&ShortNameBuffer;
UDFDOSName__(Fcb->Vcb, &ShortName, &(DirNdx->FName), Fcb->FileInfo);
if(*PtrReturnedLength < ShortName.Length) {
return(STATUS_BUFFER_OVERFLOW);
} else {
BytesToCopy = ShortName.Length;
*PtrReturnedLength -= ShortName.Length;
}
RtlCopyMemory( &(PtrBuffer->FileName),
ShortName.Buffer,
BytesToCopy );
PtrBuffer->FileNameLength = ShortName.Length;
return(STATUS_SUCCESS);
} // end UDFGetAltNameInformation()
/*
Get file position information
*/
NTSTATUS
UDFGetPositionInformation(
IN PFILE_OBJECT FileObject,
IN PFILE_POSITION_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
if(*PtrReturnedLength < (LONG)sizeof(FILE_POSITION_INFORMATION)) {
return(STATUS_BUFFER_OVERFLOW);
}
PtrBuffer->CurrentByteOffset = FileObject->CurrentByteOffset;
// Modify the local variable for BufferLength appropriately.
*PtrReturnedLength -= sizeof(FILE_POSITION_INFORMATION);
return(STATUS_SUCCESS);
} // end UDFGetAltNameInformation()
/*
Get file file stream(s) information
*/
NTSTATUS
UDFGetFileStreamInformation(
IN PtrUDFFCB Fcb,
IN PFILE_STREAM_INFORMATION PtrBuffer,
IN OUT PLONG PtrReturnedLength
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO FileInfo;
PUDF_FILE_INFO SDirInfo;
PVCB Vcb;
BOOLEAN FcbAcquired = FALSE;
uint_di i;
LONG l;
PDIR_INDEX_HDR hSDirIndex;
PDIR_INDEX_ITEM SDirIndex;
PFILE_BOTH_DIR_INFORMATION NTFileInfo = NULL;
AdPrint(("UDFGetFileStreamInformation\n"));
_SEH2_TRY {
UDFAcquireResourceExclusive(&(Fcb->Vcb->FileIdResource), TRUE);
FcbAcquired = TRUE;
FileInfo = Fcb->FileInfo;
if(!FileInfo) {
AdPrint(("!!!!!!!! Bu-u-u-u-u-g !!!!!!!!!!!\n"));
AdPrint(("!!!! UDFGetFileStreamInformation to unopened file !!!!\n"));
try_return(RC = STATUS_INVALID_PARAMETER);
}
Vcb = Fcb->Vcb;
// Zero out the supplied buffer.
RtlZeroMemory(PtrBuffer, *PtrReturnedLength);
if(!(SDirInfo = FileInfo->Dloc->SDirInfo) ||
UDFIsSDirDeleted(SDirInfo) ) {
(*PtrReturnedLength) -= (sizeof(FILE_STREAM_INFORMATION) - sizeof(WCHAR));
try_return(RC = STATUS_SUCCESS);
}
hSDirIndex = SDirInfo->Dloc->DirIndex;
NTFileInfo = (PFILE_BOTH_DIR_INFORMATION)MyAllocatePool__(NonPagedPool, sizeof(FILE_BOTH_DIR_INFORMATION)+UDF_NAME_LEN*sizeof(WCHAR));
if(!NTFileInfo) try_return(RC = STATUS_INSUFFICIENT_RESOURCES);
for(i=2; (SDirIndex = UDFDirIndex(hSDirIndex,i)); i++) {
if((SDirIndex->FI_Flags & UDF_FI_FLAG_FI_INTERNAL) ||
UDFIsDeleted(SDirIndex) ||
!SDirIndex->FName.Buffer )
continue;
// copy data to buffer
if(*PtrReturnedLength < (l = ((sizeof(FILE_STREAM_INFORMATION) - sizeof(WCHAR)) +
SDirIndex->FName.Length + 3) & (~3)) ) {
try_return(RC = STATUS_BUFFER_OVERFLOW);
}
RC = UDFFileDirInfoToNT(Vcb, SDirIndex, NTFileInfo);
PtrBuffer->NextEntryOffset = l;
PtrBuffer->StreamNameLength = SDirIndex->FName.Length;
PtrBuffer->StreamSize = NTFileInfo->EndOfFile;
PtrBuffer->StreamAllocationSize = NTFileInfo->AllocationSize;
RtlCopyMemory(&(PtrBuffer->StreamName), SDirIndex->FName.Buffer, SDirIndex->FName.Length);
*PtrReturnedLength -= l;
*((PCHAR*)(&PtrBuffer)) += l;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(FcbAcquired)
UDFReleaseResource(&(Fcb->Vcb->FileIdResource));
if(NTFileInfo)
MyFreePool__(NTFileInfo);
} _SEH2_END;
return(RC);
} // end UDFGetFileStreamInformation()
//*******************************************************************
#ifndef UDF_READ_ONLY_BUILD
/*
Set some time-stamps and file attributes supplied by the caller.
*/
NTSTATUS
UDFSetBasicInformation(
IN PtrUDFFCB Fcb,
IN PtrUDFCCB Ccb,
IN PFILE_OBJECT FileObject,
IN PFILE_BASIC_INFORMATION PtrBuffer)
{
NTSTATUS RC = STATUS_SUCCESS;
ULONG NotifyFilter = 0;
AdPrint(("UDFSetBasicInformation\n"));
_SEH2_TRY {
// Obtain a pointer to the directory entry associated with
// the FCB being modifed. The directory entry is obviously
// part of the data associated with the parent directory that
// contains this particular file stream.
if(PtrBuffer->FileAttributes) {
UDFUpdateAttrTime(Fcb->Vcb, Fcb->FileInfo);
} else
if( UDFIsADirectory(Fcb->FileInfo) &&
!(Fcb->Vcb->CompatFlags & UDF_VCB_IC_UPDATE_UCHG_DIR_ACCESS_TIME) &&
((Fcb->FileInfo->Dloc->DataLoc.Modified ||
Fcb->FileInfo->Dloc->AllocLoc.Modified ||
(Fcb->FileInfo->Dloc->FE_Flags & UDF_FE_FLAG_FE_MODIFIED) ||
Fcb->FileInfo->Dloc->FELoc.Modified))
) {
// ignore Access Time Modification for unchanged Dir
if(!PtrBuffer->CreationTime.QuadPart &&
PtrBuffer->LastAccessTime.QuadPart &&
!PtrBuffer->ChangeTime.QuadPart &&
!PtrBuffer->LastWriteTime.QuadPart)
try_return(RC);
}
UDFSetFileXTime(Fcb->FileInfo,
&(PtrBuffer->CreationTime.QuadPart),
&(PtrBuffer->LastAccessTime.QuadPart),
&(PtrBuffer->ChangeTime.QuadPart),
&(PtrBuffer->LastWriteTime.QuadPart) );
if(PtrBuffer->CreationTime.QuadPart) {
// The interesting thing here is that the user has set certain time
// fields. However, before doing this, the user may have performed
// I/O which in turn would have caused FSD to mark the fact that
// write/access time should be modifed at cleanup.
// We'll mark the fact that such updates are no longer
// required since the user has explicitly specified the values he
// wishes to see associated with the file stream.
Fcb->NTRequiredFCB->CreationTime = PtrBuffer->CreationTime;
Ccb->CCBFlags |= UDF_CCB_CREATE_TIME_SET;
NotifyFilter |= FILE_NOTIFY_CHANGE_CREATION;
}
if(PtrBuffer->LastAccessTime.QuadPart) {
Fcb->NTRequiredFCB->LastAccessTime = PtrBuffer->LastAccessTime;
Ccb->CCBFlags |= UDF_CCB_ACCESS_TIME_SET;
NotifyFilter |= FILE_NOTIFY_CHANGE_LAST_ACCESS;
}
if(PtrBuffer->ChangeTime.QuadPart) {
Fcb->NTRequiredFCB->ChangeTime = PtrBuffer->ChangeTime;
Ccb->CCBFlags |= UDF_CCB_MODIFY_TIME_SET;
}
if(PtrBuffer->LastWriteTime.QuadPart) {
Fcb->NTRequiredFCB->LastWriteTime = PtrBuffer->LastWriteTime;
Ccb->CCBFlags |= UDF_CCB_WRITE_TIME_SET;
NotifyFilter |= FILE_NOTIFY_CHANGE_LAST_WRITE;
}
// Now come the attributes.
if(PtrBuffer->FileAttributes) {
// We have a non-zero attribute value.
// The presence of a particular attribute indicates that the
// user wishes to set the attribute value. The absence indicates
// the user wishes to clear the particular attribute.
// Our routine ignores unsupported flags
PtrBuffer->FileAttributes &= ~(FILE_ATTRIBUTE_NORMAL);
// Similarly, we should pick out other invalid flag values.
if( (PtrBuffer->FileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
!(Fcb->FCBFlags & UDF_FCB_DIRECTORY))
try_return(RC = STATUS_INVALID_PARAMETER);
if(PtrBuffer->FileAttributes & FILE_ATTRIBUTE_TEMPORARY) {
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY)
try_return(RC = STATUS_INVALID_PARAMETER);
FileObject->Flags |= FO_TEMPORARY_FILE;
} else {
FileObject->Flags &= ~FO_TEMPORARY_FILE;
}
if(PtrBuffer->FileAttributes & FILE_ATTRIBUTE_READONLY) {
Fcb->FCBFlags |= UDF_FCB_READ_ONLY;
} else {
Fcb->FCBFlags &= ~UDF_FCB_READ_ONLY;
}
UDFAttributesToUDF(UDFDirIndex(UDFGetDirIndexByFileInfo(Fcb->FileInfo), Fcb->FileInfo->Index),
NULL, PtrBuffer->FileAttributes);
(UDFDirIndex(UDFGetDirIndexByFileInfo(Fcb->FileInfo), Fcb->FileInfo->Index))
->FI_Flags |= UDF_FI_FLAG_SYS_ATTR;
// If the FSD supports file compression, we may wish to
// note the user's preferences for compressing/not compressing
// the file at this time.
Ccb->CCBFlags |= UDF_CCB_ATTRIBUTES_SET;
NotifyFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
}
if(NotifyFilter) {
UDFNotifyFullReportChange( Fcb->Vcb, Fcb->FileInfo,
NotifyFilter, FILE_ACTION_MODIFIED);
UDFSetFileSizeInDirNdx(Fcb->Vcb, Fcb->FileInfo, NULL);
Fcb->FileInfo->Dloc->FE_Flags |= UDF_FE_FLAG_FE_MODIFIED;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
;
} _SEH2_END;
return(RC);
} // end UDFSetBasicInformation()
NTSTATUS
UDFMarkStreamsForDeletion(
IN PVCB Vcb,
IN PtrUDFFCB Fcb,
IN BOOLEAN ForDel
)
{
NTSTATUS RC = STATUS_SUCCESS;
PUDF_FILE_INFO SDirInfo = NULL;
PUDF_FILE_INFO FileInfo = NULL;
ULONG lc;
BOOLEAN SDirAcq = FALSE;
BOOLEAN StrAcq = FALSE;
uint_di d,i;
_SEH2_TRY {
// In some cases we needn't marking Streams for deleteion
// (Not opened or Don't exist)
if(UDFIsAStream(Fcb->FileInfo) ||
UDFIsAStreamDir(Fcb->FileInfo) ||
!UDFHasAStreamDir(Fcb->FileInfo) ||
!Fcb->FileInfo->Dloc->SDirInfo ||
UDFIsSDirDeleted(Fcb->FileInfo->Dloc->SDirInfo) ||
(UDFGetFileLinkCount(Fcb->FileInfo) > 1) )
try_return (RC /*=STATUS_SUCCESS*/);
// We shall mark Streams for deletion if there is no
// Links to the file. Otherwise we'll delete only the file.
// If we are asked to unmark Streams, we'll precess the whole Tree
RC = UDFOpenStreamDir__(Vcb, Fcb->FileInfo, &SDirInfo);
if(!NT_SUCCESS(RC))
try_return(RC);
if(SDirInfo->Fcb &&
SDirInfo->Fcb->NTRequiredFCB) {
UDF_CHECK_PAGING_IO_RESOURCE(SDirInfo->Fcb->NTRequiredFCB);
UDFAcquireResourceExclusive(&(SDirInfo->Fcb->NTRequiredFCB->MainResource),TRUE);
SDirAcq = TRUE;
}
if(!ForDel || ((lc = UDFGetFileLinkCount(Fcb->FileInfo)) < 2)) {
UDF_DIR_SCAN_CONTEXT ScanContext;
PDIR_INDEX_ITEM DirNdx;
// It is not worth checking whether the Stream can be deleted if
// Undelete requested
if(ForDel &&
// scan DirIndex
UDFDirIndexInitScan(SDirInfo, &ScanContext, 2)) {
// Check if we can delete Streams
while((DirNdx = UDFDirIndexScan(&ScanContext, &FileInfo))) {
if(!FileInfo)
continue;
if(FileInfo->Fcb) {
FileInfo->Fcb->NTRequiredFCB->AcqFlushCount++;
MmPrint((" MmFlushImageSection() for Stream\n"));
if(!MmFlushImageSection(&(FileInfo->Fcb->NTRequiredFCB->SectionObject), MmFlushForDelete)) {
FileInfo->Fcb->NTRequiredFCB->AcqFlushCount--;
try_return(RC = STATUS_CANNOT_DELETE);
}
FileInfo->Fcb->NTRequiredFCB->AcqFlushCount--;
}
}
}
// (Un)Mark Streams for deletion
// Perform sequencial Open for Streams & mark 'em
// for deletion. We should not get FileInfo pointers directly
// from DirNdx[i] to prevent great troubles with linked
// files. We should mark for deletion FI with proper ParentFile
// pointer.
d = UDFDirIndexGetLastIndex(SDirInfo->Dloc->DirIndex);
for(i=2; i<d; i++) {
RC = UDFOpenFile__(Vcb,
FALSE,TRUE,NULL,
SDirInfo,&FileInfo,&i);
ASSERT(NT_SUCCESS(RC) || (RC == STATUS_FILE_DELETED));
if(NT_SUCCESS(RC)) {
if(FileInfo->Fcb) {
if(FileInfo->Fcb->NTRequiredFCB) {
UDF_CHECK_PAGING_IO_RESOURCE(FileInfo->Fcb->NTRequiredFCB);
UDFAcquireResourceExclusive(&(FileInfo->Fcb->NTRequiredFCB->MainResource),TRUE);
StrAcq = TRUE;
}
#ifndef UDF_ALLOW_LINKS_TO_STREAMS
if(UDFGetFileLinkCount(FileInfo) >= 2) {
// Currently, UDF_INFO package doesn't
// support this case, so we'll inform developer
// about this to prevent on-disk space leaks...
BrutePoint();
try_return(RC = STATUS_CANNOT_DELETE);
}
#endif //UDF_ALLOW_LINKS_TO_STREAMS
if(ForDel) {
AdPrint((" SET stream DeleteOnClose\n"));
#ifdef UDF_DBG
ASSERT(!(FileInfo->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
if(FileInfo->ParentFile &&
FileInfo->ParentFile->Fcb) {
ASSERT(!(FileInfo->ParentFile->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
}
#endif // UDF_DBG
FileInfo->Fcb->FCBFlags |= (UDF_FCB_DELETE_ON_CLOSE |
UDF_FCB_DELETE_PARENT);
} else {
AdPrint((" CLEAR stream DeleteOnClose\n"));
FileInfo->Fcb->FCBFlags &= ~(UDF_FCB_DELETE_ON_CLOSE |
UDF_FCB_DELETE_PARENT);
}
}
UDFCloseFile__(Vcb, FileInfo);
} else
if(RC == STATUS_FILE_DELETED) {
// That's OK if STATUS_FILE_DELETED returned...
RC = STATUS_SUCCESS;
}
if(FileInfo) {
if(UDFCleanUpFile__(Vcb, FileInfo)) {
ASSERT(!StrAcq && !(FileInfo->Fcb));
MyFreePool__(FileInfo);
}
if(StrAcq) {
UDF_CHECK_PAGING_IO_RESOURCE(FileInfo->Fcb->NTRequiredFCB);
UDFReleaseResource(&(FileInfo->Fcb->NTRequiredFCB->MainResource));
StrAcq = FALSE;
}
}
FileInfo = NULL;
}
// Mark SDir for deletion
if(SDirInfo->Fcb) {
if(ForDel) {
#ifdef UDF_DBG
ASSERT(!(SDirInfo->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
if(SDirInfo->ParentFile &&
SDirInfo->ParentFile->Fcb) {
ASSERT(!(SDirInfo->ParentFile->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
}
#endif // UDF_DBG
AdPrint((" SET stream dir DeleteOnClose\n"));
SDirInfo->Fcb->FCBFlags |= (UDF_FCB_DELETE_ON_CLOSE |
UDF_FCB_DELETE_PARENT);
} else {
AdPrint((" CLEAR stream dir DeleteOnClose\n"));
SDirInfo->Fcb->FCBFlags &= ~(UDF_FCB_DELETE_ON_CLOSE |
UDF_FCB_DELETE_PARENT);
}
}
} else
if(lc >= 2) {
// if caller wants us to perform DelTree for Streams, but
// someone keeps Stream opened and there is a Link to this
// file, we can't delete it immediately (on Cleanup) & should
// not delete the whole Tree. Instead, we'll set DELETE_PARENT
// flag in SDir to kill this file later, when all the Handles
// to Streams, opened via this file, would be closed
#ifdef UDF_DBG
ASSERT(!(SDirInfo->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
if(SDirInfo->ParentFile &&
SDirInfo->ParentFile->Fcb) {
ASSERT(!(SDirInfo->ParentFile->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY));
}
#endif // UDF_DBG
if(SDirInfo->Fcb)
SDirInfo->Fcb->FCBFlags |= UDF_FCB_DELETE_PARENT;
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(FileInfo) {
UDFCloseFile__(Vcb, FileInfo);
if(UDFCleanUpFile__(Vcb, FileInfo)) {
ASSERT(!StrAcq && !(FileInfo->Fcb));
MyFreePool__(FileInfo);
}
if(StrAcq) {
UDF_CHECK_PAGING_IO_RESOURCE(FileInfo->Fcb->NTRequiredFCB);
UDFReleaseResource(&(FileInfo->Fcb->NTRequiredFCB->MainResource));
}
SDirInfo = NULL;
}
if(SDirInfo) {
UDFCloseFile__(Vcb, SDirInfo);
if(SDirAcq) {
UDF_CHECK_PAGING_IO_RESOURCE(SDirInfo->Fcb->NTRequiredFCB);
UDFReleaseResource(&(SDirInfo->Fcb->NTRequiredFCB->MainResource));
}
if(UDFCleanUpFile__(Vcb, SDirInfo)) {
MyFreePool__(SDirInfo);
}
SDirInfo = NULL;
}
} _SEH2_END;
return RC;
} // end UDFMarkStreamsForDeletion()
/*
(Un)Mark file for deletion.
*/
NTSTATUS
UDFSetDispositionInformation(
IN PtrUDFFCB Fcb,
IN PtrUDFCCB Ccb,
IN PVCB Vcb,
IN PFILE_OBJECT FileObject,
IN BOOLEAN Delete
)
{
NTSTATUS RC = STATUS_SUCCESS;
// PUDF_FILE_INFO SDirInfo = NULL;
// PUDF_FILE_INFO FileInfo = NULL;
ULONG lc;
AdPrint(("UDFSetDispositionInformation\n"));
_SEH2_TRY {
if(!Delete) {
AdPrint((" CLEAR DeleteOnClose\n"));
// "un-delete" the file.
Fcb->FCBFlags &= ~UDF_FCB_DELETE_ON_CLOSE;
if(FileObject)
FileObject->DeletePending = FALSE;
RC = UDFMarkStreamsForDeletion(Vcb, Fcb, FALSE); // Undelete
try_return(RC);
}
AdPrint((" SET DeleteOnClose\n"));
// The easy part is over. Now, we know that the user wishes to
// delete the corresponding directory entry (of course, if this
// is the only link to the file stream, any on-disk storage space
// associated with the file stream will also be released when the
// (only) link is deleted!)
// Do some checking to see if the file can even be deleted.
if(Fcb->FCBFlags & UDF_FCB_DELETE_ON_CLOSE) {
// All done!
try_return(RC);
}
if(Vcb->VCBFlags & UDF_VCB_FLAGS_VOLUME_READ_ONLY) {
try_return(RC = STATUS_CANNOT_DELETE);
}
if(Fcb->FCBFlags & UDF_FCB_READ_ONLY) {
RC = UDFCheckAccessRights(NULL, NULL, Fcb->ParentFcb, NULL, FILE_DELETE_CHILD, 0);
if(!NT_SUCCESS(RC)) {
try_return (RC = STATUS_CANNOT_DELETE);
}
}
// It would not be prudent to allow deletion of either a root
// directory or a directory that is not empty.
if(Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY)
try_return(RC = STATUS_CANNOT_DELETE);
lc = UDFGetFileLinkCount(Fcb->FileInfo);
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY) {
// Perform check to determine whether the directory
// is empty or not.
if(!UDFIsDirEmpty__(Fcb->FileInfo)) {
try_return(RC = STATUS_DIRECTORY_NOT_EMPTY);
}
} else {
// An important step is to check if the file stream has been
// mapped by any process. The delete cannot be allowed to proceed
// in this case.
MmPrint((" MmFlushImageSection()\n"));
Fcb->NTRequiredFCB->AcqFlushCount++;
if(!MmFlushImageSection(&(Fcb->NTRequiredFCB->SectionObject),
(lc > 1) ? MmFlushForWrite : MmFlushForDelete)) {
Fcb->NTRequiredFCB->AcqFlushCount--;
try_return(RC = STATUS_CANNOT_DELETE);
}
Fcb->NTRequiredFCB->AcqFlushCount--;
}
// We should also mark Streams for deletion if there are no
// Links to the file. Otherwise we'll delete only the file
if(lc > 1) {
RC = STATUS_SUCCESS;
} else {
RC = UDFMarkStreamsForDeletion(Vcb, Fcb, TRUE); // Delete
if(!NT_SUCCESS(RC))
try_return(RC);
}
// Set a flag to indicate that this directory entry will become history
// at cleanup.
Fcb->FCBFlags |= UDF_FCB_DELETE_ON_CLOSE;
if(FileObject)
FileObject->DeletePending = TRUE;
if((Fcb->FCBFlags & UDF_FCB_DIRECTORY) && Ccb) {
FsRtlNotifyFullChangeDirectory( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PVOID)Ccb, NULL, FALSE, FALSE,
0, NULL, NULL, NULL );
}
try_exit: NOTHING;
} _SEH2_FINALLY {
;
} _SEH2_END;
return(RC);
} // end UDFSetDispositionInformation()
/*
Change file allocation length.
*/
NTSTATUS
UDFSetAllocationInformation(
IN PtrUDFFCB Fcb,
IN PtrUDFCCB Ccb,
IN PVCB Vcb,
IN PFILE_OBJECT FileObject,
IN PtrUDFIrpContext PtrIrpContext,
IN PIRP Irp,
IN PFILE_ALLOCATION_INFORMATION PtrBuffer
)
{
NTSTATUS RC = STATUS_SUCCESS;
BOOLEAN TruncatedFile = FALSE;
BOOLEAN ModifiedAllocSize = FALSE;
BOOLEAN CacheMapInitialized = FALSE;
BOOLEAN AcquiredPagingIo = FALSE;
AdPrint(("UDFSetAllocationInformation\n"));
_SEH2_TRY {
// Increasing the allocation size associated with a file stream
// is relatively easy. All we have to do is execute some FSD
// specific code to check whether we have enough space available
// (and if the FSD supports user/volume quotas, whether the user
// is not exceeding quota), and then increase the file size in the
// corresponding on-disk and in-memory structures.
// Then, all we should do is inform the Cache Manager about the
// increased allocation size.
// First, do whatever error checking is appropriate here (e.g. whether
// the caller is trying the change size for a directory, etc.).
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY)
try_return(RC = STATUS_INVALID_PARAMETER);
Fcb->NTRequiredFCB->CommonFCBHeader.IsFastIoPossible = UDFIsFastIoPossible(Fcb);
if ((FileObject->SectionObjectPointer->DataSectionObject != NULL) &&
(FileObject->SectionObjectPointer->SharedCacheMap == NULL) &&
!FlagOn(Irp->Flags, IRP_PAGING_IO)) {
ASSERT( !FlagOn( FileObject->Flags, FO_CLEANUP_COMPLETE ) );
// Now initialize the cache map.
MmPrint((" CcInitializeCacheMap()\n"));
CcInitializeCacheMap( FileObject,
(PCC_FILE_SIZES)&Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize,
FALSE,
&(UDFGlobalData.CacheMgrCallBacks),
Fcb->NTRequiredFCB );
CacheMapInitialized = TRUE;
}
// Are we increasing the allocation size?
if(Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.QuadPart <
PtrBuffer->AllocationSize.QuadPart) {
// Yes. Do the FSD specific stuff i.e. increase reserved
// space on disk.
if(((LONGLONG)UDFGetFreeSpace(Vcb) << Vcb->LBlockSizeBits) < PtrBuffer->AllocationSize.QuadPart) {
try_return(RC = STATUS_DISK_FULL);
}
// RC = STATUS_SUCCESS;
ModifiedAllocSize = TRUE;
} else if(Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize.QuadPart >
PtrBuffer->AllocationSize.QuadPart) {
// This is the painful part. See if the VMM will allow us to proceed.
// The VMM will deny the request if:
// (a) any image section exists OR
// (b) a data section exists and the size of the user mapped view
// is greater than the new size
// Otherwise, the VMM should allow the request to proceed.
MmPrint((" MmCanFileBeTruncated()\n"));
if(!MmCanFileBeTruncated(&(Fcb->NTRequiredFCB->SectionObject), &(PtrBuffer->AllocationSize))) {
// VMM said no way!
try_return(RC = STATUS_USER_MAPPED_FILE);
}
// Perform our directory entry modifications. Release any on-disk
// space we may need to in the process.
ModifiedAllocSize = TRUE;
TruncatedFile = TRUE;
}
ASSERT(NT_SUCCESS(RC));
// This is a good place to check if we have performed a truncate
// operation. If we have perform a truncate (whether we extended
// or reduced file size or even leave it intact), we should update
// file time stamps.
FileObject->Flags |= FO_FILE_MODIFIED;
// Last, but not the lease, we must inform the Cache Manager of file size changes.
if(ModifiedAllocSize) {
// If we decreased the allocation size to less than the
// current file size, modify the file size value.
// Similarly, if we decreased the value to less than the
// current valid data length, modify that value as well.
AcquiredPagingIo = UDFAcquireResourceExclusiveWithCheck(&(Fcb->NTRequiredFCB->PagingIoResource));
// Update the FCB Header with the new allocation size.
if(TruncatedFile) {
if(Fcb->NTRequiredFCB->CommonFCBHeader.ValidDataLength.QuadPart >
PtrBuffer->AllocationSize.QuadPart) {
// Decrease the valid data length value.
Fcb->NTRequiredFCB->CommonFCBHeader.ValidDataLength =
PtrBuffer->AllocationSize;
}
if(Fcb->NTRequiredFCB->CommonFCBHeader.FileSize.QuadPart >
PtrBuffer->AllocationSize.QuadPart) {
// Decrease the file size value.
Fcb->NTRequiredFCB->CommonFCBHeader.FileSize =
PtrBuffer->AllocationSize;
RC = UDFResizeFile__(Vcb, Fcb->FileInfo, PtrBuffer->AllocationSize.QuadPart);
// UDFSetFileSizeInDirNdx(Vcb, Fcb->FileInfo, NULL);
}
} else {
Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize = PtrBuffer->AllocationSize;
// UDFSetFileSizeInDirNdx(Vcb, Fcb->FileInfo,
// &(PtrBuffer->AllocationSize.QuadPart));
}
if(AcquiredPagingIo) {
UDFReleaseResource(&(Fcb->NTRequiredFCB->PagingIoResource));
AcquiredPagingIo = FALSE;
}
// If the FCB has not had caching initiated, it is still valid
// for us to invoke the NT Cache Manager. It is possible in such
// situations for the call to be no'oped (unless some user has
// mapped in the file)
// NOTE: The invocation to CcSetFileSizes() will quite possibly
// result in a recursive call back into the file system.
// This is because the NT Cache Manager will typically
// perform a flush before telling the VMM to purge pages
// especially when caching has not been initiated on the
// file stream, but the user has mapped the file into
// the process' virtual address space.
MmPrint((" CcSetFileSizes()\n"));
Fcb->NTRequiredFCB->AcqFlushCount++;
CcSetFileSizes(FileObject, (PCC_FILE_SIZES)&(Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize));
Fcb->NTRequiredFCB->AcqFlushCount--;
Fcb->NTRequiredFCB->NtReqFCBFlags |= UDF_NTREQ_FCB_MODIFIED;
// Inform any pending IRPs (notify change directory).
if(UDFIsAStream(Fcb->FileInfo)) {
UDFNotifyFullReportChange( Vcb, Fcb->FileInfo,
FILE_NOTIFY_CHANGE_STREAM_SIZE,
FILE_ACTION_MODIFIED_STREAM);
} else {
UDFNotifyFullReportChange( Vcb, Fcb->FileInfo,
FILE_NOTIFY_CHANGE_SIZE,
FILE_ACTION_MODIFIED);
}
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(AcquiredPagingIo) {
UDFReleaseResource(&(Fcb->NTRequiredFCB->PagingIoResource));
AcquiredPagingIo = FALSE;
}
if (CacheMapInitialized) {
MmPrint((" CcUninitializeCacheMap()\n"));
CcUninitializeCacheMap( FileObject, NULL, NULL );
}
} _SEH2_END;
return(RC);
} // end UDFSetAllocationInformation()
/*
Set end of file (resize).
*/
NTSTATUS
UDFSetEOF(
IN PIO_STACK_LOCATION PtrSp,
IN PtrUDFFCB Fcb,
IN PtrUDFCCB Ccb,
IN PVCB Vcb,
IN PFILE_OBJECT FileObject,
IN PIRP Irp,
IN PFILE_END_OF_FILE_INFORMATION PtrBuffer
)
{
NTSTATUS RC = STATUS_SUCCESS;
BOOLEAN TruncatedFile = FALSE;
BOOLEAN ModifiedAllocSize = FALSE;
ULONG Attr;
PDIR_INDEX_ITEM DirNdx;
PtrUDFNTRequiredFCB NtReqFcb = NULL;
LONGLONG OldFileSize;
// BOOLEAN ZeroBlock;
BOOLEAN CacheMapInitialized = FALSE;
BOOLEAN AcquiredPagingIo = FALSE;
AdPrint(("UDFSetEOF\n"));
_SEH2_TRY {
// Increasing the allocation size associated with a file stream
// is relatively easy. All we have to do is execute some FSD
// specific code to check whether we have enough space available
// (and if the FSD supports user/volume quotas, whether the user
// is not exceeding quota), and then increase the file size in the
// corresponding on-disk and in-memory structures.
// Then, all we should do is inform the Cache Manager about the
// increased allocation size.
// First, do whatever error checking is appropriate here (e.g. whether
// the caller is trying the change size for a directory, etc.).
if(Fcb->FCBFlags & UDF_FCB_DIRECTORY)
try_return(RC = STATUS_INVALID_PARAMETER);
NtReqFcb = Fcb->NTRequiredFCB;
if((Fcb->FCBFlags & UDF_FCB_DELETED) ||
(NtReqFcb->NtReqFCBFlags & UDF_NTREQ_FCB_DELETED)) {
#ifdef UDF_DBG
if(UDFGetFileLinkCount(Fcb->FileInfo) < 1) {
BrutePoint();
try_return(RC = STATUS_SUCCESS);
} else
#endif // UDF_DBG
try_return(RC = STATUS_SUCCESS);
}
NtReqFcb->CommonFCBHeader.IsFastIoPossible = UDFIsFastIoPossible(Fcb);
if ((FileObject->SectionObjectPointer->DataSectionObject != NULL) &&
(FileObject->SectionObjectPointer->SharedCacheMap == NULL) &&
!(Irp->Flags & IRP_PAGING_IO)) {
ASSERT( !FlagOn( FileObject->Flags, FO_CLEANUP_COMPLETE ) );
// Now initialize the cache map.
MmPrint((" CcInitializeCacheMap()\n"));
CcInitializeCacheMap( FileObject,
(PCC_FILE_SIZES)&Fcb->NTRequiredFCB->CommonFCBHeader.AllocationSize,
FALSE,
&(UDFGlobalData.CacheMgrCallBacks),
Fcb->NTRequiredFCB );
CacheMapInitialized = TRUE;
}
AcquiredPagingIo = UDFAcquireResourceExclusiveWithCheck(&(Fcb->NTRequiredFCB->PagingIoResource));
// Do a special case here for the lazy write of file sizes.
if(PtrSp->Parameters.SetFile.AdvanceOnly) {
// Never have the dirent filesize larger than the fcb filesize
PtrBuffer->EndOfFile.QuadPart =
min(PtrBuffer->EndOfFile.QuadPart,
NtReqFcb->CommonFCBHeader.FileSize.QuadPart);
// Only advance the file size, never reduce it with this call
RC = STATUS_SUCCESS;
if(UDFGetFileSizeFromDirNdx(Vcb, Fcb->FileInfo) >=
PtrBuffer->EndOfFile.QuadPart)
try_return(RC);
UDFSetFileSizeInDirNdx(Vcb, Fcb->FileInfo, &(PtrBuffer->EndOfFile.QuadPart));
goto notify_size_changes;
}
// !!! IMPORTANT !!!
// We can get here after all Handles to the file are closed
// To prevent allocation size incoherency we should
// reference FileInfo _before_ call to UDFResizeFile__()
// and use UDFCloseFile__() _after_ that
// Are we increasing the allocation size?
OldFileSize = NtReqFcb->CommonFCBHeader.FileSize.QuadPart;
if(OldFileSize < PtrBuffer->EndOfFile.QuadPart) {
// Yes. Do the FSD specific stuff i.e. increase reserved
// space on disk.
/*
if (FileObject->PrivateCacheMap)
ZeroBlock = TRUE;
*/
// reference file to pretend that it is opened
UDFReferenceFile__(Fcb->FileInfo);
UDFInterlockedIncrement((PLONG)&(Fcb->ReferenceCount));
UDFInterlockedIncrement((PLONG)&(NtReqFcb->CommonRefCount));
// perform resize operation
RC = UDFResizeFile__(Vcb, Fcb->FileInfo, PtrBuffer->EndOfFile.QuadPart);
// dereference file
UDFCloseFile__(Vcb, Fcb->FileInfo);
UDFInterlockedDecrement((PLONG)&(Fcb->ReferenceCount));
UDFInterlockedDecrement((PLONG)&(NtReqFcb->CommonRefCount));
// update values in NtReqFcb
NtReqFcb->CommonFCBHeader.FileSize.QuadPart =
// NtReqFcb->CommonFCBHeader.ValidDataLength.QuadPart =
PtrBuffer->EndOfFile.QuadPart;
ModifiedAllocSize = TRUE;
} else if(NtReqFcb->CommonFCBHeader.FileSize.QuadPart >
PtrBuffer->EndOfFile.QuadPart) {
// This is the painful part. See if the VMM will allow us to proceed.
// The VMM will deny the request if:
// (a) any image section exists OR
// (b) a data section exists and the size of the user mapped view
// is greater than the new size
// Otherwise, the VMM should allow the request to proceed.
MmPrint((" MmCanFileBeTruncated()\n"));
if(!MmCanFileBeTruncated(&(NtReqFcb->SectionObject), &(PtrBuffer->EndOfFile))) {
// VMM said no way!
try_return(RC = STATUS_USER_MAPPED_FILE);
}
// Perform directory entry modifications. Release any on-disk
// space we may need to in the process.
UDFReferenceFile__(Fcb->FileInfo);
UDFInterlockedIncrement((PLONG)&(Fcb->ReferenceCount));
UDFInterlockedIncrement((PLONG)&(NtReqFcb->CommonRefCount));
// perform resize operation
RC = UDFResizeFile__(Vcb, Fcb->FileInfo, PtrBuffer->EndOfFile.QuadPart);
// dereference file
UDFCloseFile__(Vcb, Fcb->FileInfo);
UDFInterlockedDecrement((PLONG)&(Fcb->ReferenceCount));
UDFInterlockedDecrement((PLONG)&(NtReqFcb->CommonRefCount));
ModifiedAllocSize = TRUE;
TruncatedFile = TRUE;
}
// This is a good place to check if we have performed a truncate
// operation. If we have perform a truncate (whether we extended
// or reduced file size), we should update file time stamps.
// Last, but not the least, we must inform the Cache Manager of file size changes.
if(ModifiedAllocSize && NT_SUCCESS(RC)) {
// If we decreased the allocation size to less than the
// current file size, modify the file size value.
// Similarly, if we decreased the value to less than the
// current valid data length, modify that value as well.
if(TruncatedFile) {
if(NtReqFcb->CommonFCBHeader.ValidDataLength.QuadPart >
PtrBuffer->EndOfFile.QuadPart) {
// Decrease the valid data length value.
NtReqFcb->CommonFCBHeader.ValidDataLength =
PtrBuffer->EndOfFile;
}
if(NtReqFcb->CommonFCBHeader.FileSize.QuadPart >
PtrBuffer->EndOfFile.QuadPart) {
// Decrease the file size value.
NtReqFcb->CommonFCBHeader.FileSize =
PtrBuffer->EndOfFile;
}
UDFSetFileSizeInDirNdx(Vcb, Fcb->FileInfo, NULL);
} else {
// Update the FCB Header with the new allocation size.
// NT expects AllocationSize to be decreased on Close only
NtReqFcb->CommonFCBHeader.AllocationSize.QuadPart =
PtrBuffer->EndOfFile.QuadPart;
// UDFSysGetAllocSize(Vcb, UDFGetFileSize(Fcb->FileInfo));
UDFSetFileSizeInDirNdx(Vcb, Fcb->FileInfo, &(PtrBuffer->EndOfFile.QuadPart));
}
FileObject->Flags |= FO_FILE_MODIFIED;
// UDFGetFileAllocationSize(Vcb, Fcb->FileInfo);
// If the FCB has not had caching initiated, it is still valid
// for us to invoke the NT Cache Manager. It is possible in such
// situations for the call to be no'oped (unless some user has
// mapped in the file)
// Archive bit
if(Vcb->CompatFlags & UDF_VCB_IC_UPDATE_ARCH_BIT) {
DirNdx = UDFDirIndex(UDFGetDirIndexByFileInfo(Fcb->FileInfo), Fcb->FileInfo->Index);
Ccb->CCBFlags &= ~UDF_CCB_ATTRIBUTES_SET;
Attr = UDFAttributesToNT(DirNdx, Fcb->FileInfo->Dloc->FileEntry);
if(!(Attr & FILE_ATTRIBUTE_ARCHIVE))
UDFAttributesToUDF(DirNdx, Fcb->FileInfo->Dloc->FileEntry, Attr | FILE_ATTRIBUTE_ARCHIVE);
}
// NOTE: The invocation to CcSetFileSizes() will quite possibly
// result in a recursive call back into the file system.
// This is because the NT Cache Manager will typically
// perform a flush before telling the VMM to purge pages
// especially when caching has not been initiated on the
// file stream, but the user has mapped the file into
// the process' virtual address space.
MmPrint((" CcSetFileSizes(), thrd:%8.8x\n",PsGetCurrentThread()));
Fcb->NTRequiredFCB->AcqFlushCount++;
CcSetFileSizes(FileObject, (PCC_FILE_SIZES)&(NtReqFcb->CommonFCBHeader.AllocationSize));
Fcb->NTRequiredFCB->AcqFlushCount--;
/* if(ZeroBlock) {
UDFZeroDataEx(NtReqFcb,
OldFileSize,
PtrBuffer->EndOfFile.QuadPart - OldFileSize,
TRUE // CanWait, Vcb, FileObject);
}*/
Fcb->NTRequiredFCB->NtReqFCBFlags |= UDF_NTREQ_FCB_MODIFIED;
notify_size_changes:
if(AcquiredPagingIo) {
UDFReleaseResource(&(Fcb->NTRequiredFCB->PagingIoResource));
AcquiredPagingIo = FALSE;
}
// Inform any pending IRPs (notify change directory).
if(UDFIsAStream(Fcb->FileInfo)) {
UDFNotifyFullReportChange( Vcb, Fcb->FileInfo,
FILE_NOTIFY_CHANGE_STREAM_SIZE,
FILE_ACTION_MODIFIED_STREAM);
} else {
UDFNotifyFullReportChange( Vcb, Fcb->FileInfo,
FILE_NOTIFY_CHANGE_SIZE,
FILE_ACTION_MODIFIED);
}
}
try_exit: NOTHING;
} _SEH2_FINALLY {
if(AcquiredPagingIo) {
UDFReleaseResource(&(Fcb->NTRequiredFCB->PagingIoResource));
AcquiredPagingIo = FALSE;
}
if (CacheMapInitialized) {
MmPrint((" CcUninitializeCacheMap()\n"));
CcUninitializeCacheMap( FileObject, NULL, NULL );
}
} _SEH2_END;
return(RC);
} // end UDFSetEOF()
NTSTATUS
UDFPrepareForRenameMoveLink(
PVCB Vcb,
PBOOLEAN AcquiredVcb,
PBOOLEAN AcquiredVcbEx,
PBOOLEAN SingleDir,
PBOOLEAN AcquiredDir1,
PBOOLEAN AcquiredFcb1,
IN PtrUDFCCB Ccb1,
PUDF_FILE_INFO File1,
PUDF_FILE_INFO Dir1,
PUDF_FILE_INFO Dir2,
BOOLEAN HardLink
)
{
// convert acquisition to Exclusive
// this will prevent us from the following situation:
// There is a pair of objects among input dirs &
// one of them is a parent of another. Sequential resource
// acquisition may lead to deadlock due to concurrent
// CleanUpFcbChain() or UDFCloseFileInfoChain()
UDFInterlockedIncrement((PLONG)&(Vcb->VCBOpenCount));
UDFReleaseResource(&(Vcb->VCBResource));
(*AcquiredVcb) = FALSE;
// At first, make system to issue last Close request
// for our Source & Target ...
// we needn't flush/purge for Source on HLink
UDFRemoveFromSystemDelayedQueue(Dir2->Fcb);
if(!HardLink && (Dir2 != Dir1))
UDFRemoveFromSystemDelayedQueue(File1->Fcb);
#ifdef UDF_DELAYED_CLOSE
_SEH2_TRY {
// Do actual close for all "delayed close" calls
// ... and now remove the rest from our queue
if(!HardLink) {
UDFCloseAllDelayedInDir(Vcb, Dir1);
if(Dir2 != Dir1)
UDFCloseAllDelayedInDir(Vcb, Dir2);
} else {
UDFCloseAllDelayedInDir(Vcb, Dir2);
}
} _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
BrutePoint();
UDFInterlockedDecrement((PLONG)&(Vcb->VCBOpenCount));
return (STATUS_DRIVER_INTERNAL_ERROR);
} _SEH2_END;
#endif //UDF_DELAYED_CLOSE
(*SingleDir) = ((Dir1 == Dir2) && (Dir1->Fcb));
if(!(*SingleDir) ||
(UDFGetFileLinkCount(File1) != 1)) {
UDFAcquireResourceExclusive(&(Vcb->VCBResource), TRUE);
(*AcquiredVcb) = TRUE;
(*AcquiredVcbEx) = TRUE;
UDFInterlockedDecrement((PLONG)&(Vcb->VCBOpenCount));
} else {
UDFAcquireResourceShared(&(Vcb->VCBResource), TRUE);
(*AcquiredVcb) = TRUE;
UDFInterlockedDecrement((PLONG)&(Vcb->VCBOpenCount));
UDF_CHECK_PAGING_IO_RESOURCE(Dir1->Fcb->NTRequiredFCB);
UDFAcquireResourceExclusive(&(Dir1->Fcb->NTRequiredFCB->MainResource),TRUE);
(*AcquiredDir1) = TRUE;
UDF_CHECK_PAGING_IO_RESOURCE(File1->Fcb->NTRequiredFCB);
UDFAcquireResourceExclusive(&(File1->Fcb->NTRequiredFCB->MainResource),TRUE);
(*AcquiredFcb1) = TRUE;
}
return STATUS_SUCCESS;
} // end UDFPrepareForRenameMoveLink()
/*
Rename or move file
*/
NTSTATUS
UDFRename(
IN PIO_STACK_LOCATION PtrSp,
IN PtrUDFFCB Fcb1,
IN PtrUDFCCB Ccb1,
IN PFILE_OBJECT FileObject1, // Source File
IN PFILE_RENAME_INFORMATION PtrBuffer
)
{
// Source Directory
PFILE_OBJECT DirObject1 = FileObject1->RelatedFileObject;
// Target Directory
PFILE_OBJECT DirObject2 = PtrSp->Parameters.SetFile.FileObject;
// Overwite Flag
BOOLEAN Replace = PtrSp->Parameters.SetFile.ReplaceIfExists &&
PtrBuffer->ReplaceIfExists;
NTSTATUS RC;
PVCB Vcb = Fcb1->Vcb;
PtrUDFFCB Fcb2;
BOOLEAN ic;
BOOLEAN AcquiredVcb = TRUE;
BOOLEAN AcquiredVcbEx = FALSE;
BOOLEAN AcquiredDir1 = FALSE;
BOOLEAN AcquiredFcb1 = FALSE;
BOOLEAN SingleDir = TRUE;
BOOLEAN UseClose;
PUDF_FILE_INFO File1;
PUDF_FILE_INFO Dir1;
PUDF_FILE_INFO Dir2;
PUDF_FILE_INFO NextFileInfo, fi;
UNICODE_STRING NewName;
UNICODE_STRING LocalPath;
PtrUDFCCB CurCcb = NULL;
PLIST_ENTRY Link;
ULONG i;
ULONG DirRefCount;
ULONG FileInfoRefCount;
ULONG Attr;
PDIR_INDEX_ITEM DirNdx;
AdPrint(("UDFRename %8.8x\n", DirObject2));
LocalPath.Buffer = NULL;
_SEH2_TRY {
// do we try to rename Volume ?
#ifdef UDF_ALLOW_RENAME_MOVE
if(!(File1 = Fcb1->FileInfo))
#endif //UDF_ALLOW_RENAME_MOVE
try_return (RC = STATUS_ACCESS_DENIED);
// do we try to rename RootDir ?
if(!(Dir1 = File1->ParentFile))
try_return (RC = STATUS_ACCESS_DENIED);
// do we try to rename to RootDir or Volume ?
if(!DirObject2) {
Dir2 = File1->ParentFile;
DirObject2 = DirObject1;
} else
if(DirObject2->FsContext2 &&
(Fcb2 = ((PtrUDFCCB)(DirObject2->FsContext2))->Fcb)) {
Dir2 = ((PtrUDFCCB)(DirObject2->FsContext2))->Fcb->FileInfo;
} else {
try_return (RC = STATUS_INVALID_PARAMETER);
}
// invalid destination ?
if(!Dir2) try_return (RC = STATUS_ACCESS_DENIED);
// Stream can't be a Dir or have StreamDir
if(UDFIsAStreamDir(Dir2)) {
#ifdef UDF_ENABLE_SECURITY
if(UDFIsADirectory(File1)) {
try_return (RC = STATUS_ACCESS_DENIED);
}
// We should check whether File1 has only Internal
// (or Deleted) streams. In this case SDir should be
// removed (in UDFRenameMoveFile__()). Otherwise
// return STATUS_ACCESS_DENIED
if(UDFHasAStreamDir(File1)) {
UDFPrint(("TODO: We should remove Streams from source file\n"));
try_return (RC = STATUS_ACCESS_DENIED);
}
#else //UDF_ENABLE_SECURITY
if(UDFIsADirectory(File1) ||
UDFHasAStreamDir(File1)) {
try_return (RC = STATUS_ACCESS_DENIED);
}
#endif //UDF_ENABLE_SECURITY
}
RC = UDFPrepareForRenameMoveLink(Vcb, &AcquiredVcb, &AcquiredVcbEx,
&SingleDir,
&AcquiredDir1, &AcquiredFcb1,
Ccb1, File1,
Dir1, Dir2,
FALSE); // it is Rename operation
if(!NT_SUCCESS(RC))
try_return(RC);
// check if the source file is in use
if(Fcb1->OpenHandleCount > 1)
try_return (RC = STATUS_ACCESS_DENIED);
ASSERT(Fcb1->OpenHandleCount);
ASSERT(!Fcb1->IrpContextLite);
if(Fcb1->IrpContextLite) {
try_return (RC = STATUS_ACCESS_DENIED);
}
// Check if we have parallel/pending Close threads
if(Fcb1->CcbCount && !SingleDir) {
// if this is the 1st attempt, we'll try to
// synchronize with Close requests
// otherwise fail request
RC = STATUS_ACCESS_DENIED;
post_rename:
if(Fcb1->FCBFlags & UDF_FCB_POSTED_RENAME) {
Fcb1->FCBFlags &= ~UDF_FCB_POSTED_RENAME;
try_return (RC);
}
Fcb1->FCBFlags |= UDF_FCB_POSTED_RENAME;
try_return (RC = STATUS_PENDING);
}
if(!DirObject2) {
// Make sure the name is of legal length.
if(PtrBuffer->FileNameLength > UDF_NAME_LEN*sizeof(WCHAR)) {
try_return(RC = STATUS_OBJECT_NAME_INVALID);
}
NewName.Length = NewName.MaximumLength = (USHORT)(PtrBuffer->FileNameLength);
NewName.Buffer = (PWCHAR)&(PtrBuffer->FileName);
} else {
// This name is by definition legal.
NewName = *((PUNICODE_STRING)&DirObject2->FileName);
}
ic = (Ccb1->CCBFlags & UDF_CCB_CASE_SENSETIVE) ? FALSE : TRUE;
AdPrint((" %ws ->\n %ws\n",
Fcb1->FCBName->ObjectName.Buffer,
NewName.Buffer));
if(UDFIsDirOpened__(File1)) {
// We can't rename file because of unclean references.
// UDF_INFO package can safely do it, but NT side cannot.
// In this case NT requires STATUS_OBJECT_NAME_COLLISION
// rather than STATUS_ACCESS_DENIED
if(NT_SUCCESS(UDFFindFile__(Vcb, ic, &NewName, Dir2)))
try_return(RC = STATUS_OBJECT_NAME_COLLISION);
try_return (RC = STATUS_ACCESS_DENIED);
} else {
// Last check before Moving.
// We can't move across Dir referenced (even internally) file
if(!SingleDir) {
RC = UDFDoesOSAllowFileToBeMoved__(File1);
if(!NT_SUCCESS(RC)) {
// try_return(RC);
goto post_rename;
}
}
ASSERT_REF(Fcb1->ReferenceCount >= File1->RefCount);
ASSERT_REF(Dir1->Fcb->ReferenceCount >= Dir1->RefCount);
ASSERT_REF(Dir2->Fcb->ReferenceCount >= Dir2->RefCount);
RC = UDFRenameMoveFile__(Vcb, ic, &Replace, &NewName, Dir1, Dir2, File1);
}
if(!NT_SUCCESS(RC))
try_return (RC);
ASSERT(UDFDirIndex(File1->ParentFile->Dloc->DirIndex, File1->Index)->FileInfo == File1);
RC = MyCloneUnicodeString(&LocalPath, (Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ?
&UDFGlobalData.UnicodeStrRoot :
&(Dir2->Fcb->FCBName->ObjectName) );
if(!NT_SUCCESS(RC)) try_return (RC);
// RC = MyAppendUnicodeStringToString(&LocalPath, (Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ? &(UDFGlobalData.UnicodeStrRoot) : &(Dir2->Fcb->FCBName->ObjectName));
// if(!NT_SUCCESS(RC)) try_return (RC);
if(Dir2->ParentFile) {
RC = MyAppendUnicodeToString(&LocalPath, L"\\");
if(!NT_SUCCESS(RC)) try_return (RC);
}
RC = MyAppendUnicodeStringToStringTag(&LocalPath, &NewName, MEM_USREN_TAG);
if(!NT_SUCCESS(RC)) try_return (RC);
// Set Archive bit
DirNdx = UDFDirIndex(File1->ParentFile->Dloc->DirIndex, File1->Index);
if(Vcb->CompatFlags & UDF_VCB_IC_UPDATE_ARCH_BIT) {
Attr = UDFAttributesToNT(DirNdx, File1->Dloc->FileEntry);
if(!(Attr & FILE_ATTRIBUTE_ARCHIVE))
UDFAttributesToUDF(DirNdx, File1->Dloc->FileEntry, Attr | FILE_ATTRIBUTE_ARCHIVE);
}
// Update Parent Objects (mark 'em as modified)
if(Vcb->CompatFlags & UDF_VCB_IC_UPDATE_DIR_WRITE) {
if(DirObject1)
DirObject1->Flags |= FO_FILE_MODIFIED;
if(DirObject2) {
DirObject2->Flags |= FO_FILE_MODIFIED;
if(!Replace)
DirObject2->Flags |= FO_FILE_SIZE_CHANGED;
}
}
// report changes
if(SingleDir && !Replace) {
UDFNotifyFullReportChange( Vcb, File1,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_RENAMED_OLD_NAME);
/* UDFNotifyFullReportChange( Vcb, File2,
UDFIsADirectory(File2) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_RENAMED_NEW_NAME );*/
FsRtlNotifyFullReportChange( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PSTRING)&LocalPath,
((Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ? 0 : Dir2->Fcb->FCBName->ObjectName.Length) + sizeof(WCHAR),
NULL,NULL,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_RENAMED_NEW_NAME,
NULL);
} else {
UDFNotifyFullReportChange( Vcb, File1,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_REMOVED);
if(Replace) {
/* UDFNotifyFullReportChange( Vcb, File2,
FILE_NOTIFY_CHANGE_ATTRIBUTES |
FILE_NOTIFY_CHANGE_SIZE |
FILE_NOTIFY_CHANGE_LAST_WRITE |
FILE_NOTIFY_CHANGE_LAST_ACCESS |
FILE_NOTIFY_CHANGE_CREATION |
FILE_NOTIFY_CHANGE_EA,
FILE_ACTION_MODIFIED );*/
FsRtlNotifyFullReportChange( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PSTRING)&LocalPath,
((Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ?
0 : Dir2->Fcb->FCBName->ObjectName.Length) + sizeof(WCHAR),
NULL,NULL,
FILE_NOTIFY_CHANGE_ATTRIBUTES |
FILE_NOTIFY_CHANGE_SIZE |
FILE_NOTIFY_CHANGE_LAST_WRITE |
FILE_NOTIFY_CHANGE_LAST_ACCESS |
FILE_NOTIFY_CHANGE_CREATION |
FILE_NOTIFY_CHANGE_EA,
FILE_ACTION_MODIFIED,
NULL);
} else {
/* UDFNotifyFullReportChange( Vcb, File2,
UDFIsADirectory(File2) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_ADDED );*/
FsRtlNotifyFullReportChange( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PSTRING)&LocalPath,
((Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ?
0 : Dir2->Fcb->FCBName->ObjectName.Length) + sizeof(WCHAR),
NULL,NULL,
UDFIsADirectory(File1) ?
FILE_NOTIFY_CHANGE_DIR_NAME :
FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_ADDED,
NULL);
}
}
// this will prevent structutre release before call to
// UDFCleanUpFcbChain()
UDFInterlockedIncrement((PLONG)&(Dir1->Fcb->ReferenceCount));
UDFInterlockedIncrement((PLONG)&(Dir1->Fcb->NTRequiredFCB->CommonRefCount));
ASSERT_REF(Dir1->Fcb->ReferenceCount >= Dir1->RefCount);
// Look through Ccb list & decrement OpenHandleCounter(s)
// acquire CcbList
if(!SingleDir) {
UDFAcquireResourceExclusive(&(Fcb1->CcbListResource),TRUE);
Link = Fcb1->NextCCB.Flink;
DirRefCount = 0;
FileInfoRefCount = 0;
ASSERT(Link != &(Fcb1->NextCCB));
while (Link != &(Fcb1->NextCCB)) {
NextFileInfo = Dir1;
CurCcb = CONTAINING_RECORD(Link, UDFCCB, NextCCB);
ASSERT(CurCcb->TreeLength);
i = (CurCcb->TreeLength) ? (CurCcb->TreeLength - 1) : 0;
Link = Link->Flink;
UseClose = (CurCcb->CCBFlags & UDF_CCB_CLEANED) ? FALSE : TRUE;
AdPrint((" Ccb:%x:%s:i:%x\n", CurCcb, UseClose ? "Close" : "",i));
// cleanup old parent chain
for(; i && NextFileInfo; i--) {
// remember parent file now
// it will prevent us from data losses
// due to eventual structure release
fi = NextFileInfo->ParentFile;
if(UseClose) {
ASSERT_REF(NextFileInfo->Fcb->ReferenceCount >= NextFileInfo->RefCount);
UDFCloseFile__(Vcb, NextFileInfo);
}
ASSERT_REF(NextFileInfo->Fcb->ReferenceCount > NextFileInfo->RefCount);
ASSERT_REF(NextFileInfo->Fcb->ReferenceCount);
ASSERT_REF(NextFileInfo->Fcb->NTRequiredFCB->CommonRefCount);
UDFInterlockedDecrement((PLONG)&(NextFileInfo->Fcb->ReferenceCount));
UDFInterlockedDecrement((PLONG)&(NextFileInfo->Fcb->NTRequiredFCB->CommonRefCount));
ASSERT_REF(NextFileInfo->Fcb->ReferenceCount >= NextFileInfo->RefCount);
NextFileInfo = fi;
}
if(CurCcb->TreeLength > 1) {
DirRefCount++;
if(UseClose)
FileInfoRefCount++;
CurCcb->TreeLength = 2;
#ifdef UDF_DBG
} else {
BrutePoint();
#endif // UDF_DBG
}
}
UDFReleaseResource(&(Fcb1->CcbListResource));
ASSERT_REF(DirRefCount >= FileInfoRefCount);
// update counters & pointers
Fcb1->ParentFcb = Dir2->Fcb;
// move references to Dir2
UDFInterlockedExchangeAdd((PLONG)&(Dir2->Fcb->ReferenceCount), DirRefCount);
UDFInterlockedExchangeAdd((PLONG)&(Dir2->Fcb->NTRequiredFCB->CommonRefCount), DirRefCount);
ASSERT_REF(Dir2->Fcb->ReferenceCount > Dir2->RefCount);
UDFReferenceFileEx__(Dir2,FileInfoRefCount);
ASSERT_REF(Dir2->Fcb->ReferenceCount >= Dir2->RefCount);
}
ASSERT_REF(Dir2->Fcb->ReferenceCount >= Dir2->RefCount);
ASSERT_REF(Dir2->RefCount);
ASSERT_REF(Dir1->Fcb->ReferenceCount >= Dir1->RefCount);
// Modify name in Fcb1
if(Fcb1->FCBName) {
if(Fcb1->FCBName->ObjectName.Buffer) {
MyFreePool__(Fcb1->FCBName->ObjectName.Buffer);
}
UDFReleaseObjectName(Fcb1->FCBName);
}
Fcb1->FCBName = UDFAllocateObjectName();
if(!(Fcb1->FCBName)) {
insuf_res:
BrutePoint();
// UDFCleanUpFcbChain()...
if(AcquiredFcb1) {
UDF_CHECK_PAGING_IO_RESOURCE(Fcb1->NTRequiredFCB);
UDFReleaseResource(&(Fcb1->NTRequiredFCB->MainResource));
AcquiredDir1 = FALSE;
}
if(AcquiredDir1) {
UDF_CHECK_PAGING_IO_RESOURCE(Dir1->Fcb->NTRequiredFCB);
UDFReleaseResource(&(Dir1->Fcb->NTRequiredFCB->MainResource));
AcquiredDir1 = FALSE;
}
UDFCleanUpFcbChain(Vcb, Dir1, 1, TRUE);
try_return(RC = STATUS_INSUFFICIENT_RESOURCES);
}
RC = MyCloneUnicodeString(&(Fcb1->FCBName->ObjectName), &(Fcb2->FCBName->ObjectName));
if(!NT_SUCCESS(RC))
goto insuf_res;
/* RC = MyAppendUnicodeStringToString(&(Fcb1->FCBName->ObjectName), &(Fcb2->FCBName->ObjectName));
if(!NT_SUCCESS(RC))
goto insuf_res;*/
// if Dir2 is a RootDir, we shoud not append '\\' because
// uit will be the 2nd '\\' character (RootDir's name is also '\\')
if(Dir2->ParentFile) {
RC = MyAppendUnicodeToString(&(Fcb1->FCBName->ObjectName), L"\\");
if(!NT_SUCCESS(RC))
goto insuf_res;
}
RC = MyAppendUnicodeStringToStringTag(&(Fcb1->FCBName->ObjectName), &NewName, MEM_USREN2_TAG);
if(!NT_SUCCESS(RC))
goto insuf_res;
ASSERT_REF(Fcb1->ReferenceCount >= File1->RefCount);
ASSERT_REF(Dir1->Fcb->ReferenceCount >= Dir1->RefCount);
ASSERT_REF(Dir2->Fcb->ReferenceCount >= Dir2->RefCount);
RC = STATUS_SUCCESS;
try_exit: NOTHING;
} _SEH2_FINALLY {
if(AcquiredFcb1) {
UDF_CHECK_PAGING_IO_RESOURCE(Fcb1->NTRequiredFCB);
UDFReleaseResource(&(Fcb1->NTRequiredFCB->MainResource));
}
if(AcquiredDir1) {
UDF_CHECK_PAGING_IO_RESOURCE(Dir1->Fcb->NTRequiredFCB);
UDFReleaseResource(&(Dir1->Fcb->NTRequiredFCB->MainResource));
}
// perform protected structure release
if(NT_SUCCESS(RC) &&
(RC != STATUS_PENDING)) {
ASSERT(AcquiredVcb);
UDFCleanUpFcbChain(Vcb, Dir1, 1, TRUE);
ASSERT_REF(Fcb1->ReferenceCount >= File1->RefCount);
ASSERT_REF(Dir2->Fcb->ReferenceCount >= Dir2->RefCount);
}
if(AcquiredVcb) {
if(AcquiredVcbEx)
UDFConvertExclusiveToSharedLite(&(Vcb->VCBResource));
} else {
// caller assumes Vcb to be acquired shared
BrutePoint();
UDFAcquireResourceShared(&(Vcb->VCBResource), TRUE);
}
if(LocalPath.Buffer) {
MyFreePool__(LocalPath.Buffer);
}
} _SEH2_END;
return RC;
} // end UDFRename()
#endif //UDF_READ_ONLY_BUILD
LONG
UDFFindFileId(
IN PVCB Vcb,
IN LONGLONG Id
)
{
if(!Vcb->FileIdCache) return (-1);
for(ULONG i=0; i<Vcb->FileIdCount; i++) {
if(Vcb->FileIdCache[i].Id == Id) return i;
}
return (-1);
} // end UDFFindFileId()
LONG
UDFFindFreeFileId(
IN PVCB Vcb,
IN LONGLONG Id
)
{
if(!Vcb->FileIdCache) {
if(!(Vcb->FileIdCache = (PUDFFileIDCacheItem)MyAllocatePool__(NonPagedPool, sizeof(UDFFileIDCacheItem)*FILE_ID_CACHE_GRANULARITY)))
return (-1);
RtlZeroMemory(Vcb->FileIdCache, FILE_ID_CACHE_GRANULARITY*sizeof(UDFFileIDCacheItem));
Vcb->FileIdCount = FILE_ID_CACHE_GRANULARITY;
}
for(ULONG i=0; i<Vcb->FileIdCount; i++) {
if(!Vcb->FileIdCache[i].FullName.Buffer) return i;
}
if(!MyReallocPool__((PCHAR)(Vcb->FileIdCache), Vcb->FileIdCount*sizeof(UDFFileIDCacheItem),
(PCHAR*)&(Vcb->FileIdCache), (Vcb->FileIdCount+FILE_ID_CACHE_GRANULARITY)*sizeof(UDFFileIDCacheItem))) {
return (-1);
}
RtlZeroMemory(&(Vcb->FileIdCache[Vcb->FileIdCount]), FILE_ID_CACHE_GRANULARITY*sizeof(UDFFileIDCacheItem));
Vcb->FileIdCount += FILE_ID_CACHE_GRANULARITY;
return (Vcb->FileIdCount - FILE_ID_CACHE_GRANULARITY);
} // end UDFFindFreeFileId()
NTSTATUS
UDFStoreFileId(
IN PVCB Vcb,
IN PtrUDFCCB Ccb,
IN PUDF_FILE_INFO fi,
IN LONGLONG Id
)
{
LONG i;
NTSTATUS RC = STATUS_SUCCESS;
if((i = UDFFindFileId(Vcb, Id)) == (-1)) {
if((i = UDFFindFreeFileId(Vcb, Id)) == (-1)) return STATUS_INSUFFICIENT_RESOURCES;
} else {
return STATUS_SUCCESS;
}
Vcb->FileIdCache[i].Id = Id;
Vcb->FileIdCache[i].CaseSens = (Ccb->CCBFlags & UDF_CCB_CASE_SENSETIVE) ? TRUE : FALSE;
RC = MyCloneUnicodeString(&(Vcb->FileIdCache[i].FullName), &(Ccb->Fcb->FCBName->ObjectName));
/* if(NT_SUCCESS(RC)) {
RC = MyAppendUnicodeStringToStringTag(&(Vcb->FileIdCache[i].FullName), &(Ccb->Fcb->FCBName->ObjectName), MEM_USFIDC_TAG);
}*/
return RC;
} // end UDFStoreFileId()
NTSTATUS
UDFRemoveFileId(
IN PVCB Vcb,
IN LONGLONG Id
)
{
LONG i;
if((i = UDFFindFileId(Vcb, Id)) == (-1)) return STATUS_INVALID_PARAMETER;
MyFreePool__(Vcb->FileIdCache[i].FullName.Buffer);
RtlZeroMemory(&(Vcb->FileIdCache[i]), sizeof(UDFFileIDCacheItem));
return STATUS_SUCCESS;
} // end UDFRemoveFileId()
VOID
UDFReleaseFileIdCache(
IN PVCB Vcb
)
{
if(!Vcb->FileIdCache) return;
for(ULONG i=0; i<Vcb->FileIdCount; i++) {
if(Vcb->FileIdCache[i].FullName.Buffer) {
MyFreePool__(Vcb->FileIdCache[i].FullName.Buffer);
}
}
MyFreePool__(Vcb->FileIdCache);
Vcb->FileIdCache = NULL;
Vcb->FileIdCount = 0;
} // end UDFReleaseFileIdCache()
NTSTATUS
UDFGetOpenParamsByFileId(
IN PVCB Vcb,
IN LONGLONG Id,
OUT PUNICODE_STRING* FName,
OUT BOOLEAN* CaseSens
)
{
LONG i;
if((i = UDFFindFileId(Vcb, Id)) == (-1)) return STATUS_NOT_FOUND;
(*FName) = &(Vcb->FileIdCache[i].FullName);
(*CaseSens) = !(Vcb->FileIdCache[i].CaseSens);
return STATUS_SUCCESS;
} // end UDFGetOpenParamsByFileId()
#ifndef UDF_READ_ONLY_BUILD
#ifdef UDF_ALLOW_HARD_LINKS
/*
create hard link for the file
*/
NTSTATUS
UDFHardLink(
IN PIO_STACK_LOCATION PtrSp,
IN PtrUDFFCB Fcb1,
IN PtrUDFCCB Ccb1,
IN PFILE_OBJECT FileObject1, // Source File
IN PFILE_LINK_INFORMATION PtrBuffer
)
{
// Target Directory
PFILE_OBJECT DirObject2 = PtrSp->Parameters.SetFile.FileObject;
// Overwite Flag
BOOLEAN Replace = PtrSp->Parameters.SetFile.ReplaceIfExists &&
PtrBuffer->ReplaceIfExists;
NTSTATUS RC;
PVCB Vcb = Fcb1->Vcb;
PtrUDFFCB Fcb2;
BOOLEAN ic;
BOOLEAN AcquiredVcb = TRUE;
BOOLEAN AcquiredVcbEx = FALSE;
BOOLEAN AcquiredDir1 = FALSE;
BOOLEAN AcquiredFcb1 = FALSE;
BOOLEAN SingleDir = TRUE;
PUDF_FILE_INFO File1;
PUDF_FILE_INFO Dir1 = NULL;
PUDF_FILE_INFO Dir2;
UNICODE_STRING NewName;
UNICODE_STRING LocalPath;
// PtrUDFCCB CurCcb = NULL;
AdPrint(("UDFHardLink\n"));
LocalPath.Buffer = NULL;
_SEH2_TRY {
// do we try to link Volume ?
if(!(File1 = Fcb1->FileInfo))
try_return (RC = STATUS_ACCESS_DENIED);
// do we try to link RootDir ?
if(!(Dir1 = File1->ParentFile))
try_return (RC = STATUS_ACCESS_DENIED);
// do we try to link Stream / Stream Dir ?
#ifdef UDF_ALLOW_LINKS_TO_STREAMS
if(UDFIsAStreamDir(File1))
try_return (RC = STATUS_ACCESS_DENIED);
#else //UDF_ALLOW_LINKS_TO_STREAMS
if(UDFIsAStream(File1) || UDFIsAStreamDir(File1) /*||
UDFIsADirectory(File1) || UDFHasAStreamDir(File1)*/)
try_return (RC = STATUS_ACCESS_DENIED);
#endif // UDF_ALLOW_LINKS_TO_STREAMS
// do we try to link to RootDir or Volume ?
if(!DirObject2) {
Dir2 = File1->ParentFile;
DirObject2 = FileObject1->RelatedFileObject;
} else
if(DirObject2->FsContext2 &&
(Fcb2 = ((PtrUDFCCB)(DirObject2->FsContext2))->Fcb)) {
Dir2 = ((PtrUDFCCB)(DirObject2->FsContext2))->Fcb->FileInfo;
} else {
try_return (RC = STATUS_INVALID_PARAMETER);
}
// check target dir
if(!Dir2) try_return (RC = STATUS_ACCESS_DENIED);
// Stream can't be a Dir or have Streams
if(UDFIsAStreamDir(Dir2)) {
try_return (RC = STATUS_ACCESS_DENIED);
/* if(UDFIsADirectory(File1) ||
UDFHasAStreamDir(File1)) {
BrutePoint();
try_return (RC = STATUS_ACCESS_DENIED);
}*/
}
/* if(UDFIsAStreamDir(Dir2))
try_return (RC = STATUS_ACCESS_DENIED);*/
RC = UDFPrepareForRenameMoveLink(Vcb, &AcquiredVcb, &AcquiredVcbEx,
&SingleDir,
&AcquiredDir1, &AcquiredFcb1,
Ccb1, File1,
Dir1, Dir2,
TRUE); // it is HLink operation
if(!NT_SUCCESS(RC))
try_return(RC);
// check if the source file is used
if(!DirObject2) {
// Make sure the name is of legal length.
if(PtrBuffer->FileNameLength > UDF_NAME_LEN*sizeof(WCHAR)) {
try_return(RC = STATUS_OBJECT_NAME_INVALID);
}
NewName.Length = NewName.MaximumLength = (USHORT)(PtrBuffer->FileNameLength);
NewName.Buffer = (PWCHAR)&(PtrBuffer->FileName);
} else {
// This name is by definition legal.
NewName = *((PUNICODE_STRING)&DirObject2->FileName);
}
ic = (Ccb1->CCBFlags & UDF_CCB_CASE_SENSETIVE) ? FALSE : TRUE;
AdPrint((" %ws ->\n %ws\n",
Fcb1->FCBName->ObjectName.Buffer,
NewName.Buffer));
RC = UDFHardLinkFile__(Vcb, ic, &Replace, &NewName, Dir1, Dir2, File1);
if(!NT_SUCCESS(RC)) try_return (RC);
// Update Parent Objects (mark 'em as modified)
if(Vcb->CompatFlags & UDF_VCB_IC_UPDATE_DIR_WRITE) {
if(DirObject2) {
DirObject2->Flags |= FO_FILE_MODIFIED;
if(!Replace)
DirObject2->Flags |= FO_FILE_SIZE_CHANGED;
}
}
// report changes
UDFNotifyFullReportChange( Vcb, File1,
FILE_NOTIFY_CHANGE_LAST_WRITE |
FILE_NOTIFY_CHANGE_LAST_ACCESS,
FILE_ACTION_MODIFIED );
RC = MyCloneUnicodeString(&LocalPath, (Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ?
&UDFGlobalData.UnicodeStrRoot :
&(Dir2->Fcb->FCBName->ObjectName));
if(!NT_SUCCESS(RC)) try_return (RC);
/* RC = MyAppendUnicodeStringToString(&LocalPath, (Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ? &(UDFGlobalData.UnicodeStrRoot) : &(Dir2->Fcb->FCBName->ObjectName));
if(!NT_SUCCESS(RC)) try_return (RC);*/
// if Dir2 is a RootDir, we shoud not append '\\' because
// it will be the 2nd '\\' character (RootDir's name is also '\\')
if(Dir2->ParentFile) {
RC = MyAppendUnicodeToString(&LocalPath, L"\\");
if(!NT_SUCCESS(RC)) try_return (RC);
}
RC = MyAppendUnicodeStringToStringTag(&LocalPath, &NewName, MEM_USHL_TAG);
if(!NT_SUCCESS(RC)) try_return (RC);
if(!Replace) {
/* UDFNotifyFullReportChange( Vcb, File2,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_ADDED );*/
FsRtlNotifyFullReportChange( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PSTRING)&LocalPath,
((Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ? 0 : Dir2->Fcb->FCBName->ObjectName.Length) + sizeof(WCHAR),
NULL,NULL,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_ACTION_ADDED,
NULL);
} else {
/* UDFNotifyFullReportChange( Vcb, File2,
FILE_NOTIFY_CHANGE_ATTRIBUTES |
FILE_NOTIFY_CHANGE_SIZE |
FILE_NOTIFY_CHANGE_LAST_WRITE |
FILE_NOTIFY_CHANGE_LAST_ACCESS |
FILE_NOTIFY_CHANGE_CREATION |
FILE_NOTIFY_CHANGE_EA,
FILE_ACTION_MODIFIED );*/
FsRtlNotifyFullReportChange( Vcb->NotifyIRPMutex, &(Vcb->NextNotifyIRP),
(PSTRING)&LocalPath,
((Dir2->Fcb->FCBFlags & UDF_FCB_ROOT_DIRECTORY) ? 0 : Dir2->Fcb->FCBName->ObjectName.Length) + sizeof(WCHAR),
NULL,NULL,
UDFIsADirectory(File1) ? FILE_NOTIFY_CHANGE_DIR_NAME : FILE_NOTIFY_CHANGE_FILE_NAME,
FILE_NOTIFY_CHANGE_ATTRIBUTES |
FILE_NOTIFY_CHANGE_SIZE |
FILE_NOTIFY_CHANGE_LAST_WRITE |
FILE_NOTIFY_CHANGE_LAST_ACCESS |
FILE_NOTIFY_CHANGE_CREATION |
FILE_NOTIFY_CHANGE_EA,
NULL);
}
RC = STATUS_SUCCESS;
try_exit: NOTHING;
} _SEH2_FINALLY {
if(AcquiredFcb1) {
UDF_CHECK_PAGING_IO_RESOURCE(Fcb1->NTRequiredFCB);
UDFReleaseResource(&(Fcb1->NTRequiredFCB->MainResource));
}
if(AcquiredDir1) {
UDF_CHECK_PAGING_IO_RESOURCE(Dir1->Fcb->NTRequiredFCB);
UDFReleaseResource(&(Dir1->Fcb->NTRequiredFCB->MainResource));
}
if(AcquiredVcb) {
if(AcquiredVcbEx)
UDFConvertExclusiveToSharedLite(&(Vcb->VCBResource));
} else {
// caller assumes Vcb to be acquired shared
BrutePoint();
UDFAcquireResourceShared(&(Vcb->VCBResource), TRUE);
}
if(LocalPath.Buffer) {
MyFreePool__(LocalPath.Buffer);
}
} _SEH2_END;
return RC;
} // end UDFHardLink()
#endif //UDF_ALLOW_HARD_LINKS
#endif //UDF_READ_ONLY_BUILD
Reference in a new issue View git blame Copy permalink