Windows: memory leak AFSInitPIOCtlDirectoryCB
[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                 AFSExFreePoolWithTag( pNonPagedDirEntry, AFS_DIR_ENTRY_NP_TAG);
1188             }
1189
1190             if( pDirNode != NULL)
1191             {
1192
1193                 AFSExFreePoolWithTag( pDirNode, AFS_DIR_ENTRY_TAG);
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             AFSExFreePoolWithTag( pDirEntry, AFS_GENERIC_MEMORY_2_TAG);
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             AFSExFreePoolWithTag( pDirEntry, AFS_GENERIC_MEMORY_2_TAG);
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             AFSExFreePoolWithTag( FullPathName->Buffer, 0);
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             AFSExFreePoolWithTag( pDirEnumEntry, AFS_GENERIC_MEMORY_2_TAG);
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                 AFSExFreePoolWithTag( DirEntry->NameInformation.TargetName.Buffer, AFS_NAME_BUFFER_FIVE_TAG);
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             AFSExFreePoolWithTag( pDirEnumEntry, AFS_GENERIC_MEMORY_2_TAG);
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         AFSExFreePoolWithTag( uniFullName.Buffer, 0);
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                 AFSExFreePoolWithTag( TargetName->Buffer, AFS_NAME_BUFFER_FIVE_TAG);
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         AFSExFreePoolWithTag( NameArray, AFS_NAME_ARRAY_TAG);
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                 AFSExFreePoolWithTag( pDirNode, AFS_DIR_ENTRY_TAG);
5943             }
5944
5945             if( pNonPagedDirEntry != NULL)
5946             {
5947
5948                 ExDeleteResourceLite( &pNonPagedDirEntry->Lock);
5949
5950                 AFSExFreePoolWithTag( pNonPagedDirEntry, AFS_DIR_ENTRY_NP_TAG);
5951             }
5952
5953             if ( pObjectInfoCB != NULL)
5954             {
5955
5956                 AFSDeleteObjectInfo( pObjectInfoCB);
5957             }
5958         }
5959     }
5960
5961     return ntStatus;
5962 }
5963
5964 NTSTATUS
5965 AFSRetrieveFileAttributes( IN AFSDirectoryCB *ParentDirectoryCB,
5966                            IN AFSDirectoryCB *DirectoryCB,
5967                            IN UNICODE_STRING *ParentPathName,
5968                            IN AFSNameArrayHdr *RelatedNameArray,
5969                            IN GUID           *AuthGroup,
5970                            OUT AFSFileInfoCB *FileInfo)
5971 {
5972
5973     NTSTATUS ntStatus = STATUS_SUCCESS;
5974     AFSDirEnumEntry *pDirEntry = NULL, *pLastDirEntry = NULL;
5975     UNICODE_STRING uniFullPathName;
5976     AFSNameArrayHdr    *pNameArray = NULL;
5977     AFSVolumeCB *pVolumeCB = NULL;
5978     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
5979     WCHAR *pwchBuffer = NULL;
5980     UNICODE_STRING uniComponentName, uniRemainingPath, uniParsedName;
5981     ULONG ulNameDifference = 0;
5982     LONG lCount;
5983
5984     __Enter
5985     {
5986
5987         //
5988         // Retrieve a target name for the entry
5989         //
5990
5991         AFSAcquireShared( &DirectoryCB->NonPaged->Lock,
5992                           TRUE);
5993
5994         if( DirectoryCB->NameInformation.TargetName.Length == 0)
5995         {
5996
5997             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
5998
5999             ntStatus = AFSEvaluateTargetByID( DirectoryCB->ObjectInformation,
6000                                               AuthGroup,
6001                                               FALSE,
6002                                               &pDirEntry);
6003
6004             if( !NT_SUCCESS( ntStatus) ||
6005                 pDirEntry->TargetNameLength == 0)
6006             {
6007
6008                 if( pDirEntry != NULL)
6009                 {
6010
6011                     ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
6012                 }
6013
6014                 try_return( ntStatus);
6015             }
6016
6017             AFSAcquireExcl( &DirectoryCB->NonPaged->Lock,
6018                             TRUE);
6019
6020             if( DirectoryCB->NameInformation.TargetName.Length == 0)
6021             {
6022
6023                 //
6024                 // Update the target name
6025                 //
6026
6027                 ntStatus = AFSUpdateTargetName( &DirectoryCB->NameInformation.TargetName,
6028                                                 &DirectoryCB->Flags,
6029                                                 (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset),
6030                                                 (USHORT)pDirEntry->TargetNameLength);
6031
6032                 if( !NT_SUCCESS( ntStatus))
6033                 {
6034
6035                     AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6036
6037                     try_return( ntStatus);
6038                 }
6039             }
6040
6041             AFSConvertToShared( &DirectoryCB->NonPaged->Lock);
6042         }
6043
6044         //
6045         // Need to pass the full path in for parsing.
6046         //
6047
6048         if( AFSIsRelativeName( &DirectoryCB->NameInformation.TargetName))
6049         {
6050
6051             uniFullPathName.Length = 0;
6052             uniFullPathName.MaximumLength = ParentPathName->Length +
6053                                                     sizeof( WCHAR) +
6054                                                     DirectoryCB->NameInformation.TargetName.Length;
6055
6056             uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6057                                                                         uniFullPathName.MaximumLength,
6058                                                                         AFS_NAME_BUFFER_SIX_TAG);
6059
6060             if( uniFullPathName.Buffer == NULL)
6061             {
6062
6063                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6064
6065                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6066             }
6067
6068             pwchBuffer = uniFullPathName.Buffer;
6069
6070             RtlZeroMemory( uniFullPathName.Buffer,
6071                            uniFullPathName.MaximumLength);
6072
6073             RtlCopyMemory( uniFullPathName.Buffer,
6074                            ParentPathName->Buffer,
6075                            ParentPathName->Length);
6076
6077             uniFullPathName.Length = ParentPathName->Length;
6078
6079             if( uniFullPathName.Buffer[ (uniFullPathName.Length/sizeof( WCHAR)) - 1] != L'\\' &&
6080                 DirectoryCB->NameInformation.TargetName.Buffer[ 0] != L'\\')
6081             {
6082
6083                 uniFullPathName.Buffer[ uniFullPathName.Length/sizeof( WCHAR)] = L'\\';
6084
6085                 uniFullPathName.Length += sizeof( WCHAR);
6086             }
6087
6088             RtlCopyMemory( &uniFullPathName.Buffer[ uniFullPathName.Length/sizeof( WCHAR)],
6089                            DirectoryCB->NameInformation.TargetName.Buffer,
6090                            DirectoryCB->NameInformation.TargetName.Length);
6091
6092             uniFullPathName.Length += DirectoryCB->NameInformation.TargetName.Length;
6093
6094             uniParsedName.Length = uniFullPathName.Length - ParentPathName->Length;
6095             uniParsedName.MaximumLength = uniParsedName.Length;
6096
6097             uniParsedName.Buffer = &uniFullPathName.Buffer[ ParentPathName->Length/sizeof( WCHAR)];
6098
6099             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6100
6101             //
6102             // We populate up to the current parent
6103             //
6104
6105             if( RelatedNameArray != NULL)
6106             {
6107
6108                 pNameArray = AFSInitNameArray( NULL,
6109                                                RelatedNameArray->MaxElementCount);
6110
6111                 if( pNameArray == NULL)
6112                 {
6113
6114                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6115                 }
6116
6117                 ntStatus = AFSPopulateNameArrayFromRelatedArray( pNameArray,
6118                                                                  RelatedNameArray,
6119                                                                  ParentDirectoryCB);
6120             }
6121             else
6122             {
6123
6124                 pNameArray = AFSInitNameArray( NULL,
6125                                                0);
6126
6127                 if( pNameArray == NULL)
6128                 {
6129
6130                     try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6131                 }
6132
6133                 ntStatus = AFSPopulateNameArray( pNameArray,
6134                                                  NULL,
6135                                                  ParentDirectoryCB);
6136             }
6137
6138             if( !NT_SUCCESS( ntStatus))
6139             {
6140
6141                 try_return( ntStatus);
6142             }
6143
6144             pVolumeCB = ParentDirectoryCB->ObjectInformation->VolumeCB;
6145
6146             pParentDirEntry = ParentDirectoryCB;
6147         }
6148         else
6149         {
6150
6151             uniFullPathName.Length = 0;
6152             uniFullPathName.MaximumLength = DirectoryCB->NameInformation.TargetName.Length;
6153
6154             uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6155                                                                         uniFullPathName.MaximumLength,
6156                                                                         AFS_NAME_BUFFER_SEVEN_TAG);
6157
6158             if( uniFullPathName.Buffer == NULL)
6159             {
6160
6161                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6162
6163                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6164             }
6165
6166             pwchBuffer = uniFullPathName.Buffer;
6167
6168             RtlZeroMemory( uniFullPathName.Buffer,
6169                            uniFullPathName.MaximumLength);
6170
6171             RtlCopyMemory( uniFullPathName.Buffer,
6172                            DirectoryCB->NameInformation.TargetName.Buffer,
6173                            DirectoryCB->NameInformation.TargetName.Length);
6174
6175             uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6176
6177             //
6178             // This name should begin with the \afs server so parse it off and check it
6179             //
6180
6181             FsRtlDissectName( uniFullPathName,
6182                               &uniComponentName,
6183                               &uniRemainingPath);
6184
6185             if( RtlCompareUnicodeString( &uniComponentName,
6186                                          &AFSServerName,
6187                                          TRUE) != 0)
6188             {
6189
6190                 AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6191
6192                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
6193                               AFS_TRACE_LEVEL_ERROR,
6194                               "AFSRetrieveFileAttributes Name %wZ contains invalid server name\n",
6195                               &uniFullPathName);
6196
6197                 try_return( ntStatus = STATUS_OBJECT_PATH_INVALID);
6198             }
6199
6200             uniFullPathName = uniRemainingPath;
6201
6202             uniParsedName = uniFullPathName;
6203
6204             ulNameDifference = (ULONG)(uniFullPathName.Length > 0 ? ((char *)uniFullPathName.Buffer - (char *)pwchBuffer) : 0);
6205
6206             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6207
6208             //
6209             // Our name array
6210             //
6211
6212             pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
6213                                            0);
6214
6215             if( pNameArray == NULL)
6216             {
6217
6218                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6219             }
6220
6221             pVolumeCB = AFSGlobalRoot;
6222
6223             pParentDirEntry = AFSGlobalRoot->DirectoryCB;
6224         }
6225
6226         //
6227         // Increment the ref count on the volume and dir entry for correct processing below
6228         //
6229
6230         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
6231
6232         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6233                       AFS_TRACE_LEVEL_VERBOSE,
6234                       "AFSRetrieveFileAttributes Increment count on volume %08lX Cnt %d\n",
6235                       pVolumeCB,
6236                       lCount);
6237
6238         lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
6239
6240         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6241                       AFS_TRACE_LEVEL_VERBOSE,
6242                       "AFSRetrieveFileAttributes Increment count on %wZ DE %p Ccb %p Cnt %d\n",
6243                       &pParentDirEntry->NameInformation.FileName,
6244                       pParentDirEntry,
6245                       NULL,
6246                       lCount);
6247
6248         ntStatus = AFSLocateNameEntry( NULL,
6249                                        NULL,
6250                                        &uniFullPathName,
6251                                        &uniParsedName,
6252                                        pNameArray,
6253                                        AFS_LOCATE_FLAGS_NO_MP_TARGET_EVAL,
6254                                        &pVolumeCB,
6255                                        &pParentDirEntry,
6256                                        &pDirectoryEntry,
6257                                        NULL);
6258
6259         if( !NT_SUCCESS( ntStatus))
6260         {
6261
6262             //
6263             // The volume lock was released on failure above
6264             // Except for STATUS_OBJECT_NAME_NOT_FOUND
6265             //
6266
6267             if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
6268             {
6269
6270                 if( pVolumeCB != NULL)
6271                 {
6272
6273                     lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6274
6275                     AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6276                                   AFS_TRACE_LEVEL_VERBOSE,
6277                                   "AFSRetrieveFileAttributes Decrement count on volume %08lX Cnt %d\n",
6278                                   pVolumeCB,
6279                                   lCount);
6280                 }
6281
6282                 if( pDirectoryEntry != NULL)
6283                 {
6284
6285                     lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6286
6287                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6288                                   AFS_TRACE_LEVEL_VERBOSE,
6289                                   "AFSRetrieveFileAttributes Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6290                                   &pDirectoryEntry->NameInformation.FileName,
6291                                   pDirectoryEntry,
6292                                   NULL,
6293                                   lCount);
6294                 }
6295                 else
6296                 {
6297
6298                     lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
6299
6300                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6301                                   AFS_TRACE_LEVEL_VERBOSE,
6302                                   "AFSRetrieveFileAttributes Decrement2 count on %wZ DE %p Ccb %p Cnt %d\n",
6303                                   &pParentDirEntry->NameInformation.FileName,
6304                                   pParentDirEntry,
6305                                   NULL,
6306                                   lCount);
6307                 }
6308             }
6309
6310             pVolumeCB = NULL;
6311
6312             try_return( ntStatus);
6313         }
6314
6315         //
6316         // Store off the information
6317         //
6318
6319         FileInfo->FileAttributes = pDirectoryEntry->ObjectInformation->FileAttributes;
6320
6321         //
6322         // Check for the mount point being returned
6323         //
6324
6325         if( pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_MOUNTPOINT)
6326         {
6327
6328             FileInfo->FileAttributes = (FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_REPARSE_POINT);
6329         }
6330         else if( pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_SYMLINK ||
6331                  pDirectoryEntry->ObjectInformation->FileType == AFS_FILE_TYPE_DFSLINK)
6332         {
6333
6334             if ( FileInfo->FileAttributes == FILE_ATTRIBUTE_NORMAL)
6335             {
6336
6337                 FileInfo->FileAttributes = FILE_ATTRIBUTE_REPARSE_POINT;
6338             }
6339             else
6340             {
6341
6342                 FileInfo->FileAttributes |= FILE_ATTRIBUTE_REPARSE_POINT;
6343             }
6344         }
6345
6346         FileInfo->AllocationSize = pDirectoryEntry->ObjectInformation->AllocationSize;
6347
6348         FileInfo->EndOfFile = pDirectoryEntry->ObjectInformation->EndOfFile;
6349
6350         FileInfo->CreationTime = pDirectoryEntry->ObjectInformation->CreationTime;
6351
6352         FileInfo->LastAccessTime = pDirectoryEntry->ObjectInformation->LastAccessTime;
6353
6354         FileInfo->LastWriteTime = pDirectoryEntry->ObjectInformation->LastWriteTime;
6355
6356         FileInfo->ChangeTime = pDirectoryEntry->ObjectInformation->ChangeTime;
6357
6358         //
6359         // Remove the reference made above
6360         //
6361
6362         lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6363
6364         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6365                       AFS_TRACE_LEVEL_VERBOSE,
6366                       "AFSRetrieveFileAttributes Decrement3 count on %wZ DE %p Ccb %p Cnt %d\n",
6367                       &pDirectoryEntry->NameInformation.FileName,
6368                       pDirectoryEntry,
6369                       NULL,
6370                       lCount);
6371
6372 try_exit:
6373
6374         if( pDirEntry != NULL)
6375         {
6376
6377             AFSExFreePoolWithTag( pDirEntry, AFS_GENERIC_MEMORY_2_TAG);
6378         }
6379
6380         if( pVolumeCB != NULL)
6381         {
6382
6383             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6384
6385             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6386                           AFS_TRACE_LEVEL_VERBOSE,
6387                           "AFSRetrieveFileAttributes Decrement2 count on volume %08lX Cnt %d\n",
6388                           pVolumeCB,
6389                           lCount);
6390         }
6391
6392         if( pNameArray != NULL)
6393         {
6394
6395             AFSFreeNameArray( pNameArray);
6396         }
6397
6398         if( pwchBuffer != NULL)
6399         {
6400
6401             //
6402             // Always free the buffer that we allocated as AFSLocateNameEntry
6403             // will not free it.  If uniFullPathName.Buffer was allocated by
6404             // AFSLocateNameEntry, then we must free that as well.
6405             // Check that the uniFullPathName.Buffer in the string is not the same
6406             // offset by the length of the server name
6407             //
6408
6409             if( uniFullPathName.Length > 0 &&
6410                 pwchBuffer != (WCHAR *)((char *)uniFullPathName.Buffer - ulNameDifference))
6411             {
6412
6413                 AFSExFreePoolWithTag( uniFullPathName.Buffer, 0);
6414             }
6415
6416             AFSExFreePoolWithTag( pwchBuffer, 0);
6417         }
6418     }
6419
6420     return ntStatus;
6421 }
6422
6423 AFSObjectInfoCB *
6424 AFSAllocateObjectInfo( IN AFSObjectInfoCB *ParentObjectInfo,
6425                        IN ULONGLONG HashIndex)
6426 {
6427
6428     NTSTATUS ntStatus = STATUS_SUCCESS;
6429     AFSObjectInfoCB *pObjectInfo = NULL;
6430     LONG lCount;
6431
6432     __Enter
6433     {
6434
6435         pObjectInfo = (AFSObjectInfoCB *)AFSExAllocatePoolWithTag( PagedPool,
6436                                                                    sizeof( AFSObjectInfoCB),
6437                                                                    AFS_OBJECT_INFO_TAG);
6438
6439         if( pObjectInfo == NULL)
6440         {
6441
6442             try_return( pObjectInfo);
6443         }
6444
6445         RtlZeroMemory( pObjectInfo,
6446                        sizeof( AFSObjectInfoCB));
6447
6448         pObjectInfo->NonPagedInfo = (AFSNonPagedObjectInfoCB *)AFSExAllocatePoolWithTag( NonPagedPool,
6449                                                                                          sizeof( AFSNonPagedObjectInfoCB),
6450                                                                                          AFS_NP_OBJECT_INFO_TAG);
6451
6452         if( pObjectInfo->NonPagedInfo == NULL)
6453         {
6454
6455             AFSExFreePoolWithTag( pObjectInfo, AFS_OBJECT_INFO_TAG);
6456
6457             try_return( pObjectInfo = NULL);
6458         }
6459
6460         ExInitializeResourceLite( &pObjectInfo->NonPagedInfo->DirectoryNodeHdrLock);
6461
6462         pObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock = &pObjectInfo->NonPagedInfo->DirectoryNodeHdrLock;
6463
6464         pObjectInfo->VolumeCB = ParentObjectInfo->VolumeCB;
6465
6466         pObjectInfo->ParentObjectInformation = ParentObjectInfo;
6467
6468         if( ParentObjectInfo != NULL)
6469         {
6470             lCount = InterlockedIncrement( &ParentObjectInfo->ObjectReferenceCount);
6471         }
6472
6473         //
6474         // Initialize the access time
6475         //
6476
6477         KeQueryTickCount( &pObjectInfo->LastAccessCount);
6478
6479         if( HashIndex != 0)
6480         {
6481
6482             //
6483             // Insert the entry into the object tree and list
6484             //
6485
6486             pObjectInfo->TreeEntry.HashIndex = HashIndex;
6487
6488             if( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead == NULL)
6489             {
6490
6491                 ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead = &pObjectInfo->TreeEntry;
6492             }
6493             else
6494             {
6495
6496                 ntStatus = AFSInsertHashEntry( ParentObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
6497                                                &pObjectInfo->TreeEntry);
6498
6499                 ASSERT( NT_SUCCESS( ntStatus));
6500             }
6501
6502             //
6503             // And the object list in the volume
6504             //
6505
6506             if( ParentObjectInfo->VolumeCB->ObjectInfoListHead == NULL)
6507             {
6508
6509                 ParentObjectInfo->VolumeCB->ObjectInfoListHead = pObjectInfo;
6510             }
6511             else
6512             {
6513
6514                 ParentObjectInfo->VolumeCB->ObjectInfoListTail->ListEntry.fLink = (void *)pObjectInfo;
6515
6516                 pObjectInfo->ListEntry.bLink = (void *)ParentObjectInfo->VolumeCB->ObjectInfoListTail;
6517             }
6518
6519             ParentObjectInfo->VolumeCB->ObjectInfoListTail = pObjectInfo;
6520
6521             //
6522             // Indicate the object is in the hash tree and linked list in the volume
6523             //
6524
6525             SetFlag( pObjectInfo->Flags, AFS_OBJECT_INSERTED_HASH_TREE | AFS_OBJECT_INSERTED_VOLUME_LIST);
6526         }
6527
6528 try_exit:
6529
6530         NOTHING;
6531     }
6532
6533     return pObjectInfo;
6534 }
6535
6536 void
6537 AFSDeleteObjectInfo( IN AFSObjectInfoCB *ObjectInfo)
6538 {
6539
6540     BOOLEAN bAcquiredTreeLock = FALSE;
6541     LONG lCount;
6542
6543     if( !ExIsResourceAcquiredExclusiveLite( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock))
6544     {
6545
6546         ASSERT( !ExIsResourceAcquiredLite( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock));
6547
6548         AFSAcquireExcl( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock,
6549                         TRUE);
6550
6551         bAcquiredTreeLock = TRUE;
6552     }
6553
6554     //
6555     // Remove it from the tree and list if it was inserted
6556     //
6557
6558     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_INSERTED_HASH_TREE))
6559     {
6560
6561         AFSRemoveHashEntry( &ObjectInfo->VolumeCB->ObjectInfoTree.TreeHead,
6562                             &ObjectInfo->TreeEntry);
6563     }
6564
6565     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_INSERTED_VOLUME_LIST))
6566     {
6567
6568         if( ObjectInfo->ListEntry.fLink == NULL)
6569         {
6570
6571             ObjectInfo->VolumeCB->ObjectInfoListTail = (AFSObjectInfoCB *)ObjectInfo->ListEntry.bLink;
6572
6573             if( ObjectInfo->VolumeCB->ObjectInfoListTail != NULL)
6574             {
6575
6576                 ObjectInfo->VolumeCB->ObjectInfoListTail->ListEntry.fLink = NULL;
6577             }
6578         }
6579         else
6580         {
6581
6582             ((AFSObjectInfoCB *)(ObjectInfo->ListEntry.fLink))->ListEntry.bLink = ObjectInfo->ListEntry.bLink;
6583         }
6584
6585         if( ObjectInfo->ListEntry.bLink == NULL)
6586         {
6587
6588             ObjectInfo->VolumeCB->ObjectInfoListHead = (AFSObjectInfoCB *)ObjectInfo->ListEntry.fLink;
6589
6590             if( ObjectInfo->VolumeCB->ObjectInfoListHead != NULL)
6591             {
6592
6593                 ObjectInfo->VolumeCB->ObjectInfoListHead->ListEntry.bLink = NULL;
6594             }
6595         }
6596         else
6597         {
6598
6599             ((AFSObjectInfoCB *)(ObjectInfo->ListEntry.bLink))->ListEntry.fLink = ObjectInfo->ListEntry.fLink;
6600         }
6601     }
6602
6603     if( ObjectInfo->ParentObjectInformation != NULL)
6604     {
6605
6606         lCount = InterlockedDecrement( &ObjectInfo->ParentObjectInformation->ObjectReferenceCount);
6607     }
6608
6609     if( bAcquiredTreeLock)
6610     {
6611
6612         AFSReleaseResource( ObjectInfo->VolumeCB->ObjectInfoTree.TreeLock);
6613     }
6614
6615     //
6616     // Release the fid in the service
6617     //
6618
6619     if( BooleanFlagOn( ObjectInfo->Flags, AFS_OBJECT_HELD_IN_SERVICE))
6620     {
6621
6622         AFSReleaseFid( &ObjectInfo->FileId);
6623     }
6624
6625     ExDeleteResourceLite( &ObjectInfo->NonPagedInfo->DirectoryNodeHdrLock);
6626
6627     AFSExFreePoolWithTag( ObjectInfo->NonPagedInfo, AFS_NP_OBJECT_INFO_TAG);
6628
6629     AFSExFreePoolWithTag( ObjectInfo, AFS_OBJECT_INFO_TAG);
6630
6631     return;
6632 }
6633
6634 NTSTATUS
6635 AFSEvaluateRootEntry( IN AFSDirectoryCB *DirectoryCB,
6636                       OUT AFSDirectoryCB **TargetDirEntry)
6637 {
6638
6639     NTSTATUS ntStatus = STATUS_SUCCESS;
6640     AFSDirEnumEntry *pDirEntry = NULL, *pLastDirEntry = NULL;
6641     UNICODE_STRING uniFullPathName;
6642     AFSNameArrayHdr    *pNameArray = NULL;
6643     AFSVolumeCB *pVolumeCB = NULL;
6644     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
6645     WCHAR *pwchBuffer = NULL;
6646     UNICODE_STRING uniComponentName, uniRemainingPath, uniParsedName;
6647     ULONG ulNameDifference = 0;
6648     GUID    stAuthGroup;
6649     LONG lCount;
6650
6651     __Enter
6652     {
6653
6654         ntStatus = AFSRetrieveValidAuthGroup( NULL,
6655                                               DirectoryCB->ObjectInformation,
6656                                               FALSE,
6657                                               &stAuthGroup);
6658
6659         if( !NT_SUCCESS( ntStatus))
6660         {
6661             try_return( ntStatus);
6662         }
6663
6664         //
6665         // Retrieve a target name for the entry
6666         //
6667
6668         AFSAcquireShared( &DirectoryCB->NonPaged->Lock,
6669                           TRUE);
6670
6671         if( DirectoryCB->NameInformation.TargetName.Length == 0)
6672         {
6673
6674             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6675
6676             ntStatus = AFSEvaluateTargetByID( DirectoryCB->ObjectInformation,
6677                                               &stAuthGroup,
6678                                               FALSE,
6679                                               &pDirEntry);
6680
6681             if( !NT_SUCCESS( ntStatus) ||
6682                 pDirEntry->TargetNameLength == 0)
6683             {
6684
6685                 if( pDirEntry != NULL)
6686                 {
6687
6688                     ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
6689                 }
6690
6691                 try_return( ntStatus);
6692             }
6693
6694             AFSAcquireExcl( &DirectoryCB->NonPaged->Lock,
6695                             TRUE);
6696
6697             if( DirectoryCB->NameInformation.TargetName.Length == 0)
6698             {
6699
6700                 //
6701                 // Update the target name
6702                 //
6703
6704                 ntStatus = AFSUpdateTargetName( &DirectoryCB->NameInformation.TargetName,
6705                                                 &DirectoryCB->Flags,
6706                                                 (WCHAR *)((char *)pDirEntry + pDirEntry->TargetNameOffset),
6707                                                 (USHORT)pDirEntry->TargetNameLength);
6708
6709                 if( !NT_SUCCESS( ntStatus))
6710                 {
6711
6712                     AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6713
6714                     try_return( ntStatus);
6715                 }
6716             }
6717
6718             AFSConvertToShared( &DirectoryCB->NonPaged->Lock);
6719         }
6720
6721         //
6722         // Need to pass the full path in for parsing.
6723         //
6724
6725         uniFullPathName.Length = 0;
6726         uniFullPathName.MaximumLength = DirectoryCB->NameInformation.TargetName.Length;
6727
6728         uniFullPathName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
6729                                                                     uniFullPathName.MaximumLength,
6730                                                                     AFS_NAME_BUFFER_EIGHT_TAG);
6731
6732         if( uniFullPathName.Buffer == NULL)
6733         {
6734
6735             AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6736
6737             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6738         }
6739
6740         pwchBuffer = uniFullPathName.Buffer;
6741
6742         RtlZeroMemory( uniFullPathName.Buffer,
6743                        uniFullPathName.MaximumLength);
6744
6745         RtlCopyMemory( uniFullPathName.Buffer,
6746                        DirectoryCB->NameInformation.TargetName.Buffer,
6747                        DirectoryCB->NameInformation.TargetName.Length);
6748
6749         uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6750
6751         //
6752         // This name should begin with the \afs server so parse it off and chech it
6753         //
6754
6755         FsRtlDissectName( uniFullPathName,
6756                           &uniComponentName,
6757                           &uniRemainingPath);
6758
6759         if( RtlCompareUnicodeString( &uniComponentName,
6760                                      &AFSServerName,
6761                                      TRUE) != 0)
6762         {
6763
6764             //
6765             // Try evaluating the full path
6766             //
6767
6768             uniFullPathName.Buffer = pwchBuffer;
6769
6770             uniFullPathName.Length = DirectoryCB->NameInformation.TargetName.Length;
6771
6772             uniFullPathName.MaximumLength = uniFullPathName.Length;
6773         }
6774         else
6775         {
6776
6777             uniFullPathName = uniRemainingPath;
6778         }
6779
6780         uniParsedName = uniFullPathName;
6781
6782         ulNameDifference = (ULONG)(uniFullPathName.Length > 0 ? ((char *)uniFullPathName.Buffer - (char *)pwchBuffer) : 0);
6783
6784         AFSReleaseResource( &DirectoryCB->NonPaged->Lock);
6785
6786         //
6787         // Our name array
6788         //
6789
6790         pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
6791                                        0);
6792
6793         if( pNameArray == NULL)
6794         {
6795
6796             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
6797         }
6798
6799         pVolumeCB = AFSGlobalRoot;
6800
6801         pParentDirEntry = AFSGlobalRoot->DirectoryCB;
6802
6803         lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
6804
6805         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6806                       AFS_TRACE_LEVEL_VERBOSE,
6807                       "AFSEvaluateRootEntry Increment count on volume %08lX Cnt %d\n",
6808                       pVolumeCB,
6809                       lCount);
6810
6811         lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
6812
6813         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6814                       AFS_TRACE_LEVEL_VERBOSE,
6815                       "AFSEvaluateRootEntry Increment count on %wZ DE %p Ccb %p Cnt %d\n",
6816                       &pParentDirEntry->NameInformation.FileName,
6817                       pParentDirEntry,
6818                       NULL,
6819                       lCount);
6820
6821         ntStatus = AFSLocateNameEntry( NULL,
6822                                        NULL,
6823                                        &uniFullPathName,
6824                                        &uniParsedName,
6825                                        pNameArray,
6826                                        0,
6827                                        &pVolumeCB,
6828                                        &pParentDirEntry,
6829                                        &pDirectoryEntry,
6830                                        NULL);
6831
6832         if( !NT_SUCCESS( ntStatus))
6833         {
6834
6835             //
6836             // The volume lock was released on failure above
6837             // Except for STATUS_OBJECT_NAME_NOT_FOUND
6838             //
6839
6840             if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
6841             {
6842
6843                 if( pVolumeCB != NULL)
6844                 {
6845
6846                     lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6847
6848                     AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6849                                   AFS_TRACE_LEVEL_VERBOSE,
6850                                   "AFSEvaluateRootEntry Decrement count on volume %08lX Cnt %d\n",
6851                                   pVolumeCB,
6852                                   lCount);
6853                 }
6854
6855                 if( pDirectoryEntry != NULL)
6856                 {
6857
6858                     lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
6859
6860                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6861                                   AFS_TRACE_LEVEL_VERBOSE,
6862                                   "AFSEvaluateRootEntry Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6863                                   &pDirectoryEntry->NameInformation.FileName,
6864                                   pDirectoryEntry,
6865                                   NULL,
6866                                   lCount);
6867                 }
6868                 else
6869                 {
6870
6871                     lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
6872
6873                     AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
6874                                   AFS_TRACE_LEVEL_VERBOSE,
6875                                   "AFSEvaluateRootEntry Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
6876                                   &pParentDirEntry->NameInformation.FileName,
6877                                   pParentDirEntry,
6878                                   NULL,
6879                                   lCount);
6880                 }
6881             }
6882
6883             pVolumeCB = NULL;
6884
6885             try_return( ntStatus);
6886         }
6887
6888         //
6889         // Pass back the target dir entry for this request
6890         //
6891
6892         *TargetDirEntry = pDirectoryEntry;
6893
6894 try_exit:
6895
6896         if( pDirEntry != NULL)
6897         {
6898
6899             AFSExFreePoolWithTag( pDirEntry, AFS_GENERIC_MEMORY_2_TAG);
6900         }
6901
6902         if( pVolumeCB != NULL)
6903         {
6904
6905             lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
6906
6907             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
6908                           AFS_TRACE_LEVEL_VERBOSE,
6909                           "AFSEvaluateRootEntry2 Decrement count on volume %08lX Cnt %d\n",
6910                           pVolumeCB,
6911                           lCount);
6912         }
6913
6914         if( pNameArray != NULL)
6915         {
6916
6917             AFSFreeNameArray( pNameArray);
6918         }
6919
6920         if( pwchBuffer != NULL)
6921         {
6922
6923             //
6924             // Always free the buffer that we allocated as AFSLocateNameEntry
6925             // will not free it.  If uniFullPathName.Buffer was allocated by
6926             // AFSLocateNameEntry, then we must free that as well.
6927             // Check that the uniFullPathName.Buffer in the string is not the same
6928             // offset by the length of the server name
6929             //
6930
6931             if( uniFullPathName.Length > 0 &&
6932                 pwchBuffer != (WCHAR *)((char *)uniFullPathName.Buffer - ulNameDifference))
6933             {
6934
6935                 AFSExFreePoolWithTag( uniFullPathName.Buffer, 0);
6936             }
6937
6938             AFSExFreePoolWithTag( pwchBuffer, 0);
6939         }
6940     }
6941
6942     return ntStatus;
6943 }
6944
6945 NTSTATUS
6946 AFSCleanupFcb( IN AFSFcb *Fcb,
6947                IN BOOLEAN ForceFlush)
6948 {
6949
6950     NTSTATUS ntStatus = STATUS_SUCCESS;
6951     AFSDeviceExt *pRDRDeviceExt = NULL, *pControlDeviceExt = NULL;
6952     LARGE_INTEGER liTime;
6953     IO_STATUS_BLOCK stIoStatus;
6954
6955     __Enter
6956     {
6957
6958         pControlDeviceExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
6959
6960         pRDRDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
6961
6962         if( BooleanFlagOn( pRDRDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_REDIRECTOR_SHUTDOWN))
6963         {
6964
6965             if( !BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) &&
6966                 !BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
6967             {
6968
6969                 AFSAcquireExcl( &Fcb->NPFcb->Resource,
6970                                 TRUE);
6971
6972                 if( Fcb->OpenReferenceCount > 0)
6973                 {
6974
6975                     __try
6976                     {
6977
6978                         CcFlushCache( &Fcb->NPFcb->SectionObjectPointers,
6979                                       NULL,
6980                                       0,
6981                                       &stIoStatus);
6982
6983                         if( !NT_SUCCESS( stIoStatus.Status))
6984                         {
6985
6986                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
6987                                           AFS_TRACE_LEVEL_ERROR,
6988                                           "AFSCleanupFcb CcFlushCache [1] failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
6989                                           Fcb->ObjectInformation->FileId.Cell,
6990                                           Fcb->ObjectInformation->FileId.Volume,
6991                                           Fcb->ObjectInformation->FileId.Vnode,
6992                                           Fcb->ObjectInformation->FileId.Unique,
6993                                           stIoStatus.Status,
6994                                           stIoStatus.Information);
6995
6996                             ntStatus = stIoStatus.Status;
6997                         }
6998
6999                         if ( !CcPurgeCacheSection( &Fcb->NPFcb->SectionObjectPointers,
7000                                                    NULL,
7001                                                    0,
7002                                                    FALSE))
7003                         {
7004
7005                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
7006                                           AFS_TRACE_LEVEL_WARNING,
7007                                           "AFSCleanupFcb CcPurgeCacheSection [1] failure FID %08lX-%08lX-%08lX-%08lX\n",
7008                                           Fcb->ObjectInformation->FileId.Cell,
7009                                           Fcb->ObjectInformation->FileId.Volume,
7010                                           Fcb->ObjectInformation->FileId.Vnode,
7011                                           Fcb->ObjectInformation->FileId.Unique);
7012
7013                             SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7014                         }
7015                     }
7016                     __except( EXCEPTION_EXECUTE_HANDLER)
7017                     {
7018
7019                         ntStatus = GetExceptionCode();
7020
7021                         AFSDbgLogMsg( 0,
7022                                       0,
7023                                       "EXCEPTION - AFSCleanupFcb Cc [1] FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
7024                                       Fcb->ObjectInformation->FileId.Cell,
7025                                       Fcb->ObjectInformation->FileId.Volume,
7026                                       Fcb->ObjectInformation->FileId.Vnode,
7027                                       Fcb->ObjectInformation->FileId.Unique,
7028                                       ntStatus);
7029
7030                         SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7031                     }
7032                 }
7033
7034                 AFSReleaseResource( &Fcb->NPFcb->Resource);
7035
7036                 //
7037                 // Wait for any currently running flush or release requests to complete
7038                 //
7039
7040                 AFSWaitOnQueuedFlushes( Fcb);
7041
7042                 //
7043                 // Now perform another flush on the file
7044                 //
7045
7046                 if( !NT_SUCCESS( AFSFlushExtents( Fcb,
7047                                                   NULL)))
7048                 {
7049
7050                     AFSReleaseExtentsWithFlush( Fcb,
7051                                                 NULL,
7052                                                 TRUE);
7053                 }
7054             }
7055
7056             if( Fcb->OpenReferenceCount == 0 ||
7057                 BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) ||
7058                 BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
7059             {
7060
7061                 AFSTearDownFcbExtents( Fcb,
7062                                        NULL);
7063             }
7064
7065             try_return( ntStatus);
7066         }
7067
7068         KeQueryTickCount( &liTime);
7069
7070         //
7071         // First up are there dirty extents in the cache to flush?
7072         //
7073
7074         if( BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID) ||
7075             BooleanFlagOn( Fcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
7076         {
7077
7078             //
7079             // The file has been marked as invalid.  Dump it
7080             //
7081
7082             AFSTearDownFcbExtents( Fcb,
7083                                    NULL);
7084         }
7085         else if( ForceFlush ||
7086             ( ( Fcb->Specific.File.ExtentsDirtyCount ||
7087                 Fcb->Specific.File.ExtentCount) &&
7088               (liTime.QuadPart - Fcb->Specific.File.LastServerFlush.QuadPart)
7089                                                     >= pControlDeviceExt->Specific.Control.FcbFlushTimeCount.QuadPart))
7090         {
7091             if( !NT_SUCCESS( AFSFlushExtents( Fcb,
7092                                               NULL)) &&
7093                 Fcb->OpenReferenceCount == 0)
7094             {
7095
7096                 AFSReleaseExtentsWithFlush( Fcb,
7097                                             NULL,
7098                                             TRUE);
7099             }
7100         }
7101
7102         //
7103         // If there are extents and they haven't been used recently *and*
7104         // are not being used
7105         //
7106
7107         if( ( ForceFlush ||
7108               ( 0 != Fcb->Specific.File.ExtentCount &&
7109                 0 != Fcb->Specific.File.LastExtentAccess.QuadPart &&
7110                 (liTime.QuadPart - Fcb->Specific.File.LastExtentAccess.QuadPart) >=
7111                                         (AFS_SERVER_PURGE_SLEEP * pControlDeviceExt->Specific.Control.FcbPurgeTimeCount.QuadPart))) &&
7112             AFSAcquireExcl( &Fcb->NPFcb->Resource,
7113                             ForceFlush))
7114         {
7115
7116             __try
7117             {
7118
7119                 CcFlushCache( &Fcb->NPFcb->SectionObjectPointers,
7120                               NULL,
7121                               0,
7122                               &stIoStatus);
7123
7124                 if( !NT_SUCCESS( stIoStatus.Status))
7125                 {
7126
7127                     AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
7128                                   AFS_TRACE_LEVEL_ERROR,
7129                                   "AFSCleanupFcb CcFlushCache [2] failure FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
7130                                   Fcb->ObjectInformation->FileId.Cell,
7131                                   Fcb->ObjectInformation->FileId.Volume,
7132                                   Fcb->ObjectInformation->FileId.Vnode,
7133                                   Fcb->ObjectInformation->FileId.Unique,
7134                                   stIoStatus.Status,
7135                                   stIoStatus.Information);
7136
7137                     ntStatus = stIoStatus.Status;
7138                 }
7139
7140                 if( ForceFlush)
7141                 {
7142
7143                     if ( !CcPurgeCacheSection( &Fcb->NPFcb->SectionObjectPointers,
7144                                                NULL,
7145                                                0,
7146                                                FALSE))
7147                     {
7148
7149                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
7150                                       AFS_TRACE_LEVEL_WARNING,
7151                                       "AFSCleanupFcb CcPurgeCacheSection [2] failure FID %08lX-%08lX-%08lX-%08lX\n",
7152                                       Fcb->ObjectInformation->FileId.Cell,
7153                                       Fcb->ObjectInformation->FileId.Volume,
7154                                       Fcb->ObjectInformation->FileId.Vnode,
7155                                       Fcb->ObjectInformation->FileId.Unique);
7156
7157                         SetFlag( Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
7158                     }
7159                 }
7160             }
7161             __except( EXCEPTION_EXECUTE_HANDLER)
7162             {
7163
7164                 ntStatus = GetExceptionCode();
7165
7166                 AFSDbgLogMsg( 0,
7167                               0,
7168                               "EXCEPTION - AFSCleanupFcb Cc [2] FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
7169                               Fcb->ObjectInformation->FileId.Cell,
7170                               Fcb->ObjectInformation->FileId.Volume,
7171                               Fcb->ObjectInformation->FileId.Vnode,
7172                               Fcb->ObjectInformation->FileId.Unique,
7173                               ntStatus);
7174             }
7175
7176             AFSReleaseResource( &Fcb->NPFcb->Resource);
7177
7178             if( Fcb->OpenReferenceCount <= 0)
7179             {
7180
7181                 //
7182                 // Tear em down we'll not be needing them again
7183                 //
7184
7185                 AFSTearDownFcbExtents( Fcb,
7186                                        NULL);
7187             }
7188         }
7189
7190 try_exit:
7191
7192         NOTHING;
7193     }
7194
7195     return ntStatus;
7196 }
7197
7198 NTSTATUS
7199 AFSUpdateDirEntryName( IN AFSDirectoryCB *DirectoryCB,
7200                        IN UNICODE_STRING *NewFileName)
7201 {
7202
7203     NTSTATUS ntStatus = STATUS_SUCCESS;
7204     WCHAR *pTmpBuffer = NULL;
7205
7206     __Enter
7207     {
7208
7209         if( NewFileName->Length > DirectoryCB->NameInformation.FileName.Length)
7210         {
7211
7212             if( BooleanFlagOn( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER))
7213             {
7214
7215                 AFSExFreePoolWithTag( DirectoryCB->NameInformation.FileName.Buffer, 0);
7216
7217                 ClearFlag( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER);
7218
7219                 DirectoryCB->NameInformation.FileName.Buffer = NULL;
7220             }
7221
7222             //
7223             // OK, we need to allocate a new name buffer
7224             //
7225
7226             pTmpBuffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
7227                                                             NewFileName->Length,
7228                                                             AFS_NAME_BUFFER_NINE_TAG);
7229
7230             if( pTmpBuffer == NULL)
7231             {
7232
7233                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
7234             }
7235
7236             DirectoryCB->NameInformation.FileName.Buffer = pTmpBuffer;
7237
7238             DirectoryCB->NameInformation.FileName.MaximumLength = NewFileName->Length;
7239
7240             SetFlag( DirectoryCB->Flags, AFS_DIR_RELEASE_NAME_BUFFER);
7241         }
7242
7243         DirectoryCB->NameInformation.FileName.Length = NewFileName->Length;
7244
7245         RtlCopyMemory( DirectoryCB->NameInformation.FileName.Buffer,
7246                        NewFileName->Buffer,
7247                        NewFileName->Length);
7248
7249 try_exit:
7250
7251         NOTHING;
7252     }
7253
7254     return ntStatus;
7255 }
7256
7257 NTSTATUS
7258 AFSReadCacheFile( IN void *ReadBuffer,
7259                   IN LARGE_INTEGER *ReadOffset,
7260                   IN ULONG RequestedDataLength,
7261                   IN OUT PULONG BytesRead)
7262 {
7263
7264     NTSTATUS            ntStatus = STATUS_SUCCESS;
7265     PIRP                pIrp = NULL;
7266     KEVENT              kEvent;
7267     PIO_STACK_LOCATION  pIoStackLocation = NULL;
7268     AFSDeviceExt       *pRdrDevExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
7269     DEVICE_OBJECT      *pTargetDeviceObject = NULL;
7270     FILE_OBJECT        *pCacheFileObject = NULL;
7271
7272     __Enter
7273     {
7274
7275         pCacheFileObject = AFSReferenceCacheFileObject();
7276
7277         if( pCacheFileObject == NULL)
7278         {
7279             try_return( ntStatus = STATUS_DEVICE_NOT_READY);
7280         }
7281
7282         pTargetDeviceObject = IoGetRelatedDeviceObject( pCacheFileObject);
7283
7284         //
7285         // Initialize the event
7286         //
7287
7288         KeInitializeEvent( &kEvent,
7289                            SynchronizationEvent,
7290                            FALSE);
7291
7292         //
7293         // Allocate an irp for this request.  This could also come from a
7294         // private pool, for instance.
7295         //
7296
7297         pIrp = IoAllocateIrp( pTargetDeviceObject->StackSize,
7298                               FALSE);
7299
7300         if( pIrp == NULL)
7301         {
7302
7303             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
7304         }
7305
7306         //
7307         // Build the IRP's main body
7308         //
7309
7310         pIrp->UserBuffer = ReadBuffer;
7311
7312         pIrp->Tail.Overlay.Thread = PsGetCurrentThread();
7313         pIrp->RequestorMode = KernelMode;
7314         pIrp->Flags |= IRP_READ_OPERATION;
7315
7316         //
7317         // Set up the I/O stack location.
7318         //
7319
7320         pIoStackLocation = IoGetNextIrpStackLocation( pIrp);
7321         pIoStackLocation->MajorFunction = IRP_MJ_READ;
7322         pIoStackLocation->DeviceObject = pTargetDeviceObject;
7323         pIoStackLocation->FileObject = pCacheFileObject;
7324         pIoStackLocation->Parameters.Read.Length = RequestedDataLength;
7325
7326         pIoStackLocation->Parameters.Read.ByteOffset.QuadPart = ReadOffset->QuadPart;
7327
7328         //
7329         // Set the completion routine.
7330         //
7331
7332         IoSetCompletionRoutine( pIrp,
7333                                 AFSIrpComplete,
7334                                 &kEvent,
7335                                 TRUE,
7336                                 TRUE,
7337                                 TRUE);
7338
7339         //
7340         // Send it to the FSD
7341         //
7342
7343         ntStatus = IoCallDriver( pTargetDeviceObject,
7344                                  pIrp);
7345
7346         if( NT_SUCCESS( ntStatus))
7347         {
7348
7349             //
7350             // Wait for the I/O
7351             //
7352
7353             ntStatus = KeWaitForSingleObject( &kEvent,
7354                                               Executive,
7355                                               KernelMode,
7356                                               FALSE,
7357                                               0);
7358
7359             if( NT_SUCCESS( ntStatus))
7360             {
7361
7362                 ntStatus = pIrp->IoStatus.Status;
7363
7364                 *BytesRead = (ULONG)pIrp->IoStatus.Information;
7365             }
7366         }
7367
7368 try_exit:
7369
7370         if( pCacheFileObject != NULL)
7371         {
7372             AFSReleaseCacheFileObject( pCacheFileObject);
7373         }
7374
7375         if( pIrp != NULL)
7376         {
7377
7378             if( pIrp->MdlAddress != NULL)
7379             {
7380
7381                 if( FlagOn( pIrp->MdlAddress->MdlFlags, MDL_PAGES_LOCKED))
7382                 {
7383
7384                     MmUnlockPages( pIrp->MdlAddress);
7385                 }
7386
7387                 IoFreeMdl( pIrp->MdlAddress);
7388             }
7389
7390             pIrp->MdlAddress = NULL;
7391
7392             //
7393             // Free the Irp
7394             //
7395
7396             IoFreeIrp( pIrp);
7397         }
7398     }
7399
7400     return ntStatus;
7401 }
7402
7403 NTSTATUS
7404 AFSIrpComplete( IN PDEVICE_OBJECT DeviceObject,
7405                 IN PIRP           Irp,
7406                 IN PVOID          Context)
7407 {
7408
7409     KEVENT *pEvent = (KEVENT *)Context;
7410
7411     KeSetEvent( pEvent,
7412                 0,
7413                 FALSE);
7414
7415     return STATUS_MORE_PROCESSING_REQUIRED;
7416 }
7417
7418 BOOLEAN
7419 AFSIsDirectoryEmptyForDelete( IN AFSFcb *Fcb)
7420 {
7421
7422     BOOLEAN bIsEmpty = FALSE;
7423     AFSDirectoryCB *pDirEntry = NULL;
7424
7425     __Enter
7426     {
7427
7428         AFSAcquireShared( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
7429                           TRUE);
7430
7431         bIsEmpty = TRUE;
7432
7433         if( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeListHead != NULL)
7434         {
7435
7436             pDirEntry = Fcb->ObjectInformation->Specific.Directory.DirectoryNodeListHead;
7437
7438             while( pDirEntry != NULL)
7439             {
7440
7441                 if( !BooleanFlagOn( pDirEntry->Flags, AFS_DIR_ENTRY_FAKE) &&
7442                     !BooleanFlagOn( pDirEntry->Flags, AFS_DIR_ENTRY_DELETED))
7443                 {
7444
7445                     bIsEmpty = FALSE;
7446
7447                     break;
7448                 }
7449
7450                 pDirEntry = (AFSDirectoryCB *)pDirEntry->ListEntry.fLink;
7451             }
7452
7453         }
7454
7455         AFSReleaseResource( Fcb->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
7456     }
7457
7458     return bIsEmpty;
7459 }
7460
7461 void
7462 AFSRemoveNameEntry( IN AFSObjectInfoCB *ParentObjectInfo,
7463                     IN AFSDirectoryCB *DirEntry)
7464 {
7465
7466     NTSTATUS ntStatus = STATUS_SUCCESS;
7467
7468     __Enter
7469     {
7470
7471         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE))
7472         {
7473
7474             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7475                           AFS_TRACE_LEVEL_VERBOSE,
7476                           "AFSRemoveNameEntry DE %p for %wZ has NOT_IN flag set\n",
7477                           DirEntry,
7478                           &DirEntry->NameInformation.FileName);
7479
7480             try_return( ntStatus);
7481         }
7482
7483         ASSERT( ExIsResourceAcquiredExclusiveLite( ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.TreeLock));
7484
7485         //
7486         // Remove the entry from the parent tree
7487         //
7488
7489         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7490                       AFS_TRACE_LEVEL_VERBOSE,
7491                       "AFSRemoveNameEntry DE %p for %wZ removing from case sensitive tree\n",
7492                       DirEntry,
7493                       &DirEntry->NameInformation.FileName);
7494
7495         AFSRemoveCaseSensitiveDirEntry( &ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
7496                                         DirEntry);
7497
7498         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7499                       AFS_TRACE_LEVEL_VERBOSE,
7500                       "AFSRemoveNameEntry DE %p for %wZ removing from case insensitive tree\n",
7501                       DirEntry,
7502                       &DirEntry->NameInformation.FileName);
7503
7504         AFSRemoveCaseInsensitiveDirEntry( &ParentObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
7505                                           DirEntry);
7506
7507         if( BooleanFlagOn( DirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME))
7508         {
7509
7510             //
7511             // From the short name tree
7512             //
7513
7514             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7515                           AFS_TRACE_LEVEL_VERBOSE,
7516                           "AFSRemoveNameEntry DE %p for %wZ removing from shortname tree\n",
7517                           DirEntry,
7518                           &DirEntry->NameInformation.FileName);
7519
7520             AFSRemoveShortNameDirEntry( &ParentObjectInfo->Specific.Directory.ShortNameTree,
7521                                         DirEntry);
7522
7523             ClearFlag( DirEntry->Flags, AFS_DIR_ENTRY_INSERTED_SHORT_NAME);
7524         }
7525
7526         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
7527                       AFS_TRACE_LEVEL_VERBOSE,
7528                       "AFSRemoveNameEntry DE %p for %wZ setting NOT_IN flag\n",
7529                       DirEntry,
7530                       &DirEntry->NameInformation.FileName);
7531
7532         SetFlag( DirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE);
7533
7534         ClearFlag( DirEntry->Flags, AFS_DIR_ENTRY_CASE_INSENSTIVE_LIST_HEAD);
7535
7536 try_exit:
7537
7538         NOTHING;
7539     }
7540
7541     return;
7542 }
7543
7544 LARGE_INTEGER
7545 AFSGetAuthenticationId()
7546 {
7547
7548     LARGE_INTEGER liAuthId = {0,0};
7549     NTSTATUS ntStatus = STATUS_SUCCESS;
7550     PACCESS_TOKEN hToken = NULL;
7551     PTOKEN_STATISTICS pTokenInfo = NULL;
7552     BOOLEAN bCopyOnOpen = FALSE;
7553     BOOLEAN bEffectiveOnly = FALSE;
7554     BOOLEAN bPrimaryToken = FALSE;
7555     SECURITY_IMPERSONATION_LEVEL stImpersonationLevel;
7556
7557     __Enter
7558     {
7559
7560         hToken = PsReferenceImpersonationToken( PsGetCurrentThread(),
7561                                                 &bCopyOnOpen,
7562                                                 &bEffectiveOnly,
7563                                                 &stImpersonationLevel);
7564
7565         if( hToken == NULL)
7566         {
7567
7568             hToken = PsReferencePrimaryToken( PsGetCurrentProcess());
7569
7570             if( hToken == NULL)
7571             {
7572
7573                 AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7574                               AFS_TRACE_LEVEL_ERROR,
7575                               "AFSGetAuthenticationId Failed to retrieve impersonation or primary token\n");
7576
7577                 try_return( ntStatus);
7578             }
7579
7580             bPrimaryToken = TRUE;
7581         }
7582
7583         ntStatus = SeQueryInformationToken( hToken,
7584                                             TokenStatistics,
7585                                             (PVOID *)&pTokenInfo);
7586
7587         if( !NT_SUCCESS( ntStatus))
7588         {
7589
7590             AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7591                           AFS_TRACE_LEVEL_ERROR,
7592                           "AFSGetAuthenticationId Failed to retrieve information Status %08lX\n", ntStatus);
7593
7594             try_return( ntStatus);
7595         }
7596
7597         liAuthId.HighPart = pTokenInfo->AuthenticationId.HighPart;
7598         liAuthId.LowPart = pTokenInfo->AuthenticationId.LowPart;
7599
7600         AFSDbgLogMsg( AFS_SUBSYSTEM_NETWORK_PROVIDER,
7601                       AFS_TRACE_LEVEL_VERBOSE,
7602                       "AFSGetAuthenticationId Successfully retrieved authentication ID %I64X\n",
7603                       liAuthId.QuadPart);
7604
7605 try_exit:
7606
7607         if( hToken != NULL)
7608         {
7609
7610             if( !bPrimaryToken)
7611             {
7612
7613                 PsDereferenceImpersonationToken( hToken);
7614             }
7615             else
7616             {
7617
7618                 PsDereferencePrimaryToken( hToken);
7619             }
7620         }
7621
7622         if( pTokenInfo != NULL)
7623         {
7624
7625             ExFreePool( pTokenInfo);    // Allocated by SeQueryInformationToken
7626         }
7627     }
7628
7629     return liAuthId;
7630 }
7631
7632 void
7633 AFSUnwindFileInfo( IN AFSFcb *Fcb,
7634                    IN AFSCcb *Ccb)
7635 {
7636
7637     if( Ccb->FileUnwindInfo.FileAttributes != (ULONG)-1)
7638     {
7639         Ccb->DirectoryCB->ObjectInformation->FileAttributes = Ccb->FileUnwindInfo.FileAttributes;
7640     }
7641
7642     if( Ccb->FileUnwindInfo.CreationTime.QuadPart != (ULONGLONG)-1)
7643     {
7644         Ccb->DirectoryCB->ObjectInformation->CreationTime.QuadPart = Ccb->FileUnwindInfo.CreationTime.QuadPart;
7645     }
7646
7647     if( Ccb->FileUnwindInfo.LastAccessTime.QuadPart != (ULONGLONG)-1)
7648     {
7649         Ccb->DirectoryCB->ObjectInformation->LastAccessTime.QuadPart = Ccb->FileUnwindInfo.LastAccessTime.QuadPart;
7650     }
7651
7652     if( Ccb->FileUnwindInfo.LastWriteTime.QuadPart != (ULONGLONG)-1)
7653     {
7654         Ccb->DirectoryCB->ObjectInformation->LastWriteTime.QuadPart = Ccb->FileUnwindInfo.LastWriteTime.QuadPart;
7655     }
7656
7657     if( Ccb->FileUnwindInfo.ChangeTime.QuadPart != (ULONGLONG)-1)
7658     {
7659         Ccb->DirectoryCB->ObjectInformation->ChangeTime.QuadPart = Ccb->FileUnwindInfo.ChangeTime.QuadPart;
7660     }
7661
7662     return;
7663 }
7664
7665 BOOLEAN
7666 AFSValidateDirList( IN AFSObjectInfoCB *ObjectInfo)
7667 {
7668
7669     BOOLEAN bIsValid = TRUE;
7670     ULONG ulCount = 0;
7671     AFSDirectoryCB *pCurrentDirEntry = NULL, *pDirEntry = NULL;
7672
7673     pCurrentDirEntry = ObjectInfo->Specific.Directory.DirectoryNodeListHead;
7674
7675     while( pCurrentDirEntry != NULL)
7676     {
7677
7678         if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_FAKE))
7679         {
7680             ulCount++;
7681
7682             if( !BooleanFlagOn( pCurrentDirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE))
7683             {
7684
7685                 pDirEntry = NULL;
7686
7687                 AFSLocateCaseSensitiveDirEntry( ObjectInfo->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
7688                                                 (ULONG)pCurrentDirEntry->CaseSensitiveTreeEntry.HashIndex,
7689                                                 &pDirEntry);
7690
7691                 if( pDirEntry == NULL)
7692                 {
7693                     DbgBreakPoint();
7694                 }
7695             }
7696         }
7697
7698         pCurrentDirEntry = (AFSDirectoryCB *)pCurrentDirEntry->ListEntry.fLink;
7699     }
7700
7701     if( ulCount != ObjectInfo->Specific.Directory.DirectoryNodeCount)
7702     {
7703
7704         AFSPrint("AFSValidateDirList Count off Calc: %d Stored: %d\n",
7705                   ulCount,
7706                   ObjectInfo->Specific.Directory.DirectoryNodeCount);
7707
7708         ObjectInfo->Specific.Directory.DirectoryNodeCount = ulCount;
7709
7710         bIsValid = FALSE;
7711     }
7712
7713     return bIsValid;
7714 }
7715
7716 PFILE_OBJECT
7717 AFSReferenceCacheFileObject()
7718 {
7719
7720     AFSDeviceExt       *pRdrDevExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
7721     FILE_OBJECT        *pCacheFileObject = NULL;
7722
7723     AFSAcquireShared( &pRdrDevExt->Specific.RDR.CacheFileLock,
7724                       TRUE);
7725
7726     pCacheFileObject = pRdrDevExt->Specific.RDR.CacheFileObject;
7727
7728     if( pCacheFileObject != NULL)
7729     {
7730         ObReferenceObject( pCacheFileObject);
7731     }
7732
7733     AFSReleaseResource( &pRdrDevExt->Specific.RDR.CacheFileLock);
7734
7735     return pCacheFileObject;
7736 }
7737
7738 void
7739 AFSReleaseCacheFileObject( IN PFILE_OBJECT CacheFileObject)
7740 {
7741
7742     ASSERT( CacheFileObject != NULL);
7743
7744     ObDereferenceObject( CacheFileObject);
7745
7746     return;
7747 }
7748
7749 NTSTATUS
7750 AFSInitializeLibrary( IN AFSLibraryInitCB *LibraryInit)
7751 {
7752
7753     NTSTATUS ntStatus = STATUS_SUCCESS;
7754     AFSDeviceExt *pControlDevExt = NULL;
7755     ULONG ulTimeIncrement = 0;
7756     LONG lCount;
7757
7758     __Enter
7759     {
7760
7761         AFSControlDeviceObject = LibraryInit->AFSControlDeviceObject;
7762
7763         AFSRDRDeviceObject = LibraryInit->AFSRDRDeviceObject;
7764
7765         AFSServerName = LibraryInit->AFSServerName;
7766
7767         AFSDebugFlags = LibraryInit->AFSDebugFlags;
7768
7769         //
7770         // Callbacks in the framework
7771         //
7772
7773         AFSProcessRequest = LibraryInit->AFSProcessRequest;
7774
7775         AFSDbgLogMsg = LibraryInit->AFSDbgLogMsg;
7776
7777         AFSAddConnectionEx = LibraryInit->AFSAddConnectionEx;
7778
7779         AFSExAllocatePoolWithTag = LibraryInit->AFSExAllocatePoolWithTag;
7780
7781         AFSExFreePoolWithTag = LibraryInit->AFSExFreePoolWithTag;
7782
7783         AFSDumpTraceFilesFnc = LibraryInit->AFSDumpTraceFiles;
7784
7785         AFSRetrieveAuthGroupFnc = LibraryInit->AFSRetrieveAuthGroup;
7786
7787         AFSLibCacheManagerCallbacks = LibraryInit->AFSCacheManagerCallbacks;
7788
7789         if( LibraryInit->AFSCacheBaseAddress != NULL)
7790         {
7791
7792             SetFlag( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE);
7793
7794             AFSLibCacheBaseAddress = LibraryInit->AFSCacheBaseAddress;
7795
7796             AFSLibCacheLength = LibraryInit->AFSCacheLength;
7797         }
7798
7799         //
7800         // Initialize some flush parameters
7801         //
7802
7803         pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
7804
7805         ulTimeIncrement = KeQueryTimeIncrement();
7806
7807         pControlDevExt->Specific.Control.ObjectLifeTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_OBJECT_LIFETIME / (ULONGLONG)ulTimeIncrement);
7808         pControlDevExt->Specific.Control.FcbPurgeTimeCount.QuadPart = AFS_SERVER_PURGE_DELAY;
7809         pControlDevExt->Specific.Control.FcbPurgeTimeCount.QuadPart /= ulTimeIncrement;
7810         pControlDevExt->Specific.Control.FcbFlushTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_SERVER_FLUSH_DELAY / (ULONGLONG)ulTimeIncrement);
7811         pControlDevExt->Specific.Control.ExtentRequestTimeCount.QuadPart = (ULONGLONG)((ULONGLONG)AFS_EXTENT_REQUEST_TIME/(ULONGLONG)ulTimeIncrement);
7812
7813         //
7814         // Initialize the global root entry
7815         //
7816
7817         ntStatus = AFSInitVolume( NULL,
7818                                   &LibraryInit->GlobalRootFid,
7819                                   &AFSGlobalRoot);
7820
7821         if( !NT_SUCCESS( ntStatus))
7822         {
7823
7824             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
7825                           AFS_TRACE_LEVEL_ERROR,
7826                           "AFSInitializeLibrary AFSInitVolume failure %08lX\n",
7827                           ntStatus);
7828
7829             try_return( ntStatus);
7830         }
7831
7832         ntStatus = AFSInitRootFcb( (ULONGLONG)PsGetCurrentProcessId(),
7833                                    AFSGlobalRoot);
7834
7835         if( !NT_SUCCESS( ntStatus))
7836         {
7837
7838             AFSDbgLogMsg( AFS_SUBSYSTEM_LOAD_LIBRARY | AFS_SUBSYSTEM_INIT_PROCESSING,
7839                           AFS_TRACE_LEVEL_ERROR,
7840                           "AFSInitializeLibrary AFSInitRootFcb failure %08lX\n",
7841                           ntStatus);
7842
7843             lCount = InterlockedDecrement( &AFSGlobalRoot->VolumeReferenceCount);
7844
7845             AFSReleaseResource( AFSGlobalRoot->VolumeLock);
7846
7847             try_return( ntStatus);
7848         }
7849
7850         //
7851         // Update the node type code to AFS_ROOT_ALL
7852         //
7853
7854         AFSGlobalRoot->ObjectInformation.Fcb->Header.NodeTypeCode = AFS_ROOT_ALL;
7855
7856         SetFlag( AFSGlobalRoot->Flags, AFS_VOLUME_ACTIVE_GLOBAL_ROOT);
7857
7858         //
7859         // Invalidate all known volumes since contact with the service and therefore
7860         // the file server was lost.
7861         //
7862
7863         AFSInvalidateAllVolumes();
7864
7865         //
7866         // Drop the locks acquired above
7867         //
7868
7869         AFSInitVolumeWorker( AFSGlobalRoot);
7870
7871         lCount = InterlockedDecrement( &AFSGlobalRoot->VolumeReferenceCount);
7872
7873         AFSReleaseResource( AFSGlobalRoot->VolumeLock);
7874
7875         AFSReleaseResource( AFSGlobalRoot->ObjectInformation.Fcb->Header.Resource);
7876
7877 try_exit:
7878
7879         NOTHING;
7880     }
7881
7882     return ntStatus;
7883 }
7884
7885 NTSTATUS
7886 AFSCloseLibrary()
7887 {
7888
7889     NTSTATUS ntStatus = STATUS_SUCCESS;
7890     AFSDirectoryCB *pDirNode = NULL, *pLastDirNode = NULL;
7891
7892     __Enter
7893     {
7894
7895         if( AFSGlobalDotDirEntry != NULL)
7896         {
7897
7898             AFSDeleteObjectInfo( AFSGlobalDotDirEntry->ObjectInformation);
7899
7900             ExDeleteResourceLite( &AFSGlobalDotDirEntry->NonPaged->Lock);
7901
7902             ExFreePool( AFSGlobalDotDirEntry->NonPaged);
7903
7904             ExFreePool( AFSGlobalDotDirEntry);
7905
7906             AFSGlobalDotDirEntry = NULL;
7907         }
7908
7909         if( AFSGlobalDotDotDirEntry != NULL)
7910         {
7911
7912             AFSDeleteObjectInfo( AFSGlobalDotDotDirEntry->ObjectInformation);
7913
7914             ExDeleteResourceLite( &AFSGlobalDotDotDirEntry->NonPaged->Lock);
7915
7916             ExFreePool( AFSGlobalDotDotDirEntry->NonPaged);
7917
7918             ExFreePool( AFSGlobalDotDotDirEntry);
7919
7920             AFSGlobalDotDotDirEntry = NULL;
7921         }
7922
7923         if( AFSSpecialShareNames != NULL)
7924         {
7925
7926             pDirNode = AFSSpecialShareNames;
7927
7928             while( pDirNode != NULL)
7929             {
7930
7931                 pLastDirNode = (AFSDirectoryCB *)pDirNode->ListEntry.fLink;
7932
7933                 AFSDeleteObjectInfo( pDirNode->ObjectInformation);
7934
7935                 ExDeleteResourceLite( &pDirNode->NonPaged->Lock);
7936
7937                 ExFreePool( pDirNode->NonPaged);
7938
7939                 ExFreePool( pDirNode);
7940
7941                 pDirNode = pLastDirNode;
7942             }
7943
7944             AFSSpecialShareNames = NULL;
7945         }
7946     }
7947
7948     return ntStatus;
7949 }
7950
7951 NTSTATUS
7952 AFSDefaultLogMsg( IN ULONG Subsystem,
7953                   IN ULONG Level,
7954                   IN PCCH Format,
7955                   ...)
7956 {
7957
7958     NTSTATUS ntStatus = STATUS_SUCCESS;
7959     va_list va_args;
7960     char chDebugBuffer[ 256];
7961
7962     __Enter
7963     {
7964
7965         va_start( va_args, Format);
7966
7967         ntStatus = RtlStringCbVPrintfA( chDebugBuffer,
7968                                         256,
7969                                         Format,
7970                                         va_args);
7971
7972         if( NT_SUCCESS( ntStatus))
7973         {
7974             DbgPrint( chDebugBuffer);
7975         }
7976
7977         va_end( va_args);
7978     }
7979
7980     return ntStatus;
7981 }
7982
7983 NTSTATUS
7984 AFSGetObjectStatus( IN AFSGetStatusInfoCB *GetStatusInfo,
7985                     IN ULONG InputBufferLength,
7986                     IN AFSStatusInfoCB *StatusInfo,
7987                     OUT ULONG *ReturnLength)
7988 {
7989
7990     NTSTATUS ntStatus = STATUS_SUCCESS;
7991     AFSFcb   *pFcb = NULL;
7992     AFSVolumeCB *pVolumeCB = NULL;
7993     AFSDeviceExt *pDevExt = (AFSDeviceExt *) AFSRDRDeviceObject->DeviceExtension;
7994     AFSObjectInfoCB *pObjectInfo = NULL;
7995     ULONGLONG   ullIndex = 0;
7996     UNICODE_STRING uniFullPathName, uniRemainingPath, uniComponentName, uniParsedName;
7997     AFSNameArrayHdr *pNameArray = NULL;
7998     AFSDirectoryCB *pDirectoryEntry = NULL, *pParentDirEntry = NULL;
7999     LONG lCount;
8000
8001     __Enter
8002     {
8003
8004         //
8005         // If we are given a FID then look up the entry by that, otherwise
8006         // do it by name
8007         //
8008
8009         if( GetStatusInfo->FileID.Cell != 0 &&
8010             GetStatusInfo->FileID.Volume != 0 &&
8011             GetStatusInfo->FileID.Vnode != 0 &&
8012             GetStatusInfo->FileID.Unique != 0)
8013         {
8014
8015             AFSAcquireShared( &pDevExt->Specific.RDR.VolumeTreeLock, TRUE);
8016
8017             //
8018             // Locate the volume node
8019             //
8020
8021             ullIndex = AFSCreateHighIndex( &GetStatusInfo->FileID);
8022
8023             ntStatus = AFSLocateHashEntry( pDevExt->Specific.RDR.VolumeTree.TreeHead,
8024                                            ullIndex,
8025                                            (AFSBTreeEntry **)&pVolumeCB);
8026
8027             if( pVolumeCB != NULL)
8028             {
8029
8030                 lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
8031
8032                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8033                               AFS_TRACE_LEVEL_VERBOSE,
8034                               "AFSGetObjectStatus Increment count on volume %08lX Cnt %d\n",
8035                               pVolumeCB,
8036                               lCount);
8037             }
8038
8039             AFSReleaseResource( &pDevExt->Specific.RDR.VolumeTreeLock);
8040
8041             if( !NT_SUCCESS( ntStatus) ||
8042                 pVolumeCB == NULL)
8043             {
8044                 try_return( ntStatus = STATUS_INVALID_PARAMETER);
8045             }
8046
8047             if( AFSIsVolumeFID( &GetStatusInfo->FileID))
8048             {
8049
8050                 pObjectInfo = &pVolumeCB->ObjectInformation;
8051
8052                 lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8053
8054                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8055             }
8056             else
8057             {
8058
8059                 AFSAcquireShared( pVolumeCB->ObjectInfoTree.TreeLock,
8060                                   TRUE);
8061
8062                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8063
8064                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8065                               AFS_TRACE_LEVEL_VERBOSE,
8066                               "AFSGetObjectStatus Decrement count on volume %08lX Cnt %d\n",
8067                               pVolumeCB,
8068                               lCount);
8069
8070                 ullIndex = AFSCreateLowIndex( &GetStatusInfo->FileID);
8071
8072                 ntStatus = AFSLocateHashEntry( pVolumeCB->ObjectInfoTree.TreeHead,
8073                                                ullIndex,
8074                                                (AFSBTreeEntry **)&pObjectInfo);
8075
8076                 if( pObjectInfo != NULL)
8077                 {
8078
8079                     //
8080                     // Reference the node so it won't be torn down
8081                     //
8082
8083                     lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8084
8085                     AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
8086                                   AFS_TRACE_LEVEL_VERBOSE,
8087                                   "AFSGetObjectStatus Increment count on object %08lX Cnt %d\n",
8088                                   pObjectInfo,
8089                                   lCount);
8090                 }
8091
8092                 AFSReleaseResource( pVolumeCB->ObjectInfoTree.TreeLock);
8093
8094                 if( !NT_SUCCESS( ntStatus) ||
8095                     pObjectInfo == NULL)
8096                 {
8097                     try_return( ntStatus = STATUS_INVALID_PARAMETER);
8098                 }
8099             }
8100         }
8101         else
8102         {
8103
8104             if( GetStatusInfo->FileNameLength == 0 ||
8105                 InputBufferLength < (ULONG)(FIELD_OFFSET( AFSGetStatusInfoCB, FileName) + GetStatusInfo->FileNameLength))
8106             {
8107                 try_return( ntStatus = STATUS_INVALID_PARAMETER);
8108             }
8109
8110             uniFullPathName.Length = GetStatusInfo->FileNameLength;
8111             uniFullPathName.MaximumLength = uniFullPathName.Length;
8112
8113             uniFullPathName.Buffer = (WCHAR *)GetStatusInfo->FileName;
8114
8115             //
8116             // This name should begin with the \afs server so parse it off and check it
8117             //
8118
8119             FsRtlDissectName( uniFullPathName,
8120                               &uniComponentName,
8121                               &uniRemainingPath);
8122
8123             if( RtlCompareUnicodeString( &uniComponentName,
8124                                          &AFSServerName,
8125                                          TRUE) != 0)
8126             {
8127                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8128                               AFS_TRACE_LEVEL_ERROR,
8129                               "AFSGetObjectStatus Name %wZ contains invalid server name\n",
8130                               &uniFullPathName);
8131
8132                 try_return( ntStatus = STATUS_OBJECT_PATH_INVALID);
8133             }
8134
8135             uniFullPathName = uniRemainingPath;
8136
8137             uniParsedName = uniFullPathName;
8138
8139             //
8140             // Our name array
8141             //
8142
8143             pNameArray = AFSInitNameArray( AFSGlobalRoot->DirectoryCB,
8144                                            0);
8145
8146             if( pNameArray == NULL)
8147             {
8148                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8149             }
8150
8151             pVolumeCB = AFSGlobalRoot;
8152
8153             pParentDirEntry = AFSGlobalRoot->DirectoryCB;
8154
8155             //
8156             // Increment the ref count on the volume and dir entry for correct processing below
8157             //
8158
8159             lCount = InterlockedIncrement( &pVolumeCB->VolumeReferenceCount);
8160
8161             AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8162                           AFS_TRACE_LEVEL_VERBOSE,
8163                           "AFSGetObjectStatus Increment count on volume %08lX Cnt %d\n",
8164                           pVolumeCB,
8165                           lCount);
8166
8167             lCount = InterlockedIncrement( &pParentDirEntry->OpenReferenceCount);
8168
8169             AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8170                           AFS_TRACE_LEVEL_VERBOSE,
8171                           "AFSGetObjectStatus Increment count on %wZ DE %p Ccb %p Cnt %d\n",
8172                           &pParentDirEntry->NameInformation.FileName,
8173                           pParentDirEntry,
8174                           NULL,
8175                           lCount);
8176
8177             ntStatus = AFSLocateNameEntry( NULL,
8178                                            NULL,
8179                                            &uniFullPathName,
8180                                            &uniParsedName,
8181                                            pNameArray,
8182                                            AFS_LOCATE_FLAGS_NO_MP_TARGET_EVAL |
8183                                                         AFS_LOCATE_FLAGS_NO_SL_TARGET_EVAL,
8184                                            &pVolumeCB,
8185                                            &pParentDirEntry,
8186                                            &pDirectoryEntry,
8187                                            NULL);
8188
8189             if( !NT_SUCCESS( ntStatus))
8190             {
8191
8192                 //
8193                 // The volume lock was released on failure above
8194                 // Except for STATUS_OBJECT_NAME_NOT_FOUND
8195                 //
8196
8197                 if( ntStatus == STATUS_OBJECT_NAME_NOT_FOUND)
8198                 {
8199
8200                     if( pVolumeCB != NULL)
8201                     {
8202
8203                         lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8204
8205                         AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8206                                       AFS_TRACE_LEVEL_VERBOSE,
8207                                       "AFSGetObjectStatus Decrement count on volume %08lX Cnt %d\n",
8208                                       pVolumeCB,
8209                                       lCount);
8210                     }
8211
8212                     if( pDirectoryEntry != NULL)
8213                     {
8214
8215                         lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
8216
8217                         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8218                                       AFS_TRACE_LEVEL_VERBOSE,
8219                                       "AFSGetObjectStatus Decrement1 count on %wZ DE %p Ccb %p Cnt %d\n",
8220                                       &pDirectoryEntry->NameInformation.FileName,
8221                                       pDirectoryEntry,
8222                                       NULL,
8223                                       lCount);
8224                     }
8225                     else
8226                     {
8227
8228                         lCount = InterlockedDecrement( &pParentDirEntry->OpenReferenceCount);
8229
8230                         AFSDbgLogMsg( AFS_SUBSYSTEM_DIRENTRY_REF_COUNTING,
8231                                       AFS_TRACE_LEVEL_VERBOSE,
8232                                       "AFSGetObjectStatus Decrement2 count on %wZ DE %p Ccb %p Cnt %d\n",
8233                                       &pParentDirEntry->NameInformation.FileName,
8234                                       pParentDirEntry,
8235                                       NULL,
8236                                       lCount);
8237                     }
8238                 }
8239
8240                 pVolumeCB = NULL;
8241
8242                 try_return( ntStatus);
8243             }
8244
8245             //
8246             // Remove the reference made above
8247             //
8248
8249             lCount = InterlockedDecrement( &pDirectoryEntry->OpenReferenceCount);
8250
8251             pObjectInfo = pDirectoryEntry->ObjectInformation;
8252
8253             lCount = InterlockedIncrement( &pObjectInfo->ObjectReferenceCount);
8254
8255             if( pVolumeCB != NULL)
8256             {
8257
8258                 lCount = InterlockedDecrement( &pVolumeCB->VolumeReferenceCount);
8259
8260                 AFSDbgLogMsg( AFS_SUBSYSTEM_VOLUME_REF_COUNTING,
8261                               AFS_TRACE_LEVEL_VERBOSE,
8262                               "AFSRetrieveFileAttributes Decrement2 count on volume %08lX Cnt %d\n",
8263                               pVolumeCB,
8264                               pVolumeCB->VolumeReferenceCount);
8265             }
8266         }
8267
8268         //
8269         // At this point we have an object info block, return the information
8270         //
8271
8272         StatusInfo->FileId = pObjectInfo->FileId;
8273
8274         StatusInfo->TargetFileId = pObjectInfo->TargetFileId;
8275
8276         StatusInfo->Expiration = pObjectInfo->Expiration;
8277
8278         StatusInfo->DataVersion = pObjectInfo->DataVersion;
8279
8280         StatusInfo->FileType = pObjectInfo->FileType;
8281
8282         StatusInfo->ObjectFlags = pObjectInfo->Flags;
8283
8284         StatusInfo->CreationTime = pObjectInfo->CreationTime;
8285
8286         StatusInfo->LastAccessTime = pObjectInfo->LastAccessTime;
8287
8288         StatusInfo->LastWriteTime = pObjectInfo->LastWriteTime;
8289
8290         StatusInfo->ChangeTime = pObjectInfo->ChangeTime;
8291
8292         StatusInfo->FileAttributes = pObjectInfo->FileAttributes;
8293
8294         StatusInfo->EndOfFile = pObjectInfo->EndOfFile;
8295
8296         StatusInfo->AllocationSize = pObjectInfo->AllocationSize;
8297
8298         StatusInfo->EaSize = pObjectInfo->EaSize;
8299
8300         StatusInfo->Links = pObjectInfo->Links;
8301
8302         //
8303         // Return the information length
8304         //
8305
8306         *ReturnLength = sizeof( AFSStatusInfoCB);
8307
8308 try_exit:
8309
8310         if( pObjectInfo != NULL)
8311         {
8312
8313             lCount = InterlockedDecrement( &pObjectInfo->ObjectReferenceCount);
8314         }
8315
8316         if( pNameArray != NULL)
8317         {
8318
8319             AFSFreeNameArray( pNameArray);
8320         }
8321     }
8322
8323     return ntStatus;
8324 }
8325
8326 NTSTATUS
8327 AFSCheckSymlinkAccess( IN AFSDirectoryCB *ParentDirectoryCB,
8328                        IN UNICODE_STRING *ComponentName)
8329 {
8330
8331     NTSTATUS ntStatus = STATUS_SUCCESS;
8332     AFSDirectoryCB *pDirEntry = NULL;
8333     ULONG ulCRC = 0;
8334     LONG lCount;
8335
8336     __Enter
8337     {
8338
8339         //
8340         // Search for the entry in the parent
8341         //
8342
8343         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8344                       AFS_TRACE_LEVEL_VERBOSE_2,
8345                       "AFSCheckSymlinkAccess Searching for entry %wZ case sensitive\n",
8346                       ComponentName);
8347
8348         ulCRC = AFSGenerateCRC( ComponentName,
8349                                 FALSE);
8350
8351         AFSAcquireShared( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
8352                           TRUE);
8353
8354         AFSLocateCaseSensitiveDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
8355                                         ulCRC,
8356                                         &pDirEntry);
8357
8358         if( pDirEntry == NULL)
8359         {
8360
8361             //
8362             // Missed so perform a case insensitive lookup
8363             //
8364
8365             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8366                           AFS_TRACE_LEVEL_VERBOSE_2,
8367                           "AFSCheckSymlinkAccess Searching for entry %wZ case insensitive\n",
8368                           ComponentName);
8369
8370             ulCRC = AFSGenerateCRC( ComponentName,
8371                                     TRUE);
8372
8373             AFSLocateCaseInsensitiveDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
8374                                               ulCRC,
8375                                               &pDirEntry);
8376
8377             if( pDirEntry == NULL)
8378             {
8379
8380                 //
8381                 // OK, if this component is a valid short name then try
8382                 // a lookup in the short name tree
8383                 //
8384
8385                 if( RtlIsNameLegalDOS8Dot3( ComponentName,
8386                                             NULL,
8387                                             NULL))
8388                 {
8389
8390                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8391                                   AFS_TRACE_LEVEL_VERBOSE_2,
8392                                   "AFSCheckSymlinkAccess Searching for entry %wZ short name\n",
8393                                   ComponentName);
8394
8395                     AFSLocateShortNameDirEntry( ParentDirectoryCB->ObjectInformation->Specific.Directory.ShortNameTree,
8396                                                 ulCRC,
8397                                                 &pDirEntry);
8398                 }
8399             }
8400         }
8401
8402         if( pDirEntry != NULL)
8403         {
8404             lCount = InterlockedIncrement( &pDirEntry->OpenReferenceCount);
8405         }
8406
8407         AFSReleaseResource( ParentDirectoryCB->ObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
8408
8409         if( pDirEntry == NULL)
8410         {
8411
8412             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8413                           AFS_TRACE_LEVEL_VERBOSE_2,
8414                           "AFSCheckSymlinkAccess Failed to locate entry %wZ\n",
8415                           ComponentName);
8416
8417             try_return( ntStatus = STATUS_OBJECT_NAME_NOT_FOUND);
8418         }
8419
8420         //
8421         // We have the symlink object but previously failed to process it so return access
8422         // denied.
8423         //
8424
8425         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
8426                       AFS_TRACE_LEVEL_VERBOSE_2,
8427                       "AFSCheckSymlinkAccess Failing symlink access to entry %wZ REPARSE_POINT_NOT_RESOLVED\n",
8428                       ComponentName);
8429
8430         ntStatus = STATUS_REPARSE_POINT_NOT_RESOLVED;
8431
8432         lCount = InterlockedDecrement( &pDirEntry->OpenReferenceCount);
8433
8434 try_exit:
8435
8436         NOTHING;
8437     }
8438
8439     return ntStatus;
8440 }
8441
8442 NTSTATUS
8443 AFSRetrieveFinalComponent( IN UNICODE_STRING *FullPathName,
8444                            OUT UNICODE_STRING *ComponentName)
8445 {
8446
8447     NTSTATUS ntStatus = STATUS_SUCCESS;
8448     UNICODE_STRING uniFullPathName, uniRemainingPath, uniComponentName;
8449
8450     uniFullPathName = *FullPathName;
8451
8452     while( TRUE)
8453     {
8454
8455         FsRtlDissectName( uniFullPathName,
8456                           &uniComponentName,
8457                           &uniRemainingPath);
8458
8459         if( uniRemainingPath.Length == 0)
8460         {
8461             break;
8462         }
8463
8464         uniFullPathName = uniRemainingPath;
8465     }
8466
8467     if( uniComponentName.Length > 0)
8468     {
8469         *ComponentName = uniComponentName;
8470     }
8471
8472     return ntStatus;
8473 }
8474
8475 void
8476 AFSDumpTraceFiles_Default()
8477 {
8478     return;
8479 }
8480
8481 BOOLEAN
8482 AFSValidNameFormat( IN UNICODE_STRING *FileName)
8483 {
8484
8485     BOOLEAN bIsValidName = TRUE;
8486     USHORT usIndex = 0;
8487
8488     __Enter
8489     {
8490
8491         while( usIndex < FileName->Length/sizeof( WCHAR))
8492         {
8493
8494             if( FileName->Buffer[ usIndex] == L':' ||
8495                 FileName->Buffer[ usIndex] == L'*' ||
8496                 FileName->Buffer[ usIndex] == L'?' ||
8497                 FileName->Buffer[ usIndex] == L'"' ||
8498                 FileName->Buffer[ usIndex] == L'<' ||
8499                 FileName->Buffer[ usIndex] == L'>')
8500             {
8501                 bIsValidName = FALSE;
8502                 break;
8503             }
8504
8505             usIndex++;
8506         }
8507     }
8508
8509     return bIsValidName;
8510 }
8511
8512 NTSTATUS
8513 AFSCreateDefaultSecurityDescriptor()
8514 {
8515
8516     NTSTATUS ntStatus = STATUS_SUCCESS;
8517     PACL pSACL = NULL;
8518     ULONG ulSACLSize = 0;
8519     SYSTEM_MANDATORY_LABEL_ACE* pACE = NULL;
8520     ULONG ulACESize = 0;
8521     SECURITY_DESCRIPTOR *pSecurityDescr = NULL;
8522     ULONG ulSDLength = 0;
8523     SECURITY_DESCRIPTOR *pRelativeSecurityDescr = NULL;
8524     PSID pWorldSID = NULL;
8525     ULONG *pulSubAuthority = NULL;
8526     ULONG ulWorldSIDLEngth = 0;
8527
8528     __Enter
8529     {
8530
8531         ulWorldSIDLEngth = RtlLengthRequiredSid( 1);
8532
8533         pWorldSID = (PSID)ExAllocatePoolWithTag( PagedPool,
8534                                                  ulWorldSIDLEngth,
8535                                                  AFS_GENERIC_MEMORY_29_TAG);
8536
8537         if( pWorldSID == NULL)
8538         {
8539             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate World SID\n");
8540             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8541         }
8542
8543         RtlZeroMemory( pWorldSID,
8544                        ulWorldSIDLEngth);
8545
8546         RtlInitializeSid( pWorldSID,
8547                           &SeWorldSidAuthority,
8548                           1);
8549
8550         pulSubAuthority = RtlSubAuthoritySid(pWorldSID, 0);
8551         *pulSubAuthority = SECURITY_WORLD_RID;
8552
8553         if( AFSRtlSetSaclSecurityDescriptor == NULL)
8554         {
8555
8556             AFSPrint( "AFSCreateDefaultSecurityDescriptor AFSRtlSetSaclSecurityDescriptor == NULL\n");
8557         }
8558         else
8559         {
8560
8561             ulACESize = sizeof( SYSTEM_MANDATORY_LABEL_ACE) + 128;
8562
8563             pACE = (SYSTEM_MANDATORY_LABEL_ACE *)ExAllocatePoolWithTag( PagedPool,
8564                                                                         ulACESize,
8565                                                                         AFS_GENERIC_MEMORY_29_TAG);
8566
8567             if( pACE == NULL)
8568             {
8569
8570                 AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_29_TAG\n");
8571
8572                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8573             }
8574
8575             RtlZeroMemory( pACE,
8576                            ulACESize);
8577
8578             pACE->Header.AceFlags = OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE;
8579             pACE->Header.AceType = SYSTEM_MANDATORY_LABEL_ACE_TYPE;
8580             pACE->Header.AceSize = FIELD_OFFSET( SYSTEM_MANDATORY_LABEL_ACE, SidStart) + (USHORT)RtlLengthSid( SeExports->SeLowMandatorySid);
8581             pACE->Mask = SYSTEM_MANDATORY_LABEL_NO_WRITE_UP;
8582
8583             RtlCopySid( RtlLengthSid( SeExports->SeLowMandatorySid),
8584                         &pACE->SidStart,
8585                         SeExports->SeLowMandatorySid);
8586
8587             ulSACLSize = sizeof(ACL) + RtlLengthSid( SeExports->SeLowMandatorySid) +
8588                 FIELD_OFFSET( SYSTEM_MANDATORY_LABEL_ACE, SidStart) + ulACESize;
8589
8590             pSACL = (PACL)ExAllocatePoolWithTag( PagedPool,
8591                                                  ulSACLSize,
8592                                                  AFS_GENERIC_MEMORY_29_TAG);
8593
8594             if( pSACL == NULL)
8595             {
8596
8597                 AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_29_TAG\n");
8598
8599                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8600             }
8601
8602             ntStatus = RtlCreateAcl( pSACL,
8603                                      ulSACLSize,
8604                                      ACL_REVISION);
8605
8606             if( !NT_SUCCESS( ntStatus))
8607             {
8608
8609                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlCreateAcl ntStatus %08lX\n",
8610                           ntStatus);
8611
8612                 try_return( ntStatus);
8613             }
8614
8615             ntStatus = RtlAddAce( pSACL,
8616                                   ACL_REVISION,
8617                                   0,
8618                                   pACE,
8619                                   pACE->Header.AceSize);
8620
8621             if( !NT_SUCCESS( ntStatus))
8622             {
8623
8624                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlAddAce ntStatus %08lX\n",
8625                           ntStatus);
8626
8627                 try_return( ntStatus);
8628             }
8629         }
8630
8631         pSecurityDescr = (SECURITY_DESCRIPTOR *)ExAllocatePoolWithTag( NonPagedPool,
8632                                                                        sizeof( SECURITY_DESCRIPTOR),
8633                                                                        AFS_GENERIC_MEMORY_27_TAG);
8634
8635         if( pSecurityDescr == NULL)
8636         {
8637
8638             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_27_TAG\n");
8639
8640             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8641         }
8642
8643         ntStatus = RtlCreateSecurityDescriptor( pSecurityDescr,
8644                                                 SECURITY_DESCRIPTOR_REVISION);
8645
8646         if( !NT_SUCCESS( ntStatus))
8647         {
8648
8649             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlCreateSecurityDescriptor ntStatus %08lX\n",
8650                       ntStatus);
8651
8652             try_return( ntStatus);
8653         }
8654
8655         if( AFSRtlSetSaclSecurityDescriptor != NULL)
8656         {
8657             ntStatus = AFSRtlSetSaclSecurityDescriptor( pSecurityDescr,
8658                                                         TRUE,
8659                                                         pSACL,
8660                                                         FALSE);
8661
8662             if( !NT_SUCCESS( ntStatus))
8663             {
8664
8665                 AFSPrint( "AFSCreateDefaultSecurityDescriptor AFSRtlSetSaclSecurityDescriptor ntStatus %08lX\n",
8666                           ntStatus);
8667
8668                 try_return( ntStatus);
8669             }
8670         }
8671
8672         //
8673         // Add in the group and owner to the SD
8674         //
8675
8676         if( AFSRtlSetGroupSecurityDescriptor != NULL)
8677         {
8678             ntStatus = AFSRtlSetGroupSecurityDescriptor( pSecurityDescr,
8679                                                          pWorldSID,
8680                                                          FALSE);
8681
8682             if( !NT_SUCCESS( ntStatus))
8683             {
8684
8685                 AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlSetGroupSecurityDescriptor failed ntStatus %08lX\n",
8686                           ntStatus);
8687
8688                 try_return( ntStatus);
8689             }
8690         }
8691
8692         ntStatus = RtlSetOwnerSecurityDescriptor( pSecurityDescr,
8693                                                   pWorldSID,
8694                                                   FALSE);
8695
8696         if( !NT_SUCCESS( ntStatus))
8697         {
8698
8699             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlSetOwnerSecurityDescriptor failed ntStatus %08lX\n",
8700                       ntStatus);
8701
8702             try_return( ntStatus);
8703         }
8704
8705         if( !RtlValidSecurityDescriptor( pSecurityDescr))
8706         {
8707
8708             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlValidSecurityDescriptor NOT\n");
8709
8710             try_return( ntStatus = STATUS_INVALID_PARAMETER);
8711         }
8712
8713         pRelativeSecurityDescr = (SECURITY_DESCRIPTOR *)ExAllocatePoolWithTag( NonPagedPool,
8714                                                                                PAGE_SIZE,
8715                                                                                AFS_GENERIC_MEMORY_27_TAG);
8716
8717         if( pRelativeSecurityDescr == NULL)
8718         {
8719
8720             AFSPrint( "AFSCreateDefaultSecurityDescriptor unable to allocate AFS_GENERIC_MEMORY_27_TAG\n");
8721
8722             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
8723         }
8724
8725         ulSDLength = PAGE_SIZE;
8726
8727         ntStatus = RtlAbsoluteToSelfRelativeSD( pSecurityDescr,
8728                                                 pRelativeSecurityDescr,
8729                                                 &ulSDLength);
8730
8731         if( !NT_SUCCESS( ntStatus))
8732         {
8733
8734             AFSPrint( "AFSCreateDefaultSecurityDescriptor RtlAbsoluteToSelfRelativeSD ntStatus %08lX\n",
8735                       ntStatus);
8736
8737             try_return( ntStatus);
8738         }
8739
8740         AFSDefaultSD = pRelativeSecurityDescr;
8741
8742 try_exit:
8743
8744         if( !NT_SUCCESS( ntStatus))
8745         {
8746
8747             if( pRelativeSecurityDescr != NULL)
8748             {
8749                 ExFreePool( pRelativeSecurityDescr);
8750             }
8751         }
8752
8753         if( pSecurityDescr != NULL)
8754         {
8755             ExFreePool( pSecurityDescr);
8756         }
8757
8758         if( pSACL != NULL)
8759         {
8760             ExFreePool( pSACL);
8761         }
8762
8763         if( pACE != NULL)
8764         {
8765             ExFreePool( pACE);
8766         }
8767
8768         if( pWorldSID != NULL)
8769         {
8770             ExFreePool( pWorldSID);
8771         }
8772     }
8773
8774     return ntStatus;
8775 }
8776
8777 void
8778 AFSRetrieveParentPath( IN UNICODE_STRING *FullFileName,
8779                        OUT UNICODE_STRING *ParentPath)
8780 {
8781
8782     USHORT usIndex = 0;
8783
8784     *ParentPath = *FullFileName;
8785
8786     //
8787     // If the final character is a \, jump over it
8788     //
8789
8790     if( ParentPath->Buffer[ (ParentPath->Length/sizeof( WCHAR)) - 1] == L'\\')
8791     {
8792         ParentPath->Length -= sizeof( WCHAR);
8793     }
8794
8795     while( ParentPath->Buffer[ (ParentPath->Length/sizeof( WCHAR)) - 1] != L'\\')
8796     {
8797         ParentPath->Length -= sizeof( WCHAR);
8798     }
8799
8800     //
8801     // And the separator
8802     //
8803
8804     ParentPath->Length -= sizeof( WCHAR);
8805
8806     return;
8807 }
8808
8809 NTSTATUS
8810 AFSRetrieveValidAuthGroup( IN AFSFcb *Fcb,
8811                            IN AFSObjectInfoCB *ObjectInfo,
8812                            IN BOOLEAN WriteAccess,
8813                            OUT GUID *AuthGroup)
8814 {
8815
8816     NTSTATUS ntStatus = STATUS_SUCCESS;
8817     GUID     stAuthGroup, stZeroAuthGroup;
8818     BOOLEAN  bFoundAuthGroup = FALSE;
8819     AFSCcb  *pCcb = NULL;
8820     AFSFcb *pFcb = Fcb;
8821
8822     __Enter
8823     {
8824
8825         RtlZeroMemory( &stAuthGroup,
8826                        sizeof( GUID));
8827
8828         RtlZeroMemory( &stZeroAuthGroup,
8829                        sizeof( GUID));
8830
8831         if( Fcb == NULL)
8832         {
8833
8834             if( ObjectInfo != NULL &&
8835                 ObjectInfo->Fcb != NULL)
8836             {
8837                 pFcb = ObjectInfo->Fcb;
8838             }
8839         }
8840
8841         if( pFcb != NULL)
8842         {
8843
8844             AFSAcquireShared( &Fcb->NPFcb->CcbListLock,
8845                               TRUE);
8846
8847             pCcb = Fcb->CcbListHead;
8848
8849             while( pCcb != NULL)
8850             {
8851
8852                 if( WriteAccess &&
8853                     pCcb->GrantedAccess & FILE_WRITE_DATA)
8854                 {
8855                     RtlCopyMemory( &stAuthGroup,
8856                                    &pCcb->AuthGroup,
8857                                    sizeof( GUID));
8858
8859                     bFoundAuthGroup = TRUE;
8860
8861                     break;
8862                 }
8863                 else if( pCcb->GrantedAccess & FILE_READ_DATA)
8864                 {
8865                     //
8866                     // At least get the read-only access
8867                     //
8868
8869                     RtlCopyMemory( &stAuthGroup,
8870                                    &pCcb->AuthGroup,
8871                                    sizeof( GUID));
8872
8873                     bFoundAuthGroup = TRUE;
8874                 }
8875
8876                 pCcb = (AFSCcb *)pCcb->ListEntry.fLink;
8877             }
8878
8879             AFSReleaseResource( &Fcb->NPFcb->CcbListLock);
8880         }
8881
8882         if( !bFoundAuthGroup)
8883         {
8884
8885             AFSRetrieveAuthGroupFnc( (ULONGLONG)PsGetCurrentProcessId(),
8886                                      (ULONGLONG)PsGetCurrentThreadId(),
8887                                       &stAuthGroup);
8888
8889             if( RtlCompareMemory( &stZeroAuthGroup,
8890                                   &stAuthGroup,
8891                                   sizeof( GUID)) == sizeof( GUID))
8892             {
8893
8894                 DbgPrint("AFSRetrieveValidAuthGroup Failed to locate PAG\n");
8895
8896                 try_return( ntStatus = STATUS_ACCESS_DENIED);
8897             }
8898         }
8899
8900         RtlCopyMemory( AuthGroup,
8901                        &stAuthGroup,
8902                        sizeof( GUID));
8903
8904 try_exit:
8905
8906         NOTHING;
8907     }
8908
8909     return ntStatus;
8910 }
8911
8912 NTSTATUS
8913 AFSPerformObjectInvalidate( IN AFSObjectInfoCB *ObjectInfo,
8914                             IN ULONG InvalidateReason)
8915 {
8916
8917     NTSTATUS            ntStatus = STATUS_SUCCESS;
8918     IO_STATUS_BLOCK     stIoStatus;
8919     LIST_ENTRY         *le;
8920     AFSExtent          *pEntry;
8921     ULONG               ulProcessCount = 0;
8922     ULONG               ulCount = 0;
8923
8924     __Enter
8925     {
8926
8927         switch( InvalidateReason)
8928         {
8929
8930             case AFS_INVALIDATE_DELETED:
8931             {
8932
8933                 if( ObjectInfo->FileType == AFS_FILE_TYPE_FILE &&
8934                     ObjectInfo->Fcb != NULL)
8935                 {
8936
8937
8938                     //
8939                     // Clear out the extents
8940                     // And get rid of them (note this involves waiting
8941                     // for any writes or reads to the cache to complete)
8942                     //
8943
8944                     AFSTearDownFcbExtents( ObjectInfo->Fcb,
8945                                            NULL);
8946                 }
8947
8948                 break;
8949             }
8950
8951             case AFS_INVALIDATE_DATA_VERSION:
8952             {
8953
8954                 LARGE_INTEGER liCurrentOffset = {0,0};
8955                 LARGE_INTEGER liFlushLength = {0,0};
8956                 ULONG ulFlushLength = 0;
8957                 BOOLEAN bLocked = FALSE;
8958                 BOOLEAN bExtentsLocked = FALSE;
8959                 BOOLEAN bCleanExtents = FALSE;
8960
8961                 if( ObjectInfo->FileType == AFS_FILE_TYPE_FILE &&
8962                     ObjectInfo->Fcb != NULL)
8963                 {
8964
8965                     AFSAcquireExcl( &ObjectInfo->Fcb->NPFcb->Resource,
8966                                     TRUE);
8967
8968                     bLocked = TRUE;
8969
8970                     AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
8971                                   AFS_TRACE_LEVEL_VERBOSE,
8972                                   "AFSPerformObjectInvalidate Acquiring Fcb extents lock %08lX SHARED %08lX\n",
8973                                   &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource,
8974                                   PsGetCurrentThread());
8975
8976                     AFSAcquireShared( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource,
8977                                       TRUE);
8978
8979                     bExtentsLocked = TRUE;
8980
8981                     //
8982                     // There are several possibilities here:
8983                     //
8984                     // 0. If there are no extents or all of the extents are dirty, do nothing.
8985                     //
8986                     // 1. There could be nothing dirty and an open reference count of zero
8987                     //    in which case we can just tear down all of the extents without
8988                     //    holding any resources.
8989                     //
8990                     // 2. There could be nothing dirty and a non-zero open reference count
8991                     //    in which case we can issue a CcPurge against the entire file
8992                     //    while holding just the Fcb Resource.
8993                     //
8994                     // 3. There can be dirty extents in which case we need to identify
8995                     //    the non-dirty ranges and then perform a CcPurge on just the
8996                     //    non-dirty ranges while holding just the Fcb Resource.
8997                     //
8998
8999                     if ( ObjectInfo->Fcb->Specific.File.ExtentCount != ObjectInfo->Fcb->Specific.File.ExtentsDirtyCount)
9000                     {
9001
9002                         if ( ObjectInfo->Fcb->Specific.File.ExtentsDirtyCount == 0)
9003                         {
9004
9005                             if ( ObjectInfo->Fcb->OpenReferenceCount == 0)
9006                             {
9007
9008                                 AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9009
9010                                 bExtentsLocked = FALSE;
9011
9012                                 AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Resource);
9013
9014                                 bLocked = FALSE;
9015
9016                                 AFSTearDownFcbExtents( ObjectInfo->Fcb,
9017                                                        NULL);
9018                             }
9019                             else
9020                             {
9021
9022                                 __try
9023                                 {
9024
9025                                     AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9026
9027                                     bExtentsLocked = FALSE;
9028
9029                                     if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9030                                                               NULL,
9031                                                               0,
9032                                                               FALSE))
9033                                     {
9034
9035                                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9036                                                       AFS_TRACE_LEVEL_WARNING,
9037                                                       "AFSPerformObjectInvalidation CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9038                                                       ObjectInfo->FileId.Cell,
9039                                                       ObjectInfo->FileId.Volume,
9040                                                       ObjectInfo->FileId.Vnode,
9041                                                       ObjectInfo->FileId.Unique);
9042
9043                                         SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9044                                     }
9045                                     else
9046                                     {
9047
9048                                         bCleanExtents = TRUE;
9049                                     }
9050                                 }
9051                                 __except( EXCEPTION_EXECUTE_HANDLER)
9052                                 {
9053
9054                                     ntStatus = GetExceptionCode();
9055
9056                                     AFSDbgLogMsg( 0,
9057                                                   0,
9058                                                   "EXCEPTION - AFSPerformObjectInvalidation FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX\n",
9059                                                   ObjectInfo->FileId.Cell,
9060                                                   ObjectInfo->FileId.Volume,
9061                                                   ObjectInfo->FileId.Vnode,
9062                                                   ObjectInfo->FileId.Unique,
9063                                                   ntStatus);
9064
9065                                     SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9066                                 }
9067                             }
9068                         }
9069                         else
9070                         {
9071
9072                             //
9073                             // Must build a list of non-dirty ranges from the beginning of the file
9074                             // to the end.  There can be at most (Fcb->Specific.File.ExtentsDirtyCount + 1)
9075                             // ranges.  In all but the most extreme random data write scenario there will
9076                             // be significantly fewer.
9077                             //
9078                             // For each range we need offset and size.
9079                             //
9080
9081                             AFSByteRange * ByteRangeList = NULL;
9082                             ULONG          ulByteRangeCount = 0;
9083                             ULONG          ulIndex;
9084                             BOOLEAN        bPurgeOnClose = FALSE;
9085
9086                             __try
9087                             {
9088
9089                                 ulByteRangeCount = AFSConstructCleanByteRangeList( ObjectInfo->Fcb,
9090                                                                                    &ByteRangeList);
9091
9092                                 if ( ByteRangeList != NULL ||
9093                                      ulByteRangeCount == 0)
9094                                 {
9095
9096                                     AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9097
9098                                     bExtentsLocked = FALSE;
9099
9100                                     for ( ulIndex = 0; ulIndex < ulByteRangeCount; ulIndex++)
9101                                     {
9102
9103                                         ULONG ulSize;
9104
9105                                         do {
9106
9107                                             ulSize = ByteRangeList[ulIndex].Length.QuadPart > DWORD_MAX ? DWORD_MAX : ByteRangeList[ulIndex].Length.LowPart;
9108
9109                                             if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9110                                                                       &ByteRangeList[ulIndex].FileOffset,
9111                                                                       ulSize,
9112                                                                       FALSE))
9113                                             {
9114
9115                                                 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9116                                                               AFS_TRACE_LEVEL_WARNING,
9117                                                               "AFSPerformObjectInvalidation [1] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9118                                                               ObjectInfo->FileId.Cell,
9119                                                               ObjectInfo->FileId.Volume,
9120                                                               ObjectInfo->FileId.Vnode,
9121                                                               ObjectInfo->FileId.Unique);
9122
9123                                                 bPurgeOnClose = TRUE;
9124                                             }
9125                                             else
9126                                             {
9127
9128                                                 bCleanExtents = TRUE;
9129                                             }
9130
9131                                             ByteRangeList[ulIndex].Length.QuadPart -= ulSize;
9132
9133                                             ByteRangeList[ulIndex].FileOffset.QuadPart += ulSize;
9134
9135                                         } while ( ByteRangeList[ulIndex].Length.QuadPart > 0);
9136                                     }
9137                                 }
9138                                 else
9139                                 {
9140
9141                                     //
9142                                     // We couldn't allocate the memory to build the purge list
9143                                     // so just walk the extent list while holding the ExtentsList Resource.
9144                                     // This could deadlock but we do not have much choice.
9145                                     //
9146
9147                                     le = ObjectInfo->Fcb->Specific.File.ExtentsLists[AFS_EXTENTS_LIST].Flink;
9148
9149                                     ulProcessCount = 0;
9150
9151                                     ulCount = (ULONG)ObjectInfo->Fcb->Specific.File.ExtentCount;
9152
9153                                     if( ulCount > 0)
9154                                     {
9155                                         pEntry = ExtentFor( le, AFS_EXTENTS_LIST );
9156
9157                                         while( ulProcessCount < ulCount)
9158                                         {
9159                                             pEntry = ExtentFor( le, AFS_EXTENTS_LIST );
9160
9161                                             if( !BooleanFlagOn( pEntry->Flags, AFS_EXTENT_DIRTY))
9162                                             {
9163                                                 if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9164                                                                           &pEntry->FileOffset,
9165                                                                           pEntry->Size,
9166                                                                           FALSE))
9167                                                 {
9168
9169                                                     AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9170                                                                   AFS_TRACE_LEVEL_WARNING,
9171                                                                   "AFSPerformObjectInvalidation [2] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9172                                                                   ObjectInfo->FileId.Cell,
9173                                                                   ObjectInfo->FileId.Volume,
9174                                                                   ObjectInfo->FileId.Vnode,
9175                                                                   ObjectInfo->FileId.Unique);
9176
9177                                                     bPurgeOnClose = TRUE;
9178                                                 }
9179                                                 else
9180                                                 {
9181
9182                                                     bCleanExtents = TRUE;
9183                                                 }
9184                                             }
9185
9186                                             if( liCurrentOffset.QuadPart < pEntry->FileOffset.QuadPart)
9187                                             {
9188
9189                                                 liFlushLength.QuadPart = pEntry->FileOffset.QuadPart - liCurrentOffset.QuadPart;
9190
9191                                                 while( liFlushLength.QuadPart > 0)
9192                                                 {
9193
9194                                                     if( liFlushLength.QuadPart > 512 * 1024000)
9195                                                     {
9196                                                         ulFlushLength = 512 * 1024000;
9197                                                     }
9198                                                     else
9199                                                     {
9200                                                         ulFlushLength = liFlushLength.LowPart;
9201                                                     }
9202
9203                                                     if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9204                                                                               &liCurrentOffset,
9205                                                                               ulFlushLength,
9206                                                                               FALSE))
9207                                                     {
9208
9209                                                         AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9210                                                                       AFS_TRACE_LEVEL_WARNING,
9211                                                                       "AFSPerformObjectInvalidation [3] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9212                                                                       ObjectInfo->FileId.Cell,
9213                                                                       ObjectInfo->FileId.Volume,
9214                                                                       ObjectInfo->FileId.Vnode,
9215                                                                       ObjectInfo->FileId.Unique);
9216
9217                                                         bPurgeOnClose = TRUE;
9218                                                     }
9219                                                     else
9220                                                     {
9221
9222                                                         bCleanExtents = TRUE;
9223                                                     }
9224
9225                                                     liFlushLength.QuadPart -= ulFlushLength;
9226                                                 }
9227                                             }
9228
9229                                             liCurrentOffset.QuadPart = pEntry->FileOffset.QuadPart + pEntry->Size;
9230
9231                                             ulProcessCount++;
9232                                             le = le->Flink;
9233                                         }
9234                                     }
9235                                     else
9236                                     {
9237                                         if( !CcPurgeCacheSection( &ObjectInfo->Fcb->NPFcb->SectionObjectPointers,
9238                                                                   NULL,
9239                                                                   0,
9240                                                                   FALSE))
9241                                         {
9242
9243                                             AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
9244                                                           AFS_TRACE_LEVEL_WARNING,
9245                                                           "AFSPerformObjectInvalidation [4] CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
9246                                                           ObjectInfo->FileId.Cell,
9247                                                           ObjectInfo->FileId.Volume,
9248                                                           ObjectInfo->FileId.Vnode,
9249                                                           ObjectInfo->FileId.Unique);
9250
9251                                             bPurgeOnClose = TRUE;
9252                                         }
9253                                         else
9254                                         {
9255
9256                                             bCleanExtents = TRUE;
9257                                         }
9258                                     }
9259
9260                                     if ( bPurgeOnClose)
9261                                     {
9262
9263                                         SetFlag( ObjectInfo->Fcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
9264                                     }
9265                                 }
9266                             }
9267                             __except( EXCEPTION_EXECUTE_HANDLER)
9268                             {
9269
9270                                 ntStatus = GetExceptionCode();
9271
9272                                 AFSDbgLogMsg( 0,
9273                                               0,
9274                                               "EXCEPTION - AFSPerformObjectInvalidation FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
9275                                               ObjectInfo->FileId.Cell,
9276                                               ObjectInfo->FileId.Volume,
9277                                               ObjectInfo->FileId.Vnode,
9278                                               ObjectInfo->FileId.Unique,
9279                                               ntStatus);
9280                             }
9281                         }
9282                     }
9283
9284                     if ( bExtentsLocked)
9285                     {
9286
9287                         AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Specific.File.ExtentsResource );
9288                     }
9289
9290                     if ( bLocked)
9291                     {
9292
9293                         AFSReleaseResource( &ObjectInfo->Fcb->NPFcb->Resource);
9294                     }
9295
9296                     if ( bCleanExtents)
9297                     {
9298
9299                         AFSReleaseCleanExtents( ObjectInfo->Fcb,
9300                                                 NULL);
9301                     }
9302                 }
9303
9304                 break;
9305             }
9306         }
9307
9308         if( ObjectInfo != NULL)
9309         {
9310             InterlockedDecrement( &ObjectInfo->ObjectReferenceCount);
9311         }
9312     }
9313
9314     return ntStatus;
9315 }