/* * COPYRIGHT: See COPYRIGHT.TXT * PROJECT: Ext2 File System Driver for WinNT/2K/XP * FILE: memory.c * PROGRAMMER: Matt Wu * HOMEPAGE: http://www.ext2fsd.com * UPDATE HISTORY: */ /* INCLUDES *****************************************************************/ #include "ext2fs.h" /* GLOBALS ***************************************************************/ extern PEXT2_GLOBAL Ext2Global; /* DEFINITIONS *************************************************************/ #ifdef ALLOC_PRAGMA #pragma alloc_text(PAGE, Ext2AllocateInode) #pragma alloc_text(PAGE, Ext2DestroyInode) #pragma alloc_text(PAGE, Ext2CheckBitmapConsistency) #pragma alloc_text(PAGE, Ext2CheckSetBlock) #pragma alloc_text(PAGE, Ext2InitializeVcb) #pragma alloc_text(PAGE, Ext2TearDownStream) #pragma alloc_text(PAGE, Ext2DestroyVcb) #pragma alloc_text(PAGE, Ext2SyncUninitializeCacheMap) #pragma alloc_text(PAGE, Ext2ReaperThread) #pragma alloc_text(PAGE, Ext2StartReaper) #pragma alloc_text(PAGE, Ext2StopReaper) #endif PEXT2_IRP_CONTEXT Ext2AllocateIrpContext (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp ) { PIO_STACK_LOCATION irpSp; PEXT2_IRP_CONTEXT IrpContext; ASSERT(DeviceObject != NULL); ASSERT(Irp != NULL); irpSp = IoGetCurrentIrpStackLocation(Irp); IrpContext = (PEXT2_IRP_CONTEXT) ( ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2IrpContextLookasideList))); if (IrpContext == NULL) { return NULL; } RtlZeroMemory(IrpContext, sizeof(EXT2_IRP_CONTEXT) ); IrpContext->Identifier.Type = EXT2ICX; IrpContext->Identifier.Size = sizeof(EXT2_IRP_CONTEXT); IrpContext->Irp = Irp; IrpContext->MajorFunction = irpSp->MajorFunction; IrpContext->MinorFunction = irpSp->MinorFunction; IrpContext->DeviceObject = DeviceObject; IrpContext->FileObject = irpSp->FileObject; if (NULL != IrpContext->FileObject) { IrpContext->Fcb = (PEXT2_FCB)IrpContext->FileObject->FsContext; IrpContext->Ccb = (PEXT2_CCB)IrpContext->FileObject->FsContext2; } if (IrpContext->FileObject != NULL) { IrpContext->RealDevice = IrpContext->FileObject->DeviceObject; } else if (IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL) { if (irpSp->Parameters.MountVolume.Vpb) { IrpContext->RealDevice = irpSp->Parameters.MountVolume.Vpb->RealDevice; } } if (IsFlagOn(irpSp->Flags, SL_WRITE_THROUGH)) { SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WRITE_THROUGH); } if (IsFlagOn(irpSp->Flags, SL_OVERRIDE_VERIFY_VOLUME)) { SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_VERIFY_READ); } if (IrpContext->MajorFunction == IRP_MJ_CLEANUP || IrpContext->MajorFunction == IRP_MJ_CLOSE || IrpContext->MajorFunction == IRP_MJ_SHUTDOWN || IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL || IrpContext->MajorFunction == IRP_MJ_PNP ) { if (IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL || IrpContext->MajorFunction == IRP_MJ_PNP) { if (IoGetCurrentIrpStackLocation(Irp)->FileObject == NULL || IoIsOperationSynchronous(Irp)) { SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); } } else { SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); } } else if (IoIsOperationSynchronous(Irp)) { SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); } IrpContext->IsTopLevel = (IoGetTopLevelIrp() == Irp); IrpContext->ExceptionInProgress = FALSE; INC_IRP_COUNT(IrpContext); return IrpContext; } VOID Ext2FreeIrpContext (IN PEXT2_IRP_CONTEXT IrpContext) { ASSERT(IrpContext != NULL); ASSERT((IrpContext->Identifier.Type == EXT2ICX) && (IrpContext->Identifier.Size == sizeof(EXT2_IRP_CONTEXT))); /* free the IrpContext to NonPagedList */ IrpContext->Identifier.Type = 0; IrpContext->Identifier.Size = 0; DEC_IRP_COUNT(IrpContext); ExFreeToNPagedLookasideList(&(Ext2Global->Ext2IrpContextLookasideList), IrpContext); } PEXT2_FCB Ext2AllocateFcb ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb ) { PEXT2_FCB Fcb; ASSERT(ExIsResourceAcquiredExclusiveLite(&Vcb->FcbLock)); Fcb = (PEXT2_FCB) ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2FcbLookasideList)); if (!Fcb) { return NULL; } RtlZeroMemory(Fcb, sizeof(EXT2_FCB)); Fcb->Identifier.Type = EXT2FCB; Fcb->Identifier.Size = sizeof(EXT2_FCB); #ifndef _WIN2K_TARGET_ ExInitializeFastMutex(&Fcb->Mutex); FsRtlSetupAdvancedHeader(&Fcb->Header, &Fcb->Mutex); #endif FsRtlInitializeOplock(&Fcb->Oplock); FsRtlInitializeFileLock ( &Fcb->FileLockAnchor, NULL, NULL ); Fcb->OpenHandleCount = 0; Fcb->ReferenceCount = 0; Fcb->Vcb = Vcb; Fcb->Inode = &Mcb->Inode; ASSERT(Mcb->Fcb == NULL); Ext2ReferMcb(Mcb); Fcb->Mcb = Mcb; Mcb->Fcb = Fcb; DEBUG(DL_RES, ("Ext2AllocateFcb: Fcb %p created: %wZ.\n", Fcb, &Fcb->Mcb->FullName)); RtlZeroMemory(&Fcb->Header, sizeof(FSRTL_COMMON_FCB_HEADER)); Fcb->Header.NodeTypeCode = (USHORT) EXT2FCB; Fcb->Header.NodeByteSize = sizeof(EXT2_FCB); Fcb->Header.IsFastIoPossible = FastIoIsNotPossible; Fcb->Header.Resource = &(Fcb->MainResource); Fcb->Header.PagingIoResource = &(Fcb->PagingIoResource); Fcb->Header.FileSize.QuadPart = Mcb->Inode.i_size; Fcb->Header.ValidDataLength.QuadPart = Mcb->Inode.i_size; Fcb->Header.AllocationSize.QuadPart = CEILING_ALIGNED(ULONGLONG, Fcb->Header.FileSize.QuadPart, (ULONGLONG)Vcb->BlockSize); Fcb->SectionObject.DataSectionObject = NULL; Fcb->SectionObject.SharedCacheMap = NULL; Fcb->SectionObject.ImageSectionObject = NULL; ExInitializeResourceLite(&(Fcb->MainResource)); ExInitializeResourceLite(&(Fcb->PagingIoResource)); Ext2InsertFcb(Vcb, Fcb); INC_MEM_COUNT(PS_FCB, Fcb, sizeof(EXT2_FCB)); return Fcb; } VOID Ext2UnlinkFcb(IN PEXT2_FCB Fcb) { PEXT2_VCB Vcb = Fcb->Vcb; PEXT2_MCB Mcb; ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); Mcb = Fcb->Mcb; DEBUG(DL_INF, ("Ext2FreeFcb: Fcb (%p) to be unlinked: %wZ.\n", Fcb, Mcb ? &Mcb->FullName : NULL)); if ((Mcb != NULL) && (Mcb->Identifier.Type == EXT2MCB) && (Mcb->Identifier.Size == sizeof(EXT2_MCB))) { ASSERT (Mcb->Fcb == Fcb); if (IsMcbSpecialFile(Mcb) || IsFileDeleted(Mcb)) { ASSERT(!IsRoot(Fcb)); Ext2RemoveMcb(Vcb, Mcb); Mcb->Fcb = NULL; Ext2UnlinkMcb(Vcb, Mcb); Ext2DerefMcb(Mcb); Ext2LinkHeadMcb(Vcb, Mcb); } else { Mcb->Fcb = NULL; Ext2DerefMcb(Mcb); } Fcb->Mcb = NULL; } ExReleaseResourceLite(&Vcb->McbLock); } VOID Ext2FreeFcb (IN PEXT2_FCB Fcb) { PEXT2_VCB Vcb = Fcb->Vcb; _SEH2_TRY { ASSERT((Fcb != NULL) && (Fcb->Identifier.Type == EXT2FCB) && (Fcb->Identifier.Size == sizeof(EXT2_FCB))); ASSERT(0 == Fcb->ReferenceCount); #ifndef _WIN2K_TARGET_ FsRtlTeardownPerStreamContexts(&Fcb->Header); #endif FsRtlUninitializeFileLock(&Fcb->FileLockAnchor); FsRtlUninitializeOplock(&Fcb->Oplock); ExDeleteResourceLite(&Fcb->MainResource); ExDeleteResourceLite(&Fcb->PagingIoResource); Fcb->Identifier.Type = 0; Fcb->Identifier.Size = 0; ExFreeToNPagedLookasideList(&(Ext2Global->Ext2FcbLookasideList), Fcb); DEC_MEM_COUNT(PS_FCB, Fcb, sizeof(EXT2_FCB)); if (0 == Ext2DerefXcb(&Vcb->ReferenceCount)) { if (!IsMounted(Vcb) || IsDispending(Vcb)) { Ext2CheckDismount(NULL, Vcb, FALSE); } } } _SEH2_FINALLY { } _SEH2_END; } VOID Ext2ReleaseFcb (IN PEXT2_FCB Fcb) { PEXT2_VCB Vcb = Fcb->Vcb; PEXT2_MCB Mcb; if (0 != Ext2DerefXcb(&Fcb->ReferenceCount)) return; ExAcquireResourceExclusiveLite(&Vcb->FcbLock, TRUE); ExAcquireResourceExclusiveLite(&Fcb->MainResource, TRUE); Mcb = Fcb->Mcb; RemoveEntryList(&Fcb->Next); if (IsFlagOn(Fcb->Flags, FCB_DELETE_PENDING) || NULL == Mcb || IsFileDeleted(Mcb)) { InsertHeadList(&Vcb->FcbList, &Fcb->Next); Fcb->TsDrop.QuadPart = 0; } else { InsertTailList(&Vcb->FcbList, &Fcb->Next); KeQuerySystemTime(&Fcb->TsDrop); } ExReleaseResourceLite(&Fcb->MainResource); ExReleaseResourceLite(&Vcb->FcbLock); if ((Vcb->FcbCount >> 6) > (ULONG)(Ext2Global->MaxDepth)) { KeSetEvent(&Ext2Global->FcbReaper.Wait, 0, FALSE); } } /* Insert Fcb to Vcb->FcbList queue, with Vcb->FcbLock Acquired. */ VOID Ext2InsertFcb(PEXT2_VCB Vcb, PEXT2_FCB Fcb) { ASSERT(ExIsResourceAcquiredExclusiveLite(&Vcb->FcbLock)); KeQuerySystemTime(&Fcb->TsDrop); Ext2ReferXcb(&Vcb->FcbCount); Ext2ReferXcb(&Vcb->ReferenceCount); InsertTailList(&Vcb->FcbList, &Fcb->Next); } PEXT2_CCB Ext2AllocateCcb (ULONG Flags, PEXT2_MCB SymLink) { PEXT2_CCB Ccb; Ccb = (PEXT2_CCB) (ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2CcbLookasideList))); if (!Ccb) { return NULL; } DEBUG(DL_RES, ( "ExtAllocateCcb: Ccb created: %ph.\n", Ccb)); RtlZeroMemory(Ccb, sizeof(EXT2_CCB)); Ccb->Identifier.Type = EXT2CCB; Ccb->Identifier.Size = sizeof(EXT2_CCB); Ccb->Flags = Flags; Ccb->SymLink = SymLink; if (SymLink) { ASSERT(SymLink->Refercount > 0); Ext2ReferMcb(SymLink); DEBUG(DL_INF, ( "ExtAllocateCcb: Ccb SymLink: %wZ.\n", &Ccb->SymLink->FullName)); } Ccb->DirectorySearchPattern.Length = 0; Ccb->DirectorySearchPattern.MaximumLength = 0; Ccb->DirectorySearchPattern.Buffer = 0; INC_MEM_COUNT(PS_CCB, Ccb, sizeof(EXT2_CCB)); return Ccb; } VOID Ext2FreeCcb (IN PEXT2_VCB Vcb, IN PEXT2_CCB Ccb) { ASSERT(Ccb != NULL); ASSERT((Ccb->Identifier.Type == EXT2CCB) && (Ccb->Identifier.Size == sizeof(EXT2_CCB))); DEBUG(DL_RES, ( "Ext2FreeCcb: Ccb = %ph.\n", Ccb)); if (Ccb->SymLink) { DEBUG(DL_INF, ( "Ext2FreeCcb: Ccb SymLink: %wZ.\n", &Ccb->SymLink->FullName)); if (IsFileDeleted(Ccb->SymLink->Target)) { Ext2UnlinkMcb(Vcb, Ccb->SymLink); Ext2DerefMcb(Ccb->SymLink); Ext2LinkHeadMcb(Vcb, Ccb->SymLink); } else { Ext2DerefMcb(Ccb->SymLink); } } if (Ccb->DirectorySearchPattern.Buffer != NULL) { DEC_MEM_COUNT(PS_DIR_PATTERN, Ccb->DirectorySearchPattern.Buffer, Ccb->DirectorySearchPattern.MaximumLength ); Ext2FreePool(Ccb->DirectorySearchPattern.Buffer, EXT2_DIRSP_MAGIC); } ExFreeToNPagedLookasideList(&(Ext2Global->Ext2CcbLookasideList), Ccb); DEC_MEM_COUNT(PS_CCB, Ccb, sizeof(EXT2_CCB)); } PEXT2_INODE Ext2AllocateInode (PEXT2_VCB Vcb) { PVOID inode = NULL; inode = ExAllocateFromNPagedLookasideList( &(Vcb->InodeLookasideList)); if (!inode) { return NULL; } RtlZeroMemory(inode, INODE_SIZE); DEBUG(DL_INF, ("ExtAllocateInode: Inode created: %ph.\n", inode)); INC_MEM_COUNT(PS_EXT2_INODE, inode, INODE_SIZE); return inode; } VOID Ext2DestroyInode (IN PEXT2_VCB Vcb, IN PEXT2_INODE inode) { ASSERT(inode != NULL); DEBUG(DL_INF, ("Ext2FreeInode: Inode = %ph.\n", inode)); ExFreeToNPagedLookasideList(&(Vcb->InodeLookasideList), inode); DEC_MEM_COUNT(PS_EXT2_INODE, inode, INODE_SIZE); } struct dentry * Ext2AllocateEntry() { struct dentry *de; de = (struct dentry *)ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2DentryLookasideList)); if (!de) { return NULL; } RtlZeroMemory(de, sizeof(struct dentry)); INC_MEM_COUNT(PS_DENTRY, de, sizeof(struct dentry)); return de; } VOID Ext2FreeEntry (IN struct dentry *de) { ASSERT(de != NULL); if (de->d_name.name) ExFreePool(de->d_name.name); ExFreeToNPagedLookasideList(&(Ext2Global->Ext2DentryLookasideList), de); DEC_MEM_COUNT(PS_DENTRY, de, sizeof(struct dentry)); } struct dentry *Ext2BuildEntry(PEXT2_VCB Vcb, PEXT2_MCB Dcb, PUNICODE_STRING FileName) { OEM_STRING Oem = { 0 }; struct dentry *de = NULL; NTSTATUS Status = STATUS_INSUFFICIENT_RESOURCES; _SEH2_TRY { de = Ext2AllocateEntry(); if (!de) { DEBUG(DL_ERR, ("Ext2BuildEntry: failed to allocate dentry.\n")); _SEH2_LEAVE; } de->d_sb = &Vcb->sb; if (Dcb) de->d_parent = Dcb->de; Oem.MaximumLength = (USHORT)Ext2UnicodeToOEMSize(Vcb, FileName) + 1; Oem.Buffer = ExAllocatePool(PagedPool, Oem.MaximumLength); if (!Oem.Buffer) { DEBUG(DL_ERR, ( "Ex2BuildEntry: failed to allocate OEM name.\n")); _SEH2_LEAVE; } de->d_name.name = Oem.Buffer; RtlZeroMemory(Oem.Buffer, Oem.MaximumLength); Status = Ext2UnicodeToOEM(Vcb, &Oem, FileName); if (!NT_SUCCESS(Status)) { DEBUG(DL_CP, ("Ext2BuildEntry: failed to convert %S to OEM.\n", FileName->Buffer)); _SEH2_LEAVE; } de->d_name.len = Oem.Length; } _SEH2_FINALLY { if (!NT_SUCCESS(Status)) { if (de) Ext2FreeEntry(de); } } _SEH2_END; return de; } PEXT2_EXTENT Ext2AllocateExtent () { PEXT2_EXTENT Extent; Extent = (PEXT2_EXTENT)ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2ExtLookasideList)); if (!Extent) { return NULL; } RtlZeroMemory(Extent, sizeof(EXT2_EXTENT)); INC_MEM_COUNT(PS_EXTENT, Extent, sizeof(EXT2_EXTENT)); return Extent; } VOID Ext2FreeExtent (IN PEXT2_EXTENT Extent) { ASSERT(Extent != NULL); ExFreeToNPagedLookasideList(&(Ext2Global->Ext2ExtLookasideList), Extent); DEC_MEM_COUNT(PS_EXTENT, Extent, sizeof(EXT2_EXTENT)); } ULONG Ext2CountExtents(IN PEXT2_EXTENT Chain) { ULONG count = 0; PEXT2_EXTENT List = Chain; while (List) { count += 1; List = List->Next; } return count; } VOID Ext2JointExtents( IN PEXT2_EXTENT Chain, IN PEXT2_EXTENT Extent ) { #ifndef __REACTOS__ ULONG count = 0; #endif PEXT2_EXTENT List = Chain; while (List->Next) { List = List->Next; } List->Next = Extent; } VOID Ext2DestroyExtentChain(IN PEXT2_EXTENT Chain) { PEXT2_EXTENT Extent = NULL, List = Chain; while (List) { Extent = List->Next; Ext2FreeExtent(List); List = Extent; } } BOOLEAN Ext2ListExtents(PLARGE_MCB Extents) { if (FsRtlNumberOfRunsInLargeMcb(Extents) != 0) { LONGLONG DirtyVba; LONGLONG DirtyLba; LONGLONG DirtyLength; int i, n = 0; for (i = 0; FsRtlGetNextLargeMcbEntry( Extents, i, &DirtyVba, &DirtyLba, &DirtyLength); i++) { if (DirtyVba > 0 && DirtyLba != -1) { DEBUG(DL_EXT, ("Vba:%I64xh Lba:%I64xh Len:%I64xh.\n", DirtyVba, DirtyLba, DirtyLength)); n++; } } return n ? TRUE : FALSE; } return FALSE; } VOID Ext2CheckExtent( PLARGE_MCB Zone, LONGLONG Vbn, LONGLONG Lbn, LONGLONG Length, BOOLEAN bAdded ) { #if EXT2_DEBUG LONGLONG DirtyLbn; LONGLONG DirtyLen; LONGLONG RunStart; LONGLONG RunLength; ULONG Index; BOOLEAN bFound = FALSE; bFound = FsRtlLookupLargeMcbEntry( Zone, Vbn, &DirtyLbn, &DirtyLen, &RunStart, &RunLength, &Index ); if (!bAdded && (!bFound || DirtyLbn == -1)) { return; } if ( !bFound || (DirtyLbn == -1) || (DirtyLbn != Lbn) || (DirtyLen < Length)) { DbgBreak(); for (Index = 0; TRUE; Index++) { if (!FsRtlGetNextLargeMcbEntry( Zone, Index, &Vbn, &Lbn, &Length)) { break; } DEBUG(DL_EXT, ("Index = %xh Vbn = %I64xh Lbn = %I64xh Len = %I64xh\n", Index, Vbn, Lbn, Length )); } } #endif } VOID Ext2ClearAllExtents(PLARGE_MCB Zone) { _SEH2_TRY { FsRtlTruncateLargeMcb(Zone, (LONGLONG)0); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); } _SEH2_END; } BOOLEAN Ext2AddVcbExtent ( IN PEXT2_VCB Vcb, IN LONGLONG Vbn, IN LONGLONG Length ) { ULONG TriedTimes = 0; LONGLONG Offset = 0; BOOLEAN rc = FALSE; Offset = Vbn & (~(Vcb->IoUnitSize - 1)); Length = (Vbn - Offset + Length + Vcb->IoUnitSize - 1) & ~(Vcb->IoUnitSize - 1); ASSERT ((Offset & (Vcb->IoUnitSize - 1)) == 0); ASSERT ((Length & (Vcb->IoUnitSize - 1)) == 0); Offset = (Offset >> Vcb->IoUnitBits) + 1; Length = (Length >> Vcb->IoUnitBits); Again: _SEH2_TRY { rc = FsRtlAddLargeMcbEntry( &Vcb->Extents, Offset, Offset, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2AddVcbExtent: Vbn=%I64xh Length=%I64xh," " rc=%d Runs=%u\n", Offset, Length, rc, FsRtlNumberOfRunsInLargeMcb(&Vcb->Extents))); if (rc) { Ext2CheckExtent(&Vcb->Extents, Offset, Offset, Length, TRUE); } return rc; } BOOLEAN Ext2RemoveVcbExtent ( IN PEXT2_VCB Vcb, IN LONGLONG Vbn, IN LONGLONG Length ) { ULONG TriedTimes = 0; LONGLONG Offset = 0; BOOLEAN rc = TRUE; Offset = Vbn & (~(Vcb->IoUnitSize - 1)); Length = (Length + Vbn - Offset + Vcb->IoUnitSize - 1) & (~(Vcb->IoUnitSize - 1)); ASSERT ((Offset & (Vcb->IoUnitSize - 1)) == 0); ASSERT ((Length & (Vcb->IoUnitSize - 1)) == 0); Offset = (Offset >> Vcb->IoUnitBits) + 1; Length = (Length >> Vcb->IoUnitBits); Again: _SEH2_TRY { FsRtlRemoveLargeMcbEntry( &Vcb->Extents, Offset, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2RemoveVcbExtent: Vbn=%I64xh Length=%I64xh Runs=%u\n", Offset, Length, FsRtlNumberOfRunsInLargeMcb(&Vcb->Extents))); if (rc) { Ext2CheckExtent(&Vcb->Extents, Offset, 0, Length, FALSE); } return rc; } BOOLEAN Ext2LookupVcbExtent ( IN PEXT2_VCB Vcb, IN LONGLONG Vbn, OUT PLONGLONG Lbn, OUT PLONGLONG Length ) { LONGLONG offset; BOOLEAN rc; offset = Vbn & (~(Vcb->IoUnitSize - 1)); ASSERT ((offset & (Vcb->IoUnitSize - 1)) == 0); offset = (offset >> Vcb->IoUnitBits) + 1; rc = FsRtlLookupLargeMcbEntry( &(Vcb->Extents), offset, Lbn, Length, NULL, NULL, NULL ); if (rc) { if (Lbn && ((*Lbn) != -1)) { ASSERT((*Lbn) > 0); (*Lbn) = (((*Lbn) - 1) << Vcb->IoUnitBits); (*Lbn) += ((Vbn) & (Vcb->IoUnitSize - 1)); } if (Length && *Length) { (*Length) <<= Vcb->IoUnitBits; (*Length) -= ((Vbn) & (Vcb->IoUnitSize - 1)); } } return rc; } BOOLEAN Ext2AddMcbExtent ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN LONGLONG Vbn, IN LONGLONG Lbn, IN LONGLONG Length ) { ULONG TriedTimes = 0; LONGLONG Base = 0; UCHAR Bits = 0; BOOLEAN rc = FALSE; Base = (LONGLONG)BLOCK_SIZE; Bits = (UCHAR)BLOCK_BITS; ASSERT ((Vbn & (Base - 1)) == 0); ASSERT ((Lbn & (Base - 1)) == 0); ASSERT ((Length & (Base - 1)) == 0); Vbn = (Vbn >> Bits) + 1; Lbn = (Lbn >> Bits) + 1; Length = (Length >> Bits); Again: _SEH2_TRY { rc = FsRtlAddLargeMcbEntry( &Mcb->Extents, Vbn, Lbn, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2AddMcbExtent: Vbn=%I64xh Lbn=%I64xh Length=%I64xh," " rc=%d Runs=%u\n", Vbn, Lbn, Length, rc, FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents))); if (rc) { Ext2CheckExtent(&Mcb->Extents, Vbn, Lbn, Length, TRUE); } return rc; } BOOLEAN Ext2RemoveMcbExtent ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN LONGLONG Vbn, IN LONGLONG Length ) { ULONG TriedTimes = 0; LONGLONG Base = 0; UCHAR Bits = 0; BOOLEAN rc = TRUE; Base = (LONGLONG)BLOCK_SIZE; Bits = (UCHAR)BLOCK_BITS; ASSERT ((Vbn & (Base - 1)) == 0); ASSERT ((Length & (Base - 1)) == 0); Vbn = (Vbn >> Bits) + 1; Length = (Length >> Bits); Again: _SEH2_TRY { FsRtlRemoveLargeMcbEntry( &Mcb->Extents, Vbn, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2RemoveMcbExtent: Vbn=%I64xh Length=%I64xh Runs=%u\n", Vbn, Length, FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents))); if (rc) { Ext2CheckExtent(&Mcb->Extents, Vbn, 0, Length, FALSE); } return rc; } BOOLEAN Ext2LookupMcbExtent ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN LONGLONG Vbn, OUT PLONGLONG Lbn, OUT PLONGLONG Length ) { LONGLONG offset; BOOLEAN rc; offset = Vbn & (~((LONGLONG)BLOCK_SIZE - 1)); ASSERT ((offset & (BLOCK_SIZE - 1)) == 0); offset = (offset >> BLOCK_BITS) + 1; rc = FsRtlLookupLargeMcbEntry( &(Mcb->Extents), offset, Lbn, Length, NULL, NULL, NULL ); if (rc) { if (Lbn && ((*Lbn) != -1)) { ASSERT((*Lbn) > 0); (*Lbn) = (((*Lbn) - 1) << BLOCK_BITS); (*Lbn) += ((Vbn) & ((LONGLONG)BLOCK_SIZE - 1)); } if (Length && *Length) { (*Length) <<= BLOCK_BITS; (*Length) -= ((Vbn) & ((LONGLONG)BLOCK_SIZE - 1)); } } return rc; } BOOLEAN Ext2AddMcbMetaExts ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONG Block, IN ULONG Length ) { ULONG TriedTimes = 0; LONGLONG Lbn = Block + 1; BOOLEAN rc = TRUE; Again: _SEH2_TRY { rc = FsRtlAddLargeMcbEntry( &Mcb->MetaExts, Lbn, Lbn, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2AddMcbMetaExts: Block: %xh-%xh rc=%d Runs=%u\n", Block, Length, rc, FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts))); if (rc) { Ext2CheckExtent(&Mcb->MetaExts, Lbn, Lbn, Length, TRUE); } return rc; } BOOLEAN Ext2RemoveMcbMetaExts ( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONG Block, IN ULONG Length ) { ULONG TriedTimes = 0; LONGLONG Lbn = Block + 1; BOOLEAN rc = TRUE; Again: _SEH2_TRY { FsRtlRemoveLargeMcbEntry( &Mcb->MetaExts, Lbn, Length ); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { DbgBreak(); rc = FALSE; } _SEH2_END; if (!rc && ++TriedTimes < 10) { Ext2Sleep(TriedTimes * 100); goto Again; } DEBUG(DL_EXT, ("Ext2RemoveMcbMetaExts: Block: %xh-%xhh Runs=%u\n", Block, Length, FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts))); if (rc) { Ext2CheckExtent(&Mcb->MetaExts, Lbn, 0, Length, FALSE); } return rc; } BOOLEAN Ext2AddBlockExtent( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONG Start, IN ULONG Block, IN ULONG Number ) { LONGLONG Vbn = 0; LONGLONG Lbn = 0; LONGLONG Length = 0; Vbn = ((LONGLONG) Start) << BLOCK_BITS; Lbn = ((LONGLONG) Block) << BLOCK_BITS; Length = ((LONGLONG)Number << BLOCK_BITS); if (Mcb) { #if EXT2_DEBUG ULONG _block = 0, _mapped = 0; BOOLEAN _rc = Ext2LookupBlockExtent(Vcb, Mcb, Start, &_block, &_mapped); if (_rc && _block != 0 && (_block != Block)) { DbgBreak(); } #endif return Ext2AddMcbExtent(Vcb, Mcb, Vbn, Lbn, Length); } ASSERT(Start == Block); return Ext2AddVcbExtent(Vcb, Vbn, Length); } BOOLEAN Ext2LookupBlockExtent( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONG Start, IN PULONG Block, IN PULONG Mapped ) { LONGLONG Vbn = 0; LONGLONG Lbn = 0; LONGLONG Length = 0; BOOLEAN rc = FALSE; Vbn = ((LONGLONG) Start) << BLOCK_BITS; if (Mcb) { rc = Ext2LookupMcbExtent(Vcb, Mcb, Vbn, &Lbn, &Length); } else { rc = Ext2LookupVcbExtent(Vcb, Vbn, &Lbn, &Length); } if (rc) { *Mapped = (ULONG)(Length >> BLOCK_BITS); if (Lbn != -1 && Length > 0) { *Block = (ULONG)(Lbn >> BLOCK_BITS); } else { *Block = 0; } } return rc; } BOOLEAN Ext2RemoveBlockExtent( IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONG Start, IN ULONG Number ) { LONGLONG Vbn = 0; LONGLONG Length = 0; BOOLEAN rc; Vbn = ((LONGLONG) Start) << BLOCK_BITS; Length = ((LONGLONG)Number << BLOCK_BITS); if (Mcb) { rc = Ext2RemoveMcbExtent(Vcb, Mcb, Vbn, Length); } else { rc = Ext2RemoveVcbExtent(Vcb, Vbn, Length); } return rc; } NTSTATUS Ext2InitializeZone( IN PEXT2_IRP_CONTEXT IrpContext, IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb ) { NTSTATUS Status = STATUS_SUCCESS; ULONG Start = 0; ULONG End; ULONG Block; ULONG Mapped; Ext2ClearAllExtents(&Mcb->Extents); Ext2ClearAllExtents(&Mcb->MetaExts); ASSERT(Mcb != NULL); End = (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS); while (Start < End) { Block = Mapped = 0; /* mapping file offset to ext2 block */ if (INODE_HAS_EXTENT(&Mcb->Inode)) { Status = Ext2MapExtent( IrpContext, Vcb, Mcb, Start, FALSE, &Block, &Mapped ); } else { Status = Ext2MapIndirect( IrpContext, Vcb, Mcb, Start, FALSE, &Block, &Mapped ); } if (!NT_SUCCESS(Status)) { goto errorout; } /* skip wrong blocks, in case wrongly treating symlink target names as blocks, silly */ if (Block >= TOTAL_BLOCKS) { Block = 0; } if (Block) { if (!Ext2AddBlockExtent(Vcb, Mcb, Start, Block, Mapped)) { DbgBreak(); ClearFlag(Mcb->Flags, MCB_ZONE_INITED); Ext2ClearAllExtents(&Mcb->Extents); Status = STATUS_INSUFFICIENT_RESOURCES; goto errorout; } DEBUG(DL_MAP, ("Ext2InitializeZone %wZ: Block = %xh Mapped = %xh\n", &Mcb->FullName, Block, Mapped)); } /* Mapped is total number of continous blocks or NULL blocks */ Start += Mapped; } /* set mcb zone as initialized */ SetLongFlag(Mcb->Flags, MCB_ZONE_INITED); errorout: if (!IsZoneInited(Mcb)) { Ext2ClearAllExtents(&Mcb->Extents); Ext2ClearAllExtents(&Mcb->MetaExts); } return Status; } NTSTATUS Ext2BuildExtents( IN PEXT2_IRP_CONTEXT IrpContext, IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb, IN ULONGLONG Offset, IN ULONG Size, IN BOOLEAN bAlloc, OUT PEXT2_EXTENT * Chain ) { ULONG Start, End; ULONG Total = 0; LONGLONG Lba = 0; NTSTATUS Status = STATUS_SUCCESS; PEXT2_EXTENT Extent = NULL; PEXT2_EXTENT List = *Chain = NULL; if (!IsZoneInited(Mcb)) { Status = Ext2InitializeZone(IrpContext, Vcb, Mcb); if (!NT_SUCCESS(Status)) { DbgBreak(); } } if ((IrpContext && IrpContext->Irp) && ((IrpContext->Irp->Flags & IRP_NOCACHE) || (IrpContext->Irp->Flags & IRP_PAGING_IO))) { Size = (Size + SECTOR_SIZE - 1) & (~(SECTOR_SIZE - 1)); } Start = (ULONG)(Offset >> BLOCK_BITS); End = (ULONG)((Size + Offset + BLOCK_SIZE - 1) >> BLOCK_BITS); if (End > (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS) ) { End = (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS); } while (Size > 0 && Start < End) { ULONG Mapped = 0; ULONG Length = 0; ULONG Block = 0; BOOLEAN rc = FALSE; /* try to map file offset to ext2 block upon Extents cache */ if (IsZoneInited(Mcb)) { rc = Ext2LookupBlockExtent( Vcb, Mcb, Start, &Block, &Mapped); if (!rc) { /* we likely get a sparse file here */ Mapped = 1; Block = 0; } } /* try to BlockMap in case failed to access Extents cache */ if (!IsZoneInited(Mcb) || (bAlloc && Block == 0)) { Status = Ext2BlockMap( IrpContext, Vcb, Mcb, Start, bAlloc, &Block, &Mapped ); if (!NT_SUCCESS(Status)) { break; } /* skip wrong blocks, in case wrongly treating symlink target names as blocks, silly */ if (Block >= TOTAL_BLOCKS) { Block = 0; } /* add new allocated blocks to Mcb zone */ if (IsZoneInited(Mcb) && Block) { if (!Ext2AddBlockExtent(Vcb, Mcb, Start, Block, Mapped)) { DbgBreak(); ClearFlag(Mcb->Flags, MCB_ZONE_INITED); Ext2ClearAllExtents(&Mcb->Extents); } } } /* calculate i/o extent */ Lba = ((LONGLONG)Block << BLOCK_BITS) + Offset - ((LONGLONG)Start << BLOCK_BITS); Length = (ULONG)(((LONGLONG)(Start + Mapped) << BLOCK_BITS) - Offset); if (Length > Size) { Length = Size; } if (0 == Length) { DbgBreak(); break; } Start += Mapped; Offset = (ULONGLONG)Start << BLOCK_BITS; if (Block != 0) { if (List && List->Lba + List->Length == Lba) { /* it's continuous upon previous Extent */ List->Length += Length; } else { /* have to allocate a new Extent */ Extent = Ext2AllocateExtent(); if (!Extent) { Status = STATUS_INSUFFICIENT_RESOURCES; DbgBreak(); break; } Extent->Lba = Lba; Extent->Length = Length; Extent->Offset = Total; /* insert new Extent to chain */ if (List) { List->Next = Extent; List = Extent; } else { *Chain = List = Extent; } } } else { if (bAlloc) { DbgBreak(); } } Total += Length; Size -= Length; } return Status; } BOOLEAN Ext2BuildName( IN OUT PUNICODE_STRING Target, IN PUNICODE_STRING File, IN PUNICODE_STRING Parent ) { USHORT Length = 0; USHORT ParentLen = 0; BOOLEAN bBackslash = TRUE; /* free the original buffer */ if (Target->Buffer) { DEC_MEM_COUNT(PS_MCB_NAME, Target->Buffer, Target->MaximumLength); Ext2FreePool(Target->Buffer, EXT2_FNAME_MAGIC); Target->Length = Target->MaximumLength = 0; } /* check the parent directory's name and backslash */ if (Parent && Parent->Buffer && Parent->Length > 0) { ParentLen = Parent->Length / sizeof(WCHAR); if (Parent->Buffer[ParentLen - 1] == L'\\') { bBackslash = FALSE; } } if (Parent == NULL || File->Buffer[0] == L'\\') { /* must be root inode */ ASSERT(ParentLen == 0); bBackslash = FALSE; } /* allocate and initialize new name buffer */ Length = File->Length; Length += (ParentLen + (bBackslash ? 1 : 0)) * sizeof(WCHAR); Target->Buffer = Ext2AllocatePool( PagedPool, Length + 2, EXT2_FNAME_MAGIC ); if (!Target->Buffer) { DEBUG(DL_ERR, ( "Ex2BuildName: failed to allocate name bufer.\n")); return FALSE; } RtlZeroMemory(Target->Buffer, Length + 2); if (ParentLen) { RtlCopyMemory(&Target->Buffer[0], Parent->Buffer, ParentLen * sizeof(WCHAR)); } if (bBackslash) { Target->Buffer[ParentLen++] = L'\\'; } RtlCopyMemory( &Target->Buffer[ParentLen], File->Buffer, File->Length); INC_MEM_COUNT(PS_MCB_NAME, Target->Buffer, Length + 2); Target->Length = Length; Target->MaximumLength = Length + 2; return TRUE; } PEXT2_MCB Ext2AllocateMcb ( IN PEXT2_VCB Vcb, IN PUNICODE_STRING FileName, IN PUNICODE_STRING Parent, IN ULONG FileAttr ) { PEXT2_MCB Mcb = NULL; NTSTATUS Status = STATUS_SUCCESS; /* need wake the reaper thread if there are many Mcb allocated */ if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 4)) { KeSetEvent(&Ext2Global->McbReaper.Wait, 0, FALSE); } /* allocate Mcb from LookasideList */ Mcb = (PEXT2_MCB) (ExAllocateFromNPagedLookasideList( &(Ext2Global->Ext2McbLookasideList))); if (Mcb == NULL) { return NULL; } /* initialize Mcb header */ RtlZeroMemory(Mcb, sizeof(EXT2_MCB)); Mcb->Identifier.Type = EXT2MCB; Mcb->Identifier.Size = sizeof(EXT2_MCB); Mcb->FileAttr = FileAttr; Mcb->Inode.i_priv = (PVOID)Mcb; Mcb->Inode.i_sb = &Vcb->sb; /* initialize Mcb names */ if (FileName) { #if EXT2_DEBUG if ( FileName->Length == 2 && FileName->Buffer[0] == L'\\') { DEBUG(DL_RES, ( "Ext2AllocateMcb: Root Mcb is to be created !\n")); } if ( FileName->Length == 2 && FileName->Buffer[0] == L'.') { DbgBreak(); } if ( FileName->Length == 4 && FileName->Buffer[0] == L'.' && FileName->Buffer[1] == L'.' ) { DbgBreak(); } #endif if (( FileName->Length >= 4 && FileName->Buffer[0] == L'.') && ((FileName->Length == 4 && FileName->Buffer[1] != L'.') || FileName->Length >= 6 )) { SetFlag(Mcb->FileAttr, FILE_ATTRIBUTE_HIDDEN); } if (!Ext2BuildName(&Mcb->ShortName, FileName, NULL)) { goto errorout; } if (!Ext2BuildName(&Mcb->FullName, FileName, Parent)) { goto errorout; } } /* initialize Mcb Extents, it will raise an expcetion if failed */ _SEH2_TRY { FsRtlInitializeLargeMcb(&(Mcb->Extents), NonPagedPool); FsRtlInitializeLargeMcb(&(Mcb->MetaExts), NonPagedPool); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { Status = STATUS_INSUFFICIENT_RESOURCES; DbgBreak(); } _SEH2_END; if (!NT_SUCCESS(Status)) { goto errorout; } INC_MEM_COUNT(PS_MCB, Mcb, sizeof(EXT2_MCB)); DEBUG(DL_INF, ( "Ext2AllocateMcb: Mcb %wZ created.\n", &Mcb->FullName)); return Mcb; errorout: if (Mcb) { if (Mcb->ShortName.Buffer) { DEC_MEM_COUNT(PS_MCB_NAME, Mcb->ShortName.Buffer, Mcb->ShortName.MaximumLength); Ext2FreePool(Mcb->ShortName.Buffer, EXT2_FNAME_MAGIC); } if (Mcb->FullName.Buffer) { DEC_MEM_COUNT(PS_MCB_NAME, Mcb->FullName.Buffer, Mcb->FullName.MaximumLength); Ext2FreePool(Mcb->FullName.Buffer, EXT2_FNAME_MAGIC); } ExFreeToNPagedLookasideList(&(Ext2Global->Ext2McbLookasideList), Mcb); } return NULL; } VOID Ext2FreeMcb (IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb) { #ifndef __REACTOS__ PEXT2_MCB Parent = Mcb->Parent; #endif ASSERT(Mcb != NULL); ASSERT((Mcb->Identifier.Type == EXT2MCB) && (Mcb->Identifier.Size == sizeof(EXT2_MCB))); if ((Mcb->Identifier.Type != EXT2MCB) || (Mcb->Identifier.Size != sizeof(EXT2_MCB))) { return; } DEBUG(DL_INF, ( "Ext2FreeMcb: Mcb %wZ will be freed.\n", &Mcb->FullName)); if (IsMcbSymLink(Mcb) && Mcb->Target) { Ext2DerefMcb(Mcb->Target); } if (FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents)) { DEBUG(DL_EXT, ("List data extents for: %wZ\n", &Mcb->FullName)); Ext2ListExtents(&Mcb->Extents); } FsRtlUninitializeLargeMcb(&(Mcb->Extents)); if (FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts)) { DEBUG(DL_EXT, ("List meta extents for: %wZ\n", &Mcb->FullName)); Ext2ListExtents(&Mcb->MetaExts); } FsRtlUninitializeLargeMcb(&(Mcb->MetaExts)); ClearLongFlag(Mcb->Flags, MCB_ZONE_INITED); if (Mcb->ShortName.Buffer) { DEC_MEM_COUNT(PS_MCB_NAME, Mcb->ShortName.Buffer, Mcb->ShortName.MaximumLength); Ext2FreePool(Mcb->ShortName.Buffer, EXT2_FNAME_MAGIC); } if (Mcb->FullName.Buffer) { DEC_MEM_COUNT(PS_MCB_NAME, Mcb->FullName.Buffer, Mcb->FullName.MaximumLength); Ext2FreePool(Mcb->FullName.Buffer, EXT2_FNAME_MAGIC); } /* free dentry */ if (Mcb->de) { Ext2FreeEntry(Mcb->de); } Mcb->Identifier.Type = 0; Mcb->Identifier.Size = 0; ExFreeToNPagedLookasideList(&(Ext2Global->Ext2McbLookasideList), Mcb); DEC_MEM_COUNT(PS_MCB, Mcb, sizeof(EXT2_MCB)); } PEXT2_MCB Ext2SearchMcb( PEXT2_VCB Vcb, PEXT2_MCB Parent, PUNICODE_STRING FileName ) { BOOLEAN LockAcquired = FALSE; PEXT2_MCB Mcb = NULL; _SEH2_TRY { ExAcquireResourceSharedLite(&Vcb->McbLock, TRUE); LockAcquired = TRUE; Mcb = Ext2SearchMcbWithoutLock(Parent, FileName); } _SEH2_FINALLY { if (LockAcquired) { ExReleaseResourceLite(&Vcb->McbLock); } } _SEH2_END; return Mcb; } PEXT2_MCB Ext2SearchMcbWithoutLock( PEXT2_MCB Parent, PUNICODE_STRING FileName ) { PEXT2_MCB TmpMcb = NULL; DEBUG(DL_RES, ("Ext2SearchMcb: %wZ\n", FileName)); _SEH2_TRY { Ext2ReferMcb(Parent); if (Ext2IsDot(FileName)) { TmpMcb = Parent; Ext2ReferMcb(Parent); _SEH2_LEAVE; } if (Ext2IsDotDot(FileName)) { if (IsMcbRoot(Parent)) { TmpMcb = Parent; } else { TmpMcb = Parent->Parent; } if (TmpMcb) { Ext2ReferMcb(TmpMcb); } _SEH2_LEAVE; } if (IsMcbSymLink(Parent)) { if (Parent->Target) { TmpMcb = Parent->Target->Child; ASSERT(!IsMcbSymLink(Parent->Target)); } else { TmpMcb = NULL; _SEH2_LEAVE; } } else { TmpMcb = Parent->Child; } while (TmpMcb) { if (!RtlCompareUnicodeString( &(TmpMcb->ShortName), FileName, TRUE )) { Ext2ReferMcb(TmpMcb); break; } TmpMcb = TmpMcb->Next; } } _SEH2_FINALLY { Ext2DerefMcb(Parent); } _SEH2_END; return TmpMcb; } VOID Ext2InsertMcb ( PEXT2_VCB Vcb, PEXT2_MCB Parent, PEXT2_MCB Child ) { BOOLEAN LockAcquired = FALSE; PEXT2_MCB Mcb = NULL; _SEH2_TRY { ExAcquireResourceExclusiveLite( &Vcb->McbLock, TRUE ); LockAcquired = TRUE; /* use it's target if it's a symlink */ if (IsMcbSymLink(Parent)) { Parent = Parent->Target; ASSERT(!IsMcbSymLink(Parent)); } Mcb = Parent->Child; while (Mcb) { if (Mcb == Child) { break; } Mcb = Mcb->Next; } if (Mcb) { /* already attached in the list */ DEBUG(DL_ERR, ( "Ext2InsertMcb: Child Mcb is alreay attached.\n")); if (!IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { SetLongFlag(Child->Flags, MCB_ENTRY_TREE); DEBUG(DL_ERR, ( "Ext2InsertMcb: Child Mcb's flag isn't set.\n")); } DbgBreak(); } else { /* insert this Mcb into the head */ Child->Next = Parent->Child; Parent->Child = Child; Child->Parent = Parent; Child->de->d_parent = Parent->de; Ext2ReferMcb(Parent); SetLongFlag(Child->Flags, MCB_ENTRY_TREE); } } _SEH2_FINALLY { if (LockAcquired) { ExReleaseResourceLite(&Vcb->McbLock); } } _SEH2_END; } BOOLEAN Ext2RemoveMcb ( PEXT2_VCB Vcb, PEXT2_MCB Mcb ) { PEXT2_MCB TmpMcb = NULL; BOOLEAN LockAcquired = FALSE; BOOLEAN bLinked = FALSE; _SEH2_TRY { ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); LockAcquired = TRUE; if (Mcb->Parent) { if (Mcb->Parent->Child == Mcb) { Mcb->Parent->Child = Mcb->Next; bLinked = TRUE; } else { TmpMcb = Mcb->Parent->Child; while (TmpMcb && TmpMcb->Next != Mcb) { TmpMcb = TmpMcb->Next; } if (TmpMcb) { TmpMcb->Next = Mcb->Next; bLinked = TRUE; } else { /* we got errors: link broken */ } } if (bLinked) { if (IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { DEBUG(DL_RES, ("Mcb %p %wZ removed from Mcb %p %wZ\n", Mcb, &Mcb->FullName, Mcb->Parent, &Mcb->Parent->FullName)); Ext2DerefMcb(Mcb->Parent); ClearLongFlag(Mcb->Flags, MCB_ENTRY_TREE); } else { DbgBreak(); } } else { if (IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { ClearLongFlag(Mcb->Flags, MCB_ENTRY_TREE); } DbgBreak(); } Mcb->Parent = NULL; Mcb->de->d_parent = NULL; } } _SEH2_FINALLY { if (LockAcquired) { ExReleaseResourceLite(&Vcb->McbLock); } } _SEH2_END; return TRUE; } VOID Ext2CleanupAllMcbs(PEXT2_VCB Vcb) { BOOLEAN LockAcquired = FALSE; PEXT2_MCB Mcb = NULL; _SEH2_TRY { ExAcquireResourceExclusiveLite( &Vcb->McbLock, TRUE ); LockAcquired = TRUE; while ((Mcb = Ext2FirstUnusedMcb(Vcb, TRUE, Vcb->NumOfMcb)) != 0) { while (Mcb) { PEXT2_MCB Next = Mcb->Next; if (IsMcbSymLink(Mcb)) { Mcb->Target = NULL; } Ext2FreeMcb(Vcb, Mcb); Mcb = Next; } } Ext2FreeMcb(Vcb, Vcb->McbTree); Vcb->McbTree = NULL; } _SEH2_FINALLY { if (LockAcquired) { ExReleaseResourceLite(&Vcb->McbLock); } } _SEH2_END; } BOOLEAN Ext2CheckSetBlock(PEXT2_IRP_CONTEXT IrpContext, PEXT2_VCB Vcb, LONGLONG Block) { PEXT2_GROUP_DESC gd; struct buffer_head *gb = NULL; struct buffer_head *bh = NULL; ULONG group, dwBlk, Length; RTL_BITMAP bitmap; BOOLEAN bModified = FALSE; group = (ULONG)(Block - EXT2_FIRST_DATA_BLOCK) / BLOCKS_PER_GROUP; dwBlk = (ULONG)(Block - EXT2_FIRST_DATA_BLOCK) % BLOCKS_PER_GROUP; gd = ext4_get_group_desc(&Vcb->sb, group, &gb); if (!gd) { return FALSE; } bh = sb_getblk(&Vcb->sb, ext4_block_bitmap(&Vcb->sb, gd)); if (group == Vcb->sbi.s_groups_count - 1) { Length = (ULONG)(TOTAL_BLOCKS % BLOCKS_PER_GROUP); /* s_blocks_count is integer multiple of s_blocks_per_group */ if (Length == 0) Length = BLOCKS_PER_GROUP; } else { Length = BLOCKS_PER_GROUP; } if (dwBlk >= Length) { fini_bh(&gb); fini_bh(&bh); return FALSE; } RtlInitializeBitMap(&bitmap, (PULONG)bh->b_data, Length); if (RtlCheckBit(&bitmap, dwBlk) == 0) { DbgBreak(); RtlSetBits(&bitmap, dwBlk, 1); bModified = TRUE; mark_buffer_dirty(bh); } fini_bh(&gb); fini_bh(&bh); return (!bModified); } BOOLEAN Ext2CheckBitmapConsistency(PEXT2_IRP_CONTEXT IrpContext, PEXT2_VCB Vcb) { ULONG i, j, InodeBlocks; for (i = 0; i < Vcb->sbi.s_groups_count; i++) { PEXT2_GROUP_DESC gd; struct buffer_head *bh = NULL; gd = ext4_get_group_desc(&Vcb->sb, i, &bh); if (!gd) continue; Ext2CheckSetBlock(IrpContext, Vcb, ext4_block_bitmap(&Vcb->sb, gd)); Ext2CheckSetBlock(IrpContext, Vcb, ext4_inode_bitmap(&Vcb->sb, gd)); if (i == Vcb->sbi.s_groups_count - 1) { InodeBlocks = ((INODES_COUNT % INODES_PER_GROUP) * Vcb->InodeSize + Vcb->BlockSize - 1) / (Vcb->BlockSize); } else { InodeBlocks = (INODES_PER_GROUP * Vcb->InodeSize + Vcb->BlockSize - 1) / (Vcb->BlockSize); } for (j = 0; j < InodeBlocks; j++ ) Ext2CheckSetBlock(IrpContext, Vcb, ext4_inode_table(&Vcb->sb, gd) + j); fini_bh(&bh); } return TRUE; } /* Ext2Global->Resource should be already acquired */ VOID Ext2InsertVcb(PEXT2_VCB Vcb) { InsertTailList(&(Ext2Global->VcbList), &Vcb->Next); } /* Ext2Global->Resource should be already acquired */ VOID Ext2RemoveVcb(PEXT2_VCB Vcb) { RemoveEntryList(&Vcb->Next); InitializeListHead(&Vcb->Next); } NTSTATUS Ext2QueryVolumeParams(IN PEXT2_VCB Vcb, IN PUNICODE_STRING Params) { NTSTATUS Status; RTL_QUERY_REGISTRY_TABLE QueryTable[2]; UNICODE_STRING UniName; PUSHORT UniBuffer = NULL; USHORT UUID[50]; int i; int len = 0; /* zero params */ RtlZeroMemory(Params, sizeof(UNICODE_STRING)); /* constructing volume UUID name */ memset(UUID, 0, sizeof(USHORT) * 50); for (i=0; i < 16; i++) { if (i == 0) { swprintf((wchar_t *)&UUID[len], L"{%2.2X",Vcb->SuperBlock->s_uuid[i]); len += 3; } else if (i == 15) { swprintf((wchar_t *)&UUID[len], L"-%2.2X}", Vcb->SuperBlock->s_uuid[i]); len +=4; } else { swprintf((wchar_t *)&UUID[len], L"-%2.2X", Vcb->SuperBlock->s_uuid[i]); len += 3; } } /* allocating memory for UniBuffer */ UniBuffer = Ext2AllocatePool(PagedPool, 1024, EXT2_PARAM_MAGIC); if (NULL == UniBuffer) { Status = STATUS_INSUFFICIENT_RESOURCES; goto errorout; } RtlZeroMemory(UniBuffer, 1024); /* querying volume parameter string */ RtlZeroMemory(&QueryTable[0], sizeof(RTL_QUERY_REGISTRY_TABLE) * 2); QueryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_QUERY_REGISTRY_REQUIRED; QueryTable[0].Name = UUID; QueryTable[0].EntryContext = &(UniName); UniName.MaximumLength = 1024; UniName.Length = 0; UniName.Buffer = UniBuffer; Status = RtlQueryRegistryValues( RTL_REGISTRY_ABSOLUTE, Ext2Global->RegistryPath.Buffer, &QueryTable[0], NULL, NULL ); if (!NT_SUCCESS(Status)) { goto errorout; } errorout: if (NT_SUCCESS(Status)) { *Params = UniName; } else { if (UniBuffer) { Ext2FreePool(UniBuffer, EXT2_PARAM_MAGIC); } } return Status; } VOID Ext2ParseRegistryVolumeParams( IN PUNICODE_STRING Params, OUT PEXT2_VOLUME_PROPERTY3 Property ) { WCHAR Codepage[CODEPAGE_MAXLEN]; WCHAR Prefix[HIDINGPAT_LEN]; WCHAR Suffix[HIDINGPAT_LEN]; USHORT MountPoint[4]; UCHAR DrvLetter[4]; WCHAR wUID[8], wGID[8], wEUID[8], wEGID[8]; CHAR sUID[8], sGID[8], sEUID[8], sEGID[8]; BOOLEAN bWriteSupport = FALSE, bCheckBitmap = FALSE, bCodeName = FALSE, bMountPoint = FALSE; BOOLEAN bUID = 0, bGID = 0, bEUID = 0, bEGID = 0; struct { PWCHAR Name; /* parameters name */ PBOOLEAN bExist; /* is it contained in params */ USHORT Length; /* parameter value length */ PWCHAR uValue; /* value buffer in unicode */ PCHAR aValue; /* value buffer in ansi */ } ParamPattern[] = { /* writing support */ {READING_ONLY, &Property->bReadonly, 0, NULL, NULL}, {WRITING_SUPPORT, &bWriteSupport, 0, NULL, NULL}, {EXT3_FORCEWRITING, &Property->bExt3Writable, 0, NULL, NULL}, /* need check bitmap */ {CHECKING_BITMAP, &bCheckBitmap, 0, NULL, NULL}, /* codepage */ {CODEPAGE_NAME, &bCodeName, CODEPAGE_MAXLEN, &Codepage[0], Property->Codepage}, /* filter prefix and suffix */ {HIDING_PREFIX, &Property->bHidingPrefix, HIDINGPAT_LEN, &Prefix[0], Property->sHidingPrefix}, {HIDING_SUFFIX, &Property->bHidingSuffix, HIDINGPAT_LEN, &Suffix[0], Property->sHidingSuffix}, {MOUNT_POINT, &bMountPoint, 4, &MountPoint[0], &DrvLetter[0]}, {UID, &bUID, 8, &wUID[0], &sUID[0],}, {GID, &bGID, 8, &wGID[0], &sGID[0]}, {EUID, &bEUID, 8, &wEUID[0], &sEUID[0]}, {EGID, &bEGID, 8, &wEGID[0], &sEGID[0]}, /* end */ {NULL, NULL, 0, NULL} }; USHORT i, j, k; #ifdef __REACTOS__ RtlZeroMemory(Codepage, sizeof(WCHAR) * CODEPAGE_MAXLEN); RtlZeroMemory(Prefix, sizeof(WCHAR) * HIDINGPAT_LEN); RtlZeroMemory(Suffix, sizeof(WCHAR) * HIDINGPAT_LEN); #else RtlZeroMemory(Codepage, CODEPAGE_MAXLEN); RtlZeroMemory(Prefix, HIDINGPAT_LEN); RtlZeroMemory(Suffix, HIDINGPAT_LEN); #endif RtlZeroMemory(MountPoint, sizeof(USHORT) * 4); RtlZeroMemory(DrvLetter, sizeof(CHAR) * 4); RtlZeroMemory(Property, sizeof(EXT2_VOLUME_PROPERTY3)); Property->Magic = EXT2_VOLUME_PROPERTY_MAGIC; Property->Command = APP_CMD_SET_PROPERTY3; for (i=0; ParamPattern[i].Name != NULL; i++) { UNICODE_STRING Name1=*Params, Name2; RtlInitUnicodeString(&Name2, ParamPattern[i].Name); *ParamPattern[i].bExist = FALSE; for (j=0; j * sizeof(WCHAR) + Name2.Length <= Params->Length ; j++) { Name1.MaximumLength = Params->Length - j * sizeof(WCHAR); Name1.Length = Name2.Length; Name1.Buffer = &Params->Buffer[j]; if (!RtlCompareUnicodeString(&Name1, &Name2, TRUE)) { if (j * sizeof(WCHAR) + Name2.Length == Params->Length || Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L';' || Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L',' ) { *(ParamPattern[i].bExist) = TRUE; } else if ((j * 2 + Name2.Length < Params->Length + 2) || (Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L'=' )) { j += Name2.Length/sizeof(WCHAR) + 1; k = 0; while ( j + k < Params->Length/2 && k < ParamPattern[i].Length && Params->Buffer[j+k] != L';' && Params->Buffer[j+k] != L',' ) { #ifdef __REACTOS__ ParamPattern[i].uValue[k] = Params->Buffer[j + k]; k++; #else ParamPattern[i].uValue[k] = Params->Buffer[j + k++]; #endif } if (k) { NTSTATUS status; ANSI_STRING AnsiName; AnsiName.Length = 0; AnsiName.MaximumLength =ParamPattern[i].Length; AnsiName.Buffer = ParamPattern[i].aValue; Name2.Buffer = ParamPattern[i].uValue; Name2.MaximumLength = Name2.Length = k * sizeof(WCHAR); status = RtlUnicodeStringToAnsiString( &AnsiName, &Name2, FALSE); if (NT_SUCCESS(status)) { *(ParamPattern[i].bExist) = TRUE; } else { *ParamPattern[i].bExist = FALSE; } } } break; } } } if (bMountPoint) { Property->DrvLetter = DrvLetter[0]; Property->DrvLetter |= 0x80; } if (bUID && bGID) { SetFlag(Property->Flags2, EXT2_VPROP3_USERIDS); sUID[7] = sGID[7] = sEUID[7] = sEGID[7] = 0; Property->uid = (USHORT)atoi(sUID); Property->gid = (USHORT)atoi(sGID); if (bEUID) { Property->euid = (USHORT)atoi(sEUID); Property->egid = (USHORT)atoi(sEGID); Property->EIDS = TRUE; } else { Property->EIDS = FALSE; } } else { ClearFlag(Property->Flags2, EXT2_VPROP3_USERIDS); } } NTSTATUS Ext2PerformRegistryVolumeParams(IN PEXT2_VCB Vcb) { NTSTATUS Status; UNICODE_STRING VolumeParams; Status = Ext2QueryVolumeParams(Vcb, &VolumeParams); if (NT_SUCCESS(Status)) { /* set Vcb settings from registery */ EXT2_VOLUME_PROPERTY3 Property; Ext2ParseRegistryVolumeParams(&VolumeParams, &Property); Ext2ProcessVolumeProperty(Vcb, &Property, sizeof(Property)); } else { /* don't support auto mount */ if (IsFlagOn(Ext2Global->Flags, EXT2_AUTO_MOUNT)) { Status = STATUS_SUCCESS; } else { Status = STATUS_UNSUCCESSFUL; goto errorout; } /* set Vcb settings from Ext2Global */ if (IsFlagOn(Ext2Global->Flags, EXT2_SUPPORT_WRITING)) { if (Vcb->IsExt3fs) { if (IsFlagOn(Ext2Global->Flags, EXT3_FORCE_WRITING)) { ClearLongFlag(Vcb->Flags, VCB_READ_ONLY); } else { SetLongFlag(Vcb->Flags, VCB_READ_ONLY); } } else { ClearLongFlag(Vcb->Flags, VCB_READ_ONLY); } } else { SetLongFlag(Vcb->Flags, VCB_READ_ONLY); } /* set the default codepage */ Vcb->Codepage.PageTable = Ext2Global->Codepage.PageTable; memcpy(Vcb->Codepage.AnsiName, Ext2Global->Codepage.AnsiName, CODEPAGE_MAXLEN); Vcb->Codepage.PageTable = Ext2Global->Codepage.PageTable; if (Vcb->bHidingPrefix == Ext2Global->bHidingPrefix) { RtlCopyMemory( Vcb->sHidingPrefix, Ext2Global->sHidingPrefix, HIDINGPAT_LEN); } else { RtlZeroMemory( Vcb->sHidingPrefix, HIDINGPAT_LEN); } if (Vcb->bHidingSuffix == Ext2Global->bHidingSuffix) { RtlCopyMemory( Vcb->sHidingSuffix, Ext2Global->sHidingSuffix, HIDINGPAT_LEN); } else { RtlZeroMemory( Vcb->sHidingSuffix, HIDINGPAT_LEN); } } errorout: if (VolumeParams.Buffer) { Ext2FreePool(VolumeParams.Buffer, EXT2_PARAM_MAGIC); } return Status; } NTSTATUS Ext2InitializeLabel( IN PEXT2_VCB Vcb, IN PEXT2_SUPER_BLOCK Sb ) { NTSTATUS status; USHORT Length; UNICODE_STRING Label; OEM_STRING OemName; Label.MaximumLength = 16 * sizeof(WCHAR); Label.Length = 0; Label.Buffer = Vcb->Vpb->VolumeLabel; Vcb->Vpb->VolumeLabelLength = 0; RtlZeroMemory(Label.Buffer, Label.MaximumLength); Length = 16; while ( (Length > 0) && ((Sb->s_volume_name[Length -1] == 0x00) || (Sb->s_volume_name[Length - 1] == 0x20) ) ) { Length--; } if (Length == 0) { return STATUS_SUCCESS; } OemName.Buffer = Sb->s_volume_name; OemName.MaximumLength = 16; OemName.Length = Length; status = Ext2OEMToUnicode(Vcb, &Label, &OemName); if (NT_SUCCESS(status)) { Vcb->Vpb->VolumeLabelLength = Label.Length; } return status; } static __inline BOOLEAN Ext2IsNullUuid (__u8 * uuid) { int i; for (i = 0; i < 16; i++) { if (uuid[i]) { break; } } return (i >= 16); } #define is_power_of_2(x) ((x) != 0 && (((x) & ((x) - 1)) == 0)) NTSTATUS Ext2InitializeVcb( IN PEXT2_IRP_CONTEXT IrpContext, IN PEXT2_VCB Vcb, IN PEXT2_SUPER_BLOCK sb, IN PDEVICE_OBJECT TargetDevice, IN PDEVICE_OBJECT VolumeDevice, IN PVPB Vpb ) { NTSTATUS Status = STATUS_UNRECOGNIZED_VOLUME; ULONG IoctlSize; LONGLONG DiskSize; LONGLONG PartSize; UNICODE_STRING RootNode; USHORT Buffer[2]; ULONG ChangeCount = 0, features; CC_FILE_SIZES FileSizes; int i, has_huge_files; BOOLEAN VcbResourceInitialized = FALSE; BOOLEAN NotifySyncInitialized = FALSE; BOOLEAN ExtentsInitialized = FALSE; BOOLEAN InodeLookasideInitialized = FALSE; BOOLEAN GroupLoaded = FALSE; _SEH2_TRY { if (Vpb == NULL) { Status = STATUS_DEVICE_NOT_READY; _SEH2_LEAVE; } /* Reject mounting volume if we encounter unsupported incompat features */ if (FlagOn(sb->s_feature_incompat, ~EXT4_FEATURE_INCOMPAT_SUPP)) { Status = STATUS_UNRECOGNIZED_VOLUME; _SEH2_LEAVE; } /* Mount the volume RO if we encounter unsupported ro_compat features */ if (FlagOn(sb->s_feature_ro_compat, ~EXT4_FEATURE_RO_COMPAT_SUPP)) { SetLongFlag(Vcb->Flags, VCB_RO_COMPAT_READ_ONLY); } /* Recognize the filesystem as Ext3fs if it supports journalling */ if (IsFlagOn(sb->s_feature_compat, EXT4_FEATURE_COMPAT_HAS_JOURNAL)) { Vcb->IsExt3fs = TRUE; } /* check block size */ Vcb->BlockSize = (EXT2_MIN_BLOCK_SIZE << sb->s_log_block_size); /* we cannot handle volume with block size bigger than 64k */ if (Vcb->BlockSize > EXT2_MAX_USER_BLKSIZE) { Status = STATUS_UNRECOGNIZED_VOLUME; _SEH2_LEAVE; } if (Vcb->BlockSize >= PAGE_SIZE) { Vcb->IoUnitBits = PAGE_SHIFT; Vcb->IoUnitSize = PAGE_SIZE; } else { Vcb->IoUnitSize = Vcb->BlockSize; Vcb->IoUnitBits = Ext2Log2(Vcb->BlockSize); } /* initialize vcb header members ... */ Vcb->Header.IsFastIoPossible = FastIoIsNotPossible; Vcb->Header.Resource = &(Vcb->MainResource); Vcb->Header.PagingIoResource = &(Vcb->PagingIoResource); Vcb->OpenVolumeCount = 0; Vcb->OpenHandleCount = 0; Vcb->ReferenceCount = 0; /* initialize eresources */ ExInitializeResourceLite(&Vcb->MainResource); ExInitializeResourceLite(&Vcb->PagingIoResource); ExInitializeResourceLite(&Vcb->MetaInode); ExInitializeResourceLite(&Vcb->MetaBlock); ExInitializeResourceLite(&Vcb->McbLock); ExInitializeResourceLite(&Vcb->FcbLock); ExInitializeResourceLite(&Vcb->sbi.s_gd_lock); #ifndef _WIN2K_TARGET_ ExInitializeFastMutex(&Vcb->Mutex); FsRtlSetupAdvancedHeader(&Vcb->Header, &Vcb->Mutex); #endif VcbResourceInitialized = TRUE; /* initialize Fcb list head */ InitializeListHead(&Vcb->FcbList); /* initialize Mcb list head */ InitializeListHead(&(Vcb->McbList)); /* initialize directory notify list */ InitializeListHead(&Vcb->NotifyList); FsRtlNotifyInitializeSync(&Vcb->NotifySync); NotifySyncInitialized = TRUE; /* superblock checking */ Vcb->SuperBlock = sb; /* initialize Vpb and Label */ Vcb->DeviceObject = VolumeDevice; Vcb->TargetDeviceObject = TargetDevice; Vcb->Vpb = Vpb; Vcb->RealDevice = Vpb->RealDevice; Vpb->DeviceObject = VolumeDevice; /* set inode size */ Vcb->InodeSize = (ULONG)sb->s_inode_size; if (Vcb->InodeSize == 0) { Vcb->InodeSize = EXT2_GOOD_OLD_INODE_SIZE; } /* initialize inode lookaside list */ ExInitializeNPagedLookasideList(&(Vcb->InodeLookasideList), NULL, NULL, 0, Vcb->InodeSize, 'SNIE', 0); InodeLookasideInitialized = TRUE; /* initialize label in Vpb */ Status = Ext2InitializeLabel(Vcb, sb); if (!NT_SUCCESS(Status)) { DbgBreak(); } /* check device characteristics flags */ if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_REMOVABLE_MEDIA)) { SetLongFlag(Vcb->Flags, VCB_REMOVABLE_MEDIA); } if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_FLOPPY_DISKETTE)) { SetLongFlag(Vcb->Flags, VCB_FLOPPY_DISK); } if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_READ_ONLY_DEVICE)) { SetLongFlag(Vcb->Flags, VCB_WRITE_PROTECTED); } if (IsFlagOn(TargetDevice->Characteristics, FILE_READ_ONLY_DEVICE)) { SetLongFlag(Vcb->Flags, VCB_WRITE_PROTECTED); } /* verify device is writable ? */ if (Ext2IsMediaWriteProtected(IrpContext, TargetDevice)) { SetFlag(Vcb->Flags, VCB_WRITE_PROTECTED); } /* initialize UUID and serial number */ if (Ext2IsNullUuid(sb->s_uuid)) { ExUuidCreate((UUID *)sb->s_uuid); } Vpb->SerialNumber = ((ULONG*)sb->s_uuid)[0] + ((ULONG*)sb->s_uuid)[1] + ((ULONG*)sb->s_uuid)[2] + ((ULONG*)sb->s_uuid)[3]; /* query partition size and disk geometry parameters */ DiskSize = Vcb->DiskGeometry.Cylinders.QuadPart * Vcb->DiskGeometry.TracksPerCylinder * Vcb->DiskGeometry.SectorsPerTrack * Vcb->DiskGeometry.BytesPerSector; IoctlSize = sizeof(PARTITION_INFORMATION); Status = Ext2DiskIoControl( TargetDevice, IOCTL_DISK_GET_PARTITION_INFO, NULL, 0, &Vcb->PartitionInformation, &IoctlSize ); if (NT_SUCCESS(Status)) { PartSize = Vcb->PartitionInformation.PartitionLength.QuadPart; } else { Vcb->PartitionInformation.StartingOffset.QuadPart = 0; Vcb->PartitionInformation.PartitionLength.QuadPart = DiskSize; PartSize = DiskSize; Status = STATUS_SUCCESS; } Vcb->Header.AllocationSize.QuadPart = Vcb->Header.FileSize.QuadPart = PartSize; Vcb->Header.ValidDataLength.QuadPart = Vcb->Header.FileSize.QuadPart; /* verify count */ IoctlSize = sizeof(ULONG); Status = Ext2DiskIoControl( TargetDevice, IOCTL_DISK_CHECK_VERIFY, NULL, 0, &ChangeCount, &IoctlSize ); if (!NT_SUCCESS(Status)) { _SEH2_LEAVE; } Vcb->ChangeCount = ChangeCount; /* create the stream object for ext2 volume */ Vcb->Volume = IoCreateStreamFileObject(NULL, Vcb->Vpb->RealDevice); if (!Vcb->Volume) { Status = STATUS_UNRECOGNIZED_VOLUME; _SEH2_LEAVE; } /* initialize streaming object file */ Vcb->Volume->SectionObjectPointer = &(Vcb->SectionObject); Vcb->Volume->ReadAccess = TRUE; Vcb->Volume->WriteAccess = TRUE; Vcb->Volume->DeleteAccess = TRUE; Vcb->Volume->FsContext = (PVOID) Vcb; Vcb->Volume->FsContext2 = NULL; Vcb->Volume->Vpb = Vcb->Vpb; FileSizes.AllocationSize.QuadPart = FileSizes.FileSize.QuadPart = FileSizes.ValidDataLength.QuadPart = Vcb->Header.AllocationSize.QuadPart; CcInitializeCacheMap( Vcb->Volume, &FileSizes, TRUE, &(Ext2Global->CacheManagerNoOpCallbacks), Vcb ); /* initialize disk block LargetMcb and entry Mcb, it will raise an expcetion if failed */ _SEH2_TRY { FsRtlInitializeLargeMcb(&(Vcb->Extents), PagedPool); } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { Status = STATUS_INSUFFICIENT_RESOURCES; DbgBreak(); } _SEH2_END; if (!NT_SUCCESS(Status)) { _SEH2_LEAVE; } ExtentsInitialized = TRUE; /* set block device */ Vcb->bd.bd_dev = Vcb->RealDevice; Vcb->bd.bd_geo = Vcb->DiskGeometry; Vcb->bd.bd_part = Vcb->PartitionInformation; Vcb->bd.bd_volume = Vcb->Volume; Vcb->bd.bd_priv = (void *) Vcb; memset(&Vcb->bd.bd_bh_root, 0, sizeof(struct rb_root)); InitializeListHead(&Vcb->bd.bd_bh_free); ExInitializeResourceLite(&Vcb->bd.bd_bh_lock); KeInitializeEvent(&Vcb->bd.bd_bh_notify, NotificationEvent, TRUE); Vcb->bd.bd_bh_cache = kmem_cache_create("bd_bh_buffer", Vcb->BlockSize, 0, 0, NULL); if (!Vcb->bd.bd_bh_cache) { Status = STATUS_INSUFFICIENT_RESOURCES; _SEH2_LEAVE; } Vcb->SectorBits = Ext2Log2(SECTOR_SIZE); Vcb->sb.s_magic = sb->s_magic; Vcb->sb.s_bdev = &Vcb->bd; Vcb->sb.s_blocksize = BLOCK_SIZE; Vcb->sb.s_blocksize_bits = BLOCK_BITS; Vcb->sb.s_priv = (void *) Vcb; Vcb->sb.s_fs_info = &Vcb->sbi; Vcb->sbi.s_es = sb; Vcb->sbi.s_blocks_per_group = sb->s_blocks_per_group; Vcb->sbi.s_first_ino = sb->s_first_ino; Vcb->sbi.s_desc_size = sb->s_desc_size; if (EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, EXT4_FEATURE_INCOMPAT_64BIT)) { if (Vcb->sbi.s_desc_size < EXT4_MIN_DESC_SIZE_64BIT || Vcb->sbi.s_desc_size > EXT4_MAX_DESC_SIZE || !is_power_of_2(Vcb->sbi.s_desc_size)) { DEBUG(DL_ERR, ("EXT4-fs: unsupported descriptor size %lu\n", Vcb->sbi.s_desc_size)); Status = STATUS_DISK_CORRUPT_ERROR; _SEH2_LEAVE; } } else { Vcb->sbi.s_desc_size = EXT4_MIN_DESC_SIZE; } Vcb->sbi.s_blocks_per_group = sb->s_blocks_per_group; Vcb->sbi.s_inodes_per_group = sb->s_inodes_per_group; if (EXT3_INODES_PER_GROUP(&Vcb->sb) == 0) { Status = STATUS_DISK_CORRUPT_ERROR; _SEH2_LEAVE; } Vcb->sbi.s_inodes_per_block = BLOCK_SIZE / Vcb->InodeSize; if (Vcb->sbi.s_inodes_per_block == 0) { Status = STATUS_DISK_CORRUPT_ERROR; _SEH2_LEAVE; } Vcb->sbi.s_itb_per_group = Vcb->sbi.s_inodes_per_group / Vcb->sbi.s_inodes_per_block; Vcb->sbi.s_desc_per_block = BLOCK_SIZE / GROUP_DESC_SIZE; Vcb->sbi.s_desc_per_block_bits = ilog2(Vcb->sbi.s_desc_per_block); for (i=0; i < 4; i++) { Vcb->sbi.s_hash_seed[i] = sb->s_hash_seed[i]; } Vcb->sbi.s_def_hash_version = sb->s_def_hash_version; if (le32_to_cpu(sb->s_rev_level) == EXT3_GOOD_OLD_REV && (EXT3_HAS_COMPAT_FEATURE(&Vcb->sb, ~0U) || EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, ~0U) || EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, ~0U))) { printk(KERN_WARNING "EXT3-fs warning: feature flags set on rev 0 fs, " "running e2fsck is recommended\n"); } /* * Check feature flags regardless of the revision level, since we * previously didn't change the revision level when setting the flags, * so there is a chance incompat flags are set on a rev 0 filesystem. */ features = EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, ~EXT4_FEATURE_INCOMPAT_SUPP); if (features & EXT4_FEATURE_INCOMPAT_DIRDATA) { SetLongFlag(Vcb->Flags, VCB_READ_ONLY); ClearFlag(features, EXT4_FEATURE_INCOMPAT_DIRDATA); } if (features) { printk(KERN_ERR "EXT3-fs: %s: couldn't mount because of " "unsupported optional features (%x).\n", Vcb->sb.s_id, le32_to_cpu(features)); Status = STATUS_UNRECOGNIZED_VOLUME; _SEH2_LEAVE; } features = EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, ~EXT4_FEATURE_RO_COMPAT_SUPP); if (features) { printk(KERN_ERR "EXT3-fs: %s: unsupported optional features in this volume: (%x).\n", Vcb->sb.s_id, le32_to_cpu(features)); if (CanIWrite(Vcb)) { } else { SetLongFlag(Vcb->Flags, VCB_READ_ONLY); } } has_huge_files = EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, EXT4_FEATURE_RO_COMPAT_HUGE_FILE); Vcb->sb.s_maxbytes = ext3_max_size(BLOCK_BITS, has_huge_files); Vcb->max_bitmap_bytes = ext3_max_bitmap_size(BLOCK_BITS, has_huge_files); Vcb->max_bytes = ext3_max_size(BLOCK_BITS, has_huge_files); /* calculate maximum file bocks ... */ { ULONG dwData[EXT2_BLOCK_TYPES] = {EXT2_NDIR_BLOCKS, 1, 1, 1}; ULONG i; ASSERT(BLOCK_BITS == Ext2Log2(BLOCK_SIZE)); Vcb->sbi.s_groups_count = (ULONG)(ext3_blocks_count(sb) - sb->s_first_data_block + sb->s_blocks_per_group - 1) / sb->s_blocks_per_group; Vcb->max_data_blocks = 0; for (i = 0; i < EXT2_BLOCK_TYPES; i++) { if (BLOCK_BITS >= 12 && i == (EXT2_BLOCK_TYPES - 1)) { dwData[i] = 0x40000000; } else { dwData[i] = dwData[i] << ((BLOCK_BITS - 2) * i); } Vcb->max_blocks_per_layer[i] = dwData[i]; Vcb->max_data_blocks += Vcb->max_blocks_per_layer[i]; } } Vcb->sbi.s_gdb_count = (Vcb->sbi.s_groups_count + Vcb->sbi.s_desc_per_block - 1) / Vcb->sbi.s_desc_per_block; /* load all gorup desc */ if (!Ext2LoadGroup(Vcb)) { Status = STATUS_UNSUCCESSFUL; _SEH2_LEAVE; } GroupLoaded = TRUE; /* recovery journal since it's ext3 */ if (Vcb->IsExt3fs) { Ext2RecoverJournal(IrpContext, Vcb); if (IsFlagOn(Vcb->Flags, VCB_JOURNAL_RECOVER)) { SetLongFlag(Vcb->Flags, VCB_READ_ONLY); } } /* Now allocating the mcb for root ... */ Buffer[0] = L'\\'; Buffer[1] = 0; RootNode.Buffer = Buffer; RootNode.MaximumLength = RootNode.Length = 2; Vcb->McbTree = Ext2AllocateMcb( Vcb, &RootNode, NULL, FILE_ATTRIBUTE_DIRECTORY ); if (!Vcb->McbTree) { DbgBreak(); Status = STATUS_UNSUCCESSFUL; _SEH2_LEAVE; } Vcb->sb.s_root = Ext2BuildEntry(Vcb, NULL, &RootNode); if (!Vcb->sb.s_root) { DbgBreak(); Status = STATUS_UNSUCCESSFUL; _SEH2_LEAVE; } Vcb->sb.s_root->d_sb = &Vcb->sb; Vcb->sb.s_root->d_inode = &Vcb->McbTree->Inode; Vcb->McbTree->de = Vcb->sb.s_root; /* load root inode */ Vcb->McbTree->Inode.i_ino = EXT2_ROOT_INO; Vcb->McbTree->Inode.i_sb = &Vcb->sb; if (!Ext2LoadInode(Vcb, &Vcb->McbTree->Inode)) { DbgBreak(); Status = STATUS_CANT_WAIT; _SEH2_LEAVE; } /* initializeroot node */ Vcb->McbTree->CreationTime = Ext2NtTime(Vcb->McbTree->Inode.i_ctime); Vcb->McbTree->LastAccessTime = Ext2NtTime(Vcb->McbTree->Inode.i_atime); Vcb->McbTree->LastWriteTime = Ext2NtTime(Vcb->McbTree->Inode.i_mtime); Vcb->McbTree->ChangeTime = Ext2NtTime(Vcb->McbTree->Inode.i_mtime); /* check bitmap if user specifies it */ if (IsFlagOn(Ext2Global->Flags, EXT2_CHECKING_BITMAP)) { Ext2CheckBitmapConsistency(IrpContext, Vcb); } /* get anything doen, then refer target device */ ObReferenceObject(Vcb->TargetDeviceObject); /* query parameters from registry */ Ext2PerformRegistryVolumeParams(Vcb); SetLongFlag(Vcb->Flags, VCB_INITIALIZED); } _SEH2_FINALLY { if (!NT_SUCCESS(Status)) { if (Vcb->McbTree) { Ext2FreeMcb(Vcb, Vcb->McbTree); } if (InodeLookasideInitialized) { ExDeleteNPagedLookasideList(&(Vcb->InodeLookasideList)); } if (ExtentsInitialized) { if (Vcb->bd.bd_bh_cache) { if (GroupLoaded) Ext2PutGroup(Vcb); kmem_cache_destroy(Vcb->bd.bd_bh_cache); } FsRtlUninitializeLargeMcb(&(Vcb->Extents)); } if (Vcb->Volume) { if (Vcb->Volume->PrivateCacheMap) { Ext2SyncUninitializeCacheMap(Vcb->Volume); } ObDereferenceObject(Vcb->Volume); } if (NotifySyncInitialized) { FsRtlNotifyUninitializeSync(&Vcb->NotifySync); } if (VcbResourceInitialized) { ExDeleteResourceLite(&Vcb->FcbLock); ExDeleteResourceLite(&Vcb->McbLock); ExDeleteResourceLite(&Vcb->MetaInode); ExDeleteResourceLite(&Vcb->MetaBlock); ExDeleteResourceLite(&Vcb->sbi.s_gd_lock); ExDeleteResourceLite(&Vcb->MainResource); ExDeleteResourceLite(&Vcb->PagingIoResource); } } } _SEH2_END; return Status; } VOID Ext2TearDownStream(IN PEXT2_VCB Vcb) { PFILE_OBJECT Stream = Vcb->Volume; IO_STATUS_BLOCK IoStatus; ASSERT(Vcb != NULL); ASSERT((Vcb->Identifier.Type == EXT2VCB) && (Vcb->Identifier.Size == sizeof(EXT2_VCB))); if (Stream) { Vcb->Volume = NULL; if (IsFlagOn(Stream->Flags, FO_FILE_MODIFIED)) { CcFlushCache(&(Vcb->SectionObject), NULL, 0, &IoStatus); ClearFlag(Stream->Flags, FO_FILE_MODIFIED); } if (Stream->PrivateCacheMap) { Ext2SyncUninitializeCacheMap(Stream); } ObDereferenceObject(Stream); } } VOID Ext2DestroyVcb (IN PEXT2_VCB Vcb) { ASSERT(Vcb != NULL); ASSERT((Vcb->Identifier.Type == EXT2VCB) && (Vcb->Identifier.Size == sizeof(EXT2_VCB))); DEBUG(DL_FUN, ("Ext2DestroyVcb ...\n")); if (Vcb->Volume) { Ext2TearDownStream(Vcb); } ASSERT(NULL == Vcb->Volume); FsRtlNotifyUninitializeSync(&Vcb->NotifySync); Ext2ListExtents(&Vcb->Extents); FsRtlUninitializeLargeMcb(&(Vcb->Extents)); Ext2CleanupAllMcbs(Vcb); Ext2DropBH(Vcb); if (Vcb->bd.bd_bh_cache) kmem_cache_destroy(Vcb->bd.bd_bh_cache); ExDeleteResourceLite(&Vcb->bd.bd_bh_lock); if (Vcb->SuperBlock) { Ext2FreePool(Vcb->SuperBlock, EXT2_SB_MAGIC); Vcb->SuperBlock = NULL; } if (IsFlagOn(Vcb->Flags, VCB_NEW_VPB)) { ASSERT(Vcb->Vpb2 != NULL); DEBUG(DL_DBG, ("Ext2DestroyVcb: Vpb2 to be freed: %p\n", Vcb->Vpb2)); ExFreePoolWithTag(Vcb->Vpb2, TAG_VPB); DEC_MEM_COUNT(PS_VPB, Vcb->Vpb2, sizeof(VPB)); Vcb->Vpb2 = NULL; } ObDereferenceObject(Vcb->TargetDeviceObject); ExDeleteNPagedLookasideList(&(Vcb->InodeLookasideList)); ExDeleteResourceLite(&Vcb->FcbLock); ExDeleteResourceLite(&Vcb->McbLock); ExDeleteResourceLite(&Vcb->MetaInode); ExDeleteResourceLite(&Vcb->MetaBlock); ExDeleteResourceLite(&Vcb->sbi.s_gd_lock); ExDeleteResourceLite(&Vcb->PagingIoResource); ExDeleteResourceLite(&Vcb->MainResource); DEBUG(DL_DBG, ("Ext2DestroyVcb: DevObject=%p Vcb=%p\n", Vcb->DeviceObject, Vcb)); IoDeleteDevice(Vcb->DeviceObject); DEC_MEM_COUNT(PS_VCB, Vcb->DeviceObject, sizeof(EXT2_VCB)); } /* uninitialize cache map */ VOID Ext2SyncUninitializeCacheMap ( IN PFILE_OBJECT FileObject ) { CACHE_UNINITIALIZE_EVENT UninitializeCompleteEvent; NTSTATUS WaitStatus; LARGE_INTEGER Ext2LargeZero = {0,0}; KeInitializeEvent( &UninitializeCompleteEvent.Event, SynchronizationEvent, FALSE); CcUninitializeCacheMap( FileObject, &Ext2LargeZero, &UninitializeCompleteEvent ); WaitStatus = KeWaitForSingleObject( &UninitializeCompleteEvent.Event, Executive, KernelMode, FALSE, NULL); ASSERT (NT_SUCCESS(WaitStatus)); } /* Link Mcb to tail of Vcb->McbList queue */ VOID Ext2LinkTailMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) { if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { return; } ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); if (IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { DEBUG(DL_RES, ( "Ext2LinkTailMcb: %wZ already linked.\n", &Mcb->FullName)); } else { InsertTailList(&Vcb->McbList, &Mcb->Link); SetLongFlag(Mcb->Flags, MCB_VCB_LINK); Ext2ReferXcb(&Vcb->NumOfMcb); } ExReleaseResourceLite(&Vcb->McbLock); } /* Link Mcb to head of Vcb->McbList queue */ VOID Ext2LinkHeadMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) { if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { return; } ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); if (!IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { InsertHeadList(&Vcb->McbList, &Mcb->Link); SetLongFlag(Mcb->Flags, MCB_VCB_LINK); Ext2ReferXcb(&Vcb->NumOfMcb); } else { DEBUG(DL_RES, ( "Ext2LinkHeadMcb: %wZ already linked.\n", &Mcb->FullName)); } ExReleaseResourceLite(&Vcb->McbLock); } /* Unlink Mcb from Vcb->McbList queue */ VOID Ext2UnlinkMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) { if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { return; } ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); if (IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { RemoveEntryList(&(Mcb->Link)); ClearLongFlag(Mcb->Flags, MCB_VCB_LINK); Ext2DerefXcb(&Vcb->NumOfMcb); } else { DEBUG(DL_RES, ( "Ext2UnlinkMcb: %wZ already unlinked.\n", &Mcb->FullName)); } ExReleaseResourceLite(&Vcb->McbLock); } /* get the first Mcb record in Vcb->McbList */ PEXT2_MCB Ext2FirstUnusedMcb(PEXT2_VCB Vcb, BOOLEAN Wait, ULONG Number) { PEXT2_MCB Head = NULL; PEXT2_MCB Tail = NULL; PEXT2_MCB Mcb = NULL; PLIST_ENTRY List = NULL; ULONG i = 0; LARGE_INTEGER start, now; if (!ExAcquireResourceExclusiveLite(&Vcb->McbLock, Wait)) { return NULL; } KeQuerySystemTime(&start); while (Number--) { BOOLEAN Skip = TRUE; if (IsListEmpty(&Vcb->McbList)) { break; } while (i++ < Vcb->NumOfMcb) { KeQuerySystemTime(&now); if (now.QuadPart > start.QuadPart + (LONGLONG)10*1000*1000) { break; } List = RemoveHeadList(&Vcb->McbList); Mcb = CONTAINING_RECORD(List, EXT2_MCB, Link); ASSERT(IsFlagOn(Mcb->Flags, MCB_VCB_LINK)); if (Mcb->Fcb == NULL && !IsMcbRoot(Mcb) && Mcb->Refercount == 0 && (Mcb->Child == NULL || IsMcbSymLink(Mcb))) { Ext2RemoveMcb(Vcb, Mcb); ClearLongFlag(Mcb->Flags, MCB_VCB_LINK); Ext2DerefXcb(&Vcb->NumOfMcb); /* attach all Mcb into a chain*/ if (Head) { ASSERT(Tail != NULL); Tail->Next = Mcb; Tail = Mcb; } else { Head = Tail = Mcb; } Tail->Next = NULL; Skip = FALSE; } else { InsertTailList(&Vcb->McbList, &Mcb->Link); Mcb = NULL; } } if (Skip) break; } ExReleaseResourceLite(&Vcb->McbLock); return Head; } /* Reaper thread to release unused Mcb blocks */ VOID NTAPI Ext2McbReaperThread( PVOID Context ) { PEXT2_REAPER Reaper = Context; PLIST_ENTRY List = NULL; LARGE_INTEGER Timeout; PEXT2_VCB Vcb = NULL; PEXT2_MCB Mcb = NULL; ULONG i, NumOfMcbs; BOOLEAN GlobalAcquired = FALSE; BOOLEAN DidNothing = TRUE; BOOLEAN LastState = TRUE; BOOLEAN WaitLock; _SEH2_TRY { Reaper->Thread = PsGetCurrentThread(); /* wake up DirverEntry */ KeSetEvent(&Reaper->Engine, 0, FALSE); /* now process looping */ while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { WaitLock = FALSE; /* calculate how long we need wait */ if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 128)) { Timeout.QuadPart = (LONGLONG)-1000*1000; /* 0.1 second */ NumOfMcbs = Ext2Global->MaxDepth * 4; WaitLock = TRUE; } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 32)) { Timeout.QuadPart = (LONGLONG)-1000*1000*5; /* 0.5 second */ NumOfMcbs = Ext2Global->MaxDepth * 2; WaitLock = TRUE; } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 8)) { Timeout.QuadPart = (LONGLONG)-1000*1000*10; /* 1 second */ NumOfMcbs = Ext2Global->MaxDepth; } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 2)) { Timeout.QuadPart = (LONGLONG)-2*1000*1000*10; /* 2 second */ NumOfMcbs = Ext2Global->MaxDepth / 4; } else if (Ext2Global->PerfStat.Current.Mcb > (ULONG)Ext2Global->MaxDepth) { Timeout.QuadPart = (LONGLONG)-4*1000*1000*10; /* 4 seconds */ NumOfMcbs = Ext2Global->MaxDepth / 8; } else if (DidNothing) { Timeout.QuadPart = (LONGLONG)-8*1000*1000*10; /* 8 seconds */ if (LastState) { Timeout.QuadPart *= 2; } NumOfMcbs = Ext2Global->MaxDepth / 16; } else { Timeout.QuadPart = (LONGLONG)-5*1000*1000*10; /* 5 seconds */ if (LastState) { Timeout.QuadPart *= 2; } NumOfMcbs = Ext2Global->MaxDepth / 32; } if (NumOfMcbs == 0) NumOfMcbs = 1; LastState = DidNothing; /* wait until it is waken or it times out */ KeWaitForSingleObject( &Reaper->Wait, Executive, KernelMode, FALSE, &Timeout ); if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) break; DidNothing = TRUE; /* acquire global exclusive lock */ if (!ExAcquireResourceSharedLite(&Ext2Global->Resource, WaitLock)) { continue; } GlobalAcquired = TRUE; /* search all Vcb to get unused resources freed to system */ for (List = Ext2Global->VcbList.Flink; List != &(Ext2Global->VcbList); List = List->Flink ) { Vcb = CONTAINING_RECORD(List, EXT2_VCB, Next); Mcb = Ext2FirstUnusedMcb(Vcb, WaitLock, NumOfMcbs); while (Mcb) { PEXT2_MCB Next = Mcb->Next; DEBUG(DL_RES, ( "Ext2ReaperThread: releasing Mcb (%p): %wZ" " Total: %xh\n", Mcb, &Mcb->FullName, Ext2Global->PerfStat.Current.Mcb)); Ext2FreeMcb(Vcb, Mcb); Mcb = Next; LastState = DidNothing = FALSE; } } if (DidNothing) { KeClearEvent(&Reaper->Wait); } if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); GlobalAcquired = FALSE; } } } _SEH2_FINALLY { if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); } KeSetEvent(&Reaper->Engine, 0, FALSE); } _SEH2_END; PsTerminateSystemThread(STATUS_SUCCESS); } /* get buffer heads from global Vcb BH list */ BOOLEAN Ext2QueryUnusedBH(PEXT2_VCB Vcb, PLIST_ENTRY head) { struct buffer_head *bh = NULL; PLIST_ENTRY next = NULL; LARGE_INTEGER start, now; BOOLEAN wake = FALSE; KeQuerySystemTime(&start); ExAcquireResourceExclusiveLite(&Vcb->bd.bd_bh_lock, TRUE); while (!IsListEmpty(&Vcb->bd.bd_bh_free)) { KeQuerySystemTime(&now); if (now.QuadPart > start.QuadPart + (LONGLONG)10*1000*1000) { break; } next = RemoveHeadList(&Vcb->bd.bd_bh_free); bh = CONTAINING_RECORD(next, struct buffer_head, b_link); if (atomic_read(&bh->b_count)) { InitializeListHead(&bh->b_link); /* to be inserted by brelse */ continue; } if ( IsFlagOn(Vcb->Flags, VCB_BEING_DROPPED) || (bh->b_ts_drop.QuadPart + (LONGLONG)10*1000*1000*15) > now.QuadPart || (bh->b_ts_creat.QuadPart + (LONGLONG)10*1000*1000*180) > now.QuadPart) { InsertTailList(head, &bh->b_link); buffer_head_remove(&Vcb->bd, bh); } else { InsertHeadList(&Vcb->bd.bd_bh_free, &bh->b_link); break; } } wake = IsListEmpty(&Vcb->bd.bd_bh_free); ExReleaseResourceLite(&Vcb->bd.bd_bh_lock); if (wake) KeSetEvent(&Vcb->bd.bd_bh_notify, 0, FALSE); return IsFlagOn(Vcb->Flags, VCB_BEING_DROPPED); } /* Reaper thread to release unused buffer heads */ VOID NTAPI Ext2bhReaperThread( PVOID Context ) { PEXT2_REAPER Reaper = Context; PEXT2_VCB Vcb = NULL; LIST_ENTRY List, *Link; LARGE_INTEGER Timeout; BOOLEAN GlobalAcquired = FALSE; BOOLEAN DidNothing = FALSE; BOOLEAN NonWait = FALSE; _SEH2_TRY { Reaper->Thread = PsGetCurrentThread(); /* wake up DirverEntry */ KeSetEvent(&Reaper->Engine, 0, FALSE); /* now process looping */ while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { /* wait until it is waken or it times out */ if (NonWait) { Timeout.QuadPart = (LONGLONG)-10*1000*10; NonWait = FALSE; } else if (DidNothing) { Timeout.QuadPart = Timeout.QuadPart * 2; } else { Timeout.QuadPart = (LONGLONG)-10*1000*1000*10; /* 10 seconds */ } KeWaitForSingleObject( &Reaper->Wait, Executive, KernelMode, FALSE, &Timeout ); if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) break; InitializeListHead(&List); /* acquire global exclusive lock */ ExAcquireResourceSharedLite(&Ext2Global->Resource, TRUE); GlobalAcquired = TRUE; /* search all Vcb to get unused resources freed to system */ for (Link = Ext2Global->VcbList.Flink; Link != &(Ext2Global->VcbList); Link = Link->Flink ) { Vcb = CONTAINING_RECORD(Link, EXT2_VCB, Next); NonWait = Ext2QueryUnusedBH(Vcb, &List); } DidNothing = IsListEmpty(&List); if (DidNothing) { KeClearEvent(&Reaper->Wait); } if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); GlobalAcquired = FALSE; } while (!IsListEmpty(&List)) { struct buffer_head *bh; Link = RemoveHeadList(&List); bh = CONTAINING_RECORD(Link, struct buffer_head, b_link); ASSERT(0 == atomic_read(&bh->b_count)); free_buffer_head(bh); } } } _SEH2_FINALLY { if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); } KeSetEvent(&Reaper->Engine, 0, FALSE); } _SEH2_END; PsTerminateSystemThread(STATUS_SUCCESS); } /* get unused Fcbs to free */ BOOLEAN Ext2QueryUnusedFcb(PEXT2_VCB Vcb, PLIST_ENTRY list) { PEXT2_FCB Fcb; PLIST_ENTRY next = NULL; LARGE_INTEGER start, now; ULONG count = 0; ULONG tries = 0; BOOLEAN wake = FALSE; BOOLEAN retry = TRUE; KeQuerySystemTime(&start); ExAcquireResourceExclusiveLite(&Vcb->FcbLock, TRUE); again: KeQuerySystemTime(&now); while (!IsListEmpty(&Vcb->FcbList)) { next = RemoveHeadList(&Vcb->FcbList); Fcb = CONTAINING_RECORD(next, EXT2_FCB, Next); if (Fcb->ReferenceCount > 0) { InsertTailList(&Vcb->FcbList, &Fcb->Next); break; } retry = FALSE; if (now.QuadPart < Fcb->TsDrop.QuadPart + 10*1000*1000*120) { InsertHeadList(&Vcb->FcbList, &Fcb->Next); break; } Ext2UnlinkFcb(Fcb); Ext2DerefXcb(&Vcb->FcbCount); InsertTailList(list, &Fcb->Next); if (++count >= Ext2Global->MaxDepth) { break; } } if (start.QuadPart + 10*1000*1000 > now.QuadPart) { retry = FALSE; } if (retry) { if (++tries < (Vcb->FcbCount >> 4) ) goto again; } ExReleaseResourceLite(&Vcb->FcbLock); return 0; } /* Reaper thread to release Fcb */ VOID NTAPI Ext2FcbReaperThread( PVOID Context ) { PEXT2_REAPER Reaper = Context; PEXT2_VCB Vcb = NULL; LIST_ENTRY List, *Link; LARGE_INTEGER Timeout; BOOLEAN GlobalAcquired = FALSE; BOOLEAN DidNothing = FALSE; BOOLEAN NonWait = FALSE; _SEH2_TRY { Reaper->Thread = PsGetCurrentThread(); /* wake up DirverEntry */ KeSetEvent(&Reaper->Engine, 0, FALSE); /* now process looping */ while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { /* wait until it is waken or it times out */ if (NonWait) { Timeout.QuadPart = (LONGLONG)-10*1000*100; NonWait = FALSE; } else if (DidNothing) { Timeout.QuadPart = Timeout.QuadPart * 2; } else { Timeout.QuadPart = (LONGLONG)-10*1000*1000*20; /* 20 seconds */ } KeWaitForSingleObject( &Reaper->Wait, Executive, KernelMode, FALSE, &Timeout ); if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) break; InitializeListHead(&List); /* acquire global exclusive lock */ ExAcquireResourceSharedLite(&Ext2Global->Resource, TRUE); GlobalAcquired = TRUE; /* search all Vcb to get unused resources freed to system */ for (Link = Ext2Global->VcbList.Flink; Link != &(Ext2Global->VcbList); Link = Link->Flink ) { Vcb = CONTAINING_RECORD(Link, EXT2_VCB, Next); NonWait = Ext2QueryUnusedFcb(Vcb, &List); } DidNothing = IsListEmpty(&List); if (DidNothing) { KeClearEvent(&Reaper->Wait); } if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); GlobalAcquired = FALSE; } while (!IsListEmpty(&List)) { PEXT2_FCB Fcb; Link = RemoveHeadList(&List); Fcb = CONTAINING_RECORD(Link, EXT2_FCB, Next); ASSERT(0 == Fcb->ReferenceCount); Ext2FreeFcb(Fcb); } } } _SEH2_FINALLY { if (GlobalAcquired) { ExReleaseResourceLite(&Ext2Global->Resource); } KeSetEvent(&Reaper->Engine, 0, FALSE); } _SEH2_END; PsTerminateSystemThread(STATUS_SUCCESS); } NTSTATUS Ext2StartReaper(PEXT2_REAPER Reaper, EXT2_REAPER_RELEASE Free) { NTSTATUS status = STATUS_SUCCESS; OBJECT_ATTRIBUTES oa; HANDLE handle = 0; LARGE_INTEGER timeout; Reaper->Free = Free; /* initialize wait event */ KeInitializeEvent( &Reaper->Wait, SynchronizationEvent, FALSE ); /* Reaper thread engine event */ KeInitializeEvent( &Reaper->Engine, SynchronizationEvent, FALSE ); /* initialize oa */ InitializeObjectAttributes( &oa, NULL, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, NULL, NULL ); /* start a new system thread */ status = PsCreateSystemThread( &handle, 0, &oa, NULL, NULL, Free, (PVOID)Reaper ); if (NT_SUCCESS(status)) { ZwClose(handle); /* make sure Reaperthread is started */ timeout.QuadPart = (LONGLONG)-10*1000*1000*2; /* 2 seconds */ status = KeWaitForSingleObject( &Reaper->Engine, Executive, KernelMode, FALSE, &timeout ); if (status != STATUS_SUCCESS) { status = STATUS_INSUFFICIENT_RESOURCES; } } return status; } VOID NTAPI Ext2StopReaper(PEXT2_REAPER Reaper) { LARGE_INTEGER timeout; Reaper->Flags |= EXT2_REAPER_FLAG_STOP; KeSetEvent(&Reaper->Wait, 0, FALSE); /* make sure Reaperthread is started */ timeout.QuadPart = (LONGLONG)-10*1000*1000*2; /* 2 seconds */ KeWaitForSingleObject( &Reaper->Engine, Executive, KernelMode, FALSE, &timeout); }