Windows: Freeing SeQueryInformationToken allocations
[openafs.git] / 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 CHAR *FunctionString,
55                     IN ULONG Code,
56                     IN PEXCEPTION_POINTERS ExceptPtrs)
57 {
58
59     PEXCEPTION_RECORD ExceptRec;
60     PCONTEXT Context;
61
62     __try
63     {
64
65         ExceptRec = ExceptPtrs->ExceptionRecord;
66
67         Context = ExceptPtrs->ContextRecord;
68
69         AFSDbgLogMsg( 0,
70                       0,
71                       "AFSExceptionFilter (Library) - EXR %p CXR %p Function %s Code %08lX Address %p Routine %p\n",
72                       ExceptRec,
73                       Context,
74                       FunctionString,
75                       ExceptRec->ExceptionCode,
76                       ExceptRec->ExceptionAddress,
77                       (void *)AFSExceptionFilter);
78
79         DbgPrint("**** Exception Caught in AFS Redirector Library ****\n");
80
81         DbgPrint("\n\nPerform the following WnDbg Cmds:\n");
82         DbgPrint("\n\t.exr %p ;  .cxr %p\n\n", ExceptRec, Context);
83
84         DbgPrint("**** Exception Complete from AFS Redirector Library ****\n");
85
86         if( BooleanFlagOn( AFSDebugFlags, AFS_DBG_BUGCHECK_EXCEPTION))
87         {
88
89             KeBugCheck( (ULONG)-2);
90         }
91         else
92         {
93
94             AFSBreakPoint();
95         }
96     }
97     __except( EXCEPTION_EXECUTE_HANDLER)
98     {
99
100         NOTHING;
101     }
102
103     return EXCEPTION_EXECUTE_HANDLER;
104 }
105
106 //
107 // Function: AFSLibExAllocatePoolWithTag()
108 //
109 // Purpose: Allocate Pool Memory.  If BugCheck Exception flag
110 //          is configured on, then bugcheck the system if
111 //          a memory allocation fails.  The routine should be
112 //          used for all memory allocations that are to be freed
113 //          when the library is unloaded.  Memory allocations that
114 //          are to survive library unload and reload should be
115 //          performed using AFSExAllocatePoolWithTag() which is
116 //          provided by the AFS Framework.
117 //
118 // Parameters:
119 //                POOL_TYPE PoolType - Paged or NonPaged
120 //                SIZE_T NumberOfBytes - requested allocation size
121 //                ULONG  Tag - Pool Allocation Tag to be applied for tracking
122 //
123 // Return:
124 //                void * - the memory allocation
125 //
126
127 void *
128 AFSLibExAllocatePoolWithTag( IN POOL_TYPE  PoolType,
129                              IN SIZE_T  NumberOfBytes,
130                              IN ULONG  Tag)
131 {
132
133     void *pBuffer = NULL;
134
135     pBuffer = ExAllocatePoolWithTag( PoolType,
136                                      NumberOfBytes,
137                                      Tag);
138
139     if( pBuffer == NULL)
140     {
141
142         if( BooleanFlagOn( AFSDebugFlags, AFS_DBG_BUGCHECK_EXCEPTION))
143         {
144
145             KeBugCheck( (ULONG)-2);
146         }
147         else
148         {
149
150             AFSDbgLogMsg( 0,
151                           0,
152                           "AFSLibExAllocatePoolWithTag failure Type %08lX Size %08lX Tag %08lX %08lX\n",
153                           PoolType,
154                           NumberOfBytes,
155                           Tag,
156                           PsGetCurrentThread());
157
158             AFSBreakPoint();
159         }
160     }
161
162     return pBuffer;
163 }
164
165 //
166 // Function: AFSAcquireExcl()
167 //
168 // Purpose: Called to acquire a resource exclusive with optional wait
169 //
170 // Parameters:
171 //                PERESOURCE Resource - Resource to acquire
172 //                BOOLEAN Wait - Whether to block
173 //
174 // Return:
175 //                BOOLEAN - Whether the mask was acquired
176 //
177
178 BOOLEAN
179 AFSAcquireExcl( IN PERESOURCE Resource,
180                 IN BOOLEAN wait)
181 {
182
183     BOOLEAN bStatus = FALSE;
184
185     //
186     // Normal kernel APCs must be disabled before calling
187     // ExAcquireResourceExclusiveLite. Otherwise a bugcheck occurs.
188     //
189
190     KeEnterCriticalRegion();
191
192     bStatus = ExAcquireResourceExclusiveLite( Resource,
193                                               wait);
194
195     if( !bStatus)
196     {
197
198         KeLeaveCriticalRegion();
199     }
200
201     return bStatus;
202 }
203
204 BOOLEAN
205 AFSAcquireSharedStarveExclusive( IN PERESOURCE Resource,
206                                  IN BOOLEAN Wait)
207 {
208
209     BOOLEAN bStatus = FALSE;
210
211     KeEnterCriticalRegion();
212
213     bStatus = ExAcquireSharedStarveExclusive( Resource,
214                                               Wait);
215
216     if( !bStatus)
217     {
218
219         KeLeaveCriticalRegion();
220     }
221
222     return bStatus;
223 }
224
225 //
226 // Function: AFSAcquireShared()
227 //
228 // Purpose: Called to acquire a resource shared with optional wait
229 //
230 // Parameters:
231 //                PERESOURCE Resource - Resource to acquire
232 //                BOOLEAN Wait - Whether to block
233 //
234 // Return:
235 //                BOOLEAN - Whether the mask was acquired
236 //
237
238 BOOLEAN
239 AFSAcquireShared( IN PERESOURCE Resource,
240                   IN BOOLEAN wait)
241 {
242
243     BOOLEAN bStatus = FALSE;
244
245     KeEnterCriticalRegion();
246
247     bStatus = ExAcquireResourceSharedLite( Resource,
248                                            wait);
249
250     if( !bStatus)
251     {
252
253         KeLeaveCriticalRegion();
254     }
255
256     return bStatus;
257 }
258
259 //
260 // Function: AFSReleaseResource()
261 //
262 // Purpose: Called to release a resource
263 //
264 // Parameters:
265 //                PERESOURCE Resource - Resource to release
266 //
267 // Return:
268 //                None
269 //
270
271 void
272 AFSReleaseResource( IN PERESOURCE Resource)
273 {
274
275     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
276                   AFS_TRACE_LEVEL_VERBOSE,
277                   "AFSReleaseResource Releasing lock %08lX Thread %08lX\n",
278                   Resource,
279                   PsGetCurrentThread());
280
281     ExReleaseResourceLite( Resource);
282
283     KeLeaveCriticalRegion();
284
285     return;
286 }
287
288 void
289 AFSConvertToShared( IN PERESOURCE Resource)
290 {
291
292     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
293                   AFS_TRACE_LEVEL_VERBOSE,
294                   "AFSConvertToShared Converting lock %08lX Thread %08lX\n",
295                   Resource,
296                   PsGetCurrentThread());
297
298     ExConvertExclusiveToSharedLite( Resource);
299
300     return;
301 }
302
303 //
304 // Function: AFSCompleteRequest
305 //
306 // Description:
307 //
308 //      This function completes irps
309 //
310 // Return:
311 //
312 //      A status is returned for the function
313 //
314
315 void
316 AFSCompleteRequest( IN PIRP Irp,
317                     IN ULONG Status)
318 {
319
320     Irp->IoStatus.Status = Status;
321
322     IoCompleteRequest( Irp,
323                        IO_NO_INCREMENT);
324
325     return;
326 }
327
328 //
329 // Function: AFSGenerateCRC
330 //
331 // Description:
332 //
333 //      Given a device and filename this function generates a CRC
334 //
335 // Return:
336 //
337 //      A status is returned for the function
338 //
339
340 ULONG
341 AFSGenerateCRC( IN PUNICODE_STRING FileName,
342                 IN BOOLEAN UpperCaseName)
343 {
344
345     ULONG ulCRC = 0;
346     NTSTATUS ntStatus = STATUS_SUCCESS;
347
348     ntStatus = RtlHashUnicodeString( FileName,
349                                      UpperCaseName,
350                                      HASH_STRING_ALGORITHM_DEFAULT,
351                                      &ulCRC);
352
353     if( !NT_SUCCESS( ntStatus))
354     {
355         ulCRC = 0;
356     }
357
358     return ulCRC;
359 }
360
361 void *
362 AFSLockSystemBuffer( IN PIRP Irp,
363                      IN ULONG Length)
364 {
365
366     NTSTATUS Status = STATUS_SUCCESS;
367     void *pAddress = NULL;
368
369     if( Irp->MdlAddress != NULL)
370     {
371
372         pAddress = MmGetSystemAddressForMdlSafe( Irp->MdlAddress,
373                                                  NormalPagePriority);
374     }
375     else if( Irp->AssociatedIrp.SystemBuffer != NULL)
376     {
377
378         pAddress = Irp->AssociatedIrp.SystemBuffer;
379     }
380     else if( Irp->UserBuffer != NULL)
381     {
382
383         Irp->MdlAddress = IoAllocateMdl( Irp->UserBuffer,
384                                          Length,
385                                          FALSE,
386                                          FALSE,
387                                          Irp);
388
389         if( Irp->MdlAddress != NULL)
390         {
391
392             //
393             //  Lock the new Mdl in memory.
394             //
395
396             __try
397             {
398                 PIO_STACK_LOCATION pIoStack;
399                 pIoStack = IoGetCurrentIrpStackLocation( Irp);
400
401
402                 MmProbeAndLockPages( Irp->MdlAddress, KernelMode,
403                                      (pIoStack->MajorFunction == IRP_MJ_READ) ? IoWriteAccess : IoReadAccess);
404
405                 pAddress = MmGetSystemAddressForMdlSafe( Irp->MdlAddress, NormalPagePriority );
406
407             }
408             __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
409             {
410
411                 AFSDumpTraceFilesFnc();
412
413                 IoFreeMdl( Irp->MdlAddress );
414                 Irp->MdlAddress = NULL;
415                 pAddress = NULL;
416             }
417         }
418     }
419
420     return pAddress;
421 }
422
423 void *
424 AFSLockUserBuffer( IN void *UserBuffer,
425                    IN ULONG BufferLength,
426                                    OUT MDL ** Mdl)
427 {
428
429     NTSTATUS ntStatus = STATUS_SUCCESS;
430     void *pAddress = NULL;
431     MDL *pMdl = NULL;
432
433     __Enter
434     {
435
436         pMdl = IoAllocateMdl( UserBuffer,
437                               BufferLength,
438                               FALSE,
439                               FALSE,
440                               NULL);
441
442             if( pMdl == NULL)
443             {
444
445                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
446             }
447
448         //
449         //  Lock the new Mdl in memory.
450         //
451
452         __try
453         {
454
455             MmProbeAndLockPages( pMdl,
456                                  KernelMode,
457                                  IoWriteAccess);
458
459             pAddress = MmGetSystemAddressForMdlSafe( pMdl,
460                                                      NormalPagePriority);
461         }
462         __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
463         {
464
465             AFSDumpTraceFilesFnc();
466
467             IoFreeMdl( pMdl);
468             pMdl = NULL;
469             pAddress = NULL;
470         }
471
472         if( pMdl != NULL)
473         {
474
475             *Mdl = pMdl;
476         }
477
478 try_exit:
479
480         NOTHING;
481     }
482
483     return pAddress;
484 }
485
486 void *
487 AFSMapToService( IN PIRP Irp,
488                  IN ULONG ByteCount)
489 {
490
491     NTSTATUS ntStatus = STATUS_SUCCESS;
492     void *pMappedBuffer = NULL;
493     AFSDeviceExt *pDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
494     KAPC stApcState;
495
496     __Enter
497     {
498
499         if( pDevExt->Specific.Control.ServiceProcess == NULL)
500         {
501
502             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
503         }
504
505         if( Irp->MdlAddress == NULL)
506         {
507
508             if( AFSLockSystemBuffer( Irp,
509                                      ByteCount) == NULL)
510             {
511
512                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
513             }
514         }
515
516         //
517         // Attach to the service process for mapping
518         //
519
520         KeStackAttachProcess( pDevExt->Specific.Control.ServiceProcess,
521                               (PRKAPC_STATE)&stApcState);
522
523         pMappedBuffer = MmMapLockedPagesSpecifyCache( Irp->MdlAddress,
524                                                       UserMode,
525                                                       MmCached,
526                                                       NULL,
527                                                       FALSE,
528                                                       NormalPagePriority);
529
530         KeUnstackDetachProcess( (PRKAPC_STATE)&stApcState);
531
532 try_exit:
533
534         NOTHING;
535     }
536
537     return pMappedBuffer;
538 }
539
540 NTSTATUS
541 AFSUnmapServiceMappedBuffer( IN void *MappedBuffer,
542                              IN PMDL Mdl)
543 {
544
545     NTSTATUS ntStatus = STATUS_SUCCESS;
546     void *pMappedBuffer = NULL;
547     AFSDeviceExt *pDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
548     KAPC stApcState;
549
550     __Enter
551     {
552
553         if( pDevExt->Specific.Control.ServiceProcess == NULL)
554         {
555
556             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
557         }
558
559         if( Mdl != NULL)
560         {
561
562             //
563             // Attach to the service process for mapping
564             //
565
566             KeStackAttachProcess( pDevExt->Specific.Control.ServiceProcess,
567                                   (PRKAPC_STATE)&stApcState);
568
569             MmUnmapLockedPages( MappedBuffer,
570                                 Mdl);
571
572             KeUnstackDetachProcess( (PRKAPC_STATE)&stApcState);
573         }
574
575 try_exit:
576
577         NOTHING;
578     }
579
580     return ntStatus;
581 }
582
583 NTSTATUS
584 AFSInitializeLibraryDevice()
585 {
586
587     NTSTATUS ntStatus = STATUS_SUCCESS;
588     AFSDeviceExt *pDeviceExt = NULL;
589
590     __Enter
591     {
592
593         pDeviceExt = (AFSDeviceExt *)AFSLibraryDeviceObject->DeviceExtension;
594
595         //
596         // The PIOCtl file name
597         //
598
599         RtlInitUnicodeString( &AFSPIOCtlName,
600                               AFS_PIOCTL_FILE_INTERFACE_NAME);
601
602         //
603         // And the global root share name
604         //
605
606         RtlInitUnicodeString( &AFSGlobalRootName,
607                               AFS_GLOBAL_ROOT_SHARE_NAME);
608
609     }
610
611     return ntStatus;
612 }
613
614 NTSTATUS
615 AFSRemoveLibraryDevice()
616 {
617
618     NTSTATUS ntStatus = STATUS_SUCCESS;
619
620     __Enter
621     {
622
623     }
624
625     return ntStatus;
626 }
627
628 NTSTATUS
629 AFSDefaultDispatch( IN PDEVICE_OBJECT DeviceObject,
630                     IN PIRP Irp)
631 {
632
633     NTSTATUS            ntStatus = STATUS_INVALID_DEVICE_REQUEST;
634     PIO_STACK_LOCATION  pIrpSp = IoGetCurrentIrpStackLocation( Irp);
635
636     AFSCompleteRequest( Irp,
637                         ntStatus);
638
639     return ntStatus;
640 }
641
642 NTSTATUS
643 AFSInitializeGlobalDirectoryEntries()
644 {
645
646     NTSTATUS ntStatus = STATUS_SUCCESS;
647     AFSDirectoryCB *pDirNode = NULL;
648     ULONG ulEntryLength = 0;
649     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
650     AFSObjectInfoCB *pObjectInfoCB = NULL;
651     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
652     LONG lCount;
653
654     __Enter
655     {
656
657         //
658         // Initialize the global . entry
659         //
660
661         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
662                                                0);
663
664         if( pObjectInfoCB == NULL)
665         {
666
667             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
668                           AFS_TRACE_LEVEL_ERROR,
669                           "AFSInitializeGlobalDirectory AFSAllocateObjectInfo failure %08lX\n",
670                           ntStatus);
671
672             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
673         }
674
675         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
676
677         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
678                       AFS_TRACE_LEVEL_VERBOSE,
679                       "AFSInitializeGlobalDirectoryEntries Increment count on object %08lX Cnt %d\n",
680                       pObjectInfoCB,
681                       lCount);
682
683         ntStatus = STATUS_SUCCESS;
684
685         ulEntryLength = sizeof( AFSDirectoryCB) +
686                                      sizeof( WCHAR);
687
688         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
689                                                                   ulEntryLength,
690                                                                   AFS_DIR_ENTRY_TAG);
691
692         if( pDirNode == NULL)
693         {
694
695             AFSDeleteObjectInfo( pObjectInfoCB);
696
697             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
698                           AFS_TRACE_LEVEL_ERROR,
699                           "AFSInitializeGlobalDirectory AFS_DIR_ENTRY_TAG allocation failure\n");
700
701             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
702         }
703
704         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
705                                                                                    sizeof( AFSNonPagedDirectoryCB),
706                                                                                    AFS_DIR_ENTRY_NP_TAG);
707
708         if( pNonPagedDirEntry == NULL)
709         {
710
711             ExFreePool( pDirNode);
712
713             AFSDeleteObjectInfo( pObjectInfoCB);
714
715             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
716                           AFS_TRACE_LEVEL_ERROR,
717                           "AFSInitializeGlobalDirectory AFS_DIR_ENTRY_NP_TAG allocation failure\n");
718
719             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
720         }
721
722         RtlZeroMemory( pDirNode,
723                        ulEntryLength);
724
725         RtlZeroMemory( pNonPagedDirEntry,
726                        sizeof( AFSNonPagedDirectoryCB));
727
728         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
729
730         pDirNode->NonPaged = pNonPagedDirEntry;
731
732         pDirNode->ObjectInformation = pObjectInfoCB;
733
734         //
735         // Set valid entry
736         //
737
738         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE | AFS_DIR_ENTRY_FAKE | AFS_DIR_ENTRY_VALID);
739
740         pDirNode->FileIndex = (ULONG)AFS_DIR_ENTRY_DOT_INDEX;
741
742         //
743         // Setup the names in the entry
744         //
745
746         pDirNode->NameInformation.FileName.Length = sizeof( WCHAR);
747
748         pDirNode->NameInformation.FileName.MaximumLength = sizeof( WCHAR);
749
750         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
751
752         pDirNode->NameInformation.FileName.Buffer[ 0] = L'.';
753
754         //
755         // Populate the rest of the data
756         //
757
758         pObjectInfoCB->FileType = AFS_FILE_TYPE_DIRECTORY;
759
760         pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
761
762         AFSGlobalDotDirEntry = pDirNode;
763
764         //
765         // Now the .. entry
766         //
767
768         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
769                                                0);
770
771         if( pObjectInfoCB == NULL)
772         {
773
774             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
775                           AFS_TRACE_LEVEL_ERROR,
776                           "AFSInitializeGlobalDirectory AFSAllocateObjectInfo (2) failure %08lX\n",
777                           ntStatus);
778
779             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
780         }
781
782         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
783
784         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
785                       AFS_TRACE_LEVEL_VERBOSE,
786                       "AFSInitializeGlobalDirectoryEntries Increment count on object %08lX Cnt %d\n",
787                       pObjectInfoCB,
788                       lCount);
789
790         ntStatus = STATUS_SUCCESS;
791
792         ulEntryLength = sizeof( AFSDirectoryCB) +
793                                      ( 2 * sizeof( WCHAR));
794
795         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
796                                                                   ulEntryLength,
797                                                                   AFS_DIR_ENTRY_TAG);
798
799         if( pDirNode == NULL)
800         {
801
802             AFSDeleteObjectInfo( pObjectInfoCB);
803
804             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
805         }
806
807         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
808                                                                                    sizeof( AFSNonPagedDirectoryCB),
809                                                                                    AFS_DIR_ENTRY_NP_TAG);
810
811         if( pNonPagedDirEntry == NULL)
812         {
813
814             ExFreePool( pDirNode);
815
816             AFSDeleteObjectInfo( pObjectInfoCB);
817
818             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
819         }
820
821         RtlZeroMemory( pDirNode,
822                        ulEntryLength);
823
824         RtlZeroMemory( pNonPagedDirEntry,
825                        sizeof( AFSNonPagedDirectoryCB));
826
827         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
828
829         pDirNode->NonPaged = pNonPagedDirEntry;
830
831         pDirNode->ObjectInformation = pObjectInfoCB;
832
833         //
834         // Set valid entry
835         //
836
837         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE | AFS_DIR_ENTRY_FAKE | AFS_DIR_ENTRY_VALID);
838
839         pDirNode->FileIndex = (ULONG)AFS_DIR_ENTRY_DOT_DOT_INDEX;
840
841         //
842         // Setup the names in the entry
843         //
844
845         pDirNode->NameInformation.FileName.Length = 2 * sizeof( WCHAR);
846
847         pDirNode->NameInformation.FileName.MaximumLength = 2 * sizeof( WCHAR);
848
849         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
850
851         pDirNode->NameInformation.FileName.Buffer[ 0] = L'.';
852
853         pDirNode->NameInformation.FileName.Buffer[ 1] = L'.';
854
855         //
856         // Populate the rest of the data
857         //
858
859         pObjectInfoCB->FileType = AFS_FILE_TYPE_DIRECTORY;
860
861         pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_DIRECTORY;
862
863         AFSGlobalDotDotDirEntry = pDirNode;
864
865 try_exit:
866
867         if( !NT_SUCCESS( ntStatus))
868         {
869
870             if( AFSGlobalDotDirEntry != NULL)
871             {
872
873                 AFSDeleteObjectInfo( AFSGlobalDotDirEntry->ObjectInformation);
874
875                 ExDeleteResourceLite( &AFSGlobalDotDirEntry->NonPaged->Lock);
876
877                 ExFreePool( AFSGlobalDotDirEntry->NonPaged);
878
879                 ExFreePool( AFSGlobalDotDirEntry);
880
881                 AFSGlobalDotDirEntry = NULL;
882             }
883
884             if( AFSGlobalDotDotDirEntry != NULL)
885             {
886
887                 AFSDeleteObjectInfo( AFSGlobalDotDotDirEntry->ObjectInformation);
888
889                 ExDeleteResourceLite( &AFSGlobalDotDotDirEntry->NonPaged->Lock);
890
891                 ExFreePool( AFSGlobalDotDotDirEntry->NonPaged);
892
893                 ExFreePool( AFSGlobalDotDotDirEntry);
894
895                 AFSGlobalDotDotDirEntry = NULL;
896             }
897         }
898     }
899
900     return ntStatus;
901 }
902
903 AFSDirectoryCB *
904 AFSInitDirEntry( IN AFSObjectInfoCB *ParentObjectInfo,
905                  IN PUNICODE_STRING FileName,
906                  IN PUNICODE_STRING TargetName,
907                  IN AFSDirEnumEntry *DirEnumEntry,
908                  IN ULONG FileIndex)
909 {
910
911     AFSDirectoryCB *pDirNode = NULL;
912     NTSTATUS ntStatus = STATUS_SUCCESS;
913     ULONG ulEntryLength = 0;
914     AFSDirEnumEntry *pDirEnumCB = NULL;
915     AFSFileID stTargetFileID;
916     AFSFcb *pVcb = NULL;
917     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
918     AFSObjectInfoCB *pObjectInfoCB = NULL;
919     BOOLEAN bAllocatedObjectCB = FALSE;
920     ULONGLONG ullIndex = 0;
921     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
922     LONG lCount;
923
924     __Enter
925     {
926
927         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
928                       AFS_TRACE_LEVEL_VERBOSE,
929                       "AFSInitDirEntry Initializing entry %wZ parent FID %08lX-%08lX-%08lX-%08lX\n",
930                       FileName,
931                       ParentObjectInfo->FileId.Cell,
932                       ParentObjectInfo->FileId.Volume,
933                       ParentObjectInfo->FileId.Vnode,
934                       ParentObjectInfo->FileId.Unique);
935
936         //
937         // First thing is to locate/create our object information block
938         // for this entry
939         //
940
941         AFSAcquireExcl( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock,
942                         TRUE);
943
944         ullIndex = AFSCreateLowIndex( &DirEnumEntry->FileId);
945
946         ntStatus = AFSLocateHashEntry( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
947                                        ullIndex,
948                                        (AFSBTreeEntry **)&pObjectInfoCB);
949
950         if( !NT_SUCCESS( ntStatus) ||
951             pObjectInfoCB == NULL)
952         {
953
954             //
955             // Allocate our object info cb
956             //
957
958             pObjectInfoCB = AFSAllocateObjectInfo( ParentObjectInfo,
959                                                    ullIndex);
960
961             if( pObjectInfoCB == NULL)
962             {
963
964                 AFSReleaseResource( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
965
966                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
967             }
968
969             bAllocatedObjectCB = TRUE;
970
971             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
972                           AFS_TRACE_LEVEL_VERBOSE,
973                           "AFSInitDirEntry initialized object %08lX Parent Object %08lX for %wZ\n",
974                           pObjectInfoCB,
975                           ParentObjectInfo,
976                           FileName);
977         }
978
979         lCount = InterlockedIncrement( &pObjectInfoCB->ObjectReferenceCount);
980
981         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
982                       AFS_TRACE_LEVEL_VERBOSE,
983                       "AFSInitDirEntry Increment count on object %08lX Cnt %d\n",
984                       pObjectInfoCB,
985                       lCount);
986
987         AFSReleaseResource( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
988
989         ntStatus = STATUS_SUCCESS;
990
991         ulEntryLength = sizeof( AFSDirectoryCB) +
992                                      FileName->Length;
993
994         if( TargetName != NULL)
995         {
996
997             ulEntryLength += TargetName->Length;
998         }
999
1000         pDirNode = (AFSDirectoryCB *)AFSExAllocatePoolWithTag( PagedPool,
1001                                                                ulEntryLength,
1002                                                                AFS_DIR_ENTRY_TAG);
1003
1004         if( pDirNode == NULL)
1005         {
1006
1007             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1008         }
1009
1010         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSExAllocatePoolWithTag( NonPagedPool,
1011                                                                                 sizeof( AFSNonPagedDirectoryCB),
1012                                                                                 AFS_DIR_ENTRY_NP_TAG);
1013
1014         if( pNonPagedDirEntry == NULL)
1015         {
1016
1017             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1018         }
1019
1020         RtlZeroMemory( pDirNode,
1021                        ulEntryLength);
1022
1023         RtlZeroMemory( pNonPagedDirEntry,
1024                        sizeof( AFSNonPagedDirectoryCB));
1025
1026         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
1027
1028         pDirNode->NonPaged = pNonPagedDirEntry;
1029
1030         pDirNode->ObjectInformation = pObjectInfoCB;
1031
1032         //
1033         // Set valid entry and NOT_IN_PARENT flag
1034         //
1035
1036         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_NOT_IN_PARENT_TREE);
1037
1038         pDirNode->FileIndex = FileIndex;
1039
1040         //
1041         // Setup the names in the entry
1042         //
1043
1044         if( FileName->Length > 0)
1045         {
1046
1047             pDirNode->NameInformation.FileName.Length = FileName->Length;
1048
1049             pDirNode->NameInformation.FileName.MaximumLength = FileName->Length;
1050
1051             pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
1052
1053             RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
1054                            FileName->Buffer,
1055                            pDirNode->NameInformation.FileName.Length);
1056
1057             //
1058             // Create a CRC for the file
1059             //
1060
1061             pDirNode->CaseSensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
1062                                                                          FALSE);
1063
1064             pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
1065                                                                            TRUE);
1066         }
1067
1068         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1069                       AFS_TRACE_LEVEL_VERBOSE,
1070                       "AFSInitDirEntry Initialized DE %p for %wZ in parent FID %08lX-%08lX-%08lX-%08lX\n",
1071                       pDirNode,
1072                       FileName,
1073                       ParentObjectInfo->FileId.Cell,
1074                       ParentObjectInfo->FileId.Volume,
1075                       ParentObjectInfo->FileId.Vnode,
1076                       ParentObjectInfo->FileId.Unique);
1077
1078         if( TargetName != NULL &&
1079             TargetName->Length > 0)
1080         {
1081
1082             pDirNode->NameInformation.TargetName.Length = TargetName->Length;
1083
1084             pDirNode->NameInformation.TargetName.MaximumLength = pDirNode->NameInformation.TargetName.Length;
1085
1086             pDirNode->NameInformation.TargetName.Buffer = (WCHAR *)((char *)pDirNode +
1087                                                                             sizeof( AFSDirectoryCB) +
1088                                                                             pDirNode->NameInformation.FileName.Length);
1089
1090             RtlCopyMemory( pDirNode->NameInformation.TargetName.Buffer,
1091                            TargetName->Buffer,
1092                            pDirNode->NameInformation.TargetName.Length);
1093         }
1094
1095         //
1096         // If we allocated the object information cb then update the information
1097         //
1098
1099         if( bAllocatedObjectCB)
1100         {
1101
1102             //
1103             // Populate the rest of the data
1104             //
1105
1106             pObjectInfoCB->FileId = DirEnumEntry->FileId;
1107
1108             pObjectInfoCB->TargetFileId = DirEnumEntry->TargetFileId;
1109
1110             pObjectInfoCB->FileType = DirEnumEntry->FileType;
1111
1112             pObjectInfoCB->CreationTime = DirEnumEntry->CreationTime;
1113
1114             pObjectInfoCB->LastAccessTime = DirEnumEntry->LastAccessTime;
1115
1116             pObjectInfoCB->LastWriteTime = DirEnumEntry->LastWriteTime;
1117
1118             pObjectInfoCB->ChangeTime = DirEnumEntry->ChangeTime;
1119
1120             pObjectInfoCB->EndOfFile = DirEnumEntry->EndOfFile;
1121
1122             pObjectInfoCB->AllocationSize = DirEnumEntry->AllocationSize;
1123
1124             pObjectInfoCB->FileAttributes = DirEnumEntry->FileAttributes;
1125
1126             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1127             {
1128
1129                 pObjectInfoCB->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1130             }
1131
1132             if (pObjectInfoCB->FileType == AFS_FILE_TYPE_SYMLINK ||
1133                 pObjectInfoCB->FileType == AFS_FILE_TYPE_DFSLINK)
1134             {
1135
1136                 pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1137             }
1138
1139             pObjectInfoCB->EaSize = DirEnumEntry->EaSize;
1140
1141             //
1142             // Object specific information
1143             //
1144
1145             pObjectInfoCB->Links = DirEnumEntry->Links;
1146
1147             pObjectInfoCB->Expiration = DirEnumEntry->Expiration;
1148
1149             pObjectInfoCB->DataVersion = DirEnumEntry->DataVersion;
1150
1151             //
1152             // Check for the case where we have a filetype of SymLink but both the TargetFid and the
1153             // TargetName are empty. In this case set the filetype to zero so we evaluate it later in
1154             // the code
1155             //
1156
1157             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_SYMLINK &&
1158                 pObjectInfoCB->TargetFileId.Vnode == 0 &&
1159                 pObjectInfoCB->TargetFileId.Unique == 0 &&
1160                 pDirNode->NameInformation.TargetName.Length == 0)
1161             {
1162
1163                 //
1164                 // This will ensure we perform a validation on the node
1165                 //
1166
1167                 pObjectInfoCB->FileType = AFS_FILE_TYPE_UNKNOWN;
1168             }
1169
1170             if( pObjectInfoCB->FileType == AFS_FILE_TYPE_UNKNOWN)
1171             {
1172
1173                 SetFlag( pObjectInfoCB->Flags, AFS_OBJECT_FLAGS_NOT_EVALUATED);
1174             }
1175         }
1176
1177 try_exit:
1178
1179         if( !NT_SUCCESS( ntStatus))
1180         {
1181
1182             if( pNonPagedDirEntry != NULL)
1183             {
1184
1185                 ExDeleteResourceLite( &pNonPagedDirEntry->Lock);
1186
1187                 AFSExFreePool( pNonPagedDirEntry);
1188             }
1189
1190             if( pDirNode != NULL)
1191             {
1192
1193                 AFSExFreePool( pDirNode);
1194
1195                 pDirNode = NULL;
1196             }
1197
1198             //
1199             // Dereference our object info block if we have one
1200             //
1201
1202             if( pObjectInfoCB != NULL)
1203             {
1204
1205                 lCount = InterlockedDecrement( &pObjectInfoCB->ObjectReferenceCount);
1206
1207                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
1208                               AFS_TRACE_LEVEL_VERBOSE,
1209                               "AFSInitDirEntry Decrement count on object %08lX Cnt %d\n",
1210                               pObjectInfoCB,
1211                               lCount);
1212
1213                 if( bAllocatedObjectCB)
1214                 {
1215
1216                     ASSERT( pObjectInfoCB->ObjectReferenceCount == 0);
1217
1218                     AFSDeleteObjectInfo( pObjectInfoCB);
1219                 }
1220             }
1221         }
1222     }
1223
1224     return pDirNode;
1225 }
1226
1227 BOOLEAN
1228 AFSCheckForReadOnlyAccess( IN ACCESS_MASK DesiredAccess,
1229                            IN BOOLEAN DirectoryEntry)
1230 {
1231
1232     BOOLEAN bReturn = TRUE;
1233     ACCESS_MASK stAccessMask = 0;
1234
1235     //
1236     // Get rid of anything we don't know about
1237     //
1238
1239     DesiredAccess = (DesiredAccess   &
1240                           ( DELETE |
1241                             READ_CONTROL |
1242                             WRITE_OWNER |
1243                             WRITE_DAC |
1244                             SYNCHRONIZE |
1245                             ACCESS_SYSTEM_SECURITY |
1246                             FILE_WRITE_DATA |
1247                             FILE_READ_EA |
1248                             FILE_WRITE_EA |
1249                             FILE_READ_ATTRIBUTES |
1250                             FILE_WRITE_ATTRIBUTES |
1251                             FILE_LIST_DIRECTORY |
1252                             FILE_TRAVERSE |
1253                             FILE_DELETE_CHILD |
1254                             FILE_APPEND_DATA));
1255
1256     //
1257     // Our 'read only' access mask. These are the accesses we will
1258     // allow for a read only file
1259     //
1260
1261     stAccessMask = DELETE |
1262                         READ_CONTROL |
1263                         WRITE_OWNER |
1264                         WRITE_DAC |
1265                         SYNCHRONIZE |
1266                         ACCESS_SYSTEM_SECURITY |
1267                         FILE_READ_DATA |
1268                         FILE_READ_EA |
1269                         FILE_WRITE_EA |
1270                         FILE_READ_ATTRIBUTES |
1271                         FILE_WRITE_ATTRIBUTES |
1272                         FILE_EXECUTE |
1273                         FILE_LIST_DIRECTORY |
1274                         FILE_TRAVERSE;
1275
1276     //
1277     // For a directory, add in the directory specific accesses
1278     //
1279
1280     if( DirectoryEntry)
1281     {
1282
1283         stAccessMask |= FILE_ADD_SUBDIRECTORY |
1284                                 FILE_ADD_FILE |
1285                                 FILE_DELETE_CHILD;
1286     }
1287
1288     if( FlagOn( DesiredAccess, ~stAccessMask))
1289     {
1290
1291         //
1292         // A write access is set ...
1293         //
1294
1295         bReturn = FALSE;
1296     }
1297
1298     return bReturn;
1299 }
1300
1301 NTSTATUS
1302 AFSEvaluateNode( IN GUID *AuthGroup,
1303                  IN AFSDirectoryCB *DirEntry)
1304 {
1305
1306     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1307     NTSTATUS ntStatus = STATUS_SUCCESS;
1308     AFSDirEnumEntry *pDirEntry = NULL;
1309     UNICODE_STRING uniTargetName;
1310
1311     __Enter
1312     {
1313
1314         ntStatus = AFSEvaluateTargetByID( DirEntry->ObjectInformation,
1315                                           AuthGroup,
1316                                           FALSE,
1317                                           &pDirEntry);
1318
1319         if( !NT_SUCCESS( ntStatus))
1320         {
1321
1322             try_return( ntStatus);
1323         }
1324
1325         DirEntry->ObjectInformation->TargetFileId = pDirEntry->TargetFileId;
1326
1327         DirEntry->ObjectInformation->Expiration = pDirEntry->Expiration;
1328
1329         DirEntry->ObjectInformation->DataVersion = pDirEntry->DataVersion;
1330
1331         DirEntry->ObjectInformation->FileType = pDirEntry->FileType;
1332
1333         DirEntry->ObjectInformation->CreationTime = pDirEntry->CreationTime;
1334
1335         DirEntry->ObjectInformation->LastAccessTime = pDirEntry->LastAccessTime;
1336
1337         DirEntry->ObjectInformation->LastWriteTime = pDirEntry->LastWriteTime;
1338
1339         DirEntry->ObjectInformation->ChangeTime = pDirEntry->ChangeTime;
1340
1341         DirEntry->ObjectInformation->EndOfFile = pDirEntry->EndOfFile;
1342
1343         DirEntry->ObjectInformation->AllocationSize = pDirEntry->AllocationSize;
1344
1345         DirEntry->ObjectInformation->FileAttributes = pDirEntry->FileAttributes;
1346
1347         if( pDirEntry->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1348         {
1349
1350             DirEntry->ObjectInformation->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1351         }
1352
1353         if( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK ||
1354             pDirEntry->FileType == AFS_FILE_TYPE_DFSLINK)
1355         {
1356
1357             DirEntry->ObjectInformation->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1358         }
1359
1360         DirEntry->ObjectInformation->EaSize = pDirEntry->EaSize;
1361
1362         DirEntry->ObjectInformation->Links = pDirEntry->Links;
1363
1364         //
1365         // If we have a target name then see if it needs updating ...
1366         //
1367
1368         if( pDirEntry->TargetNameLength > 0)
1369         {
1370
1371             //
1372             // Update the target name information if needed
1373             //
1374
1375             uniTargetName.Length = (USHORT)pDirEntry->TargetNameLength;
1376
1377             uniTargetName.MaximumLength = uniTargetName.Length;
1378
1379             uniTargetName.Buffer = (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset);
1380
1381             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
1382                             TRUE);
1383
1384             if( DirEntry->NameInformation.TargetName.Length == 0 ||
1385                 RtlCompareUnicodeString( &uniTargetName,
1386                                          &DirEntry->NameInformation.TargetName,
1387                                          TRUE) != 0)
1388             {
1389
1390                 //
1391                 // Update the target name
1392                 //
1393
1394                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
1395                                                 &DirEntry->Flags,
1396                                                 uniTargetName.Buffer,
1397                                                 uniTargetName.Length);
1398
1399                 if( !NT_SUCCESS( ntStatus))
1400                 {
1401
1402                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
1403
1404                     try_return( ntStatus);
1405                 }
1406             }
1407
1408             AFSReleaseResource( &DirEntry->NonPaged->Lock);
1409         }
1410
1411 try_exit:
1412
1413         if( pDirEntry != NULL)
1414         {
1415
1416             AFSExFreePool( pDirEntry);
1417         }
1418     }
1419
1420     return ntStatus;
1421 }
1422
1423 NTSTATUS
1424 AFSValidateSymLink( IN GUID *AuthGroup,
1425                     IN AFSDirectoryCB *DirEntry)
1426 {
1427
1428     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1429     NTSTATUS ntStatus = STATUS_SUCCESS;
1430     AFSDirEnumEntry *pDirEntry = NULL;
1431     UNICODE_STRING uniTargetName;
1432
1433     __Enter
1434     {
1435
1436         ntStatus = AFSEvaluateTargetByID( DirEntry->ObjectInformation,
1437                                           AuthGroup,
1438                                           FALSE,
1439                                           &pDirEntry);
1440
1441         if( !NT_SUCCESS( ntStatus))
1442         {
1443
1444             try_return( ntStatus);
1445         }
1446
1447         if( pDirEntry->FileType == AFS_FILE_TYPE_UNKNOWN ||
1448             pDirEntry->FileType == AFS_FILE_TYPE_INVALID)
1449         {
1450
1451             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
1452         }
1453
1454         DirEntry->ObjectInformation->TargetFileId = pDirEntry->TargetFileId;
1455
1456         DirEntry->ObjectInformation->Expiration = pDirEntry->Expiration;
1457
1458         DirEntry->ObjectInformation->DataVersion = pDirEntry->DataVersion;
1459
1460         //
1461         // Update the target name information if needed
1462         //
1463
1464         uniTargetName.Length = (USHORT)pDirEntry->TargetNameLength;
1465
1466         uniTargetName.MaximumLength = uniTargetName.Length;
1467
1468         uniTargetName.Buffer = (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset);
1469
1470         if( uniTargetName.Length > 0)
1471         {
1472
1473             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
1474                             TRUE);
1475
1476             if( DirEntry->NameInformation.TargetName.Length == 0 ||
1477                 RtlCompareUnicodeString( &uniTargetName,
1478                                          &DirEntry->NameInformation.TargetName,
1479                                          TRUE) != 0)
1480             {
1481
1482                 //
1483                 // Update the target name
1484                 //
1485
1486                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
1487                                                 &DirEntry->Flags,
1488                                                 uniTargetName.Buffer,
1489                                                 uniTargetName.Length);
1490
1491                 if( !NT_SUCCESS( ntStatus))
1492                 {
1493
1494                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
1495
1496                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1497                 }
1498             }
1499
1500             AFSReleaseResource( &DirEntry->NonPaged->Lock);
1501         }
1502
1503         //
1504         // If the FileType is the same then nothing to do since it IS
1505         // a SymLink
1506         //
1507
1508         if( pDirEntry->FileType == DirEntry->ObjectInformation->FileType)
1509         {
1510
1511             ASSERT( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK);
1512
1513             try_return( ntStatus = STATUS_SUCCESS);
1514         }
1515
1516         DirEntry->ObjectInformation->FileType = pDirEntry->FileType;
1517
1518         DirEntry->ObjectInformation->CreationTime = pDirEntry->CreationTime;
1519
1520         DirEntry->ObjectInformation->LastAccessTime = pDirEntry->LastAccessTime;
1521
1522         DirEntry->ObjectInformation->LastWriteTime = pDirEntry->LastWriteTime;
1523
1524         DirEntry->ObjectInformation->ChangeTime = pDirEntry->ChangeTime;
1525
1526         DirEntry->ObjectInformation->EndOfFile = pDirEntry->EndOfFile;
1527
1528         DirEntry->ObjectInformation->AllocationSize = pDirEntry->AllocationSize;
1529
1530         DirEntry->ObjectInformation->FileAttributes = pDirEntry->FileAttributes;
1531
1532         if( pDirEntry->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1533         {
1534
1535             DirEntry->ObjectInformation->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
1536         }
1537
1538         if( pDirEntry->FileType == AFS_FILE_TYPE_SYMLINK ||
1539             pDirEntry->FileType == AFS_FILE_TYPE_DFSLINK)
1540         {
1541
1542             DirEntry->ObjectInformation->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
1543         }
1544
1545         DirEntry->ObjectInformation->EaSize = pDirEntry->EaSize;
1546
1547         DirEntry->ObjectInformation->Links = pDirEntry->Links;
1548
1549 try_exit:
1550
1551         if( pDirEntry != NULL)
1552         {
1553
1554             AFSExFreePool( pDirEntry);
1555         }
1556     }
1557
1558     return ntStatus;
1559 }
1560
1561 NTSTATUS
1562 AFSInvalidateObject( IN OUT AFSObjectInfoCB **ppObjectInfo,
1563                      IN     ULONG Reason)
1564 {
1565
1566     NTSTATUS ntStatus = STATUS_SUCCESS;
1567     IO_STATUS_BLOCK stIoStatus;
1568     ULONG ulFilter = 0;
1569
1570     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1571                   AFS_TRACE_LEVEL_VERBOSE,
1572                   "AFSInvalidateObject Invalidation on node type %d for fid %08lX-%08lX-%08lX-%08lX Reason %d\n",
1573                   (*ppObjectInfo)->FileType,
1574                   (*ppObjectInfo)->FileId.Cell,
1575                   (*ppObjectInfo)->FileId.Volume,
1576                   (*ppObjectInfo)->FileId.Vnode,
1577                   (*ppObjectInfo)->FileId.Unique,
1578                   Reason);
1579
1580     if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_SYMLINK ||
1581         (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DFSLINK ||
1582         (*ppObjectInfo)->FileType == AFS_FILE_TYPE_MOUNTPOINT)
1583     {
1584         //
1585         // We only act on the mount point itself, not the target. If the
1586         // node has been deleted then mark it as such otherwise indicate
1587         // it requires verification
1588         //
1589
1590         if( Reason == AFS_INVALIDATE_DELETED)
1591         {
1592             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
1593         }
1594         else
1595         {
1596
1597             if( Reason == AFS_INVALIDATE_FLUSHED)
1598             {
1599
1600                 (*ppObjectInfo)->DataVersion.QuadPart = (ULONGLONG)-1;
1601
1602                 SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
1603             }
1604
1605             (*ppObjectInfo)->Expiration.QuadPart = 0;
1606
1607             (*ppObjectInfo)->TargetFileId.Vnode = 0;
1608
1609             (*ppObjectInfo)->TargetFileId.Unique = 0;
1610
1611             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1612                           AFS_TRACE_LEVEL_VERBOSE,
1613                           "AFSInvalidateObject Setting VERIFY flag on fid %08lX-%08lX-%08lX-%08lX\n",
1614                           (*ppObjectInfo)->FileId.Cell,
1615                           (*ppObjectInfo)->FileId.Volume,
1616                           (*ppObjectInfo)->FileId.Vnode,
1617                           (*ppObjectInfo)->FileId.Unique);
1618
1619             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY);
1620         }
1621
1622         ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1623
1624         if( Reason == AFS_INVALIDATE_CREDS)
1625         {
1626             ulFilter |= FILE_NOTIFY_CHANGE_SECURITY;
1627         }
1628
1629         if( Reason == AFS_INVALIDATE_DATA_VERSION ||
1630             Reason == AFS_INVALIDATE_FLUSHED)
1631         {
1632             ulFilter |= FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE;
1633         }
1634         else
1635         {
1636             ulFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1637         }
1638
1639         AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1640                                         NULL,
1641                                         ulFilter,
1642                                         FILE_ACTION_MODIFIED);
1643
1644         try_return( ntStatus);
1645     }
1646
1647     //
1648     // Depending on the reason for invalidation then perform work on the node
1649     //
1650
1651     switch( Reason)
1652     {
1653
1654     case AFS_INVALIDATE_DELETED:
1655         {
1656
1657             //
1658             // Mark this node as invalid
1659             //
1660
1661             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_DELETED);
1662
1663             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1664                           AFS_TRACE_LEVEL_VERBOSE,
1665                           "AFSInvalidateObject Set DELETE flag on fid %08lX-%08lX-%08lX-%08lX\n",
1666                           (*ppObjectInfo)->FileId.Cell,
1667                           (*ppObjectInfo)->FileId.Volume,
1668                           (*ppObjectInfo)->FileId.Vnode,
1669                           (*ppObjectInfo)->FileId.Unique);
1670
1671             if( (*ppObjectInfo)->ParentObjectInformation != NULL)
1672             {
1673
1674                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1675                               AFS_TRACE_LEVEL_VERBOSE,
1676                               "AFSInvalidateObject Set VERIFY flag on parent fid %08lX-%08lX-%08lX-%08lX\n",
1677                               (*ppObjectInfo)->ParentObjectInformation->FileId.Cell,
1678                               (*ppObjectInfo)->ParentObjectInformation->FileId.Volume,
1679                               (*ppObjectInfo)->ParentObjectInformation->FileId.Vnode,
1680                               (*ppObjectInfo)->ParentObjectInformation->FileId.Unique);
1681
1682                 SetFlag( (*ppObjectInfo)->ParentObjectInformation->Flags, AFS_OBJECT_FLAGS_VERIFY);
1683
1684                 (*ppObjectInfo)->ParentObjectInformation->DataVersion.QuadPart = (ULONGLONG)-1;
1685
1686                 (*ppObjectInfo)->ParentObjectInformation->Expiration.QuadPart = 0;
1687             }
1688
1689             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DIRECTORY)
1690             {
1691                 ulFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
1692             }
1693             else
1694             {
1695                 ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1696             }
1697
1698             AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1699                                             NULL,
1700                                             ulFilter,
1701                                             FILE_ACTION_REMOVED);
1702
1703             if( NT_SUCCESS( AFSQueueInvalidateObject( (*ppObjectInfo),
1704                                                       Reason)))
1705             {
1706                 (*ppObjectInfo) = NULL; // We'll dec the count in the worker item
1707             }
1708
1709             break;
1710         }
1711
1712     case AFS_INVALIDATE_FLUSHED:
1713         {
1714
1715             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE &&
1716                 (*ppObjectInfo)->Fcb != NULL)
1717             {
1718
1719                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1720                               AFS_TRACE_LEVEL_VERBOSE,
1721                               "AFSInvalidateObject Flush/purge file fid %08lX-%08lX-%08lX-%08lX\n",
1722                               (*ppObjectInfo)->FileId.Cell,
1723                               (*ppObjectInfo)->FileId.Volume,
1724                               (*ppObjectInfo)->FileId.Vnode,
1725                               (*ppObjectInfo)->FileId.Unique);
1726
1727                 AFSAcquireExcl( &(*ppObjectInfo)->Fcb->NPFcb->Resource,
1728                                 TRUE);
1729
1730                 __try
1731                 {
1732
1733                     CcFlushCache( &(*ppObjectInfo)->Fcb->NPFcb->SectionObjectPointers,
1734                                   NULL,
1735                                   0,
1736                                   &stIoStatus);
1737
1738                     if( !NT_SUCCESS( stIoStatus.Status))
1739                     {
1740
1741                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
1742                                       AFS_TRACE_LEVEL_ERROR,
1743                                       "AFSInvalidateObject CcFlushCache failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
1744                                       (*ppObjectInfo)->FileId.Cell,
1745                                       (*ppObjectInfo)->FileId.Volume,
1746                                       (*ppObjectInfo)->FileId.Vnode,
1747                                       (*ppObjectInfo)->FileId.Unique,
1748                                       stIoStatus.Status,
1749                                       stIoStatus.Information);
1750
1751                         ntStatus = stIoStatus.Status;
1752                     }
1753
1754                     if ( !CcPurgeCacheSection( &(*ppObjectInfo)->Fcb->NPFcb->SectionObjectPointers,
1755                                                NULL,
1756                                                0,
1757                                                FALSE))
1758                     {
1759
1760                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
1761                                       AFS_TRACE_LEVEL_WARNING,
1762                                       "AFSInvalidateObject CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
1763                                       (*ppObjectInfo)->FileId.Cell,
1764                                       (*ppObjectInfo)->FileId.Volume,
1765                                       (*ppObjectInfo)->FileId.Vnode,
1766                                       (*ppObjectInfo)->FileId.Unique);
1767
1768                         SetFlag( (*ppObjectInfo)->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
1769                     }
1770                 }
1771                 __except( EXCEPTION_EXECUTE_HANDLER)
1772                 {
1773
1774                     ntStatus = GetExceptionCode();
1775
1776                     AFSDbgLogMsg( 0,
1777                                   0,
1778                                   "EXCEPTION - AFSInvalidateObject Cc FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
1779                                   (*ppObjectInfo)->FileId.Cell,
1780                                   (*ppObjectInfo)->FileId.Volume,
1781                                   (*ppObjectInfo)->FileId.Vnode,
1782                                   (*ppObjectInfo)->FileId.Unique,
1783                                   ntStatus);
1784
1785                     SetFlag( (*ppObjectInfo)->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
1786                 }
1787
1788                 AFSReleaseResource( &(*ppObjectInfo)->Fcb->NPFcb->Resource);
1789
1790                 //
1791                 // Clear out the extents
1792                 // Get rid of them (note this involves waiting
1793                 // for any writes or reads to the cache to complete)
1794                 //
1795
1796                 AFSTearDownFcbExtents( (*ppObjectInfo)->Fcb,
1797                                        NULL);
1798             }
1799
1800             (*ppObjectInfo)->DataVersion.QuadPart = (ULONGLONG)-1;
1801
1802
1803             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE)
1804             {
1805
1806                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1807                               AFS_TRACE_LEVEL_VERBOSE,
1808                               "AFSInvalidateObject Setting VERIFY_DATA flag on fid %08lX-%08lX-%08lX-%08lX\n",
1809                               (*ppObjectInfo)->FileId.Cell,
1810                               (*ppObjectInfo)->FileId.Volume,
1811                               (*ppObjectInfo)->FileId.Vnode,
1812                               (*ppObjectInfo)->FileId.Unique);
1813
1814                 SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
1815             }
1816
1817             // Fall through to the default processing
1818         }
1819
1820     default:
1821         {
1822
1823             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DIRECTORY)
1824             {
1825                 ulFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
1826             }
1827             else
1828             {
1829                 ulFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
1830             }
1831
1832             if( Reason == AFS_INVALIDATE_CREDS)
1833             {
1834                 ulFilter |= FILE_NOTIFY_CHANGE_SECURITY;
1835             }
1836
1837             if( Reason == AFS_INVALIDATE_DATA_VERSION)
1838             {
1839                 ulFilter |= FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE;
1840             }
1841             else
1842             {
1843                 ulFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1844             }
1845
1846             if( (*ppObjectInfo)->FileType == AFS_FILE_TYPE_DIRECTORY)
1847             {
1848
1849                 AFSFsRtlNotifyFullReportChange( (*ppObjectInfo),
1850                                                 NULL,
1851                                                 ulFilter,
1852                                                 FILE_ACTION_MODIFIED);
1853             }
1854             else
1855             {
1856
1857                 AFSFsRtlNotifyFullReportChange( (*ppObjectInfo)->ParentObjectInformation,
1858                                                 NULL,
1859                                                 ulFilter,
1860                                                 FILE_ACTION_MODIFIED);
1861             }
1862
1863             //
1864             // Indicate this node requires re-evaluation for the remaining reasons
1865             //
1866
1867             (*ppObjectInfo)->Expiration.QuadPart = 0;
1868
1869             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1870                           AFS_TRACE_LEVEL_VERBOSE,
1871                           "AFSInvalidateObject Setting VERIFY flag on fid %08lX-%08lX-%08lX-%08lX\n",
1872                           (*ppObjectInfo)->FileId.Cell,
1873                           (*ppObjectInfo)->FileId.Volume,
1874                           (*ppObjectInfo)->FileId.Vnode,
1875                           (*ppObjectInfo)->FileId.Unique);
1876
1877             SetFlag( (*ppObjectInfo)->Flags, AFS_OBJECT_FLAGS_VERIFY);
1878
1879             if( Reason == AFS_INVALIDATE_DATA_VERSION ||
1880                 (*ppObjectInfo)->FileType == AFS_FILE_TYPE_FILE &&
1881                 ( Reason == AFS_INVALIDATE_CALLBACK ||
1882                   Reason == AFS_INVALIDATE_EXPIRED))
1883             {
1884                 if ( NT_SUCCESS( AFSQueueInvalidateObject( (*ppObjectInfo),
1885                                                            AFS_INVALIDATE_DATA_VERSION)))
1886                 {
1887
1888                     (*ppObjectInfo) = NULL; // We'll dec the count in the worker item
1889                 }
1890             }
1891
1892             break;
1893         }
1894     }
1895
1896   try_exit:
1897
1898     return ntStatus;
1899 }
1900
1901 NTSTATUS
1902 AFSInvalidateCache( IN AFSInvalidateCacheCB *InvalidateCB)
1903 {
1904
1905     NTSTATUS ntStatus = STATUS_SUCCESS;
1906     AFSFcb      *pDcb = NULL, *pFcb = NULL, *pNextFcb = NULL;
1907     AFSVolumeCB *pVolumeCB = NULL;
1908     AFSFcb      *pTargetDcb = NULL;
1909     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
1910     AFSDirectoryCB *pCurrentDirEntry = NULL;
1911     BOOLEAN     bIsChild = FALSE;
1912     ULONGLONG   ullIndex = 0;
1913     AFSObjectInfoCB *pObjectInfo = NULL;
1914     LONG lCount;
1915
1916     __Enter
1917     {
1918
1919         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1920                       AFS_TRACE_LEVEL_VERBOSE,
1921                       "AFSInvalidateCache Invalidation FID %08lX-%08lX-%08lX-%08lX Type %d WholeVolume %d Reason %d\n",
1922                       InvalidateCB->FileID.Cell,
1923                       InvalidateCB->FileID.Volume,
1924                       InvalidateCB->FileID.Vnode,
1925                       InvalidateCB->FileID.Unique,
1926                       InvalidateCB->FileType,
1927                       InvalidateCB->WholeVolume,
1928                       InvalidateCB->Reason);
1929
1930         //
1931         // Need to locate the Fcb for the directory to purge
1932         //
1933
1934         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
1935                       AFS_TRACE_LEVEL_VERBOSE,
1936                       "AFSInvalidateCache Acquiring RDR VolumeTreeLock lock %08lX SHARED %08lX\n",
1937                       &pDevExt->Specific.RDR.VolumeTreeLock,
1938                       PsGetCurrentThread());
1939
1940         //
1941         // Starve any exclusive waiters on this paticular call
1942         //
1943
1944         AFSAcquireSharedStarveExclusive( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
1945
1946         //
1947         // Locate the volume node
1948         //
1949
1950         ullIndex = AFSCreateHighIndex( &InvalidateCB->FileID);
1951
1952         ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
1953                                        ullIndex,
1954                                        (AFSBTreeEntry **)&pVolumeCB);
1955
1956         if( pVolumeCB != NULL)
1957         {
1958
1959             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
1960
1961             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
1962                           AFS_TRACE_LEVEL_VERBOSE,
1963                           "AFSInvalidateCache Increment count on volume %08lX Cnt %d\n",
1964                           pVolumeCB,
1965                           lCount);
1966         }
1967
1968         AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
1969
1970         if( !NT_SUCCESS( ntStatus) ||
1971             pVolumeCB == NULL)
1972         {
1973
1974             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1975                           AFS_TRACE_LEVEL_WARNING,
1976                           "AFSInvalidateCache Invalidation FAILURE Unable to locate volume node FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
1977                           InvalidateCB->FileID.Cell,
1978                           InvalidateCB->FileID.Volume,
1979                           InvalidateCB->FileID.Vnode,
1980                           InvalidateCB->FileID.Unique,
1981                           ntStatus);
1982
1983             try_return( ntStatus = STATUS_SUCCESS);
1984         }
1985
1986         //
1987         // If this is a whole volume invalidation then go do it now
1988         //
1989
1990         if( InvalidateCB->WholeVolume)
1991         {
1992
1993             ntStatus = AFSInvalidateVolume( pVolumeCB,
1994                                             InvalidateCB->Reason);
1995
1996             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
1997
1998             try_return( ntStatus);
1999         }
2000
2001         AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
2002                           TRUE);
2003
2004         if ( AFSIsVolumeFID( &InvalidateCB->FileID))
2005         {
2006
2007             pObjectInfo = &pVolumeCB->ObjectInformation;
2008         }
2009         else
2010         {
2011
2012             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
2013
2014             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
2015                           AFS_TRACE_LEVEL_VERBOSE,
2016                           "AFSInvalidateCache Decrement count on volume %08lX Cnt %d\n",
2017                           pVolumeCB,
2018                           pVolumeCB->VolumeReferenceCount);
2019
2020             ullIndex = AFSCreateLowIndex( &InvalidateCB->FileID);
2021
2022             ntStatus = AFSLocateHashEntry( pVolumeCB->ObjectInfoTree.TreeHead,
2023                                            ullIndex,
2024                                            (AFSBTreeEntry **)&pObjectInfo);
2025         }
2026
2027         if( pObjectInfo != NULL)
2028         {
2029
2030             //
2031             // Reference the node so it won't be torn down
2032             //
2033
2034             lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
2035
2036             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2037                           AFS_TRACE_LEVEL_VERBOSE,
2038                           "AFSInvalidateCache Increment count on object %08lX Cnt %d\n",
2039                           pObjectInfo,
2040                           lCount);
2041         }
2042
2043         AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
2044
2045         if( !NT_SUCCESS( ntStatus) ||
2046             pObjectInfo == NULL)
2047         {
2048
2049             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2050                           AFS_TRACE_LEVEL_VERBOSE,
2051                           "AFSInvalidateCache Invalidation FAILURE Unable to locate object FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
2052                           InvalidateCB->FileID.Cell,
2053                           InvalidateCB->FileID.Volume,
2054                           InvalidateCB->FileID.Vnode,
2055                           InvalidateCB->FileID.Unique,
2056                           ntStatus);
2057
2058             try_return( ntStatus = STATUS_SUCCESS);
2059         }
2060
2061         AFSInvalidateObject( &pObjectInfo,
2062                              InvalidateCB->Reason);
2063
2064 try_exit:
2065
2066         if( pObjectInfo != NULL)
2067         {
2068
2069             lCount = InterlockedDecrement( &pObjectInfo->ObjectReferenceCount);
2070
2071             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2072                           AFS_TRACE_LEVEL_VERBOSE,
2073                           "AFSInvalidateCache Decrement count on object %08lX Cnt %d\n",
2074                           pObjectInfo,
2075                           lCount);
2076         }
2077     }
2078
2079     return ntStatus;
2080 }
2081
2082 BOOLEAN
2083 AFSIsChildOfParent( IN AFSFcb *Dcb,
2084                     IN AFSFcb *Fcb)
2085 {
2086
2087     BOOLEAN bIsChild = FALSE;
2088     AFSFcb *pCurrentFcb = Fcb;
2089
2090     while( pCurrentFcb != NULL)
2091     {
2092
2093         if( pCurrentFcb->ObjectInformation->ParentObjectInformation == Dcb->ObjectInformation)
2094         {
2095
2096             bIsChild = TRUE;
2097
2098             break;
2099         }
2100
2101         pCurrentFcb = pCurrentFcb->ObjectInformation->ParentObjectInformation->Fcb;
2102     }
2103
2104     return bIsChild;
2105 }
2106
2107 inline
2108 ULONGLONG
2109 AFSCreateHighIndex( IN AFSFileID *FileID)
2110 {
2111
2112     ULONGLONG ullIndex = 0;
2113
2114     ullIndex = (((ULONGLONG)FileID->Cell << 32) | FileID->Volume);
2115
2116     return ullIndex;
2117 }
2118
2119 inline
2120 ULONGLONG
2121 AFSCreateLowIndex( IN AFSFileID *FileID)
2122 {
2123
2124     ULONGLONG ullIndex = 0;
2125
2126     ullIndex = (((ULONGLONG)FileID->Vnode << 32) | FileID->Unique);
2127
2128     return ullIndex;
2129 }
2130
2131 BOOLEAN
2132 AFSCheckAccess( IN ACCESS_MASK DesiredAccess,
2133                 IN ACCESS_MASK GrantedAccess,
2134                 IN BOOLEAN DirectoryEntry)
2135 {
2136
2137     BOOLEAN bAccessGranted = TRUE;
2138
2139     //
2140     // Check if we are asking for read/write and granted only read only
2141     // NOTE: There will be more checks here
2142     //
2143
2144     if( !AFSCheckForReadOnlyAccess( DesiredAccess,
2145                                     DirectoryEntry) &&
2146         AFSCheckForReadOnlyAccess( GrantedAccess,
2147                                    DirectoryEntry))
2148     {
2149
2150         bAccessGranted = FALSE;
2151     }
2152
2153     return bAccessGranted;
2154 }
2155
2156 NTSTATUS
2157 AFSGetDriverStatus( IN AFSDriverStatusRespCB *DriverStatus)
2158 {
2159
2160     NTSTATUS         ntStatus = STATUS_SUCCESS;
2161     AFSDeviceExt    *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
2162
2163     //
2164     // Start with read
2165     //
2166
2167     DriverStatus->Status = AFS_DRIVER_STATUS_READY;
2168
2169     if( AFSGlobalRoot == NULL)
2170     {
2171
2172         //
2173         // We are not ready
2174         //
2175
2176         DriverStatus->Status = AFS_DRIVER_STATUS_NOT_READY;
2177     }
2178
2179     if( pControlDevExt->Specific.Control.CommServiceCB.IrpPoolControlFlag != POOL_ACTIVE)
2180     {
2181
2182         //
2183         // No service yet
2184         //
2185
2186         DriverStatus->Status = AFS_DRIVER_STATUS_NO_SERVICE;
2187     }
2188
2189     return ntStatus;
2190 }
2191
2192 NTSTATUS
2193 AFSSubstituteSysName( IN UNICODE_STRING *ComponentName,
2194                       IN UNICODE_STRING *SubstituteName,
2195                       IN ULONG StringIndex)
2196 {
2197
2198     NTSTATUS ntStatus = STATUS_SUCCESS;
2199     AFSDeviceExt    *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
2200     AFSSysNameCB    *pSysName = NULL;
2201     ERESOURCE       *pSysNameLock = NULL;
2202     ULONG            ulIndex = 1;
2203     USHORT           usIndex = 0;
2204     UNICODE_STRING   uniSysName;
2205
2206     __Enter
2207     {
2208
2209 #if defined(_WIN64)
2210
2211         if( IoIs32bitProcess( NULL))
2212         {
2213
2214             pSysNameLock = &pControlDevExt->Specific.Control.SysName32ListLock;
2215
2216             pSysName = pControlDevExt->Specific.Control.SysName32ListHead;
2217         }
2218         else
2219         {
2220
2221             pSysNameLock = &pControlDevExt->Specific.Control.SysName64ListLock;
2222
2223             pSysName = pControlDevExt->Specific.Control.SysName64ListHead;
2224         }
2225 #else
2226
2227         pSysNameLock = &pControlDevExt->Specific.Control.SysName32ListLock;
2228
2229         pSysName = pControlDevExt->Specific.Control.SysName32ListHead;
2230
2231 #endif
2232
2233         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2234                       AFS_TRACE_LEVEL_VERBOSE,
2235                       "AFSSubstituteSysName Acquiring SysName lock %08lX SHARED %08lX\n",
2236                       pSysNameLock,
2237                       PsGetCurrentThread());
2238
2239         AFSAcquireShared( pSysNameLock,
2240                           TRUE);
2241
2242         //
2243         // Find where we are in the list
2244         //
2245
2246         while( pSysName != NULL &&
2247             ulIndex < StringIndex)
2248         {
2249
2250             pSysName = pSysName->fLink;
2251
2252             ulIndex++;
2253         }
2254
2255         if( pSysName == NULL)
2256         {
2257
2258             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
2259         }
2260
2261         RtlInitUnicodeString( &uniSysName,
2262                               L"@SYS");
2263         //
2264         // If it is a full component of @SYS then just substitue the
2265         // name in
2266         //
2267
2268         if( RtlCompareUnicodeString( &uniSysName,
2269                                      ComponentName,
2270                                      TRUE) == 0)
2271         {
2272
2273             SubstituteName->Length = pSysName->SysName.Length;
2274             SubstituteName->MaximumLength = SubstituteName->Length;
2275
2276             SubstituteName->Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2277                                                                         SubstituteName->Length,
2278                                                                         AFS_SUBST_BUFFER_TAG);
2279
2280             if( SubstituteName->Buffer == NULL)
2281             {
2282
2283                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2284             }
2285
2286             RtlCopyMemory( SubstituteName->Buffer,
2287                            pSysName->SysName.Buffer,
2288                            pSysName->SysName.Length);
2289         }
2290         else
2291         {
2292
2293             usIndex = 0;
2294
2295             while( ComponentName->Buffer[ usIndex] != L'@')
2296             {
2297
2298                 usIndex++;
2299             }
2300
2301             SubstituteName->Length = (usIndex * sizeof( WCHAR)) + pSysName->SysName.Length;
2302             SubstituteName->MaximumLength = SubstituteName->Length;
2303
2304             SubstituteName->Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2305                                                                         SubstituteName->Length,
2306                                                                         AFS_SUBST_BUFFER_TAG);
2307
2308             if( SubstituteName->Buffer == NULL)
2309             {
2310
2311                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2312             }
2313
2314             RtlCopyMemory( SubstituteName->Buffer,
2315                            ComponentName->Buffer,
2316                            usIndex * sizeof( WCHAR));
2317
2318             RtlCopyMemory( &SubstituteName->Buffer[ usIndex],
2319                            pSysName->SysName.Buffer,
2320                            pSysName->SysName.Length);
2321         }
2322
2323 try_exit:
2324
2325         AFSReleaseResource( pSysNameLock);
2326     }
2327
2328     return ntStatus;
2329 }
2330
2331 NTSTATUS
2332 AFSSubstituteNameInPath( IN OUT UNICODE_STRING *FullPathName,
2333                          IN OUT UNICODE_STRING *ComponentName,
2334                          IN UNICODE_STRING *SubstituteName,
2335                          IN OUT UNICODE_STRING *RemainingPath,
2336                          IN BOOLEAN FreePathName)
2337 {
2338
2339     NTSTATUS ntStatus = STATUS_SUCCESS;
2340     UNICODE_STRING uniPathName;
2341     USHORT usPrefixNameLen = 0;
2342     SHORT  sNameLenDelta = 0;
2343
2344     __Enter
2345     {
2346
2347         //
2348         // If the passed in name can handle the additional length
2349         // then just moves things around
2350         //
2351
2352         sNameLenDelta = SubstituteName->Length - ComponentName->Length;
2353
2354         usPrefixNameLen = (USHORT)(ComponentName->Buffer - FullPathName->Buffer);
2355
2356         if( FullPathName->MaximumLength > FullPathName->Length + sNameLenDelta)
2357         {
2358
2359             if( FullPathName->Length > usPrefixNameLen + ComponentName->Length)
2360             {
2361
2362                 RtlMoveMemory( &FullPathName->Buffer[ ((usPrefixNameLen*sizeof( WCHAR) + SubstituteName->Length)/sizeof( WCHAR))],
2363                                &FullPathName->Buffer[ ((usPrefixNameLen*sizeof( WCHAR) + ComponentName->Length)/sizeof( WCHAR))],
2364                                FullPathName->Length - usPrefixNameLen*sizeof( WCHAR) - ComponentName->Length);
2365             }
2366
2367             RtlCopyMemory( &FullPathName->Buffer[ usPrefixNameLen],
2368                            SubstituteName->Buffer,
2369                            SubstituteName->Length);
2370
2371             FullPathName->Length += sNameLenDelta;
2372
2373             ComponentName->Length += sNameLenDelta;
2374
2375             ComponentName->MaximumLength = ComponentName->Length;
2376
2377             if ( RemainingPath->Buffer)
2378             {
2379
2380                 RemainingPath->Buffer += sNameLenDelta/sizeof( WCHAR);
2381             }
2382
2383             try_return( ntStatus);
2384         }
2385
2386         //
2387         // Need to re-allocate the buffer
2388         //
2389
2390         uniPathName.Length = FullPathName->Length -
2391                                          ComponentName->Length +
2392                                          SubstituteName->Length;
2393
2394         uniPathName.MaximumLength = FullPathName->MaximumLength + PAGE_SIZE;
2395
2396         uniPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2397                                                                 uniPathName.MaximumLength,
2398                                                                 AFS_NAME_BUFFER_FOUR_TAG);
2399
2400         if( uniPathName.Buffer == NULL)
2401         {
2402
2403             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2404         }
2405
2406         usPrefixNameLen = (USHORT)(ComponentName->Buffer - FullPathName->Buffer);
2407
2408         usPrefixNameLen *= sizeof( WCHAR);
2409
2410         RtlZeroMemory( uniPathName.Buffer,
2411                        uniPathName.MaximumLength);
2412
2413         RtlCopyMemory( uniPathName.Buffer,
2414                        FullPathName->Buffer,
2415                        usPrefixNameLen);
2416
2417         RtlCopyMemory( &uniPathName.Buffer[ (usPrefixNameLen/sizeof( WCHAR))],
2418                        SubstituteName->Buffer,
2419                        SubstituteName->Length);
2420
2421         if( FullPathName->Length > usPrefixNameLen + ComponentName->Length)
2422         {
2423
2424             RtlCopyMemory( &uniPathName.Buffer[ (usPrefixNameLen + SubstituteName->Length)/sizeof( WCHAR)],
2425                            &FullPathName->Buffer[ (usPrefixNameLen + ComponentName->Length)/sizeof( WCHAR)],
2426                            FullPathName->Length - usPrefixNameLen - ComponentName->Length);
2427         }
2428
2429         ComponentName->Buffer = uniPathName.Buffer + (ComponentName->Buffer - FullPathName->Buffer);
2430
2431         ComponentName->Length += sNameLenDelta;
2432
2433         ComponentName->MaximumLength = ComponentName->Length;
2434
2435         if ( RemainingPath->Buffer)
2436         {
2437
2438             RemainingPath->Buffer = uniPathName.Buffer
2439                 + (RemainingPath->Buffer - FullPathName->Buffer)
2440                 + sNameLenDelta/sizeof( WCHAR);
2441         }
2442
2443         if( FreePathName)
2444         {
2445             AFSExFreePool( FullPathName->Buffer);
2446         }
2447
2448         *FullPathName = uniPathName;
2449
2450 try_exit:
2451
2452         NOTHING;
2453     }
2454
2455     return ntStatus;
2456 }
2457
2458 NTSTATUS
2459 AFSInvalidateVolume( IN AFSVolumeCB *VolumeCB,
2460                      IN ULONG Reason)
2461 {
2462
2463     NTSTATUS ntStatus = STATUS_SUCCESS;
2464     AFSObjectInfoCB *pCurrentObject = NULL;
2465     AFSObjectInfoCB *pNextObject = NULL;
2466     LONG lCount;
2467     AFSFcb *pFcb = NULL;
2468     ULONG ulFilter = 0;
2469
2470     __Enter
2471     {
2472
2473         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2474                       AFS_TRACE_LEVEL_VERBOSE,
2475                       "AFSInvalidateVolume Invalidate volume fid %08lX-%08lX-%08lX-%08lX Reason %08lX\n",
2476                       VolumeCB->ObjectInformation.FileId.Cell,
2477                       VolumeCB->ObjectInformation.FileId.Volume,
2478                       VolumeCB->ObjectInformation.FileId.Vnode,
2479                       VolumeCB->ObjectInformation.FileId.Unique,
2480                       Reason);
2481
2482         //
2483         // Depending on the reason for invalidation then perform work on the node
2484         //
2485
2486         switch( Reason)
2487         {
2488
2489             case AFS_INVALIDATE_DELETED:
2490             {
2491
2492                 //
2493                 // Mark this volume as invalid
2494                 //
2495
2496                 SetFlag( VolumeCB->ObjectInformation.Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
2497
2498                 SetFlag( VolumeCB->Flags, AFS_VOLUME_FLAGS_OFFLINE);
2499
2500                 break;
2501             }
2502         }
2503
2504         //
2505         // Invalidate the volume root directory
2506         //
2507
2508         pCurrentObject = &VolumeCB->ObjectInformation;
2509
2510         if ( pCurrentObject )
2511         {
2512
2513             lCount = InterlockedIncrement( &pCurrentObject->ObjectReferenceCount);
2514
2515             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2516                           AFS_TRACE_LEVEL_VERBOSE,
2517                           "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2518                           pCurrentObject,
2519                           lCount);
2520
2521             AFSInvalidateObject( &pCurrentObject,
2522                                  Reason);
2523
2524             if ( pCurrentObject)
2525             {
2526
2527                 lCount = InterlockedDecrement( &pCurrentObject->ObjectReferenceCount);
2528
2529                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2530                               AFS_TRACE_LEVEL_VERBOSE,
2531                               "AFSInvalidateVolumeObjects Decrement count on object %08lX Cnt %d\n",
2532                               pCurrentObject,
2533                               lCount);
2534             }
2535         }
2536
2537         //
2538         // Apply invalidation to all other volume objects
2539         //
2540
2541         AFSAcquireShared( VolumeCB->ObjectInfoTree.TreeLock,
2542                           TRUE);
2543
2544         pCurrentObject = VolumeCB->ObjectInfoListHead;
2545
2546         if ( pCurrentObject)
2547         {
2548
2549             //
2550             // Reference the node so it won't be torn down
2551             //
2552
2553             lCount = InterlockedIncrement( &pCurrentObject->ObjectReferenceCount);
2554
2555             AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2556                           AFS_TRACE_LEVEL_VERBOSE,
2557                           "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2558                           pCurrentObject,
2559                           lCount);
2560         }
2561
2562         while( pCurrentObject != NULL)
2563         {
2564
2565             pNextObject = (AFSObjectInfoCB *)pCurrentObject->ListEntry.fLink;
2566
2567             if ( pNextObject)
2568             {
2569
2570                 //
2571                 // Reference the node so it won't be torn down
2572                 //
2573
2574                 lCount = InterlockedIncrement( &pNextObject->ObjectReferenceCount);
2575
2576                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2577                               AFS_TRACE_LEVEL_VERBOSE,
2578                               "AFSInvalidateVolumeObjects Increment count on object %08lX Cnt %d\n",
2579                               pNextObject,
2580                               lCount);
2581             }
2582
2583             AFSReleaseResource( VolumeCB->ObjectInfoTree.TreeLock);
2584
2585             AFSInvalidateObject( &pCurrentObject,
2586                                  Reason);
2587
2588             if ( pCurrentObject )
2589             {
2590
2591                 lCount = InterlockedDecrement( &pCurrentObject->ObjectReferenceCount);
2592
2593                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2594                               AFS_TRACE_LEVEL_VERBOSE,
2595                               "AFSInvalidateVolumeObjects Decrement count on object %08lX Cnt %d\n",
2596                               pCurrentObject,
2597                               lCount);
2598             }
2599
2600             AFSAcquireShared( VolumeCB->ObjectInfoTree.TreeLock,
2601                               TRUE);
2602
2603             pCurrentObject = pNextObject;
2604         }
2605
2606         AFSReleaseResource( VolumeCB->ObjectInfoTree.TreeLock);
2607     }
2608
2609     return ntStatus;
2610 }
2611
2612 VOID
2613 AFSInvalidateAllVolumes( VOID)
2614 {
2615     AFSVolumeCB *pVolumeCB = NULL;
2616     AFSVolumeCB *pNextVolumeCB = NULL;
2617     AFSDeviceExt *pRDRDeviceExt = NULL;
2618     LONG lCount;
2619
2620     pRDRDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
2621
2622     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2623                   AFS_TRACE_LEVEL_VERBOSE,
2624                   "AFSInvalidateAllVolumes Acquiring RDR VolumeListLock lock %08lX SHARED %08lX\n",
2625                   &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2626                   PsGetCurrentThread());
2627
2628     AFSAcquireShared( &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2629                       TRUE);
2630
2631     pVolumeCB = pRDRDeviceExt->Specific.RDR.VolumeListHead;
2632
2633     if ( pVolumeCB)
2634     {
2635
2636         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2637                       AFS_TRACE_LEVEL_VERBOSE,
2638                       "AFSInvalidateAllVolumes Acquiring VolumeRoot ObjectInfoTree lock %08lX SHARED %08lX\n",
2639                       pVolumeCB->ObjectInfoTree.TreeLock,
2640                       PsGetCurrentThread());
2641
2642         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
2643     }
2644
2645     while( pVolumeCB != NULL)
2646     {
2647
2648         pNextVolumeCB = (AFSVolumeCB *)pVolumeCB->ListEntry.fLink;
2649
2650         if ( pNextVolumeCB)
2651         {
2652
2653             lCount = InterlockedIncrement( &pNextVolumeCB->VolumeReferenceCount);
2654         }
2655
2656         AFSReleaseResource( &pRDRDeviceExt->Specific.RDR.VolumeListLock);
2657
2658         // do I need to hold the volume lock here?
2659
2660         AFSInvalidateVolume( pVolumeCB, AFS_INVALIDATE_EXPIRED);
2661
2662         AFSAcquireShared( &pRDRDeviceExt->Specific.RDR.VolumeListLock,
2663                           TRUE);
2664
2665         lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
2666
2667         pVolumeCB = pNextVolumeCB;
2668     }
2669
2670     AFSReleaseResource( &pRDRDeviceExt->Specific.RDR.VolumeListLock);
2671 }
2672
2673 NTSTATUS
2674 AFSVerifyEntry( IN GUID *AuthGroup,
2675                 IN AFSDirectoryCB *DirEntry)
2676 {
2677
2678     NTSTATUS ntStatus = STATUS_SUCCESS;
2679     AFSDirEnumEntry *pDirEnumEntry = NULL;
2680     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
2681     IO_STATUS_BLOCK stIoStatus;
2682
2683     __Enter
2684     {
2685
2686         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2687                       AFS_TRACE_LEVEL_VERBOSE_2,
2688                       "AFSVerifyEntry Verifying entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2689                       &DirEntry->NameInformation.FileName,
2690                       pObjectInfo->FileId.Cell,
2691                       pObjectInfo->FileId.Volume,
2692                       pObjectInfo->FileId.Vnode,
2693                       pObjectInfo->FileId.Unique);
2694
2695         ntStatus = AFSEvaluateTargetByID( pObjectInfo,
2696                                           AuthGroup,
2697                                           FALSE,
2698                                           &pDirEnumEntry);
2699
2700         if( !NT_SUCCESS( ntStatus))
2701         {
2702
2703             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2704                           AFS_TRACE_LEVEL_ERROR,
2705                           "AFSVerifyEntry Evaluate Target failed %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
2706                           &DirEntry->NameInformation.FileName,
2707                           pObjectInfo->FileId.Cell,
2708                           pObjectInfo->FileId.Volume,
2709                           pObjectInfo->FileId.Vnode,
2710                           pObjectInfo->FileId.Unique,
2711                           ntStatus);
2712
2713             try_return( ntStatus);
2714         }
2715
2716         //
2717         // Check the data version of the file
2718         //
2719
2720         if( pObjectInfo->DataVersion.QuadPart == pDirEnumEntry->DataVersion.QuadPart)
2721         {
2722             if ( !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA))
2723             {
2724
2725                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2726                               AFS_TRACE_LEVEL_VERBOSE,
2727                               "AFSVerifyEntry No DV change %I64X for Fcb %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2728                               pObjectInfo->DataVersion.QuadPart,
2729                               &DirEntry->NameInformation.FileName,
2730                               pObjectInfo->FileId.Cell,
2731                               pObjectInfo->FileId.Volume,
2732                               pObjectInfo->FileId.Vnode,
2733                               pObjectInfo->FileId.Unique);
2734
2735                 //
2736                 // We are ok, just get out
2737                 //
2738
2739                 ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2740
2741                 try_return( ntStatus = STATUS_SUCCESS);
2742             }
2743         }
2744
2745         //
2746         // New data version so we will need to process the node based on the type
2747         //
2748
2749         switch( pDirEnumEntry->FileType)
2750         {
2751
2752             case AFS_FILE_TYPE_MOUNTPOINT:
2753             {
2754
2755                 //
2756                 // For a mount point we need to ensure the target is the same
2757                 //
2758
2759                 if( !AFSIsEqualFID( &pObjectInfo->TargetFileId,
2760                                     &pDirEnumEntry->TargetFileId))
2761                 {
2762
2763                 }
2764
2765                 //
2766                 // Update the metadata for the entry
2767                 //
2768
2769                 ntStatus = AFSUpdateMetaData( DirEntry,
2770                                               pDirEnumEntry);
2771
2772                 if( NT_SUCCESS( ntStatus))
2773                 {
2774
2775                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2776                 }
2777
2778                 break;
2779             }
2780
2781             case AFS_FILE_TYPE_SYMLINK:
2782             {
2783
2784                 ASSERT( pDirEnumEntry->TargetNameLength > 0);
2785
2786                 //
2787                 // Update the metadata for the entry
2788                 //
2789
2790                 ntStatus = AFSUpdateMetaData( DirEntry,
2791                                               pDirEnumEntry);
2792
2793                 if( NT_SUCCESS( ntStatus))
2794                 {
2795
2796                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
2797                 }
2798
2799                 break;
2800             }
2801
2802             case AFS_FILE_TYPE_FILE:
2803             {
2804                 FILE_OBJECT * pCCFileObject = NULL;
2805                 BOOLEAN bPurgeExtents = FALSE;
2806
2807                 if ( pObjectInfo->DataVersion.QuadPart != pDirEnumEntry->DataVersion.QuadPart)
2808                 {
2809
2810                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2811                                   AFS_TRACE_LEVEL_VERBOSE,
2812                                   "AFSVerifyEntry DV Change %wZ FID %08lX-%08lX-%08lX-%08lX (%08lX != %08lX)\n",
2813                                   &DirEntry->NameInformation.FileName,
2814                                   pObjectInfo->FileId.Cell,
2815                                   pObjectInfo->FileId.Volume,
2816                                   pObjectInfo->FileId.Vnode,
2817                                   pObjectInfo->FileId.Unique,
2818                                   pObjectInfo->DataVersion.LowPart,
2819                                   pDirEnumEntry->DataVersion.LowPart
2820                                   );
2821
2822                     bPurgeExtents = TRUE;
2823                 }
2824
2825                 if ( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA))
2826                 {
2827
2828                     bPurgeExtents = TRUE;
2829
2830                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2831                                   AFS_TRACE_LEVEL_VERBOSE,
2832                                   "AFSVerifyEntry Clearing VERIFY_DATA flag %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2833                                   &DirEntry->NameInformation.FileName,
2834                                   pObjectInfo->FileId.Cell,
2835                                   pObjectInfo->FileId.Volume,
2836                                   pObjectInfo->FileId.Vnode,
2837                                   pObjectInfo->FileId.Unique);
2838
2839                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
2840                 }
2841
2842                 if( pObjectInfo->Fcb != NULL)
2843                 {
2844
2845                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2846                                   AFS_TRACE_LEVEL_VERBOSE,
2847                                   "AFSVerifyEntry Flush/purge entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2848                                   &DirEntry->NameInformation.FileName,
2849                                   pObjectInfo->FileId.Cell,
2850                                   pObjectInfo->FileId.Volume,
2851                                   pObjectInfo->FileId.Vnode,
2852                                   pObjectInfo->FileId.Unique);
2853
2854                     AFSAcquireExcl( &pObjectInfo->Fcb->NPFcb->Resource,
2855                                     TRUE);
2856
2857                     __try
2858                     {
2859
2860                         CcFlushCache( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers,
2861                                       NULL,
2862                                       0,
2863                                       &stIoStatus);
2864
2865                         if( !NT_SUCCESS( stIoStatus.Status))
2866                         {
2867
2868                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
2869                                           AFS_TRACE_LEVEL_ERROR,
2870                                           "AFSVerifyEntry CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
2871                                           &DirEntry->NameInformation.FileName,
2872                                           pObjectInfo->FileId.Cell,
2873                                           pObjectInfo->FileId.Volume,
2874                                           pObjectInfo->FileId.Vnode,
2875                                           pObjectInfo->FileId.Unique,
2876                                           stIoStatus.Status,
2877                                           stIoStatus.Information);
2878
2879                             ntStatus = stIoStatus.Status;
2880                         }
2881
2882                         if ( bPurgeExtents)
2883                         {
2884
2885                             if ( !CcPurgeCacheSection( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers,
2886                                                        NULL,
2887                                                        0,
2888                                                        FALSE))
2889                             {
2890
2891                                 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
2892                                               AFS_TRACE_LEVEL_WARNING,
2893                                               "AFSVerifyEntry CcPurgeCacheSection failure %wZ FID %08lX-%08lX-%08lX-%08lX\n",
2894                                               &DirEntry->NameInformation.FileName,
2895                                               pObjectInfo->FileId.Cell,
2896                                               pObjectInfo->FileId.Volume,
2897                                               pObjectInfo->FileId.Vnode,
2898                                               pObjectInfo->FileId.Unique);
2899
2900                                 SetFlag( pObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
2901                             }
2902                         }
2903                     }
2904                     __except( EXCEPTION_EXECUTE_HANDLER)
2905                     {
2906                         ntStatus = GetExceptionCode();
2907
2908                         AFSDbgLogMsg( 0,
2909                                       0,
2910                                       "EXCEPTION - AFSVerifyEntry CcFlushCache or CcPurgeCacheSection %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
2911                                       &DirEntry->NameInformation.FileName,
2912                                       pObjectInfo->FileId.Cell,
2913                                       pObjectInfo->FileId.Volume,
2914                                       pObjectInfo->FileId.Vnode,
2915                                       pObjectInfo->FileId.Unique,
2916                                       ntStatus);
2917
2918                         SetFlag( pObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
2919                     }
2920
2921                     AFSReleaseResource( &pObjectInfo->Fcb->NPFcb->Resource);
2922
2923                     if ( bPurgeExtents)
2924                     {
2925                         AFSFlushExtents( pObjectInfo->Fcb,
2926                                          AuthGroup);
2927                     }
2928
2929                     //
2930                     // Reacquire the Fcb to purge the cache
2931                     //
2932
2933                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2934                                   AFS_TRACE_LEVEL_VERBOSE,
2935                                   "AFSVerifyEntry Acquiring Fcb lock %08lX EXCL %08lX\n",
2936                                   &pObjectInfo->Fcb->NPFcb->Resource,
2937                                   PsGetCurrentThread());
2938
2939                     AFSAcquireExcl( &pObjectInfo->Fcb->NPFcb->Resource,
2940                                     TRUE);
2941
2942                     //
2943                     // Update the metadata for the entry
2944                     //
2945
2946                     ntStatus = AFSUpdateMetaData( DirEntry,
2947                                                   pDirEnumEntry);
2948
2949                     if( !NT_SUCCESS( ntStatus))
2950                     {
2951
2952                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2953                                       AFS_TRACE_LEVEL_ERROR,
2954                                       "AFSVerifyEntry Meta Data Update failed %wZ FID %08lX-%08lX-%08lX-%08lX ntStatus %08lX\n",
2955                                       &DirEntry->NameInformation.FileName,
2956                                       pObjectInfo->FileId.Cell,
2957                                       pObjectInfo->FileId.Volume,
2958                                       pObjectInfo->FileId.Vnode,
2959                                       pObjectInfo->FileId.Unique,
2960                                       ntStatus);
2961
2962                         break;
2963                     }
2964
2965                     //
2966                     // Update file sizes
2967                     //
2968
2969                     pObjectInfo->Fcb->Header.AllocationSize.QuadPart  = pObjectInfo->AllocationSize.QuadPart;
2970                     pObjectInfo->Fcb->Header.FileSize.QuadPart        = pObjectInfo->EndOfFile.QuadPart;
2971                     pObjectInfo->Fcb->Header.ValidDataLength.QuadPart = pObjectInfo->EndOfFile.QuadPart;
2972
2973                     pCCFileObject = CcGetFileObjectFromSectionPtrs( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers);
2974
2975                     if ( pCCFileObject != NULL)
2976                     {
2977                         CcSetFileSizes( pCCFileObject,
2978                                         (PCC_FILE_SIZES)&pObjectInfo->Fcb->Header.AllocationSize);
2979                     }
2980
2981                     AFSReleaseResource( &pObjectInfo->Fcb->NPFcb->Resource);
2982                 }
2983                 else
2984                 {
2985
2986                     //
2987                     // Update the metadata for the entry
2988                     //
2989
2990                     ntStatus = AFSUpdateMetaData( DirEntry,
2991                                                   pDirEnumEntry);
2992
2993                     if( !NT_SUCCESS( ntStatus))
2994                     {
2995
2996                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2997                                       AFS_TRACE_LEVEL_ERROR,
2998                                       "AFSVerifyEntry Meta Data Update failed %wZ FID %08lX-%08lX-%08lX-%08lX ntStatus %08lX\n",
2999                                       &DirEntry->NameInformation.FileName,
3000                                       pObjectInfo->FileId.Cell,
3001                                       pObjectInfo->FileId.Volume,
3002                                       pObjectInfo->FileId.Vnode,
3003                                       pObjectInfo->FileId.Unique,
3004                                       ntStatus);
3005
3006                         break;
3007                     }
3008
3009                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3010                                   AFS_TRACE_LEVEL_WARNING,
3011                                   "AFSVerifyEntry Fcb NULL %wZ FID %08lX-%08lX-%08lX-%08lX\n",
3012                                   &DirEntry->NameInformation.FileName,
3013                                   pObjectInfo->FileId.Cell,
3014                                   pObjectInfo->FileId.Volume,
3015                                   pObjectInfo->FileId.Vnode,
3016                                   pObjectInfo->FileId.Unique);
3017                 }
3018
3019                 ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
3020
3021                 break;
3022             }
3023
3024             case AFS_FILE_TYPE_DIRECTORY:
3025             {
3026
3027                 AFSFcb *pCurrentFcb = NULL;
3028                 AFSDirectoryCB *pCurrentDirEntry = NULL;
3029
3030                 //
3031                 // For a directory or root entry flush the content of
3032                 // the directory enumeration.
3033                 //
3034
3035                 if( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED))
3036                 {
3037
3038                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3039                                   AFS_TRACE_LEVEL_VERBOSE_2,
3040                                   "AFSVerifyEntry Validating directory content for entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
3041                                   &DirEntry->NameInformation.FileName,
3042                                   pObjectInfo->FileId.Cell,
3043                                   pObjectInfo->FileId.Volume,
3044                                   pObjectInfo->FileId.Vnode,
3045                                   pObjectInfo->FileId.Unique);
3046
3047                     AFSAcquireExcl( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
3048                                     TRUE);
3049
3050                     ntStatus = AFSValidateDirectoryCache( pObjectInfo,
3051                                                           AuthGroup);
3052
3053                     AFSReleaseResource( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock);
3054
3055                     if ( !NT_SUCCESS( ntStatus))
3056                     {
3057
3058                         try_return( ntStatus);
3059                     }
3060                 }
3061
3062                 //
3063                 // Update the metadata for the entry
3064                 //
3065
3066                 ntStatus = AFSUpdateMetaData( DirEntry,
3067                                               pDirEnumEntry);
3068
3069                 if( NT_SUCCESS( ntStatus))
3070                 {
3071
3072                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
3073                 }
3074
3075                 break;
3076             }
3077
3078             case AFS_FILE_TYPE_DFSLINK:
3079             {
3080
3081                 UNICODE_STRING uniTargetName;
3082
3083                 //
3084                 // For a DFS link need to check the target name has not changed
3085                 //
3086
3087                 uniTargetName.Length = (USHORT)pDirEnumEntry->TargetNameLength;
3088
3089                 uniTargetName.MaximumLength = uniTargetName.Length;
3090
3091                 uniTargetName.Buffer = (WCHAR *)((char *)pDirEnumEntry + pDirEnumEntry->TargetNameOffset);
3092
3093                 AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3094                                 TRUE);
3095
3096                 if( DirEntry->NameInformation.TargetName.Length == 0 ||
3097                     RtlCompareUnicodeString( &uniTargetName,
3098                                              &DirEntry->NameInformation.TargetName,
3099                                              TRUE) != 0)
3100                 {
3101
3102                     //
3103                     // Update the target name
3104                     //
3105
3106                     ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
3107                                                     &DirEntry->Flags,
3108                                                     uniTargetName.Buffer,
3109                                                     uniTargetName.Length);
3110
3111                     if( !NT_SUCCESS( ntStatus))
3112                     {
3113
3114                         AFSReleaseResource( &DirEntry->NonPaged->Lock);
3115
3116                         break;
3117                     }
3118                 }
3119
3120                 AFSReleaseResource( &DirEntry->NonPaged->Lock);
3121
3122                 //
3123                 // Update the metadata for the entry
3124                 //
3125
3126                 ntStatus = AFSUpdateMetaData( DirEntry,
3127                                               pDirEnumEntry);
3128
3129                 if( NT_SUCCESS( ntStatus))
3130                 {
3131
3132                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY);
3133                 }
3134
3135                 break;
3136             }
3137
3138             default:
3139
3140                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3141                               AFS_TRACE_LEVEL_WARNING,
3142                               "AFSVerifyEntry Attempt to verify node of type %d %wZ FID %08lX-%08lX-%08lX-%08lX\n",
3143                               pObjectInfo->FileType,
3144                               &DirEntry->NameInformation.FileName,
3145                               pObjectInfo->FileId.Cell,
3146                               pObjectInfo->FileId.Volume,
3147                               pObjectInfo->FileId.Vnode,
3148                               pObjectInfo->FileId.Unique);
3149
3150                 break;
3151         }
3152
3153  try_exit:
3154
3155         if( pDirEnumEntry != NULL)
3156         {
3157
3158             AFSExFreePool( pDirEnumEntry);
3159         }
3160     }
3161
3162     return ntStatus;
3163 }
3164
3165 NTSTATUS
3166 AFSSetVolumeState( IN AFSVolumeStatusCB *VolumeStatus)
3167 {
3168
3169     NTSTATUS ntStatus = STATUS_SUCCESS;
3170     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
3171     ULONGLONG   ullIndex = 0;
3172     AFSVolumeCB *pVolumeCB = NULL;
3173     AFSFcb *pFcb = NULL;
3174     AFSObjectInfoCB *pCurrentObject = NULL;
3175     LONG lCount;
3176
3177     __Enter
3178     {
3179
3180         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3181                       AFS_TRACE_LEVEL_VERBOSE,
3182                       "AFSSetVolumeState Marking volume state %d Volume Cell %08lX Volume %08lX\n",
3183                       VolumeStatus->Online,
3184                       VolumeStatus->FileID.Cell,
3185                       VolumeStatus->FileID.Volume);
3186
3187         //
3188         // Need to locate the Fcb for the directory to purge
3189         //
3190
3191         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3192                       AFS_TRACE_LEVEL_VERBOSE,
3193                       "AFSSetVolumeState Acquiring RDR VolumeTreeLock lock %08lX SHARED %08lX\n",
3194                       &pDevExt->Specific.RDR.VolumeTreeLock,
3195                       PsGetCurrentThread());
3196
3197         AFSAcquireShared( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
3198
3199         //
3200         // Locate the volume node
3201         //
3202
3203         ullIndex = AFSCreateHighIndex( &VolumeStatus->FileID);
3204
3205         ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
3206                                        ullIndex,
3207                                        (AFSBTreeEntry **)&pVolumeCB);
3208
3209         if( pVolumeCB != NULL)
3210         {
3211
3212             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
3213
3214             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
3215
3216             //
3217             // Set the volume state accordingly
3218             //
3219
3220             if( VolumeStatus->Online)
3221             {
3222
3223                 InterlockedAnd( (LONG *)&(pVolumeCB->Flags), ~AFS_VOLUME_FLAGS_OFFLINE);
3224             }
3225             else
3226             {
3227
3228                 InterlockedOr( (LONG *)&(pVolumeCB->Flags), AFS_VOLUME_FLAGS_OFFLINE);
3229             }
3230
3231             AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
3232                               TRUE);
3233
3234             pCurrentObject = pVolumeCB->ObjectInfoListHead;;
3235
3236             while( pCurrentObject != NULL)
3237             {
3238
3239                 if( VolumeStatus->Online)
3240                 {
3241
3242                     ClearFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
3243
3244                     SetFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_VERIFY);
3245
3246                     pCurrentObject->DataVersion.QuadPart = (ULONGLONG)-1;
3247                 }
3248                 else
3249                 {
3250
3251                     SetFlag( pCurrentObject->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID);
3252                 }
3253
3254                 pFcb = pCurrentObject->Fcb;
3255
3256                 if( pFcb != NULL &&
3257                     !(VolumeStatus->Online) &&
3258                     pFcb->Header.NodeTypeCode == AFS_FILE_FCB)
3259                 {
3260
3261                     AFSDbgLogMsg( AFS_SUBSYSTEM_EXTENT_PROCESSING,
3262                                   AFS_TRACE_LEVEL_ERROR,
3263                                   "AFSSetVolumeState Marking volume offline and canceling extents Volume Cell %08lX Volume %08lX\n",
3264                                   VolumeStatus->FileID.Cell,
3265                                   VolumeStatus->FileID.Volume);
3266
3267                     //
3268                     // Clear out the extents
3269                     //
3270
3271                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3272                                   AFS_TRACE_LEVEL_VERBOSE,
3273                                   "AFSSetVolumeState Acquiring Fcb extents lock %08lX EXCL %08lX\n",
3274                                   &pFcb->NPFcb->Specific.File.ExtentsResource,
3275                                   PsGetCurrentThread());
3276
3277                     AFSAcquireExcl( &pFcb->NPFcb->Specific.File.ExtentsResource,
3278                                     TRUE);
3279
3280                     pFcb->NPFcb->Specific.File.ExtentsRequestStatus = STATUS_CANCELLED;
3281
3282                     KeSetEvent( &pFcb->NPFcb->Specific.File.ExtentsRequestComplete,
3283                                 0,
3284                                 FALSE);
3285
3286                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3287                                   AFS_TRACE_LEVEL_VERBOSE,
3288                                   "AFSSetVolumeState Releasing Fcb extents lock %08lX EXCL %08lX\n",
3289                                   &pFcb->NPFcb->Specific.File.ExtentsResource,
3290                                   PsGetCurrentThread());
3291
3292                     AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource);
3293
3294                     //
3295                     // And get rid of them (note this involves waiting
3296                     // for any writes or reads to the cache to complete)
3297                     //
3298
3299                     AFSTearDownFcbExtents( pFcb,
3300                                            NULL);
3301                 }
3302
3303                 pCurrentObject = (AFSObjectInfoCB *)pCurrentObject->ListEntry.fLink;
3304             }
3305
3306             AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
3307
3308             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
3309         }
3310         else
3311         {
3312
3313             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
3314         }
3315     }
3316
3317     return ntStatus;
3318 }
3319
3320 NTSTATUS
3321 AFSSetNetworkState( IN AFSNetworkStatusCB *NetworkStatus)
3322 {
3323
3324     NTSTATUS ntStatus = STATUS_SUCCESS;
3325
3326     __Enter
3327     {
3328
3329         if( AFSGlobalRoot == NULL)
3330         {
3331
3332             try_return( ntStatus);
3333         }
3334
3335         AFSAcquireExcl( AFSGlobalRoot->VolumeLock,
3336                         TRUE);
3337
3338         //
3339         // Set the network state according to the information
3340         //
3341
3342         if( NetworkStatus->Online)
3343         {
3344
3345             ClearFlag( AFSGlobalRoot->Flags, AFS_VOLUME_FLAGS_OFFLINE);
3346         }
3347         else
3348         {
3349
3350             SetFlag( AFSGlobalRoot->Flags, AFS_VOLUME_FLAGS_OFFLINE);
3351         }
3352
3353         AFSReleaseResource( AFSGlobalRoot->VolumeLock);
3354
3355 try_exit:
3356
3357         NOTHING;
3358     }
3359
3360     return ntStatus;
3361 }
3362
3363 NTSTATUS
3364 AFSValidateDirectoryCache( IN AFSObjectInfoCB *ObjectInfo,
3365                            IN GUID *AuthGroup)
3366 {
3367
3368     NTSTATUS ntStatus = STATUS_SUCCESS;
3369     BOOLEAN  bAcquiredLock = FALSE;
3370     AFSDirectoryCB *pCurrentDirEntry = NULL, *pNextDirEntry = NULL;
3371     AFSFcb *pFcb = NULL;
3372
3373     __Enter
3374     {
3375
3376         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3377                       AFS_TRACE_LEVEL_VERBOSE,
3378                       "AFSValidateDirectoryCache Validating content for FID %08lX-%08lX-%08lX-%08lX\n",
3379                       ObjectInfo->FileId.Cell,
3380                       ObjectInfo->FileId.Volume,
3381                       ObjectInfo->FileId.Vnode,
3382                       ObjectInfo->FileId.Unique);
3383
3384         if( !ExIsResourceAcquiredLite( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock))
3385         {
3386
3387             AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3388                           AFS_TRACE_LEVEL_VERBOSE,
3389                           "AFSValidateDirectoryCache Acquiring DirectoryNodeHdr.TreeLock lock %08lX EXCL %08lX\n",
3390                           ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
3391                           PsGetCurrentThread());
3392
3393             AFSAcquireExcl( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
3394                             TRUE);
3395
3396             bAcquiredLock = TRUE;
3397         }
3398
3399         //
3400         // Check for inconsistency between DirectoryNodeList and DirectoryNodeCount
3401         //
3402
3403         if ( ObjectInfo->Specific.Directory.DirectoryNodeListHead == NULL &&
3404              ObjectInfo->Specific.Directory.DirectoryNodeCount > 0)
3405         {
3406
3407             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3408                           AFS_TRACE_LEVEL_ERROR,
3409                           "AFSValidateDirectoryCache Empty Node List but Non-Zero Node Count %08lX for dir FID %08lX-%08lX-%08lX-%08lX\n",
3410                           ObjectInfo->Specific.Directory.DirectoryNodeCount,
3411                           ObjectInfo->FileId.Cell,
3412                           ObjectInfo->FileId.Volume,
3413                           ObjectInfo->FileId.Vnode,
3414                           ObjectInfo->FileId.Unique);
3415         }
3416
3417         //
3418         // Reset the directory list information by clearing all valid entries
3419         //
3420
3421         pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
3422
3423         while( pCurrentDirEntry != NULL)
3424         {
3425
3426             pNextDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
3427
3428             if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_FAKE))
3429             {
3430
3431                 //
3432                 // If this entry has been deleted then process it here
3433                 //
3434
3435                 if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_DELETED) &&
3436                     pCurrentDirEntry->OpenReferenceCount == 0)
3437                 {
3438
3439                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3440                                   AFS_TRACE_LEVEL_VERBOSE,
3441                                   "AFSValidateDirectoryCache Deleting dir entry %p name %wZ\n",
3442                                   pCurrentDirEntry,
3443                                   &pCurrentDirEntry->NameInformation.FileName);
3444
3445                     AFSDeleteDirEntry( ObjectInfo,
3446                                        pCurrentDirEntry);
3447                 }
3448                 else
3449                 {
3450
3451                     ClearFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_VALID);
3452
3453                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3454                                   AFS_TRACE_LEVEL_VERBOSE,
3455                                   "AFSValidateDirectoryCache Clear VALID flag on DE %p Reference count %08lX\n",
3456                                   pCurrentDirEntry,
3457                                   pCurrentDirEntry->OpenReferenceCount);
3458
3459                     //
3460                     // We pull the short name from the parent tree since it could change below
3461                     //
3462
3463                     if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME))
3464                     {
3465
3466                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3467                                       AFS_TRACE_LEVEL_VERBOSE,
3468                                       "AFSValidateDirectoryCache Removing DE %p (%08lX) from shortname tree for %wZ\n",
3469                                       pCurrentDirEntry,
3470                                       pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex,
3471                                       &pCurrentDirEntry->NameInformation.FileName);
3472
3473                         AFSRemoveShortNameDirEntry( &ObjectInfo->Specific.Directory.ShortNameTree,
3474                                                     pCurrentDirEntry);
3475
3476                         ClearFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3477                     }
3478                 }
3479             }
3480
3481             pCurrentDirEntry = pNextDirEntry;
3482         }
3483
3484         //
3485         // Reget the directory contents
3486         //
3487
3488         ntStatus = AFSVerifyDirectoryContent( ObjectInfo,
3489                                               AuthGroup);
3490
3491         if ( !NT_SUCCESS( ntStatus))
3492         {
3493             try_return( ntStatus);
3494         }
3495
3496         //
3497         // Now start again and tear down any entries not valid
3498         //
3499
3500         pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
3501
3502         while( pCurrentDirEntry != NULL)
3503         {
3504
3505             pNextDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
3506
3507             if( BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_VALID))
3508             {
3509
3510                 if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME) &&
3511                     pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex > 0)
3512                 {
3513
3514                     if( ObjectInfo->Specific.Directory.ShortNameTree == NULL)
3515                     {
3516
3517                         ObjectInfo->Specific.Directory.ShortNameTree = pCurrentDirEntry;
3518
3519                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3520                                       AFS_TRACE_LEVEL_VERBOSE,
3521                                       "AFSValidateDirectoryCache Insert DE %p to head of shortname tree for %wZ\n",
3522                                       pCurrentDirEntry,
3523                                       &pCurrentDirEntry->NameInformation.FileName);
3524
3525                         SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3526                     }
3527                     else
3528                     {
3529
3530                         if( !NT_SUCCESS( AFSInsertShortNameDirEntry( ObjectInfo->Specific.Directory.ShortNameTree,
3531                                                                      pCurrentDirEntry)))
3532                         {
3533                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3534                                           AFS_TRACE_LEVEL_VERBOSE,
3535                                           "AFSValidateDirectoryCache Failed to insert DE %p (%08lX) to shortname tree for %wZ\n",
3536                                           pCurrentDirEntry,
3537                                           pCurrentDirEntry->Type.Data.ShortNameTreeEntry.HashIndex,
3538                                           &pCurrentDirEntry->NameInformation.FileName);
3539                         }
3540                         else
3541                         {
3542                             SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
3543
3544                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3545                                           AFS_TRACE_LEVEL_VERBOSE,
3546                                           "AFSValidateDirectoryCache Insert DE %p to shortname tree for %wZ\n",
3547                                           pCurrentDirEntry,
3548                                           &pCurrentDirEntry->NameInformation.FileName);
3549                         }
3550                     }
3551                 }
3552
3553                 pCurrentDirEntry = pNextDirEntry;
3554
3555                 continue;
3556             }
3557
3558             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3559                           AFS_TRACE_LEVEL_VERBOSE,
3560                           "AFSValidateDirectoryCache Processing INVALID DE %p Reference count %08lX\n",
3561                           pCurrentDirEntry,
3562                           pCurrentDirEntry->OpenReferenceCount);
3563
3564             if( pCurrentDirEntry->OpenReferenceCount == 0)
3565             {
3566
3567                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3568                               AFS_TRACE_LEVEL_VERBOSE,
3569                               "AFSValidateDirectoryCache Deleting dir entry %wZ from parent FID %08lX-%08lX-%08lX-%08lX\n",
3570                               &pCurrentDirEntry->NameInformation.FileName,
3571                               ObjectInfo->FileId.Cell,
3572                               ObjectInfo->FileId.Volume,
3573                               ObjectInfo->FileId.Vnode,
3574                               ObjectInfo->FileId.Unique);
3575
3576                 AFSDeleteDirEntry( ObjectInfo,
3577                                    pCurrentDirEntry);
3578             }
3579             else
3580             {
3581
3582                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3583                               AFS_TRACE_LEVEL_VERBOSE,
3584                               "AFSValidateDirectoryCache Setting dir entry %p Name %wZ DELETED in parent FID %08lX-%08lX-%08lX-%08lX\n",
3585                               pCurrentDirEntry,
3586                               &pCurrentDirEntry->NameInformation.FileName,
3587                               ObjectInfo->FileId.Cell,
3588                               ObjectInfo->FileId.Volume,
3589                               ObjectInfo->FileId.Vnode,
3590                               ObjectInfo->FileId.Unique);
3591
3592                 SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_DELETED);
3593
3594                 AFSRemoveNameEntry( ObjectInfo,
3595                                     pCurrentDirEntry);
3596             }
3597
3598             pCurrentDirEntry = pNextDirEntry;
3599         }
3600
3601 #if DBG
3602         if( !AFSValidateDirList( ObjectInfo))
3603         {
3604
3605             AFSPrint("AFSValidateDirectoryCache Invalid count ...\n");
3606         }
3607 #endif
3608
3609 try_exit:
3610
3611         if( bAcquiredLock)
3612         {
3613
3614             AFSReleaseResource( ObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock);
3615         }
3616     }
3617
3618     return ntStatus;
3619 }
3620
3621 BOOLEAN
3622 AFSIsVolumeFID( IN AFSFileID *FileID)
3623 {
3624
3625     BOOLEAN bIsVolume = FALSE;
3626
3627     if( FileID->Vnode == 1 &&
3628         FileID->Unique == 1)
3629     {
3630
3631         bIsVolume = TRUE;
3632     }
3633
3634     return bIsVolume;
3635 }
3636
3637 BOOLEAN
3638 AFSIsFinalNode( IN AFSFcb *Fcb)
3639 {
3640
3641     BOOLEAN bIsFinalNode = FALSE;
3642
3643     if( Fcb->Header.NodeTypeCode == AFS_ROOT_FCB ||
3644         Fcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB ||
3645         Fcb->Header.NodeTypeCode == AFS_FILE_FCB ||
3646         Fcb->Header.NodeTypeCode == AFS_DFS_LINK_FCB ||
3647         Fcb->Header.NodeTypeCode == AFS_INVALID_FCB )
3648     {
3649
3650         bIsFinalNode = TRUE;
3651     }
3652     else
3653     {
3654
3655         ASSERT( Fcb->Header.NodeTypeCode == AFS_MOUNT_POINT_FCB ||
3656                 Fcb->Header.NodeTypeCode == AFS_SYMBOLIC_LINK_FCB);
3657     }
3658
3659     return bIsFinalNode;
3660 }
3661
3662 NTSTATUS
3663 AFSUpdateMetaData( IN AFSDirectoryCB *DirEntry,
3664                    IN AFSDirEnumEntry *DirEnumEntry)
3665 {
3666
3667     NTSTATUS ntStatus = STATUS_SUCCESS;
3668     UNICODE_STRING uniTargetName;
3669     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
3670
3671     __Enter
3672     {
3673
3674         pObjectInfo->TargetFileId = DirEnumEntry->TargetFileId;
3675
3676         pObjectInfo->Expiration = DirEnumEntry->Expiration;
3677
3678         pObjectInfo->DataVersion = DirEnumEntry->DataVersion;
3679
3680         pObjectInfo->FileType = DirEnumEntry->FileType;
3681
3682         pObjectInfo->CreationTime = DirEnumEntry->CreationTime;
3683
3684         pObjectInfo->LastAccessTime = DirEnumEntry->LastAccessTime;
3685
3686         pObjectInfo->LastWriteTime = DirEnumEntry->LastWriteTime;
3687
3688         pObjectInfo->ChangeTime = DirEnumEntry->ChangeTime;
3689
3690         pObjectInfo->EndOfFile = DirEnumEntry->EndOfFile;
3691
3692         pObjectInfo->AllocationSize = DirEnumEntry->AllocationSize;
3693
3694         pObjectInfo->FileAttributes = DirEnumEntry->FileAttributes;
3695
3696         if( pObjectInfo->FileType == AFS_FILE_TYPE_MOUNTPOINT)
3697         {
3698
3699             pObjectInfo->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
3700         }
3701
3702         if( pObjectInfo->FileType == AFS_FILE_TYPE_SYMLINK ||
3703             pObjectInfo->FileType == AFS_FILE_TYPE_DFSLINK)
3704         {
3705
3706             pObjectInfo->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
3707         }
3708
3709         pObjectInfo->EaSize = DirEnumEntry->EaSize;
3710
3711         pObjectInfo->Links = DirEnumEntry->Links;
3712
3713         if( DirEnumEntry->TargetNameLength > 0)
3714         {
3715
3716             //
3717             // Update the target name information if needed
3718             //
3719
3720             uniTargetName.Length = (USHORT)DirEnumEntry->TargetNameLength;
3721
3722             uniTargetName.MaximumLength = uniTargetName.Length;
3723
3724             uniTargetName.Buffer = (WCHAR *)((char *)DirEnumEntry + DirEnumEntry->TargetNameOffset);
3725
3726             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3727                             TRUE);
3728
3729             if( DirEntry->NameInformation.TargetName.Length == 0 ||
3730                 RtlCompareUnicodeString( &uniTargetName,
3731                                          &DirEntry->NameInformation.TargetName,
3732                                          TRUE) != 0)
3733             {
3734
3735                 //
3736                 // Update the target name
3737                 //
3738
3739                 ntStatus = AFSUpdateTargetName( &DirEntry->NameInformation.TargetName,
3740                                                 &DirEntry->Flags,
3741                                                 uniTargetName.Buffer,
3742                                                 uniTargetName.Length);
3743
3744                 if( !NT_SUCCESS( ntStatus))
3745                 {
3746
3747                     AFSReleaseResource( &DirEntry->NonPaged->Lock);
3748
3749                     try_return( ntStatus);
3750                 }
3751             }
3752
3753             AFSReleaseResource( &DirEntry->NonPaged->Lock);
3754         }
3755         else if( DirEntry->NameInformation.TargetName.Length > 0)
3756         {
3757
3758             AFSAcquireExcl( &DirEntry->NonPaged->Lock,
3759                             TRUE);
3760
3761             if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER) &&
3762                 DirEntry->NameInformation.TargetName.Buffer != NULL)
3763             {
3764                 AFSExFreePool( DirEntry->NameInformation.TargetName.Buffer);
3765             }
3766
3767             ClearFlag( DirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER);
3768
3769             DirEntry->NameInformation.TargetName.Length = 0;
3770             DirEntry->NameInformation.TargetName.MaximumLength = 0;
3771             DirEntry->NameInformation.TargetName.Buffer = NULL;
3772
3773             AFSReleaseResource( &DirEntry->NonPaged->Lock);
3774         }
3775
3776 try_exit:
3777
3778         NOTHING;
3779     }
3780
3781     return ntStatus;
3782 }
3783
3784 NTSTATUS
3785 AFSValidateEntry( IN AFSDirectoryCB *DirEntry,
3786                   IN GUID *AuthGroup,
3787                   IN BOOLEAN FastCall)
3788 {
3789
3790     NTSTATUS ntStatus = STATUS_SUCCESS;
3791     LARGE_INTEGER liSystemTime;
3792     AFSDirEnumEntry *pDirEnumEntry = NULL;
3793     AFSFcb *pCurrentFcb = NULL;
3794     BOOLEAN bReleaseFcb = FALSE;
3795     AFSObjectInfoCB *pObjectInfo = DirEntry->ObjectInformation;
3796
3797     __Enter
3798     {
3799
3800         //
3801         // If we have an Fcb hanging off the directory entry then be sure to acquire the locks in the
3802         // correct order
3803         //
3804
3805         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3806                       AFS_TRACE_LEVEL_VERBOSE_2,
3807                       "AFSValidateEntry Validating entry %wZ FID %08lX-%08lX-%08lX-%08lX FastCall %u\n",
3808                       &DirEntry->NameInformation.FileName,
3809                       pObjectInfo->FileId.Cell,
3810                       pObjectInfo->FileId.Volume,
3811                       pObjectInfo->FileId.Vnode,
3812                       pObjectInfo->FileId.Unique,
3813                       FastCall);
3814
3815         //
3816         // If this is a fake node then bail since the service knows nothing about it
3817         //
3818
3819         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_FAKE))
3820         {
3821
3822             try_return( ntStatus);
3823         }
3824
3825         //
3826         // This routine ensures that the current entry is valid by:
3827         //
3828         //      1) Checking that the expiration time is non-zero and after where we
3829         //         currently are
3830         //
3831
3832         KeQuerySystemTime( &liSystemTime);
3833
3834         if( !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_NOT_EVALUATED) &&
3835             !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY) &&
3836             !BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA) &&
3837             pObjectInfo->Expiration.QuadPart >= liSystemTime.QuadPart)
3838         {
3839
3840             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3841                           AFS_TRACE_LEVEL_VERBOSE_2,
3842                           "AFSValidateEntry Directory entry %wZ FID %08lX-%08lX-%08lX-%08lX VALID\n",
3843                           &DirEntry->NameInformation.FileName,
3844                           pObjectInfo->FileId.Cell,
3845                           pObjectInfo->FileId.Volume,
3846                           pObjectInfo->FileId.Vnode,
3847                           pObjectInfo->FileId.Unique);
3848
3849             try_return( ntStatus);
3850         }
3851
3852         //
3853         // This node requires updating
3854         //
3855
3856         ntStatus = AFSEvaluateTargetByID( pObjectInfo,
3857                                           AuthGroup,
3858                                           FastCall,
3859                                           &pDirEnumEntry);
3860
3861         if( !NT_SUCCESS( ntStatus))
3862         {
3863
3864             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3865                           AFS_TRACE_LEVEL_ERROR,
3866                           "AFSValidateEntry Failed to evaluate entry FastCall %d %wZ FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
3867                           FastCall,
3868                           &DirEntry->NameInformation.FileName,
3869                           pObjectInfo->FileId.Cell,
3870                           pObjectInfo->FileId.Volume,
3871                           pObjectInfo->FileId.Vnode,
3872                           pObjectInfo->FileId.Unique,
3873                           ntStatus);
3874
3875             //
3876             // Failed validation of node so return access-denied
3877             //
3878
3879             try_return( ntStatus);
3880         }
3881
3882         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3883                       AFS_TRACE_LEVEL_VERBOSE,
3884                       "AFSValidateEntry Validating entry FastCall %d %wZ FID %08lX-%08lX-%08lX-%08lX DV %I64X returned DV %I64X FT %d\n",
3885                       FastCall,
3886                       &DirEntry->NameInformation.FileName,
3887                       pObjectInfo->FileId.Cell,
3888                       pObjectInfo->FileId.Volume,
3889                       pObjectInfo->FileId.Vnode,
3890                       pObjectInfo->FileId.Unique,
3891                       pObjectInfo->DataVersion.QuadPart,
3892                       pDirEnumEntry->DataVersion.QuadPart,
3893                       pDirEnumEntry->FileType);
3894
3895
3896         //
3897         // Based on the file type, process the node
3898         //
3899
3900         switch( pDirEnumEntry->FileType)
3901         {
3902
3903             case AFS_FILE_TYPE_MOUNTPOINT:
3904             {
3905
3906                 //
3907                 // Update the metadata for the entry
3908                 //
3909
3910                 ntStatus = AFSUpdateMetaData( DirEntry,
3911                                               pDirEnumEntry);
3912
3913                 if( NT_SUCCESS( ntStatus))
3914                 {
3915
3916                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY | AFS_OBJECT_FLAGS_NOT_EVALUATED);
3917                 }
3918
3919                 break;
3920             }
3921
3922             case AFS_FILE_TYPE_SYMLINK:
3923             case AFS_FILE_TYPE_DFSLINK:
3924             {
3925
3926                 //
3927                 // Update the metadata for the entry
3928                 //
3929
3930                 ntStatus = AFSUpdateMetaData( DirEntry,
3931                                               pDirEnumEntry);
3932
3933                 if( NT_SUCCESS( ntStatus))
3934                 {
3935
3936                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY | AFS_OBJECT_FLAGS_NOT_EVALUATED);
3937                 }
3938
3939                 break;
3940             }
3941
3942             case AFS_FILE_TYPE_FILE:
3943             {
3944
3945                 //
3946                 // For a file where the data version has become invalid we need to
3947                 // fail any current extent requests and purge the cache for the file
3948                 // Can't hold the Fcb resource while doing this
3949                 //
3950
3951                 if( pObjectInfo->Fcb != NULL &&
3952                     (pObjectInfo->DataVersion.QuadPart != pDirEnumEntry->DataVersion.QuadPart ||
3953                       BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA)))
3954                 {
3955
3956                     pCurrentFcb = pObjectInfo->Fcb;
3957
3958                     if( !ExIsResourceAcquiredLite( &pCurrentFcb->NPFcb->Resource))
3959                     {
3960
3961                         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
3962                                       AFS_TRACE_LEVEL_VERBOSE,
3963                                       "AFSValidateEntry Acquiring Fcb lock %08lX EXCL %08lX\n",
3964                                       &pCurrentFcb->NPFcb->Resource,
3965                                       PsGetCurrentThread());
3966
3967                         AFSAcquireExcl( &pCurrentFcb->NPFcb->Resource,
3968                                         TRUE);
3969
3970                         bReleaseFcb = TRUE;
3971                     }
3972
3973                     if( pCurrentFcb != NULL)
3974                     {
3975
3976                         IO_STATUS_BLOCK stIoStatus;
3977                         BOOLEAN bPurgeExtents = FALSE;
3978
3979                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3980                                       AFS_TRACE_LEVEL_VERBOSE_2,
3981                                       "AFSValidateEntry Flush/purge entry %wZ FID %08lX-%08lX-%08lX-%08lX\n",
3982                                       &DirEntry->NameInformation.FileName,
3983                                       pObjectInfo->FileId.Cell,
3984                                       pObjectInfo->FileId.Volume,
3985                                       pObjectInfo->FileId.Vnode,
3986                                       pObjectInfo->FileId.Unique);
3987
3988                         if ( pObjectInfo->DataVersion.QuadPart != pDirEnumEntry->DataVersion.QuadPart)
3989                         {
3990
3991                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
3992                                           AFS_TRACE_LEVEL_VERBOSE,
3993                                           "AFSValidateEntry DV Change %wZ FID %08lX-%08lX-%08lX-%08lX (%08lX != %08lX)\n",
3994                                           &DirEntry->NameInformation.FileName,
3995                                           pObjectInfo->FileId.Cell,
3996                                           pObjectInfo->FileId.Volume,
3997                                           pObjectInfo->FileId.Vnode,
3998                                           pObjectInfo->FileId.Unique,
3999                                           pObjectInfo->DataVersion.LowPart,
4000                                           pDirEnumEntry->DataVersion.LowPart
4001                                           );
4002
4003                             bPurgeExtents = TRUE;
4004                         }
4005
4006                         if ( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA))
4007                         {
4008                             bPurgeExtents = TRUE;
4009
4010                             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
4011                                           AFS_TRACE_LEVEL_VERBOSE,
4012                                           "AFSVerifyEntry Clearing VERIFY_DATA flag %wZ FID %08lX-%08lX-%08lX-%08lX\n",
4013                                           &DirEntry->NameInformation.FileName,
4014                                           pObjectInfo->FileId.Cell,
4015                                           pObjectInfo->FileId.Volume,
4016                                           pObjectInfo->FileId.Vnode,
4017                                           pObjectInfo->FileId.Unique);
4018
4019                             ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY_DATA);
4020                         }
4021
4022                         __try
4023                         {
4024
4025                             CcFlushCache( &pCurrentFcb->NPFcb->SectionObjectPointers,
4026                                           NULL,
4027                                           0,
4028                                           &stIoStatus);
4029
4030                             if( !NT_SUCCESS( stIoStatus.Status))
4031                             {
4032
4033                                 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
4034                                               AFS_TRACE_LEVEL_ERROR,
4035                                               "AFSValidateEntry CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
4036                                               &DirEntry->NameInformation.FileName,
4037                                               pObjectInfo->FileId.Cell,
4038                                               pObjectInfo->FileId.Volume,
4039                                               pObjectInfo->FileId.Vnode,
4040                                               pObjectInfo->FileId.Unique,
4041                                               stIoStatus.Status,
4042                                               stIoStatus.Information);
4043
4044                                 ntStatus = stIoStatus.Status;
4045                             }
4046
4047                             if ( bPurgeExtents)
4048                             {
4049
4050                                 if ( !CcPurgeCacheSection( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers,
4051                                                            NULL,
4052                                                            0,
4053                                                            FALSE))
4054                                 {
4055
4056                                     AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
4057                                                   AFS_TRACE_LEVEL_WARNING,
4058                                                   "AFSValidateEntry CcPurgeCacheSection failure %wZ FID %08lX-%08lX-%08lX-%08lX\n",
4059                                                   &DirEntry->NameInformation.FileName,
4060                                                   pObjectInfo->FileId.Cell,
4061                                                   pObjectInfo->FileId.Volume,
4062                                                   pObjectInfo->FileId.Vnode,
4063                                                   pObjectInfo->FileId.Unique);
4064
4065                                     SetFlag( pObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
4066                                 }
4067                             }
4068                         }
4069                         __except( EXCEPTION_EXECUTE_HANDLER)
4070                         {
4071                             ntStatus = GetExceptionCode();
4072
4073                             AFSDbgLogMsg( 0,
4074                                           0,
4075                                           "EXCEPTION - AFSValidateEntry CcFlushCache or CcPurgeCacheSection %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
4076                                           &DirEntry->NameInformation.FileName,
4077                                           pObjectInfo->FileId.Cell,
4078                                           pObjectInfo->FileId.Volume,
4079                                           pObjectInfo->FileId.Vnode,
4080                                           pObjectInfo->FileId.Unique,
4081                                           ntStatus);
4082
4083                             SetFlag( pObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
4084                         }
4085
4086                         AFSReleaseResource( &pCurrentFcb->NPFcb->Resource);
4087
4088                         bReleaseFcb = FALSE;
4089
4090                         if ( bPurgeExtents)
4091                         {
4092                             AFSFlushExtents( pCurrentFcb,
4093                                              AuthGroup);
4094                         }
4095                     }
4096                 }
4097
4098                 //
4099                 // Update the metadata for the entry
4100                 //
4101
4102                 ntStatus = AFSUpdateMetaData( DirEntry,
4103                                               pDirEnumEntry);
4104
4105                 if( !NT_SUCCESS( ntStatus))
4106                 {
4107
4108                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
4109                                   AFS_TRACE_LEVEL_ERROR,
4110                                   "AFSValidateEntry Meta Data Update failed %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
4111                                   &DirEntry->NameInformation.FileName,
4112                                   pObjectInfo->FileId.Cell,
4113                                   pObjectInfo->FileId.Volume,
4114                                   pObjectInfo->FileId.Vnode,
4115                                   pObjectInfo->FileId.Unique,
4116                                   ntStatus);
4117
4118                     break;
4119                 }
4120
4121                 ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY | AFS_OBJECT_FLAGS_NOT_EVALUATED);
4122
4123                 //
4124                 // Update file sizes
4125                 //
4126
4127                 if( pObjectInfo->Fcb != NULL)
4128                 {
4129                     FILE_OBJECT *pCCFileObject = CcGetFileObjectFromSectionPtrs( &pObjectInfo->Fcb->NPFcb->SectionObjectPointers);
4130
4131                     pObjectInfo->Fcb->Header.AllocationSize.QuadPart  = pObjectInfo->AllocationSize.QuadPart;
4132                     pObjectInfo->Fcb->Header.FileSize.QuadPart        = pObjectInfo->EndOfFile.QuadPart;
4133                     pObjectInfo->Fcb->Header.ValidDataLength.QuadPart = pObjectInfo->EndOfFile.QuadPart;
4134
4135                     if ( pCCFileObject != NULL)
4136                     {
4137                         CcSetFileSizes( pCCFileObject,
4138                                         (PCC_FILE_SIZES)&pObjectInfo->Fcb->Header.AllocationSize);
4139                     }
4140                 }
4141                 break;
4142             }
4143
4144             case AFS_FILE_TYPE_DIRECTORY:
4145             {
4146
4147                 AFSDirectoryCB *pCurrentDirEntry = NULL;
4148
4149                 if( pObjectInfo->DataVersion.QuadPart != pDirEnumEntry->DataVersion.QuadPart)
4150                 {
4151
4152                     //
4153                     // For a directory or root entry flush the content of
4154                     // the directory enumeration.
4155                     //
4156
4157                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
4158                                   AFS_TRACE_LEVEL_VERBOSE,
4159                                   "AFSValidateEntry Acquiring DirectoryNodeHdr.TreeLock lock %08lX EXCL %08lX\n",
4160                                   pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
4161                                   PsGetCurrentThread());
4162
4163                     if( BooleanFlagOn( pObjectInfo->Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED))
4164                     {
4165
4166                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
4167                                       AFS_TRACE_LEVEL_VERBOSE_2,
4168                                       "AFSValidateEntry Validating directory content for %wZ FID %08lX-%08lX-%08lX-%08lX\n",
4169                                       &DirEntry->NameInformation.FileName,
4170                                       pObjectInfo->FileId.Cell,
4171                                       pObjectInfo->FileId.Volume,
4172                                       pObjectInfo->FileId.Vnode,
4173                                       pObjectInfo->FileId.Unique);
4174
4175                         AFSAcquireExcl( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock,
4176                                         TRUE);
4177
4178                         ntStatus = AFSValidateDirectoryCache( pObjectInfo,
4179                                                               AuthGroup);
4180
4181                         AFSReleaseResource( pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock);
4182                     }
4183
4184                     if( !NT_SUCCESS( ntStatus))
4185                     {
4186
4187                         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
4188                                       AFS_TRACE_LEVEL_ERROR,
4189                                       "AFSValidateEntry Failed to re-enumerate %wZ FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
4190                                       &DirEntry->NameInformation.FileName,
4191                                       pObjectInfo->FileId.Cell,
4192                                       pObjectInfo->FileId.Volume,
4193                                       pObjectInfo->FileId.Vnode,
4194                                       pObjectInfo->FileId.Unique,
4195                                       ntStatus);
4196
4197                         break;
4198                     }
4199                 }
4200
4201                 //
4202                 // Update the metadata for the entry
4203                 //
4204
4205                 ntStatus = AFSUpdateMetaData( DirEntry,
4206                                               pDirEnumEntry);
4207
4208                 if( NT_SUCCESS( ntStatus))
4209                 {
4210
4211                     ClearFlag( pObjectInfo->Flags, AFS_OBJECT_FLAGS_VERIFY | AFS_OBJECT_FLAGS_NOT_EVALUATED);
4212                 }
4213
4214                 break;
4215             }
4216
4217             default:
4218
4219                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
4220                               AFS_TRACE_LEVEL_WARNING,
4221                               "AFSValidateEntry Attempt to verify node of type %d FastCall %d %wZ FID %08lX-%08lX-%08lX-%08lX\n",
4222                               pObjectInfo->FileType,
4223                               FastCall,
4224                               &DirEntry->NameInformation.FileName,
4225                               pObjectInfo->FileId.Cell,
4226                               pObjectInfo->FileId.Volume,
4227                               pObjectInfo->FileId.Vnode,
4228                               pObjectInfo->FileId.Unique);
4229
4230                 break;
4231         }
4232
4233  try_exit:
4234
4235         if( bReleaseFcb)
4236         {
4237
4238             AFSReleaseResource( &pCurrentFcb->NPFcb->Resource);
4239         }
4240
4241         if( pDirEnumEntry != NULL)
4242         {
4243
4244             AFSExFreePool( pDirEnumEntry);
4245         }
4246     }
4247
4248     return ntStatus;
4249 }
4250
4251 NTSTATUS
4252 AFSInitializeSpecialShareNameList()
4253 {
4254
4255     NTSTATUS ntStatus = STATUS_SUCCESS;
4256     AFSDirectoryCB *pDirNode = NULL, *pLastDirNode = NULL;
4257     AFSObjectInfoCB *pObjectInfoCB = NULL;
4258     UNICODE_STRING uniShareName;
4259     ULONG ulEntryLength = 0;
4260     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
4261
4262     __Enter
4263     {
4264
4265         RtlInitUnicodeString( &uniShareName,
4266                               L"PIPE\\srvsvc");
4267
4268         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
4269                                                0);
4270
4271         if( pObjectInfoCB == NULL)
4272         {
4273
4274             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4275         }
4276
4277         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
4278                       AFS_TRACE_LEVEL_VERBOSE,
4279                       "AFSInitializeSpecialShareNameList (srvsvc) Initializing count (1) on object %08lX\n",
4280                       pObjectInfoCB);
4281
4282         pObjectInfoCB->ObjectReferenceCount = 1;
4283
4284         pObjectInfoCB->FileType = AFS_FILE_TYPE_SPECIAL_SHARE_NAME;
4285
4286         ulEntryLength = sizeof( AFSDirectoryCB) +
4287                                      uniShareName.Length;
4288
4289         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
4290                                                                   ulEntryLength,
4291                                                                   AFS_DIR_ENTRY_TAG);
4292
4293         if( pDirNode == NULL)
4294         {
4295
4296             AFSDeleteObjectInfo( pObjectInfoCB);
4297
4298             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4299         }
4300
4301         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
4302                                                                                    sizeof( AFSNonPagedDirectoryCB),
4303                                                                                    AFS_DIR_ENTRY_NP_TAG);
4304
4305         if( pNonPagedDirEntry == NULL)
4306         {
4307
4308             ExFreePool( pDirNode);
4309
4310             AFSDeleteObjectInfo( pObjectInfoCB);
4311
4312             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4313         }
4314
4315         RtlZeroMemory( pDirNode,
4316                        ulEntryLength);
4317
4318         RtlZeroMemory( pNonPagedDirEntry,
4319                        sizeof( AFSNonPagedDirectoryCB));
4320
4321         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
4322
4323         pDirNode->NonPaged = pNonPagedDirEntry;
4324
4325         pDirNode->ObjectInformation = pObjectInfoCB;
4326
4327         //
4328         // Set valid entry
4329         //
4330
4331         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_SERVER_SERVICE);
4332
4333         pDirNode->NameInformation.FileName.Length = uniShareName.Length;
4334
4335         pDirNode->NameInformation.FileName.MaximumLength = uniShareName.Length;
4336
4337         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
4338
4339         RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
4340                        uniShareName.Buffer,
4341                        pDirNode->NameInformation.FileName.Length);
4342
4343         pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
4344                                                                        TRUE);
4345
4346         AFSSpecialShareNames = pDirNode;
4347
4348         pLastDirNode = pDirNode;
4349
4350         RtlInitUnicodeString( &uniShareName,
4351                               L"PIPE\\wkssvc");
4352
4353         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
4354                                                0);
4355
4356         if( pObjectInfoCB == NULL)
4357         {
4358
4359             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4360         }
4361
4362         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
4363                       AFS_TRACE_LEVEL_VERBOSE,
4364                       "AFSInitializeSpecialShareNameList (wkssvc) Initializing count (1) on object %08lX\n",
4365                       pObjectInfoCB);
4366
4367         pObjectInfoCB->ObjectReferenceCount = 1;
4368
4369         pObjectInfoCB->FileType = AFS_FILE_TYPE_SPECIAL_SHARE_NAME;
4370
4371         ulEntryLength = sizeof( AFSDirectoryCB) +
4372                                      uniShareName.Length;
4373
4374         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
4375                                                                   ulEntryLength,
4376                                                                   AFS_DIR_ENTRY_TAG);
4377
4378         if( pDirNode == NULL)
4379         {
4380
4381             AFSDeleteObjectInfo( pObjectInfoCB);
4382
4383             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4384         }
4385
4386         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
4387                                                                                    sizeof( AFSNonPagedDirectoryCB),
4388                                                                                    AFS_DIR_ENTRY_NP_TAG);
4389
4390         if( pNonPagedDirEntry == NULL)
4391         {
4392
4393             ExFreePool( pDirNode);
4394
4395             AFSDeleteObjectInfo( pObjectInfoCB);
4396
4397             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4398         }
4399
4400         RtlZeroMemory( pDirNode,
4401                        ulEntryLength);
4402
4403         RtlZeroMemory( pNonPagedDirEntry,
4404                        sizeof( AFSNonPagedDirectoryCB));
4405
4406         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
4407
4408         pDirNode->NonPaged = pNonPagedDirEntry;
4409
4410         pDirNode->ObjectInformation = pObjectInfoCB;
4411
4412         //
4413         // Set valid entry
4414         //
4415
4416         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_WORKSTATION_SERVICE);
4417
4418         pDirNode->NameInformation.FileName.Length = uniShareName.Length;
4419
4420         pDirNode->NameInformation.FileName.MaximumLength = uniShareName.Length;
4421
4422         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
4423
4424         RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
4425                        uniShareName.Buffer,
4426                        pDirNode->NameInformation.FileName.Length);
4427
4428         pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
4429                                                                        TRUE);
4430
4431         pLastDirNode->ListEntry.fLink = pDirNode;
4432
4433         pDirNode->ListEntry.bLink = pLastDirNode;
4434
4435         pLastDirNode = pDirNode;
4436
4437         RtlInitUnicodeString( &uniShareName,
4438                               L"IPC$");
4439
4440         pObjectInfoCB = AFSAllocateObjectInfo( &AFSGlobalRoot->ObjectInformation,
4441                                                0);
4442
4443         if( pObjectInfoCB == NULL)
4444         {
4445
4446             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4447         }
4448
4449         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
4450                       AFS_TRACE_LEVEL_VERBOSE,
4451                       "AFSInitializeSpecialShareNameList (ipc$) Initializing count (1) on object %08lX\n",
4452                       pObjectInfoCB);
4453
4454         pObjectInfoCB->ObjectReferenceCount = 1;
4455
4456         pObjectInfoCB->FileType = AFS_FILE_TYPE_SPECIAL_SHARE_NAME;
4457
4458         ulEntryLength = sizeof( AFSDirectoryCB) +
4459                                      uniShareName.Length;
4460
4461         pDirNode = (AFSDirectoryCB *)AFSLibExAllocatePoolWithTag( PagedPool,
4462                                                                   ulEntryLength,
4463                                                                   AFS_DIR_ENTRY_TAG);
4464
4465         if( pDirNode == NULL)
4466         {
4467
4468             AFSDeleteObjectInfo( pObjectInfoCB);
4469
4470             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4471         }
4472
4473         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSLibExAllocatePoolWithTag( NonPagedPool,
4474                                                                                    sizeof( AFSNonPagedDirectoryCB),
4475                                                                                    AFS_DIR_ENTRY_NP_TAG);
4476
4477         if( pNonPagedDirEntry == NULL)
4478         {
4479
4480             ExFreePool( pDirNode);
4481
4482             AFSDeleteObjectInfo( pObjectInfoCB);
4483
4484             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4485         }
4486
4487         RtlZeroMemory( pDirNode,
4488                        ulEntryLength);
4489
4490         RtlZeroMemory( pNonPagedDirEntry,
4491                        sizeof( AFSNonPagedDirectoryCB));
4492
4493         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
4494
4495         pDirNode->NonPaged = pNonPagedDirEntry;
4496
4497         pDirNode->ObjectInformation = pObjectInfoCB;
4498
4499         //
4500         // Set valid entry
4501         //
4502
4503         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_IPC);
4504
4505         pDirNode->NameInformation.FileName.Length = uniShareName.Length;
4506
4507         pDirNode->NameInformation.FileName.MaximumLength = uniShareName.Length;
4508
4509         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
4510
4511         RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
4512                        uniShareName.Buffer,
4513                        pDirNode->NameInformation.FileName.Length);
4514
4515         pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
4516                                                                        TRUE);
4517
4518         pLastDirNode->ListEntry.fLink = pDirNode;
4519
4520         pDirNode->ListEntry.bLink = pLastDirNode;
4521
4522 try_exit:
4523
4524         if( !NT_SUCCESS( ntStatus))
4525         {
4526
4527             if( AFSSpecialShareNames != NULL)
4528             {
4529
4530                 pDirNode = AFSSpecialShareNames;
4531
4532                 while( pDirNode != NULL)
4533                 {
4534
4535                     pLastDirNode = (AFSDirectoryCB *)pDirNode->ListEntry.fLink;
4536
4537                     AFSDeleteObjectInfo( pDirNode->ObjectInformation);
4538
4539                     ExDeleteResourceLite( &pDirNode->NonPaged->Lock);
4540
4541                     ExFreePool( pDirNode->NonPaged);
4542
4543                     ExFreePool( pDirNode);
4544
4545                     pDirNode = pLastDirNode;
4546                 }
4547
4548                 AFSSpecialShareNames = NULL;
4549             }
4550         }
4551     }
4552
4553     return ntStatus;
4554 }
4555
4556 AFSDirectoryCB *
4557 AFSGetSpecialShareNameEntry( IN UNICODE_STRING *ShareName,
4558                              IN UNICODE_STRING *SecondaryName)
4559 {
4560
4561     AFSDirectoryCB *pDirectoryCB = NULL;
4562     ULONGLONG ullHash = 0;
4563     UNICODE_STRING uniFullShareName;
4564
4565     __Enter
4566     {
4567
4568         //
4569         // Build up the entire name here. We are guaranteed that if there is a
4570         // secondary name, it is pointing to a portion of the share name buffer
4571         //
4572
4573         if( SecondaryName->Length > 0 &&
4574             SecondaryName->Buffer != NULL)
4575         {
4576
4577             uniFullShareName = *SecondaryName;
4578
4579             //
4580             // The calling routine strips off the leading slash so add it back in
4581             //
4582
4583             uniFullShareName.Buffer--;
4584             uniFullShareName.Length += sizeof( WCHAR);
4585             uniFullShareName.MaximumLength += sizeof( WCHAR);
4586
4587             //
4588             // And the share name
4589             //
4590
4591             uniFullShareName.Buffer -= (ShareName->Length/sizeof( WCHAR));
4592             uniFullShareName.Length += ShareName->Length;
4593             uniFullShareName.MaximumLength += ShareName->Length;
4594         }
4595         else
4596         {
4597
4598             uniFullShareName = *ShareName;
4599         }
4600
4601         //
4602         // Generate our hash value
4603         //
4604
4605         ullHash = AFSGenerateCRC( &uniFullShareName,
4606                                   TRUE);
4607
4608         //
4609         // Loop through our special share names to see if this is one of them
4610         //
4611
4612         pDirectoryCB = AFSSpecialShareNames;
4613
4614         while( pDirectoryCB != NULL)
4615         {
4616
4617             if( ullHash == pDirectoryCB->CaseInsensitiveTreeEntry.HashIndex)
4618             {
4619
4620                 break;
4621             }
4622
4623             pDirectoryCB = (AFSDirectoryCB *)pDirectoryCB->ListEntry.fLink;
4624         }
4625     }
4626
4627     return pDirectoryCB;
4628 }
4629
4630 void
4631 AFSWaitOnQueuedFlushes( IN AFSFcb *Fcb)
4632 {
4633
4634     //
4635     // Block on the queue flush event
4636     //
4637
4638     KeWaitForSingleObject( &Fcb->NPFcb->Specific.File.QueuedFlushEvent,
4639                            Executive,
4640                            KernelMode,
4641                            FALSE,
4642                            NULL);
4643
4644     return;
4645 }
4646
4647 void
4648 AFSWaitOnQueuedReleases()
4649 {
4650
4651     AFSDeviceExt *pRDRDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
4652
4653     //
4654     // Block on the queue flush event
4655     //
4656
4657     KeWaitForSingleObject( &pRDRDeviceExt->Specific.RDR.QueuedReleaseExtentEvent,
4658                            Executive,
4659                            KernelMode,
4660                            FALSE,
4661                            NULL);
4662
4663     return;
4664 }
4665
4666 BOOLEAN
4667 AFSIsEqualFID( IN AFSFileID *FileId1,
4668                IN AFSFileID *FileId2)
4669 {
4670
4671     BOOLEAN bIsEqual = FALSE;
4672
4673     if( FileId1->Hash == FileId2->Hash &&
4674         FileId1->Unique == FileId2->Unique &&
4675         FileId1->Vnode == FileId2->Vnode &&
4676         FileId1->Volume == FileId2->Volume &&
4677         FileId1->Cell == FileId2->Cell)
4678     {
4679
4680         bIsEqual = TRUE;
4681     }
4682
4683     return bIsEqual;
4684 }
4685
4686 NTSTATUS
4687 AFSResetDirectoryContent( IN AFSObjectInfoCB *ObjectInfoCB)
4688 {
4689
4690     NTSTATUS ntStatus = STATUS_SUCCESS;
4691     AFSDirectoryCB *pCurrentDirEntry = NULL, *pNextDirEntry = NULL;
4692
4693     __Enter
4694     {
4695
4696         ASSERT( ExIsResourceAcquiredExclusiveLite( ObjectInfoCB->Specific.Directory.DirectoryNodeHdr.TreeLock));
4697
4698         //
4699         // Reset the directory list information
4700         //
4701
4702         pCurrentDirEntry = ObjectInfoCB->Specific.Directory.DirectoryNodeListHead;
4703
4704         while( pCurrentDirEntry != NULL)
4705         {
4706
4707             pNextDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
4708
4709             if( pCurrentDirEntry->OpenReferenceCount == 0)
4710             {
4711
4712                 AFSDbgLogMsg( AFS_SUBSYSTEM_CLEANUP_PROCESSING,
4713                               AFS_TRACE_LEVEL_VERBOSE,
4714                               "AFSResetDirectoryContent Deleting dir entry %p for %wZ\n",
4715                               pCurrentDirEntry,
4716                               &pCurrentDirEntry->NameInformation.FileName);
4717
4718                 AFSDeleteDirEntry( ObjectInfoCB,
4719                                    pCurrentDirEntry);
4720             }
4721             else
4722             {
4723
4724                 AFSDbgLogMsg( AFS_SUBSYSTEM_CLEANUP_PROCESSING,
4725                               AFS_TRACE_LEVEL_VERBOSE,
4726                               "AFSResetDirectoryContent Setting DELETE flag in dir entry %p for %wZ\n",
4727                               pCurrentDirEntry,
4728                               &pCurrentDirEntry->NameInformation.FileName);
4729
4730                 SetFlag( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_DELETED);
4731
4732                 AFSRemoveNameEntry( ObjectInfoCB,
4733                                     pCurrentDirEntry);
4734             }
4735
4736             pCurrentDirEntry = pNextDirEntry;
4737         }
4738
4739         ObjectInfoCB->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead = NULL;
4740
4741         ObjectInfoCB->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead = NULL;
4742
4743         ObjectInfoCB->Specific.Directory.ShortNameTree = NULL;
4744
4745         ObjectInfoCB->Specific.Directory.DirectoryNodeListHead = NULL;
4746
4747         ObjectInfoCB->Specific.Directory.DirectoryNodeListTail = NULL;
4748
4749         ObjectInfoCB->Specific.Directory.DirectoryNodeCount = 0;
4750
4751         AFSDbgLogMsg( AFS_SUBSYSTEM_DIR_NODE_COUNT,
4752                       AFS_TRACE_LEVEL_VERBOSE,
4753                       "AFSResetDirectoryContent Reset count to 0 on parent FID %08lX-%08lX-%08lX-%08lX\n",
4754                       ObjectInfoCB->FileId.Cell,
4755                       ObjectInfoCB->FileId.Volume,
4756                       ObjectInfoCB->FileId.Vnode,
4757                       ObjectInfoCB->FileId.Unique);
4758     }
4759
4760     return ntStatus;
4761 }
4762
4763 NTSTATUS
4764 AFSEnumerateGlobalRoot( IN GUID *AuthGroup)
4765 {
4766
4767     NTSTATUS ntStatus = STATUS_SUCCESS;
4768     AFSDirectoryCB *pDirGlobalDirNode = NULL;
4769     UNICODE_STRING uniFullName;
4770
4771     __Enter
4772     {
4773
4774         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
4775                       AFS_TRACE_LEVEL_VERBOSE,
4776                       "AFSEnumerateGlobalRoot Acquiring GlobalRoot DirectoryNodeHdr.TreeLock lock %08lX EXCL %08lX\n",
4777                       AFSGlobalRoot->ObjectInformation.Specific.Directory.DirectoryNodeHdr.TreeLock,
4778                       PsGetCurrentThread());
4779
4780         AFSAcquireExcl( AFSGlobalRoot->ObjectInformation.Specific.Directory.DirectoryNodeHdr.TreeLock,
4781                         TRUE);
4782
4783         if( BooleanFlagOn( AFSGlobalRoot->ObjectInformation.Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED))
4784         {
4785
4786             try_return( ntStatus);
4787         }
4788
4789         //
4790         // Initialize the root information
4791         //
4792
4793         AFSGlobalRoot->ObjectInformation.Specific.Directory.DirectoryNodeHdr.ContentIndex = 1;
4794
4795         //
4796         // Enumerate the shares in the volume
4797         //
4798
4799         ntStatus = AFSEnumerateDirectory( AuthGroup,
4800                                           &AFSGlobalRoot->ObjectInformation,
4801                                           TRUE);
4802
4803         if( !NT_SUCCESS( ntStatus))
4804         {
4805
4806             try_return( ntStatus);
4807         }
4808
4809         pDirGlobalDirNode = AFSGlobalRoot->ObjectInformation.Specific.Directory.DirectoryNodeListHead;
4810
4811         //
4812         // Indicate the node is initialized
4813         //
4814
4815         SetFlag( AFSGlobalRoot->ObjectInformation.Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED);
4816
4817         uniFullName.MaximumLength = PAGE_SIZE;
4818         uniFullName.Length = 0;
4819
4820         uniFullName.Buffer = (WCHAR *)AFSLibExAllocatePoolWithTag( PagedPool,
4821                                                                    uniFullName.MaximumLength,
4822                                                                    AFS_GENERIC_MEMORY_12_TAG);
4823
4824         if( uniFullName.Buffer == NULL)
4825         {
4826
4827             //
4828             // Reset the directory content
4829             //
4830
4831             AFSResetDirectoryContent( &AFSGlobalRoot->ObjectInformation);
4832
4833             ClearFlag( AFSGlobalRoot->ObjectInformation.Flags, AFS_OBJECT_FLAGS_DIRECTORY_ENUMERATED);
4834
4835             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4836         }
4837
4838         //
4839         // Populate our list of entries in the NP enumeration list
4840         //
4841
4842         while( pDirGlobalDirNode != NULL)
4843         {
4844
4845             uniFullName.Buffer[ 0] = L'\\';
4846             uniFullName.Buffer[ 1] = L'\\';
4847
4848             uniFullName.Length = 2 * sizeof( WCHAR);
4849
4850             RtlCopyMemory( &uniFullName.Buffer[ 2],
4851                            AFSServerName.Buffer,
4852                            AFSServerName.Length);
4853
4854             uniFullName.Length += AFSServerName.Length;
4855
4856             uniFullName.Buffer[ uniFullName.Length/sizeof( WCHAR)] = L'\\';
4857
4858             uniFullName.Length += sizeof( WCHAR);
4859
4860             RtlCopyMemory( &uniFullName.Buffer[ uniFullName.Length/sizeof( WCHAR)],
4861                            pDirGlobalDirNode->NameInformation.FileName.Buffer,
4862                            pDirGlobalDirNode->NameInformation.FileName.Length);
4863
4864             uniFullName.Length += pDirGlobalDirNode->NameInformation.FileName.Length;
4865
4866             AFSAddConnectionEx( &uniFullName,
4867                                 RESOURCEDISPLAYTYPE_SHARE,
4868                                 0);
4869
4870             pDirGlobalDirNode = (AFSDirectoryCB *)pDirGlobalDirNode->ListEntry.fLink;
4871         }
4872
4873         AFSExFreePool( uniFullName.Buffer);
4874
4875 try_exit:
4876
4877         AFSReleaseResource( AFSGlobalRoot->ObjectInformation.Specific.Directory.DirectoryNodeHdr.TreeLock);
4878     }
4879
4880     return ntStatus;
4881 }
4882
4883 BOOLEAN
4884 AFSIsRelativeName( IN UNICODE_STRING *Name)
4885 {
4886
4887     BOOLEAN bIsRelative = FALSE;
4888
4889     if( Name->Buffer[ 0] != L'\\')
4890     {
4891
4892         bIsRelative = TRUE;
4893     }
4894
4895     return bIsRelative;
4896 }
4897
4898 void
4899 AFSUpdateName( IN UNICODE_STRING *Name)
4900 {
4901
4902     USHORT usIndex = 0;
4903
4904     while( usIndex < Name->Length/sizeof( WCHAR))
4905     {
4906
4907         if( Name->Buffer[ usIndex] == L'/')
4908         {
4909
4910             Name->Buffer[ usIndex] = L'\\';
4911         }
4912
4913         usIndex++;
4914     }
4915
4916     return;
4917 }
4918
4919 NTSTATUS
4920 AFSUpdateTargetName( IN OUT UNICODE_STRING *TargetName,
4921                      IN OUT ULONG *Flags,
4922                      IN WCHAR *NameBuffer,
4923                      IN USHORT NameLength)
4924 {
4925
4926     NTSTATUS ntStatus = STATUS_SUCCESS;
4927     WCHAR *pTmpBuffer = NULL;
4928
4929     __Enter
4930     {
4931
4932         //
4933         // If we have enough space then just move in the name otherwise
4934         // allocate a new buffer
4935         //
4936
4937         if( TargetName->Length < NameLength)
4938         {
4939
4940             pTmpBuffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
4941                                                             NameLength,
4942                                                             AFS_NAME_BUFFER_FIVE_TAG);
4943
4944             if( pTmpBuffer == NULL)
4945             {
4946
4947                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
4948             }
4949
4950             if( BooleanFlagOn( *Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER))
4951             {
4952
4953                 AFSExFreePool( TargetName->Buffer);
4954             }
4955
4956             TargetName->MaximumLength = NameLength;
4957
4958             TargetName->Buffer = pTmpBuffer;
4959
4960             SetFlag( *Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER);
4961         }
4962
4963         TargetName->Length = NameLength;
4964
4965         RtlCopyMemory( TargetName->Buffer,
4966                        NameBuffer,
4967                        TargetName->Length);
4968
4969         //
4970         // Update the name in the buffer
4971         //
4972
4973         AFSUpdateName( TargetName);
4974
4975 try_exit:
4976
4977         NOTHING;
4978     }
4979
4980     return ntStatus;
4981 }
4982
4983 AFSNameArrayHdr *
4984 AFSInitNameArray( IN AFSDirectoryCB *DirectoryCB,
4985                   IN ULONG InitialElementCount)
4986 {
4987
4988     AFSNameArrayHdr *pNameArray = NULL;
4989     AFSNameArrayCB *pCurrentElement = NULL;
4990     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
4991     LONG lCount;
4992
4993     __Enter
4994     {
4995
4996         if( InitialElementCount == 0)
4997         {
4998
4999             InitialElementCount = pDevExt->Specific.RDR.NameArrayLength;
5000         }
5001
5002         pNameArray = (AFSNameArrayHdr *)AFSExAllocatePoolWithTag( PagedPool,
5003                                                                   sizeof( AFSNameArrayHdr) +
5004                                                                     (InitialElementCount * sizeof( AFSNameArrayCB)),
5005                                                                   AFS_NAME_ARRAY_TAG);
5006
5007         if( pNameArray == NULL)
5008         {
5009
5010             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5011                           AFS_TRACE_LEVEL_ERROR,
5012                           "AFSInitNameArray Failed to allocate name array\n");
5013
5014             try_return( pNameArray);
5015         }
5016
5017         RtlZeroMemory( pNameArray,
5018                        sizeof( AFSNameArrayHdr) +
5019                           (InitialElementCount * sizeof( AFSNameArrayCB)));
5020
5021         pNameArray->MaxElementCount = InitialElementCount;
5022
5023         if( DirectoryCB != NULL)
5024         {
5025
5026             pCurrentElement = &pNameArray->ElementArray[ 0];
5027
5028             pNameArray->CurrentEntry = pCurrentElement;
5029
5030             pNameArray->Count = 1;
5031
5032             pNameArray->LinkCount = 0;
5033
5034             lCount = InterlockedIncrement( &DirectoryCB->OpenReferenceCount);
5035
5036             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5037                           AFS_TRACE_LEVEL_VERBOSE,
5038                           "AFSInitNameArray Increment count on %wZ DE %p Cnt %d\n",
5039                           &DirectoryCB->NameInformation.FileName,
5040                           DirectoryCB,
5041                           lCount);
5042
5043             pCurrentElement->DirectoryCB = DirectoryCB;
5044
5045             pCurrentElement->Component = DirectoryCB->NameInformation.FileName;
5046
5047             pCurrentElement->FileId = DirectoryCB->ObjectInformation->FileId;
5048
5049             if( pCurrentElement->FileId.Vnode == 1)
5050             {
5051
5052                 SetFlag( pCurrentElement->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5053             }
5054
5055             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5056                           AFS_TRACE_LEVEL_VERBOSE,
5057                           "AFSInitNameArray [NA:%p] Element[0] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5058                           pNameArray,
5059                           pCurrentElement->DirectoryCB,
5060                           pCurrentElement->FileId.Cell,
5061                           pCurrentElement->FileId.Volume,
5062                           pCurrentElement->FileId.Vnode,
5063                           pCurrentElement->FileId.Unique,
5064                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5065                           pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5066         }
5067
5068 try_exit:
5069
5070         NOTHING;
5071     }
5072
5073     return pNameArray;
5074 }
5075
5076 NTSTATUS
5077 AFSPopulateNameArray( IN AFSNameArrayHdr *NameArray,
5078                       IN UNICODE_STRING *Path,
5079                       IN AFSDirectoryCB *DirectoryCB)
5080 {
5081
5082     NTSTATUS ntStatus = STATUS_SUCCESS;
5083     AFSNameArrayCB *pCurrentElement = NULL;
5084     UNICODE_STRING uniComponentName, uniRemainingPath;
5085     AFSObjectInfoCB *pCurrentObject = NULL;
5086     ULONG  ulTotalCount = 0;
5087     ULONG ulIndex = 0;
5088     USHORT usLength = 0;
5089     LONG lCount;
5090
5091     __Enter
5092     {
5093
5094         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5095                       AFS_TRACE_LEVEL_VERBOSE,
5096                       "AFSPopulateNameArray [NA:%p] passed Path %wZ DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5097                       NameArray,
5098                       &Path,
5099                       DirectoryCB,
5100                       DirectoryCB->ObjectInformation->FileId.Cell,
5101                       DirectoryCB->ObjectInformation->FileId.Volume,
5102                       DirectoryCB->ObjectInformation->FileId.Vnode,
5103                       DirectoryCB->ObjectInformation->FileId.Unique,
5104                       &DirectoryCB->NameInformation.FileName,
5105                       DirectoryCB->ObjectInformation->FileType);
5106
5107         //
5108         // Init some info in the header
5109         //
5110
5111         pCurrentElement = &NameArray->ElementArray[ 0];
5112
5113         NameArray->CurrentEntry = pCurrentElement;
5114
5115         //
5116         // The first entry points at the root
5117         //
5118
5119         pCurrentElement->DirectoryCB = DirectoryCB->ObjectInformation->VolumeCB->DirectoryCB;
5120
5121         lCount = InterlockedIncrement( &pCurrentElement->DirectoryCB->OpenReferenceCount);
5122
5123         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5124                       AFS_TRACE_LEVEL_VERBOSE,
5125                       "AFSPopulateNameArray Increment count on volume %wZ DE %p Cnt %d\n",
5126                       &pCurrentElement->DirectoryCB->NameInformation.FileName,
5127                       pCurrentElement->DirectoryCB,
5128                       lCount);
5129
5130         pCurrentElement->Component = DirectoryCB->ObjectInformation->VolumeCB->DirectoryCB->NameInformation.FileName;
5131
5132         pCurrentElement->FileId = DirectoryCB->ObjectInformation->VolumeCB->ObjectInformation.FileId;
5133
5134         pCurrentElement->Flags = 0;
5135
5136         if( pCurrentElement->FileId.Vnode == 1)
5137         {
5138
5139             SetFlag( pCurrentElement->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5140         }
5141
5142         NameArray->Count = 1;
5143
5144         NameArray->LinkCount = 0;
5145
5146         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5147                       AFS_TRACE_LEVEL_VERBOSE,
5148                       "AFSPopulateNameArray [NA:%p] Element[0] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5149                       NameArray,
5150                       pCurrentElement->DirectoryCB,
5151                       pCurrentElement->FileId.Cell,
5152                       pCurrentElement->FileId.Volume,
5153                       pCurrentElement->FileId.Vnode,
5154                       pCurrentElement->FileId.Unique,
5155                       &pCurrentElement->DirectoryCB->NameInformation.FileName,
5156                       pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5157
5158         //
5159         // If the root is the parent then we are done ...
5160         //
5161
5162         if( &DirectoryCB->ObjectInformation->VolumeCB->ObjectInformation == DirectoryCB->ObjectInformation)
5163         {
5164             try_return( ntStatus);
5165         }
5166
5167 try_exit:
5168
5169         NOTHING;
5170     }
5171
5172     return ntStatus;
5173 }
5174
5175 NTSTATUS
5176 AFSPopulateNameArrayFromRelatedArray( IN AFSNameArrayHdr *NameArray,
5177                                       IN AFSNameArrayHdr *RelatedNameArray,
5178                                       IN AFSDirectoryCB *DirectoryCB)
5179 {
5180
5181     NTSTATUS ntStatus = STATUS_SUCCESS;
5182     AFSNameArrayCB *pCurrentElement = NULL, *pCurrentRelatedElement = NULL;
5183     UNICODE_STRING uniComponentName, uniRemainingPath;
5184     AFSObjectInfoCB *pObjectInfo = NULL;
5185     ULONG  ulTotalCount = 0;
5186     ULONG ulIndex = 0;
5187     USHORT usLength = 0;
5188     LONG lCount;
5189
5190     __Enter
5191     {
5192
5193         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5194                       AFS_TRACE_LEVEL_VERBOSE,
5195                       "AFSPopulateNameArray [NA:%p] passed RelatedNameArray %p DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5196                       NameArray,
5197                       RelatedNameArray,
5198                       DirectoryCB,
5199                       DirectoryCB->ObjectInformation->FileId.Cell,
5200                       DirectoryCB->ObjectInformation->FileId.Volume,
5201                       DirectoryCB->ObjectInformation->FileId.Vnode,
5202                       DirectoryCB->ObjectInformation->FileId.Unique,
5203                       &DirectoryCB->NameInformation.FileName,
5204                       DirectoryCB->ObjectInformation->FileType);
5205
5206         //
5207         // Init some info in the header
5208         //
5209
5210         pCurrentElement = &NameArray->ElementArray[ 0];
5211
5212         pCurrentRelatedElement = &RelatedNameArray->ElementArray[ 0];
5213
5214         NameArray->Count = 0;
5215
5216         NameArray->LinkCount = RelatedNameArray->LinkCount;
5217
5218         //
5219         // Populate the name array with the data from the related array
5220         //
5221
5222         while( TRUE)
5223         {
5224
5225             pCurrentElement->DirectoryCB = pCurrentRelatedElement->DirectoryCB;
5226
5227             pCurrentElement->Component = pCurrentRelatedElement->DirectoryCB->NameInformation.FileName;
5228
5229             pCurrentElement->FileId    = pCurrentElement->DirectoryCB->ObjectInformation->FileId;
5230
5231             pCurrentElement->Flags = 0;
5232
5233             if( pCurrentElement->FileId.Vnode == 1)
5234             {
5235
5236                 SetFlag( pCurrentElement->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5237             }
5238
5239             lCount = InterlockedIncrement( &pCurrentElement->DirectoryCB->OpenReferenceCount);
5240
5241             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5242                           AFS_TRACE_LEVEL_VERBOSE,
5243                           "AFSPopulateNameArrayFromRelatedArray Increment count on %wZ DE %p Cnt %d\n",
5244                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5245                           pCurrentElement->DirectoryCB,
5246                           lCount);
5247
5248             lCount = InterlockedIncrement( &NameArray->Count);
5249
5250             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5251                           AFS_TRACE_LEVEL_VERBOSE,
5252                           "AFSPopulateNameArrayFromRelatedArray [NA:%p] Element[%d] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5253                           NameArray,
5254                           lCount - 1,
5255                           pCurrentElement->DirectoryCB,
5256                           pCurrentElement->FileId.Cell,
5257                           pCurrentElement->FileId.Volume,
5258                           pCurrentElement->FileId.Vnode,
5259                           pCurrentElement->FileId.Unique,
5260                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5261                           pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5262
5263             if( pCurrentElement->DirectoryCB == DirectoryCB ||
5264                 NameArray->Count == RelatedNameArray->Count)
5265             {
5266
5267                 //
5268                 // Done ...
5269                 //
5270
5271                 break;
5272             }
5273
5274             pCurrentElement++;
5275
5276             pCurrentRelatedElement++;
5277         }
5278
5279         NameArray->CurrentEntry = NameArray->Count > 0 ? pCurrentElement : NULL;
5280     }
5281
5282     return ntStatus;
5283 }
5284
5285 NTSTATUS
5286 AFSFreeNameArray( IN AFSNameArrayHdr *NameArray)
5287 {
5288
5289     NTSTATUS ntStatus = STATUS_SUCCESS;
5290     AFSNameArrayCB *pCurrentElement = NULL;
5291     LONG lCount, lElement;
5292
5293     __Enter
5294     {
5295
5296         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5297                       AFS_TRACE_LEVEL_VERBOSE,
5298                       "AFSFreeNameArray [NA:%p]\n",
5299                       NameArray);
5300
5301         for ( lElement = 0; lElement < NameArray->Count; lElement++)
5302         {
5303
5304             pCurrentElement = &NameArray->ElementArray[ lElement];
5305
5306             lCount = InterlockedDecrement( &pCurrentElement->DirectoryCB->OpenReferenceCount);
5307
5308             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5309                           AFS_TRACE_LEVEL_VERBOSE,
5310                           "AFSFreeNameArray Decrement count on %wZ DE %p Cnt %d\n",
5311                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5312                           pCurrentElement->DirectoryCB,
5313                           lCount);
5314         }
5315
5316         AFSExFreePool( NameArray);
5317     }
5318
5319     return ntStatus;
5320 }
5321
5322 NTSTATUS
5323 AFSInsertNextElement( IN AFSNameArrayHdr *NameArray,
5324                       IN AFSDirectoryCB *DirectoryCB)
5325 {
5326
5327     NTSTATUS ntStatus = STATUS_SUCCESS;
5328     AFSNameArrayCB *pCurrentElement = NULL;
5329     LONG lCount;
5330
5331     __Enter
5332     {
5333
5334         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5335                       AFS_TRACE_LEVEL_VERBOSE,
5336                       "AFSInsertNextElement [NA:%p] passed DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5337                       NameArray,
5338                       DirectoryCB,
5339                       DirectoryCB->ObjectInformation->FileId.Cell,
5340                       DirectoryCB->ObjectInformation->FileId.Volume,
5341                       DirectoryCB->ObjectInformation->FileId.Vnode,
5342                       DirectoryCB->ObjectInformation->FileId.Unique,
5343                       &DirectoryCB->NameInformation.FileName,
5344                       DirectoryCB->ObjectInformation->FileType);
5345
5346         if( NameArray->Count == NameArray->MaxElementCount)
5347         {
5348
5349             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5350                           AFS_TRACE_LEVEL_ERROR,
5351                           "AFSInsertNextElement [NA:%p] Name has reached Maximum Size\n",
5352                           NameArray);
5353
5354             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
5355         }
5356
5357         for ( lCount = 0; lCount < NameArray->Count; lCount++)
5358         {
5359
5360             if ( AFSIsEqualFID( &NameArray->ElementArray[ lCount].FileId,
5361                                 &DirectoryCB->ObjectInformation->FileId) )
5362             {
5363
5364                 AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5365                               AFS_TRACE_LEVEL_WARNING,
5366                               "AFSInsertNextElement [NA:%p] DE %p recursion Status %08X\n",
5367                               NameArray,
5368                               DirectoryCB,
5369                               STATUS_ACCESS_DENIED);
5370
5371                 try_return( ntStatus = STATUS_ACCESS_DENIED);
5372             }
5373         }
5374
5375         if( NameArray->Count > 0)
5376         {
5377
5378             NameArray->CurrentEntry++;
5379         }
5380         else
5381         {
5382             NameArray->CurrentEntry = &NameArray->ElementArray[ 0];
5383         }
5384
5385         pCurrentElement = NameArray->CurrentEntry;
5386
5387         lCount = InterlockedIncrement( &NameArray->Count);
5388
5389         lCount = InterlockedIncrement( &DirectoryCB->OpenReferenceCount);
5390
5391         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5392                       AFS_TRACE_LEVEL_VERBOSE,
5393                       "AFSInsertNextElement Increment count on %wZ DE %p Cnt %d\n",
5394                       &DirectoryCB->NameInformation.FileName,
5395                       DirectoryCB,
5396                       lCount);
5397
5398         pCurrentElement->DirectoryCB = DirectoryCB;
5399
5400         pCurrentElement->Component = DirectoryCB->NameInformation.FileName;
5401
5402         pCurrentElement->FileId = DirectoryCB->ObjectInformation->FileId;
5403
5404         pCurrentElement->Flags = 0;
5405
5406         if( pCurrentElement->FileId.Vnode == 1)
5407         {
5408
5409             SetFlag( pCurrentElement->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5410         }
5411
5412         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5413                       AFS_TRACE_LEVEL_VERBOSE,
5414                       "AFSInsertNextElement [NA:%p] Element[%d] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5415                       NameArray,
5416                       NameArray->Count - 1,
5417                       pCurrentElement->DirectoryCB,
5418                       pCurrentElement->FileId.Cell,
5419                       pCurrentElement->FileId.Volume,
5420                       pCurrentElement->FileId.Vnode,
5421                       pCurrentElement->FileId.Unique,
5422                       &pCurrentElement->DirectoryCB->NameInformation.FileName,
5423                       pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5424
5425 try_exit:
5426
5427         NOTHING;
5428     }
5429
5430     return ntStatus;
5431 }
5432
5433 AFSDirectoryCB *
5434 AFSBackupEntry( IN AFSNameArrayHdr *NameArray)
5435 {
5436
5437     AFSDirectoryCB *pDirectoryCB = NULL;
5438     AFSNameArrayCB *pCurrentElement = NULL;
5439     BOOLEAN         bVolumeRoot = FALSE;
5440     LONG lCount;
5441
5442     __Enter
5443     {
5444
5445         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5446                       AFS_TRACE_LEVEL_VERBOSE,
5447                       "AFSBackupEntry [NA:%p]\n",
5448                       NameArray);
5449
5450         if( NameArray->Count == 0)
5451         {
5452
5453             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5454                           AFS_TRACE_LEVEL_ERROR,
5455                           "AFSBackupEntry [NA:%p] No more entries\n",
5456                           NameArray);
5457
5458             try_return( pCurrentElement);
5459         }
5460
5461         lCount = InterlockedDecrement( &NameArray->CurrentEntry->DirectoryCB->OpenReferenceCount);
5462
5463         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5464                       AFS_TRACE_LEVEL_VERBOSE,
5465                       "AFSBackupEntry Decrement count on %wZ DE %p Cnt %d\n",
5466                       &NameArray->CurrentEntry->DirectoryCB->NameInformation.FileName,
5467                       NameArray->CurrentEntry->DirectoryCB,
5468                       lCount);
5469
5470         NameArray->CurrentEntry->DirectoryCB = NULL;
5471
5472         lCount = InterlockedDecrement( &NameArray->Count);
5473
5474         if( lCount == 0)
5475         {
5476             NameArray->CurrentEntry = NULL;
5477
5478             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5479                           AFS_TRACE_LEVEL_ERROR,
5480                           "AFSBackupEntry [NA:%p] No more entries\n",
5481                           NameArray);
5482         }
5483         else
5484         {
5485
5486             bVolumeRoot = BooleanFlagOn( NameArray->CurrentEntry->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5487
5488             NameArray->CurrentEntry--;
5489
5490             pCurrentElement = NameArray->CurrentEntry;
5491
5492             pDirectoryCB = pCurrentElement->DirectoryCB;
5493
5494             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5495                           AFS_TRACE_LEVEL_VERBOSE,
5496                           "AFSBackupEntry [NA:%p] Returning Element[%d] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5497                           NameArray,
5498                           NameArray->Count - 1,
5499                           pCurrentElement->DirectoryCB,
5500                           pCurrentElement->FileId.Cell,
5501                           pCurrentElement->FileId.Volume,
5502                           pCurrentElement->FileId.Vnode,
5503                           pCurrentElement->FileId.Unique,
5504                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5505                           pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5506
5507             //
5508             // If the entry we are removing is a volume root,
5509             // we must remove the mount point entry as well.
5510             // If the NameArray was constructed by checking the
5511             // share name via the service, the name array can
5512             // contain two volume roots in sequence without a
5513             // mount point separating them.
5514             //
5515
5516             if ( bVolumeRoot &&
5517                  !BooleanFlagOn( NameArray->CurrentEntry->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT))
5518             {
5519
5520                 pDirectoryCB = AFSBackupEntry( NameArray);
5521             }
5522         }
5523
5524
5525 try_exit:
5526
5527         NOTHING;
5528     }
5529
5530     return pDirectoryCB;
5531 }
5532
5533 AFSDirectoryCB *
5534 AFSGetParentEntry( IN AFSNameArrayHdr *NameArray)
5535 {
5536
5537     AFSDirectoryCB *pDirEntry = NULL;
5538     AFSNameArrayCB *pElement = NULL;
5539
5540     __Enter
5541     {
5542
5543         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5544                       AFS_TRACE_LEVEL_VERBOSE,
5545                       "AFSGetParentEntry [NA:%p]\n",
5546                       NameArray);
5547
5548         if( NameArray->Count == 0 ||
5549             NameArray->Count == 1)
5550         {
5551
5552             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5553                           AFS_TRACE_LEVEL_ERROR,
5554                           "AFSGetParentEntry [NA:%p] No more entries\n",
5555                           NameArray);
5556
5557             try_return( pDirEntry = NULL);
5558         }
5559
5560         pElement = &NameArray->ElementArray[ NameArray->Count - 2];
5561
5562         pDirEntry = pElement->DirectoryCB;
5563
5564         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5565                       AFS_TRACE_LEVEL_VERBOSE,
5566                       "AFSGetParentEntry [NA:%p] Returning Element[%d] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5567                       NameArray,
5568                       NameArray->Count - 2,
5569                       pElement->DirectoryCB,
5570                       pElement->FileId.Cell,
5571                       pElement->FileId.Volume,
5572                       pElement->FileId.Vnode,
5573                       pElement->FileId.Unique,
5574                       &pElement->DirectoryCB->NameInformation.FileName,
5575                       pElement->DirectoryCB->ObjectInformation->FileType);
5576
5577 try_exit:
5578
5579         NOTHING;
5580     }
5581
5582     return pDirEntry;
5583 }
5584
5585 void
5586 AFSResetNameArray( IN AFSNameArrayHdr *NameArray,
5587                    IN AFSDirectoryCB *DirectoryCB)
5588 {
5589
5590     AFSNameArrayCB *pCurrentElement = NULL;
5591     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
5592     LONG lCount, lElement;
5593
5594     __Enter
5595     {
5596
5597         AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5598                       AFS_TRACE_LEVEL_VERBOSE,
5599                       "AFSResetNameArray [NA:%p] passed DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5600                       NameArray,
5601                       DirectoryCB,
5602                       DirectoryCB->ObjectInformation->FileId.Cell,
5603                       DirectoryCB->ObjectInformation->FileId.Volume,
5604                       DirectoryCB->ObjectInformation->FileId.Vnode,
5605                       DirectoryCB->ObjectInformation->FileId.Unique,
5606                       &DirectoryCB->NameInformation.FileName,
5607                       DirectoryCB->ObjectInformation->FileType);
5608         //
5609         // Dereference previous name array contents
5610         //
5611
5612         for ( lElement = 0; lElement < NameArray->Count; lElement++)
5613         {
5614
5615             pCurrentElement = &NameArray->ElementArray[ lElement];
5616
5617             lCount = InterlockedDecrement( &pCurrentElement->DirectoryCB->OpenReferenceCount);
5618
5619             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5620                           AFS_TRACE_LEVEL_VERBOSE,
5621                           "AFSResetNameArray Decrement count on %wZ DE %p Cnt %d\n",
5622                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5623                           pCurrentElement->DirectoryCB,
5624                           lCount);
5625         }
5626
5627         RtlZeroMemory( NameArray,
5628                        sizeof( AFSNameArrayHdr) +
5629                           ((pDevExt->Specific.RDR.NameArrayLength - 1) * sizeof( AFSNameArrayCB)));
5630
5631         NameArray->MaxElementCount = pDevExt->Specific.RDR.NameArrayLength;
5632
5633         if( DirectoryCB != NULL)
5634         {
5635
5636             pCurrentElement = &NameArray->ElementArray[ 0];
5637
5638             NameArray->CurrentEntry = pCurrentElement;
5639
5640             NameArray->Count = 1;
5641
5642             NameArray->LinkCount = 0;
5643
5644             lCount = InterlockedIncrement( &DirectoryCB->OpenReferenceCount);
5645
5646             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
5647                           AFS_TRACE_LEVEL_VERBOSE,
5648                           "AFSResetNameArray Increment count on %wZ DE %p Cnt %d\n",
5649                           &DirectoryCB->NameInformation.FileName,
5650                           DirectoryCB,
5651                           lCount);
5652
5653             pCurrentElement->DirectoryCB = DirectoryCB;
5654
5655             pCurrentElement->Component = DirectoryCB->NameInformation.FileName;
5656
5657             pCurrentElement->FileId = DirectoryCB->ObjectInformation->FileId;
5658
5659             pCurrentElement->Flags  = 0;
5660
5661             if( pCurrentElement->FileId.Vnode == 1)
5662             {
5663
5664                 SetFlag( pCurrentElement->Flags, AFS_NAME_ARRAY_FLAG_ROOT_ELEMENT);
5665             }
5666
5667             AFSDbgLogMsg( AFS_SUBSYSTEM_NAME_ARRAY_PROCESSING,
5668                           AFS_TRACE_LEVEL_VERBOSE,
5669                           "AFSResetNameArray [NA:%p] Element[0] DE %p FID %08lX-%08lX-%08lX-%08lX %wZ Type %d\n",
5670                           NameArray,
5671                           pCurrentElement->DirectoryCB,
5672                           pCurrentElement->FileId.Cell,
5673                           pCurrentElement->FileId.Volume,
5674                           pCurrentElement->FileId.Vnode,
5675                           pCurrentElement->FileId.Unique,
5676                           &pCurrentElement->DirectoryCB->NameInformation.FileName,
5677                           pCurrentElement->DirectoryCB->ObjectInformation->FileType);
5678         }
5679     }
5680
5681     return;
5682 }
5683
5684 void
5685 AFSDumpNameArray( IN AFSNameArrayHdr *NameArray)
5686 {
5687
5688     AFSNameArrayCB *pCurrentElement = NULL;
5689
5690     pCurrentElement = &NameArray->ElementArray[ 0];
5691
5692     AFSPrint("AFSDumpNameArray Start (%d)\n", NameArray->Count);
5693
5694     while( pCurrentElement->DirectoryCB != NULL)
5695     {
5696
5697         AFSPrint("FID %08lX-%08lX-%08lX-%08lX %wZ\n",
5698                   pCurrentElement->FileId.Cell,
5699                   pCurrentElement->FileId.Volume,
5700                   pCurrentElement->FileId.Vnode,
5701                   pCurrentElement->FileId.Unique,
5702                   &pCurrentElement->DirectoryCB->NameInformation.FileName);
5703
5704         pCurrentElement++;
5705     }
5706
5707     AFSPrint("AFSDumpNameArray End\n\n");
5708
5709     return;
5710 }
5711
5712 void
5713 AFSSetEnumerationEvent( IN AFSFcb *Fcb)
5714 {
5715     LONG lCount;
5716
5717     //
5718     // Depending on the type of node, set the event
5719     //
5720
5721     switch( Fcb->Header.NodeTypeCode)
5722     {
5723
5724         case AFS_DIRECTORY_FCB:
5725         case AFS_ROOT_FCB:
5726         case AFS_ROOT_ALL:
5727         {
5728
5729             lCount = InterlockedIncrement( &Fcb->NPFcb->Specific.Directory.DirectoryEnumCount);
5730
5731             break;
5732         }
5733     }
5734
5735     return;
5736 }
5737
5738 void
5739 AFSClearEnumerationEvent( IN AFSFcb *Fcb)
5740 {
5741
5742     LONG lCount;
5743
5744     //
5745     // Depending on the type of node, set the event
5746     //
5747
5748     switch( Fcb->Header.NodeTypeCode)
5749     {
5750
5751         case AFS_DIRECTORY_FCB:
5752         case AFS_ROOT_FCB:
5753         case AFS_ROOT_ALL:
5754         {
5755
5756             ASSERT( Fcb->NPFcb->Specific.Directory.DirectoryEnumCount > 0);
5757
5758             lCount = InterlockedDecrement( &Fcb->NPFcb->Specific.Directory.DirectoryEnumCount);
5759
5760             break;
5761         }
5762     }
5763
5764     return;
5765 }
5766
5767 BOOLEAN
5768 AFSIsEnumerationInProcess( IN AFSObjectInfoCB *ObjectInfo)
5769 {
5770
5771     BOOLEAN bIsInProcess = FALSE;
5772
5773     __Enter
5774     {
5775
5776         if( ObjectInfo->Fcb == NULL)
5777         {
5778
5779             try_return( bIsInProcess);
5780         }
5781
5782         switch( ObjectInfo->Fcb->Header.NodeTypeCode)
5783         {
5784
5785             case AFS_DIRECTORY_FCB:
5786             case AFS_ROOT_FCB:
5787             case AFS_ROOT_ALL:
5788             {
5789
5790                 if( ObjectInfo->Fcb->NPFcb->Specific.Directory.DirectoryEnumCount > 0)
5791                 {
5792
5793                     bIsInProcess = TRUE;
5794                 }
5795
5796                 break;
5797             }
5798         }
5799
5800 try_exit:
5801
5802         NOTHING;
5803     }
5804
5805     return bIsInProcess;
5806 }
5807
5808 NTSTATUS
5809 AFSVerifyVolume( IN ULONGLONG ProcessId,
5810                  IN AFSVolumeCB *VolumeCB)
5811 {
5812
5813     NTSTATUS ntStatus = STATUS_SUCCESS;
5814
5815
5816     return ntStatus;
5817 }
5818
5819 NTSTATUS
5820 AFSInitPIOCtlDirectoryCB( IN AFSObjectInfoCB *ObjectInfo)
5821 {
5822
5823     NTSTATUS ntStatus = STATUS_SUCCESS;
5824     AFSObjectInfoCB *pObjectInfoCB = NULL;
5825     AFSDirectoryCB *pDirNode = NULL;
5826     ULONG ulEntryLength = 0;
5827     AFSNonPagedDirectoryCB *pNonPagedDirEntry = NULL;
5828     LONG lCount;
5829
5830     __Enter
5831     {
5832
5833         pObjectInfoCB = AFSAllocateObjectInfo( ObjectInfo,
5834                                                0);
5835
5836         if( pObjectInfoCB == NULL)
5837         {
5838
5839             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
5840         }
5841
5842         AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
5843                       AFS_TRACE_LEVEL_VERBOSE,
5844                       "AFSInitPIOCtlDirectoryCB Initializing count (1) on object %08lX\n",
5845                       pObjectInfoCB);
5846
5847         pObjectInfoCB->ObjectReferenceCount = 1;
5848
5849         pObjectInfoCB->FileType = AFS_FILE_TYPE_PIOCTL;
5850
5851         pObjectInfoCB->FileAttributes = FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM;
5852
5853         ulEntryLength = sizeof( AFSDirectoryCB) + AFSPIOCtlName.Length;
5854
5855         pDirNode = (AFSDirectoryCB *)AFSExAllocatePoolWithTag( PagedPool,
5856                                                                ulEntryLength,
5857                                                                AFS_DIR_ENTRY_TAG);
5858
5859         if( pDirNode == NULL)
5860         {
5861
5862             AFSDeleteObjectInfo( pObjectInfoCB);
5863
5864             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
5865         }
5866
5867         pNonPagedDirEntry = (AFSNonPagedDirectoryCB *)AFSExAllocatePoolWithTag( NonPagedPool,
5868                                                                                 sizeof( AFSNonPagedDirectoryCB),
5869                                                                                 AFS_DIR_ENTRY_NP_TAG);
5870
5871         if( pNonPagedDirEntry == NULL)
5872         {
5873
5874             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
5875         }
5876
5877         RtlZeroMemory( pDirNode,
5878                        ulEntryLength);
5879
5880         RtlZeroMemory( pNonPagedDirEntry,
5881                        sizeof( AFSNonPagedDirectoryCB));
5882
5883         ExInitializeResourceLite( &pNonPagedDirEntry->Lock);
5884
5885         pDirNode->NonPaged = pNonPagedDirEntry;
5886
5887         pDirNode->ObjectInformation = pObjectInfoCB;
5888
5889         pDirNode->FileIndex = (ULONG)AFS_DIR_ENTRY_PIOCTL_INDEX;
5890
5891         //
5892         // Set valid entry
5893         //
5894
5895         SetFlag( pDirNode->Flags, AFS_DIR_ENTRY_VALID | AFS_DIR_ENTRY_FAKE);
5896
5897         pDirNode->NameInformation.FileName.Length = AFSPIOCtlName.Length;
5898
5899         pDirNode->NameInformation.FileName.MaximumLength = AFSPIOCtlName.Length;
5900
5901         pDirNode->NameInformation.FileName.Buffer = (WCHAR *)((char *)pDirNode + sizeof( AFSDirectoryCB));
5902
5903         RtlCopyMemory( pDirNode->NameInformation.FileName.Buffer,
5904                        AFSPIOCtlName.Buffer,
5905                        pDirNode->NameInformation.FileName.Length);
5906
5907         pDirNode->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pDirNode->NameInformation.FileName,
5908                                                                        TRUE);
5909
5910         if ( InterlockedCompareExchangePointer( (PVOID *)&ObjectInfo->Specific.Directory.PIOCtlDirectoryCB, pDirNode, NULL) != NULL)
5911         {
5912
5913             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
5914                           AFS_TRACE_LEVEL_WARNING,
5915                           "AFSInitPIOCtlDirectoryCB Raced PIOCtlDirectoryCB %08lX pFcb %08lX\n",
5916                           ObjectInfo->Specific.Directory.PIOCtlDirectoryCB,
5917                           pDirNode);
5918
5919             //
5920             // Increment the open reference and handle on the node
5921             //
5922
5923             lCount = InterlockedIncrement( &pDirNode->ObjectInformation->ObjectReferenceCount);
5924
5925             AFSDbgLogMsg( AFS_SUBSYSTEM_FCB_REF_COUNTING,
5926                           AFS_TRACE_LEVEL_VERBOSE,
5927                           "AFSInitPIOCtlDirectoryCB Increment count on Object %08lX Cnt %d\n",
5928                           pDirNode->ObjectInformation,
5929                           lCount);
5930
5931             try_return( ntStatus = STATUS_REPARSE);
5932         }
5933
5934 try_exit:
5935
5936         if ( ntStatus != STATUS_SUCCESS)
5937         {
5938
5939             if ( pDirNode != NULL)
5940             {
5941
5942                 AFSExFreePool( pDirNode);
5943             }
5944
5945             if ( pObjectInfoCB != NULL)
5946             {
5947
5948                 AFSDeleteObjectInfo( pObjectInfoCB);
5949             }
5950         }
5951     }
5952
5953     return ntStatus;
5954 }
5955
5956 NTSTATUS
5957 AFSRetrieveFileAttributes( IN AFSDirectoryCB *ParentDirectoryCB,
5958                            IN AFSDirectoryCB *DirectoryCB,
5959                            IN UNICODE_STRING *ParentPathName,
5960                            IN AFSNameArrayHdr *RelatedNameArray,
5961                            IN GUID           *AuthGroup,
5962                            OUT AFSFileInfoCB *FileInfo)
5963 {
5964
5965     NTSTATUS ntStatus = STATUS_SUCCESS;
5966     AFSDirEnumEntry *pDirEntry = NULL, *pLastDirEntry = NULL;
5967     UNICODE_STRING uniFullPathName;
5968     AFSNameArrayHdr    *pNameArray = NULL;
5969     AFSVolumeCB *pVolumeCB = NULL;
5970     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
5971     WCHAR *pwchBuffer = NULL;
5972     UNICODE_STRING uniComponentName, uniRemainingPath, uniParsedName;
5973     ULONG ulNameDifference = 0;
5974     LONG lCount;
5975
5976     __Enter
5977     {
5978
5979         //
5980         // Retrieve a target name for the entry
5981         //
5982
5983         AFSAcquireShared( &DirectoryCB->NonPaged->Lock,
5984                           TRUE);
5985
5986         if( DirectoryCB->NameInformation.TargetName.Length == 0)
5987         {
5988
5989             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
5990
5991             ntStatus = AFSEvaluateTargetByID( DirectoryCB->ObjectInformation,
5992                                               AuthGroup,
5993                                               FALSE,
5994                                               &pDirEntry);
5995
5996             if( !NT_SUCCESS( ntStatus) ||
5997                 pDirEntry->TargetNameLength == 0)
5998             {
5999
6000                 if( pDirEntry != NULL)
6001                 {
6002
6003                     ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
6004                 }
6005
6006                 try_return( ntStatus);
6007             }
6008
6009             AFSAcquireExcl( &DirectoryCB->NonPaged->Lock,
6010                             TRUE);
6011
6012             if( DirectoryCB->NameInformation.TargetName.Length == 0)
6013             {
6014
6015                 //
6016                 // Update the target name
6017                 //
6018
6019                 ntStatus = AFSUpdateTargetName( &DirectoryCB->NameInformation.TargetName,
6020                                                 &DirectoryCB->Flags,
6021                                                 (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset),
6022                                                 (USHORT)pDirEntry->TargetNameLength);
6023
6024                 if( !NT_SUCCESS( ntStatus))
6025                 {
6026
6027                     AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6028
6029                     try_return( ntStatus);
6030                 }
6031             }
6032
6033             AFSConvertToShared( &DirectoryCB->NonPaged->Lock);
6034         }
6035
6036         //
6037         // Need to pass the full path in for parsing.
6038         //
6039
6040         if( AFSIsRelativeName( &DirectoryCB->NameInformation.TargetName))
6041         {
6042
6043             uniFullPathName.Length = 0;
6044             uniFullPathName.MaximumLength = ParentPathName->Length +
6045                                                     sizeof( WCHAR) +
6046                                                     DirectoryCB->NameInformation.TargetName.Length;
6047
6048             uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6049                                                                         uniFullPathName.MaximumLength,
6050                                                                         AFS_NAME_BUFFER_SIX_TAG);
6051
6052             if( uniFullPathName.Buffer == NULL)
6053             {
6054
6055                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6056
6057                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6058             }
6059
6060             pwchBuffer = uniFullPathName.Buffer;
6061
6062             RtlZeroMemory( uniFullPathName.Buffer,
6063                            uniFullPathName.MaximumLength);
6064
6065             RtlCopyMemory( uniFullPathName.Buffer,
6066                            ParentPathName->Buffer,
6067                            ParentPathName->Length);
6068
6069             uniFullPathName.Length = ParentPathName->Length;
6070
6071             if( uniFullPathName.Buffer[ (uniFullPathName.Length/sizeof( WCHAR)) - 1] != L'\\' &&
6072                 DirectoryCB->NameInformation.TargetName.Buffer[ 0] != L'\\')
6073             {
6074
6075                 uniFullPathName.Buffer[ uniFullPathName.Length/sizeof( WCHAR)] = L'\\';
6076
6077                 uniFullPathName.Length += sizeof( WCHAR);
6078             }
6079
6080             RtlCopyMemory( &uniFullPathName.Buffer[ uniFullPathName.Length/sizeof( WCHAR)],
6081                            DirectoryCB->NameInformation.TargetName.Buffer,
6082                            DirectoryCB->NameInformation.TargetName.Length);
6083
6084             uniFullPathName.Length += DirectoryCB->NameInformation.TargetName.Length;
6085
6086             uniParsedName.Length = uniFullPathName.Length - ParentPathName->Length;
6087             uniParsedName.MaximumLength = uniParsedName.Length;
6088
6089             uniParsedName.Buffer = &uniFullPathName.Buffer[ ParentPathName->Length/sizeof( WCHAR)];
6090
6091             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6092
6093             //
6094             // We populate up to the current parent
6095             //
6096
6097             if( RelatedNameArray != NULL)
6098             {
6099
6100                 pNameArray = AFSInitNameArray( NULL,
6101                                                RelatedNameArray->MaxElementCount);
6102
6103                 if( pNameArray == NULL)
6104                 {
6105
6106                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6107                 }
6108
6109                 ntStatus = AFSPopulateNameArrayFromRelatedArray( pNameArray,
6110                                                                  RelatedNameArray,
6111                                                                  ParentDirectoryCB);
6112             }
6113             else
6114             {
6115
6116                 pNameArray = AFSInitNameArray( NULL,
6117                                                0);
6118
6119                 if( pNameArray == NULL)
6120                 {
6121
6122                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6123                 }
6124
6125                 ntStatus = AFSPopulateNameArray( pNameArray,
6126                                                  NULL,
6127                                                  ParentDirectoryCB);
6128             }
6129
6130             if( !NT_SUCCESS( ntStatus))
6131             {
6132
6133                 try_return( ntStatus);
6134             }
6135
6136             pVolumeCB = ParentDirectoryCB->ObjectInformation->VolumeCB;
6137
6138             pParentDirEntry = ParentDirectoryCB;
6139         }
6140         else
6141         {
6142
6143             uniFullPathName.Length = 0;
6144             uniFullPathName.MaximumLength = DirectoryCB->NameInformation.TargetName.Length;
6145
6146             uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6147                                                                         uniFullPathName.MaximumLength,
6148                                                                         AFS_NAME_BUFFER_SEVEN_TAG);
6149
6150             if( uniFullPathName.Buffer == NULL)
6151             {
6152
6153                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6154
6155                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6156             }
6157
6158             pwchBuffer = uniFullPathName.Buffer;
6159
6160             RtlZeroMemory( uniFullPathName.Buffer,
6161                            uniFullPathName.MaximumLength);
6162
6163             RtlCopyMemory( uniFullPathName.Buffer,
6164                            DirectoryCB->NameInformation.TargetName.Buffer,
6165                            DirectoryCB->NameInformation.TargetName.Length);
6166
6167             uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6168
6169             //
6170             // This name should begin with the \afs server so parse it off and check it
6171             //
6172
6173             FsRtlDissectName( uniFullPathName,
6174                               &uniComponentName,
6175                               &uniRemainingPath);
6176
6177             if( RtlCompareUnicodeString( &uniComponentName,
6178                                          &AFSServerName,
6179                                          TRUE) != 0)
6180             {
6181
6182                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6183
6184                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
6185                               AFS_TRACE_LEVEL_ERROR,
6186                               "AFSRetrieveFileAttributes Name %wZ contains invalid server name\n",
6187                               &uniFullPathName);
6188
6189                 try_return( ntStatus = STATUS_OBJECT_PATH_INVALID);
6190             }
6191
6192             uniFullPathName = uniRemainingPath;
6193
6194             uniParsedName = uniFullPathName;
6195
6196             ulNameDifference = (ULONG)(uniFullPathName.Length > 0 ? ((char *)uniFullPathName.Buffer - (char *)pwchBuffer) : 0);
6197
6198             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6199
6200             //
6201             // Our name array
6202             //
6203
6204             pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
6205                                            0);
6206
6207             if( pNameArray == NULL)
6208             {
6209
6210                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6211             }
6212
6213             pVolumeCB = AFSGlobalRoot;
6214
6215             pParentDirEntry = AFSGlobalRoot->DirectoryCB;
6216         }
6217
6218         //
6219         // Increment the ref count on the volume and dir entry for correct processing below
6220         //
6221
6222         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
6223
6224         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6225                       AFS_TRACE_LEVEL_VERBOSE,
6226                       "AFSRetrieveFileAttributes Increment count on volume %08lX Cnt %d\n",
6227                       pVolumeCB,
6228                       lCount);
6229
6230         lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
6231
6232         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6233                       AFS_TRACE_LEVEL_VERBOSE,
6234                       "AFSRetrieveFileAttributes Increment count on %wZ DE %p Ccb %p Cnt %d\n",
6235                       &pParentDirEntry->NameInformation.FileName,
6236                       pParentDirEntry,
6237                       NULL,
6238                       lCount);
6239
6240         ntStatus = AFSLocateNameEntry( NULL,
6241                                        NULL,
6242                                        &uniFullPathName,
6243                                        &uniParsedName,
6244                                        pNameArray,
6245                                        AFS_LOCATE_FLAGS_NO_MP_TARGET_EVAL,
6246                                        &pVolumeCB,
6247                                        &pParentDirEntry,
6248                                        &pDirectoryEntry,
6249                                        NULL);
6250
6251         if( !NT_SUCCESS( ntStatus))
6252         {
6253
6254             //
6255             // The volume lock was released on failure above
6256             // Except for STATUS_OBJECT_NAME_NOT_FOUND
6257             //
6258
6259             if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
6260             {
6261
6262                 if( pVolumeCB != NULL)
6263                 {
6264
6265                     lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6266
6267                     AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6268                                   AFS_TRACE_LEVEL_VERBOSE,
6269                                   "AFSRetrieveFileAttributes Decrement count on volume %08lX Cnt %d\n",
6270                                   pVolumeCB,
6271                                   lCount);
6272                 }
6273
6274                 if( pDirectoryEntry != NULL)
6275                 {
6276
6277                     lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6278
6279                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6280                                   AFS_TRACE_LEVEL_VERBOSE,
6281                                   "AFSRetrieveFileAttributes Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6282                                   &pDirectoryEntry->NameInformation.FileName,
6283                                   pDirectoryEntry,
6284                                   NULL,
6285                                   lCount);
6286                 }
6287                 else
6288                 {
6289
6290                     lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
6291
6292                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6293                                   AFS_TRACE_LEVEL_VERBOSE,
6294                                   "AFSRetrieveFileAttributes Decrement2 count on %wZ DE %p Ccb %p Cnt %d\n",
6295                                   &pParentDirEntry->NameInformation.FileName,
6296                                   pParentDirEntry,
6297                                   NULL,
6298                                   lCount);
6299                 }
6300             }
6301
6302             pVolumeCB = NULL;
6303
6304             try_return( ntStatus);
6305         }
6306
6307         //
6308         // Store off the information
6309         //
6310
6311         FileInfo->FileAttributes = pDirectoryEntry->ObjectInformation->FileAttributes;
6312
6313         //
6314         // Check for the mount point being returned
6315         //
6316
6317         if( pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_MOUNTPOINT)
6318         {
6319
6320             FileInfo->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
6321         }
6322         else if( pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_SYMLINK ||
6323                  pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_DFSLINK)
6324         {
6325
6326             if ( FileInfo->FileAttributes == FILE_ATTRIBUTE_NORMAL)
6327             {
6328
6329                 FileInfo->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
6330             }
6331             else
6332             {
6333
6334                 FileInfo->FileAttributes |= FILE_ATTRIBUTE_REPARSE_POINT;
6335             }
6336         }
6337
6338         FileInfo->AllocationSize = pDirectoryEntry->ObjectInformation->AllocationSize;
6339
6340         FileInfo->EndOfFile = pDirectoryEntry->ObjectInformation->EndOfFile;
6341
6342         FileInfo->CreationTime = pDirectoryEntry->ObjectInformation->CreationTime;
6343
6344         FileInfo->LastAccessTime = pDirectoryEntry->ObjectInformation->LastAccessTime;
6345
6346         FileInfo->LastWriteTime = pDirectoryEntry->ObjectInformation->LastWriteTime;
6347
6348         FileInfo->ChangeTime = pDirectoryEntry->ObjectInformation->ChangeTime;
6349
6350         //
6351         // Remove the reference made above
6352         //
6353
6354         lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6355
6356         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6357                       AFS_TRACE_LEVEL_VERBOSE,
6358                       "AFSRetrieveFileAttributes Decrement3 count on %wZ DE %p Ccb %p Cnt %d\n",
6359                       &pDirectoryEntry->NameInformation.FileName,
6360                       pDirectoryEntry,
6361                       NULL,
6362                       lCount);
6363
6364 try_exit:
6365
6366         if( pDirEntry != NULL)
6367         {
6368
6369             AFSExFreePool( pDirEntry);
6370         }
6371
6372         if( pVolumeCB != NULL)
6373         {
6374
6375             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6376
6377             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6378                           AFS_TRACE_LEVEL_VERBOSE,
6379                           "AFSRetrieveFileAttributes Decrement2 count on volume %08lX Cnt %d\n",
6380                           pVolumeCB,
6381                           lCount);
6382         }
6383
6384         if( pNameArray != NULL)
6385         {
6386
6387             AFSFreeNameArray( pNameArray);
6388         }
6389
6390         if( pwchBuffer != NULL)
6391         {
6392
6393             //
6394             // Always free the buffer that we allocated as AFSLocateNameEntry
6395             // will not free it.  If uniFullPathName.Buffer was allocated by
6396             // AFSLocateNameEntry, then we must free that as well.
6397             // Check that the uniFullPathName.Buffer in the string is not the same
6398             // offset by the length of the server name
6399             //
6400
6401             if( uniFullPathName.Length > 0 &&
6402                 pwchBuffer != (WCHAR *)((char *)uniFullPathName.Buffer - ulNameDifference))
6403             {
6404
6405                 AFSExFreePool( uniFullPathName.Buffer);
6406             }
6407
6408             AFSExFreePool( pwchBuffer);
6409         }
6410     }
6411
6412     return ntStatus;
6413 }
6414
6415 AFSObjectInfoCB *
6416 AFSAllocateObjectInfo( IN AFSObjectInfoCB *ParentObjectInfo,
6417                        IN ULONGLONG HashIndex)
6418 {
6419
6420     NTSTATUS ntStatus = STATUS_SUCCESS;
6421     AFSObjectInfoCB *pObjectInfo = NULL;
6422     LONG lCount;
6423
6424     __Enter
6425     {
6426
6427         pObjectInfo = (AFSObjectInfoCB *)AFSExAllocatePoolWithTag( PagedPool,
6428                                                                    sizeof( AFSObjectInfoCB),
6429                                                                    AFS_OBJECT_INFO_TAG);
6430
6431         if( pObjectInfo == NULL)
6432         {
6433
6434             try_return( pObjectInfo);
6435         }
6436
6437         RtlZeroMemory( pObjectInfo,
6438                        sizeof( AFSObjectInfoCB));
6439
6440         pObjectInfo->NonPagedInfo = (AFSNonPagedObjectInfoCB *)AFSExAllocatePoolWithTag( NonPagedPool,
6441                                                                                          sizeof( AFSNonPagedObjectInfoCB),
6442                                                                                          AFS_NP_OBJECT_INFO_TAG);
6443
6444         if( pObjectInfo->NonPagedInfo == NULL)
6445         {
6446
6447             AFSExFreePool( pObjectInfo);
6448
6449             try_return( pObjectInfo = NULL);
6450         }
6451
6452         ExInitializeResourceLite( &pObjectInfo->NonPagedInfo->DirectoryNodeHdrLock);
6453
6454         pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock = &pObjectInfo->NonPagedInfo->DirectoryNodeHdrLock;
6455
6456         pObjectInfo->VolumeCB = ParentObjectInfo->VolumeCB;
6457
6458         pObjectInfo->ParentObjectInformation = ParentObjectInfo;
6459
6460         if( ParentObjectInfo != NULL)
6461         {
6462             lCount = InterlockedIncrement( &ParentObjectInfo->ObjectReferenceCount);
6463         }
6464
6465         //
6466         // Initialize the access time
6467         //
6468
6469         KeQueryTickCount( &pObjectInfo->LastAccessCount);
6470
6471         if( HashIndex != 0)
6472         {
6473
6474             //
6475             // Insert the entry into the object tree and list
6476             //
6477
6478             pObjectInfo->TreeEntry.HashIndex = HashIndex;
6479
6480             if( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead == NULL)
6481             {
6482
6483                 ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead = &pObjectInfo->TreeEntry;
6484             }
6485             else
6486             {
6487
6488                 ntStatus = AFSInsertHashEntry( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
6489                                                &pObjectInfo->TreeEntry);
6490
6491                 ASSERT( NT_SUCCESS( ntStatus));
6492             }
6493
6494             //
6495             // And the object list in the volume
6496             //
6497
6498             if( ParentObjectInfo->VolumeCB->ObjectInfoListHead == NULL)
6499             {
6500
6501                 ParentObjectInfo->VolumeCB->ObjectInfoListHead = pObjectInfo;
6502             }
6503             else
6504             {
6505
6506                 ParentObjectInfo->VolumeCB->ObjectInfoListTail->ListEntry.fLink = (void *)pObjectInfo;
6507
6508                 pObjectInfo->ListEntry.bLink = (void *)ParentObjectInfo->VolumeCB->ObjectInfoListTail;
6509             }
6510
6511             ParentObjectInfo->VolumeCB->ObjectInfoListTail = pObjectInfo;
6512
6513             //
6514             // Indicate the object is in the hash tree and linked list in the volume
6515             //
6516
6517             SetFlag( pObjectInfo->Flags, AFS_OBJECT_INSERTED_HASH_TREE | AFS_OBJECT_INSERTED_VOLUME_LIST);
6518         }
6519
6520 try_exit:
6521
6522         NOTHING;
6523     }
6524
6525     return pObjectInfo;
6526 }
6527
6528 void
6529 AFSDeleteObjectInfo( IN AFSObjectInfoCB *ObjectInfo)
6530 {
6531
6532     BOOLEAN bAcquiredTreeLock = FALSE;
6533     LONG lCount;
6534
6535     if( !ExIsResourceAcquiredExclusiveLite( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock))
6536     {
6537
6538         ASSERT( !ExIsResourceAcquiredLite( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock));
6539
6540         AFSAcquireExcl( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock,
6541                         TRUE);
6542
6543         bAcquiredTreeLock = TRUE;
6544     }
6545
6546     //
6547     // Remove it from the tree and list if it was inserted
6548     //
6549
6550     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_INSERTED_HASH_TREE))
6551     {
6552
6553         AFSRemoveHashEntry( &ObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
6554                             &ObjectInfo->TreeEntry);
6555     }
6556
6557     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_INSERTED_VOLUME_LIST))
6558     {
6559
6560         if( ObjectInfo->ListEntry.fLink == NULL)
6561         {
6562
6563             ObjectInfo->VolumeCB->ObjectInfoListTail = (AFSObjectInfoCB *)ObjectInfo->ListEntry.bLink;
6564
6565             if( ObjectInfo->VolumeCB->ObjectInfoListTail != NULL)
6566             {
6567
6568                 ObjectInfo->VolumeCB->ObjectInfoListTail->ListEntry.fLink = NULL;
6569             }
6570         }
6571         else
6572         {
6573
6574             ((AFSObjectInfoCB *)(ObjectInfo->ListEntry.fLink))->ListEntry.bLink = ObjectInfo->ListEntry.bLink;
6575         }
6576
6577         if( ObjectInfo->ListEntry.bLink == NULL)
6578         {
6579
6580             ObjectInfo->VolumeCB->ObjectInfoListHead = (AFSObjectInfoCB *)ObjectInfo->ListEntry.fLink;
6581
6582             if( ObjectInfo->VolumeCB->ObjectInfoListHead != NULL)
6583             {
6584
6585                 ObjectInfo->VolumeCB->ObjectInfoListHead->ListEntry.bLink = NULL;
6586             }
6587         }
6588         else
6589         {
6590
6591             ((AFSObjectInfoCB *)(ObjectInfo->ListEntry.bLink))->ListEntry.fLink = ObjectInfo->ListEntry.fLink;
6592         }
6593     }
6594
6595     if( ObjectInfo->ParentObjectInformation != NULL)
6596     {
6597
6598         lCount = InterlockedDecrement( &ObjectInfo->ParentObjectInformation->ObjectReferenceCount);
6599     }
6600
6601     if( bAcquiredTreeLock)
6602     {
6603
6604         AFSReleaseResource( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
6605     }
6606
6607     //
6608     // Release the fid in the service
6609     //
6610
6611     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_HELD_IN_SERVICE))
6612     {
6613
6614         AFSReleaseFid( &ObjectInfo->FileId);
6615     }
6616
6617     ExDeleteResourceLite( &ObjectInfo->NonPagedInfo->DirectoryNodeHdrLock);
6618
6619     AFSExFreePool( ObjectInfo->NonPagedInfo);
6620
6621     AFSExFreePool( ObjectInfo);
6622
6623     return;
6624 }
6625
6626 NTSTATUS
6627 AFSEvaluateRootEntry( IN AFSDirectoryCB *DirectoryCB,
6628                       OUT AFSDirectoryCB **TargetDirEntry)
6629 {
6630
6631     NTSTATUS ntStatus = STATUS_SUCCESS;
6632     AFSDirEnumEntry *pDirEntry = NULL, *pLastDirEntry = NULL;
6633     UNICODE_STRING uniFullPathName;
6634     AFSNameArrayHdr    *pNameArray = NULL;
6635     AFSVolumeCB *pVolumeCB = NULL;
6636     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
6637     WCHAR *pwchBuffer = NULL;
6638     UNICODE_STRING uniComponentName, uniRemainingPath, uniParsedName;
6639     ULONG ulNameDifference = 0;
6640     GUID    stAuthGroup;
6641     LONG lCount;
6642
6643     __Enter
6644     {
6645
6646         ntStatus = AFSRetrieveValidAuthGroup( NULL,
6647                                               DirectoryCB->ObjectInformation,
6648                                               FALSE,
6649                                               &stAuthGroup);
6650
6651         if( !NT_SUCCESS( ntStatus))
6652         {
6653             try_return( ntStatus);
6654         }
6655
6656         //
6657         // Retrieve a target name for the entry
6658         //
6659
6660         AFSAcquireShared( &DirectoryCB->NonPaged->Lock,
6661                           TRUE);
6662
6663         if( DirectoryCB->NameInformation.TargetName.Length == 0)
6664         {
6665
6666             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6667
6668             ntStatus = AFSEvaluateTargetByID( DirectoryCB->ObjectInformation,
6669                                               &stAuthGroup,
6670                                               FALSE,
6671                                               &pDirEntry);
6672
6673             if( !NT_SUCCESS( ntStatus) ||
6674                 pDirEntry->TargetNameLength == 0)
6675             {
6676
6677                 if( pDirEntry != NULL)
6678                 {
6679
6680                     ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
6681                 }
6682
6683                 try_return( ntStatus);
6684             }
6685
6686             AFSAcquireExcl( &DirectoryCB->NonPaged->Lock,
6687                             TRUE);
6688
6689             if( DirectoryCB->NameInformation.TargetName.Length == 0)
6690             {
6691
6692                 //
6693                 // Update the target name
6694                 //
6695
6696                 ntStatus = AFSUpdateTargetName( &DirectoryCB->NameInformation.TargetName,
6697                                                 &DirectoryCB->Flags,
6698                                                 (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset),
6699                                                 (USHORT)pDirEntry->TargetNameLength);
6700
6701                 if( !NT_SUCCESS( ntStatus))
6702                 {
6703
6704                     AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6705
6706                     try_return( ntStatus);
6707                 }
6708             }
6709
6710             AFSConvertToShared( &DirectoryCB->NonPaged->Lock);
6711         }
6712
6713         //
6714         // Need to pass the full path in for parsing.
6715         //
6716
6717         uniFullPathName.Length = 0;
6718         uniFullPathName.MaximumLength = DirectoryCB->NameInformation.TargetName.Length;
6719
6720         uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6721                                                                     uniFullPathName.MaximumLength,
6722                                                                     AFS_NAME_BUFFER_EIGHT_TAG);
6723
6724         if( uniFullPathName.Buffer == NULL)
6725         {
6726
6727             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6728
6729             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6730         }
6731
6732         pwchBuffer = uniFullPathName.Buffer;
6733
6734         RtlZeroMemory( uniFullPathName.Buffer,
6735                        uniFullPathName.MaximumLength);
6736
6737         RtlCopyMemory( uniFullPathName.Buffer,
6738                        DirectoryCB->NameInformation.TargetName.Buffer,
6739                        DirectoryCB->NameInformation.TargetName.Length);
6740
6741         uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6742
6743         //
6744         // This name should begin with the \afs server so parse it off and chech it
6745         //
6746
6747         FsRtlDissectName( uniFullPathName,
6748                           &uniComponentName,
6749                           &uniRemainingPath);
6750
6751         if( RtlCompareUnicodeString( &uniComponentName,
6752                                      &AFSServerName,
6753                                      TRUE) != 0)
6754         {
6755
6756             //
6757             // Try evaluating the full path
6758             //
6759
6760             uniFullPathName.Buffer = pwchBuffer;
6761
6762             uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6763
6764             uniFullPathName.MaximumLength = uniFullPathName.Length;
6765         }
6766         else
6767         {
6768
6769             uniFullPathName = uniRemainingPath;
6770         }
6771
6772         uniParsedName = uniFullPathName;
6773
6774         ulNameDifference = (ULONG)(uniFullPathName.Length > 0 ? ((char *)uniFullPathName.Buffer - (char *)pwchBuffer) : 0);
6775
6776         AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6777
6778         //
6779         // Our name array
6780         //
6781
6782         pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
6783                                        0);
6784
6785         if( pNameArray == NULL)
6786         {
6787
6788             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6789         }
6790
6791         pVolumeCB = AFSGlobalRoot;
6792
6793         pParentDirEntry = AFSGlobalRoot->DirectoryCB;
6794
6795         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
6796
6797         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6798                       AFS_TRACE_LEVEL_VERBOSE,
6799                       "AFSEvaluateRootEntry Increment count on volume %08lX Cnt %d\n",
6800                       pVolumeCB,
6801                       lCount);
6802
6803         lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
6804
6805         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6806                       AFS_TRACE_LEVEL_VERBOSE,
6807                       "AFSEvaluateRootEntry Increment count on %wZ DE %p Ccb %p Cnt %d\n",
6808                       &pParentDirEntry->NameInformation.FileName,
6809                       pParentDirEntry,
6810                       NULL,
6811                       lCount);
6812
6813         ntStatus = AFSLocateNameEntry( NULL,
6814                                        NULL,
6815                                        &uniFullPathName,
6816                                        &uniParsedName,
6817                                        pNameArray,
6818                                        0,
6819                                        &pVolumeCB,
6820                                        &pParentDirEntry,
6821                                        &pDirectoryEntry,
6822                                        NULL);
6823
6824         if( !NT_SUCCESS( ntStatus))
6825         {
6826
6827             //
6828             // The volume lock was released on failure above
6829             // Except for STATUS_OBJECT_NAME_NOT_FOUND
6830             //
6831
6832             if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
6833             {
6834
6835                 if( pVolumeCB != NULL)
6836                 {
6837
6838                     lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6839
6840                     AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6841                                   AFS_TRACE_LEVEL_VERBOSE,
6842                                   "AFSEvaluateRootEntry Decrement count on volume %08lX Cnt %d\n",
6843                                   pVolumeCB,
6844                                   lCount);
6845                 }
6846
6847                 if( pDirectoryEntry != NULL)
6848                 {
6849
6850                     lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6851
6852                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6853                                   AFS_TRACE_LEVEL_VERBOSE,
6854                                   "AFSEvaluateRootEntry Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6855                                   &pDirectoryEntry->NameInformation.FileName,
6856                                   pDirectoryEntry,
6857                                   NULL,
6858                                   lCount);
6859                 }
6860                 else
6861                 {
6862
6863                     lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
6864
6865                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6866                                   AFS_TRACE_LEVEL_VERBOSE,
6867                                   "AFSEvaluateRootEntry Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6868                                   &pParentDirEntry->NameInformation.FileName,
6869                                   pParentDirEntry,
6870                                   NULL,
6871                                   lCount);
6872                 }
6873             }
6874
6875             pVolumeCB = NULL;
6876
6877             try_return( ntStatus);
6878         }
6879
6880         //
6881         // Pass back the target dir entry for this request
6882         //
6883
6884         *TargetDirEntry = pDirectoryEntry;
6885
6886 try_exit:
6887
6888         if( pDirEntry != NULL)
6889         {
6890
6891             AFSExFreePool( pDirEntry);
6892         }
6893
6894         if( pVolumeCB != NULL)
6895         {
6896
6897             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6898
6899             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6900                           AFS_TRACE_LEVEL_VERBOSE,
6901                           "AFSEvaluateRootEntry2 Decrement count on volume %08lX Cnt %d\n",
6902                           pVolumeCB,
6903                           lCount);
6904         }
6905
6906         if( pNameArray != NULL)
6907         {
6908
6909             AFSFreeNameArray( pNameArray);
6910         }
6911
6912         if( pwchBuffer != NULL)
6913         {
6914
6915             //
6916             // Always free the buffer that we allocated as AFSLocateNameEntry
6917             // will not free it.  If uniFullPathName.Buffer was allocated by
6918             // AFSLocateNameEntry, then we must free that as well.
6919             // Check that the uniFullPathName.Buffer in the string is not the same
6920             // offset by the length of the server name
6921             //
6922
6923             if( uniFullPathName.Length > 0 &&
6924                 pwchBuffer != (WCHAR *)((char *)uniFullPathName.Buffer - ulNameDifference))
6925             {
6926
6927                 AFSExFreePool( uniFullPathName.Buffer);
6928             }
6929
6930             AFSExFreePool( pwchBuffer);
6931         }
6932     }
6933
6934     return ntStatus;
6935 }
6936
6937 NTSTATUS
6938 AFSCleanupFcb( IN AFSFcb *Fcb,
6939                IN BOOLEAN ForceFlush)
6940 {
6941
6942     NTSTATUS ntStatus = STATUS_SUCCESS;
6943     AFSDeviceExt *pRDRDeviceExt = NULL, *pControlDeviceExt = NULL;
6944     LARGE_INTEGER liTime;
6945     IO_STATUS_BLOCK stIoStatus;
6946
6947     __Enter
6948     {
6949
6950         pControlDeviceExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
6951
6952         pRDRDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
6953
6954         if( BooleanFlagOn( pRDRDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_REDIRECTOR_SHUTDOWN))
6955         {
6956
6957             if( !BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) &&
6958                 !BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
6959             {
6960
6961                 AFSAcquireExcl( &Fcb->NPFcb->Resource,
6962                                 TRUE);
6963
6964                 if( Fcb->OpenReferenceCount > 0)
6965                 {
6966
6967                     __try
6968                     {
6969
6970                         CcFlushCache( &Fcb->NPFcb->SectionObjectPointers,
6971                                       NULL,
6972                                       0,
6973                                       &stIoStatus);
6974
6975                         if( !NT_SUCCESS( stIoStatus.Status))
6976                         {
6977
6978                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
6979                                           AFS_TRACE_LEVEL_ERROR,
6980                                           "AFSCleanupFcb CcFlushCache [1] failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
6981                                           Fcb->ObjectInformation->FileId.Cell,
6982                                           Fcb->ObjectInformation->FileId.Volume,
6983                                           Fcb->ObjectInformation->FileId.Vnode,
6984                                           Fcb->ObjectInformation->FileId.Unique,
6985                                           stIoStatus.Status,
6986                                           stIoStatus.Information);
6987
6988                             ntStatus = stIoStatus.Status;
6989                         }
6990
6991                         if ( !CcPurgeCacheSection( &Fcb->NPFcb->SectionObjectPointers,
6992                                                    NULL,
6993                                                    0,
6994                                                    FALSE))
6995                         {
6996
6997                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
6998                                           AFS_TRACE_LEVEL_WARNING,
6999                                           "AFSCleanupFcb CcPurgeCacheSection [1] failure FID %08lX-%08lX-%08lX-%08lX\n",
7000                                           Fcb->ObjectInformation->FileId.Cell,
7001                                           Fcb->ObjectInformation->FileId.Volume,
7002                                           Fcb->ObjectInformation->FileId.Vnode,
7003                                           Fcb->ObjectInformation->FileId.Unique);
7004
7005                             SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7006                         }
7007                     }
7008                     __except( EXCEPTION_EXECUTE_HANDLER)
7009                     {
7010
7011                         ntStatus = GetExceptionCode();
7012
7013                         AFSDbgLogMsg( 0,
7014                                       0,
7015                                       "EXCEPTION - AFSCleanupFcb Cc [1] FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
7016                                       Fcb->ObjectInformation->FileId.Cell,
7017                                       Fcb->ObjectInformation->FileId.Volume,
7018                                       Fcb->ObjectInformation->FileId.Vnode,
7019                                       Fcb->ObjectInformation->FileId.Unique,
7020                                       ntStatus);
7021
7022                         SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7023                     }
7024                 }
7025
7026                 AFSReleaseResource( &Fcb->NPFcb->Resource);
7027
7028                 //
7029                 // Wait for any currently running flush or release requests to complete
7030                 //
7031
7032                 AFSWaitOnQueuedFlushes( Fcb);
7033
7034                 //
7035                 // Now perform another flush on the file
7036                 //
7037
7038                 if( !NT_SUCCESS( AFSFlushExtents( Fcb,
7039                                                   NULL)))
7040                 {
7041
7042                     AFSReleaseExtentsWithFlush( Fcb,
7043                                                 NULL,
7044                                                 TRUE);
7045                 }
7046             }
7047
7048             if( Fcb->OpenReferenceCount == 0 ||
7049                 BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) ||
7050                 BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
7051             {
7052
7053                 AFSTearDownFcbExtents( Fcb,
7054                                        NULL);
7055             }
7056
7057             try_return( ntStatus);
7058         }
7059
7060         KeQueryTickCount( &liTime);
7061
7062         //
7063         // First up are there dirty extents in the cache to flush?
7064         //
7065
7066         if( BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) ||
7067             BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
7068         {
7069
7070             //
7071             // The file has been marked as invalid.  Dump it
7072             //
7073
7074             AFSTearDownFcbExtents( Fcb,
7075                                    NULL);
7076         }
7077         else if( ForceFlush ||
7078             ( ( Fcb->Specific.File.ExtentsDirtyCount ||
7079                 Fcb->Specific.File.ExtentCount) &&
7080               (liTime.QuadPart - Fcb->Specific.File.LastServerFlush.QuadPart)
7081                                                     >= pControlDeviceExt->Specific.Control.FcbFlushTimeCount.QuadPart))
7082         {
7083             if( !NT_SUCCESS( AFSFlushExtents( Fcb,
7084                                               NULL)) &&
7085                 Fcb->OpenReferenceCount == 0)
7086             {
7087
7088                 AFSReleaseExtentsWithFlush( Fcb,
7089                                             NULL,
7090                                             TRUE);
7091             }
7092         }
7093
7094         //
7095         // If there are extents and they haven't been used recently *and*
7096         // are not being used
7097         //
7098
7099         if( ( ForceFlush ||
7100               ( 0 != Fcb->Specific.File.ExtentCount &&
7101                 0 != Fcb->Specific.File.LastExtentAccess.QuadPart &&
7102                 (liTime.QuadPart - Fcb->Specific.File.LastExtentAccess.QuadPart) >=
7103                                         (AFS_SERVER_PURGE_SLEEP * pControlDeviceExt->Specific.Control.FcbPurgeTimeCount.QuadPart))) &&
7104             AFSAcquireExcl( &Fcb->NPFcb->Resource,
7105                             ForceFlush))
7106         {
7107
7108             __try
7109             {
7110
7111                 CcFlushCache( &Fcb->NPFcb->SectionObjectPointers,
7112                               NULL,
7113                               0,
7114                               &stIoStatus);
7115
7116                 if( !NT_SUCCESS( stIoStatus.Status))
7117                 {
7118
7119                     AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
7120                                   AFS_TRACE_LEVEL_ERROR,
7121                                   "AFSCleanupFcb CcFlushCache [2] failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
7122                                   Fcb->ObjectInformation->FileId.Cell,
7123                                   Fcb->ObjectInformation->FileId.Volume,
7124                                   Fcb->ObjectInformation->FileId.Vnode,
7125                                   Fcb->ObjectInformation->FileId.Unique,
7126                                   stIoStatus.Status,
7127                                   stIoStatus.Information);
7128
7129                     ntStatus = stIoStatus.Status;
7130                 }
7131
7132                 if( ForceFlush)
7133                 {
7134
7135                     if ( !CcPurgeCacheSection( &Fcb->NPFcb->SectionObjectPointers,
7136                                                NULL,
7137                                                0,
7138                                                FALSE))
7139                     {
7140
7141                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
7142                                       AFS_TRACE_LEVEL_WARNING,
7143                                       "AFSCleanupFcb CcPurgeCacheSection [2] failure FID %08lX-%08lX-%08lX-%08lX\n",
7144                                       Fcb->ObjectInformation->FileId.Cell,
7145                                       Fcb->ObjectInformation->FileId.Volume,
7146                                       Fcb->ObjectInformation->FileId.Vnode,
7147                                       Fcb->ObjectInformation->FileId.Unique);
7148
7149                         SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7150                     }
7151                 }
7152             }
7153             __except( EXCEPTION_EXECUTE_HANDLER)
7154             {
7155
7156                 ntStatus = GetExceptionCode();
7157
7158                 AFSDbgLogMsg( 0,
7159                               0,
7160                               "EXCEPTION - AFSCleanupFcb Cc [2] FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
7161                               Fcb->ObjectInformation->FileId.Cell,
7162                               Fcb->ObjectInformation->FileId.Volume,
7163                               Fcb->ObjectInformation->FileId.Vnode,
7164                               Fcb->ObjectInformation->FileId.Unique,
7165                               ntStatus);
7166             }
7167
7168             AFSReleaseResource( &Fcb->NPFcb->Resource);
7169
7170             if( Fcb->OpenReferenceCount <= 0)
7171             {
7172
7173                 //
7174                 // Tear em down we'll not be needing them again
7175                 //
7176
7177                 AFSTearDownFcbExtents( Fcb,
7178                                        NULL);
7179             }
7180         }
7181
7182 try_exit:
7183
7184         NOTHING;
7185     }
7186
7187     return ntStatus;
7188 }
7189
7190 NTSTATUS
7191 AFSUpdateDirEntryName( IN AFSDirectoryCB *DirectoryCB,
7192                        IN UNICODE_STRING *NewFileName)
7193 {
7194
7195     NTSTATUS ntStatus = STATUS_SUCCESS;
7196     WCHAR *pTmpBuffer = NULL;
7197
7198     __Enter
7199     {
7200
7201         if( NewFileName->Length > DirectoryCB->NameInformation.FileName.Length)
7202         {
7203
7204             if( BooleanFlagOn( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER))
7205             {
7206
7207                 AFSExFreePool( DirectoryCB->NameInformation.FileName.Buffer);
7208
7209                 ClearFlag( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER);
7210
7211                 DirectoryCB->NameInformation.FileName.Buffer = NULL;
7212             }
7213
7214             //
7215             // OK, we need to allocate a new name buffer
7216             //
7217
7218             pTmpBuffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
7219                                                             NewFileName->Length,
7220                                                             AFS_NAME_BUFFER_NINE_TAG);
7221
7222             if( pTmpBuffer == NULL)
7223             {
7224
7225                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
7226             }
7227
7228             DirectoryCB->NameInformation.FileName.Buffer = pTmpBuffer;
7229
7230             DirectoryCB->NameInformation.FileName.MaximumLength = NewFileName->Length;
7231
7232             SetFlag( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER);
7233         }
7234
7235         DirectoryCB->NameInformation.FileName.Length = NewFileName->Length;
7236
7237         RtlCopyMemory( DirectoryCB->NameInformation.FileName.Buffer,
7238                        NewFileName->Buffer,
7239                        NewFileName->Length);
7240
7241 try_exit:
7242
7243         NOTHING;
7244     }
7245
7246     return ntStatus;
7247 }
7248
7249 NTSTATUS
7250 AFSReadCacheFile( IN void *ReadBuffer,
7251                   IN LARGE_INTEGER *ReadOffset,
7252                   IN ULONG RequestedDataLength,
7253                   IN OUT PULONG BytesRead)
7254 {
7255
7256     NTSTATUS            ntStatus = STATUS_SUCCESS;
7257     PIRP                pIrp = NULL;
7258     KEVENT              kEvent;
7259     PIO_STACK_LOCATION  pIoStackLocation = NULL;
7260     AFSDeviceExt       *pRdrDevExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
7261     DEVICE_OBJECT      *pTargetDeviceObject = NULL;
7262     FILE_OBJECT        *pCacheFileObject = NULL;
7263
7264     __Enter
7265     {
7266
7267         pCacheFileObject = AFSReferenceCacheFileObject();
7268
7269         if( pCacheFileObject == NULL)
7270         {
7271             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
7272         }
7273
7274         pTargetDeviceObject = IoGetRelatedDeviceObject( pCacheFileObject);
7275
7276         //
7277         // Initialize the event
7278         //
7279
7280         KeInitializeEvent( &kEvent,
7281                            SynchronizationEvent,
7282                            FALSE);
7283
7284         //
7285         // Allocate an irp for this request.  This could also come from a
7286         // private pool, for instance.
7287         //
7288
7289         pIrp = IoAllocateIrp( pTargetDeviceObject->StackSize,
7290                               FALSE);
7291
7292         if( pIrp == NULL)
7293         {
7294
7295             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
7296         }
7297
7298         //
7299         // Build the IRP's main body
7300         //
7301
7302         pIrp->UserBuffer = ReadBuffer;
7303
7304         pIrp->Tail.Overlay.Thread = PsGetCurrentThread();
7305         pIrp->RequestorMode = KernelMode;
7306         pIrp->Flags |= IRP_READ_OPERATION;
7307
7308         //
7309         // Set up the I/O stack location.
7310         //
7311
7312         pIoStackLocation = IoGetNextIrpStackLocation( pIrp);
7313         pIoStackLocation->MajorFunction = IRP_MJ_READ;
7314         pIoStackLocation->DeviceObject = pTargetDeviceObject;
7315         pIoStackLocation->FileObject = pCacheFileObject;
7316         pIoStackLocation->Parameters.Read.Length = RequestedDataLength;
7317
7318         pIoStackLocation->Parameters.Read.ByteOffset.QuadPart = ReadOffset->QuadPart;
7319
7320         //
7321         // Set the completion routine.
7322         //
7323
7324         IoSetCompletionRoutine( pIrp,
7325                                 AFSIrpComplete,
7326                                 &kEvent,
7327                                 TRUE,
7328                                 TRUE,
7329                                 TRUE);
7330
7331         //
7332         // Send it to the FSD
7333         //
7334
7335         ntStatus = IoCallDriver( pTargetDeviceObject,
7336                                  pIrp);
7337
7338         if( NT_SUCCESS( ntStatus))
7339         {
7340
7341             //
7342             // Wait for the I/O
7343             //
7344
7345             ntStatus = KeWaitForSingleObject( &kEvent,
7346                                               Executive,
7347                                               KernelMode,
7348                                               FALSE,
7349                                               0);
7350
7351             if( NT_SUCCESS( ntStatus))
7352             {
7353
7354                 ntStatus = pIrp->IoStatus.Status;
7355
7356                 *BytesRead = (ULONG)pIrp->IoStatus.Information;
7357             }
7358         }
7359
7360 try_exit:
7361
7362         if( pCacheFileObject != NULL)
7363         {
7364             AFSReleaseCacheFileObject( pCacheFileObject);
7365         }
7366
7367         if( pIrp != NULL)
7368         {
7369
7370             if( pIrp->MdlAddress != NULL)
7371             {
7372
7373                 if( FlagOn( pIrp->MdlAddress->MdlFlags, MDL_PAGES_LOCKED))
7374                 {
7375
7376                     MmUnlockPages( pIrp->MdlAddress);
7377                 }
7378
7379                 IoFreeMdl( pIrp->MdlAddress);
7380             }
7381
7382             pIrp->MdlAddress = NULL;
7383
7384             //
7385             // Free the Irp
7386             //
7387
7388             IoFreeIrp( pIrp);
7389         }
7390     }
7391
7392     return ntStatus;
7393 }
7394
7395 NTSTATUS
7396 AFSIrpComplete( IN PDEVICE_OBJECT DeviceObject,
7397                 IN PIRP           Irp,
7398                 IN PVOID          Context)
7399 {
7400
7401     KEVENT *pEvent = (KEVENT *)Context;
7402
7403     KeSetEvent( pEvent,
7404                 0,
7405                 FALSE);
7406
7407     return STATUS_MORE_PROCESSING_REQUIRED;
7408 }
7409
7410 BOOLEAN
7411 AFSIsDirectoryEmptyForDelete( IN AFSFcb *Fcb)
7412 {
7413
7414     BOOLEAN bIsEmpty = FALSE;
7415     AFSDirectoryCB *pDirEntry = NULL;
7416
7417     __Enter
7418     {
7419
7420         AFSAcquireShared( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
7421                           TRUE);
7422
7423         bIsEmpty = TRUE;
7424
7425         if( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeListHead != NULL)
7426         {
7427
7428             pDirEntry = Fcb->ObjectInformation->Specific.Directory.DirectoryNodeListHead;
7429
7430             while( pDirEntry != NULL)
7431             {
7432
7433                 if( !BooleanFlagOn( pDirEntry->Flags, AFS_DIR_ENTRY_FAKE) &&
7434                     !BooleanFlagOn( pDirEntry->Flags, AFS_DIR_ENTRY_DELETED))
7435                 {
7436
7437                     bIsEmpty = FALSE;
7438
7439                     break;
7440                 }
7441
7442                 pDirEntry = (AFSDirectoryCB *)pDirEntry->ListEntry.fLink;
7443             }
7444
7445         }
7446
7447         AFSReleaseResource( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
7448     }
7449
7450     return bIsEmpty;
7451 }
7452
7453 void
7454 AFSRemoveNameEntry( IN AFSObjectInfoCB *ParentObjectInfo,
7455                     IN AFSDirectoryCB *DirEntry)
7456 {
7457
7458     NTSTATUS ntStatus = STATUS_SUCCESS;
7459
7460     __Enter
7461     {
7462
7463         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE))
7464         {
7465
7466             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7467                           AFS_TRACE_LEVEL_VERBOSE,
7468                           "AFSRemoveNameEntry DE %p for %wZ has NOT_IN flag set\n",
7469                           DirEntry,
7470                           &DirEntry->NameInformation.FileName);
7471
7472             try_return( ntStatus);
7473         }
7474
7475         ASSERT( ExIsResourceAcquiredExclusiveLite( ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock));
7476
7477         //
7478         // Remove the entry from the parent tree
7479         //
7480
7481         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7482                       AFS_TRACE_LEVEL_VERBOSE,
7483                       "AFSRemoveNameEntry DE %p for %wZ removing from case sensitive tree\n",
7484                       DirEntry,
7485                       &DirEntry->NameInformation.FileName);
7486
7487         AFSRemoveCaseSensitiveDirEntry( &ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
7488                                         DirEntry);
7489
7490         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7491                       AFS_TRACE_LEVEL_VERBOSE,
7492                       "AFSRemoveNameEntry DE %p for %wZ removing from case insensitive tree\n",
7493                       DirEntry,
7494                       &DirEntry->NameInformation.FileName);
7495
7496         AFSRemoveCaseInsensitiveDirEntry( &ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
7497                                           DirEntry);
7498
7499         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME))
7500         {
7501
7502             //
7503             // From the short name tree
7504             //
7505
7506             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7507                           AFS_TRACE_LEVEL_VERBOSE,
7508                           "AFSRemoveNameEntry DE %p for %wZ removing from shortname tree\n",
7509                           DirEntry,
7510                           &DirEntry->NameInformation.FileName);
7511
7512             AFSRemoveShortNameDirEntry( &ParentObjectInfo->Specific.Directory.ShortNameTree,
7513                                         DirEntry);
7514
7515             ClearFlag( DirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
7516         }
7517
7518         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7519                       AFS_TRACE_LEVEL_VERBOSE,
7520                       "AFSRemoveNameEntry DE %p for %wZ setting NOT_IN flag\n",
7521                       DirEntry,
7522                       &DirEntry->NameInformation.FileName);
7523
7524         SetFlag( DirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE);
7525
7526         ClearFlag( DirEntry->Flags, AFS_DIR_ENTRY_CASE_INSENSTIVE_LIST_HEAD);
7527
7528 try_exit:
7529
7530         NOTHING;
7531     }
7532
7533     return;
7534 }
7535
7536 LARGE_INTEGER
7537 AFSGetAuthenticationId()
7538 {
7539
7540     LARGE_INTEGER liAuthId = {0,0};
7541     NTSTATUS ntStatus = STATUS_SUCCESS;
7542     PACCESS_TOKEN hToken = NULL;
7543     PTOKEN_STATISTICS pTokenInfo = NULL;
7544     BOOLEAN bCopyOnOpen = FALSE;
7545     BOOLEAN bEffectiveOnly = FALSE;
7546     BOOLEAN bPrimaryToken = FALSE;
7547     SECURITY_IMPERSONATION_LEVEL stImpersonationLevel;
7548
7549     __Enter
7550     {
7551
7552         hToken = PsReferenceImpersonationToken( PsGetCurrentThread(),
7553                                                 &bCopyOnOpen,
7554                                                 &bEffectiveOnly,
7555                                                 &stImpersonationLevel);
7556
7557         if( hToken == NULL)
7558         {
7559
7560             hToken = PsReferencePrimaryToken( PsGetCurrentProcess());
7561
7562             if( hToken == NULL)
7563             {
7564
7565                 AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7566                               AFS_TRACE_LEVEL_ERROR,
7567                               "AFSGetAuthenticationId Failed to retrieve impersonation or primary token\n");
7568
7569                 try_return( ntStatus);
7570             }
7571
7572             bPrimaryToken = TRUE;
7573         }
7574
7575         ntStatus = SeQueryInformationToken( hToken,
7576                                             TokenStatistics,
7577                                             (PVOID *)&pTokenInfo);
7578
7579         if( !NT_SUCCESS( ntStatus))
7580         {
7581
7582             AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7583                           AFS_TRACE_LEVEL_ERROR,
7584                           "AFSGetAuthenticationId Failed to retrieve information Status %08lX\n", ntStatus);
7585
7586             try_return( ntStatus);
7587         }
7588
7589         liAuthId.HighPart = pTokenInfo->AuthenticationId.HighPart;
7590         liAuthId.LowPart = pTokenInfo->AuthenticationId.LowPart;
7591
7592         AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7593                       AFS_TRACE_LEVEL_VERBOSE,
7594                       "AFSGetAuthenticationId Successfully retrieved authentication ID %I64X\n",
7595                       liAuthId.QuadPart);
7596
7597 try_exit:
7598
7599         if( hToken != NULL)
7600         {
7601
7602             if( !bPrimaryToken)
7603             {
7604
7605                 PsDereferenceImpersonationToken( hToken);
7606             }
7607             else
7608             {
7609
7610                 PsDereferencePrimaryToken( hToken);
7611             }
7612         }
7613
7614         if( pTokenInfo != NULL)
7615         {
7616
7617             ExFreePool( pTokenInfo);    // Allocated by SeQueryInformationToken
7618         }
7619     }
7620
7621     return liAuthId;
7622 }
7623
7624 void
7625 AFSUnwindFileInfo( IN AFSFcb *Fcb,
7626                    IN AFSCcb *Ccb)
7627 {
7628
7629     if( Ccb->FileUnwindInfo.FileAttributes != (ULONG)-1)
7630     {
7631         Ccb->DirectoryCB->ObjectInformation->FileAttributes = Ccb->FileUnwindInfo.FileAttributes;
7632     }
7633
7634     if( Ccb->FileUnwindInfo.CreationTime.QuadPart != (ULONGLONG)-1)
7635     {
7636         Ccb->DirectoryCB->ObjectInformation->CreationTime.QuadPart = Ccb->FileUnwindInfo.CreationTime.QuadPart;
7637     }
7638
7639     if( Ccb->FileUnwindInfo.LastAccessTime.QuadPart != (ULONGLONG)-1)
7640     {
7641         Ccb->DirectoryCB->ObjectInformation->LastAccessTime.QuadPart = Ccb->FileUnwindInfo.LastAccessTime.QuadPart;
7642     }
7643
7644     if( Ccb->FileUnwindInfo.LastWriteTime.QuadPart != (ULONGLONG)-1)
7645     {
7646         Ccb->DirectoryCB->ObjectInformation->LastWriteTime.QuadPart = Ccb->FileUnwindInfo.LastWriteTime.QuadPart;
7647     }
7648
7649     if( Ccb->FileUnwindInfo.ChangeTime.QuadPart != (ULONGLONG)-1)
7650     {
7651         Ccb->DirectoryCB->ObjectInformation->ChangeTime.QuadPart = Ccb->FileUnwindInfo.ChangeTime.QuadPart;
7652     }
7653
7654     return;
7655 }
7656
7657 BOOLEAN
7658 AFSValidateDirList( IN AFSObjectInfoCB *ObjectInfo)
7659 {
7660
7661     BOOLEAN bIsValid = TRUE;
7662     ULONG ulCount = 0;
7663     AFSDirectoryCB *pCurrentDirEntry = NULL, *pDirEntry = NULL;
7664
7665     pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
7666
7667     while( pCurrentDirEntry != NULL)
7668     {
7669
7670         if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_FAKE))
7671         {
7672             ulCount++;
7673
7674             if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE))
7675             {
7676
7677                 pDirEntry = NULL;
7678
7679                 AFSLocateCaseSensitiveDirEntry( ObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
7680                                                 (ULONG)pCurrentDirEntry->CaseSensitiveTreeEntry.HashIndex,
7681                                                 &pDirEntry);
7682
7683                 if( pDirEntry == NULL)
7684                 {
7685                     DbgBreakPoint();
7686                 }
7687             }
7688         }
7689
7690         pCurrentDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
7691     }
7692
7693     if( ulCount != ObjectInfo->Specific.Directory.DirectoryNodeCount)
7694     {
7695
7696         AFSPrint("AFSValidateDirList Count off Calc: %d Stored: %d\n",
7697                   ulCount,
7698                   ObjectInfo->Specific.Directory.DirectoryNodeCount);
7699
7700         ObjectInfo->Specific.Directory.DirectoryNodeCount = ulCount;
7701
7702         bIsValid = FALSE;
7703     }
7704
7705     return bIsValid;
7706 }
7707
7708 PFILE_OBJECT
7709 AFSReferenceCacheFileObject()
7710 {
7711
7712     AFSDeviceExt       *pRdrDevExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
7713     FILE_OBJECT        *pCacheFileObject = NULL;
7714
7715     AFSAcquireShared( &pRdrDevExt->Specific.RDR.CacheFileLock,
7716                       TRUE);
7717
7718     pCacheFileObject = pRdrDevExt->Specific.RDR.CacheFileObject;
7719
7720     if( pCacheFileObject != NULL)
7721     {
7722         ObReferenceObject( pCacheFileObject);
7723     }
7724
7725     AFSReleaseResource( &pRdrDevExt->Specific.RDR.CacheFileLock);
7726
7727     return pCacheFileObject;
7728 }
7729
7730 void
7731 AFSReleaseCacheFileObject( IN PFILE_OBJECT CacheFileObject)
7732 {
7733
7734     ASSERT( CacheFileObject != NULL);
7735
7736     ObDereferenceObject( CacheFileObject);
7737
7738     return;
7739 }
7740
7741 NTSTATUS
7742 AFSInitializeLibrary( IN AFSLibraryInitCB *LibraryInit)
7743 {
7744
7745     NTSTATUS ntStatus = STATUS_SUCCESS;
7746     AFSDeviceExt *pControlDevExt = NULL;
7747     ULONG ulTimeIncrement = 0;
7748     LONG lCount;
7749
7750     __Enter
7751     {
7752
7753         AFSControlDeviceObject = LibraryInit->AFSControlDeviceObject;
7754
7755         AFSRDRDeviceObject = LibraryInit->AFSRDRDeviceObject;
7756
7757         AFSServerName = LibraryInit->AFSServerName;
7758
7759         AFSDebugFlags = LibraryInit->AFSDebugFlags;
7760
7761         //
7762         // Callbacks in the framework
7763         //
7764
7765         AFSProcessRequest = LibraryInit->AFSProcessRequest;
7766
7767         AFSDbgLogMsg = LibraryInit->AFSDbgLogMsg;
7768
7769         AFSAddConnectionEx = LibraryInit->AFSAddConnectionEx;
7770
7771         AFSExAllocatePoolWithTag = LibraryInit->AFSExAllocatePoolWithTag;
7772
7773         AFSExFreePool = LibraryInit->AFSExFreePool;
7774
7775         AFSDumpTraceFilesFnc = LibraryInit->AFSDumpTraceFiles;
7776
7777         AFSRetrieveAuthGroupFnc = LibraryInit->AFSRetrieveAuthGroup;
7778
7779         AFSLibCacheManagerCallbacks = LibraryInit->AFSCacheManagerCallbacks;
7780
7781         if( LibraryInit->AFSCacheBaseAddress != NULL)
7782         {
7783
7784             SetFlag( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE);
7785
7786             AFSLibCacheBaseAddress = LibraryInit->AFSCacheBaseAddress;
7787
7788             AFSLibCacheLength = LibraryInit->AFSCacheLength;
7789         }
7790
7791         //
7792         // Initialize some flush parameters
7793         //
7794
7795         pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
7796
7797         ulTimeIncrement = KeQueryTimeIncrement();
7798
7799         pControlDevExt->Specific.Control.ObjectLifeTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_OBJECT_LIFETIME / (ULONGLONG)ulTimeIncrement);
7800         pControlDevExt->Specific.Control.FcbPurgeTimeCount.QuadPart = AFS_SERVER_PURGE_DELAY;
7801         pControlDevExt->Specific.Control.FcbPurgeTimeCount.QuadPart /= ulTimeIncrement;
7802         pControlDevExt->Specific.Control.FcbFlushTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_SERVER_FLUSH_DELAY / (ULONGLONG)ulTimeIncrement);
7803         pControlDevExt->Specific.Control.ExtentRequestTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_EXTENT_REQUEST_TIME/(ULONGLONG)ulTimeIncrement);
7804
7805         //
7806         // Initialize the global root entry
7807         //
7808
7809         ntStatus = AFSInitVolume( NULL,
7810                                   &LibraryInit->GlobalRootFid,
7811                                   &AFSGlobalRoot);
7812
7813         if( !NT_SUCCESS( ntStatus))
7814         {
7815
7816             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
7817                           AFS_TRACE_LEVEL_ERROR,
7818                           "AFSInitializeLibrary AFSInitVolume failure %08lX\n",
7819                           ntStatus);
7820
7821             try_return( ntStatus);
7822         }
7823
7824         ntStatus = AFSInitRootFcb( (ULONGLONG)PsGetCurrentProcessId(),
7825                                    AFSGlobalRoot);
7826
7827         if( !NT_SUCCESS( ntStatus))
7828         {
7829
7830             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
7831                           AFS_TRACE_LEVEL_ERROR,
7832                           "AFSInitializeLibrary AFSInitRootFcb failure %08lX\n",
7833                           ntStatus);
7834
7835             lCount = InterlockedDecrement( &AFSGlobalRoot->VolumeReferenceCount);
7836
7837             AFSReleaseResource( AFSGlobalRoot->VolumeLock);
7838
7839             try_return( ntStatus);
7840         }
7841
7842         //
7843         // Update the node type code to AFS_ROOT_ALL
7844         //
7845
7846         AFSGlobalRoot->ObjectInformation.Fcb->Header.NodeTypeCode = AFS_ROOT_ALL;
7847
7848         SetFlag( AFSGlobalRoot->Flags, AFS_VOLUME_ACTIVE_GLOBAL_ROOT);
7849
7850         //
7851         // Invalidate all known volumes since contact with the service and therefore
7852         // the file server was lost.
7853         //
7854
7855         AFSInvalidateAllVolumes();
7856
7857         //
7858         // Drop the locks acquired above
7859         //
7860
7861         AFSInitVolumeWorker( AFSGlobalRoot);
7862
7863         lCount = InterlockedDecrement( &AFSGlobalRoot->VolumeReferenceCount);
7864
7865         AFSReleaseResource( AFSGlobalRoot->VolumeLock);
7866
7867         AFSReleaseResource( AFSGlobalRoot->ObjectInformation.Fcb->Header.Resource);
7868
7869 try_exit:
7870
7871         NOTHING;
7872     }
7873
7874     return ntStatus;
7875 }
7876
7877 NTSTATUS
7878 AFSCloseLibrary()
7879 {
7880
7881     NTSTATUS ntStatus = STATUS_SUCCESS;
7882     AFSDirectoryCB *pDirNode = NULL, *pLastDirNode = NULL;
7883
7884     __Enter
7885     {
7886
7887         if( AFSGlobalDotDirEntry != NULL)
7888         {
7889
7890             AFSDeleteObjectInfo( AFSGlobalDotDirEntry->ObjectInformation);
7891
7892             ExDeleteResourceLite( &AFSGlobalDotDirEntry->NonPaged->Lock);
7893
7894             ExFreePool( AFSGlobalDotDirEntry->NonPaged);
7895
7896             ExFreePool( AFSGlobalDotDirEntry);
7897
7898             AFSGlobalDotDirEntry = NULL;
7899         }
7900
7901         if( AFSGlobalDotDotDirEntry != NULL)
7902         {
7903
7904             AFSDeleteObjectInfo( AFSGlobalDotDotDirEntry->ObjectInformation);
7905
7906             ExDeleteResourceLite( &AFSGlobalDotDotDirEntry->NonPaged->Lock);
7907
7908             ExFreePool( AFSGlobalDotDotDirEntry->NonPaged);
7909
7910             ExFreePool( AFSGlobalDotDotDirEntry);
7911
7912             AFSGlobalDotDotDirEntry = NULL;
7913         }
7914
7915         if( AFSSpecialShareNames != NULL)
7916         {
7917
7918             pDirNode = AFSSpecialShareNames;
7919
7920             while( pDirNode != NULL)
7921             {
7922
7923                 pLastDirNode = (AFSDirectoryCB *)pDirNode->ListEntry.fLink;
7924
7925                 AFSDeleteObjectInfo( pDirNode->ObjectInformation);
7926
7927                 ExDeleteResourceLite( &pDirNode->NonPaged->Lock);
7928
7929                 ExFreePool( pDirNode->NonPaged);
7930
7931                 ExFreePool( pDirNode);
7932
7933                 pDirNode = pLastDirNode;
7934             }
7935
7936             AFSSpecialShareNames = NULL;
7937         }
7938     }
7939
7940     return ntStatus;
7941 }
7942
7943 NTSTATUS
7944 AFSDefaultLogMsg( IN ULONG Subsystem,
7945                   IN ULONG Level,
7946                   IN PCCH Format,
7947                   ...)
7948 {
7949
7950     NTSTATUS ntStatus = STATUS_SUCCESS;
7951     va_list va_args;
7952     char chDebugBuffer[ 256];
7953
7954     __Enter
7955     {
7956
7957         va_start( va_args, Format);
7958
7959         ntStatus = RtlStringCbVPrintfA( chDebugBuffer,
7960                                         256,
7961                                         Format,
7962                                         va_args);
7963
7964         if( NT_SUCCESS( ntStatus))
7965         {
7966             DbgPrint( chDebugBuffer);
7967         }
7968
7969         va_end( va_args);
7970     }
7971
7972     return ntStatus;
7973 }
7974
7975 NTSTATUS
7976 AFSGetObjectStatus( IN AFSGetStatusInfoCB *GetStatusInfo,
7977                     IN ULONG InputBufferLength,
7978                     IN AFSStatusInfoCB *StatusInfo,
7979                     OUT ULONG *ReturnLength)
7980 {
7981
7982     NTSTATUS ntStatus = STATUS_SUCCESS;
7983     AFSFcb   *pFcb = NULL;
7984     AFSVolumeCB *pVolumeCB = NULL;
7985     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
7986     AFSObjectInfoCB *pObjectInfo = NULL;
7987     ULONGLONG   ullIndex = 0;
7988     UNICODE_STRING uniFullPathName, uniRemainingPath, uniComponentName, uniParsedName;
7989     AFSNameArrayHdr *pNameArray = NULL;
7990     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
7991     LONG lCount;
7992
7993     __Enter
7994     {
7995
7996         //
7997         // If we are given a FID then look up the entry by that, otherwise
7998         // do it by name
7999         //
8000
8001         if( GetStatusInfo->FileID.Cell != 0 &&
8002             GetStatusInfo->FileID.Volume != 0 &&
8003             GetStatusInfo->FileID.Vnode != 0 &&
8004             GetStatusInfo->FileID.Unique != 0)
8005         {
8006
8007             AFSAcquireShared( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
8008
8009             //
8010             // Locate the volume node
8011             //
8012
8013             ullIndex = AFSCreateHighIndex( &GetStatusInfo->FileID);
8014
8015             ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
8016                                            ullIndex,
8017                                            (AFSBTreeEntry **)&pVolumeCB);
8018
8019             if( pVolumeCB != NULL)
8020             {
8021
8022                 lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
8023
8024                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8025                               AFS_TRACE_LEVEL_VERBOSE,
8026                               "AFSGetObjectStatus Increment count on volume %08lX Cnt %d\n",
8027                               pVolumeCB,
8028                               lCount);
8029             }
8030
8031             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
8032
8033             if( !NT_SUCCESS( ntStatus) ||
8034                 pVolumeCB == NULL)
8035             {
8036                 try_return( ntStatus = STATUS_INVALID_PARAMETER);
8037             }
8038
8039             if( AFSIsVolumeFID( &GetStatusInfo->FileID))
8040             {
8041
8042                 pObjectInfo = &pVolumeCB->ObjectInformation;
8043
8044                 lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8045
8046                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8047             }
8048             else
8049             {
8050
8051                 AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
8052                                   TRUE);
8053
8054                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8055
8056                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8057                               AFS_TRACE_LEVEL_VERBOSE,
8058                               "AFSGetObjectStatus Decrement count on volume %08lX Cnt %d\n",
8059                               pVolumeCB,
8060                               lCount);
8061
8062                 ullIndex = AFSCreateLowIndex( &GetStatusInfo->FileID);
8063
8064                 ntStatus = AFSLocateHashEntry( pVolumeCB->ObjectInfoTree.TreeHead,
8065                                                ullIndex,
8066                                                (AFSBTreeEntry **)&pObjectInfo);
8067
8068                 if( pObjectInfo != NULL)
8069                 {
8070
8071                     //
8072                     // Reference the node so it won't be torn down
8073                     //
8074
8075                     lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8076
8077                     AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
8078                                   AFS_TRACE_LEVEL_VERBOSE,
8079                                   "AFSGetObjectStatus Increment count on object %08lX Cnt %d\n",
8080                                   pObjectInfo,
8081                                   lCount);
8082                 }
8083
8084                 AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
8085
8086                 if( !NT_SUCCESS( ntStatus) ||
8087                     pObjectInfo == NULL)
8088                 {
8089                     try_return( ntStatus = STATUS_INVALID_PARAMETER);
8090                 }
8091             }
8092         }
8093         else
8094         {
8095
8096             if( GetStatusInfo->FileNameLength == 0 ||
8097                 InputBufferLength < (ULONG)(FIELD_OFFSET( AFSGetStatusInfoCB, FileName) + GetStatusInfo->FileNameLength))
8098             {
8099                 try_return( ntStatus = STATUS_INVALID_PARAMETER);
8100             }
8101
8102             uniFullPathName.Length = GetStatusInfo->FileNameLength;
8103             uniFullPathName.MaximumLength = uniFullPathName.Length;
8104
8105             uniFullPathName.Buffer = (WCHAR *)GetStatusInfo->FileName;
8106
8107             //
8108             // This name should begin with the \afs server so parse it off and check it
8109             //
8110
8111             FsRtlDissectName( uniFullPathName,
8112                               &uniComponentName,
8113                               &uniRemainingPath);
8114
8115             if( RtlCompareUnicodeString( &uniComponentName,
8116                                          &AFSServerName,
8117                                          TRUE) != 0)
8118             {
8119                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8120                               AFS_TRACE_LEVEL_ERROR,
8121                               "AFSGetObjectStatus Name %wZ contains invalid server name\n",
8122                               &uniFullPathName);
8123
8124                 try_return( ntStatus = STATUS_OBJECT_PATH_INVALID);
8125             }
8126
8127             uniFullPathName = uniRemainingPath;
8128
8129             uniParsedName = uniFullPathName;
8130
8131             //
8132             // Our name array
8133             //
8134
8135             pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
8136                                            0);
8137
8138             if( pNameArray == NULL)
8139             {
8140                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8141             }
8142
8143             pVolumeCB = AFSGlobalRoot;
8144
8145             pParentDirEntry = AFSGlobalRoot->DirectoryCB;
8146
8147             //
8148             // Increment the ref count on the volume and dir entry for correct processing below
8149             //
8150
8151             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
8152
8153             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8154                           AFS_TRACE_LEVEL_VERBOSE,
8155                           "AFSGetObjectStatus Increment count on volume %08lX Cnt %d\n",
8156                           pVolumeCB,
8157                           lCount);
8158
8159             lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
8160
8161             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8162                           AFS_TRACE_LEVEL_VERBOSE,
8163                           "AFSGetObjectStatus Increment count on %wZ DE %p Ccb %p Cnt %d\n",
8164                           &pParentDirEntry->NameInformation.FileName,
8165                           pParentDirEntry,
8166                           NULL,
8167                           lCount);
8168
8169             ntStatus = AFSLocateNameEntry( NULL,
8170                                            NULL,
8171                                            &uniFullPathName,
8172                                            &uniParsedName,
8173                                            pNameArray,
8174                                            AFS_LOCATE_FLAGS_NO_MP_TARGET_EVAL |
8175                                                         AFS_LOCATE_FLAGS_NO_SL_TARGET_EVAL,
8176                                            &pVolumeCB,
8177                                            &pParentDirEntry,
8178                                            &pDirectoryEntry,
8179                                            NULL);
8180
8181             if( !NT_SUCCESS( ntStatus))
8182             {
8183
8184                 //
8185                 // The volume lock was released on failure above
8186                 // Except for STATUS_OBJECT_NAME_NOT_FOUND
8187                 //
8188
8189                 if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
8190                 {
8191
8192                     if( pVolumeCB != NULL)
8193                     {
8194
8195                         lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8196
8197                         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8198                                       AFS_TRACE_LEVEL_VERBOSE,
8199                                       "AFSGetObjectStatus Decrement count on volume %08lX Cnt %d\n",
8200                                       pVolumeCB,
8201                                       lCount);
8202                     }
8203
8204                     if( pDirectoryEntry != NULL)
8205                     {
8206
8207                         lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
8208
8209                         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8210                                       AFS_TRACE_LEVEL_VERBOSE,
8211                                       "AFSGetObjectStatus Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
8212                                       &pDirectoryEntry->NameInformation.FileName,
8213                                       pDirectoryEntry,
8214                                       NULL,
8215                                       lCount);
8216                     }
8217                     else
8218                     {
8219
8220                         lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
8221
8222                         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8223                                       AFS_TRACE_LEVEL_VERBOSE,
8224                                       "AFSGetObjectStatus Decrement2 count on %wZ DE %p Ccb %p Cnt %d\n",
8225                                       &pParentDirEntry->NameInformation.FileName,
8226                                       pParentDirEntry,
8227                                       NULL,
8228                                       lCount);
8229                     }
8230                 }
8231
8232                 pVolumeCB = NULL;
8233
8234                 try_return( ntStatus);
8235             }
8236
8237             //
8238             // Remove the reference made above
8239             //
8240
8241             lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
8242
8243             pObjectInfo = pDirectoryEntry->ObjectInformation;
8244
8245             lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8246
8247             if( pVolumeCB != NULL)
8248             {
8249
8250                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8251
8252                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8253                               AFS_TRACE_LEVEL_VERBOSE,
8254                               "AFSRetrieveFileAttributes Decrement2 count on volume %08lX Cnt %d\n",
8255                               pVolumeCB,
8256                               pVolumeCB->VolumeReferenceCount);
8257             }
8258         }
8259
8260         //
8261         // At this point we have an object info block, return the information
8262         //
8263
8264         StatusInfo->FileId = pObjectInfo->FileId;
8265
8266         StatusInfo->TargetFileId = pObjectInfo->TargetFileId;
8267
8268         StatusInfo->Expiration = pObjectInfo->Expiration;
8269
8270         StatusInfo->DataVersion = pObjectInfo->DataVersion;
8271
8272         StatusInfo->FileType = pObjectInfo->FileType;
8273
8274         StatusInfo->ObjectFlags = pObjectInfo->Flags;
8275
8276         StatusInfo->CreationTime = pObjectInfo->CreationTime;
8277
8278         StatusInfo->LastAccessTime = pObjectInfo->LastAccessTime;
8279
8280         StatusInfo->LastWriteTime = pObjectInfo->LastWriteTime;
8281
8282         StatusInfo->ChangeTime = pObjectInfo->ChangeTime;
8283
8284         StatusInfo->FileAttributes = pObjectInfo->FileAttributes;
8285
8286         StatusInfo->EndOfFile = pObjectInfo->EndOfFile;
8287
8288         StatusInfo->AllocationSize = pObjectInfo->AllocationSize;
8289
8290         StatusInfo->EaSize = pObjectInfo->EaSize;
8291
8292         StatusInfo->Links = pObjectInfo->Links;
8293
8294         //
8295         // Return the information length
8296         //
8297
8298         *ReturnLength = sizeof( AFSStatusInfoCB);
8299
8300 try_exit:
8301
8302         if( pObjectInfo != NULL)
8303         {
8304
8305             lCount = InterlockedDecrement( &pObjectInfo->ObjectReferenceCount);
8306         }
8307
8308         if( pNameArray != NULL)
8309         {
8310
8311             AFSFreeNameArray( pNameArray);
8312         }
8313     }
8314
8315     return ntStatus;
8316 }
8317
8318 NTSTATUS
8319 AFSCheckSymlinkAccess( IN AFSDirectoryCB *ParentDirectoryCB,
8320                        IN UNICODE_STRING *ComponentName)
8321 {
8322
8323     NTSTATUS ntStatus = STATUS_SUCCESS;
8324     AFSDirectoryCB *pDirEntry = NULL;
8325     ULONG ulCRC = 0;
8326     LONG lCount;
8327
8328     __Enter
8329     {
8330
8331         //
8332         // Search for the entry in the parent
8333         //
8334
8335         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8336                       AFS_TRACE_LEVEL_VERBOSE_2,
8337                       "AFSCheckSymlinkAccess Searching for entry %wZ case sensitive\n",
8338                       ComponentName);
8339
8340         ulCRC = AFSGenerateCRC( ComponentName,
8341                                 FALSE);
8342
8343         AFSAcquireShared( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
8344                           TRUE);
8345
8346         AFSLocateCaseSensitiveDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
8347                                         ulCRC,
8348                                         &pDirEntry);
8349
8350         if( pDirEntry == NULL)
8351         {
8352
8353             //
8354             // Missed so perform a case insensitive lookup
8355             //
8356
8357             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8358                           AFS_TRACE_LEVEL_VERBOSE_2,
8359                           "AFSCheckSymlinkAccess Searching for entry %wZ case insensitive\n",
8360                           ComponentName);
8361
8362             ulCRC = AFSGenerateCRC( ComponentName,
8363                                     TRUE);
8364
8365             AFSLocateCaseInsensitiveDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
8366                                               ulCRC,
8367                                               &pDirEntry);
8368
8369             if( pDirEntry == NULL)
8370             {
8371
8372                 //
8373                 // OK, if this component is a valid short name then try
8374                 // a lookup in the short name tree
8375                 //
8376
8377                 if( RtlIsNameLegalDOS8Dot3( ComponentName,
8378                                             NULL,
8379                                             NULL))
8380                 {
8381
8382                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8383                                   AFS_TRACE_LEVEL_VERBOSE_2,
8384                                   "AFSCheckSymlinkAccess Searching for entry %wZ short name\n",
8385                                   ComponentName);
8386
8387                     AFSLocateShortNameDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.ShortNameTree,
8388                                                 ulCRC,
8389                                                 &pDirEntry);
8390                 }
8391             }
8392         }
8393
8394         if( pDirEntry != NULL)
8395         {
8396             lCount = InterlockedIncrement( &pDirEntry->OpenReferenceCount);
8397         }
8398
8399         AFSReleaseResource( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
8400
8401         if( pDirEntry == NULL)
8402         {
8403
8404             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8405                           AFS_TRACE_LEVEL_VERBOSE_2,
8406                           "AFSCheckSymlinkAccess Failed to locate entry %wZ\n",
8407                           ComponentName);
8408
8409             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
8410         }
8411
8412         //
8413         // We have the symlink object but previously failed to process it so return access
8414         // denied.
8415         //
8416
8417         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8418                       AFS_TRACE_LEVEL_VERBOSE_2,
8419                       "AFSCheckSymlinkAccess Failing symlink access to entry %wZ REPARSE_POINT_NOT_RESOLVED\n",
8420                       ComponentName);
8421
8422         ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
8423
8424         lCount = InterlockedDecrement( &pDirEntry->OpenReferenceCount);
8425
8426 try_exit:
8427
8428         NOTHING;
8429     }
8430
8431     return ntStatus;
8432 }
8433
8434 NTSTATUS
8435 AFSRetrieveFinalComponent( IN UNICODE_STRING *FullPathName,
8436                            OUT UNICODE_STRING *ComponentName)
8437 {
8438
8439     NTSTATUS ntStatus = STATUS_SUCCESS;
8440     UNICODE_STRING uniFullPathName, uniRemainingPath, uniComponentName;
8441
8442     uniFullPathName = *FullPathName;
8443
8444     while( TRUE)
8445     {
8446
8447         FsRtlDissectName( uniFullPathName,
8448                           &uniComponentName,
8449                           &uniRemainingPath);
8450
8451         if( uniRemainingPath.Length == 0)
8452         {
8453             break;
8454         }
8455
8456         uniFullPathName = uniRemainingPath;
8457     }
8458
8459     if( uniComponentName.Length > 0)
8460     {
8461         *ComponentName = uniComponentName;
8462     }
8463
8464     return ntStatus;
8465 }
8466
8467 void
8468 AFSDumpTraceFiles_Default()
8469 {
8470     return;
8471 }
8472
8473 BOOLEAN
8474 AFSValidNameFormat( IN UNICODE_STRING *FileName)
8475 {
8476
8477     BOOLEAN bIsValidName = TRUE;
8478     USHORT usIndex = 0;
8479
8480     __Enter
8481     {
8482
8483         while( usIndex < FileName->Length/sizeof( WCHAR))
8484         {
8485
8486             if( FileName->Buffer[ usIndex] == L':' ||
8487                 FileName->Buffer[ usIndex] == L'*' ||
8488                 FileName->Buffer[ usIndex] == L'?' ||
8489                 FileName->Buffer[ usIndex] == L'"' ||
8490                 FileName->Buffer[ usIndex] == L'<' ||
8491                 FileName->Buffer[ usIndex] == L'>')
8492             {
8493                 bIsValidName = FALSE;
8494                 break;
8495             }
8496
8497             usIndex++;
8498         }
8499     }
8500
8501     return bIsValidName;
8502 }
8503
8504 NTSTATUS
8505 AFSCreateDefaultSecurityDescriptor()
8506 {
8507
8508     NTSTATUS ntStatus = STATUS_SUCCESS;
8509     PACL pSACL = NULL;
8510     ULONG ulSACLSize = 0;
8511     SYSTEM_MANDATORY_LABEL_ACE* pACE = NULL;
8512     ULONG ulACESize = 0;
8513     SECURITY_DESCRIPTOR *pSecurityDescr = NULL;
8514     ULONG ulSDLength = 0;
8515     SECURITY_DESCRIPTOR *pRelativeSecurityDescr = NULL;
8516     PSID pWorldSID = NULL;
8517     ULONG *pulSubAuthority = NULL;
8518     ULONG ulWorldSIDLEngth = 0;
8519
8520     __Enter
8521     {
8522
8523         ulWorldSIDLEngth = RtlLengthRequiredSid( 1);
8524
8525         pWorldSID = (PSID)ExAllocatePoolWithTag( PagedPool,
8526                                                  ulWorldSIDLEngth,
8527                                                  AFS_GENERIC_MEMORY_29_TAG);
8528
8529         if( pWorldSID == NULL)
8530         {
8531             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate World SID\n");
8532             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8533         }
8534
8535         RtlZeroMemory( pWorldSID,
8536                        ulWorldSIDLEngth);
8537
8538         RtlInitializeSid( pWorldSID,
8539                           &SeWorldSidAuthority,
8540                           1);
8541
8542         pulSubAuthority = RtlSubAuthoritySid(pWorldSID, 0);
8543         *pulSubAuthority = SECURITY_WORLD_RID;
8544
8545         if( AFSRtlSetSaclSecurityDescriptor == NULL)
8546         {
8547
8548             AFSPrint( "AFSCreateDefaultSecurityDescriptor AFSRtlSetSaclSecurityDescriptor == NULL\n");
8549         }
8550         else
8551         {
8552
8553             ulACESize = sizeof( SYSTEM_MANDATORY_LABEL_ACE) + 128;
8554
8555             pACE = (SYSTEM_MANDATORY_LABEL_ACE *)ExAllocatePoolWithTag( PagedPool,
8556                                                                         ulACESize,
8557                                                                         AFS_GENERIC_MEMORY_29_TAG);
8558
8559             if( pACE == NULL)
8560             {
8561
8562                 AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_29_TAG\n");
8563
8564                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8565             }
8566
8567             RtlZeroMemory( pACE,
8568                            ulACESize);
8569
8570             pACE->Header.AceFlags = OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE;
8571             pACE->Header.AceType = SYSTEM_MANDATORY_LABEL_ACE_TYPE;
8572             pACE->Header.AceSize = FIELD_OFFSET( SYSTEM_MANDATORY_LABEL_ACE, SidStart) + (USHORT)RtlLengthSid( SeExports->SeLowMandatorySid);
8573             pACE->Mask = SYSTEM_MANDATORY_LABEL_NO_WRITE_UP;
8574
8575             RtlCopySid( RtlLengthSid( SeExports->SeLowMandatorySid),
8576                         &pACE->SidStart,
8577                         SeExports->SeLowMandatorySid);
8578
8579             ulSACLSize = sizeof(ACL) + RtlLengthSid( SeExports->SeLowMandatorySid) +
8580                 FIELD_OFFSET( SYSTEM_MANDATORY_LABEL_ACE, SidStart) + ulACESize;
8581
8582             pSACL = (PACL)ExAllocatePoolWithTag( PagedPool,
8583                                                  ulSACLSize,
8584                                                  AFS_GENERIC_MEMORY_29_TAG);
8585
8586             if( pSACL == NULL)
8587             {
8588
8589                 AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_29_TAG\n");
8590
8591                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8592             }
8593
8594             ntStatus = RtlCreateAcl( pSACL,
8595                                      ulSACLSize,
8596                                      ACL_REVISION);
8597
8598             if( !NT_SUCCESS( ntStatus))
8599             {
8600
8601                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlCreateAcl ntStatus %08lX\n",
8602                           ntStatus);
8603
8604                 try_return( ntStatus);
8605             }
8606
8607             ntStatus = RtlAddAce( pSACL,
8608                                   ACL_REVISION,
8609                                   0,
8610                                   pACE,
8611                                   pACE->Header.AceSize);
8612
8613             if( !NT_SUCCESS( ntStatus))
8614             {
8615
8616                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlAddAce ntStatus %08lX\n",
8617                           ntStatus);
8618
8619                 try_return( ntStatus);
8620             }
8621         }
8622
8623         pSecurityDescr = (SECURITY_DESCRIPTOR *)ExAllocatePoolWithTag( NonPagedPool,
8624                                                                        sizeof( SECURITY_DESCRIPTOR),
8625                                                                        AFS_GENERIC_MEMORY_27_TAG);
8626
8627         if( pSecurityDescr == NULL)
8628         {
8629
8630             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_27_TAG\n");
8631
8632             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8633         }
8634
8635         ntStatus = RtlCreateSecurityDescriptor( pSecurityDescr,
8636                                                 SECURITY_DESCRIPTOR_REVISION);
8637
8638         if( !NT_SUCCESS( ntStatus))
8639         {
8640
8641             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlCreateSecurityDescriptor ntStatus %08lX\n",
8642                       ntStatus);
8643
8644             try_return( ntStatus);
8645         }
8646
8647         if( AFSRtlSetSaclSecurityDescriptor != NULL)
8648         {
8649             ntStatus = AFSRtlSetSaclSecurityDescriptor( pSecurityDescr,
8650                                                         TRUE,
8651                                                         pSACL,
8652                                                         FALSE);
8653
8654             if( !NT_SUCCESS( ntStatus))
8655             {
8656
8657                 AFSPrint( "AFSCreateDefaultSecurityDescriptor AFSRtlSetSaclSecurityDescriptor ntStatus %08lX\n",
8658                           ntStatus);
8659
8660                 try_return( ntStatus);
8661             }
8662         }
8663
8664         //
8665         // Add in the group and owner to the SD
8666         //
8667
8668         if( AFSRtlSetGroupSecurityDescriptor != NULL)
8669         {
8670             ntStatus = AFSRtlSetGroupSecurityDescriptor( pSecurityDescr,
8671                                                          pWorldSID,
8672                                                          FALSE);
8673
8674             if( !NT_SUCCESS( ntStatus))
8675             {
8676
8677                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlSetGroupSecurityDescriptor failed ntStatus %08lX\n",
8678                           ntStatus);
8679
8680                 try_return( ntStatus);
8681             }
8682         }
8683
8684         ntStatus = RtlSetOwnerSecurityDescriptor( pSecurityDescr,
8685                                                   pWorldSID,
8686                                                   FALSE);
8687
8688         if( !NT_SUCCESS( ntStatus))
8689         {
8690
8691             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlSetOwnerSecurityDescriptor failed ntStatus %08lX\n",
8692                       ntStatus);
8693
8694             try_return( ntStatus);
8695         }
8696
8697         if( !RtlValidSecurityDescriptor( pSecurityDescr))
8698         {
8699
8700             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlValidSecurityDescriptor NOT\n");
8701
8702             try_return( ntStatus = STATUS_INVALID_PARAMETER);
8703         }
8704
8705         pRelativeSecurityDescr = (SECURITY_DESCRIPTOR *)ExAllocatePoolWithTag( NonPagedPool,
8706                                                                                PAGE_SIZE,
8707                                                                                AFS_GENERIC_MEMORY_27_TAG);
8708
8709         if( pRelativeSecurityDescr == NULL)
8710         {
8711
8712             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_27_TAG\n");
8713
8714             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8715         }
8716
8717         ulSDLength = PAGE_SIZE;
8718
8719         ntStatus = RtlAbsoluteToSelfRelativeSD( pSecurityDescr,
8720                                                 pRelativeSecurityDescr,
8721                                                 &ulSDLength);
8722
8723         if( !NT_SUCCESS( ntStatus))
8724         {
8725
8726             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlAbsoluteToSelfRelativeSD ntStatus %08lX\n",
8727                       ntStatus);
8728
8729             try_return( ntStatus);
8730         }
8731
8732         AFSDefaultSD = pRelativeSecurityDescr;
8733
8734 try_exit:
8735
8736         if( !NT_SUCCESS( ntStatus))
8737         {
8738
8739             if( pRelativeSecurityDescr != NULL)
8740             {
8741                 ExFreePool( pRelativeSecurityDescr);
8742             }
8743         }
8744
8745         if( pSecurityDescr != NULL)
8746         {
8747             ExFreePool( pSecurityDescr);
8748         }
8749
8750         if( pSACL != NULL)
8751         {
8752             ExFreePool( pSACL);
8753         }
8754
8755         if( pACE != NULL)
8756         {
8757             ExFreePool( pACE);
8758         }
8759
8760         if( pWorldSID != NULL)
8761         {
8762             ExFreePool( pWorldSID);
8763         }
8764     }
8765
8766     return ntStatus;
8767 }
8768
8769 void
8770 AFSRetrieveParentPath( IN UNICODE_STRING *FullFileName,
8771                        OUT UNICODE_STRING *ParentPath)
8772 {
8773
8774     USHORT usIndex = 0;
8775
8776     *ParentPath = *FullFileName;
8777
8778     //
8779     // If the final character is a \, jump over it
8780     //
8781
8782     if( ParentPath->Buffer[ (ParentPath->Length/sizeof( WCHAR)) - 1] == L'\\')
8783     {
8784         ParentPath->Length -= sizeof( WCHAR);
8785     }
8786
8787     while( ParentPath->Buffer[ (ParentPath->Length/sizeof( WCHAR)) - 1] != L'\\')
8788     {
8789         ParentPath->Length -= sizeof( WCHAR);
8790     }
8791
8792     //
8793     // And the separator
8794     //
8795
8796     ParentPath->Length -= sizeof( WCHAR);
8797
8798     return;
8799 }
8800
8801 NTSTATUS
8802 AFSRetrieveValidAuthGroup( IN AFSFcb *Fcb,
8803                            IN AFSObjectInfoCB *ObjectInfo,
8804                            IN BOOLEAN WriteAccess,
8805                            OUT GUID *AuthGroup)
8806 {
8807
8808     NTSTATUS ntStatus = STATUS_SUCCESS;
8809     GUID     stAuthGroup, stZeroAuthGroup;
8810     BOOLEAN  bFoundAuthGroup = FALSE;
8811     AFSCcb  *pCcb = NULL;
8812     AFSFcb *pFcb = Fcb;
8813
8814     __Enter
8815     {
8816
8817         RtlZeroMemory( &stAuthGroup,
8818                        sizeof( GUID));
8819
8820         RtlZeroMemory( &stZeroAuthGroup,
8821                        sizeof( GUID));
8822
8823         if( Fcb == NULL)
8824         {
8825
8826             if( ObjectInfo != NULL &&
8827                 ObjectInfo->Fcb != NULL)
8828             {
8829                 pFcb = ObjectInfo->Fcb;
8830             }
8831         }
8832
8833         if( pFcb != NULL)
8834         {
8835
8836             AFSAcquireShared( &Fcb->NPFcb->CcbListLock,
8837                               TRUE);
8838
8839             pCcb = Fcb->CcbListHead;
8840
8841             while( pCcb != NULL)
8842             {
8843
8844                 if( WriteAccess &&
8845                     pCcb->GrantedAccess & FILE_WRITE_DATA)
8846                 {
8847                     RtlCopyMemory( &stAuthGroup,
8848                                    &pCcb->AuthGroup,
8849                                    sizeof( GUID));
8850
8851                     bFoundAuthGroup = TRUE;
8852
8853                     break;
8854                 }
8855                 else if( pCcb->GrantedAccess & FILE_READ_DATA)
8856                 {
8857                     //
8858                     // At least get the read-only access
8859                     //
8860
8861                     RtlCopyMemory( &stAuthGroup,
8862                                    &pCcb->AuthGroup,
8863                                    sizeof( GUID));
8864
8865                     bFoundAuthGroup = TRUE;
8866                 }
8867
8868                 pCcb = (AFSCcb *)pCcb->ListEntry.fLink;
8869             }
8870
8871             AFSReleaseResource( &Fcb->NPFcb->CcbListLock);
8872         }
8873
8874         if( !bFoundAuthGroup)
8875         {
8876
8877             AFSRetrieveAuthGroupFnc( (ULONGLONG)PsGetCurrentProcessId(),
8878                                      (ULONGLONG)PsGetCurrentThreadId(),
8879                                       &stAuthGroup);
8880
8881             if( RtlCompareMemory( &stZeroAuthGroup,
8882                                   &stAuthGroup,
8883                                   sizeof( GUID)) == sizeof( GUID))
8884             {
8885
8886                 DbgPrint("AFSRetrieveValidAuthGroup Failed to locate PAG\n");
8887
8888                 try_return( ntStatus = STATUS_ACCESS_DENIED);
8889             }
8890         }
8891
8892         RtlCopyMemory( AuthGroup,
8893                        &stAuthGroup,
8894                        sizeof( GUID));
8895
8896 try_exit:
8897
8898         NOTHING;
8899     }
8900
8901     return ntStatus;
8902 }
8903
8904 NTSTATUS
8905 AFSPerformObjectInvalidate( IN AFSObjectInfoCB *ObjectInfo,
8906                             IN ULONG InvalidateReason)
8907 {
8908
8909     NTSTATUS            ntStatus = STATUS_SUCCESS;
8910     IO_STATUS_BLOCK     stIoStatus;
8911     LIST_ENTRY         *le;
8912     AFSExtent          *pEntry;
8913     ULONG               ulProcessCount = 0;
8914     ULONG               ulCount = 0;
8915
8916     __Enter
8917     {
8918
8919         switch( InvalidateReason)
8920         {
8921
8922             case AFS_INVALIDATE_DELETED:
8923             {
8924
8925                 if( ObjectInfo->FileType == AFS_FILE_TYPE_FILE &&
8926                     ObjectInfo->Fcb != NULL)
8927                 {
8928
8929
8930                     //
8931                     // Clear out the extents
8932                     // And get rid of them (note this involves waiting
8933                     // for any writes or reads to the cache to complete)
8934                     //
8935
8936                     AFSTearDownFcbExtents( ObjectInfo->Fcb,
8937                                            NULL);
8938                 }
8939
8940                 break;
8941             }
8942
8943             case AFS_INVALIDATE_DATA_VERSION:
8944             {
8945
8946                 LARGE_INTEGER liCurrentOffset = {0,0};
8947                 LARGE_INTEGER liFlushLength = {0,0};
8948                 ULONG ulFlushLength = 0;
8949                 BOOLEAN bLocked = FALSE;
8950                 BOOLEAN bExtentsLocked = FALSE;
8951                 BOOLEAN bCleanExtents = FALSE;
8952
8953                 if( ObjectInfo->FileType == AFS_FILE_TYPE_FILE &&
8954                     ObjectInfo->Fcb != NULL)
8955                 {
8956
8957                     AFSAcquireExcl( &ObjectInfo->Fcb->NPFcb->Resource,
8958                                     TRUE);
8959
8960                     bLocked = TRUE;
8961
8962                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
8963                                   AFS_TRACE_LEVEL_VERBOSE,
8964                                   "AFSPerformObjectInvalidate Acquiring Fcb extents lock %08lX SHARED %08lX\n",
8965                                   &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource,
8966                                   PsGetCurrentThread());
8967
8968                     AFSAcquireShared( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource,
8969                                       TRUE);
8970
8971                     bExtentsLocked = TRUE;
8972
8973                     //
8974                     // There are several possibilities here:
8975                     //
8976                     // 0. If there are no extents or all of the extents are dirty, do nothing.
8977                     //
8978                     // 1. There could be nothing dirty and an open reference count of zero
8979                     //    in which case we can just tear down all of the extents without
8980                     //    holding any resources.
8981                     //
8982                     // 2. There could be nothing dirty and a non-zero open reference count
8983                     //    in which case we can issue a CcPurge against the entire file
8984                     //    while holding just the Fcb Resource.
8985                     //
8986                     // 3. There can be dirty extents in which case we need to identify
8987                     //    the non-dirty ranges and then perform a CcPurge on just the
8988                     //    non-dirty ranges while holding just the Fcb Resource.
8989                     //
8990
8991                     if ( ObjectInfo->Fcb->Specific.File.ExtentCount != ObjectInfo->Fcb->Specific.File.ExtentsDirtyCount)
8992                     {
8993
8994                         if ( ObjectInfo->Fcb->Specific.File.ExtentsDirtyCount == 0)
8995                         {
8996
8997                             if ( ObjectInfo->Fcb->OpenReferenceCount == 0)
8998                             {
8999
9000                                 AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9001
9002                                 bExtentsLocked = FALSE;
9003
9004                                 AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Resource);
9005
9006                                 bLocked = FALSE;
9007
9008                                 AFSTearDownFcbExtents( ObjectInfo->Fcb,
9009                                                        NULL);
9010                             }
9011                             else
9012                             {
9013
9014                                 __try
9015                                 {
9016
9017                                     AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9018
9019                                     bExtentsLocked = FALSE;
9020
9021                                     if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9022                                                               NULL,
9023                                                               0,
9024                                                               FALSE))
9025                                     {
9026
9027                                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9028                                                       AFS_TRACE_LEVEL_WARNING,
9029                                                       "AFSPerformObjectInvalidation CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9030                                                       ObjectInfo->FileId.Cell,
9031                                                       ObjectInfo->FileId.Volume,
9032                                                       ObjectInfo->FileId.Vnode,
9033                                                       ObjectInfo->FileId.Unique);
9034
9035                                         SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9036                                     }
9037                                     else
9038                                     {
9039
9040                                         bCleanExtents = TRUE;
9041                                     }
9042                                 }
9043                                 __except( EXCEPTION_EXECUTE_HANDLER)
9044                                 {
9045
9046                                     ntStatus = GetExceptionCode();
9047
9048                                     AFSDbgLogMsg( 0,
9049                                                   0,
9050                                                   "EXCEPTION - AFSPerformObjectInvalidation FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
9051                                                   ObjectInfo->FileId.Cell,
9052                                                   ObjectInfo->FileId.Volume,
9053                                                   ObjectInfo->FileId.Vnode,
9054                                                   ObjectInfo->FileId.Unique,
9055                                                   ntStatus);
9056
9057                                     SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9058                                 }
9059                             }
9060                         }
9061                         else
9062                         {
9063
9064                             //
9065                             // Must build a list of non-dirty ranges from the beginning of the file
9066                             // to the end.  There can be at most (Fcb->Specific.File.ExtentsDirtyCount + 1)
9067                             // ranges.  In all but the most extreme random data write scenario there will
9068                             // be significantly fewer.
9069                             //
9070                             // For each range we need offset and size.
9071                             //
9072
9073                             AFSByteRange * ByteRangeList = NULL;
9074                             ULONG          ulByteRangeCount = 0;
9075                             ULONG          ulIndex;
9076                             BOOLEAN        bPurgeOnClose = FALSE;
9077
9078                             __try
9079                             {
9080
9081                                 ulByteRangeCount = AFSConstructCleanByteRangeList( ObjectInfo->Fcb,
9082                                                                                    &ByteRangeList);
9083
9084                                 if ( ByteRangeList != NULL ||
9085                                      ulByteRangeCount == 0)
9086                                 {
9087
9088                                     AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9089
9090                                     bExtentsLocked = FALSE;
9091
9092                                     for ( ulIndex = 0; ulIndex < ulByteRangeCount; ulIndex++)
9093                                     {
9094
9095                                         ULONG ulSize;
9096
9097                                         do {
9098
9099                                             ulSize = ByteRangeList[ulIndex].Length.QuadPart > DWORD_MAX ? DWORD_MAX : ByteRangeList[ulIndex].Length.LowPart;
9100
9101                                             if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9102                                                                       &ByteRangeList[ulIndex].FileOffset,
9103                                                                       ulSize,
9104                                                                       FALSE))
9105                                             {
9106
9107                                                 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9108                                                               AFS_TRACE_LEVEL_WARNING,
9109                                                               "AFSPerformObjectInvalidation [1] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9110                                                               ObjectInfo->FileId.Cell,
9111                                                               ObjectInfo->FileId.Volume,
9112                                                               ObjectInfo->FileId.Vnode,
9113                                                               ObjectInfo->FileId.Unique);
9114
9115                                                 bPurgeOnClose = TRUE;
9116                                             }
9117                                             else
9118                                             {
9119
9120                                                 bCleanExtents = TRUE;
9121                                             }
9122
9123                                             ByteRangeList[ulIndex].Length.QuadPart -= ulSize;
9124
9125                                             ByteRangeList[ulIndex].FileOffset.QuadPart += ulSize;
9126
9127                                         } while ( ByteRangeList[ulIndex].Length.QuadPart > 0);
9128                                     }
9129                                 }
9130                                 else
9131                                 {
9132
9133                                     //
9134                                     // We couldn't allocate the memory to build the purge list
9135                                     // so just walk the extent list while holding the ExtentsList Resource.
9136                                     // This could deadlock but we do not have much choice.
9137                                     //
9138
9139                                     le = ObjectInfo->Fcb->Specific.File.ExtentsLists[AFS_EXTENTS_LIST].Flink;
9140
9141                                     ulProcessCount = 0;
9142
9143                                     ulCount = (ULONG)ObjectInfo->Fcb->Specific.File.ExtentCount;
9144
9145                                     if( ulCount > 0)
9146                                     {
9147                                         pEntry = ExtentFor( le, AFS_EXTENTS_LIST );
9148
9149                                         while( ulProcessCount < ulCount)
9150                                         {
9151                                             pEntry = ExtentFor( le, AFS_EXTENTS_LIST );
9152
9153                                             if( !BooleanFlagOn( pEntry->Flags, AFS_EXTENT_DIRTY))
9154                                             {
9155                                                 if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9156                                                                           &pEntry->FileOffset,
9157                                                                           pEntry->Size,
9158                                                                           FALSE))
9159                                                 {
9160
9161                                                     AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9162                                                                   AFS_TRACE_LEVEL_WARNING,
9163                                                                   "AFSPerformObjectInvalidation [2] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9164                                                                   ObjectInfo->FileId.Cell,
9165                                                                   ObjectInfo->FileId.Volume,
9166                                                                   ObjectInfo->FileId.Vnode,
9167                                                                   ObjectInfo->FileId.Unique);
9168
9169                                                     bPurgeOnClose = TRUE;
9170                                                 }
9171                                                 else
9172                                                 {
9173
9174                                                     bCleanExtents = TRUE;
9175                                                 }
9176                                             }
9177
9178                                             if( liCurrentOffset.QuadPart < pEntry->FileOffset.QuadPart)
9179                                             {
9180
9181                                                 liFlushLength.QuadPart = pEntry->FileOffset.QuadPart - liCurrentOffset.QuadPart;
9182
9183                                                 while( liFlushLength.QuadPart > 0)
9184                                                 {
9185
9186                                                     if( liFlushLength.QuadPart > 512 * 1024000)
9187                                                     {
9188                                                         ulFlushLength = 512 * 1024000;
9189                                                     }
9190                                                     else
9191                                                     {
9192                                                         ulFlushLength = liFlushLength.LowPart;
9193                                                     }
9194
9195                                                     if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9196                                                                               &liCurrentOffset,
9197                                                                               ulFlushLength,
9198                                                                               FALSE))
9199                                                     {
9200
9201                                                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9202                                                                       AFS_TRACE_LEVEL_WARNING,
9203                                                                       "AFSPerformObjectInvalidation [3] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9204                                                                       ObjectInfo->FileId.Cell,
9205                                                                       ObjectInfo->FileId.Volume,
9206                                                                       ObjectInfo->FileId.Vnode,
9207                                                                       ObjectInfo->FileId.Unique);
9208
9209                                                         bPurgeOnClose = TRUE;
9210                                                     }
9211                                                     else
9212                                                     {
9213
9214                                                         bCleanExtents = TRUE;
9215                                                     }
9216
9217                                                     liFlushLength.QuadPart -= ulFlushLength;
9218                                                 }
9219                                             }
9220
9221                                             liCurrentOffset.QuadPart = pEntry->FileOffset.QuadPart + pEntry->Size;
9222
9223                                             ulProcessCount++;
9224                                             le = le->Flink;
9225                                         }
9226                                     }
9227                                     else
9228                                     {
9229                                         if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9230                                                                   NULL,
9231                                                                   0,
9232                                                                   FALSE))
9233                                         {
9234
9235                                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9236                                                           AFS_TRACE_LEVEL_WARNING,
9237                                                           "AFSPerformObjectInvalidation [4] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9238                                                           ObjectInfo->FileId.Cell,
9239                                                           ObjectInfo->FileId.Volume,
9240                                                           ObjectInfo->FileId.Vnode,
9241                                                           ObjectInfo->FileId.Unique);
9242
9243                                             bPurgeOnClose = TRUE;
9244                                         }
9245                                         else
9246                                         {
9247
9248                                             bCleanExtents = TRUE;
9249                                         }
9250                                     }
9251
9252                                     if ( bPurgeOnClose)
9253                                     {
9254
9255                                         SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9256                                     }
9257                                 }
9258                             }
9259                             __except( EXCEPTION_EXECUTE_HANDLER)
9260                             {
9261
9262                                 ntStatus = GetExceptionCode();
9263
9264                                 AFSDbgLogMsg( 0,
9265                                               0,
9266                                               "EXCEPTION - AFSPerformObjectInvalidation FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
9267                                               ObjectInfo->FileId.Cell,
9268                                               ObjectInfo->FileId.Volume,
9269                                               ObjectInfo->FileId.Vnode,
9270                                               ObjectInfo->FileId.Unique,
9271                                               ntStatus);
9272                             }
9273                         }
9274                     }
9275
9276                     if ( bExtentsLocked)
9277                     {
9278
9279                         AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9280                     }
9281
9282                     if ( bLocked)
9283                     {
9284
9285                         AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Resource);
9286                     }
9287
9288                     if ( bCleanExtents)
9289                     {
9290
9291                         AFSReleaseCleanExtents( ObjectInfo->Fcb,
9292                                                 NULL);
9293                     }
9294                 }
9295
9296                 break;
9297             }
9298         }
9299
9300         if( ObjectInfo != NULL)
9301         {
9302             InterlockedDecrement( &ObjectInfo->ObjectReferenceCount);
9303         }
9304     }
9305
9306     return ntStatus;
9307 }