src/WINNT/afsrdr/kernel/lib/AFSGeneric.cpp

   1 /*
   2  * Copyright (c) 2008, 2009, 2010, 2011 Kernel Drivers, LLC.
   3  * Copyright (c) 2009, 2010, 2011 Your File System, Inc.
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * - Redistributions of source code must retain the above copyright notice,
  11  *   this list of conditions and the following disclaimer.
  12  * - Redistributions in binary form must reproduce the above copyright
  13  *   notice,
  14  *   this list of conditions and the following disclaimer in the
  15  *   documentation
  16  *   and/or other materials provided with the distribution.
  17  * - Neither the names of Kernel Drivers, LLC and Your File System, Inc.
  18  *   nor the names of their contributors may be used to endorse or promote
  19  *   products derived from this software without specific prior written
  20  *   permission from Kernel Drivers, LLC and Your File System, Inc.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  23  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  24  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  25  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
  26  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  27  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  28  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  29  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  30  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  31  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  32  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  33  */
  34
  35 //
  36 // File: AFSGeneric.cpp
  37 //
  38
  39 #include "AFSCommon.h"
  40
  41 //
  42 // Function: AFSExceptionFilter
  43 //
  44 // Description:
  45 //
  46 //      This function is the exception handler
  47 //
  48 // Return:
  49 //
  50 //      A status is returned for the function
  51 //
  52
  53 ULONG
  54 AFSExceptionFilter( IN ULONG Code,
  55                     IN PEXCEPTION_POINTERS ExceptPtrs)
  56 {
  57
  58     PEXCEPTION_RECORD ExceptRec;
  59     PCONTEXT Context;
  60
  61     __try
  62     {
  63
  64         ExceptRec = ExceptPtrs->ExceptionRecord;
  65
  66         Context = ExceptPtrs->ContextRecord;
  67
  68         AFSDbgLogMsg( 0,
  69                       0,
  70                       "AFSExceptionFilter (Library) - EXR %p CXR %p Code %08lX Address %p Routine %p\n",
  71                       ExceptRec,
  72                       Context,
  73                       ExceptRec->ExceptionCode,
  74                       ExceptRec->ExceptionAddress,
  75                       (void *)AFSExceptionFilter);
  76
  77         DbgPrint("**** Exception Caught in AFS Redirector Library ****\n");
  78
  79         DbgPrint("\n\nPerform the following WnDbg Cmds:\n");
  80         DbgPrint("\n\t.exr %p ;  .cxr %p\n\n", ExceptRec, Context);
  81
  82         DbgPrint("**** Exception Complete from AFS Redirector Library ****\n");
  83
  84         if( BooleanFlagOn( AFSDebugFlags, AFS_DBG_BUGCHECK_EXCEPTION))
  85         {
  86
  87             KeBugCheck( (ULONG)-2);
  88         }
  89         else
  90         {
  91
  92             AFSBreakPoint();
  93         }
  94     }
  95     __except( EXCEPTION_EXECUTE_HANDLER)
  96     {
  97
  98         NOTHING;
  99     }
 100
 101     return EXCEPTION_EXECUTE_HANDLER;
 102 }
 103
 104 //
 105 // Function: AFSLibExAllocatePoolWithTag()
 106 //
 107 // Purpose: Allocate Pool Memory.  If BugCheck Exception flag
 108 //          is configured on, then bugcheck the system if
 109 //          a memory allocation fails.  The routine should be
 110 //          used for all memory allocations that are to be freed
 111 //          when the library is unloaded.  Memory allocations that
 112 //          are to survive library unload and reload should be
 113 //          performed using AFSExAllocatePoolWithTag() which is
 114 //          provided by the AFS Framework.
 115 //
 116 // Parameters:
 117 //                POOL_TYPE PoolType - Paged or NonPaged
 118 //                SIZE_T NumberOfBytes - requested allocation size
 119 //                ULONG  Tag - Pool Allocation Tag to be applied for tracking
 120 //
 121 // Return:
 122 //                void * - the memory allocation
 123 //
 124
 125 void *
 126 AFSLibExAllocatePoolWithTag( IN POOL_TYPE  PoolType,
 127                              IN SIZE_T  NumberOfBytes,
 128                              IN ULONG  Tag)
 129 {
 130
 131     void *pBuffer = NULL;
 132
 133     pBuffer = ExAllocatePoolWithTag( PoolType,
 134                                      NumberOfBytes,
 135                                      Tag);
 136
 137     if( pBuffer == NULL)
 138     {
 139
 140         if( BooleanFlagOn( AFSDebugFlags, AFS_DBG_BUGCHECK_EXCEPTION))
 141         {
 142
 143             KeBugCheck( (ULONG)-2);
 144         }
 145         else
 146         {
 147
 148             AFSDbgLogMsg( 0,
 149                           0,
 150                           "AFSLibExAllocatePoolWithTag failure Type %08lX Size %08lX Tag %08lX %08lX\n",
 151                           PoolType,
 152                           NumberOfBytes,
 153                           Tag,
 154                           PsGetCurrentThread());
 155
 156             AFSBreakPoint();
 157         }
 158     }
 159
 160     return pBuffer;
 161 }
 162
 163 //
 164 // Function: AFSAcquireExcl()
 165 //
 166 // Purpose: Called to acquire a resource exclusive with optional wait
 167 //
 168 // Parameters:
 169 //                PERESOURCE Resource - Resource to acquire
 170 //                BOOLEAN Wait - Whether to block
 171 //
 172 // Return:
 173 //                BOOLEAN - Whether the mask was acquired
 174 //
 175
 176 BOOLEAN
 177 AFSAcquireExcl( IN PERESOURCE Resource,
 178                 IN BOOLEAN wait)
 179 {
 180
 181     BOOLEAN bStatus = FALSE;
 182
 183     //
 184     // Normal kernel APCs must be disabled before calling
 185     // ExAcquireResourceExclusiveLite. Otherwise a bugcheck occurs.
 186     //
 187
 188     KeEnterCriticalRegion();
 189
 190     bStatus = ExAcquireResourceExclusiveLite( Resource,
 191                                               wait);
 192
 193     if( !bStatus)
 194     {
 195
 196         KeLeaveCriticalRegion();
 197     }
 198
 199     return bStatus;
 200 }
 201
 202 BOOLEAN
 203 AFSAcquireSharedStarveExclusive( IN PERESOURCE Resource,
 204                                  IN BOOLEAN Wait)
 205 {
 206
 207     BOOLEAN bStatus = FALSE;
 208
 209     KeEnterCriticalRegion();
 210
 211     bStatus = ExAcquireSharedStarveExclusive( Resource,
 212                                               Wait);
 213
 214     if( !bStatus)
 215     {
 216
 217         KeLeaveCriticalRegion();
 218     }
 219
 220     return bStatus;
 221 }
 222
 223 //
 224 // Function: AFSAcquireShared()
 225 //
 226 // Purpose: Called to acquire a resource shared with optional wait
 227 //
 228 // Parameters:
 229 //                PERESOURCE Resource - Resource to acquire
 230 //                BOOLEAN Wait - Whether to block
 231 //
 232 // Return:
 233 //                BOOLEAN - Whether the mask was acquired
 234 //
 235
 236 BOOLEAN
 237 AFSAcquireShared( IN PERESOURCE Resource,
 238                   IN BOOLEAN wait)
 239 {
 240
 241     BOOLEAN bStatus = FALSE;
 242
 243     KeEnterCriticalRegion();
 244
 245     bStatus = ExAcquireResourceSharedLite( Resource,
 246                                            wait);
 247
 248     if( !bStatus)
 249     {
 250
 251         KeLeaveCriticalRegion();
 252     }
 253
 254     return bStatus;
 255 }
 256
 257 //
 258 // Function: AFSReleaseResource()
 259 //
 260 // Purpose: Called to release a resource
 261 //
 262 // Parameters:
 263 //                PERESOURCE Resource - Resource to release
 264 //
 265 // Return:
 266 //                None
 267 //
 268
 269 void
 270 AFSReleaseResource( IN PERESOURCE Resource)
 271 {
 272
 273     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
 274                   AFS_TRACE_LEVEL_VERBOSE,
 275                   "AFSReleaseResource Releasing lock %08lX Thread %08lX\n",
 276                   Resource,
 277                   PsGetCurrentThread());
 278
 279     ExReleaseResourceLite( Resource);
 280
 281     KeLeaveCriticalRegion();
 282
 283     return;
 284 }
 285
 286 void
 287 AFSConvertToShared( IN PERESOURCE Resource)
 288 {
 289
 290     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
 291                   AFS_TRACE_LEVEL_VERBOSE,
 292                   "AFSConvertToShared Converting lock %08lX Thread %08lX\n",
 293                   Resource,
 294                   PsGetCurrentThread());
 295
 296     ExConvertExclusiveToSharedLite( Resource);
 297
 298     return;
 299 }
 300
 301 //
 302 // Function: AFSCompleteRequest
 303 //
 304 // Description:
 305 //
 306 //      This function completes irps
 307 //
 308 // Return:
 309 //
 310 //      A status is returned for the function
 311 //
 312
 313 void
 314 AFSCompleteRequest( IN PIRP Irp,
 315                     IN ULONG Status)
 316 {
 317
 318     Irp->IoStatus.Status = Status;
 319
 320     IoCompleteRequest( Irp,
 321                        IO_NO_INCREMENT);
 322
 323     return;
 324 }
 325
 326 //
 327 // Function: AFSGenerateCRC
 328 //
 329 // Description:
 330 //
 331 //      Given a device and filename this function generates a CRC
 332 //
 333 // Return:
 334 //
 335 //      A status is returned for the function
 336 //
 337
 338 ULONG
 339 AFSGenerateCRC( IN PUNICODE_STRING FileName,
 340                 IN BOOLEAN UpperCaseName)
 341 {
 342
 343     ULONG ulCRC = 0;
 344     NTSTATUS ntStatus = STATUS_SUCCESS;
 345
 346     ntStatus = RtlHashUnicodeString( FileName,
 347                                      UpperCaseName,
 348                                      HASH_STRING_ALGORITHM_DEFAULT,
 349                                      &ulCRC);
 350
 351     if( !NT_SUCCESS( ntStatus))
 352     {
 353         ulCRC = 0;
 354     }
 355
 356     return ulCRC;
 357 }
 358
 359 void *
 360 AFSLockSystemBuffer( IN PIRP Irp,
 361                      IN ULONG Length)
 362 {
 363
 364     NTSTATUS Status = STATUS_SUCCESS;
 365     void *pAddress = NULL;
 366
 367     if( Irp->MdlAddress != NULL)
 368     {
 369
 370         pAddress = MmGetSystemAddressForMdlSafe( Irp->MdlAddress,
 371                                                  NormalPagePriority);
 372     }
 373     else if( Irp->AssociatedIrp.SystemBuffer != NULL)
 374     {
 375
 376         pAddress = Irp->AssociatedIrp.SystemBuffer;
 377     }
 378     else if( Irp->UserBuffer != NULL)
 379     {
 380
 381         Irp->MdlAddress = IoAllocateMdl( Irp->UserBuffer,
 382                                          Length,
 383                                          FALSE,
 384                                          FALSE,
 385                                          Irp);
 386
 387         if( Irp->MdlAddress != NULL)
 388         {
 389
 390             //
 391             //  Lock the new Mdl in memory.
 392             //
 393
 394             __try
 395             {
 396                 PIO_STACK_LOCATION pIoStack;
 397                 pIoStack = IoGetCurrentIrpStackLocation( Irp);
 398
 399
 400                 MmProbeAndLockPages( Irp->MdlAddress, KernelMode,
 401                                      (pIoStack->MajorFunction == IRP_MJ_READ) ? IoWriteAccess : IoReadAccess);
 402
 403                 pAddress = MmGetSystemAddressForMdlSafe( Irp->MdlAddress, NormalPagePriority );
 404
 405             }
 406             __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
 407             {
 408
 409                 IoFreeMdl( Irp->MdlAddress );
 410                 Irp->MdlAddress = NULL;
 411                 pAddress = NULL;
 412             }
 413         }
 414     }
 415
 416     return pAddress;
 417 }
 418
 419 void *
 420 AFSLockUserBuffer( IN void *UserBuffer,
 421                    IN ULONG BufferLength,
 422                                    OUT MDL ** Mdl)
 423 {
 424
 425     NTSTATUS ntStatus = STATUS_SUCCESS;
 426     void *pAddress = NULL;
 427     MDL *pMdl = NULL;
 428
 429     __Enter
 430     {
 431
 432         pMdl = IoAllocateMdl( UserBuffer,
 433                               BufferLength,
 434                               FALSE,
 435                               FALSE,
 436                               NULL);
 437
 438             if( pMdl == NULL)
 439             {
 440
 441                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 442             }
 443
 444         //
 445         //  Lock the new Mdl in memory.
 446         //
 447
 448         __try
 449         {
 450
 451             MmProbeAndLockPages( pMdl,
 452                                  KernelMode,
 453                                  IoWriteAccess);
 454
 455             pAddress = MmGetSystemAddressForMdlSafe( pMdl,
 456                                                      NormalPagePriority);
 457         }
 458         __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
 459         {
 460
 461             IoFreeMdl( pMdl);
 462             pMdl = NULL;
 463             pAddress = NULL;
 464         }
 465
 466         if( pMdl != NULL)
 467         {
 468
 469             *Mdl = pMdl;
 470         }
 471
 472 try_exit:
 473
 474         NOTHING;
 475     }
 476
 477     return pAddress;
 478 }
 479
 480 void *
 481 AFSMapToService( IN PIRP Irp,
 482                  IN ULONG ByteCount)
 483 {
 484
 485     NTSTATUS ntStatus = STATUS_SUCCESS;
 486     void *pMappedBuffer = NULL;
 487     AFSDeviceExt *pDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
 488     KAPC stApcState;
 489
 490     __Enter
 491     {
 492
 493         if( pDevExt->Specific.Control.ServiceProcess == NULL)
 494         {
 495
 496             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
 497         }
 498
 499         if( Irp->MdlAddress == NULL)
 500         {
 501
 502             if( AFSLockSystemBuffer( Irp,
 503                                      ByteCount) == NULL)
 504             {
 505
 506                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 507             }
 508         }
 509
 510         //
 511         // Attach to the service process for mapping
 512         //
 513
 514         KeStackAttachProcess( pDevExt->Specific.Control.ServiceProcess,
 515                               (PRKAPC_STATE)&stApcState);
 516
 517         pMappedBuffer = MmMapLockedPagesSpecifyCache( Irp->MdlAddress,
 518                                                       UserMode,
 519                                                       MmCached,
 520                                                       NULL,
 521                                                       FALSE,
 522                                                       NormalPagePriority);
 523
 524         KeUnstackDetachProcess( (PRKAPC_STATE)&stApcState);
 525
 526 try_exit:
 527
 528         NOTHING;
 529     }
 530
 531     return pMappedBuffer;
 532 }
 533
 534 NTSTATUS
 535 AFSUnmapServiceMappedBuffer( IN void *MappedBuffer,
 536                              IN PMDL Mdl)
 537 {
 538
 539     NTSTATUS ntStatus = STATUS_SUCCESS;
 540     void *pMappedBuffer = NULL;
 541     AFSDeviceExt *pDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
 542     KAPC stApcState;
 543
 544     __Enter
 545     {
 546
 547         if( pDevExt->Specific.Control.ServiceProcess == NULL)
 548         {
 549
 550             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
 551         }
 552
 553         if( Mdl != NULL)
 554         {
 555
 556             //
 557             // Attach to the service process for mapping
 558             //
 559
 560             KeStackAttachProcess( pDevExt->Specific.Control.ServiceProcess,
 561                                   (PRKAPC_STATE)&stApcState);
 562
 563             MmUnmapLockedPages( MappedBuffer,
 564                                 Mdl);
 565
 566             KeUnstackDetachProcess( (PRKAPC_STATE)&stApcState);
 567         }
 568
 569 try_exit:
 570
 571         NOTHING;
 572     }
 573
 574     return ntStatus;
 575 }
 576
 577 NTSTATUS
 578 AFSInitializeLibraryDevice()
 579 {
 580
 581     NTSTATUS ntStatus = STATUS_SUCCESS;
 582     AFSDeviceExt *pDeviceExt = NULL;
 583
 584     __Enter
 585     {
 586
 587         pDeviceExt = (AFSDeviceExt *)AFSLibraryDeviceObject->DeviceExtension;
 588
 589         //
 590         // The PIOCtl file name
 591         //
 592
 593         RtlInitUnicodeString( &AFSPIOCtlName,
 594                               AFS_PIOCTL_FILE_INTERFACE_NAME);
 595
 596         //
 597         // And the global root share name
 598         //
 599
 600         RtlInitUnicodeString( &AFSGlobalRootName,
 601                               AFS_GLOBAL_ROOT_SHARE_NAME);
 602
 603     }
 604
 605     return ntStatus;
 606 }
 607
 608 NTSTATUS
 609 AFSRemoveLibraryDevice()
 610 {
 611
 612     NTSTATUS ntStatus = STATUS_SUCCESS;
 613
 614     __Enter
 615     {
 616
 617     }
 618
 619     return ntStatus;
 620 }
 621
 622 NTSTATUS
 623 AFSDefaultDispatch( IN PDEVICE_OBJECT DeviceObject,
 624                     IN PIRP Irp)
 625 {
 626
 627     NTSTATUS            ntStatus = STATUS_INVALID_DEVICE_REQUEST;
 628     PIO_STACK_LOCATION  pIrpSp = IoGetCurrentIrpStackLocation( Irp);
 629
 630     AFSCompleteRequest( Irp,
 631                         ntStatus);
 632
 633     return ntStatus;
 634 }
 635
 636 NTSTATUS
 637 AFSInitializeGlobalDirectoryEntries()
 638 {
 639
 640     NTSTATUS ntStatus = STATUS_SUCCESS;
 641     AFSDirectoryCB *pDirNode = NULL;
 642     ULONG ulEntryLength = 0;
 643     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
 644     AFSObjectInfoCB *pObjectInfoCB = NULL;
 645     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
 646     LONG lCount;
 647
 648     __Enter
 649     {
 650
 651         //
 652         // Initialize the global . entry
 653         //
 654
 655         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
 656                                                0);
 657
 658         if( pObjectInfoCB == NULL)
 659         {
 660
 661             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
 662                           AFS_TRACE_LEVEL_ERROR,
 663                           "AFSInitializeGlobalDirectory AFSAllocateObjectInfo failure %08lX\n",
 664                           ntStatus);
 665
 666             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 667         }
 668
 669         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
 670
 671         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
 672                       AFS_TRACE_LEVEL_VERBOSE,
 673                       "AFSInitializeGlobalDirectoryEntries Increment count on object %08lX Cnt %d\n",
 674                       pObjectInfoCB,
 675                       lCount);
 676
 677         ntStatus = STATUS_SUCCESS;
 678
 679         ulEntryLength = sizeof( AFSDirectoryCB) +
 680                                      sizeof( WCHAR);
 681
 682         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
 683                                                                   ulEntryLength,
 684                                                                   AFS_DIR_ENTRY_TAG);
 685
 686         if( pDirNode == NULL)
 687         {
 688
 689             AFSDeleteObjectInfo( pObjectInfoCB);
 690
 691             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
 692                           AFS_TRACE_LEVEL_ERROR,
 693                           "AFSInitializeGlobalDirectory AFS_DIR_ENTRY_TAG allocation failure\n");
 694
 695             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 696         }
 697
 698         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
 699                                                                                    sizeof( AFSNonPagedDirectoryCB),
 700                                                                                    AFS_DIR_ENTRY_NP_TAG);
 701
 702         if( pNonPagedDirEntry == NULL)
 703         {
 704
 705             ExFreePool( pDirNode);
 706
 707             AFSDeleteObjectInfo( pObjectInfoCB);
 708
 709             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
 710                           AFS_TRACE_LEVEL_ERROR,
 711                           "AFSInitializeGlobalDirectory AFS_DIR_ENTRY_NP_TAG allocation failure\n");
 712
 713             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 714         }
 715
 716         RtlZeroMemory( pDirNode,
 717                        ulEntryLength);
 718
 719         RtlZeroMemory( pNonPagedDirEntry,
 720                        sizeof( AFSNonPagedDirectoryCB));
 721
 722         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
 723
 724         pDirNode->NonPaged = pNonPagedDirEntry;
 725
 726         pDirNode->ObjectInformation = pObjectInfoCB;
 727
 728         //
 729         // Set valid entry
 730         //
 731
 732         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE | AFS_DIR_ENTRY_FAKE | AFS_DIR_ENTRY_VALID);
 733
 734         pDirNode->FileIndex = (ULONG)AFS_DIR_ENTRY_DOT_INDEX;
 735
 736         //
 737         // Setup the names in the entry
 738         //
 739
 740         pDirNode->NameInformation.FileName.Length = sizeof( WCHAR);
 741
 742         pDirNode->NameInformation.FileName.MaximumLength = sizeof( WCHAR);
 743
 744         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
 745
 746         pDirNode->NameInformation.FileName.Buffer[ 0] = L'.';
 747
 748         //
 749         // Populate the rest of the data
 750         //
 751
 752         pObjectInfoCB->FileType = AFS_FILE_TYPE_DIRECTORY;
 753
 754         pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
 755
 756         AFSGlobalDotDirEntry = pDirNode;
 757
 758         //
 759         // Now the .. entry
 760         //
 761
 762         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
 763                                                0);
 764
 765         if( pObjectInfoCB == NULL)
 766         {
 767
 768             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
 769                           AFS_TRACE_LEVEL_ERROR,
 770                           "AFSInitializeGlobalDirectory AFSAllocateObjectInfo (2) failure %08lX\n",
 771                           ntStatus);
 772
 773             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 774         }
 775
 776         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
 777
 778         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
 779                       AFS_TRACE_LEVEL_VERBOSE,
 780                       "AFSInitializeGlobalDirectoryEntries Increment count on object %08lX Cnt %d\n",
 781                       pObjectInfoCB,
 782                       lCount);
 783
 784         ntStatus = STATUS_SUCCESS;
 785
 786         ulEntryLength = sizeof( AFSDirectoryCB) +
 787                                      ( 2 * sizeof( WCHAR));
 788
 789         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
 790                                                                   ulEntryLength,
 791                                                                   AFS_DIR_ENTRY_TAG);
 792
 793         if( pDirNode == NULL)
 794         {
 795
 796             AFSDeleteObjectInfo( pObjectInfoCB);
 797
 798             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 799         }
 800
 801         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
 802                                                                                    sizeof( AFSNonPagedDirectoryCB),
 803                                                                                    AFS_DIR_ENTRY_NP_TAG);
 804
 805         if( pNonPagedDirEntry == NULL)
 806         {
 807
 808             ExFreePool( pDirNode);
 809
 810             AFSDeleteObjectInfo( pObjectInfoCB);
 811
 812             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 813         }
 814
 815         RtlZeroMemory( pDirNode,
 816                        ulEntryLength);
 817
 818         RtlZeroMemory( pNonPagedDirEntry,
 819                        sizeof( AFSNonPagedDirectoryCB));
 820
 821         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
 822
 823         pDirNode->NonPaged = pNonPagedDirEntry;
 824
 825         pDirNode->ObjectInformation = pObjectInfoCB;
 826
 827         //
 828         // Set valid entry
 829         //
 830
 831         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE | AFS_DIR_ENTRY_FAKE | AFS_DIR_ENTRY_VALID);
 832
 833         pDirNode->FileIndex = (ULONG)AFS_DIR_ENTRY_DOT_DOT_INDEX;
 834
 835         //
 836         // Setup the names in the entry
 837         //
 838
 839         pDirNode->NameInformation.FileName.Length = 2 * sizeof( WCHAR);
 840
 841         pDirNode->NameInformation.FileName.MaximumLength = 2 * sizeof( WCHAR);
 842
 843         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
 844
 845         pDirNode->NameInformation.FileName.Buffer[ 0] = L'.';
 846
 847         pDirNode->NameInformation.FileName.Buffer[ 1] = L'.';
 848
 849         //
 850         // Populate the rest of the data
 851         //
 852
 853         pObjectInfoCB->FileType = AFS_FILE_TYPE_DIRECTORY;
 854
 855         pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
 856
 857         AFSGlobalDotDotDirEntry = pDirNode;
 858
 859 try_exit:
 860
 861         if( !NT_SUCCESS( ntStatus))
 862         {
 863
 864             if( AFSGlobalDotDirEntry != NULL)
 865             {
 866
 867                 AFSDeleteObjectInfo( AFSGlobalDotDirEntry->ObjectInformation);
 868
 869                 ExDeleteResourceLite( &AFSGlobalDotDirEntry->NonPaged->Lock);
 870
 871                 ExFreePool( AFSGlobalDotDirEntry->NonPaged);
 872
 873                 ExFreePool( AFSGlobalDotDirEntry);
 874
 875                 AFSGlobalDotDirEntry = NULL;
 876             }
 877
 878             if( AFSGlobalDotDotDirEntry != NULL)
 879             {
 880
 881                 AFSDeleteObjectInfo( AFSGlobalDotDotDirEntry->ObjectInformation);
 882
 883                 ExDeleteResourceLite( &AFSGlobalDotDotDirEntry->NonPaged->Lock);
 884
 885                 ExFreePool( AFSGlobalDotDotDirEntry->NonPaged);
 886
 887                 ExFreePool( AFSGlobalDotDotDirEntry);
 888
 889                 AFSGlobalDotDotDirEntry = NULL;
 890             }
 891         }
 892     }
 893
 894     return ntStatus;
 895 }
 896
 897 AFSDirectoryCB *
 898 AFSInitDirEntry( IN AFSObjectInfoCB *ParentObjectInfo,
 899                  IN PUNICODE_STRING FileName,
 900                  IN PUNICODE_STRING TargetName,
 901                  IN AFSDirEnumEntry *DirEnumEntry,
 902                  IN ULONG FileIndex)
 903 {
 904
 905     AFSDirectoryCB *pDirNode = NULL;
 906     NTSTATUS ntStatus = STATUS_SUCCESS;
 907     ULONG ulEntryLength = 0;
 908     AFSDirEnumEntry *pDirEnumCB = NULL;
 909     AFSFileID stTargetFileID;
 910     AFSFcb *pVcb = NULL;
 911     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
 912     AFSObjectInfoCB *pObjectInfoCB = NULL;
 913     BOOLEAN bAllocatedObjectCB = FALSE;
 914     ULONGLONG ullIndex = 0;
 915     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
 916     LONG lCount;
 917
 918     __Enter
 919     {
 920
 921         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
 922                       AFS_TRACE_LEVEL_VERBOSE,
 923                       "AFSInitDirEntry Initializing entry %wZ parent FID %08lX-%08lX-%08lX-%08lX\n",
 924                       FileName,
 925                       ParentObjectInfo->FileId.Cell,
 926                       ParentObjectInfo->FileId.Volume,
 927                       ParentObjectInfo->FileId.Vnode,
 928                       ParentObjectInfo->FileId.Unique);
 929
 930         //
 931         // First thing is to locate/create our object information block
 932         // for this entry
 933         //
 934
 935         AFSAcquireExcl( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock,
 936                         TRUE);
 937
 938         ullIndex = AFSCreateLowIndex( &DirEnumEntry->FileId);
 939
 940         ntStatus = AFSLocateHashEntry( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
 941                                        ullIndex,
 942                                        (AFSBTreeEntry **)&pObjectInfoCB);
 943
 944         if( !NT_SUCCESS( ntStatus) ||
 945             pObjectInfoCB == NULL)
 946         {
 947
 948             //
 949             // Allocate our object info cb
 950             //
 951
 952             pObjectInfoCB = AFSAllocateObjectInfo( ParentObjectInfo,
 953                                                    ullIndex);
 954
 955             if( pObjectInfoCB == NULL)
 956             {
 957
 958                 AFSReleaseResource( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
 959
 960                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 961             }
 962
 963             bAllocatedObjectCB = TRUE;
 964
 965             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
 966                           AFS_TRACE_LEVEL_VERBOSE,
 967                           "AFSInitDirEntry initialized object %08lX Parent Object %08lX for %wZ\n",
 968                           pObjectInfoCB,
 969                           ParentObjectInfo,
 970                           FileName);
 971         }
 972
 973         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
 974
 975         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
 976                       AFS_TRACE_LEVEL_VERBOSE,
 977                       "AFSInitDirEntry Increment count on object %08lX Cnt %d\n",
 978                       pObjectInfoCB,
 979                       lCount);
 980
 981         AFSReleaseResource( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
 982
 983         ntStatus = STATUS_SUCCESS;
 984
 985         ulEntryLength = sizeof( AFSDirectoryCB) +
 986                                      FileName->Length;
 987
 988         if( TargetName != NULL)
 989         {
 990
 991             ulEntryLength += TargetName->Length;
 992         }
 993
 994         pDirNode = (AFSDirectoryCB *)AFSExAllocatePoolWithTag( PagedPool,
 995                                                                ulEntryLength,
 996                                                                AFS_DIR_ENTRY_TAG);
 997
 998         if( pDirNode == NULL)
 999         {
1000
1001             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1002         }
1003
1004         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSExAllocatePoolWithTag( NonPagedPool,
1005                                                                                 sizeof( AFSNonPagedDirectoryCB),
1006                                                                                 AFS_DIR_ENTRY_NP_TAG);
1007
1008         if( pNonPagedDirEntry == NULL)
1009         {
1010
1011             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1012         }
1013
1014         RtlZeroMemory( pDirNode,
1015                        ulEntryLength);
1016
1017         RtlZeroMemory( pNonPagedDirEntry,
1018                        sizeof( AFSNonPagedDirectoryCB));
1019
1020         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
1021
1022         pDirNode->NonPaged = pNonPagedDirEntry;
1023
1024         pDirNode->ObjectInformation = pObjectInfoCB;
1025
1026         //
1027         // Set valid entry and NOT_IN_PARENT flag
1028         //
1029
1030         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_NOT_IN_PARENT_TREE);
1031
1032         pDirNode->FileIndex = FileIndex;
1033
1034         //
1035         // Setup the names in the entry
1036         //
1037
1038         if( FileName->Length > 0)
1039         {
1040
1041             pDirNode->NameInformation.FileName.Length = FileName->Length;
1042
1043             pDirNode->NameInformation.FileName.MaximumLength = FileName->Length;
1044
1045             pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
1046
1047             RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
1048                            FileName->Buffer,
1049                            pDirNode->NameInformation.FileName.Length);
1050
1051             //
1052             // Create a CRC for the file
1053             //
1054
1055             pDirNode->CaseSensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
1056                                                                          FALSE);
1057
1058             pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
1059                                                                            TRUE);
1060         }
1061
1062         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1063                       AFS_TRACE_LEVEL_VERBOSE,
1064                       "AFSInitDirEntry Initialized DE %p for %wZ in parent FID %08lX-%08lX-%08lX-%08lX\n",
1065                       pDirNode,
1066                       FileName,
1067                       ParentObjectInfo->FileId.Cell,
1068                       ParentObjectInfo->FileId.Volume,
1069                       ParentObjectInfo->FileId.Vnode,
1070                       ParentObjectInfo->FileId.Unique);
1071
1072         if( TargetName != NULL &&
1073             TargetName->Length > 0)
1074         {
1075
1076             pDirNode->NameInformation.TargetName.Length = TargetName->Length;
1077
1078             pDirNode->NameInformation.TargetName.MaximumLength = pDirNode->NameInformation.TargetName.Length;
1079
1080             pDirNode->NameInformation.TargetName.Buffer = (WCHAR *)((char *)pDirNode +
1081                                                                             sizeof( AFSDirectoryCB) +
1082                                                                             pDirNode->NameInformation.FileName.Length);
1083
1084             RtlCopyMemory( pDirNode->NameInformation.TargetName.Buffer,
1085                            TargetName->Buffer,
1086                            pDirNode->NameInformation.TargetName.Length);
1087         }
1088
1089         //
1090         // If we allocated the object information cb then update the information
1091         //
1092
1093         if( bAllocatedObjectCB)
1094         {
1095
1096             //
1097             // Populate the rest of the data
1098             //
1099
1100             pObjectInfoCB->FileId = DirEnumEntry->FileId;
1101
1102             pObjectInfoCB->TargetFileId = DirEnumEntry->TargetFileId;
1103
1104             pObjectInfoCB->FileType = DirEnumEntry->FileType;
1105
1106             pObjectInfoCB->CreationTime = DirEnumEntry->CreationTime;
1107
1108             pObjectInfoCB->LastAccessTime = DirEnumEntry->LastAccessTime;
1109
1110             pObjectInfoCB->LastWriteTime = DirEnumEntry->LastWriteTime;
1111
1112             pObjectInfoCB->ChangeTime = DirEnumEntry->ChangeTime;
1113
1114             pObjectInfoCB->EndOfFile = DirEnumEntry->EndOfFile;
1115
1116             pObjectInfoCB->AllocationSize = DirEnumEntry->AllocationSize;
1117
1118             pObjectInfoCB->FileAttributes = DirEnumEntry->FileAttributes;
1119
1120             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1121             {
1122
1123                 pObjectInfoCB->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1124             }
1125
1126             if (pObjectInfoCB->FileType == AFS_FILE_TYPE_SYMLINK ||
1127                 pObjectInfoCB->FileType == AFS_FILE_TYPE_DFSLINK)
1128             {
1129
1130                 pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1131             }
1132
1133             pObjectInfoCB->EaSize = DirEnumEntry->EaSize;
1134
1135             //
1136             // Object specific information
1137             //
1138
1139             pObjectInfoCB->Links = DirEnumEntry->Links;
1140
1141             pObjectInfoCB->Expiration = DirEnumEntry->Expiration;
1142
1143             pObjectInfoCB->DataVersion = DirEnumEntry->DataVersion;
1144
1145             //
1146             // Check for the case where we have a filetype of SymLink but both the TargetFid and the
1147             // TargetName are empty. In this case set the filetype to zero so we evaluate it later in
1148             // the code
1149             //
1150
1151             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_SYMLINK &&
1152                 pObjectInfoCB->TargetFileId.Vnode == 0 &&
1153                 pObjectInfoCB->TargetFileId.Unique == 0 &&
1154                 pDirNode->NameInformation.TargetName.Length == 0)
1155             {
1156
1157                 //
1158                 // This will ensure we perform a validation on the node
1159                 //
1160
1161                 pObjectInfoCB->FileType = AFS_FILE_TYPE_UNKNOWN;
1162             }
1163
1164             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_UNKNOWN)
1165             {
1166
1167                 SetFlag( pObjectInfoCB->Flags, AFS_OBJECT_FLAGS_NOT_EVALUATED);
1168             }
1169         }
1170
1171 try_exit:
1172
1173         if( !NT_SUCCESS( ntStatus))
1174         {
1175
1176             if( pNonPagedDirEntry != NULL)
1177             {
1178
1179                 ExDeleteResourceLite( &pNonPagedDirEntry->Lock);
1180
1181                 AFSExFreePool( pNonPagedDirEntry);
1182             }
1183
1184             if( pDirNode != NULL)
1185             {
1186
1187                 AFSExFreePool( pDirNode);
1188
1189                 pDirNode = NULL;
1190             }
1191
1192             //
1193             // Dereference our object info block if we have one
1194             //
1195
1196             if( pObjectInfoCB != NULL)
1197             {
1198
1199                 lCount = InterlockedDecrement( &pObjectInfoCB->ObjectReferenceCount);
1200
1201                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
1202                               AFS_TRACE_LEVEL_VERBOSE,
1203                               "AFSInitDirEntry Decrement count on object %08lX Cnt %d\n",
1204                               pObjectInfoCB,
1205                               lCount);
1206
1207                 if( bAllocatedObjectCB)
1208                 {
1209
1210                     ASSERT( pObjectInfoCB->ObjectReferenceCount == 0);
1211
1212                     AFSDeleteObjectInfo( pObjectInfoCB);
1213                 }
1214             }
1215         }
1216     }
1217
1218     return pDirNode;
1219 }
1220
1221 BOOLEAN
1222 AFSCheckForReadOnlyAccess( IN ACCESS_MASK DesiredAccess,
1223                            IN BOOLEAN DirectoryEntry)
1224 {
1225
1226     BOOLEAN bReturn = TRUE;
1227     ACCESS_MASK stAccessMask = 0;
1228
1229     //
1230     // Get rid of anything we don't know about
1231     //
1232
1233     DesiredAccess = (DesiredAccess   &
1234                           ( DELETE |
1235                             READ_CONTROL |
1236                             WRITE_OWNER |
1237                             WRITE_DAC |
1238                             SYNCHRONIZE |
1239                             ACCESS_SYSTEM_SECURITY |
1240                             FILE_WRITE_DATA |
1241                             FILE_READ_EA |
1242                             FILE_WRITE_EA |
1243                             FILE_READ_ATTRIBUTES |
1244                             FILE_WRITE_ATTRIBUTES |
1245                             FILE_LIST_DIRECTORY |
1246                             FILE_TRAVERSE |
1247                             FILE_DELETE_CHILD |
1248                             FILE_APPEND_DATA));
1249
1250     //
1251     // Our 'read only' access mask. These are the accesses we will
1252     // allow for a read only file
1253     //
1254
1255     stAccessMask = DELETE |
1256                         READ_CONTROL |
1257                         WRITE_OWNER |
1258                         WRITE_DAC |
1259                         SYNCHRONIZE |
1260                         ACCESS_SYSTEM_SECURITY |
1261                         FILE_READ_DATA |
1262                         FILE_READ_EA |
1263                         FILE_WRITE_EA |
1264                         FILE_READ_ATTRIBUTES |
1265                         FILE_WRITE_ATTRIBUTES |
1266                         FILE_EXECUTE |
1267                         FILE_LIST_DIRECTORY |
1268                         FILE_TRAVERSE;
1269
1270     //
1271     // For a directory, add in the directory specific accesses
1272     //
1273
1274     if( DirectoryEntry)
1275     {
1276
1277         stAccessMask |= FILE_ADD_SUBDIRECTORY |
1278                                 FILE_ADD_FILE |
1279                                 FILE_DELETE_CHILD;
1280     }
1281
1282     if( FlagOn( DesiredAccess, ~stAccessMask))
1283     {
1284
1285         //
1286         // A write access is set ...
1287         //
1288
1289         bReturn = FALSE;
1290     }
1291
1292     return bReturn;
1293 }
1294
1295 NTSTATUS
1296 AFSEvaluateNode( IN GUID *AuthGroup,
1297                  IN AFSDirectoryCB *DirEntry)
1298 {
1299
1300     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1301     NTSTATUS ntStatus = STATUS_SUCCESS;
1302     AFSDirEnumEntry *pDirEntry = NULL;
1303     UNICODE_STRING uniTargetName;
1304
1305     __Enter
1306     {
1307
1308         ntStatus = AFSEvaluateTargetByID( DirEntry->ObjectInformation,
1309                                           AuthGroup,
1310                                           FALSE,
1311                                           &pDirEntry);
1312
1313         if( !NT_SUCCESS( ntStatus))
1314         {
1315
1316             try_return( ntStatus);
1317         }
1318
1319         DirEntry->ObjectInformation->TargetFileId = pDirEntry->TargetFileId;
1320
1321         DirEntry->ObjectInformation->Expiration = pDirEntry->Expiration;
1322
1323         DirEntry->ObjectInformation->DataVersion = pDirEntry->DataVersion;
1324
1325         DirEntry->ObjectInformation->FileType = pDirEntry->FileType;
1326
1327         DirEntry->ObjectInformation->CreationTime = pDirEntry->CreationTime;
1328
1329         DirEntry->ObjectInformation->LastAccessTime = pDirEntry->LastAccessTime;
1330
1331         DirEntry->ObjectInformation->LastWriteTime = pDirEntry->LastWriteTime;
1332
1333         DirEntry->ObjectInformation->ChangeTime = pDirEntry->ChangeTime;
1334
1335         DirEntry->ObjectInformation->EndOfFile = pDirEntry->EndOfFile;
1336
1337         DirEntry->ObjectInformation->AllocationSize = pDirEntry->AllocationSize;
1338
1339         DirEntry->ObjectInformation->FileAttributes = pDirEntry->FileAttributes;
1340
1341         if( pDirEntry->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1342         {
1343
1344             DirEntry->ObjectInformation->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1345         }
1346
1347         if( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK ||
1348             pDirEntry->FileType == AFS_FILE_TYPE_DFSLINK)
1349         {
1350
1351             DirEntry->ObjectInformation->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1352         }
1353
1354         DirEntry->ObjectInformation->EaSize = pDirEntry->EaSize;
1355
1356         DirEntry->ObjectInformation->Links = pDirEntry->Links;
1357
1358         //
1359         // If we have a target name then see if it needs updating ...
1360         //
1361
1362         if( pDirEntry->TargetNameLength > 0)
1363         {
1364
1365             //
1366             // Update the target name information if needed
1367             //
1368
1369             uniTargetName.Length = (USHORT)pDirEntry->TargetNameLength;
1370
1371             uniTargetName.MaximumLength = uniTargetName.Length;
1372
1373             uniTargetName.Buffer = (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset);
1374
1375             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
1376                             TRUE);
1377
1378             if( DirEntry->NameInformation.TargetName.Length == 0 ||
1379                 RtlCompareUnicodeString( &uniTargetName,
1380                                          &DirEntry->NameInformation.TargetName,
1381                                          TRUE) != 0)
1382             {
1383
1384                 //
1385                 // Update the target name
1386                 //
1387
1388                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
1389                                                 &DirEntry->Flags,
1390                                                 uniTargetName.Buffer,
1391                                                 uniTargetName.Length);
1392
1393                 if( !NT_SUCCESS( ntStatus))
1394                 {
1395
1396                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
1397
1398                     try_return( ntStatus);
1399                 }
1400             }
1401
1402             AFSReleaseResource( &DirEntry->NonPaged->Lock);
1403         }
1404
1405 try_exit:
1406
1407         if( pDirEntry != NULL)
1408         {
1409
1410             AFSExFreePool( pDirEntry);
1411         }
1412     }
1413
1414     return ntStatus;
1415 }
1416
1417 NTSTATUS
1418 AFSValidateSymLink( IN GUID *AuthGroup,
1419                     IN AFSDirectoryCB *DirEntry)
1420 {
1421
1422     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1423     NTSTATUS ntStatus = STATUS_SUCCESS;
1424     AFSDirEnumEntry *pDirEntry = NULL;
1425     UNICODE_STRING uniTargetName;
1426
1427     __Enter
1428     {
1429
1430         ntStatus = AFSEvaluateTargetByID( DirEntry->ObjectInformation,
1431                                           AuthGroup,
1432                                           FALSE,
1433                                           &pDirEntry);
1434
1435         if( !NT_SUCCESS( ntStatus))
1436         {
1437
1438             try_return( ntStatus);
1439         }
1440
1441         if( pDirEntry->FileType == AFS_FILE_TYPE_UNKNOWN ||
1442             pDirEntry->FileType == AFS_FILE_TYPE_INVALID)
1443         {
1444
1445             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
1446         }
1447
1448         DirEntry->ObjectInformation->TargetFileId = pDirEntry->TargetFileId;
1449
1450         DirEntry->ObjectInformation->Expiration = pDirEntry->Expiration;
1451
1452         DirEntry->ObjectInformation->DataVersion = pDirEntry->DataVersion;
1453
1454         //
1455         // Update the target name information if needed
1456         //
1457
1458         uniTargetName.Length = (USHORT)pDirEntry->TargetNameLength;
1459
1460         uniTargetName.MaximumLength = uniTargetName.Length;
1461
1462         uniTargetName.Buffer = (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset);
1463
1464         if( uniTargetName.Length > 0)
1465         {
1466
1467             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
1468                             TRUE);
1469
1470             if( DirEntry->NameInformation.TargetName.Length == 0 ||
1471                 RtlCompareUnicodeString( &uniTargetName,
1472                                          &DirEntry->NameInformation.TargetName,
1473                                          TRUE) != 0)
1474             {
1475
1476                 //
1477                 // Update the target name
1478                 //
1479
1480                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
1481                                                 &DirEntry->Flags,
1482                                                 uniTargetName.Buffer,
1483                                                 uniTargetName.Length);
1484
1485                 if( !NT_SUCCESS( ntStatus))
1486                 {
1487
1488                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
1489
1490                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1491                 }
1492             }
1493
1494             AFSReleaseResource( &DirEntry->NonPaged->Lock);
1495         }
1496
1497         //
1498         // If the FileType is the same then nothing to do since it IS
1499         // a SymLink
1500         //
1501
1502         if( pDirEntry->FileType == DirEntry->ObjectInformation->FileType)
1503         {
1504
1505             ASSERT( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK);
1506
1507             try_return( ntStatus = STATUS_SUCCESS);
1508         }
1509
1510         DirEntry->ObjectInformation->FileType = pDirEntry->FileType;
1511
1512         DirEntry->ObjectInformation->CreationTime = pDirEntry->CreationTime;
1513
1514         DirEntry->ObjectInformation->LastAccessTime = pDirEntry->LastAccessTime;
1515
1516         DirEntry->ObjectInformation->LastWriteTime = pDirEntry->LastWriteTime;
1517
1518         DirEntry->ObjectInformation->ChangeTime = pDirEntry->ChangeTime;
1519
1520         DirEntry->ObjectInformation->EndOfFile = pDirEntry->EndOfFile;
1521
1522         DirEntry->ObjectInformation->AllocationSize = pDirEntry->AllocationSize;
1523
1524         DirEntry->ObjectInformation->FileAttributes = pDirEntry->FileAttributes;
1525
1526         if( pDirEntry->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1527         {
1528
1529             DirEntry->ObjectInformation->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1530         }
1531
1532         if( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK ||
1533             pDirEntry->FileType == AFS_FILE_TYPE_DFSLINK)
1534         {
1535
1536             DirEntry->ObjectInformation->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1537         }
1538
1539         DirEntry->ObjectInformation->EaSize = pDirEntry->EaSize;
1540
1541         DirEntry->ObjectInformation->Links = pDirEntry->Links;
1542
1543 try_exit:
1544
1545         if( pDirEntry != NULL)
1546         {
1547
1548             AFSExFreePool( pDirEntry);
1549         }
1550     }
1551
1552     return ntStatus;
1553 }
1554
1555 NTSTATUS
1556 AFSInvalidateObject( IN OUT AFSObjectInfoCB **ppObjectInfo,
1557                      IN     ULONG Reason)
1558 {
1559
1560     NTSTATUS ntStatus = STATUS_SUCCESS;
1561     IO_STATUS_BLOCK stIoStatus;
1562     ULONG ulFilter = 0;
1563
1564     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1565                   AFS_TRACE_LEVEL_VERBOSE,
1566                   "AFSInvalidateObject Invalidation on node type %d for fid %08lX-%08lX-%08lX-%08lX Reason %d\n",
1567                   (*ppObjectInfo)->FileType,
1568                   (*ppObjectInfo)->FileId.Cell,
1569                   (*ppObjectInfo)->FileId.Volume,
1570                   (*ppObjectInfo)->FileId.Vnode,
1571                   (*ppObjectInfo)->FileId.Unique,
1572                   Reason);
1573
1574     if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_SYMLINK ||
1575         (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DFSLINK ||
1576         (*ppObjectInfo)->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1577     {
1578         //
1579         // We only act on the mount point itself, not the target. If the
1580         // node has been deleted then mark it as such otherwise indicate
1581         // it requires verification
1582         //
1583
1584         if( Reason == AFS_INVALIDATE_DELETED)
1585         {
1586             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
1587         }
1588         else
1589         {
1590
1591             if( Reason == AFS_INVALIDATE_FLUSHED)
1592             {
1593
1594                 (*ppObjectInfo)->DataVersion.QuadPart = (ULONGLONG)-1;
1595
1596                 SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
1597             }
1598
1599             (*ppObjectInfo)->Expiration.QuadPart = 0;
1600
1601             (*ppObjectInfo)->TargetFileId.Vnode = 0;
1602
1603             (*ppObjectInfo)->TargetFileId.Unique = 0;
1604
1605             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1606                           AFS_TRACE_LEVEL_VERBOSE,
1607                           "AFSInvalidateObject Setting VERIFY flag on fid %08lX-%08lX-%08lX-%08lX\n",
1608                           (*ppObjectInfo)->FileId.Cell,
1609                           (*ppObjectInfo)->FileId.Volume,
1610                           (*ppObjectInfo)->FileId.Vnode,
1611                           (*ppObjectInfo)->FileId.Unique);
1612
1613             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY);
1614         }
1615
1616         ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1617
1618         if( Reason == AFS_INVALIDATE_CREDS)
1619         {
1620             ulFilter |= FILE_NOTIFY_CHANGE_SECURITY;
1621         }
1622
1623         if( Reason == AFS_INVALIDATE_DATA_VERSION ||
1624             Reason == AFS_INVALIDATE_FLUSHED)
1625         {
1626             ulFilter |= FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE;
1627         }
1628         else
1629         {
1630             ulFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1631         }
1632
1633         AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1634                                         NULL,
1635                                         ulFilter,
1636                                         FILE_ACTION_MODIFIED);
1637
1638         try_return( ntStatus);
1639     }
1640
1641     //
1642     // Depending on the reason for invalidation then perform work on the node
1643     //
1644
1645     switch( Reason)
1646     {
1647
1648     case AFS_INVALIDATE_DELETED:
1649         {
1650
1651             //
1652             // Mark this node as invalid
1653             //
1654
1655             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_DELETED);
1656
1657             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1658                           AFS_TRACE_LEVEL_VERBOSE,
1659                           "AFSInvalidateObject Set DELETE flag on fid %08lX-%08lX-%08lX-%08lX\n",
1660                           (*ppObjectInfo)->FileId.Cell,
1661                           (*ppObjectInfo)->FileId.Volume,
1662                           (*ppObjectInfo)->FileId.Vnode,
1663                           (*ppObjectInfo)->FileId.Unique);
1664
1665             if( (*ppObjectInfo)->ParentObjectInformation != NULL)
1666             {
1667
1668                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1669                               AFS_TRACE_LEVEL_VERBOSE,
1670                               "AFSInvalidateObject Set VERIFY flag on parent fid %08lX-%08lX-%08lX-%08lX\n",
1671                               (*ppObjectInfo)->ParentObjectInformation->FileId.Cell,
1672                               (*ppObjectInfo)->ParentObjectInformation->FileId.Volume,
1673                               (*ppObjectInfo)->ParentObjectInformation->FileId.Vnode,
1674                               (*ppObjectInfo)->ParentObjectInformation->FileId.Unique);
1675
1676                 SetFlag( (*ppObjectInfo)->ParentObjectInformation->Flags, AFS_OBJECT_FLAGS_VERIFY);
1677
1678                 (*ppObjectInfo)->ParentObjectInformation->DataVersion.QuadPart = (ULONGLONG)-1;
1679
1680                 (*ppObjectInfo)->ParentObjectInformation->Expiration.QuadPart = 0;
1681             }
1682
1683             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DIRECTORY)
1684             {
1685                 ulFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
1686             }
1687             else
1688             {
1689                 ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1690             }
1691
1692             AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1693                                             NULL,
1694                                             ulFilter,
1695                                             FILE_ACTION_REMOVED);
1696
1697             if( NT_SUCCESS( AFSQueueInvalidateObject( (*ppObjectInfo),
1698                                                       Reason)))
1699             {
1700                 (*ppObjectInfo) = NULL; // We'll dec the count in the worker item
1701             }
1702
1703             break;
1704         }
1705
1706     case AFS_INVALIDATE_FLUSHED:
1707         {
1708
1709             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE &&
1710                 (*ppObjectInfo)->Fcb != NULL)
1711             {
1712
1713                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1714                               AFS_TRACE_LEVEL_VERBOSE,
1715                               "AFSInvalidateObject Flush/purge file fid %08lX-%08lX-%08lX-%08lX\n",
1716                               (*ppObjectInfo)->FileId.Cell,
1717                               (*ppObjectInfo)->FileId.Volume,
1718                               (*ppObjectInfo)->FileId.Vnode,
1719                               (*ppObjectInfo)->FileId.Unique);
1720
1721                 AFSAcquireExcl( &(*ppObjectInfo)->Fcb->NPFcb->Resource,
1722                                 TRUE);
1723
1724                 __try
1725                 {
1726
1727                     CcFlushCache( &(*ppObjectInfo)->Fcb->NPFcb->SectionObjectPointers,
1728                                   NULL,
1729                                   0,
1730                                   &stIoStatus);
1731
1732                     if( !NT_SUCCESS( stIoStatus.Status))
1733                     {
1734
1735                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
1736                                       AFS_TRACE_LEVEL_ERROR,
1737                                       "AFSInvalidateObject CcFlushCache failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
1738                                       (*ppObjectInfo)->FileId.Cell,
1739                                       (*ppObjectInfo)->FileId.Volume,
1740                                       (*ppObjectInfo)->FileId.Vnode,
1741                                       (*ppObjectInfo)->FileId.Unique,
1742                                       stIoStatus.Status,
1743                                       stIoStatus.Information);
1744
1745                         ntStatus = stIoStatus.Status;
1746                     }
1747
1748                     CcPurgeCacheSection( &(*ppObjectInfo)->Fcb->NPFcb->SectionObjectPointers,
1749                                          NULL,
1750                                          0,
1751                                          FALSE);
1752                 }
1753                 __except( EXCEPTION_EXECUTE_HANDLER)
1754                 {
1755
1756                     ntStatus = GetExceptionCode();
1757                 }
1758
1759                 AFSReleaseResource( &(*ppObjectInfo)->Fcb->NPFcb->Resource);
1760
1761                 //
1762                 // Clear out the extents
1763                 // Get rid of them (note this involves waiting
1764                 // for any writes or reads to the cache to complete)
1765                 //
1766
1767                 (VOID) AFSTearDownFcbExtents( (*ppObjectInfo)->Fcb,
1768                                               NULL);
1769             }
1770
1771             (*ppObjectInfo)->DataVersion.QuadPart = (ULONGLONG)-1;
1772
1773
1774             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE)
1775             {
1776
1777                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1778                               AFS_TRACE_LEVEL_VERBOSE,
1779                               "AFSInvalidateObject Setting VERIFY_DATA flag on fid %08lX-%08lX-%08lX-%08lX\n",
1780                               (*ppObjectInfo)->FileId.Cell,
1781                               (*ppObjectInfo)->FileId.Volume,
1782                               (*ppObjectInfo)->FileId.Vnode,
1783                               (*ppObjectInfo)->FileId.Unique);
1784
1785                 SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
1786             }
1787
1788             // Fall through to the default processing
1789         }
1790
1791     default:
1792         {
1793
1794             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DIRECTORY)
1795             {
1796                 ulFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
1797             }
1798             else
1799             {
1800                 ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1801             }
1802
1803             if( Reason == AFS_INVALIDATE_CREDS)
1804             {
1805                 ulFilter |= FILE_NOTIFY_CHANGE_SECURITY;
1806             }
1807
1808             if( Reason == AFS_INVALIDATE_DATA_VERSION)
1809             {
1810                 ulFilter |= FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE;
1811             }
1812             else
1813             {
1814                 ulFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1815             }
1816
1817             AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1818                                             NULL,
1819                                             ulFilter,
1820                                             FILE_ACTION_MODIFIED);
1821
1822             //
1823             // Indicate this node requires re-evaluation for the remaining reasons
1824             //
1825
1826             (*ppObjectInfo)->Expiration.QuadPart = 0;
1827
1828             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1829                           AFS_TRACE_LEVEL_VERBOSE,
1830                           "AFSInvalidateObject Setting VERIFY flag on fid %08lX-%08lX-%08lX-%08lX\n",
1831                           (*ppObjectInfo)->FileId.Cell,
1832                           (*ppObjectInfo)->FileId.Volume,
1833                           (*ppObjectInfo)->FileId.Vnode,
1834                           (*ppObjectInfo)->FileId.Unique);
1835
1836             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY);
1837
1838             if( Reason == AFS_INVALIDATE_DATA_VERSION ||
1839                 (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE &&
1840                 ( Reason == AFS_INVALIDATE_CALLBACK ||
1841                   Reason == AFS_INVALIDATE_EXPIRED))
1842             {
1843                 if ( NT_SUCCESS( AFSQueueInvalidateObject( (*ppObjectInfo),
1844                                                            AFS_INVALIDATE_DATA_VERSION)))
1845                 {
1846
1847                     (*ppObjectInfo) = NULL; // We'll dec the count in the worker item
1848                 }
1849             }
1850
1851             break;
1852         }
1853     }
1854
1855   try_exit:
1856
1857     return ntStatus;
1858 }
1859
1860 NTSTATUS
1861 AFSInvalidateCache( IN AFSInvalidateCacheCB *InvalidateCB)
1862 {
1863
1864     NTSTATUS ntStatus = STATUS_SUCCESS;
1865     AFSFcb      *pDcb = NULL, *pFcb = NULL, *pNextFcb = NULL;
1866     AFSVolumeCB *pVolumeCB = NULL;
1867     AFSFcb      *pTargetDcb = NULL;
1868     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
1869     AFSDirectoryCB *pCurrentDirEntry = NULL;
1870     BOOLEAN     bIsChild = FALSE;
1871     ULONGLONG   ullIndex = 0;
1872     AFSObjectInfoCB *pObjectInfo = NULL;
1873     LONG lCount;
1874
1875     __Enter
1876     {
1877
1878         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1879                       AFS_TRACE_LEVEL_VERBOSE,
1880                       "AFSInvalidateCache Invalidation FID %08lX-%08lX-%08lX-%08lX Type %d WholeVolume %d Reason %d\n",
1881                       InvalidateCB->FileID.Cell,
1882                       InvalidateCB->FileID.Volume,
1883                       InvalidateCB->FileID.Vnode,
1884                       InvalidateCB->FileID.Unique,
1885                       InvalidateCB->FileType,
1886                       InvalidateCB->WholeVolume,
1887                       InvalidateCB->Reason);
1888
1889         //
1890         // Need to locate the Fcb for the directory to purge
1891         //
1892
1893         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
1894                       AFS_TRACE_LEVEL_VERBOSE,
1895                       "AFSInvalidateCache Acquiring RDR VolumeTreeLock lock %08lX SHARED %08lX\n",
1896                       &pDevExt->Specific.RDR.VolumeTreeLock,
1897                       PsGetCurrentThread());
1898
1899         //
1900         // Starve any exclusive waiters on this paticular call
1901         //
1902
1903         AFSAcquireSharedStarveExclusive( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
1904
1905         //
1906         // Locate the volume node
1907         //
1908
1909         ullIndex = AFSCreateHighIndex( &InvalidateCB->FileID);
1910
1911         ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
1912                                        ullIndex,
1913                                        (AFSBTreeEntry **)&pVolumeCB);
1914
1915         if( pVolumeCB != NULL)
1916         {
1917
1918             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
1919
1920             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
1921                           AFS_TRACE_LEVEL_VERBOSE,
1922                           "AFSInvalidateCache Increment count on volume %08lX Cnt %d\n",
1923                           pVolumeCB,
1924                           lCount);
1925         }
1926
1927         AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
1928
1929         if( !NT_SUCCESS( ntStatus) ||
1930             pVolumeCB == NULL)
1931         {
1932
1933             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1934                           AFS_TRACE_LEVEL_WARNING,
1935                           "AFSInvalidateCache Invalidation FAILURE Unable to locate volume node FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
1936                           InvalidateCB->FileID.Cell,
1937                           InvalidateCB->FileID.Volume,
1938                           InvalidateCB->FileID.Vnode,
1939                           InvalidateCB->FileID.Unique,
1940                           ntStatus);
1941
1942             try_return( ntStatus = STATUS_SUCCESS);
1943         }
1944
1945         //
1946         // If this is a whole volume invalidation then go do it now
1947         //
1948
1949         if( InvalidateCB->WholeVolume)
1950         {
1951
1952             ntStatus = AFSInvalidateVolume( pVolumeCB,
1953                                             InvalidateCB->Reason);
1954
1955             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
1956
1957             try_return( ntStatus);
1958         }
1959
1960         AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
1961                           TRUE);
1962
1963         if ( AFSIsVolumeFID( &InvalidateCB->FileID))
1964         {
1965
1966             pObjectInfo = &pVolumeCB->ObjectInformation;
1967         }
1968         else
1969         {
1970
1971             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
1972
1973             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
1974                           AFS_TRACE_LEVEL_VERBOSE,
1975                           "AFSInvalidateCache Decrement count on volume %08lX Cnt %d\n",
1976                           pVolumeCB,
1977                           pVolumeCB->VolumeReferenceCount);
1978
1979             ullIndex = AFSCreateLowIndex( &InvalidateCB->FileID);
1980
1981             ntStatus = AFSLocateHashEntry( pVolumeCB->ObjectInfoTree.TreeHead,
1982                                            ullIndex,
1983                                            (AFSBTreeEntry **)&pObjectInfo);
1984         }
1985
1986         if( pObjectInfo != NULL)
1987         {
1988
1989             //
1990             // Reference the node so it won't be torn down
1991             //
1992
1993             lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
1994
1995             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
1996                           AFS_TRACE_LEVEL_VERBOSE,
1997                           "AFSInvalidateCache Increment count on object %08lX Cnt %d\n",
1998                           pObjectInfo,
1999                           lCount);
2000         }
2001
2002         AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
2003
2004         if( !NT_SUCCESS( ntStatus) ||
2005             pObjectInfo == NULL)
2006         {
2007
2008             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2009                           AFS_TRACE_LEVEL_WARNING,
2010                           "AFSInvalidateCache Invalidation FAILURE Unable to locate object FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
2011                           InvalidateCB->FileID.Cell,
2012                           InvalidateCB->FileID.Volume,
2013                           InvalidateCB->FileID.Vnode,
2014                           InvalidateCB->FileID.Unique,
2015                           ntStatus);
2016
2017             try_return( ntStatus = STATUS_SUCCESS);
2018         }
2019
2020         AFSInvalidateObject( &pObjectInfo,
2021                              InvalidateCB->Reason);
2022
2023 try_exit:
2024
2025         if( pObjectInfo != NULL)
2026         {
2027
2028             lCount = InterlockedDecrement( &pObjectInfo->ObjectReferenceCount);
2029
2030             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2031                           AFS_TRACE_LEVEL_VERBOSE,
2032                           "AFSInvalidateCache Decrement count on object %08lX Cnt %d\n",
2033                           pObjectInfo,
2034                           lCount);
2035         }
2036     }
2037
2038     return ntStatus;
2039 }
2040
2041 BOOLEAN
2042 AFSIsChildOfParent( IN AFSFcb *Dcb,
2043                     IN AFSFcb *Fcb)
2044 {
2045
2046     BOOLEAN bIsChild = FALSE;
2047     AFSFcb *pCurrentFcb = Fcb;
2048
2049     while( pCurrentFcb != NULL)
2050     {
2051
2052         if( pCurrentFcb->ObjectInformation->ParentObjectInformation == Dcb->ObjectInformation)
2053         {
2054
2055             bIsChild = TRUE;
2056
2057             break;
2058         }
2059
2060         pCurrentFcb = pCurrentFcb->ObjectInformation->ParentObjectInformation->Fcb;
2061     }
2062
2063     return bIsChild;
2064 }
2065
2066 inline
2067 ULONGLONG
2068 AFSCreateHighIndex( IN AFSFileID *FileID)
2069 {
2070
2071     ULONGLONG ullIndex = 0;
2072
2073     ullIndex = (((ULONGLONG)FileID->Cell << 32) | FileID->Volume);
2074
2075     return ullIndex;
2076 }
2077
2078 inline
2079 ULONGLONG
2080 AFSCreateLowIndex( IN AFSFileID *FileID)
2081 {
2082
2083     ULONGLONG ullIndex = 0;
2084
2085     ullIndex = (((ULONGLONG)FileID->Vnode << 32) | FileID->Unique);
2086
2087     return ullIndex;
2088 }
2089
2090 BOOLEAN
2091 AFSCheckAccess( IN ACCESS_MASK DesiredAccess,
2092                 IN ACCESS_MASK GrantedAccess,
2093                 IN BOOLEAN DirectoryEntry)
2094 {
2095
2096     BOOLEAN bAccessGranted = TRUE;
2097
2098     //
2099     // Check if we are asking for read/write and granted only read only
2100     // NOTE: There will be more checks here
2101     //
2102
2103     if( !AFSCheckForReadOnlyAccess( DesiredAccess,
2104                                     DirectoryEntry) &&
2105         AFSCheckForReadOnlyAccess( GrantedAccess,
2106                                    DirectoryEntry))
2107     {
2108
2109         bAccessGranted = FALSE;
2110     }
2111
2112     return bAccessGranted;
2113 }
2114
2115 NTSTATUS
2116 AFSGetDriverStatus( IN AFSDriverStatusRespCB *DriverStatus)
2117 {
2118
2119     NTSTATUS         ntStatus = STATUS_SUCCESS;
2120     AFSDeviceExt    *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
2121
2122     //
2123     // Start with read
2124     //
2125
2126     DriverStatus->Status = AFS_DRIVER_STATUS_READY;
2127
2128     if( AFSGlobalRoot == NULL)
2129     {
2130
2131         //
2132         // We are not ready
2133         //
2134
2135         DriverStatus->Status = AFS_DRIVER_STATUS_NOT_READY;
2136     }
2137
2138     if( pControlDevExt->Specific.Control.CommServiceCB.IrpPoolControlFlag != POOL_ACTIVE)
2139     {
2140
2141         //
2142         // No service yet
2143         //
2144
2145         DriverStatus->Status = AFS_DRIVER_STATUS_NO_SERVICE;
2146     }
2147
2148     return ntStatus;
2149 }
2150
2151 NTSTATUS
2152 AFSSubstituteSysName( IN UNICODE_STRING *ComponentName,
2153                       IN UNICODE_STRING *SubstituteName,
2154                       IN ULONG StringIndex)
2155 {
2156
2157     NTSTATUS ntStatus = STATUS_SUCCESS;
2158     AFSDeviceExt    *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
2159     AFSSysNameCB    *pSysName = NULL;
2160     ERESOURCE       *pSysNameLock = NULL;
2161     ULONG            ulIndex = 1;
2162     USHORT           usIndex = 0;
2163     UNICODE_STRING   uniSysName;
2164
2165     __Enter
2166     {
2167
2168 #if defined(_WIN64)
2169
2170         if( IoIs32bitProcess( NULL))
2171         {
2172
2173             pSysNameLock = &pControlDevExt->Specific.Control.SysName32ListLock;
2174
2175             pSysName = pControlDevExt->Specific.Control.SysName32ListHead;
2176         }
2177         else
2178         {
2179
2180             pSysNameLock = &pControlDevExt->Specific.Control.SysName64ListLock;
2181
2182             pSysName = pControlDevExt->Specific.Control.SysName64ListHead;
2183         }
2184 #else
2185
2186         pSysNameLock = &pControlDevExt->Specific.Control.SysName32ListLock;
2187
2188         pSysName = pControlDevExt->Specific.Control.SysName32ListHead;
2189
2190 #endif
2191
2192         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2193                       AFS_TRACE_LEVEL_VERBOSE,
2194                       "AFSSubstituteSysName Acquiring SysName lock %08lX SHARED %08lX\n",
2195                       pSysNameLock,
2196                       PsGetCurrentThread());
2197
2198         AFSAcquireShared( pSysNameLock,
2199                           TRUE);
2200
2201         //
2202         // Find where we are in the list
2203         //
2204
2205         while( pSysName != NULL &&
2206             ulIndex < StringIndex)
2207         {
2208
2209             pSysName = pSysName->fLink;
2210
2211             ulIndex++;
2212         }
2213
2214         if( pSysName == NULL)
2215         {
2216
2217             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
2218         }
2219
2220         RtlInitUnicodeString( &uniSysName,
2221                               L"@SYS");
2222         //
2223         // If it is a full component of @SYS then just substitue the
2224         // name in
2225         //
2226
2227         if( RtlCompareUnicodeString( &uniSysName,
2228                                      ComponentName,
2229                                      TRUE) == 0)
2230         {
2231
2232             SubstituteName->Length = pSysName->SysName.Length;
2233             SubstituteName->MaximumLength = SubstituteName->Length;
2234
2235             SubstituteName->Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2236                                                                         SubstituteName->Length,
2237                                                                         AFS_SUBST_BUFFER_TAG);
2238
2239             if( SubstituteName->Buffer == NULL)
2240             {
2241
2242                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2243             }
2244
2245             RtlCopyMemory( SubstituteName->Buffer,
2246                            pSysName->SysName.Buffer,
2247                            pSysName->SysName.Length);
2248         }
2249         else
2250         {
2251
2252             usIndex = 0;
2253
2254             while( ComponentName->Buffer[ usIndex] != L'@')
2255             {
2256
2257                 usIndex++;
2258             }
2259
2260             SubstituteName->Length = (usIndex * sizeof( WCHAR)) + pSysName->SysName.Length;
2261             SubstituteName->MaximumLength = SubstituteName->Length;
2262
2263             SubstituteName->Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2264                                                                         SubstituteName->Length,
2265                                                                         AFS_SUBST_BUFFER_TAG);
2266
2267             if( SubstituteName->Buffer == NULL)
2268             {
2269
2270                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2271             }
2272
2273             RtlCopyMemory( SubstituteName->Buffer,
2274                            ComponentName->Buffer,
2275                            usIndex * sizeof( WCHAR));
2276
2277             RtlCopyMemory( &SubstituteName->Buffer[ usIndex],
2278                            pSysName->SysName.Buffer,
2279                            pSysName->SysName.Length);
2280         }
2281
2282 try_exit:
2283
2284         AFSReleaseResource( pSysNameLock);
2285     }
2286
2287     return ntStatus;
2288 }
2289
2290 NTSTATUS
2291 AFSSubstituteNameInPath( IN OUT UNICODE_STRING *FullPathName,
2292                          IN OUT UNICODE_STRING *ComponentName,
2293                          IN UNICODE_STRING *SubstituteName,
2294                          IN OUT UNICODE_STRING *RemainingPath,
2295                          IN BOOLEAN FreePathName)
2296 {
2297
2298     NTSTATUS ntStatus = STATUS_SUCCESS;
2299     UNICODE_STRING uniPathName;
2300     USHORT usPrefixNameLen = 0;
2301     SHORT  sNameLenDelta = 0;
2302
2303     __Enter
2304     {
2305
2306         //
2307         // If the passed in name can handle the additional length
2308         // then just moves things around
2309         //
2310
2311         sNameLenDelta = SubstituteName->Length - ComponentName->Length;
2312
2313         usPrefixNameLen = (USHORT)(ComponentName->Buffer - FullPathName->Buffer);
2314
2315         if( FullPathName->MaximumLength > FullPathName->Length + sNameLenDelta)
2316         {
2317
2318             if( FullPathName->Length > usPrefixNameLen + ComponentName->Length)
2319             {
2320
2321                 RtlMoveMemory( &FullPathName->Buffer[ ((usPrefixNameLen*sizeof( WCHAR) + SubstituteName->Length)/sizeof( WCHAR))],
2322                                &FullPathName->Buffer[ ((usPrefixNameLen*sizeof( WCHAR) + ComponentName->Length)/sizeof( WCHAR))],
2323                                FullPathName->Length - usPrefixNameLen*sizeof( WCHAR) - ComponentName->Length);
2324             }
2325
2326             RtlCopyMemory( &FullPathName->Buffer[ usPrefixNameLen],
2327                            SubstituteName->Buffer,
2328                            SubstituteName->Length);
2329
2330             FullPathName->Length += sNameLenDelta;
2331
2332             ComponentName->Length += sNameLenDelta;
2333
2334             ComponentName->MaximumLength = ComponentName->Length;
2335
2336             if ( RemainingPath->Buffer)
2337             {
2338
2339                 RemainingPath->Buffer += sNameLenDelta/sizeof( WCHAR);
2340             }
2341
2342             try_return( ntStatus);
2343         }
2344
2345         //
2346         // Need to re-allocate the buffer
2347         //
2348
2349         uniPathName.Length = FullPathName->Length -
2350                                          ComponentName->Length +
2351                                          SubstituteName->Length;
2352
2353         uniPathName.MaximumLength = FullPathName->MaximumLength + PAGE_SIZE;
2354
2355         uniPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2356                                                                 uniPathName.MaximumLength,
2357                                                                 AFS_NAME_BUFFER_FOUR_TAG);
2358
2359         if( uniPathName.Buffer == NULL)
2360         {
2361
2362             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2363         }
2364
2365         usPrefixNameLen = (USHORT)(ComponentName->Buffer - FullPathName->Buffer);
2366
2367         usPrefixNameLen *= sizeof( WCHAR);
2368
2369         RtlZeroMemory( uniPathName.Buffer,
2370                        uniPathName.MaximumLength);
2371
2372         RtlCopyMemory( uniPathName.Buffer,
2373                        FullPathName->Buffer,
2374                        usPrefixNameLen);
2375
2376         RtlCopyMemory( &uniPathName.Buffer[ (usPrefixNameLen/sizeof( WCHAR))],
2377                        SubstituteName->Buffer,
2378                        SubstituteName->Length);
2379
2380         if( FullPathName->Length > usPrefixNameLen + ComponentName->Length)
2381         {
2382
2383             RtlCopyMemory( &uniPathName.Buffer[ (usPrefixNameLen + SubstituteName->Length)/sizeof( WCHAR)],
2384                            &FullPathName->Buffer[ (usPrefixNameLen + ComponentName->Length)/sizeof( WCHAR)],
2385                            FullPathName->Length - usPrefixNameLen - ComponentName->Length);
2386         }
2387
2388         ComponentName->Buffer = uniPathName.Buffer + (ComponentName->Buffer - FullPathName->Buffer);
2389
2390         ComponentName->Length += sNameLenDelta;
2391
2392         ComponentName->MaximumLength = ComponentName->Length;
2393
2394         if ( RemainingPath->Buffer)
2395         {
2396
2397             RemainingPath->Buffer = uniPathName.Buffer
2398                 + (RemainingPath->Buffer - FullPathName->Buffer)
2399                 + sNameLenDelta/sizeof( WCHAR);
2400         }
2401
2402         if( FreePathName)
2403         {
2404             AFSExFreePool( FullPathName->Buffer);
2405         }
2406
2407         *FullPathName = uniPathName;
2408
2409 try_exit:
2410
2411         NOTHING;
2412     }
2413
2414     return ntStatus;
2415 }
2416
2417 NTSTATUS
2418 AFSInvalidateVolume( IN AFSVolumeCB *VolumeCB,
2419                      IN ULONG Reason)
2420 {
2421
2422     NTSTATUS ntStatus = STATUS_SUCCESS;
2423     AFSObjectInfoCB *pCurrentObject = NULL;
2424     AFSObjectInfoCB *pNextObject = NULL;
2425     LONG lCount;
2426     AFSFcb *pFcb = NULL;
2427     ULONG ulFilter = 0;
2428
2429     __Enter
2430     {
2431
2432         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2433                       AFS_TRACE_LEVEL_VERBOSE,
2434                       "AFSInvalidateVolume Invalidate volume fid %08lX-%08lX-%08lX-%08lX Reason %08lX\n",
2435                       VolumeCB->ObjectInformation.FileId.Cell,
2436                       VolumeCB->ObjectInformation.FileId.Volume,
2437                       VolumeCB->ObjectInformation.FileId.Vnode,
2438                       VolumeCB->ObjectInformation.FileId.Unique,
2439                       Reason);
2440
2441         //
2442         // Depending on the reason for invalidation then perform work on the node
2443         //
2444
2445         switch( Reason)
2446         {
2447
2448             case AFS_INVALIDATE_DELETED:
2449             {
2450
2451                 //
2452                 // Mark this volume as invalid
2453                 //
2454
2455                 SetFlag( VolumeCB->ObjectInformation.Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
2456
2457                 SetFlag( VolumeCB->Flags, AFS_VOLUME_FLAGS_OFFLINE);
2458
2459                 break;
2460             }
2461         }
2462
2463         //
2464         // Invalidate the volume root directory
2465         //
2466
2467         pCurrentObject = &VolumeCB->ObjectInformation;
2468
2469         if ( pCurrentObject )
2470         {
2471
2472             lCount = InterlockedIncrement( &pCurrentObject->ObjectReferenceCount);
2473
2474             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2475                           AFS_TRACE_LEVEL_VERBOSE,
2476                           "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2477                           pCurrentObject,
2478                           lCount);
2479
2480             AFSInvalidateObject( &pCurrentObject,
2481                                  Reason);
2482
2483             if ( pCurrentObject)
2484             {
2485
2486                 lCount = InterlockedDecrement( &pCurrentObject->ObjectReferenceCount);
2487
2488                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2489                               AFS_TRACE_LEVEL_VERBOSE,
2490                               "AFSInvalidateVolumeObjects Decrement count on object %08lX Cnt %d\n",
2491                               pCurrentObject,
2492                               lCount);
2493             }
2494         }
2495
2496         //
2497         // Apply invalidation to all other volume objects
2498         //
2499
2500         AFSAcquireShared( VolumeCB->ObjectInfoTree.TreeLock,
2501                           TRUE);
2502
2503         pCurrentObject = VolumeCB->ObjectInfoListHead;
2504
2505         if ( pCurrentObject)
2506         {
2507
2508             //
2509             // Reference the node so it won't be torn down
2510             //
2511
2512             lCount = InterlockedIncrement( &pCurrentObject->ObjectReferenceCount);
2513
2514             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2515                           AFS_TRACE_LEVEL_VERBOSE,
2516                           "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2517                           pCurrentObject,
2518                           lCount);
2519         }
2520
2521         while( pCurrentObject != NULL)
2522         {
2523
2524             pNextObject = (AFSObjectInfoCB *)pCurrentObject->ListEntry.fLink;
2525
2526             if ( pNextObject)
2527             {
2528
2529                 //
2530                 // Reference the node so it won't be torn down
2531                 //
2532
2533                 lCount = InterlockedIncrement( &pNextObject->ObjectReferenceCount);
2534
2535                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2536                               AFS_TRACE_LEVEL_VERBOSE,
2537                               "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2538                               pNextObject,
2539                               lCount);
2540             }
2541
2542             AFSReleaseResource( VolumeCB->ObjectInfoTree.TreeLock);
2543
2544             AFSInvalidateObject( &pCurrentObject,
2545                                  Reason);
2546
2547             if ( pCurrentObject )
2548             {
2549
2550                 lCount = InterlockedDecrement( &pCurrentObject->ObjectReferenceCount);
2551
2552                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2553                               AFS_TRACE_LEVEL_VERBOSE,
2554                               "AFSInvalidateVolumeObjects Decrement count on object %08lX Cnt %d\n",
2555                               pCurrentObject,
2556                               lCount);
2557             }
2558
2559             AFSAcquireShared( VolumeCB->ObjectInfoTree.TreeLock,
2560                               TRUE);
2561
2562             pCurrentObject = pNextObject;
2563         }
2564
2565         AFSReleaseResource( VolumeCB->ObjectInfoTree.TreeLock);
2566     }
2567
2568     return ntStatus;
2569 }
2570
2571 VOID
2572 AFSInvalidateAllVolumes( VOID)
2573 {
2574     AFSVolumeCB *pVolumeCB = NULL;
2575     AFSVolumeCB *pNextVolumeCB = NULL;
2576     AFSDeviceExt *pRDRDeviceExt = NULL;
2577     LONG lCount;
2578
2579     pRDRDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
2580
2581     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2582                   AFS_TRACE_LEVEL_VERBOSE,
2583                   "AFSInvalidateAllVolumes Acquiring RDR VolumeListLock lock %08lX SHARED %08lX\n",
2584                   &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2585                   PsGetCurrentThread());
2586
2587     AFSAcquireShared( &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2588                       TRUE);
2589
2590     pVolumeCB = pRDRDeviceExt->Specific.RDR.VolumeListHead;
2591
2592     if ( pVolumeCB)
2593     {
2594
2595         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2596                       AFS_TRACE_LEVEL_VERBOSE,
2597                       "AFSInvalidateAllVolumes Acquiring VolumeRoot ObjectInfoTree lock %08lX SHARED %08lX\n",
2598                       pVolumeCB->ObjectInfoTree.TreeLock,
2599                       PsGetCurrentThread());
2600
2601         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
2602     }
2603
2604     while( pVolumeCB != NULL)
2605     {
2606
2607         pNextVolumeCB = (AFSVolumeCB *)pVolumeCB->ListEntry.fLink;
2608
2609         if ( pNextVolumeCB)
2610         {
2611
2612             lCount = InterlockedIncrement( &pNextVolumeCB->VolumeReferenceCount);
2613         }
2614
2615         AFSReleaseResource( &pRDRDeviceExt->Specific.RDR.VolumeListLock);
2616
2617         // do I need to hold the volume lock here?
2618
2619         AFSInvalidateVolume( pVolumeCB, AFS_INVALIDATE_EXPIRED);
2620
2621         AFSAcquireShared( &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2622                           TRUE);
2623
2624         lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
2625
2626         pVolumeCB = pNextVolumeCB;
2627     }
2628
2629     AFSReleaseResource( &pRDRDeviceExt->Specific.RDR.VolumeListLock);
2630 }
2631
2632 NTSTATUS
2633 AFSVerifyEntry( IN GUID *AuthGroup,
2634                 IN AFSDirectoryCB *DirEntry)
2635 {
2636
2637     NTSTATUS ntStatus = STATUS_SUCCESS;
2638     AFSDirEnumEntry *pDirEnumEntry = NULL;
2639     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
2640     IO_STATUS_BLOCK stIoStatus;
2641
2642     __Enter
2643     {
2644
2645         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2646                       AFS_TRACE_LEVEL_VERBOSE_2,
2647                       "AFSVerifyEntry Verifying entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2648                       &DirEntry->NameInformation.FileName,
2649                       pObjectInfo->FileId.Cell,
2650                       pObjectInfo->FileId.Volume,
2651                       pObjectInfo->FileId.Vnode,
2652                       pObjectInfo->FileId.Unique);
2653
2654         ntStatus = AFSEvaluateTargetByID( pObjectInfo,
2655                                           AuthGroup,
2656                                           FALSE,
2657                                           &pDirEnumEntry);
2658
2659         if( !NT_SUCCESS( ntStatus))
2660         {
2661
2662             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2663                           AFS_TRACE_LEVEL_ERROR,
2664                           "AFSVerifyEntry Evaluate Target failed %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
2665                           &DirEntry->NameInformation.FileName,
2666                           pObjectInfo->FileId.Cell,
2667                           pObjectInfo->FileId.Volume,
2668                           pObjectInfo->FileId.Vnode,
2669                           pObjectInfo->FileId.Unique,
2670                           ntStatus);
2671
2672             try_return( ntStatus);
2673         }
2674
2675         //
2676         // Check the data version of the file
2677         //
2678
2679         if( pObjectInfo->DataVersion.QuadPart == pDirEnumEntry->DataVersion.QuadPart)
2680         {
2681             if ( !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA))
2682             {
2683
2684                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2685                               AFS_TRACE_LEVEL_VERBOSE,
2686                               "AFSVerifyEntry No DV change %I64X for Fcb %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2687                               pObjectInfo->DataVersion.QuadPart,
2688                               &DirEntry->NameInformation.FileName,
2689                               pObjectInfo->FileId.Cell,
2690                               pObjectInfo->FileId.Volume,
2691                               pObjectInfo->FileId.Vnode,
2692                               pObjectInfo->FileId.Unique);
2693
2694                 //
2695                 // We are ok, just get out
2696                 //
2697
2698                 ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2699
2700                 try_return( ntStatus = STATUS_SUCCESS);
2701             }
2702         }
2703
2704         //
2705         // New data version so we will need to process the node based on the type
2706         //
2707
2708         switch( pDirEnumEntry->FileType)
2709         {
2710
2711             case AFS_FILE_TYPE_MOUNTPOINT:
2712             {
2713
2714                 //
2715                 // For a mount point we need to ensure the target is the same
2716                 //
2717
2718                 if( !AFSIsEqualFID( &pObjectInfo->TargetFileId,
2719                                     &pDirEnumEntry->TargetFileId))
2720                 {
2721
2722                 }
2723
2724                 //
2725                 // Update the metadata for the entry
2726                 //
2727
2728                 ntStatus = AFSUpdateMetaData( DirEntry,
2729                                               pDirEnumEntry);
2730
2731                 if( NT_SUCCESS( ntStatus))
2732                 {
2733
2734                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2735                 }
2736
2737                 break;
2738             }
2739
2740             case AFS_FILE_TYPE_SYMLINK:
2741             {
2742
2743                 ASSERT( pDirEnumEntry->TargetNameLength > 0);
2744
2745                 //
2746                 // Update the metadata for the entry
2747                 //
2748
2749                 ntStatus = AFSUpdateMetaData( DirEntry,
2750                                               pDirEnumEntry);
2751
2752                 if( NT_SUCCESS( ntStatus))
2753                 {
2754
2755                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2756                 }
2757
2758                 break;
2759             }
2760
2761             case AFS_FILE_TYPE_FILE:
2762             {
2763                 FILE_OBJECT * pCCFileObject = NULL;
2764                 BOOLEAN bPurgeExtents = FALSE;
2765
2766                 if ( pObjectInfo->DataVersion.QuadPart != pDirEnumEntry->DataVersion.QuadPart)
2767                 {
2768
2769                     bPurgeExtents = TRUE;
2770                 }
2771
2772                 if ( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA))
2773                 {
2774
2775                     bPurgeExtents = TRUE;
2776
2777                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2778                                   AFS_TRACE_LEVEL_VERBOSE,
2779                                   "AFSVerifyEntry Clearing VERIFY_DATA flag %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2780                                   &DirEntry->NameInformation.FileName,
2781                                   pObjectInfo->FileId.Cell,
2782                                   pObjectInfo->FileId.Volume,
2783                                   pObjectInfo->FileId.Vnode,
2784                                   pObjectInfo->FileId.Unique);
2785
2786                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
2787                 }
2788
2789                 if( pObjectInfo->Fcb != NULL)
2790                 {
2791
2792                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2793                                   AFS_TRACE_LEVEL_VERBOSE,
2794                                   "AFSVerifyEntry Flush/purge entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2795                                   &DirEntry->NameInformation.FileName,
2796                                   pObjectInfo->FileId.Cell,
2797                                   pObjectInfo->FileId.Volume,
2798                                   pObjectInfo->FileId.Vnode,
2799                                   pObjectInfo->FileId.Unique);
2800
2801                     AFSAcquireExcl( &pObjectInfo->Fcb->NPFcb->Resource,
2802                                     TRUE);
2803
2804                     __try
2805                     {
2806
2807                         CcFlushCache( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers,
2808                                       NULL,
2809                                       0,
2810                                       &stIoStatus);
2811
2812                         if( !NT_SUCCESS( stIoStatus.Status))
2813                         {
2814
2815                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
2816                                           AFS_TRACE_LEVEL_ERROR,
2817                                           "AFSVerifyEntry CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
2818                                           &DirEntry->NameInformation.FileName,
2819                                           pObjectInfo->FileId.Cell,
2820                                           pObjectInfo->FileId.Volume,
2821                                           pObjectInfo->FileId.Vnode,
2822                                           pObjectInfo->FileId.Unique,
2823                                           stIoStatus.Status,
2824                                           stIoStatus.Information);
2825
2826                             ntStatus = stIoStatus.Status;
2827                         }
2828
2829                         if ( bPurgeExtents)
2830                         {
2831
2832                             CcPurgeCacheSection( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers,
2833                                                  NULL,
2834                                                  0,
2835                                                  FALSE);
2836                         }
2837                     }
2838                     __except( EXCEPTION_EXECUTE_HANDLER)
2839                     {
2840                         ntStatus = GetExceptionCode();
2841
2842                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
2843                                       AFS_TRACE_LEVEL_ERROR,
2844                                       "AFSVerifyEntry CcFlushCache or CcPurgeCacheSection Exception %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
2845                                       &DirEntry->NameInformation.FileName,
2846                                       pObjectInfo->FileId.Cell,
2847                                       pObjectInfo->FileId.Volume,
2848                                       pObjectInfo->FileId.Vnode,
2849                                       pObjectInfo->FileId.Unique,
2850                                       ntStatus);
2851                     }
2852
2853                     AFSReleaseResource( &pObjectInfo->Fcb->NPFcb->Resource);
2854
2855                     if ( bPurgeExtents)
2856                     {
2857                         AFSFlushExtents( pObjectInfo->Fcb,
2858                                          AuthGroup);
2859                     }
2860
2861                     //
2862                     // Reacquire the Fcb to purge the cache
2863                     //
2864
2865                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2866                                   AFS_TRACE_LEVEL_VERBOSE,
2867                                   "AFSVerifyEntry Acquiring Fcb lock %08lX EXCL %08lX\n",
2868                                   &pObjectInfo->Fcb->NPFcb->Resource,
2869                                   PsGetCurrentThread());
2870
2871                     AFSAcquireExcl( &pObjectInfo->Fcb->NPFcb->Resource,
2872                                     TRUE);
2873
2874                     //
2875                     // Update the metadata for the entry
2876                     //
2877
2878                     ntStatus = AFSUpdateMetaData( DirEntry,
2879                                                   pDirEnumEntry);
2880
2881                     if( !NT_SUCCESS( ntStatus))
2882                     {
2883
2884                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2885                                       AFS_TRACE_LEVEL_ERROR,
2886                                       "AFSVerifyEntry Meta Data Update failed %wZ FID %08lX-%08lX-%08lX-%08lX ntStatus %08lX\n",
2887                                       &DirEntry->NameInformation.FileName,
2888                                       pObjectInfo->FileId.Cell,
2889                                       pObjectInfo->FileId.Volume,
2890                                       pObjectInfo->FileId.Vnode,
2891                                       pObjectInfo->FileId.Unique,
2892                                       ntStatus);
2893
2894                         break;
2895                     }
2896
2897                     //
2898                     // Update file sizes
2899                     //
2900
2901                     pObjectInfo->Fcb->Header.AllocationSize.QuadPart  = pObjectInfo->AllocationSize.QuadPart;
2902                     pObjectInfo->Fcb->Header.FileSize.QuadPart        = pObjectInfo->EndOfFile.QuadPart;
2903                     pObjectInfo->Fcb->Header.ValidDataLength.QuadPart = pObjectInfo->EndOfFile.QuadPart;
2904
2905                     pCCFileObject = CcGetFileObjectFromSectionPtrs( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers);
2906
2907                     if ( pCCFileObject != NULL)
2908                     {
2909                         CcSetFileSizes( pCCFileObject,
2910                                         (PCC_FILE_SIZES)&pObjectInfo->Fcb->Header.AllocationSize);
2911                     }
2912
2913                     AFSReleaseResource( &pObjectInfo->Fcb->NPFcb->Resource);
2914                 }
2915                 else
2916                 {
2917
2918                     //
2919                     // Update the metadata for the entry
2920                     //
2921
2922                     ntStatus = AFSUpdateMetaData( DirEntry,
2923                                                   pDirEnumEntry);
2924
2925                     if( !NT_SUCCESS( ntStatus))
2926                     {
2927
2928                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2929                                       AFS_TRACE_LEVEL_ERROR,
2930                                       "AFSVerifyEntry Meta Data Update failed %wZ FID %08lX-%08lX-%08lX-%08lX ntStatus %08lX\n",
2931                                       &DirEntry->NameInformation.FileName,
2932                                       pObjectInfo->FileId.Cell,
2933                                       pObjectInfo->FileId.Volume,
2934                                       pObjectInfo->FileId.Vnode,
2935                                       pObjectInfo->FileId.Unique,
2936                                       ntStatus);
2937
2938                         break;
2939                     }
2940
2941                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2942                                   AFS_TRACE_LEVEL_WARNING,
2943                                   "AFSVerifyEntry Fcb NULL %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2944                                   &DirEntry->NameInformation.FileName,
2945                                   pObjectInfo->FileId.Cell,
2946                                   pObjectInfo->FileId.Volume,
2947                                   pObjectInfo->FileId.Vnode,
2948                                   pObjectInfo->FileId.Unique);
2949                 }
2950
2951                 ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2952
2953                 break;
2954             }
2955
2956             case AFS_FILE_TYPE_DIRECTORY:
2957             {
2958
2959                 AFSFcb *pCurrentFcb = NULL;
2960                 AFSDirectoryCB *pCurrentDirEntry = NULL;
2961
2962                 //
2963                 // For a directory or root entry flush the content of
2964                 // the directory enumeration.
2965                 //
2966
2967                 if( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED))
2968                 {
2969
2970                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2971                                   AFS_TRACE_LEVEL_VERBOSE_2,
2972                                   "AFSVerifyEntry Validating directory content for entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2973                                   &DirEntry->NameInformation.FileName,
2974                                   pObjectInfo->FileId.Cell,
2975                                   pObjectInfo->FileId.Volume,
2976                                   pObjectInfo->FileId.Vnode,
2977                                   pObjectInfo->FileId.Unique);
2978
2979                     AFSAcquireExcl( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
2980                                     TRUE);
2981
2982                     ntStatus = AFSValidateDirectoryCache( pObjectInfo,
2983                                                           AuthGroup);
2984
2985                     AFSReleaseResource( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock);
2986
2987                     if ( !NT_SUCCESS( ntStatus))
2988                     {
2989
2990                         try_return( ntStatus);
2991                     }
2992                 }
2993
2994                 //
2995                 // Update the metadata for the entry
2996                 //
2997
2998                 ntStatus = AFSUpdateMetaData( DirEntry,
2999                                               pDirEnumEntry);
3000
3001                 if( NT_SUCCESS( ntStatus))
3002                 {
3003
3004                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
3005                 }
3006
3007                 break;
3008             }
3009
3010             case AFS_FILE_TYPE_DFSLINK:
3011             {
3012
3013                 UNICODE_STRING uniTargetName;
3014
3015                 //
3016                 // For a DFS link need to check the target name has not changed
3017                 //
3018
3019                 uniTargetName.Length = (USHORT)pDirEnumEntry->TargetNameLength;
3020
3021                 uniTargetName.MaximumLength = uniTargetName.Length;
3022
3023                 uniTargetName.Buffer = (WCHAR *)((char *)pDirEnumEntry + pDirEnumEntry->TargetNameOffset);
3024
3025                 AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3026                                 TRUE);
3027
3028                 if( DirEntry->NameInformation.TargetName.Length == 0 ||
3029                     RtlCompareUnicodeString( &uniTargetName,
3030                                              &DirEntry->NameInformation.TargetName,
3031                                              TRUE) != 0)
3032                 {
3033
3034                     //
3035                     // Update the target name
3036                     //
3037
3038                     ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
3039                                                     &DirEntry->Flags,
3040                                                     uniTargetName.Buffer,
3041                                                     uniTargetName.Length);
3042
3043                     if( !NT_SUCCESS( ntStatus))
3044                     {
3045
3046                         AFSReleaseResource( &DirEntry->NonPaged->Lock);
3047
3048                         break;
3049                     }
3050                 }
3051
3052                 AFSReleaseResource( &DirEntry->NonPaged->Lock);
3053
3054                 //
3055                 // Update the metadata for the entry
3056                 //
3057
3058                 ntStatus = AFSUpdateMetaData( DirEntry,
3059                                               pDirEnumEntry);
3060
3061                 if( NT_SUCCESS( ntStatus))
3062                 {
3063
3064                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
3065                 }
3066
3067                 break;
3068             }
3069
3070             default:
3071
3072                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3073                               AFS_TRACE_LEVEL_WARNING,
3074                               "AFSVerifyEntry Attempt to verify node of type %d %wZ FID %08lX-%08lX-%08lX-%08lX\n",
3075                               pObjectInfo->FileType,
3076                               &DirEntry->NameInformation.FileName,
3077                               pObjectInfo->FileId.Cell,
3078                               pObjectInfo->FileId.Volume,
3079                               pObjectInfo->FileId.Vnode,
3080                               pObjectInfo->FileId.Unique);
3081
3082                 break;
3083         }
3084
3085  try_exit:
3086
3087         if( pDirEnumEntry != NULL)
3088         {
3089
3090             AFSExFreePool( pDirEnumEntry);
3091         }
3092     }
3093
3094     return ntStatus;
3095 }
3096
3097 NTSTATUS
3098 AFSSetVolumeState( IN AFSVolumeStatusCB *VolumeStatus)
3099 {
3100
3101     NTSTATUS ntStatus = STATUS_SUCCESS;
3102     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
3103     ULONGLONG   ullIndex = 0;
3104     AFSVolumeCB *pVolumeCB = NULL;
3105     AFSFcb *pFcb = NULL;
3106     AFSObjectInfoCB *pCurrentObject = NULL;
3107     LONG lCount;
3108
3109     __Enter
3110     {
3111
3112         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3113                       AFS_TRACE_LEVEL_VERBOSE,
3114                       "AFSSetVolumeState Marking volume state %d Volume Cell %08lX Volume %08lX\n",
3115                       VolumeStatus->Online,
3116                       VolumeStatus->FileID.Cell,
3117                       VolumeStatus->FileID.Volume);
3118
3119         //
3120         // Need to locate the Fcb for the directory to purge
3121         //
3122
3123         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3124                       AFS_TRACE_LEVEL_VERBOSE,
3125                       "AFSSetVolumeState Acquiring RDR VolumeTreeLock lock %08lX SHARED %08lX\n",
3126                       &pDevExt->Specific.RDR.VolumeTreeLock,
3127                       PsGetCurrentThread());
3128
3129         AFSAcquireShared( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
3130
3131         //
3132         // Locate the volume node
3133         //
3134
3135         ullIndex = AFSCreateHighIndex( &VolumeStatus->FileID);
3136
3137         ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
3138                                        ullIndex,
3139                                        (AFSBTreeEntry **)&pVolumeCB);
3140
3141         if( pVolumeCB != NULL)
3142         {
3143
3144             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
3145
3146             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
3147
3148             //
3149             // Set the volume state accordingly
3150             //
3151
3152             if( VolumeStatus->Online)
3153             {
3154
3155                 InterlockedAnd( (LONG *)&(pVolumeCB->Flags), ~AFS_VOLUME_FLAGS_OFFLINE);
3156             }
3157             else
3158             {
3159
3160                 InterlockedOr( (LONG *)&(pVolumeCB->Flags), AFS_VOLUME_FLAGS_OFFLINE);
3161             }
3162
3163             AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
3164                               TRUE);
3165
3166             pCurrentObject = pVolumeCB->ObjectInfoListHead;;
3167
3168             while( pCurrentObject != NULL)
3169             {
3170
3171                 if( VolumeStatus->Online)
3172                 {
3173
3174                     ClearFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
3175
3176                     SetFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_VERIFY);
3177
3178                     pCurrentObject->DataVersion.QuadPart = (ULONGLONG)-1;
3179                 }
3180                 else
3181                 {
3182
3183                     SetFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
3184                 }
3185
3186                 pFcb = pCurrentObject->Fcb;
3187
3188                 if( pFcb != NULL &&
3189                     !(VolumeStatus->Online) &&
3190                     pFcb->Header.NodeTypeCode == AFS_FILE_FCB)
3191                 {
3192
3193                     AFSDbgLogMsg( AFS_SUBSYSTEM_EXTENT_PROCESSING,
3194                                   AFS_TRACE_LEVEL_ERROR,
3195                                   "AFSSetVolumeState Marking volume offline and canceling extents Volume Cell %08lX Volume %08lX\n",
3196                                   VolumeStatus->FileID.Cell,
3197                                   VolumeStatus->FileID.Volume);
3198
3199                     //
3200                     // Clear out the extents
3201                     //
3202
3203                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3204                                   AFS_TRACE_LEVEL_VERBOSE,
3205                                   "AFSSetVolumeState Acquiring Fcb extents lock %08lX EXCL %08lX\n",
3206                                   &pFcb->NPFcb->Specific.File.ExtentsResource,
3207                                   PsGetCurrentThread());
3208
3209                     AFSAcquireExcl( &pFcb->NPFcb->Specific.File.ExtentsResource,
3210                                     TRUE);
3211
3212                     pFcb->NPFcb->Specific.File.ExtentsRequestStatus = STATUS_CANCELLED;
3213
3214                     KeSetEvent( &pFcb->NPFcb->Specific.File.ExtentsRequestComplete,
3215                                 0,
3216                                 FALSE);
3217
3218                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3219                                   AFS_TRACE_LEVEL_VERBOSE,
3220                                   "AFSSetVolumeState Releasing Fcb extents lock %08lX EXCL %08lX\n",
3221                                   &pFcb->NPFcb->Specific.File.ExtentsResource,
3222                                   PsGetCurrentThread());
3223
3224                     AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource);
3225
3226                     //
3227                     // And get rid of them (note this involves waiting
3228                     // for any writes or reads to the cache to complete)
3229                     //
3230
3231                     (VOID) AFSTearDownFcbExtents( pFcb,
3232                                                   NULL);
3233                 }
3234
3235                 pCurrentObject = (AFSObjectInfoCB *)pCurrentObject->ListEntry.fLink;
3236             }
3237
3238             AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
3239
3240             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
3241         }
3242         else
3243         {
3244
3245             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
3246         }
3247     }
3248
3249     return ntStatus;
3250 }
3251
3252 NTSTATUS
3253 AFSSetNetworkState( IN AFSNetworkStatusCB *NetworkStatus)
3254 {
3255
3256     NTSTATUS ntStatus = STATUS_SUCCESS;
3257
3258     __Enter
3259     {
3260
3261         if( AFSGlobalRoot == NULL)
3262         {
3263
3264             try_return( ntStatus);
3265         }
3266
3267         AFSAcquireExcl( AFSGlobalRoot->VolumeLock,
3268                         TRUE);
3269
3270         //
3271         // Set the network state according to the information
3272         //
3273
3274         if( NetworkStatus->Online)
3275         {
3276
3277             ClearFlag( AFSGlobalRoot->Flags, AFS_VOLUME_FLAGS_OFFLINE);
3278         }
3279         else
3280         {
3281
3282             SetFlag( AFSGlobalRoot->Flags, AFS_VOLUME_FLAGS_OFFLINE);
3283         }
3284
3285         AFSReleaseResource( AFSGlobalRoot->VolumeLock);
3286
3287 try_exit:
3288
3289         NOTHING;
3290     }
3291
3292     return ntStatus;
3293 }
3294
3295 NTSTATUS
3296 AFSValidateDirectoryCache( IN AFSObjectInfoCB *ObjectInfo,
3297                            IN GUID *AuthGroup)
3298 {
3299
3300     NTSTATUS ntStatus = STATUS_SUCCESS;
3301     BOOLEAN  bAcquiredLock = FALSE;
3302     AFSDirectoryCB *pCurrentDirEntry = NULL, *pNextDirEntry = NULL;
3303     AFSFcb *pFcb = NULL;
3304
3305     __Enter
3306     {
3307
3308         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3309                       AFS_TRACE_LEVEL_VERBOSE,
3310                       "AFSValidateDirectoryCache Validating content for FID %08lX-%08lX-%08lX-%08lX\n",
3311                       ObjectInfo->FileId.Cell,
3312                       ObjectInfo->FileId.Volume,
3313                       ObjectInfo->FileId.Vnode,
3314                       ObjectInfo->FileId.Unique);
3315
3316         if( !ExIsResourceAcquiredLite( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock))
3317         {
3318
3319             AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3320                           AFS_TRACE_LEVEL_VERBOSE,
3321                           "AFSValidateDirectoryCache Acquiring DirectoryNodeHdr.TreeLock lock %08lX EXCL %08lX\n",
3322                           ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
3323                           PsGetCurrentThread());
3324
3325             AFSAcquireExcl( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
3326                             TRUE);
3327
3328             bAcquiredLock = TRUE;
3329         }
3330
3331         //
3332         // Check for inconsistency between DirectoryNodeList and DirectoryNodeCount
3333         //
3334
3335         if ( ObjectInfo->Specific.Directory.DirectoryNodeListHead == NULL &&
3336              ObjectInfo->Specific.Directory.DirectoryNodeCount > 0)
3337         {
3338
3339             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3340                           AFS_TRACE_LEVEL_ERROR,
3341                           "AFSValidateDirectoryCache Empty Node List but Non-Zero Node Count %08lX for dir FID %08lX-%08lX-%08lX-%08lX\n",
3342                           ObjectInfo->Specific.Directory.DirectoryNodeCount,
3343                           ObjectInfo->FileId.Cell,
3344                           ObjectInfo->FileId.Volume,
3345                           ObjectInfo->FileId.Vnode,
3346                           ObjectInfo->FileId.Unique);
3347         }
3348
3349         //
3350         // Reset the directory list information by clearing all valid entries
3351         //
3352
3353         pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
3354
3355         while( pCurrentDirEntry != NULL)
3356         {
3357
3358             pNextDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
3359
3360             if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_FAKE))
3361             {
3362
3363                 //
3364                 // If this entry has been deleted then process it here
3365                 //
3366
3367                 if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_DELETED) &&
3368                     pCurrentDirEntry->OpenReferenceCount == 0)
3369                 {
3370
3371                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3372                                   AFS_TRACE_LEVEL_VERBOSE,
3373                                   "AFSValidateDirectoryCache Deleting dir entry %p name %wZ\n",
3374                                   pCurrentDirEntry,
3375                                   &pCurrentDirEntry->NameInformation.FileName);
3376
3377                     AFSDeleteDirEntry( ObjectInfo,
3378                                        pCurrentDirEntry);
3379                 }
3380                 else
3381                 {
3382
3383                     ClearFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_VALID);
3384
3385                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3386                                   AFS_TRACE_LEVEL_VERBOSE,
3387                                   "AFSValidateDirectoryCache Clear VALID flag on DE %p Reference count %08lX\n",
3388                                   pCurrentDirEntry,
3389                                   pCurrentDirEntry->OpenReferenceCount);
3390
3391                     //
3392                     // We pull the short name from the parent tree since it could change below
3393                     //
3394
3395                     if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME))
3396                     {
3397
3398                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3399                                       AFS_TRACE_LEVEL_VERBOSE,
3400                                       "AFSValidateDirectoryCache Removing DE %p (%08lX) from shortname tree for %wZ\n",
3401                                       pCurrentDirEntry,
3402                                       pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex,
3403                                       &pCurrentDirEntry->NameInformation.FileName);
3404
3405                         AFSRemoveShortNameDirEntry( &ObjectInfo->Specific.Directory.ShortNameTree,
3406                                                     pCurrentDirEntry);
3407
3408                         ClearFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3409                     }
3410                 }
3411             }
3412
3413             pCurrentDirEntry = pNextDirEntry;
3414         }
3415
3416         //
3417         // Reget the directory contents
3418         //
3419
3420         ntStatus = AFSVerifyDirectoryContent( ObjectInfo,
3421                                               AuthGroup);
3422
3423         if ( !NT_SUCCESS( ntStatus))
3424         {
3425             try_return( ntStatus);
3426         }
3427
3428         //
3429         // Now start again and tear down any entries not valid
3430         //
3431
3432         pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
3433
3434         while( pCurrentDirEntry != NULL)
3435         {
3436
3437             pNextDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
3438
3439             if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_VALID))
3440             {
3441
3442                 if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME) &&
3443                     pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex > 0)
3444                 {
3445
3446                     if( ObjectInfo->Specific.Directory.ShortNameTree == NULL)
3447                     {
3448
3449                         ObjectInfo->Specific.Directory.ShortNameTree = pCurrentDirEntry;
3450
3451                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3452                                       AFS_TRACE_LEVEL_VERBOSE,
3453                                       "AFSValidateDirectoryCache Insert DE %p to head of shortname tree for %wZ\n",
3454                                       pCurrentDirEntry,
3455                                       &pCurrentDirEntry->NameInformation.FileName);
3456
3457                         SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3458                     }
3459                     else
3460                     {
3461
3462                         if( !NT_SUCCESS( AFSInsertShortNameDirEntry( ObjectInfo->Specific.Directory.ShortNameTree,
3463                                                                      pCurrentDirEntry)))
3464                         {
3465                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3466                                           AFS_TRACE_LEVEL_VERBOSE,
3467                                           "AFSValidateDirectoryCache Failed to insert DE %p (%08lX) to shortname tree for %wZ\n",
3468                                           pCurrentDirEntry,
3469                                           pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex,
3470                                           &pCurrentDirEntry->NameInformation.FileName);
3471                         }
3472                         else
3473                         {
3474                             SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3475
3476                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3477                                           AFS_TRACE_LEVEL_VERBOSE,
3478                                           "AFSValidateDirectoryCache Insert DE %p to shortname tree for %wZ\n",
3479                                           pCurrentDirEntry,
3480                                           &pCurrentDirEntry->NameInformation.FileName);
3481                         }
3482                     }
3483                 }
3484
3485                 pCurrentDirEntry = pNextDirEntry;
3486
3487                 continue;
3488             }
3489
3490             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3491                           AFS_TRACE_LEVEL_VERBOSE,
3492                           "AFSValidateDirectoryCache Processing INVALID DE %p Reference count %08lX\n",
3493                           pCurrentDirEntry,
3494                           pCurrentDirEntry->OpenReferenceCount);
3495
3496             if( pCurrentDirEntry->OpenReferenceCount == 0)
3497             {
3498
3499                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3500                               AFS_TRACE_LEVEL_VERBOSE,
3501                               "AFSValidateDirectoryCache Deleting dir entry %wZ from parent FID %08lX-%08lX-%08lX-%08lX\n",
3502                               &pCurrentDirEntry->NameInformation.FileName,
3503                               ObjectInfo->FileId.Cell,
3504                               ObjectInfo->FileId.Volume,
3505                               ObjectInfo->FileId.Vnode,
3506                               ObjectInfo->FileId.Unique);
3507
3508                 AFSDeleteDirEntry( ObjectInfo,
3509                                    pCurrentDirEntry);
3510             }
3511             else
3512             {
3513
3514                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3515                               AFS_TRACE_LEVEL_VERBOSE,
3516                               "AFSValidateDirectoryCache Setting dir entry %p Name %wZ DELETED in parent FID %08lX-%08lX-%08lX-%08lX\n",
3517                               pCurrentDirEntry,
3518                               &pCurrentDirEntry->NameInformation.FileName,
3519                               ObjectInfo->FileId.Cell,
3520                               ObjectInfo->FileId.Volume,
3521                               ObjectInfo->FileId.Vnode,
3522                               ObjectInfo->FileId.Unique);
3523
3524                 SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_DELETED);
3525
3526                 AFSRemoveNameEntry( ObjectInfo,
3527                                     pCurrentDirEntry);
3528             }
3529
3530             pCurrentDirEntry = pNextDirEntry;
3531         }
3532
3533 #if DBG
3534         if( !AFSValidateDirList( ObjectInfo))
3535         {
3536
3537             AFSPrint("AFSValidateDirectoryCache Invalid count ...\n");
3538         }
3539 #endif
3540
3541 try_exit:
3542
3543         if( bAcquiredLock)
3544         {
3545
3546             AFSReleaseResource( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock);
3547         }
3548     }
3549
3550     return ntStatus;
3551 }
3552
3553 BOOLEAN
3554 AFSIsVolumeFID( IN AFSFileID *FileID)
3555 {
3556
3557     BOOLEAN bIsVolume = FALSE;
3558
3559     if( FileID->Vnode == 1 &&
3560         FileID->Unique == 1)
3561     {
3562
3563         bIsVolume = TRUE;
3564     }
3565
3566     return bIsVolume;
3567 }
3568
3569 BOOLEAN
3570 AFSIsFinalNode( IN AFSFcb *Fcb)
3571 {
3572
3573     BOOLEAN bIsFinalNode = FALSE;
3574
3575     if( Fcb->Header.NodeTypeCode == AFS_ROOT_FCB ||
3576         Fcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB ||
3577         Fcb->Header.NodeTypeCode == AFS_FILE_FCB ||
3578         Fcb->Header.NodeTypeCode == AFS_DFS_LINK_FCB ||
3579         Fcb->Header.NodeTypeCode == AFS_INVALID_FCB )
3580     {
3581
3582         bIsFinalNode = TRUE;
3583     }
3584     else
3585     {
3586
3587         ASSERT( Fcb->Header.NodeTypeCode == AFS_MOUNT_POINT_FCB ||
3588                 Fcb->Header.NodeTypeCode == AFS_SYMBOLIC_LINK_FCB);
3589     }
3590
3591     return bIsFinalNode;
3592 }
3593
3594 NTSTATUS
3595 AFSUpdateMetaData( IN AFSDirectoryCB *DirEntry,
3596                    IN AFSDirEnumEntry *DirEnumEntry)
3597 {
3598
3599     NTSTATUS ntStatus = STATUS_SUCCESS;
3600     UNICODE_STRING uniTargetName;
3601     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
3602
3603     __Enter
3604     {
3605
3606         pObjectInfo->TargetFileId = DirEnumEntry->TargetFileId;
3607
3608         pObjectInfo->Expiration = DirEnumEntry->Expiration;
3609
3610         pObjectInfo->DataVersion = DirEnumEntry->DataVersion;
3611
3612         pObjectInfo->FileType = DirEnumEntry->FileType;
3613
3614         pObjectInfo->CreationTime = DirEnumEntry->CreationTime;
3615
3616         pObjectInfo->LastAccessTime = DirEnumEntry->LastAccessTime;
3617
3618         pObjectInfo->LastWriteTime = DirEnumEntry->LastWriteTime;
3619
3620         pObjectInfo->ChangeTime = DirEnumEntry->ChangeTime;
3621
3622         pObjectInfo->EndOfFile = DirEnumEntry->EndOfFile;
3623
3624         pObjectInfo->AllocationSize = DirEnumEntry->AllocationSize;
3625
3626         pObjectInfo->FileAttributes = DirEnumEntry->FileAttributes;
3627
3628         if( pObjectInfo->FileType == AFS_FILE_TYPE_MOUNTPOINT)
3629         {
3630
3631             pObjectInfo->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
3632         }
3633
3634         if( pObjectInfo->FileType == AFS_FILE_TYPE_SYMLINK ||
3635             pObjectInfo->FileType == AFS_FILE_TYPE_DFSLINK)
3636         {
3637
3638             pObjectInfo->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
3639         }
3640
3641         pObjectInfo->EaSize = DirEnumEntry->EaSize;
3642
3643         pObjectInfo->Links = DirEnumEntry->Links;
3644
3645         if( DirEnumEntry->TargetNameLength > 0)
3646         {
3647
3648             //
3649             // Update the target name information if needed
3650             //
3651
3652             uniTargetName.Length = (USHORT)DirEnumEntry->TargetNameLength;
3653
3654             uniTargetName.MaximumLength = uniTargetName.Length;
3655
3656             uniTargetName.Buffer = (WCHAR *)((char *)DirEnumEntry + DirEnumEntry->TargetNameOffset);
3657
3658             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3659                             TRUE);
3660
3661             if( DirEntry->NameInformation.TargetName.Length == 0 ||
3662                 RtlCompareUnicodeString( &uniTargetName,
3663                                          &DirEntry->NameInformation.TargetName,
3664                                          TRUE) != 0)
3665             {
3666
3667                 //
3668                 // Update the target name
3669                 //
3670
3671                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
3672                                                 &DirEntry->Flags,
3673                                                 uniTargetName.Buffer,
3674                                                 uniTargetName.Length);
3675
3676                 if( !NT_SUCCESS( ntStatus))
3677                 {
3678
3679                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
3680
3681                     try_return( ntStatus);
3682                 }
3683             }
3684
3685             AFSReleaseResource( &DirEntry->NonPaged->Lock);
3686         }
3687         else if( DirEntry->NameInformation.TargetName.Length > 0)
3688         {
3689
3690             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3691                             TRUE);
3692
3693             if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER) &&
3694                 DirEntry->NameInformation.TargetName.Buffer != NULL)
3695             {
3696                 AFSExFreePool( DirEntry->NameInformation.TargetName.Buffer);
3697             }
3698
3699             ClearFlag( DirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER);
3700
3701             DirEntry->NameInformation.TargetName.Length = 0;
3702             DirEntry->NameInformation.TargetName.MaximumLength = 0;
3703             DirEntry->NameInformation.TargetName.Buffer = NULL;
3704
3705             AFSReleaseResource( &DirEntry->NonPaged->Lock);
3706         }
3707
3708 try_exit:
3709
3710         NOTHING;
3711     }
3712
3713     return ntStatus;
3714 }
3715
3716 NTSTATUS
3717 AFSValidateEntry( IN AFSDirectoryCB *DirEntry,
3718                   IN GUID *AuthGroup,
3719                   IN BOOLEAN FastCall)
3720 {
3721
3722     NTSTATUS ntStatus = STATUS_SUCCESS;
3723     LARGE_INTEGER liSystemTime;
3724     AFSDirEnumEntry *pDirEnumEntry = NULL;
3725     AFSFcb *pCurrentFcb = NULL;
3726     BOOLEAN bReleaseFcb = FALSE;
3727     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
3728
3729     __Enter
3730     {
3731
3732         //
3733         // If we have an Fcb hanging off the directory entry then be sure to acquire the locks in the
3734         // correct order
3735         //
3736
3737         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3738                       AFS_TRACE_LEVEL_VERBOSE_2,
3739                       "AFSValidateEntry Validating entry %wZ FID %08lX-%08lX-%08lX-%08lX FastCall %u\n",
3740                       &DirEntry->NameInformation.FileName,
3741                       pObjectInfo->FileId.Cell,
3742                       pObjectInfo->FileId.Volume,
3743                       pObjectInfo->FileId.Vnode,
3744                       pObjectInfo->FileId.Unique,
3745                       FastCall);
3746
3747         //
3748         // If this is a fake node then bail since the service knows nothing about it
3749         //
3750
3751         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_FAKE))
3752         {
3753
3754             try_return( ntStatus);
3755         }
3756
3757         //
3758         // This routine ensures that the current entry is valid by:
3759         //
3760         //      1) Checking that the expiration time is non-zero and after where we
3761         //         currently are
3762         //
3763
3764         KeQuerySystemTime( &liSystemTime);
3765
3766         if( !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_NOT_EVALUATED) &&
3767             !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY) &&
3768             !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA) &&
3769             pObjectInfo->Expiration.QuadPart >= liSystemTime.QuadPart)
3770         {
3771
3772             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3773                           AFS_TRACE_LEVEL_VERBOSE_2,
3774                           "AFSValidateEntry Directory entry %wZ FID %08lX-%08lX-%08lX-%08lX VALID\n",
3775                           &DirEntry->NameInformation.FileName,
3776                           pObjectInfo->FileId.Cell,
3777                           pObjectInfo->FileId.Volume,
3778                           pObjectInfo->FileId.Vnode,
3779                           pObjectInfo->FileId.Unique);
3780
3781             try_return( ntStatus);
3782         }
3783
3784         //
3785         // This node requires updating
3786         //
3787
3788         ntStatus = AFSEvaluateTargetByID( pObjectInfo,
3789  &n