1 /* 2 * COPYRIGHT: See COPYRIGHT.TXT 3 * PROJECT: Ext2 File System Driver for WinNT/2K/XP 4 * FILE: memory.c 5 * PROGRAMMER: Matt Wu <mattwu@163.com> 6 * HOMEPAGE: http://www.ext2fsd.com 7 * UPDATE HISTORY: 8 */ 9 10 /* INCLUDES *****************************************************************/ 11 12 #include "ext2fs.h" 13 14 /* GLOBALS ***************************************************************/ 15 16 extern PEXT2_GLOBAL Ext2Global; 17 18 /* DEFINITIONS *************************************************************/ 19 20 #ifdef ALLOC_PRAGMA 21 #pragma alloc_text(PAGE, Ext2AllocateInode) 22 #pragma alloc_text(PAGE, Ext2DestroyInode) 23 #pragma alloc_text(PAGE, Ext2CheckBitmapConsistency) 24 #pragma alloc_text(PAGE, Ext2CheckSetBlock) 25 #pragma alloc_text(PAGE, Ext2InitializeVcb) 26 #pragma alloc_text(PAGE, Ext2TearDownStream) 27 #pragma alloc_text(PAGE, Ext2DestroyVcb) 28 #pragma alloc_text(PAGE, Ext2SyncUninitializeCacheMap) 29 #pragma alloc_text(PAGE, Ext2ReaperThread) 30 #pragma alloc_text(PAGE, Ext2StartReaper) 31 #pragma alloc_text(PAGE, Ext2StopReaper) 32 #endif 33 34 PEXT2_IRP_CONTEXT 35 Ext2AllocateIrpContext (IN PDEVICE_OBJECT DeviceObject, 36 IN PIRP Irp ) 37 { 38 PIO_STACK_LOCATION irpSp; 39 PEXT2_IRP_CONTEXT IrpContext; 40 41 ASSERT(DeviceObject != NULL); 42 ASSERT(Irp != NULL); 43 44 irpSp = IoGetCurrentIrpStackLocation(Irp); 45 46 IrpContext = (PEXT2_IRP_CONTEXT) ( 47 ExAllocateFromNPagedLookasideList( 48 &(Ext2Global->Ext2IrpContextLookasideList))); 49 50 if (IrpContext == NULL) { 51 return NULL; 52 } 53 54 RtlZeroMemory(IrpContext, sizeof(EXT2_IRP_CONTEXT) ); 55 56 IrpContext->Identifier.Type = EXT2ICX; 57 IrpContext->Identifier.Size = sizeof(EXT2_IRP_CONTEXT); 58 59 IrpContext->Irp = Irp; 60 IrpContext->MajorFunction = irpSp->MajorFunction; 61 IrpContext->MinorFunction = irpSp->MinorFunction; 62 IrpContext->DeviceObject = DeviceObject; 63 IrpContext->FileObject = irpSp->FileObject; 64 if (NULL != IrpContext->FileObject) { 65 IrpContext->Fcb = (PEXT2_FCB)IrpContext->FileObject->FsContext; 66 IrpContext->Ccb = (PEXT2_CCB)IrpContext->FileObject->FsContext2; 67 } 68 69 if (IrpContext->FileObject != NULL) { 70 IrpContext->RealDevice = IrpContext->FileObject->DeviceObject; 71 } else if (IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL) { 72 if (irpSp->Parameters.MountVolume.Vpb) { 73 IrpContext->RealDevice = irpSp->Parameters.MountVolume.Vpb->RealDevice; 74 } 75 } 76 77 if (IsFlagOn(irpSp->Flags, SL_WRITE_THROUGH)) { 78 SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WRITE_THROUGH); 79 } 80 81 if (IsFlagOn(irpSp->Flags, SL_OVERRIDE_VERIFY_VOLUME)) { 82 SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_VERIFY_READ); 83 } 84 85 if (IrpContext->MajorFunction == IRP_MJ_CLEANUP || 86 IrpContext->MajorFunction == IRP_MJ_CLOSE || 87 IrpContext->MajorFunction == IRP_MJ_SHUTDOWN || 88 IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL || 89 IrpContext->MajorFunction == IRP_MJ_PNP ) { 90 91 if (IrpContext->MajorFunction == IRP_MJ_FILE_SYSTEM_CONTROL || 92 IrpContext->MajorFunction == IRP_MJ_PNP) { 93 if (IoGetCurrentIrpStackLocation(Irp)->FileObject == NULL || 94 IoIsOperationSynchronous(Irp)) { 95 SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); 96 } 97 } else { 98 SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); 99 } 100 101 } else if (IoIsOperationSynchronous(Irp)) { 102 103 SetFlag(IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT); 104 } 105 106 IrpContext->IsTopLevel = (IoGetTopLevelIrp() == Irp); 107 IrpContext->ExceptionInProgress = FALSE; 108 INC_IRP_COUNT(IrpContext); 109 110 return IrpContext; 111 } 112 113 VOID 114 Ext2FreeIrpContext (IN PEXT2_IRP_CONTEXT IrpContext) 115 { 116 ASSERT(IrpContext != NULL); 117 118 ASSERT((IrpContext->Identifier.Type == EXT2ICX) && 119 (IrpContext->Identifier.Size == sizeof(EXT2_IRP_CONTEXT))); 120 121 /* free the IrpContext to NonPagedList */ 122 IrpContext->Identifier.Type = 0; 123 IrpContext->Identifier.Size = 0; 124 125 DEC_IRP_COUNT(IrpContext); 126 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2IrpContextLookasideList), IrpContext); 127 } 128 129 130 PEXT2_FCB 131 Ext2AllocateFcb ( 132 IN PEXT2_VCB Vcb, 133 IN PEXT2_MCB Mcb 134 ) 135 { 136 PEXT2_FCB Fcb; 137 138 ASSERT(ExIsResourceAcquiredExclusiveLite(&Vcb->FcbLock)); 139 140 Fcb = (PEXT2_FCB) ExAllocateFromNPagedLookasideList( 141 &(Ext2Global->Ext2FcbLookasideList)); 142 143 if (!Fcb) { 144 return NULL; 145 } 146 147 RtlZeroMemory(Fcb, sizeof(EXT2_FCB)); 148 Fcb->Identifier.Type = EXT2FCB; 149 Fcb->Identifier.Size = sizeof(EXT2_FCB); 150 151 #ifndef _WIN2K_TARGET_ 152 ExInitializeFastMutex(&Fcb->Mutex); 153 FsRtlSetupAdvancedHeader(&Fcb->Header, &Fcb->Mutex); 154 #endif 155 156 FsRtlInitializeOplock(&Fcb->Oplock); 157 FsRtlInitializeFileLock ( 158 &Fcb->FileLockAnchor, 159 NULL, 160 NULL ); 161 162 Fcb->OpenHandleCount = 0; 163 Fcb->ReferenceCount = 0; 164 Fcb->Vcb = Vcb; 165 Fcb->Inode = &Mcb->Inode; 166 167 ASSERT(Mcb->Fcb == NULL); 168 Ext2ReferMcb(Mcb); 169 Fcb->Mcb = Mcb; 170 Mcb->Fcb = Fcb; 171 172 DEBUG(DL_RES, ("Ext2AllocateFcb: Fcb %p created: %wZ.\n", 173 Fcb, &Fcb->Mcb->FullName)); 174 175 RtlZeroMemory(&Fcb->Header, sizeof(FSRTL_COMMON_FCB_HEADER)); 176 Fcb->Header.NodeTypeCode = (USHORT) EXT2FCB; 177 Fcb->Header.NodeByteSize = sizeof(EXT2_FCB); 178 Fcb->Header.IsFastIoPossible = FastIoIsNotPossible; 179 Fcb->Header.Resource = &(Fcb->MainResource); 180 Fcb->Header.PagingIoResource = &(Fcb->PagingIoResource); 181 182 Fcb->Header.FileSize.QuadPart = Mcb->Inode.i_size; 183 Fcb->Header.ValidDataLength.QuadPart = Mcb->Inode.i_size; 184 Fcb->Header.AllocationSize.QuadPart = CEILING_ALIGNED(ULONGLONG, 185 Fcb->Header.FileSize.QuadPart, (ULONGLONG)Vcb->BlockSize); 186 187 Fcb->SectionObject.DataSectionObject = NULL; 188 Fcb->SectionObject.SharedCacheMap = NULL; 189 Fcb->SectionObject.ImageSectionObject = NULL; 190 191 ExInitializeResourceLite(&(Fcb->MainResource)); 192 ExInitializeResourceLite(&(Fcb->PagingIoResource)); 193 194 Ext2InsertFcb(Vcb, Fcb); 195 196 INC_MEM_COUNT(PS_FCB, Fcb, sizeof(EXT2_FCB)); 197 198 return Fcb; 199 } 200 201 VOID 202 Ext2UnlinkFcb(IN PEXT2_FCB Fcb) 203 { 204 PEXT2_VCB Vcb = Fcb->Vcb; 205 PEXT2_MCB Mcb; 206 207 ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); 208 Mcb = Fcb->Mcb; 209 210 DEBUG(DL_INF, ("Ext2FreeFcb: Fcb (%p) to be unlinked: %wZ.\n", 211 Fcb, Mcb ? &Mcb->FullName : NULL)); 212 213 if ((Mcb != NULL) && 214 (Mcb->Identifier.Type == EXT2MCB) && 215 (Mcb->Identifier.Size == sizeof(EXT2_MCB))) { 216 217 ASSERT (Mcb->Fcb == Fcb); 218 if (IsMcbSpecialFile(Mcb) || 219 IsFileDeleted(Mcb)) { 220 221 ASSERT(!IsRoot(Fcb)); 222 Ext2RemoveMcb(Vcb, Mcb); 223 Mcb->Fcb = NULL; 224 225 Ext2UnlinkMcb(Vcb, Mcb); 226 Ext2DerefMcb(Mcb); 227 Ext2LinkHeadMcb(Vcb, Mcb); 228 229 } else { 230 Mcb->Fcb = NULL; 231 Ext2DerefMcb(Mcb); 232 } 233 Fcb->Mcb = NULL; 234 } 235 236 ExReleaseResourceLite(&Vcb->McbLock); 237 } 238 239 VOID 240 Ext2FreeFcb (IN PEXT2_FCB Fcb) 241 { 242 PEXT2_VCB Vcb = Fcb->Vcb; 243 244 _SEH2_TRY { 245 246 ASSERT((Fcb != NULL) && (Fcb->Identifier.Type == EXT2FCB) && 247 (Fcb->Identifier.Size == sizeof(EXT2_FCB))); 248 ASSERT(0 == Fcb->ReferenceCount); 249 250 #ifndef _WIN2K_TARGET_ 251 FsRtlTeardownPerStreamContexts(&Fcb->Header); 252 #endif 253 254 FsRtlUninitializeFileLock(&Fcb->FileLockAnchor); 255 FsRtlUninitializeOplock(&Fcb->Oplock); 256 ExDeleteResourceLite(&Fcb->MainResource); 257 ExDeleteResourceLite(&Fcb->PagingIoResource); 258 259 Fcb->Identifier.Type = 0; 260 Fcb->Identifier.Size = 0; 261 262 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2FcbLookasideList), Fcb); 263 DEC_MEM_COUNT(PS_FCB, Fcb, sizeof(EXT2_FCB)); 264 265 if (0 == Ext2DerefXcb(&Vcb->ReferenceCount)) { 266 if (!IsMounted(Vcb) || IsDispending(Vcb)) { 267 Ext2CheckDismount(NULL, Vcb, FALSE); 268 } 269 } 270 271 } _SEH2_FINALLY { 272 } _SEH2_END; 273 } 274 275 VOID 276 Ext2ReleaseFcb (IN PEXT2_FCB Fcb) 277 { 278 PEXT2_VCB Vcb = Fcb->Vcb; 279 PEXT2_MCB Mcb; 280 281 if (0 != Ext2DerefXcb(&Fcb->ReferenceCount)) 282 return; 283 284 ExAcquireResourceExclusiveLite(&Vcb->FcbLock, TRUE); 285 ExAcquireResourceExclusiveLite(&Fcb->MainResource, TRUE); 286 287 Mcb = Fcb->Mcb; 288 RemoveEntryList(&Fcb->Next); 289 290 if (IsFlagOn(Fcb->Flags, FCB_DELETE_PENDING) || 291 NULL == Mcb || IsFileDeleted(Mcb)) { 292 InsertHeadList(&Vcb->FcbList, &Fcb->Next); 293 Fcb->TsDrop.QuadPart = 0; 294 } else { 295 InsertTailList(&Vcb->FcbList, &Fcb->Next); 296 KeQuerySystemTime(&Fcb->TsDrop); 297 } 298 ExReleaseResourceLite(&Fcb->MainResource); 299 ExReleaseResourceLite(&Vcb->FcbLock); 300 301 if ((Vcb->FcbCount >> 6) > (ULONG)(Ext2Global->MaxDepth)) { 302 KeSetEvent(&Ext2Global->FcbReaper.Wait, 0, FALSE); 303 } 304 } 305 306 /* Insert Fcb to Vcb->FcbList queue, with Vcb->FcbLock Acquired. */ 307 308 VOID 309 Ext2InsertFcb(PEXT2_VCB Vcb, PEXT2_FCB Fcb) 310 { 311 ASSERT(ExIsResourceAcquiredExclusiveLite(&Vcb->FcbLock)); 312 313 KeQuerySystemTime(&Fcb->TsDrop); 314 Ext2ReferXcb(&Vcb->FcbCount); 315 Ext2ReferXcb(&Vcb->ReferenceCount); 316 InsertTailList(&Vcb->FcbList, &Fcb->Next); 317 } 318 319 PEXT2_CCB 320 Ext2AllocateCcb (ULONG Flags, PEXT2_MCB SymLink) 321 { 322 PEXT2_CCB Ccb; 323 324 Ccb = (PEXT2_CCB) (ExAllocateFromNPagedLookasideList( 325 &(Ext2Global->Ext2CcbLookasideList))); 326 if (!Ccb) { 327 return NULL; 328 } 329 330 DEBUG(DL_RES, ( "ExtAllocateCcb: Ccb created: %ph.\n", Ccb)); 331 332 RtlZeroMemory(Ccb, sizeof(EXT2_CCB)); 333 334 Ccb->Identifier.Type = EXT2CCB; 335 Ccb->Identifier.Size = sizeof(EXT2_CCB); 336 Ccb->Flags = Flags; 337 338 Ccb->SymLink = SymLink; 339 if (SymLink) { 340 ASSERT(SymLink->Refercount > 0); 341 Ext2ReferMcb(SymLink); 342 DEBUG(DL_INF, ( "ExtAllocateCcb: Ccb SymLink: %wZ.\n", 343 &Ccb->SymLink->FullName)); 344 } 345 346 Ccb->DirectorySearchPattern.Length = 0; 347 Ccb->DirectorySearchPattern.MaximumLength = 0; 348 Ccb->DirectorySearchPattern.Buffer = 0; 349 350 INC_MEM_COUNT(PS_CCB, Ccb, sizeof(EXT2_CCB)); 351 352 return Ccb; 353 } 354 355 VOID 356 Ext2FreeCcb (IN PEXT2_VCB Vcb, IN PEXT2_CCB Ccb) 357 { 358 ASSERT(Ccb != NULL); 359 ASSERT((Ccb->Identifier.Type == EXT2CCB) && 360 (Ccb->Identifier.Size == sizeof(EXT2_CCB))); 361 362 DEBUG(DL_RES, ( "Ext2FreeCcb: Ccb = %ph.\n", Ccb)); 363 364 if (Ccb->SymLink) { 365 DEBUG(DL_INF, ( "Ext2FreeCcb: Ccb SymLink: %wZ.\n", 366 &Ccb->SymLink->FullName)); 367 if (IsFileDeleted(Ccb->SymLink->Target)) { 368 Ext2UnlinkMcb(Vcb, Ccb->SymLink); 369 Ext2DerefMcb(Ccb->SymLink); 370 Ext2LinkHeadMcb(Vcb, Ccb->SymLink); 371 } else { 372 Ext2DerefMcb(Ccb->SymLink); 373 } 374 } 375 376 if (Ccb->DirectorySearchPattern.Buffer != NULL) { 377 DEC_MEM_COUNT(PS_DIR_PATTERN, Ccb->DirectorySearchPattern.Buffer, 378 Ccb->DirectorySearchPattern.MaximumLength ); 379 Ext2FreePool(Ccb->DirectorySearchPattern.Buffer, EXT2_DIRSP_MAGIC); 380 } 381 382 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2CcbLookasideList), Ccb); 383 DEC_MEM_COUNT(PS_CCB, Ccb, sizeof(EXT2_CCB)); 384 } 385 386 PEXT2_INODE 387 Ext2AllocateInode (PEXT2_VCB Vcb) 388 { 389 PVOID inode = NULL; 390 391 inode = ExAllocateFromNPagedLookasideList( 392 &(Vcb->InodeLookasideList)); 393 if (!inode) { 394 return NULL; 395 } 396 397 RtlZeroMemory(inode, INODE_SIZE); 398 399 DEBUG(DL_INF, ("ExtAllocateInode: Inode created: %ph.\n", inode)); 400 INC_MEM_COUNT(PS_EXT2_INODE, inode, INODE_SIZE); 401 402 return inode; 403 } 404 405 VOID 406 Ext2DestroyInode (IN PEXT2_VCB Vcb, IN PEXT2_INODE inode) 407 { 408 ASSERT(inode != NULL); 409 410 DEBUG(DL_INF, ("Ext2FreeInode: Inode = %ph.\n", inode)); 411 412 ExFreeToNPagedLookasideList(&(Vcb->InodeLookasideList), inode); 413 DEC_MEM_COUNT(PS_EXT2_INODE, inode, INODE_SIZE); 414 } 415 416 struct dentry * Ext2AllocateEntry() 417 { 418 struct dentry *de; 419 420 de = (struct dentry *)ExAllocateFromNPagedLookasideList( 421 &(Ext2Global->Ext2DentryLookasideList)); 422 if (!de) { 423 return NULL; 424 } 425 426 RtlZeroMemory(de, sizeof(struct dentry)); 427 INC_MEM_COUNT(PS_DENTRY, de, sizeof(struct dentry)); 428 429 return de; 430 } 431 432 VOID Ext2FreeEntry (IN struct dentry *de) 433 { 434 ASSERT(de != NULL); 435 436 if (de->d_name.name) 437 ExFreePool(de->d_name.name); 438 439 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2DentryLookasideList), de); 440 DEC_MEM_COUNT(PS_DENTRY, de, sizeof(struct dentry)); 441 } 442 443 444 struct dentry *Ext2BuildEntry(PEXT2_VCB Vcb, PEXT2_MCB Dcb, PUNICODE_STRING FileName) 445 { 446 OEM_STRING Oem = { 0 }; 447 struct dentry *de = NULL; 448 NTSTATUS Status = STATUS_INSUFFICIENT_RESOURCES; 449 450 _SEH2_TRY { 451 452 de = Ext2AllocateEntry(); 453 if (!de) { 454 DEBUG(DL_ERR, ("Ext2BuildEntry: failed to allocate dentry.\n")); 455 _SEH2_LEAVE; 456 } 457 de->d_sb = &Vcb->sb; 458 if (Dcb) 459 de->d_parent = Dcb->de; 460 461 Oem.MaximumLength = (USHORT)Ext2UnicodeToOEMSize(Vcb, FileName) + 1; 462 Oem.Buffer = ExAllocatePool(PagedPool, Oem.MaximumLength); 463 if (!Oem.Buffer) { 464 DEBUG(DL_ERR, ( "Ex2BuildEntry: failed to allocate OEM name.\n")); 465 _SEH2_LEAVE; 466 } 467 de->d_name.name = Oem.Buffer; 468 RtlZeroMemory(Oem.Buffer, Oem.MaximumLength); 469 Status = Ext2UnicodeToOEM(Vcb, &Oem, FileName); 470 if (!NT_SUCCESS(Status)) { 471 DEBUG(DL_CP, ("Ext2BuildEntry: failed to convert %S to OEM.\n", FileName->Buffer)); 472 _SEH2_LEAVE; 473 } 474 de->d_name.len = Oem.Length; 475 476 } _SEH2_FINALLY { 477 478 if (!NT_SUCCESS(Status)) { 479 if (de) 480 Ext2FreeEntry(de); 481 } 482 } _SEH2_END; 483 484 return de; 485 } 486 487 PEXT2_EXTENT 488 Ext2AllocateExtent () 489 { 490 PEXT2_EXTENT Extent; 491 492 Extent = (PEXT2_EXTENT)ExAllocateFromNPagedLookasideList( 493 &(Ext2Global->Ext2ExtLookasideList)); 494 if (!Extent) { 495 return NULL; 496 } 497 498 RtlZeroMemory(Extent, sizeof(EXT2_EXTENT)); 499 INC_MEM_COUNT(PS_EXTENT, Extent, sizeof(EXT2_EXTENT)); 500 501 return Extent; 502 } 503 504 VOID 505 Ext2FreeExtent (IN PEXT2_EXTENT Extent) 506 { 507 ASSERT(Extent != NULL); 508 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2ExtLookasideList), Extent); 509 DEC_MEM_COUNT(PS_EXTENT, Extent, sizeof(EXT2_EXTENT)); 510 } 511 512 ULONG 513 Ext2CountExtents(IN PEXT2_EXTENT Chain) 514 { 515 ULONG count = 0; 516 PEXT2_EXTENT List = Chain; 517 518 while (List) { 519 count += 1; 520 List = List->Next; 521 } 522 523 return count; 524 } 525 526 VOID 527 Ext2JointExtents( 528 IN PEXT2_EXTENT Chain, 529 IN PEXT2_EXTENT Extent 530 ) 531 { 532 #ifndef __REACTOS__ 533 ULONG count = 0; 534 #endif 535 PEXT2_EXTENT List = Chain; 536 537 while (List->Next) { 538 List = List->Next; 539 } 540 541 List->Next = Extent; 542 } 543 544 545 VOID 546 Ext2DestroyExtentChain(IN PEXT2_EXTENT Chain) 547 { 548 PEXT2_EXTENT Extent = NULL, List = Chain; 549 550 while (List) { 551 Extent = List->Next; 552 Ext2FreeExtent(List); 553 List = Extent; 554 } 555 } 556 557 BOOLEAN 558 Ext2ListExtents(PLARGE_MCB Extents) 559 { 560 if (FsRtlNumberOfRunsInLargeMcb(Extents) != 0) { 561 562 LONGLONG DirtyVba; 563 LONGLONG DirtyLba; 564 LONGLONG DirtyLength; 565 int i, n = 0; 566 567 for (i = 0; FsRtlGetNextLargeMcbEntry( 568 Extents, i, &DirtyVba, 569 &DirtyLba, &DirtyLength); i++) { 570 if (DirtyVba > 0 && DirtyLba != -1) { 571 DEBUG(DL_EXT, ("Vba:%I64xh Lba:%I64xh Len:%I64xh.\n", DirtyVba, DirtyLba, DirtyLength)); 572 n++; 573 } 574 } 575 576 return n ? TRUE : FALSE; 577 } 578 579 return FALSE; 580 } 581 582 VOID 583 Ext2CheckExtent( 584 PLARGE_MCB Zone, 585 LONGLONG Vbn, 586 LONGLONG Lbn, 587 LONGLONG Length, 588 BOOLEAN bAdded 589 ) 590 { 591 #if EXT2_DEBUG 592 LONGLONG DirtyLbn; 593 LONGLONG DirtyLen; 594 LONGLONG RunStart; 595 LONGLONG RunLength; 596 ULONG Index; 597 BOOLEAN bFound = FALSE; 598 599 bFound = FsRtlLookupLargeMcbEntry( 600 Zone, 601 Vbn, 602 &DirtyLbn, 603 &DirtyLen, 604 &RunStart, 605 &RunLength, 606 &Index ); 607 608 if (!bAdded && (!bFound || DirtyLbn == -1)) { 609 return; 610 } 611 612 if ( !bFound || (DirtyLbn == -1) || 613 (DirtyLbn != Lbn) || 614 (DirtyLen < Length)) { 615 616 DbgBreak(); 617 618 for (Index = 0; TRUE; Index++) { 619 620 if (!FsRtlGetNextLargeMcbEntry( 621 Zone, 622 Index, 623 &Vbn, 624 &Lbn, 625 &Length)) { 626 break; 627 } 628 629 DEBUG(DL_EXT, ("Index = %xh Vbn = %I64xh Lbn = %I64xh Len = %I64xh\n", 630 Index, Vbn, Lbn, Length )); 631 } 632 } 633 #endif 634 } 635 636 VOID 637 Ext2ClearAllExtents(PLARGE_MCB Zone) 638 { 639 _SEH2_TRY { 640 FsRtlTruncateLargeMcb(Zone, (LONGLONG)0); 641 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 642 DbgBreak(); 643 } _SEH2_END; 644 } 645 646 647 BOOLEAN 648 Ext2AddVcbExtent ( 649 IN PEXT2_VCB Vcb, 650 IN LONGLONG Vbn, 651 IN LONGLONG Length 652 ) 653 { 654 ULONG TriedTimes = 0; 655 656 LONGLONG Offset = 0; 657 BOOLEAN rc = FALSE; 658 659 Offset = Vbn & (~(Vcb->IoUnitSize - 1)); 660 Length = (Vbn - Offset + Length + Vcb->IoUnitSize - 1) & 661 ~(Vcb->IoUnitSize - 1); 662 663 ASSERT ((Offset & (Vcb->IoUnitSize - 1)) == 0); 664 ASSERT ((Length & (Vcb->IoUnitSize - 1)) == 0); 665 666 Offset = (Offset >> Vcb->IoUnitBits) + 1; 667 Length = (Length >> Vcb->IoUnitBits); 668 669 Again: 670 671 _SEH2_TRY { 672 rc = FsRtlAddLargeMcbEntry( 673 &Vcb->Extents, 674 Offset, 675 Offset, 676 Length 677 ); 678 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 679 DbgBreak(); 680 rc = FALSE; 681 } _SEH2_END; 682 683 if (!rc && ++TriedTimes < 10) { 684 Ext2Sleep(TriedTimes * 100); 685 goto Again; 686 } 687 688 DEBUG(DL_EXT, ("Ext2AddVcbExtent: Vbn=%I64xh Length=%I64xh," 689 " rc=%d Runs=%u\n", Offset, Length, rc, 690 FsRtlNumberOfRunsInLargeMcb(&Vcb->Extents))); 691 692 if (rc) { 693 Ext2CheckExtent(&Vcb->Extents, Offset, Offset, Length, TRUE); 694 } 695 696 return rc; 697 } 698 699 BOOLEAN 700 Ext2RemoveVcbExtent ( 701 IN PEXT2_VCB Vcb, 702 IN LONGLONG Vbn, 703 IN LONGLONG Length 704 ) 705 { 706 ULONG TriedTimes = 0; 707 LONGLONG Offset = 0; 708 BOOLEAN rc = TRUE; 709 710 Offset = Vbn & (~(Vcb->IoUnitSize - 1)); 711 Length = (Length + Vbn - Offset + Vcb->IoUnitSize - 1) & (~(Vcb->IoUnitSize - 1)); 712 713 ASSERT ((Offset & (Vcb->IoUnitSize - 1)) == 0); 714 ASSERT ((Length & (Vcb->IoUnitSize - 1)) == 0); 715 716 Offset = (Offset >> Vcb->IoUnitBits) + 1; 717 Length = (Length >> Vcb->IoUnitBits); 718 719 Again: 720 721 _SEH2_TRY { 722 FsRtlRemoveLargeMcbEntry( 723 &Vcb->Extents, 724 Offset, 725 Length 726 ); 727 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 728 DbgBreak(); 729 rc = FALSE; 730 } _SEH2_END; 731 732 if (!rc && ++TriedTimes < 10) { 733 Ext2Sleep(TriedTimes * 100); 734 goto Again; 735 } 736 737 DEBUG(DL_EXT, ("Ext2RemoveVcbExtent: Vbn=%I64xh Length=%I64xh Runs=%u\n", 738 Offset, Length, FsRtlNumberOfRunsInLargeMcb(&Vcb->Extents))); 739 if (rc) { 740 Ext2CheckExtent(&Vcb->Extents, Offset, 0, Length, FALSE); 741 } 742 743 return rc; 744 } 745 746 BOOLEAN 747 Ext2LookupVcbExtent ( 748 IN PEXT2_VCB Vcb, 749 IN LONGLONG Vbn, 750 OUT PLONGLONG Lbn, 751 OUT PLONGLONG Length 752 ) 753 { 754 LONGLONG offset; 755 BOOLEAN rc; 756 757 offset = Vbn & (~(Vcb->IoUnitSize - 1)); 758 ASSERT ((offset & (Vcb->IoUnitSize - 1)) == 0); 759 offset = (offset >> Vcb->IoUnitBits) + 1; 760 761 rc = FsRtlLookupLargeMcbEntry( 762 &(Vcb->Extents), 763 offset, 764 Lbn, 765 Length, 766 NULL, 767 NULL, 768 NULL 769 ); 770 771 if (rc) { 772 773 if (Lbn && ((*Lbn) != -1)) { 774 ASSERT((*Lbn) > 0); 775 (*Lbn) = (((*Lbn) - 1) << Vcb->IoUnitBits); 776 (*Lbn) += ((Vbn) & (Vcb->IoUnitSize - 1)); 777 } 778 779 if (Length && *Length) { 780 (*Length) <<= Vcb->IoUnitBits; 781 (*Length) -= ((Vbn) & (Vcb->IoUnitSize - 1)); 782 } 783 } 784 785 return rc; 786 } 787 788 789 BOOLEAN 790 Ext2AddMcbExtent ( 791 IN PEXT2_VCB Vcb, 792 IN PEXT2_MCB Mcb, 793 IN LONGLONG Vbn, 794 IN LONGLONG Lbn, 795 IN LONGLONG Length 796 ) 797 { 798 ULONG TriedTimes = 0; 799 LONGLONG Base = 0; 800 UCHAR Bits = 0; 801 BOOLEAN rc = FALSE; 802 803 Base = (LONGLONG)BLOCK_SIZE; 804 Bits = (UCHAR)BLOCK_BITS; 805 806 ASSERT ((Vbn & (Base - 1)) == 0); 807 ASSERT ((Lbn & (Base - 1)) == 0); 808 ASSERT ((Length & (Base - 1)) == 0); 809 810 Vbn = (Vbn >> Bits) + 1; 811 Lbn = (Lbn >> Bits) + 1; 812 Length = (Length >> Bits); 813 814 Again: 815 816 _SEH2_TRY { 817 818 rc = FsRtlAddLargeMcbEntry( 819 &Mcb->Extents, 820 Vbn, 821 Lbn, 822 Length 823 ); 824 825 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 826 827 DbgBreak(); 828 rc = FALSE; 829 } _SEH2_END; 830 831 if (!rc && ++TriedTimes < 10) { 832 Ext2Sleep(TriedTimes * 100); 833 goto Again; 834 } 835 836 DEBUG(DL_EXT, ("Ext2AddMcbExtent: Vbn=%I64xh Lbn=%I64xh Length=%I64xh," 837 " rc=%d Runs=%u\n", Vbn, Lbn, Length, rc, 838 FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents))); 839 840 if (rc) { 841 Ext2CheckExtent(&Mcb->Extents, Vbn, Lbn, Length, TRUE); 842 } 843 844 return rc; 845 } 846 847 BOOLEAN 848 Ext2RemoveMcbExtent ( 849 IN PEXT2_VCB Vcb, 850 IN PEXT2_MCB Mcb, 851 IN LONGLONG Vbn, 852 IN LONGLONG Length 853 ) 854 { 855 ULONG TriedTimes = 0; 856 LONGLONG Base = 0; 857 UCHAR Bits = 0; 858 BOOLEAN rc = TRUE; 859 860 Base = (LONGLONG)BLOCK_SIZE; 861 Bits = (UCHAR)BLOCK_BITS; 862 863 ASSERT ((Vbn & (Base - 1)) == 0); 864 ASSERT ((Length & (Base - 1)) == 0); 865 866 Vbn = (Vbn >> Bits) + 1; 867 Length = (Length >> Bits); 868 869 Again: 870 871 _SEH2_TRY { 872 FsRtlRemoveLargeMcbEntry( 873 &Mcb->Extents, 874 Vbn, 875 Length 876 ); 877 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 878 DbgBreak(); 879 rc = FALSE; 880 } _SEH2_END; 881 882 if (!rc && ++TriedTimes < 10) { 883 Ext2Sleep(TriedTimes * 100); 884 goto Again; 885 } 886 887 DEBUG(DL_EXT, ("Ext2RemoveMcbExtent: Vbn=%I64xh Length=%I64xh Runs=%u\n", 888 Vbn, Length, FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents))); 889 if (rc) { 890 Ext2CheckExtent(&Mcb->Extents, Vbn, 0, Length, FALSE); 891 } 892 893 return rc; 894 } 895 896 BOOLEAN 897 Ext2LookupMcbExtent ( 898 IN PEXT2_VCB Vcb, 899 IN PEXT2_MCB Mcb, 900 IN LONGLONG Vbn, 901 OUT PLONGLONG Lbn, 902 OUT PLONGLONG Length 903 ) 904 { 905 LONGLONG offset; 906 BOOLEAN rc; 907 908 offset = Vbn & (~((LONGLONG)BLOCK_SIZE - 1)); 909 ASSERT ((offset & (BLOCK_SIZE - 1)) == 0); 910 offset = (offset >> BLOCK_BITS) + 1; 911 912 rc = FsRtlLookupLargeMcbEntry( 913 &(Mcb->Extents), 914 offset, 915 Lbn, 916 Length, 917 NULL, 918 NULL, 919 NULL 920 ); 921 922 if (rc) { 923 924 if (Lbn && ((*Lbn) != -1)) { 925 ASSERT((*Lbn) > 0); 926 (*Lbn) = (((*Lbn) - 1) << BLOCK_BITS); 927 (*Lbn) += ((Vbn) & ((LONGLONG)BLOCK_SIZE - 1)); 928 } 929 930 if (Length && *Length) { 931 (*Length) <<= BLOCK_BITS; 932 (*Length) -= ((Vbn) & ((LONGLONG)BLOCK_SIZE - 1)); 933 } 934 } 935 936 return rc; 937 } 938 939 940 BOOLEAN 941 Ext2AddMcbMetaExts ( 942 IN PEXT2_VCB Vcb, 943 IN PEXT2_MCB Mcb, 944 IN ULONG Block, 945 IN ULONG Length 946 ) 947 { 948 ULONG TriedTimes = 0; 949 LONGLONG Lbn = Block + 1; 950 BOOLEAN rc = TRUE; 951 952 Again: 953 954 _SEH2_TRY { 955 956 rc = FsRtlAddLargeMcbEntry( 957 &Mcb->MetaExts, 958 Lbn, 959 Lbn, 960 Length 961 ); 962 963 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 964 965 DbgBreak(); 966 rc = FALSE; 967 } _SEH2_END; 968 969 if (!rc && ++TriedTimes < 10) { 970 Ext2Sleep(TriedTimes * 100); 971 goto Again; 972 } 973 974 DEBUG(DL_EXT, ("Ext2AddMcbMetaExts: Block: %xh-%xh rc=%d Runs=%u\n", Block, 975 Length, rc, FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts))); 976 977 if (rc) { 978 Ext2CheckExtent(&Mcb->MetaExts, Lbn, Lbn, Length, TRUE); 979 } 980 981 return rc; 982 } 983 984 BOOLEAN 985 Ext2RemoveMcbMetaExts ( 986 IN PEXT2_VCB Vcb, 987 IN PEXT2_MCB Mcb, 988 IN ULONG Block, 989 IN ULONG Length 990 ) 991 { 992 ULONG TriedTimes = 0; 993 LONGLONG Lbn = Block + 1; 994 BOOLEAN rc = TRUE; 995 996 Again: 997 998 _SEH2_TRY { 999 1000 FsRtlRemoveLargeMcbEntry( 1001 &Mcb->MetaExts, 1002 Lbn, 1003 Length 1004 ); 1005 1006 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 1007 DbgBreak(); 1008 rc = FALSE; 1009 } _SEH2_END; 1010 1011 if (!rc && ++TriedTimes < 10) { 1012 Ext2Sleep(TriedTimes * 100); 1013 goto Again; 1014 } 1015 1016 DEBUG(DL_EXT, ("Ext2RemoveMcbMetaExts: Block: %xh-%xhh Runs=%u\n", Block, 1017 Length, FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts))); 1018 if (rc) { 1019 Ext2CheckExtent(&Mcb->MetaExts, Lbn, 0, Length, FALSE); 1020 } 1021 1022 return rc; 1023 } 1024 1025 1026 BOOLEAN 1027 Ext2AddBlockExtent( 1028 IN PEXT2_VCB Vcb, 1029 IN PEXT2_MCB Mcb, 1030 IN ULONG Start, 1031 IN ULONG Block, 1032 IN ULONG Number 1033 ) 1034 { 1035 LONGLONG Vbn = 0; 1036 LONGLONG Lbn = 0; 1037 LONGLONG Length = 0; 1038 1039 Vbn = ((LONGLONG) Start) << BLOCK_BITS; 1040 Lbn = ((LONGLONG) Block) << BLOCK_BITS; 1041 Length = ((LONGLONG)Number << BLOCK_BITS); 1042 1043 if (Mcb) { 1044 #if EXT2_DEBUG 1045 ULONG _block = 0, _mapped = 0; 1046 BOOLEAN _rc = Ext2LookupBlockExtent(Vcb, Mcb, Start, &_block, &_mapped); 1047 if (_rc && _block != 0 && (_block != Block)) { 1048 DbgBreak(); 1049 } 1050 #endif 1051 return Ext2AddMcbExtent(Vcb, Mcb, Vbn, Lbn, Length); 1052 1053 } 1054 1055 ASSERT(Start == Block); 1056 return Ext2AddVcbExtent(Vcb, Vbn, Length); 1057 } 1058 1059 1060 BOOLEAN 1061 Ext2LookupBlockExtent( 1062 IN PEXT2_VCB Vcb, 1063 IN PEXT2_MCB Mcb, 1064 IN ULONG Start, 1065 IN PULONG Block, 1066 IN PULONG Mapped 1067 ) 1068 { 1069 LONGLONG Vbn = 0; 1070 LONGLONG Lbn = 0; 1071 LONGLONG Length = 0; 1072 1073 BOOLEAN rc = FALSE; 1074 1075 Vbn = ((LONGLONG) Start) << BLOCK_BITS; 1076 1077 if (Mcb) { 1078 rc = Ext2LookupMcbExtent(Vcb, Mcb, Vbn, &Lbn, &Length); 1079 } else { 1080 rc = Ext2LookupVcbExtent(Vcb, Vbn, &Lbn, &Length); 1081 } 1082 1083 if (rc) { 1084 *Mapped = (ULONG)(Length >> BLOCK_BITS); 1085 if (Lbn != -1 && Length > 0) { 1086 *Block = (ULONG)(Lbn >> BLOCK_BITS); 1087 } else { 1088 *Block = 0; 1089 } 1090 } 1091 1092 return rc; 1093 } 1094 1095 1096 BOOLEAN 1097 Ext2RemoveBlockExtent( 1098 IN PEXT2_VCB Vcb, 1099 IN PEXT2_MCB Mcb, 1100 IN ULONG Start, 1101 IN ULONG Number 1102 ) 1103 { 1104 LONGLONG Vbn = 0; 1105 LONGLONG Length = 0; 1106 BOOLEAN rc; 1107 1108 Vbn = ((LONGLONG) Start) << BLOCK_BITS; 1109 Length = ((LONGLONG)Number << BLOCK_BITS); 1110 1111 if (Mcb) { 1112 rc = Ext2RemoveMcbExtent(Vcb, Mcb, Vbn, Length); 1113 } else { 1114 rc = Ext2RemoveVcbExtent(Vcb, Vbn, Length); 1115 } 1116 1117 return rc; 1118 } 1119 1120 NTSTATUS 1121 Ext2InitializeZone( 1122 IN PEXT2_IRP_CONTEXT IrpContext, 1123 IN PEXT2_VCB Vcb, 1124 IN PEXT2_MCB Mcb 1125 ) 1126 { 1127 NTSTATUS Status = STATUS_SUCCESS; 1128 1129 ULONG Start = 0; 1130 ULONG End; 1131 ULONG Block; 1132 ULONG Mapped; 1133 1134 Ext2ClearAllExtents(&Mcb->Extents); 1135 Ext2ClearAllExtents(&Mcb->MetaExts); 1136 1137 ASSERT(Mcb != NULL); 1138 End = (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS); 1139 1140 while (Start < End) { 1141 1142 Block = Mapped = 0; 1143 1144 /* mapping file offset to ext2 block */ 1145 if (INODE_HAS_EXTENT(&Mcb->Inode)) { 1146 Status = Ext2MapExtent( 1147 IrpContext, 1148 Vcb, 1149 Mcb, 1150 Start, 1151 FALSE, 1152 &Block, 1153 &Mapped 1154 ); 1155 } else { 1156 Status = Ext2MapIndirect( 1157 IrpContext, 1158 Vcb, 1159 Mcb, 1160 Start, 1161 FALSE, 1162 &Block, 1163 &Mapped 1164 ); 1165 } 1166 1167 if (!NT_SUCCESS(Status)) { 1168 goto errorout; 1169 } 1170 1171 /* skip wrong blocks, in case wrongly treating symlink 1172 target names as blocks, silly */ 1173 if (Block >= TOTAL_BLOCKS) { 1174 Block = 0; 1175 } 1176 1177 if (Block) { 1178 if (!Ext2AddBlockExtent(Vcb, Mcb, Start, Block, Mapped)) { 1179 DbgBreak(); 1180 ClearFlag(Mcb->Flags, MCB_ZONE_INITED); 1181 Ext2ClearAllExtents(&Mcb->Extents); 1182 Status = STATUS_INSUFFICIENT_RESOURCES; 1183 goto errorout; 1184 } 1185 DEBUG(DL_MAP, ("Ext2InitializeZone %wZ: Block = %xh Mapped = %xh\n", 1186 &Mcb->FullName, Block, Mapped)); 1187 } 1188 1189 /* Mapped is total number of continous blocks or NULL blocks */ 1190 Start += Mapped; 1191 } 1192 1193 /* set mcb zone as initialized */ 1194 SetLongFlag(Mcb->Flags, MCB_ZONE_INITED); 1195 1196 errorout: 1197 1198 if (!IsZoneInited(Mcb)) { 1199 Ext2ClearAllExtents(&Mcb->Extents); 1200 Ext2ClearAllExtents(&Mcb->MetaExts); 1201 } 1202 1203 return Status; 1204 } 1205 1206 NTSTATUS 1207 Ext2BuildExtents( 1208 IN PEXT2_IRP_CONTEXT IrpContext, 1209 IN PEXT2_VCB Vcb, 1210 IN PEXT2_MCB Mcb, 1211 IN ULONGLONG Offset, 1212 IN ULONG Size, 1213 IN BOOLEAN bAlloc, 1214 OUT PEXT2_EXTENT * Chain 1215 ) 1216 { 1217 ULONG Start, End; 1218 ULONG Total = 0; 1219 1220 LONGLONG Lba = 0; 1221 NTSTATUS Status = STATUS_SUCCESS; 1222 1223 PEXT2_EXTENT Extent = NULL; 1224 PEXT2_EXTENT List = *Chain = NULL; 1225 1226 if (!IsZoneInited(Mcb)) { 1227 Status = Ext2InitializeZone(IrpContext, Vcb, Mcb); 1228 if (!NT_SUCCESS(Status)) { 1229 DbgBreak(); 1230 } 1231 } 1232 1233 if ((IrpContext && IrpContext->Irp) && 1234 ((IrpContext->Irp->Flags & IRP_NOCACHE) || 1235 (IrpContext->Irp->Flags & IRP_PAGING_IO))) { 1236 Size = (Size + SECTOR_SIZE - 1) & (~(SECTOR_SIZE - 1)); 1237 } 1238 1239 Start = (ULONG)(Offset >> BLOCK_BITS); 1240 End = (ULONG)((Size + Offset + BLOCK_SIZE - 1) >> BLOCK_BITS); 1241 1242 if (End > (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS) ) { 1243 End = (ULONG)((Mcb->Inode.i_size + BLOCK_SIZE - 1) >> BLOCK_BITS); 1244 } 1245 1246 while (Size > 0 && Start < End) { 1247 1248 ULONG Mapped = 0; 1249 ULONG Length = 0; 1250 ULONG Block = 0; 1251 1252 BOOLEAN rc = FALSE; 1253 1254 /* try to map file offset to ext2 block upon Extents cache */ 1255 if (IsZoneInited(Mcb)) { 1256 rc = Ext2LookupBlockExtent( 1257 Vcb, 1258 Mcb, 1259 Start, 1260 &Block, 1261 &Mapped); 1262 1263 if (!rc) { 1264 /* we likely get a sparse file here */ 1265 Mapped = 1; 1266 Block = 0; 1267 } 1268 } 1269 1270 /* try to BlockMap in case failed to access Extents cache */ 1271 if (!IsZoneInited(Mcb) || (bAlloc && Block == 0)) { 1272 1273 Status = Ext2BlockMap( 1274 IrpContext, 1275 Vcb, 1276 Mcb, 1277 Start, 1278 bAlloc, 1279 &Block, 1280 &Mapped 1281 ); 1282 if (!NT_SUCCESS(Status)) { 1283 break; 1284 } 1285 1286 /* skip wrong blocks, in case wrongly treating symlink 1287 target names as blocks, silly */ 1288 if (Block >= TOTAL_BLOCKS) { 1289 Block = 0; 1290 } 1291 1292 /* add new allocated blocks to Mcb zone */ 1293 if (IsZoneInited(Mcb) && Block) { 1294 if (!Ext2AddBlockExtent(Vcb, Mcb, Start, Block, Mapped)) { 1295 DbgBreak(); 1296 ClearFlag(Mcb->Flags, MCB_ZONE_INITED); 1297 Ext2ClearAllExtents(&Mcb->Extents); 1298 } 1299 } 1300 } 1301 1302 /* calculate i/o extent */ 1303 Lba = ((LONGLONG)Block << BLOCK_BITS) + Offset - ((LONGLONG)Start << BLOCK_BITS); 1304 Length = (ULONG)(((LONGLONG)(Start + Mapped) << BLOCK_BITS) - Offset); 1305 if (Length > Size) { 1306 Length = Size; 1307 } 1308 1309 if (0 == Length) { 1310 DbgBreak(); 1311 break; 1312 } 1313 1314 Start += Mapped; 1315 Offset = (ULONGLONG)Start << BLOCK_BITS; 1316 1317 if (Block != 0) { 1318 1319 if (List && List->Lba + List->Length == Lba) { 1320 1321 /* it's continuous upon previous Extent */ 1322 List->Length += Length; 1323 1324 } else { 1325 1326 /* have to allocate a new Extent */ 1327 Extent = Ext2AllocateExtent(); 1328 if (!Extent) { 1329 Status = STATUS_INSUFFICIENT_RESOURCES; 1330 DbgBreak(); 1331 break; 1332 } 1333 1334 Extent->Lba = Lba; 1335 Extent->Length = Length; 1336 Extent->Offset = Total; 1337 1338 /* insert new Extent to chain */ 1339 if (List) { 1340 List->Next = Extent; 1341 List = Extent; 1342 } else { 1343 *Chain = List = Extent; 1344 } 1345 } 1346 } else { 1347 if (bAlloc) { 1348 DbgBreak(); 1349 } 1350 } 1351 1352 Total += Length; 1353 Size -= Length; 1354 } 1355 1356 return Status; 1357 } 1358 1359 1360 BOOLEAN 1361 Ext2BuildName( 1362 IN OUT PUNICODE_STRING Target, 1363 IN PUNICODE_STRING File, 1364 IN PUNICODE_STRING Parent 1365 ) 1366 { 1367 USHORT Length = 0; 1368 USHORT ParentLen = 0; 1369 BOOLEAN bBackslash = TRUE; 1370 1371 /* free the original buffer */ 1372 if (Target->Buffer) { 1373 DEC_MEM_COUNT(PS_MCB_NAME, Target->Buffer, Target->MaximumLength); 1374 Ext2FreePool(Target->Buffer, EXT2_FNAME_MAGIC); 1375 Target->Length = Target->MaximumLength = 0; 1376 } 1377 1378 /* check the parent directory's name and backslash */ 1379 if (Parent && Parent->Buffer && Parent->Length > 0) { 1380 ParentLen = Parent->Length / sizeof(WCHAR); 1381 if (Parent->Buffer[ParentLen - 1] == L'\\') { 1382 bBackslash = FALSE; 1383 } 1384 } 1385 1386 if (Parent == NULL || File->Buffer[0] == L'\\') { 1387 /* must be root inode */ 1388 ASSERT(ParentLen == 0); 1389 bBackslash = FALSE; 1390 } 1391 1392 /* allocate and initialize new name buffer */ 1393 Length = File->Length; 1394 Length += (ParentLen + (bBackslash ? 1 : 0)) * sizeof(WCHAR); 1395 1396 Target->Buffer = Ext2AllocatePool( 1397 PagedPool, 1398 Length + 2, 1399 EXT2_FNAME_MAGIC 1400 ); 1401 1402 if (!Target->Buffer) { 1403 DEBUG(DL_ERR, ( "Ex2BuildName: failed to allocate name bufer.\n")); 1404 return FALSE; 1405 } 1406 RtlZeroMemory(Target->Buffer, Length + 2); 1407 1408 if (ParentLen) { 1409 RtlCopyMemory(&Target->Buffer[0], 1410 Parent->Buffer, 1411 ParentLen * sizeof(WCHAR)); 1412 } 1413 1414 if (bBackslash) { 1415 Target->Buffer[ParentLen++] = L'\\'; 1416 } 1417 1418 RtlCopyMemory( &Target->Buffer[ParentLen], 1419 File->Buffer, 1420 File->Length); 1421 1422 INC_MEM_COUNT(PS_MCB_NAME, Target->Buffer, Length + 2); 1423 Target->Length = Length; 1424 Target->MaximumLength = Length + 2; 1425 1426 return TRUE; 1427 } 1428 1429 PEXT2_MCB 1430 Ext2AllocateMcb ( 1431 IN PEXT2_VCB Vcb, 1432 IN PUNICODE_STRING FileName, 1433 IN PUNICODE_STRING Parent, 1434 IN ULONG FileAttr 1435 ) 1436 { 1437 PEXT2_MCB Mcb = NULL; 1438 NTSTATUS Status = STATUS_SUCCESS; 1439 1440 /* need wake the reaper thread if there are many Mcb allocated */ 1441 if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 4)) { 1442 KeSetEvent(&Ext2Global->McbReaper.Wait, 0, FALSE); 1443 } 1444 1445 /* allocate Mcb from LookasideList */ 1446 Mcb = (PEXT2_MCB) (ExAllocateFromNPagedLookasideList( 1447 &(Ext2Global->Ext2McbLookasideList))); 1448 1449 if (Mcb == NULL) { 1450 return NULL; 1451 } 1452 1453 /* initialize Mcb header */ 1454 RtlZeroMemory(Mcb, sizeof(EXT2_MCB)); 1455 Mcb->Identifier.Type = EXT2MCB; 1456 Mcb->Identifier.Size = sizeof(EXT2_MCB); 1457 Mcb->FileAttr = FileAttr; 1458 1459 Mcb->Inode.i_priv = (PVOID)Mcb; 1460 Mcb->Inode.i_sb = &Vcb->sb; 1461 1462 /* initialize Mcb names */ 1463 if (FileName) { 1464 1465 #if EXT2_DEBUG 1466 if ( FileName->Length == 2 && 1467 FileName->Buffer[0] == L'\\') { 1468 DEBUG(DL_RES, ( "Ext2AllocateMcb: Root Mcb is to be created !\n")); 1469 } 1470 1471 if ( FileName->Length == 2 && 1472 FileName->Buffer[0] == L'.') { 1473 DbgBreak(); 1474 } 1475 1476 if ( FileName->Length == 4 && 1477 FileName->Buffer[0] == L'.' && 1478 FileName->Buffer[1] == L'.' ) { 1479 DbgBreak(); 1480 } 1481 #endif 1482 1483 if (( FileName->Length >= 4 && FileName->Buffer[0] == L'.') && 1484 ((FileName->Length == 4 && FileName->Buffer[1] != L'.') || 1485 FileName->Length >= 6 )) { 1486 SetFlag(Mcb->FileAttr, FILE_ATTRIBUTE_HIDDEN); 1487 } 1488 1489 if (!Ext2BuildName(&Mcb->ShortName, FileName, NULL)) { 1490 goto errorout; 1491 } 1492 if (!Ext2BuildName(&Mcb->FullName, FileName, Parent)) { 1493 goto errorout; 1494 } 1495 } 1496 1497 /* initialize Mcb Extents, it will raise an expcetion if failed */ 1498 _SEH2_TRY { 1499 FsRtlInitializeLargeMcb(&(Mcb->Extents), NonPagedPool); 1500 FsRtlInitializeLargeMcb(&(Mcb->MetaExts), NonPagedPool); 1501 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 1502 Status = STATUS_INSUFFICIENT_RESOURCES; 1503 DbgBreak(); 1504 } _SEH2_END; 1505 1506 if (!NT_SUCCESS(Status)) { 1507 goto errorout; 1508 } 1509 1510 INC_MEM_COUNT(PS_MCB, Mcb, sizeof(EXT2_MCB)); 1511 DEBUG(DL_INF, ( "Ext2AllocateMcb: Mcb %wZ created.\n", &Mcb->FullName)); 1512 1513 return Mcb; 1514 1515 errorout: 1516 1517 if (Mcb) { 1518 1519 if (Mcb->ShortName.Buffer) { 1520 DEC_MEM_COUNT(PS_MCB_NAME, Mcb->ShortName.Buffer, 1521 Mcb->ShortName.MaximumLength); 1522 Ext2FreePool(Mcb->ShortName.Buffer, EXT2_FNAME_MAGIC); 1523 } 1524 1525 if (Mcb->FullName.Buffer) { 1526 DEC_MEM_COUNT(PS_MCB_NAME, Mcb->FullName.Buffer, 1527 Mcb->FullName.MaximumLength); 1528 Ext2FreePool(Mcb->FullName.Buffer, EXT2_FNAME_MAGIC); 1529 } 1530 1531 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2McbLookasideList), Mcb); 1532 } 1533 1534 return NULL; 1535 } 1536 1537 VOID 1538 Ext2FreeMcb (IN PEXT2_VCB Vcb, IN PEXT2_MCB Mcb) 1539 { 1540 #ifndef __REACTOS__ 1541 PEXT2_MCB Parent = Mcb->Parent; 1542 #endif 1543 1544 ASSERT(Mcb != NULL); 1545 1546 ASSERT((Mcb->Identifier.Type == EXT2MCB) && 1547 (Mcb->Identifier.Size == sizeof(EXT2_MCB))); 1548 1549 if ((Mcb->Identifier.Type != EXT2MCB) || 1550 (Mcb->Identifier.Size != sizeof(EXT2_MCB))) { 1551 return; 1552 } 1553 1554 DEBUG(DL_INF, ( "Ext2FreeMcb: Mcb %wZ will be freed.\n", &Mcb->FullName)); 1555 1556 if (IsMcbSymLink(Mcb) && Mcb->Target) { 1557 Ext2DerefMcb(Mcb->Target); 1558 } 1559 1560 if (FsRtlNumberOfRunsInLargeMcb(&Mcb->Extents)) { 1561 DEBUG(DL_EXT, ("List data extents for: %wZ\n", &Mcb->FullName)); 1562 Ext2ListExtents(&Mcb->Extents); 1563 } 1564 FsRtlUninitializeLargeMcb(&(Mcb->Extents)); 1565 if (FsRtlNumberOfRunsInLargeMcb(&Mcb->MetaExts)) { 1566 DEBUG(DL_EXT, ("List meta extents for: %wZ\n", &Mcb->FullName)); 1567 Ext2ListExtents(&Mcb->MetaExts); 1568 } 1569 FsRtlUninitializeLargeMcb(&(Mcb->MetaExts)); 1570 ClearLongFlag(Mcb->Flags, MCB_ZONE_INITED); 1571 1572 if (Mcb->ShortName.Buffer) { 1573 DEC_MEM_COUNT(PS_MCB_NAME, Mcb->ShortName.Buffer, 1574 Mcb->ShortName.MaximumLength); 1575 Ext2FreePool(Mcb->ShortName.Buffer, EXT2_FNAME_MAGIC); 1576 } 1577 1578 if (Mcb->FullName.Buffer) { 1579 DEC_MEM_COUNT(PS_MCB_NAME, Mcb->FullName.Buffer, 1580 Mcb->FullName.MaximumLength); 1581 Ext2FreePool(Mcb->FullName.Buffer, EXT2_FNAME_MAGIC); 1582 } 1583 1584 /* free dentry */ 1585 if (Mcb->de) { 1586 Ext2FreeEntry(Mcb->de); 1587 } 1588 1589 Mcb->Identifier.Type = 0; 1590 Mcb->Identifier.Size = 0; 1591 1592 ExFreeToNPagedLookasideList(&(Ext2Global->Ext2McbLookasideList), Mcb); 1593 DEC_MEM_COUNT(PS_MCB, Mcb, sizeof(EXT2_MCB)); 1594 } 1595 1596 1597 PEXT2_MCB 1598 Ext2SearchMcb( 1599 PEXT2_VCB Vcb, 1600 PEXT2_MCB Parent, 1601 PUNICODE_STRING FileName 1602 ) 1603 { 1604 BOOLEAN LockAcquired = FALSE; 1605 PEXT2_MCB Mcb = NULL; 1606 1607 _SEH2_TRY { 1608 ExAcquireResourceSharedLite(&Vcb->McbLock, TRUE); 1609 LockAcquired = TRUE; 1610 Mcb = Ext2SearchMcbWithoutLock(Parent, FileName); 1611 } _SEH2_FINALLY { 1612 if (LockAcquired) { 1613 ExReleaseResourceLite(&Vcb->McbLock); 1614 } 1615 } _SEH2_END; 1616 1617 return Mcb; 1618 } 1619 1620 1621 PEXT2_MCB 1622 Ext2SearchMcbWithoutLock( 1623 PEXT2_MCB Parent, 1624 PUNICODE_STRING FileName 1625 ) 1626 { 1627 PEXT2_MCB TmpMcb = NULL; 1628 1629 DEBUG(DL_RES, ("Ext2SearchMcb: %wZ\n", FileName)); 1630 1631 _SEH2_TRY { 1632 1633 Ext2ReferMcb(Parent); 1634 1635 if (Ext2IsDot(FileName)) { 1636 TmpMcb = Parent; 1637 Ext2ReferMcb(Parent); 1638 _SEH2_LEAVE; 1639 } 1640 1641 if (Ext2IsDotDot(FileName)) { 1642 if (IsMcbRoot(Parent)) { 1643 TmpMcb = Parent; 1644 } else { 1645 TmpMcb = Parent->Parent; 1646 } 1647 if (TmpMcb) { 1648 Ext2ReferMcb(TmpMcb); 1649 } 1650 _SEH2_LEAVE; 1651 } 1652 1653 if (IsMcbSymLink(Parent)) { 1654 if (Parent->Target) { 1655 TmpMcb = Parent->Target->Child; 1656 ASSERT(!IsMcbSymLink(Parent->Target)); 1657 } else { 1658 TmpMcb = NULL; 1659 _SEH2_LEAVE; 1660 } 1661 } else { 1662 TmpMcb = Parent->Child; 1663 } 1664 1665 while (TmpMcb) { 1666 1667 if (!RtlCompareUnicodeString( 1668 &(TmpMcb->ShortName), 1669 FileName, TRUE )) { 1670 Ext2ReferMcb(TmpMcb); 1671 break; 1672 } 1673 1674 TmpMcb = TmpMcb->Next; 1675 } 1676 1677 } _SEH2_FINALLY { 1678 1679 Ext2DerefMcb(Parent); 1680 } _SEH2_END; 1681 1682 return TmpMcb; 1683 } 1684 1685 VOID 1686 Ext2InsertMcb ( 1687 PEXT2_VCB Vcb, 1688 PEXT2_MCB Parent, 1689 PEXT2_MCB Child 1690 ) 1691 { 1692 BOOLEAN LockAcquired = FALSE; 1693 PEXT2_MCB Mcb = NULL; 1694 1695 _SEH2_TRY { 1696 1697 ExAcquireResourceExclusiveLite( 1698 &Vcb->McbLock, 1699 TRUE ); 1700 LockAcquired = TRUE; 1701 1702 /* use it's target if it's a symlink */ 1703 if (IsMcbSymLink(Parent)) { 1704 Parent = Parent->Target; 1705 ASSERT(!IsMcbSymLink(Parent)); 1706 } 1707 1708 Mcb = Parent->Child; 1709 while (Mcb) { 1710 if (Mcb == Child) { 1711 break; 1712 } 1713 Mcb = Mcb->Next; 1714 } 1715 1716 if (Mcb) { 1717 /* already attached in the list */ 1718 DEBUG(DL_ERR, ( "Ext2InsertMcb: Child Mcb is alreay attached.\n")); 1719 if (!IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { 1720 SetLongFlag(Child->Flags, MCB_ENTRY_TREE); 1721 DEBUG(DL_ERR, ( "Ext2InsertMcb: Child Mcb's flag isn't set.\n")); 1722 } 1723 1724 DbgBreak(); 1725 1726 } else { 1727 1728 /* insert this Mcb into the head */ 1729 Child->Next = Parent->Child; 1730 Parent->Child = Child; 1731 Child->Parent = Parent; 1732 Child->de->d_parent = Parent->de; 1733 Ext2ReferMcb(Parent); 1734 SetLongFlag(Child->Flags, MCB_ENTRY_TREE); 1735 } 1736 1737 } _SEH2_FINALLY { 1738 1739 if (LockAcquired) { 1740 ExReleaseResourceLite(&Vcb->McbLock); 1741 } 1742 } _SEH2_END; 1743 } 1744 1745 BOOLEAN 1746 Ext2RemoveMcb ( 1747 PEXT2_VCB Vcb, 1748 PEXT2_MCB Mcb 1749 ) 1750 { 1751 PEXT2_MCB TmpMcb = NULL; 1752 BOOLEAN LockAcquired = FALSE; 1753 BOOLEAN bLinked = FALSE; 1754 1755 _SEH2_TRY { 1756 1757 ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); 1758 LockAcquired = TRUE; 1759 1760 if (Mcb->Parent) { 1761 1762 if (Mcb->Parent->Child == Mcb) { 1763 Mcb->Parent->Child = Mcb->Next; 1764 bLinked = TRUE; 1765 } else { 1766 TmpMcb = Mcb->Parent->Child; 1767 1768 while (TmpMcb && TmpMcb->Next != Mcb) { 1769 TmpMcb = TmpMcb->Next; 1770 } 1771 1772 if (TmpMcb) { 1773 TmpMcb->Next = Mcb->Next; 1774 bLinked = TRUE; 1775 } else { 1776 /* we got errors: link broken */ 1777 } 1778 } 1779 1780 if (bLinked) { 1781 if (IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { 1782 DEBUG(DL_RES, ("Mcb %p %wZ removed from Mcb %p %wZ\n", Mcb, 1783 &Mcb->FullName, Mcb->Parent, &Mcb->Parent->FullName)); 1784 Ext2DerefMcb(Mcb->Parent); 1785 ClearLongFlag(Mcb->Flags, MCB_ENTRY_TREE); 1786 } else { 1787 DbgBreak(); 1788 } 1789 } else { 1790 if (IsFlagOn(Mcb->Flags, MCB_ENTRY_TREE)) { 1791 ClearLongFlag(Mcb->Flags, MCB_ENTRY_TREE); 1792 } 1793 DbgBreak(); 1794 } 1795 Mcb->Parent = NULL; 1796 Mcb->de->d_parent = NULL; 1797 } 1798 1799 } _SEH2_FINALLY { 1800 1801 if (LockAcquired) { 1802 ExReleaseResourceLite(&Vcb->McbLock); 1803 } 1804 } _SEH2_END; 1805 1806 return TRUE; 1807 } 1808 1809 VOID 1810 Ext2CleanupAllMcbs(PEXT2_VCB Vcb) 1811 { 1812 BOOLEAN LockAcquired = FALSE; 1813 PEXT2_MCB Mcb = NULL; 1814 1815 _SEH2_TRY { 1816 1817 ExAcquireResourceExclusiveLite( 1818 &Vcb->McbLock, 1819 TRUE ); 1820 LockAcquired = TRUE; 1821 1822 #ifdef __REACTOS__ 1823 while ((Mcb = Ext2FirstUnusedMcb(Vcb, TRUE, Vcb->NumOfMcb)) != 0) { 1824 #else 1825 while (Mcb = Ext2FirstUnusedMcb(Vcb, TRUE, Vcb->NumOfMcb)) { 1826 #endif 1827 while (Mcb) { 1828 PEXT2_MCB Next = Mcb->Next; 1829 if (IsMcbSymLink(Mcb)) { 1830 Mcb->Target = NULL; 1831 } 1832 Ext2FreeMcb(Vcb, Mcb); 1833 Mcb = Next; 1834 } 1835 } 1836 Ext2FreeMcb(Vcb, Vcb->McbTree); 1837 Vcb->McbTree = NULL; 1838 1839 } _SEH2_FINALLY { 1840 1841 if (LockAcquired) { 1842 ExReleaseResourceLite(&Vcb->McbLock); 1843 } 1844 } _SEH2_END; 1845 } 1846 1847 BOOLEAN 1848 Ext2CheckSetBlock(PEXT2_IRP_CONTEXT IrpContext, PEXT2_VCB Vcb, LONGLONG Block) 1849 { 1850 PEXT2_GROUP_DESC gd; 1851 struct buffer_head *gb = NULL; 1852 struct buffer_head *bh = NULL; 1853 ULONG group, dwBlk, Length; 1854 RTL_BITMAP bitmap; 1855 BOOLEAN bModified = FALSE; 1856 1857 group = (ULONG)(Block - EXT2_FIRST_DATA_BLOCK) / BLOCKS_PER_GROUP; 1858 dwBlk = (ULONG)(Block - EXT2_FIRST_DATA_BLOCK) % BLOCKS_PER_GROUP; 1859 1860 gd = ext4_get_group_desc(&Vcb->sb, group, &gb); 1861 if (!gd) { 1862 return FALSE; 1863 } 1864 bh = sb_getblk(&Vcb->sb, ext4_block_bitmap(&Vcb->sb, gd)); 1865 1866 if (group == Vcb->sbi.s_groups_count - 1) { 1867 Length = (ULONG)(TOTAL_BLOCKS % BLOCKS_PER_GROUP); 1868 1869 /* s_blocks_count is integer multiple of s_blocks_per_group */ 1870 if (Length == 0) 1871 Length = BLOCKS_PER_GROUP; 1872 } else { 1873 Length = BLOCKS_PER_GROUP; 1874 } 1875 1876 if (dwBlk >= Length) { 1877 fini_bh(&gb); 1878 fini_bh(&bh); 1879 return FALSE; 1880 } 1881 1882 1883 RtlInitializeBitMap(&bitmap, (PULONG)bh->b_data, Length); 1884 1885 if (RtlCheckBit(&bitmap, dwBlk) == 0) { 1886 DbgBreak(); 1887 RtlSetBits(&bitmap, dwBlk, 1); 1888 bModified = TRUE; 1889 mark_buffer_dirty(bh); 1890 } 1891 1892 fini_bh(&gb); 1893 fini_bh(&bh); 1894 1895 return (!bModified); 1896 } 1897 1898 BOOLEAN 1899 Ext2CheckBitmapConsistency(PEXT2_IRP_CONTEXT IrpContext, PEXT2_VCB Vcb) 1900 { 1901 ULONG i, j, InodeBlocks; 1902 1903 for (i = 0; i < Vcb->sbi.s_groups_count; i++) { 1904 1905 PEXT2_GROUP_DESC gd; 1906 struct buffer_head *bh = NULL; 1907 1908 gd = ext4_get_group_desc(&Vcb->sb, i, &bh); 1909 if (!gd) 1910 continue; 1911 Ext2CheckSetBlock(IrpContext, Vcb, ext4_block_bitmap(&Vcb->sb, gd)); 1912 Ext2CheckSetBlock(IrpContext, Vcb, ext4_inode_bitmap(&Vcb->sb, gd)); 1913 1914 1915 if (i == Vcb->sbi.s_groups_count - 1) { 1916 InodeBlocks = ((INODES_COUNT % INODES_PER_GROUP) * 1917 Vcb->InodeSize + Vcb->BlockSize - 1) / 1918 (Vcb->BlockSize); 1919 } else { 1920 InodeBlocks = (INODES_PER_GROUP * Vcb->InodeSize + 1921 Vcb->BlockSize - 1) / (Vcb->BlockSize); 1922 } 1923 1924 for (j = 0; j < InodeBlocks; j++ ) 1925 Ext2CheckSetBlock(IrpContext, Vcb, ext4_inode_table(&Vcb->sb, gd) + j); 1926 1927 fini_bh(&bh); 1928 } 1929 1930 return TRUE; 1931 } 1932 1933 /* Ext2Global->Resource should be already acquired */ 1934 VOID 1935 Ext2InsertVcb(PEXT2_VCB Vcb) 1936 { 1937 InsertTailList(&(Ext2Global->VcbList), &Vcb->Next); 1938 } 1939 1940 1941 /* Ext2Global->Resource should be already acquired */ 1942 VOID 1943 Ext2RemoveVcb(PEXT2_VCB Vcb) 1944 { 1945 RemoveEntryList(&Vcb->Next); 1946 InitializeListHead(&Vcb->Next); 1947 } 1948 1949 NTSTATUS 1950 Ext2QueryVolumeParams(IN PEXT2_VCB Vcb, IN PUNICODE_STRING Params) 1951 { 1952 NTSTATUS Status; 1953 RTL_QUERY_REGISTRY_TABLE QueryTable[2]; 1954 1955 UNICODE_STRING UniName; 1956 PUSHORT UniBuffer = NULL; 1957 1958 USHORT UUID[50]; 1959 1960 int i; 1961 int len = 0; 1962 1963 /* zero params */ 1964 RtlZeroMemory(Params, sizeof(UNICODE_STRING)); 1965 1966 /* constructing volume UUID name */ 1967 memset(UUID, 0, sizeof(USHORT) * 50); 1968 for (i=0; i < 16; i++) { 1969 if (i == 0) { 1970 swprintf((wchar_t *)&UUID[len], L"{%2.2X",Vcb->SuperBlock->s_uuid[i]); 1971 len += 3; 1972 } else if (i == 15) { 1973 swprintf((wchar_t *)&UUID[len], L"-%2.2X}", Vcb->SuperBlock->s_uuid[i]); 1974 len +=4; 1975 } else { 1976 swprintf((wchar_t *)&UUID[len], L"-%2.2X", Vcb->SuperBlock->s_uuid[i]); 1977 len += 3; 1978 } 1979 } 1980 1981 /* allocating memory for UniBuffer */ 1982 UniBuffer = Ext2AllocatePool(PagedPool, 1024, EXT2_PARAM_MAGIC); 1983 if (NULL == UniBuffer) { 1984 Status = STATUS_INSUFFICIENT_RESOURCES; 1985 goto errorout; 1986 } 1987 RtlZeroMemory(UniBuffer, 1024); 1988 1989 /* querying volume parameter string */ 1990 RtlZeroMemory(&QueryTable[0], sizeof(RTL_QUERY_REGISTRY_TABLE) * 2); 1991 QueryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_QUERY_REGISTRY_REQUIRED; 1992 QueryTable[0].Name = UUID; 1993 QueryTable[0].EntryContext = &(UniName); 1994 UniName.MaximumLength = 1024; 1995 UniName.Length = 0; 1996 UniName.Buffer = UniBuffer; 1997 1998 Status = RtlQueryRegistryValues( 1999 RTL_REGISTRY_ABSOLUTE, 2000 Ext2Global->RegistryPath.Buffer, 2001 &QueryTable[0], 2002 NULL, 2003 NULL 2004 ); 2005 2006 if (!NT_SUCCESS(Status)) { 2007 goto errorout; 2008 } 2009 2010 errorout: 2011 2012 if (NT_SUCCESS(Status)) { 2013 *Params = UniName; 2014 } else { 2015 if (UniBuffer) { 2016 Ext2FreePool(UniBuffer, EXT2_PARAM_MAGIC); 2017 } 2018 } 2019 2020 return Status; 2021 } 2022 2023 VOID 2024 Ext2ParseRegistryVolumeParams( 2025 IN PUNICODE_STRING Params, 2026 OUT PEXT2_VOLUME_PROPERTY3 Property 2027 ) 2028 { 2029 WCHAR Codepage[CODEPAGE_MAXLEN]; 2030 WCHAR Prefix[HIDINGPAT_LEN]; 2031 WCHAR Suffix[HIDINGPAT_LEN]; 2032 USHORT MountPoint[4]; 2033 UCHAR DrvLetter[4]; 2034 WCHAR wUID[8], wGID[8], wEUID[8], wEGID[8]; 2035 CHAR sUID[8], sGID[8], sEUID[8], sEGID[8]; 2036 2037 BOOLEAN bWriteSupport = FALSE, 2038 bCheckBitmap = FALSE, 2039 bCodeName = FALSE, 2040 bMountPoint = FALSE; 2041 BOOLEAN bUID = 0, bGID = 0, bEUID = 0, bEGID = 0; 2042 2043 struct { 2044 PWCHAR Name; /* parameters name */ 2045 PBOOLEAN bExist; /* is it contained in params */ 2046 USHORT Length; /* parameter value length */ 2047 PWCHAR uValue; /* value buffer in unicode */ 2048 PCHAR aValue; /* value buffer in ansi */ 2049 } ParamPattern[] = { 2050 /* writing support */ 2051 {READING_ONLY, &Property->bReadonly, 0, NULL, NULL}, 2052 {WRITING_SUPPORT, &bWriteSupport, 0, NULL, NULL}, 2053 {EXT3_FORCEWRITING, &Property->bExt3Writable, 0, NULL, NULL}, 2054 2055 /* need check bitmap */ 2056 {CHECKING_BITMAP, &bCheckBitmap, 0, NULL, NULL}, 2057 /* codepage */ 2058 {CODEPAGE_NAME, &bCodeName, CODEPAGE_MAXLEN, 2059 &Codepage[0], Property->Codepage}, 2060 /* filter prefix and suffix */ 2061 {HIDING_PREFIX, &Property->bHidingPrefix, HIDINGPAT_LEN, 2062 &Prefix[0], Property->sHidingPrefix}, 2063 {HIDING_SUFFIX, &Property->bHidingSuffix, HIDINGPAT_LEN, 2064 &Suffix[0], Property->sHidingSuffix}, 2065 {MOUNT_POINT, &bMountPoint, 4, 2066 &MountPoint[0], &DrvLetter[0]}, 2067 2068 {UID, &bUID, 8, &wUID[0], &sUID[0],}, 2069 {GID, &bGID, 8, &wGID[0], &sGID[0]}, 2070 {EUID, &bEUID, 8, &wEUID[0], &sEUID[0]}, 2071 {EGID, &bEGID, 8, &wEGID[0], &sEGID[0]}, 2072 2073 /* end */ 2074 {NULL, NULL, 0, NULL} 2075 }; 2076 2077 USHORT i, j, k; 2078 2079 #ifdef __REACTOS__ 2080 RtlZeroMemory(Codepage, sizeof(WCHAR) * CODEPAGE_MAXLEN); 2081 RtlZeroMemory(Prefix, sizeof(WCHAR) * HIDINGPAT_LEN); 2082 RtlZeroMemory(Suffix, sizeof(WCHAR) * HIDINGPAT_LEN); 2083 #else 2084 RtlZeroMemory(Codepage, CODEPAGE_MAXLEN); 2085 RtlZeroMemory(Prefix, HIDINGPAT_LEN); 2086 RtlZeroMemory(Suffix, HIDINGPAT_LEN); 2087 #endif 2088 RtlZeroMemory(MountPoint, sizeof(USHORT) * 4); 2089 RtlZeroMemory(DrvLetter, sizeof(CHAR) * 4); 2090 2091 RtlZeroMemory(Property, sizeof(EXT2_VOLUME_PROPERTY3)); 2092 Property->Magic = EXT2_VOLUME_PROPERTY_MAGIC; 2093 Property->Command = APP_CMD_SET_PROPERTY3; 2094 2095 for (i=0; ParamPattern[i].Name != NULL; i++) { 2096 2097 UNICODE_STRING Name1=*Params, Name2; 2098 RtlInitUnicodeString(&Name2, ParamPattern[i].Name); 2099 *ParamPattern[i].bExist = FALSE; 2100 2101 for (j=0; j * sizeof(WCHAR) + Name2.Length <= Params->Length ; j++) { 2102 2103 Name1.MaximumLength = Params->Length - j * sizeof(WCHAR); 2104 Name1.Length = Name2.Length; 2105 Name1.Buffer = &Params->Buffer[j]; 2106 2107 if (!RtlCompareUnicodeString(&Name1, &Name2, TRUE)) { 2108 if (j * sizeof(WCHAR) + Name2.Length == Params->Length || 2109 Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L';' || 2110 Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L',' ) { 2111 *(ParamPattern[i].bExist) = TRUE; 2112 } else if ((j * 2 + Name2.Length < Params->Length + 2) || 2113 (Name1.Buffer[Name2.Length/sizeof(WCHAR)] == L'=' )) { 2114 j += Name2.Length/sizeof(WCHAR) + 1; 2115 k = 0; 2116 while ( j + k < Params->Length/2 && 2117 k < ParamPattern[i].Length && 2118 Params->Buffer[j+k] != L';' && 2119 Params->Buffer[j+k] != L',' ) { 2120 #ifdef __REACTOS__ 2121 ParamPattern[i].uValue[k] = Params->Buffer[j + k]; 2122 k++; 2123 #else 2124 ParamPattern[i].uValue[k] = Params->Buffer[j + k++]; 2125 #endif 2126 } 2127 if (k) { 2128 NTSTATUS status; 2129 ANSI_STRING AnsiName; 2130 AnsiName.Length = 0; 2131 AnsiName.MaximumLength =ParamPattern[i].Length; 2132 AnsiName.Buffer = ParamPattern[i].aValue; 2133 2134 Name2.Buffer = ParamPattern[i].uValue; 2135 Name2.MaximumLength = Name2.Length = k * sizeof(WCHAR); 2136 status = RtlUnicodeStringToAnsiString( 2137 &AnsiName, &Name2, FALSE); 2138 if (NT_SUCCESS(status)) { 2139 *(ParamPattern[i].bExist) = TRUE; 2140 } else { 2141 *ParamPattern[i].bExist = FALSE; 2142 } 2143 } 2144 } 2145 break; 2146 } 2147 } 2148 } 2149 2150 if (bMountPoint) { 2151 Property->DrvLetter = DrvLetter[0]; 2152 Property->DrvLetter |= 0x80; 2153 } 2154 2155 if (bUID && bGID) { 2156 SetFlag(Property->Flags2, EXT2_VPROP3_USERIDS); 2157 sUID[7] = sGID[7] = sEUID[7] = sEGID[7] = 0; 2158 Property->uid = (USHORT)atoi(sUID); 2159 Property->gid = (USHORT)atoi(sGID); 2160 if (bEUID) { 2161 Property->euid = (USHORT)atoi(sEUID); 2162 Property->egid = (USHORT)atoi(sEGID); 2163 Property->EIDS = TRUE; 2164 } else { 2165 Property->EIDS = FALSE; 2166 } 2167 } else { 2168 ClearFlag(Property->Flags2, EXT2_VPROP3_USERIDS); 2169 } 2170 } 2171 2172 NTSTATUS 2173 Ext2PerformRegistryVolumeParams(IN PEXT2_VCB Vcb) 2174 { 2175 NTSTATUS Status; 2176 UNICODE_STRING VolumeParams; 2177 2178 Status = Ext2QueryVolumeParams(Vcb, &VolumeParams); 2179 if (NT_SUCCESS(Status)) { 2180 2181 /* set Vcb settings from registery */ 2182 EXT2_VOLUME_PROPERTY3 Property; 2183 Ext2ParseRegistryVolumeParams(&VolumeParams, &Property); 2184 Ext2ProcessVolumeProperty(Vcb, &Property, sizeof(Property)); 2185 2186 } else { 2187 2188 /* don't support auto mount */ 2189 if (IsFlagOn(Ext2Global->Flags, EXT2_AUTO_MOUNT)) { 2190 Status = STATUS_SUCCESS; 2191 } else { 2192 Status = STATUS_UNSUCCESSFUL; 2193 goto errorout; 2194 } 2195 2196 /* set Vcb settings from Ext2Global */ 2197 if (IsFlagOn(Ext2Global->Flags, EXT2_SUPPORT_WRITING)) { 2198 if (Vcb->IsExt3fs) { 2199 if (IsFlagOn(Ext2Global->Flags, EXT3_FORCE_WRITING)) { 2200 ClearLongFlag(Vcb->Flags, VCB_READ_ONLY); 2201 } else { 2202 SetLongFlag(Vcb->Flags, VCB_READ_ONLY); 2203 } 2204 } else { 2205 ClearLongFlag(Vcb->Flags, VCB_READ_ONLY); 2206 } 2207 } else { 2208 SetLongFlag(Vcb->Flags, VCB_READ_ONLY); 2209 } 2210 2211 /* set the default codepage */ 2212 Vcb->Codepage.PageTable = Ext2Global->Codepage.PageTable; 2213 memcpy(Vcb->Codepage.AnsiName, Ext2Global->Codepage.AnsiName, CODEPAGE_MAXLEN); 2214 Vcb->Codepage.PageTable = Ext2Global->Codepage.PageTable; 2215 2216 #ifdef __REACTOS__ 2217 if (Vcb->bHidingPrefix == Ext2Global->bHidingPrefix) { 2218 #else 2219 if (Vcb->bHidingPrefix = Ext2Global->bHidingPrefix) { 2220 #endif 2221 RtlCopyMemory( Vcb->sHidingPrefix, 2222 Ext2Global->sHidingPrefix, 2223 HIDINGPAT_LEN); 2224 } else { 2225 RtlZeroMemory( Vcb->sHidingPrefix, 2226 HIDINGPAT_LEN); 2227 } 2228 2229 #ifdef __REACTOS__ 2230 if (Vcb->bHidingSuffix == Ext2Global->bHidingSuffix) { 2231 #else 2232 if (Vcb->bHidingSuffix = Ext2Global->bHidingSuffix) { 2233 #endif 2234 RtlCopyMemory( Vcb->sHidingSuffix, 2235 Ext2Global->sHidingSuffix, 2236 HIDINGPAT_LEN); 2237 } else { 2238 RtlZeroMemory( Vcb->sHidingSuffix, 2239 HIDINGPAT_LEN); 2240 } 2241 } 2242 2243 errorout: 2244 2245 if (VolumeParams.Buffer) { 2246 Ext2FreePool(VolumeParams.Buffer, EXT2_PARAM_MAGIC); 2247 } 2248 2249 return Status; 2250 } 2251 2252 NTSTATUS 2253 Ext2InitializeLabel( 2254 IN PEXT2_VCB Vcb, 2255 IN PEXT2_SUPER_BLOCK Sb 2256 ) 2257 { 2258 NTSTATUS status; 2259 2260 USHORT Length; 2261 UNICODE_STRING Label; 2262 OEM_STRING OemName; 2263 2264 Label.MaximumLength = 16 * sizeof(WCHAR); 2265 Label.Length = 0; 2266 Label.Buffer = Vcb->Vpb->VolumeLabel; 2267 Vcb->Vpb->VolumeLabelLength = 0; 2268 RtlZeroMemory(Label.Buffer, Label.MaximumLength); 2269 2270 Length = 16; 2271 while ( (Length > 0) && 2272 ((Sb->s_volume_name[Length -1] == 0x00) || 2273 (Sb->s_volume_name[Length - 1] == 0x20) ) 2274 ) { 2275 Length--; 2276 } 2277 2278 if (Length == 0) { 2279 return STATUS_SUCCESS; 2280 } 2281 2282 OemName.Buffer = Sb->s_volume_name; 2283 OemName.MaximumLength = 16; 2284 OemName.Length = Length; 2285 2286 status = Ext2OEMToUnicode(Vcb, &Label, &OemName); 2287 if (NT_SUCCESS(status)) { 2288 Vcb->Vpb->VolumeLabelLength = Label.Length; 2289 } 2290 2291 return status; 2292 } 2293 2294 static __inline BOOLEAN Ext2IsNullUuid (__u8 * uuid) 2295 { 2296 int i; 2297 for (i = 0; i < 16; i++) { 2298 if (uuid[i]) { 2299 break; 2300 } 2301 } 2302 2303 return (i >= 16); 2304 } 2305 2306 #define is_power_of_2(x) ((x) != 0 && (((x) & ((x) - 1)) == 0)) 2307 2308 NTSTATUS 2309 Ext2InitializeVcb( IN PEXT2_IRP_CONTEXT IrpContext, 2310 IN PEXT2_VCB Vcb, 2311 IN PEXT2_SUPER_BLOCK sb, 2312 IN PDEVICE_OBJECT TargetDevice, 2313 IN PDEVICE_OBJECT VolumeDevice, 2314 IN PVPB Vpb ) 2315 { 2316 NTSTATUS Status = STATUS_UNRECOGNIZED_VOLUME; 2317 ULONG IoctlSize; 2318 LONGLONG DiskSize; 2319 LONGLONG PartSize; 2320 UNICODE_STRING RootNode; 2321 USHORT Buffer[2]; 2322 ULONG ChangeCount = 0, features; 2323 CC_FILE_SIZES FileSizes; 2324 int i, has_huge_files; 2325 2326 BOOLEAN VcbResourceInitialized = FALSE; 2327 BOOLEAN NotifySyncInitialized = FALSE; 2328 BOOLEAN ExtentsInitialized = FALSE; 2329 BOOLEAN InodeLookasideInitialized = FALSE; 2330 BOOLEAN GroupLoaded = FALSE; 2331 2332 _SEH2_TRY { 2333 2334 if (Vpb == NULL) { 2335 Status = STATUS_DEVICE_NOT_READY; 2336 _SEH2_LEAVE; 2337 } 2338 2339 /* Reject mounting volume if we encounter unsupported incompat features */ 2340 if (FlagOn(sb->s_feature_incompat, ~EXT4_FEATURE_INCOMPAT_SUPP)) { 2341 Status = STATUS_UNRECOGNIZED_VOLUME; 2342 _SEH2_LEAVE; 2343 } 2344 2345 /* Mount the volume RO if we encounter unsupported ro_compat features */ 2346 if (FlagOn(sb->s_feature_ro_compat, ~EXT4_FEATURE_RO_COMPAT_SUPP)) { 2347 SetLongFlag(Vcb->Flags, VCB_RO_COMPAT_READ_ONLY); 2348 } 2349 2350 /* Recognize the filesystem as Ext3fs if it supports journalling */ 2351 if (IsFlagOn(sb->s_feature_compat, EXT4_FEATURE_COMPAT_HAS_JOURNAL)) { 2352 Vcb->IsExt3fs = TRUE; 2353 } 2354 2355 /* check block size */ 2356 Vcb->BlockSize = (EXT2_MIN_BLOCK_SIZE << sb->s_log_block_size); 2357 /* we cannot handle volume with block size bigger than 64k */ 2358 if (Vcb->BlockSize > EXT2_MAX_USER_BLKSIZE) { 2359 Status = STATUS_UNRECOGNIZED_VOLUME; 2360 _SEH2_LEAVE; 2361 } 2362 2363 if (Vcb->BlockSize >= PAGE_SIZE) { 2364 Vcb->IoUnitBits = PAGE_SHIFT; 2365 Vcb->IoUnitSize = PAGE_SIZE; 2366 } else { 2367 Vcb->IoUnitSize = Vcb->BlockSize; 2368 Vcb->IoUnitBits = Ext2Log2(Vcb->BlockSize); 2369 } 2370 2371 /* initialize vcb header members ... */ 2372 Vcb->Header.IsFastIoPossible = FastIoIsNotPossible; 2373 Vcb->Header.Resource = &(Vcb->MainResource); 2374 Vcb->Header.PagingIoResource = &(Vcb->PagingIoResource); 2375 Vcb->OpenVolumeCount = 0; 2376 Vcb->OpenHandleCount = 0; 2377 Vcb->ReferenceCount = 0; 2378 2379 /* initialize eresources */ 2380 ExInitializeResourceLite(&Vcb->MainResource); 2381 ExInitializeResourceLite(&Vcb->PagingIoResource); 2382 ExInitializeResourceLite(&Vcb->MetaInode); 2383 ExInitializeResourceLite(&Vcb->MetaBlock); 2384 ExInitializeResourceLite(&Vcb->McbLock); 2385 ExInitializeResourceLite(&Vcb->FcbLock); 2386 ExInitializeResourceLite(&Vcb->sbi.s_gd_lock); 2387 #ifndef _WIN2K_TARGET_ 2388 ExInitializeFastMutex(&Vcb->Mutex); 2389 FsRtlSetupAdvancedHeader(&Vcb->Header, &Vcb->Mutex); 2390 #endif 2391 VcbResourceInitialized = TRUE; 2392 2393 /* initialize Fcb list head */ 2394 InitializeListHead(&Vcb->FcbList); 2395 2396 /* initialize Mcb list head */ 2397 InitializeListHead(&(Vcb->McbList)); 2398 2399 /* initialize directory notify list */ 2400 InitializeListHead(&Vcb->NotifyList); 2401 FsRtlNotifyInitializeSync(&Vcb->NotifySync); 2402 NotifySyncInitialized = TRUE; 2403 2404 /* superblock checking */ 2405 Vcb->SuperBlock = sb; 2406 2407 /* initialize Vpb and Label */ 2408 Vcb->DeviceObject = VolumeDevice; 2409 Vcb->TargetDeviceObject = TargetDevice; 2410 Vcb->Vpb = Vpb; 2411 Vcb->RealDevice = Vpb->RealDevice; 2412 Vpb->DeviceObject = VolumeDevice; 2413 2414 /* set inode size */ 2415 Vcb->InodeSize = (ULONG)sb->s_inode_size; 2416 if (Vcb->InodeSize == 0) { 2417 Vcb->InodeSize = EXT2_GOOD_OLD_INODE_SIZE; 2418 } 2419 2420 /* initialize inode lookaside list */ 2421 ExInitializeNPagedLookasideList(&(Vcb->InodeLookasideList), 2422 NULL, NULL, 0, Vcb->InodeSize, 2423 'SNIE', 0); 2424 2425 InodeLookasideInitialized = TRUE; 2426 2427 /* initialize label in Vpb */ 2428 Status = Ext2InitializeLabel(Vcb, sb); 2429 if (!NT_SUCCESS(Status)) { 2430 DbgBreak(); 2431 } 2432 2433 /* check device characteristics flags */ 2434 if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_REMOVABLE_MEDIA)) { 2435 SetLongFlag(Vcb->Flags, VCB_REMOVABLE_MEDIA); 2436 } 2437 2438 if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_FLOPPY_DISKETTE)) { 2439 SetLongFlag(Vcb->Flags, VCB_FLOPPY_DISK); 2440 } 2441 2442 if (IsFlagOn(Vpb->RealDevice->Characteristics, FILE_READ_ONLY_DEVICE)) { 2443 SetLongFlag(Vcb->Flags, VCB_WRITE_PROTECTED); 2444 } 2445 2446 if (IsFlagOn(TargetDevice->Characteristics, FILE_READ_ONLY_DEVICE)) { 2447 SetLongFlag(Vcb->Flags, VCB_WRITE_PROTECTED); 2448 } 2449 2450 /* verify device is writable ? */ 2451 if (Ext2IsMediaWriteProtected(IrpContext, TargetDevice)) { 2452 SetFlag(Vcb->Flags, VCB_WRITE_PROTECTED); 2453 } 2454 2455 /* initialize UUID and serial number */ 2456 if (Ext2IsNullUuid(sb->s_uuid)) { 2457 ExUuidCreate((UUID *)sb->s_uuid); 2458 } 2459 Vpb->SerialNumber = ((ULONG*)sb->s_uuid)[0] + 2460 ((ULONG*)sb->s_uuid)[1] + 2461 ((ULONG*)sb->s_uuid)[2] + 2462 ((ULONG*)sb->s_uuid)[3]; 2463 2464 /* query partition size and disk geometry parameters */ 2465 DiskSize = Vcb->DiskGeometry.Cylinders.QuadPart * 2466 Vcb->DiskGeometry.TracksPerCylinder * 2467 Vcb->DiskGeometry.SectorsPerTrack * 2468 Vcb->DiskGeometry.BytesPerSector; 2469 2470 IoctlSize = sizeof(PARTITION_INFORMATION); 2471 Status = Ext2DiskIoControl( 2472 TargetDevice, 2473 IOCTL_DISK_GET_PARTITION_INFO, 2474 NULL, 2475 0, 2476 &Vcb->PartitionInformation, 2477 &IoctlSize ); 2478 if (NT_SUCCESS(Status)) { 2479 PartSize = Vcb->PartitionInformation.PartitionLength.QuadPart; 2480 } else { 2481 Vcb->PartitionInformation.StartingOffset.QuadPart = 0; 2482 Vcb->PartitionInformation.PartitionLength.QuadPart = DiskSize; 2483 PartSize = DiskSize; 2484 Status = STATUS_SUCCESS; 2485 } 2486 Vcb->Header.AllocationSize.QuadPart = 2487 Vcb->Header.FileSize.QuadPart = PartSize; 2488 2489 Vcb->Header.ValidDataLength.QuadPart = 2490 Vcb->Header.FileSize.QuadPart; 2491 2492 /* verify count */ 2493 IoctlSize = sizeof(ULONG); 2494 Status = Ext2DiskIoControl( 2495 TargetDevice, 2496 IOCTL_DISK_CHECK_VERIFY, 2497 NULL, 2498 0, 2499 &ChangeCount, 2500 &IoctlSize ); 2501 2502 if (!NT_SUCCESS(Status)) { 2503 _SEH2_LEAVE; 2504 } 2505 Vcb->ChangeCount = ChangeCount; 2506 2507 /* create the stream object for ext2 volume */ 2508 Vcb->Volume = IoCreateStreamFileObject(NULL, Vcb->Vpb->RealDevice); 2509 if (!Vcb->Volume) { 2510 Status = STATUS_UNRECOGNIZED_VOLUME; 2511 _SEH2_LEAVE; 2512 } 2513 2514 /* initialize streaming object file */ 2515 Vcb->Volume->SectionObjectPointer = &(Vcb->SectionObject); 2516 Vcb->Volume->ReadAccess = TRUE; 2517 Vcb->Volume->WriteAccess = TRUE; 2518 Vcb->Volume->DeleteAccess = TRUE; 2519 Vcb->Volume->FsContext = (PVOID) Vcb; 2520 Vcb->Volume->FsContext2 = NULL; 2521 Vcb->Volume->Vpb = Vcb->Vpb; 2522 2523 FileSizes.AllocationSize.QuadPart = 2524 FileSizes.FileSize.QuadPart = 2525 FileSizes.ValidDataLength.QuadPart = 2526 Vcb->Header.AllocationSize.QuadPart; 2527 2528 CcInitializeCacheMap( Vcb->Volume, 2529 &FileSizes, 2530 TRUE, 2531 &(Ext2Global->CacheManagerNoOpCallbacks), 2532 Vcb ); 2533 2534 /* initialize disk block LargetMcb and entry Mcb, 2535 it will raise an expcetion if failed */ 2536 _SEH2_TRY { 2537 FsRtlInitializeLargeMcb(&(Vcb->Extents), PagedPool); 2538 } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) { 2539 Status = STATUS_INSUFFICIENT_RESOURCES; 2540 DbgBreak(); 2541 } _SEH2_END; 2542 if (!NT_SUCCESS(Status)) { 2543 _SEH2_LEAVE; 2544 } 2545 ExtentsInitialized = TRUE; 2546 2547 /* set block device */ 2548 Vcb->bd.bd_dev = Vcb->RealDevice; 2549 Vcb->bd.bd_geo = Vcb->DiskGeometry; 2550 Vcb->bd.bd_part = Vcb->PartitionInformation; 2551 Vcb->bd.bd_volume = Vcb->Volume; 2552 Vcb->bd.bd_priv = (void *) Vcb; 2553 memset(&Vcb->bd.bd_bh_root, 0, sizeof(struct rb_root)); 2554 InitializeListHead(&Vcb->bd.bd_bh_free); 2555 ExInitializeResourceLite(&Vcb->bd.bd_bh_lock); 2556 KeInitializeEvent(&Vcb->bd.bd_bh_notify, 2557 NotificationEvent, TRUE); 2558 Vcb->bd.bd_bh_cache = kmem_cache_create("bd_bh_buffer", 2559 Vcb->BlockSize, 0, 0, NULL); 2560 if (!Vcb->bd.bd_bh_cache) { 2561 Status = STATUS_INSUFFICIENT_RESOURCES; 2562 _SEH2_LEAVE; 2563 } 2564 2565 Vcb->SectorBits = Ext2Log2(SECTOR_SIZE); 2566 Vcb->sb.s_magic = sb->s_magic; 2567 Vcb->sb.s_bdev = &Vcb->bd; 2568 Vcb->sb.s_blocksize = BLOCK_SIZE; 2569 Vcb->sb.s_blocksize_bits = BLOCK_BITS; 2570 Vcb->sb.s_priv = (void *) Vcb; 2571 Vcb->sb.s_fs_info = &Vcb->sbi; 2572 2573 Vcb->sbi.s_es = sb; 2574 Vcb->sbi.s_blocks_per_group = sb->s_blocks_per_group; 2575 Vcb->sbi.s_first_ino = sb->s_first_ino; 2576 Vcb->sbi.s_desc_size = sb->s_desc_size; 2577 2578 if (EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, EXT4_FEATURE_INCOMPAT_64BIT)) { 2579 if (Vcb->sbi.s_desc_size < EXT4_MIN_DESC_SIZE_64BIT || 2580 Vcb->sbi.s_desc_size > EXT4_MAX_DESC_SIZE || 2581 !is_power_of_2(Vcb->sbi.s_desc_size)) { 2582 DEBUG(DL_ERR, ("EXT4-fs: unsupported descriptor size %lu\n", Vcb->sbi.s_desc_size)); 2583 Status = STATUS_DISK_CORRUPT_ERROR; 2584 _SEH2_LEAVE; 2585 } 2586 } else { 2587 Vcb->sbi.s_desc_size = EXT4_MIN_DESC_SIZE; 2588 } 2589 2590 Vcb->sbi.s_blocks_per_group = sb->s_blocks_per_group; 2591 Vcb->sbi.s_inodes_per_group = sb->s_inodes_per_group; 2592 if (EXT3_INODES_PER_GROUP(&Vcb->sb) == 0) { 2593 Status = STATUS_DISK_CORRUPT_ERROR; 2594 _SEH2_LEAVE; 2595 } 2596 Vcb->sbi.s_inodes_per_block = BLOCK_SIZE / Vcb->InodeSize; 2597 if (Vcb->sbi.s_inodes_per_block == 0) { 2598 Status = STATUS_DISK_CORRUPT_ERROR; 2599 _SEH2_LEAVE; 2600 } 2601 Vcb->sbi.s_itb_per_group = Vcb->sbi.s_inodes_per_group / 2602 Vcb->sbi.s_inodes_per_block; 2603 2604 2605 Vcb->sbi.s_desc_per_block = BLOCK_SIZE / GROUP_DESC_SIZE; 2606 Vcb->sbi.s_desc_per_block_bits = ilog2(Vcb->sbi.s_desc_per_block); 2607 2608 for (i=0; i < 4; i++) { 2609 Vcb->sbi.s_hash_seed[i] = sb->s_hash_seed[i]; 2610 } 2611 Vcb->sbi.s_def_hash_version = sb->s_def_hash_version; 2612 2613 if (le32_to_cpu(sb->s_rev_level) == EXT3_GOOD_OLD_REV && 2614 (EXT3_HAS_COMPAT_FEATURE(&Vcb->sb, ~0U) || 2615 EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, ~0U) || 2616 EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, ~0U))) { 2617 printk(KERN_WARNING 2618 "EXT3-fs warning: feature flags set on rev 0 fs, " 2619 "running e2fsck is recommended\n"); 2620 } 2621 2622 /* 2623 * Check feature flags regardless of the revision level, since we 2624 * previously didn't change the revision level when setting the flags, 2625 * so there is a chance incompat flags are set on a rev 0 filesystem. 2626 */ 2627 features = EXT3_HAS_INCOMPAT_FEATURE(&Vcb->sb, ~EXT4_FEATURE_INCOMPAT_SUPP); 2628 if (features & EXT4_FEATURE_INCOMPAT_DIRDATA) { 2629 SetLongFlag(Vcb->Flags, VCB_READ_ONLY); 2630 ClearFlag(features, EXT4_FEATURE_INCOMPAT_DIRDATA); 2631 } 2632 if (features) { 2633 printk(KERN_ERR "EXT3-fs: %s: couldn't mount because of " 2634 "unsupported optional features (%x).\n", 2635 Vcb->sb.s_id, le32_to_cpu(features)); 2636 Status = STATUS_UNRECOGNIZED_VOLUME; 2637 _SEH2_LEAVE; 2638 } 2639 2640 features = EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, ~EXT4_FEATURE_RO_COMPAT_SUPP); 2641 if (features) { 2642 printk(KERN_ERR "EXT3-fs: %s: unsupported optional features in this volume: (%x).\n", 2643 Vcb->sb.s_id, le32_to_cpu(features)); 2644 if (CanIWrite(Vcb)) { 2645 } else { 2646 SetLongFlag(Vcb->Flags, VCB_READ_ONLY); 2647 } 2648 } 2649 2650 has_huge_files = EXT3_HAS_RO_COMPAT_FEATURE(&Vcb->sb, EXT4_FEATURE_RO_COMPAT_HUGE_FILE); 2651 2652 Vcb->sb.s_maxbytes = ext3_max_size(BLOCK_BITS, has_huge_files); 2653 Vcb->max_bitmap_bytes = ext3_max_bitmap_size(BLOCK_BITS, 2654 has_huge_files); 2655 Vcb->max_bytes = ext3_max_size(BLOCK_BITS, has_huge_files); 2656 2657 /* calculate maximum file bocks ... */ 2658 { 2659 ULONG dwData[EXT2_BLOCK_TYPES] = {EXT2_NDIR_BLOCKS, 1, 1, 1}; 2660 ULONG i; 2661 2662 ASSERT(BLOCK_BITS == Ext2Log2(BLOCK_SIZE)); 2663 2664 Vcb->sbi.s_groups_count = (ULONG)(ext3_blocks_count(sb) - sb->s_first_data_block + 2665 sb->s_blocks_per_group - 1) / sb->s_blocks_per_group; 2666 2667 Vcb->max_data_blocks = 0; 2668 for (i = 0; i < EXT2_BLOCK_TYPES; i++) { 2669 if (BLOCK_BITS >= 12 && i == (EXT2_BLOCK_TYPES - 1)) { 2670 dwData[i] = 0x40000000; 2671 } else { 2672 dwData[i] = dwData[i] << ((BLOCK_BITS - 2) * i); 2673 } 2674 Vcb->max_blocks_per_layer[i] = dwData[i]; 2675 Vcb->max_data_blocks += Vcb->max_blocks_per_layer[i]; 2676 } 2677 } 2678 2679 Vcb->sbi.s_gdb_count = (Vcb->sbi.s_groups_count + Vcb->sbi.s_desc_per_block - 1) / 2680 Vcb->sbi.s_desc_per_block; 2681 /* load all gorup desc */ 2682 if (!Ext2LoadGroup(Vcb)) { 2683 Status = STATUS_UNSUCCESSFUL; 2684 _SEH2_LEAVE; 2685 } 2686 GroupLoaded = TRUE; 2687 2688 /* recovery journal since it's ext3 */ 2689 if (Vcb->IsExt3fs) { 2690 Ext2RecoverJournal(IrpContext, Vcb); 2691 if (IsFlagOn(Vcb->Flags, VCB_JOURNAL_RECOVER)) { 2692 SetLongFlag(Vcb->Flags, VCB_READ_ONLY); 2693 } 2694 } 2695 2696 /* Now allocating the mcb for root ... */ 2697 Buffer[0] = L'\\'; 2698 Buffer[1] = 0; 2699 RootNode.Buffer = Buffer; 2700 RootNode.MaximumLength = RootNode.Length = 2; 2701 Vcb->McbTree = Ext2AllocateMcb( 2702 Vcb, &RootNode, NULL, 2703 FILE_ATTRIBUTE_DIRECTORY 2704 ); 2705 if (!Vcb->McbTree) { 2706 DbgBreak(); 2707 Status = STATUS_UNSUCCESSFUL; 2708 _SEH2_LEAVE; 2709 } 2710 2711 Vcb->sb.s_root = Ext2BuildEntry(Vcb, NULL, &RootNode); 2712 if (!Vcb->sb.s_root) { 2713 DbgBreak(); 2714 Status = STATUS_UNSUCCESSFUL; 2715 _SEH2_LEAVE; 2716 } 2717 Vcb->sb.s_root->d_sb = &Vcb->sb; 2718 Vcb->sb.s_root->d_inode = &Vcb->McbTree->Inode; 2719 Vcb->McbTree->de = Vcb->sb.s_root; 2720 2721 /* load root inode */ 2722 Vcb->McbTree->Inode.i_ino = EXT2_ROOT_INO; 2723 Vcb->McbTree->Inode.i_sb = &Vcb->sb; 2724 if (!Ext2LoadInode(Vcb, &Vcb->McbTree->Inode)) { 2725 DbgBreak(); 2726 Status = STATUS_CANT_WAIT; 2727 _SEH2_LEAVE; 2728 } 2729 2730 /* initializeroot node */ 2731 Vcb->McbTree->CreationTime = Ext2NtTime(Vcb->McbTree->Inode.i_ctime); 2732 Vcb->McbTree->LastAccessTime = Ext2NtTime(Vcb->McbTree->Inode.i_atime); 2733 Vcb->McbTree->LastWriteTime = Ext2NtTime(Vcb->McbTree->Inode.i_mtime); 2734 Vcb->McbTree->ChangeTime = Ext2NtTime(Vcb->McbTree->Inode.i_mtime); 2735 2736 /* check bitmap if user specifies it */ 2737 if (IsFlagOn(Ext2Global->Flags, EXT2_CHECKING_BITMAP)) { 2738 Ext2CheckBitmapConsistency(IrpContext, Vcb); 2739 } 2740 2741 /* get anything doen, then refer target device */ 2742 ObReferenceObject(Vcb->TargetDeviceObject); 2743 2744 /* query parameters from registry */ 2745 Ext2PerformRegistryVolumeParams(Vcb); 2746 2747 SetLongFlag(Vcb->Flags, VCB_INITIALIZED); 2748 2749 } _SEH2_FINALLY { 2750 2751 if (!NT_SUCCESS(Status)) { 2752 2753 if (Vcb->McbTree) { 2754 Ext2FreeMcb(Vcb, Vcb->McbTree); 2755 } 2756 2757 if (InodeLookasideInitialized) { 2758 ExDeleteNPagedLookasideList(&(Vcb->InodeLookasideList)); 2759 } 2760 2761 if (ExtentsInitialized) { 2762 if (Vcb->bd.bd_bh_cache) { 2763 if (GroupLoaded) 2764 Ext2PutGroup(Vcb); 2765 kmem_cache_destroy(Vcb->bd.bd_bh_cache); 2766 } 2767 FsRtlUninitializeLargeMcb(&(Vcb->Extents)); 2768 } 2769 2770 if (Vcb->Volume) { 2771 if (Vcb->Volume->PrivateCacheMap) { 2772 Ext2SyncUninitializeCacheMap(Vcb->Volume); 2773 } 2774 ObDereferenceObject(Vcb->Volume); 2775 } 2776 2777 if (NotifySyncInitialized) { 2778 FsRtlNotifyUninitializeSync(&Vcb->NotifySync); 2779 } 2780 2781 if (VcbResourceInitialized) { 2782 ExDeleteResourceLite(&Vcb->FcbLock); 2783 ExDeleteResourceLite(&Vcb->McbLock); 2784 ExDeleteResourceLite(&Vcb->MetaInode); 2785 ExDeleteResourceLite(&Vcb->MetaBlock); 2786 ExDeleteResourceLite(&Vcb->sbi.s_gd_lock); 2787 ExDeleteResourceLite(&Vcb->MainResource); 2788 ExDeleteResourceLite(&Vcb->PagingIoResource); 2789 } 2790 } 2791 } _SEH2_END; 2792 2793 return Status; 2794 } 2795 2796 2797 VOID 2798 Ext2TearDownStream(IN PEXT2_VCB Vcb) 2799 { 2800 PFILE_OBJECT Stream = Vcb->Volume; 2801 IO_STATUS_BLOCK IoStatus; 2802 2803 ASSERT(Vcb != NULL); 2804 ASSERT((Vcb->Identifier.Type == EXT2VCB) && 2805 (Vcb->Identifier.Size == sizeof(EXT2_VCB))); 2806 2807 if (Stream) { 2808 2809 Vcb->Volume = NULL; 2810 2811 if (IsFlagOn(Stream->Flags, FO_FILE_MODIFIED)) { 2812 CcFlushCache(&(Vcb->SectionObject), NULL, 0, &IoStatus); 2813 ClearFlag(Stream->Flags, FO_FILE_MODIFIED); 2814 } 2815 2816 if (Stream->PrivateCacheMap) { 2817 Ext2SyncUninitializeCacheMap(Stream); 2818 } 2819 2820 ObDereferenceObject(Stream); 2821 } 2822 } 2823 2824 VOID 2825 Ext2DestroyVcb (IN PEXT2_VCB Vcb) 2826 { 2827 ASSERT(Vcb != NULL); 2828 ASSERT((Vcb->Identifier.Type == EXT2VCB) && 2829 (Vcb->Identifier.Size == sizeof(EXT2_VCB))); 2830 2831 DEBUG(DL_FUN, ("Ext2DestroyVcb ...\n")); 2832 2833 if (Vcb->Volume) { 2834 Ext2TearDownStream(Vcb); 2835 } 2836 ASSERT(NULL == Vcb->Volume); 2837 2838 FsRtlNotifyUninitializeSync(&Vcb->NotifySync); 2839 Ext2ListExtents(&Vcb->Extents); 2840 FsRtlUninitializeLargeMcb(&(Vcb->Extents)); 2841 2842 Ext2CleanupAllMcbs(Vcb); 2843 2844 Ext2DropBH(Vcb); 2845 2846 if (Vcb->bd.bd_bh_cache) 2847 kmem_cache_destroy(Vcb->bd.bd_bh_cache); 2848 ExDeleteResourceLite(&Vcb->bd.bd_bh_lock); 2849 2850 if (Vcb->SuperBlock) { 2851 Ext2FreePool(Vcb->SuperBlock, EXT2_SB_MAGIC); 2852 Vcb->SuperBlock = NULL; 2853 } 2854 2855 if (IsFlagOn(Vcb->Flags, VCB_NEW_VPB)) { 2856 ASSERT(Vcb->Vpb2 != NULL); 2857 DEBUG(DL_DBG, ("Ext2DestroyVcb: Vpb2 to be freed: %p\n", Vcb->Vpb2)); 2858 ExFreePoolWithTag(Vcb->Vpb2, TAG_VPB); 2859 DEC_MEM_COUNT(PS_VPB, Vcb->Vpb2, sizeof(VPB)); 2860 Vcb->Vpb2 = NULL; 2861 } 2862 2863 ObDereferenceObject(Vcb->TargetDeviceObject); 2864 2865 ExDeleteNPagedLookasideList(&(Vcb->InodeLookasideList)); 2866 ExDeleteResourceLite(&Vcb->FcbLock); 2867 ExDeleteResourceLite(&Vcb->McbLock); 2868 ExDeleteResourceLite(&Vcb->MetaInode); 2869 ExDeleteResourceLite(&Vcb->MetaBlock); 2870 ExDeleteResourceLite(&Vcb->sbi.s_gd_lock); 2871 ExDeleteResourceLite(&Vcb->PagingIoResource); 2872 ExDeleteResourceLite(&Vcb->MainResource); 2873 2874 DEBUG(DL_DBG, ("Ext2DestroyVcb: DevObject=%p Vcb=%p\n", Vcb->DeviceObject, Vcb)); 2875 IoDeleteDevice(Vcb->DeviceObject); 2876 DEC_MEM_COUNT(PS_VCB, Vcb->DeviceObject, sizeof(EXT2_VCB)); 2877 } 2878 2879 2880 /* uninitialize cache map */ 2881 2882 VOID 2883 Ext2SyncUninitializeCacheMap ( 2884 IN PFILE_OBJECT FileObject 2885 ) 2886 { 2887 CACHE_UNINITIALIZE_EVENT UninitializeCompleteEvent; 2888 NTSTATUS WaitStatus; 2889 LARGE_INTEGER Ext2LargeZero = {0,0}; 2890 2891 2892 KeInitializeEvent( &UninitializeCompleteEvent.Event, 2893 SynchronizationEvent, 2894 FALSE); 2895 2896 CcUninitializeCacheMap( FileObject, 2897 &Ext2LargeZero, 2898 &UninitializeCompleteEvent ); 2899 2900 WaitStatus = KeWaitForSingleObject( &UninitializeCompleteEvent.Event, 2901 Executive, 2902 KernelMode, 2903 FALSE, 2904 NULL); 2905 2906 ASSERT (NT_SUCCESS(WaitStatus)); 2907 } 2908 2909 /* Link Mcb to tail of Vcb->McbList queue */ 2910 2911 VOID 2912 Ext2LinkTailMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) 2913 { 2914 if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { 2915 return; 2916 } 2917 2918 ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); 2919 2920 if (IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { 2921 DEBUG(DL_RES, ( "Ext2LinkTailMcb: %wZ already linked.\n", 2922 &Mcb->FullName)); 2923 } else { 2924 InsertTailList(&Vcb->McbList, &Mcb->Link); 2925 SetLongFlag(Mcb->Flags, MCB_VCB_LINK); 2926 Ext2ReferXcb(&Vcb->NumOfMcb); 2927 } 2928 2929 ExReleaseResourceLite(&Vcb->McbLock); 2930 } 2931 2932 /* Link Mcb to head of Vcb->McbList queue */ 2933 2934 VOID 2935 Ext2LinkHeadMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) 2936 { 2937 if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { 2938 return; 2939 } 2940 2941 ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); 2942 2943 if (!IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { 2944 InsertHeadList(&Vcb->McbList, &Mcb->Link); 2945 SetLongFlag(Mcb->Flags, MCB_VCB_LINK); 2946 Ext2ReferXcb(&Vcb->NumOfMcb); 2947 } else { 2948 DEBUG(DL_RES, ( "Ext2LinkHeadMcb: %wZ already linked.\n", 2949 &Mcb->FullName)); 2950 } 2951 ExReleaseResourceLite(&Vcb->McbLock); 2952 } 2953 2954 /* Unlink Mcb from Vcb->McbList queue */ 2955 2956 VOID 2957 Ext2UnlinkMcb(PEXT2_VCB Vcb, PEXT2_MCB Mcb) 2958 { 2959 if (Mcb->Inode.i_ino == EXT2_ROOT_INO) { 2960 return; 2961 } 2962 2963 ExAcquireResourceExclusiveLite(&Vcb->McbLock, TRUE); 2964 2965 if (IsFlagOn(Mcb->Flags, MCB_VCB_LINK)) { 2966 RemoveEntryList(&(Mcb->Link)); 2967 ClearLongFlag(Mcb->Flags, MCB_VCB_LINK); 2968 Ext2DerefXcb(&Vcb->NumOfMcb); 2969 } else { 2970 DEBUG(DL_RES, ( "Ext2UnlinkMcb: %wZ already unlinked.\n", 2971 &Mcb->FullName)); 2972 } 2973 ExReleaseResourceLite(&Vcb->McbLock); 2974 } 2975 2976 /* get the first Mcb record in Vcb->McbList */ 2977 2978 PEXT2_MCB 2979 Ext2FirstUnusedMcb(PEXT2_VCB Vcb, BOOLEAN Wait, ULONG Number) 2980 { 2981 PEXT2_MCB Head = NULL; 2982 PEXT2_MCB Tail = NULL; 2983 PEXT2_MCB Mcb = NULL; 2984 PLIST_ENTRY List = NULL; 2985 ULONG i = 0; 2986 LARGE_INTEGER start, now; 2987 2988 if (!ExAcquireResourceExclusiveLite(&Vcb->McbLock, Wait)) { 2989 return NULL; 2990 } 2991 2992 KeQuerySystemTime(&start); 2993 2994 while (Number--) { 2995 2996 BOOLEAN Skip = TRUE; 2997 2998 if (IsListEmpty(&Vcb->McbList)) { 2999 break; 3000 } 3001 3002 while (i++ < Vcb->NumOfMcb) { 3003 3004 KeQuerySystemTime(&now); 3005 if (now.QuadPart > start.QuadPart + (LONGLONG)10*1000*1000) { 3006 break; 3007 } 3008 3009 List = RemoveHeadList(&Vcb->McbList); 3010 Mcb = CONTAINING_RECORD(List, EXT2_MCB, Link); 3011 ASSERT(IsFlagOn(Mcb->Flags, MCB_VCB_LINK)); 3012 3013 if (Mcb->Fcb == NULL && !IsMcbRoot(Mcb) && 3014 Mcb->Refercount == 0 && 3015 (Mcb->Child == NULL || IsMcbSymLink(Mcb))) { 3016 3017 Ext2RemoveMcb(Vcb, Mcb); 3018 ClearLongFlag(Mcb->Flags, MCB_VCB_LINK); 3019 Ext2DerefXcb(&Vcb->NumOfMcb); 3020 3021 /* attach all Mcb into a chain*/ 3022 if (Head) { 3023 ASSERT(Tail != NULL); 3024 Tail->Next = Mcb; 3025 Tail = Mcb; 3026 } else { 3027 Head = Tail = Mcb; 3028 } 3029 Tail->Next = NULL; 3030 Skip = FALSE; 3031 3032 } else { 3033 3034 InsertTailList(&Vcb->McbList, &Mcb->Link); 3035 Mcb = NULL; 3036 } 3037 } 3038 3039 if (Skip) 3040 break; 3041 } 3042 3043 ExReleaseResourceLite(&Vcb->McbLock); 3044 3045 return Head; 3046 } 3047 3048 3049 /* Reaper thread to release unused Mcb blocks */ 3050 #ifdef __REACTOS__ 3051 VOID NTAPI 3052 #else 3053 VOID 3054 #endif 3055 Ext2McbReaperThread( 3056 PVOID Context 3057 ) 3058 { 3059 PEXT2_REAPER Reaper = Context; 3060 PLIST_ENTRY List = NULL; 3061 LARGE_INTEGER Timeout; 3062 3063 PEXT2_VCB Vcb = NULL; 3064 PEXT2_MCB Mcb = NULL; 3065 3066 ULONG i, NumOfMcbs; 3067 3068 BOOLEAN GlobalAcquired = FALSE; 3069 3070 BOOLEAN DidNothing = TRUE; 3071 BOOLEAN LastState = TRUE; 3072 BOOLEAN WaitLock; 3073 3074 _SEH2_TRY { 3075 3076 Reaper->Thread = PsGetCurrentThread(); 3077 3078 /* wake up DirverEntry */ 3079 KeSetEvent(&Reaper->Engine, 0, FALSE); 3080 3081 /* now process looping */ 3082 while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { 3083 3084 WaitLock = FALSE; 3085 3086 /* calculate how long we need wait */ 3087 if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 128)) { 3088 Timeout.QuadPart = (LONGLONG)-1000*1000; /* 0.1 second */ 3089 NumOfMcbs = Ext2Global->MaxDepth * 4; 3090 WaitLock = TRUE; 3091 } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 32)) { 3092 Timeout.QuadPart = (LONGLONG)-1000*1000*5; /* 0.5 second */ 3093 NumOfMcbs = Ext2Global->MaxDepth * 2; 3094 WaitLock = TRUE; 3095 } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 8)) { 3096 Timeout.QuadPart = (LONGLONG)-1000*1000*10; /* 1 second */ 3097 NumOfMcbs = Ext2Global->MaxDepth; 3098 } else if (Ext2Global->PerfStat.Current.Mcb > (((ULONG)Ext2Global->MaxDepth) * 2)) { 3099 Timeout.QuadPart = (LONGLONG)-2*1000*1000*10; /* 2 second */ 3100 NumOfMcbs = Ext2Global->MaxDepth / 4; 3101 } else if (Ext2Global->PerfStat.Current.Mcb > (ULONG)Ext2Global->MaxDepth) { 3102 Timeout.QuadPart = (LONGLONG)-4*1000*1000*10; /* 4 seconds */ 3103 NumOfMcbs = Ext2Global->MaxDepth / 8; 3104 } else if (DidNothing) { 3105 Timeout.QuadPart = (LONGLONG)-8*1000*1000*10; /* 8 seconds */ 3106 if (LastState) { 3107 Timeout.QuadPart *= 2; 3108 } 3109 NumOfMcbs = Ext2Global->MaxDepth / 16; 3110 } else { 3111 Timeout.QuadPart = (LONGLONG)-5*1000*1000*10; /* 5 seconds */ 3112 if (LastState) { 3113 Timeout.QuadPart *= 2; 3114 } 3115 NumOfMcbs = Ext2Global->MaxDepth / 32; 3116 } 3117 3118 if (NumOfMcbs == 0) 3119 NumOfMcbs = 1; 3120 3121 LastState = DidNothing; 3122 3123 /* wait until it is waken or it times out */ 3124 KeWaitForSingleObject( 3125 &Reaper->Wait, 3126 Executive, 3127 KernelMode, 3128 FALSE, 3129 &Timeout 3130 ); 3131 3132 if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) 3133 break; 3134 3135 DidNothing = TRUE; 3136 3137 /* acquire global exclusive lock */ 3138 if (!ExAcquireResourceSharedLite(&Ext2Global->Resource, WaitLock)) { 3139 continue; 3140 } 3141 GlobalAcquired = TRUE; 3142 3143 /* search all Vcb to get unused resources freed to system */ 3144 for (List = Ext2Global->VcbList.Flink; 3145 List != &(Ext2Global->VcbList); 3146 List = List->Flink ) { 3147 3148 Vcb = CONTAINING_RECORD(List, EXT2_VCB, Next); 3149 3150 Mcb = Ext2FirstUnusedMcb(Vcb, WaitLock, NumOfMcbs); 3151 while (Mcb) { 3152 PEXT2_MCB Next = Mcb->Next; 3153 DEBUG(DL_RES, ( "Ext2ReaperThread: releasing Mcb (%p): %wZ" 3154 " Total: %xh\n", Mcb, &Mcb->FullName, 3155 Ext2Global->PerfStat.Current.Mcb)); 3156 Ext2FreeMcb(Vcb, Mcb); 3157 Mcb = Next; 3158 LastState = DidNothing = FALSE; 3159 } 3160 } 3161 if (DidNothing) { 3162 KeClearEvent(&Reaper->Wait); 3163 } 3164 if (GlobalAcquired) { 3165 ExReleaseResourceLite(&Ext2Global->Resource); 3166 GlobalAcquired = FALSE; 3167 } 3168 } 3169 3170 } _SEH2_FINALLY { 3171 3172 if (GlobalAcquired) { 3173 ExReleaseResourceLite(&Ext2Global->Resource); 3174 } 3175 3176 KeSetEvent(&Reaper->Engine, 0, FALSE); 3177 } _SEH2_END; 3178 3179 PsTerminateSystemThread(STATUS_SUCCESS); 3180 } 3181 3182 3183 /* get buffer heads from global Vcb BH list */ 3184 3185 BOOLEAN 3186 Ext2QueryUnusedBH(PEXT2_VCB Vcb, PLIST_ENTRY head) 3187 { 3188 struct buffer_head *bh = NULL; 3189 PLIST_ENTRY next = NULL; 3190 LARGE_INTEGER start, now; 3191 BOOLEAN wake = FALSE; 3192 3193 KeQuerySystemTime(&start); 3194 3195 ExAcquireResourceExclusiveLite(&Vcb->bd.bd_bh_lock, TRUE); 3196 3197 while (!IsListEmpty(&Vcb->bd.bd_bh_free)) { 3198 3199 KeQuerySystemTime(&now); 3200 if (now.QuadPart > start.QuadPart + (LONGLONG)10*1000*1000) { 3201 break; 3202 } 3203 3204 next = RemoveHeadList(&Vcb->bd.bd_bh_free); 3205 bh = CONTAINING_RECORD(next, struct buffer_head, b_link); 3206 if (atomic_read(&bh->b_count)) { 3207 InitializeListHead(&bh->b_link); 3208 /* to be inserted by brelse */ 3209 continue; 3210 } 3211 3212 if ( IsFlagOn(Vcb->Flags, VCB_BEING_DROPPED) || 3213 (bh->b_ts_drop.QuadPart + (LONGLONG)10*1000*1000*15) > now.QuadPart || 3214 (bh->b_ts_creat.QuadPart + (LONGLONG)10*1000*1000*180) > now.QuadPart) { 3215 InsertTailList(head, &bh->b_link); 3216 buffer_head_remove(&Vcb->bd, bh); 3217 } else { 3218 InsertHeadList(&Vcb->bd.bd_bh_free, &bh->b_link); 3219 break; 3220 } 3221 } 3222 3223 wake = IsListEmpty(&Vcb->bd.bd_bh_free); 3224 ExReleaseResourceLite(&Vcb->bd.bd_bh_lock); 3225 3226 if (wake) 3227 KeSetEvent(&Vcb->bd.bd_bh_notify, 0, FALSE); 3228 3229 return IsFlagOn(Vcb->Flags, VCB_BEING_DROPPED); 3230 } 3231 3232 3233 /* Reaper thread to release unused buffer heads */ 3234 #ifdef __REACTOS__ 3235 VOID NTAPI 3236 #else 3237 VOID 3238 #endif 3239 Ext2bhReaperThread( 3240 PVOID Context 3241 ) 3242 { 3243 PEXT2_REAPER Reaper = Context; 3244 PEXT2_VCB Vcb = NULL; 3245 LIST_ENTRY List, *Link; 3246 LARGE_INTEGER Timeout; 3247 3248 BOOLEAN GlobalAcquired = FALSE; 3249 BOOLEAN DidNothing = FALSE; 3250 BOOLEAN NonWait = FALSE; 3251 3252 _SEH2_TRY { 3253 3254 Reaper->Thread = PsGetCurrentThread(); 3255 3256 /* wake up DirverEntry */ 3257 KeSetEvent(&Reaper->Engine, 0, FALSE); 3258 3259 /* now process looping */ 3260 while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { 3261 3262 /* wait until it is waken or it times out */ 3263 if (NonWait) { 3264 Timeout.QuadPart = (LONGLONG)-10*1000*10; 3265 NonWait = FALSE; 3266 } else if (DidNothing) { 3267 Timeout.QuadPart = Timeout.QuadPart * 2; 3268 } else { 3269 Timeout.QuadPart = (LONGLONG)-10*1000*1000*10; /* 10 seconds */ 3270 } 3271 KeWaitForSingleObject( 3272 &Reaper->Wait, 3273 Executive, 3274 KernelMode, 3275 FALSE, 3276 &Timeout 3277 ); 3278 3279 if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) 3280 break; 3281 3282 InitializeListHead(&List); 3283 3284 /* acquire global exclusive lock */ 3285 ExAcquireResourceSharedLite(&Ext2Global->Resource, TRUE); 3286 GlobalAcquired = TRUE; 3287 /* search all Vcb to get unused resources freed to system */ 3288 for (Link = Ext2Global->VcbList.Flink; 3289 Link != &(Ext2Global->VcbList); 3290 Link = Link->Flink ) { 3291 3292 Vcb = CONTAINING_RECORD(Link, EXT2_VCB, Next); 3293 NonWait = Ext2QueryUnusedBH(Vcb, &List); 3294 } 3295 DidNothing = IsListEmpty(&List); 3296 if (DidNothing) { 3297 KeClearEvent(&Reaper->Wait); 3298 } 3299 if (GlobalAcquired) { 3300 ExReleaseResourceLite(&Ext2Global->Resource); 3301 GlobalAcquired = FALSE; 3302 } 3303 3304 while (!IsListEmpty(&List)) { 3305 struct buffer_head *bh; 3306 Link = RemoveHeadList(&List); 3307 bh = CONTAINING_RECORD(Link, struct buffer_head, b_link); 3308 ASSERT(0 == atomic_read(&bh->b_count)); 3309 free_buffer_head(bh); 3310 } 3311 } 3312 3313 } _SEH2_FINALLY { 3314 3315 if (GlobalAcquired) { 3316 ExReleaseResourceLite(&Ext2Global->Resource); 3317 } 3318 3319 KeSetEvent(&Reaper->Engine, 0, FALSE); 3320 } _SEH2_END; 3321 3322 PsTerminateSystemThread(STATUS_SUCCESS); 3323 } 3324 3325 /* get unused Fcbs to free */ 3326 3327 BOOLEAN 3328 Ext2QueryUnusedFcb(PEXT2_VCB Vcb, PLIST_ENTRY list) 3329 { 3330 PEXT2_FCB Fcb; 3331 PLIST_ENTRY next = NULL; 3332 LARGE_INTEGER start, now; 3333 3334 ULONG count = 0; 3335 ULONG tries = 0; 3336 BOOLEAN wake = FALSE; 3337 BOOLEAN retry = TRUE; 3338 3339 KeQuerySystemTime(&start); 3340 3341 ExAcquireResourceExclusiveLite(&Vcb->FcbLock, TRUE); 3342 3343 again: 3344 3345 KeQuerySystemTime(&now); 3346 while (!IsListEmpty(&Vcb->FcbList)) { 3347 3348 next = RemoveHeadList(&Vcb->FcbList); 3349 Fcb = CONTAINING_RECORD(next, EXT2_FCB, Next); 3350 3351 if (Fcb->ReferenceCount > 0) { 3352 InsertTailList(&Vcb->FcbList, &Fcb->Next); 3353 break; 3354 } 3355 3356 retry = FALSE; 3357 3358 if (now.QuadPart < Fcb->TsDrop.QuadPart + 10*1000*1000*120) { 3359 InsertHeadList(&Vcb->FcbList, &Fcb->Next); 3360 break; 3361 } 3362 3363 Ext2UnlinkFcb(Fcb); 3364 Ext2DerefXcb(&Vcb->FcbCount); 3365 InsertTailList(list, &Fcb->Next); 3366 if (++count >= Ext2Global->MaxDepth) { 3367 break; 3368 } 3369 } 3370 3371 if (start.QuadPart + 10*1000*1000 > now.QuadPart) { 3372 retry = FALSE; 3373 } 3374 3375 if (retry) { 3376 if (++tries < (Vcb->FcbCount >> 4) ) 3377 goto again; 3378 } 3379 3380 ExReleaseResourceLite(&Vcb->FcbLock); 3381 3382 return 0; 3383 } 3384 3385 /* Reaper thread to release Fcb */ 3386 #ifdef __REACTOS__ 3387 VOID NTAPI 3388 #else 3389 VOID 3390 #endif 3391 Ext2FcbReaperThread( 3392 PVOID Context 3393 ) 3394 { 3395 PEXT2_REAPER Reaper = Context; 3396 PEXT2_VCB Vcb = NULL; 3397 LIST_ENTRY List, *Link; 3398 LARGE_INTEGER Timeout; 3399 3400 BOOLEAN GlobalAcquired = FALSE; 3401 BOOLEAN DidNothing = FALSE; 3402 BOOLEAN NonWait = FALSE; 3403 3404 _SEH2_TRY { 3405 3406 Reaper->Thread = PsGetCurrentThread(); 3407 3408 /* wake up DirverEntry */ 3409 KeSetEvent(&Reaper->Engine, 0, FALSE); 3410 3411 /* now process looping */ 3412 while (!IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) { 3413 3414 /* wait until it is waken or it times out */ 3415 if (NonWait) { 3416 Timeout.QuadPart = (LONGLONG)-10*1000*100; 3417 NonWait = FALSE; 3418 } else if (DidNothing) { 3419 Timeout.QuadPart = Timeout.QuadPart * 2; 3420 } else { 3421 Timeout.QuadPart = (LONGLONG)-10*1000*1000*20; /* 20 seconds */ 3422 } 3423 KeWaitForSingleObject( 3424 &Reaper->Wait, 3425 Executive, 3426 KernelMode, 3427 FALSE, 3428 &Timeout 3429 ); 3430 3431 if (IsFlagOn(Reaper->Flags, EXT2_REAPER_FLAG_STOP)) 3432 break; 3433 3434 InitializeListHead(&List); 3435 3436 /* acquire global exclusive lock */ 3437 ExAcquireResourceSharedLite(&Ext2Global->Resource, TRUE); 3438 GlobalAcquired = TRUE; 3439 /* search all Vcb to get unused resources freed to system */ 3440 for (Link = Ext2Global->VcbList.Flink; 3441 Link != &(Ext2Global->VcbList); 3442 Link = Link->Flink ) { 3443 3444 Vcb = CONTAINING_RECORD(Link, EXT2_VCB, Next); 3445 NonWait = Ext2QueryUnusedFcb(Vcb, &List); 3446 } 3447 DidNothing = IsListEmpty(&List); 3448 if (DidNothing) { 3449 KeClearEvent(&Reaper->Wait); 3450 } 3451 if (GlobalAcquired) { 3452 ExReleaseResourceLite(&Ext2Global->Resource); 3453 GlobalAcquired = FALSE; 3454 } 3455 3456 while (!IsListEmpty(&List)) { 3457 PEXT2_FCB Fcb; 3458 Link = RemoveHeadList(&List); 3459 Fcb = CONTAINING_RECORD(Link, EXT2_FCB, Next); 3460 ASSERT(0 == Fcb->ReferenceCount); 3461 Ext2FreeFcb(Fcb); 3462 } 3463 } 3464 3465 } _SEH2_FINALLY { 3466 3467 if (GlobalAcquired) { 3468 ExReleaseResourceLite(&Ext2Global->Resource); 3469 } 3470 3471 KeSetEvent(&Reaper->Engine, 0, FALSE); 3472 } _SEH2_END; 3473 3474 PsTerminateSystemThread(STATUS_SUCCESS); 3475 } 3476 3477 NTSTATUS 3478 Ext2StartReaper(PEXT2_REAPER Reaper, EXT2_REAPER_RELEASE Free) 3479 { 3480 NTSTATUS status = STATUS_SUCCESS; 3481 OBJECT_ATTRIBUTES oa; 3482 HANDLE handle = 0; 3483 LARGE_INTEGER timeout; 3484 3485 Reaper->Free = Free; 3486 3487 /* initialize wait event */ 3488 KeInitializeEvent( 3489 &Reaper->Wait, 3490 SynchronizationEvent, FALSE 3491 ); 3492 3493 /* Reaper thread engine event */ 3494 KeInitializeEvent( 3495 &Reaper->Engine, 3496 SynchronizationEvent, FALSE 3497 ); 3498 3499 /* initialize oa */ 3500 InitializeObjectAttributes( 3501 &oa, 3502 NULL, 3503 OBJ_CASE_INSENSITIVE | 3504 OBJ_KERNEL_HANDLE, 3505 NULL, 3506 NULL 3507 ); 3508 3509 /* start a new system thread */ 3510 status = PsCreateSystemThread( 3511 &handle, 3512 0, 3513 &oa, 3514 NULL, 3515 NULL, 3516 Free, 3517 (PVOID)Reaper 3518 ); 3519 3520 if (NT_SUCCESS(status)) { 3521 ZwClose(handle); 3522 3523 /* make sure Reaperthread is started */ 3524 timeout.QuadPart = (LONGLONG)-10*1000*1000*2; /* 2 seconds */ 3525 status = KeWaitForSingleObject( 3526 &Reaper->Engine, 3527 Executive, 3528 KernelMode, 3529 FALSE, 3530 &timeout 3531 ); 3532 if (status != STATUS_SUCCESS) { 3533 status = STATUS_INSUFFICIENT_RESOURCES; 3534 } 3535 } 3536 3537 return status; 3538 } 3539 3540 3541 #ifdef __REACTOS__ 3542 VOID NTAPI 3543 #else 3544 VOID 3545 #endif 3546 Ext2StopReaper(PEXT2_REAPER Reaper) 3547 { 3548 LARGE_INTEGER timeout; 3549 3550 Reaper->Flags |= EXT2_REAPER_FLAG_STOP; 3551 KeSetEvent(&Reaper->Wait, 0, FALSE); 3552 3553 /* make sure Reaperthread is started */ 3554 timeout.QuadPart = (LONGLONG)-10*1000*1000*2; /* 2 seconds */ 3555 KeWaitForSingleObject( 3556 &Reaper->Engine, 3557 Executive, 3558 KernelMode, 3559 FALSE, 3560 &timeout); 3561 } 3562