Windows: AFSRedirLib.sys file system library driver
[openafs.git] / src / WINNT / afsrdr / kernel / lib / AFSFileInfo.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: AFSFileInfo.cpp
37 //
38
39 #include "AFSCommon.h"
40
41 //
42 // Function: AFSQueryFileInfo
43 //
44 // Description:
45 //
46 //      This function is the dispatch handler for the IRP_MJ_QUERY_FILE_INFORMATION request
47 //
48 // Return:
49 //
50 //      A status is returned for the function
51 //
52
53 NTSTATUS
54 AFSQueryFileInfo( IN PDEVICE_OBJECT LibDeviceObject,
55                   IN PIRP Irp)
56 {
57
58     NTSTATUS ntStatus = STATUS_SUCCESS;
59     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
60     ULONG ulRequestType = 0;
61     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
62     AFSFcb *pFcb = NULL;
63     AFSCcb *pCcb = NULL;
64     PFILE_OBJECT pFileObject;
65     BOOLEAN bReleaseMain = FALSE;
66     LONG lLength;
67     FILE_INFORMATION_CLASS stFileInformationClass;
68     PVOID pBuffer;
69
70     __try
71     {
72
73         //
74         // Determine the type of request this request is
75         //
76
77         pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
78
79         pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
80
81         if( pFcb == NULL)
82         {
83
84             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
85                           AFS_TRACE_LEVEL_ERROR,
86                           "AFSQueryFileInfo Attempted access (%08lX) when pFcb == NULL\n",
87                           Irp);
88
89             try_return( ntStatus = STATUS_INVALID_DEVICE_REQUEST);
90         }
91
92         lLength = (LONG)pIrpSp->Parameters.QueryFile.Length;
93         stFileInformationClass = pIrpSp->Parameters.QueryFile.FileInformationClass;
94         pBuffer = Irp->AssociatedIrp.SystemBuffer;
95
96         //
97         // Grab the main shared right off the bat
98         //
99
100         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
101                       AFS_TRACE_LEVEL_VERBOSE,
102                       "AFSQueryFileInfo Acquiring Fcb lock %08lX SHARED %08lX\n",
103                       &pFcb->NPFcb->Resource,
104                       PsGetCurrentThread());
105
106         AFSAcquireShared( &pFcb->NPFcb->Resource,
107                           TRUE);
108
109         bReleaseMain = TRUE;
110
111         //
112         // Don't allow requests against IOCtl nodes
113         //
114
115         if( pFcb->Header.NodeTypeCode == AFS_SPECIAL_SHARE_FCB)
116         {
117
118             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
119                           AFS_TRACE_LEVEL_VERBOSE,
120                           "AFSQueryFileInfo Processing request against SpecialShare Fcb\n");
121
122             ntStatus = AFSProcessShareQueryInfo( Irp,
123                                                  pFcb,
124                                                  pCcb);
125
126             try_return( ntStatus);
127         }
128         else if( pFcb->Header.NodeTypeCode == AFS_IOCTL_FCB)
129         {
130             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
131                           AFS_TRACE_LEVEL_VERBOSE,
132                           "AFSQueryFileInfo request against PIOCtl Fcb\n");
133
134             ntStatus = AFSProcessPIOCtlQueryInfo( Irp,
135                                                   pFcb,
136                                                   pCcb,
137                                                   &lLength);
138
139             try_return( ntStatus);
140         }
141
142         //
143         // Process the request
144         //
145
146         switch( stFileInformationClass)
147         {
148
149             case FileAllInformation:
150             {
151
152                 PFILE_ALL_INFORMATION pAllInfo;
153
154                 //
155                 //  For the all information class we'll typecast a local
156                 //  pointer to the output buffer and then call the
157                 //  individual routines to fill in the buffer.
158                 //
159
160                 pAllInfo = (PFILE_ALL_INFORMATION)pBuffer;
161
162                 ntStatus = AFSQueryBasicInfo( Irp,
163                                               pCcb->DirectoryCB,
164                                               &pAllInfo->BasicInformation,
165                                               &lLength);
166
167                 if( !NT_SUCCESS( ntStatus))
168                 {
169
170                     try_return( ntStatus);
171                 }
172
173                 ntStatus = AFSQueryStandardInfo( Irp,
174                                                  pCcb->DirectoryCB,
175                                                  &pAllInfo->StandardInformation,
176                                                  &lLength);
177
178                 if( !NT_SUCCESS( ntStatus))
179                 {
180
181                     try_return( ntStatus);
182                 }
183
184                 ntStatus = AFSQueryInternalInfo( Irp,
185                                                  pFcb,
186                                                  &pAllInfo->InternalInformation,
187                                                  &lLength);
188
189                 if( !NT_SUCCESS( ntStatus))
190                 {
191
192                     try_return( ntStatus);
193                 }
194
195                 ntStatus = AFSQueryEaInfo( Irp,
196                                            pCcb->DirectoryCB,
197                                            &pAllInfo->EaInformation,
198                                            &lLength);
199
200                 if( !NT_SUCCESS( ntStatus))
201                 {
202
203                     try_return( ntStatus);
204                 }
205
206                 ntStatus = AFSQueryAccess( Irp,
207                                            pFcb,
208                                            &pAllInfo->AccessInformation,
209                                            &lLength);
210
211                 if( !NT_SUCCESS( ntStatus))
212                 {
213
214                     try_return( ntStatus);
215                 }
216
217                 ntStatus = AFSQueryPositionInfo( Irp,
218                                                  pFcb,
219                                                  &pAllInfo->PositionInformation,
220                                                  &lLength);
221
222                 if( !NT_SUCCESS( ntStatus))
223                 {
224
225                     try_return( ntStatus);
226                 }
227
228                 ntStatus = AFSQueryMode( Irp,
229                                          pFcb,
230                                          &pAllInfo->ModeInformation,
231                                          &lLength);
232
233                 if( !NT_SUCCESS( ntStatus))
234                 {
235
236                     try_return( ntStatus);
237                 }
238
239                 ntStatus = AFSQueryAlignment( Irp,
240                                               pFcb,
241                                               &pAllInfo->AlignmentInformation,
242                                               &lLength);
243
244                 if( !NT_SUCCESS( ntStatus))
245                 {
246
247                     try_return( ntStatus);
248                 }
249
250                 ntStatus = AFSQueryNameInfo( Irp,
251                                              pCcb->DirectoryCB,
252                                              &pAllInfo->NameInformation,
253                                              &lLength);
254
255                 if( !NT_SUCCESS( ntStatus))
256                 {
257
258                     try_return( ntStatus);
259                 }
260
261                 break;
262             }
263
264             case FileBasicInformation:
265             {
266
267                 ntStatus = AFSQueryBasicInfo( Irp,
268                                               pCcb->DirectoryCB,
269                                               (PFILE_BASIC_INFORMATION)pBuffer,
270                                               &lLength);
271
272                 break;
273             }
274
275             case FileStandardInformation:
276             {
277
278                 ntStatus = AFSQueryStandardInfo( Irp,
279                                                  pCcb->DirectoryCB,
280                                                  (PFILE_STANDARD_INFORMATION)pBuffer,
281                                                  &lLength);
282
283                 break;
284             }
285
286             case FileInternalInformation:
287             {
288
289                 ntStatus = AFSQueryInternalInfo( Irp,
290                                                  pFcb,
291                                                  (PFILE_INTERNAL_INFORMATION)pBuffer,
292                                                  &lLength);
293
294                 break;
295             }
296
297             case FileEaInformation:
298             {
299
300                 ntStatus = AFSQueryEaInfo( Irp,
301                                            pCcb->DirectoryCB,
302                                            (PFILE_EA_INFORMATION)pBuffer,
303                                            &lLength);
304
305                 break;
306             }
307
308             case FilePositionInformation:
309             {
310
311                 ntStatus = AFSQueryPositionInfo( Irp,
312                                       pFcb,
313                                       (PFILE_POSITION_INFORMATION)pBuffer,
314                                       &lLength);
315
316                 break;
317             }
318
319             case FileNameInformation:
320             {
321
322                 ntStatus = AFSQueryNameInfo( Irp,
323                                   pCcb->DirectoryCB,
324                                   (PFILE_NAME_INFORMATION)pBuffer,
325                                   &lLength);
326
327                 break;
328             }
329
330             case FileAlternateNameInformation:
331             {
332
333                 ntStatus = AFSQueryShortNameInfo( Irp,
334                                        pCcb->DirectoryCB,
335                                        (PFILE_NAME_INFORMATION)pBuffer,
336                                        &lLength);
337
338                 break;
339             }
340
341             case FileNetworkOpenInformation:
342             {
343
344                 ntStatus = AFSQueryNetworkInfo( Irp,
345                                      pCcb->DirectoryCB,
346                                      (PFILE_NETWORK_OPEN_INFORMATION)pBuffer,
347                                      &lLength);
348
349                 break;
350             }
351
352             case FileStreamInformation:
353             {
354
355                 ntStatus = AFSQueryStreamInfo( Irp,
356                                                pCcb->DirectoryCB,
357                                                (FILE_STREAM_INFORMATION *)pBuffer,
358                                                &lLength);
359
360                 break;
361             }
362
363
364             case FileAttributeTagInformation:
365             {
366
367                 ntStatus = AFSQueryAttribTagInfo( Irp,
368                                                   pCcb->DirectoryCB,
369                                                   (FILE_ATTRIBUTE_TAG_INFORMATION *)pBuffer,
370                                                   &lLength);
371
372                 break;
373             }
374
375             case FileRemoteProtocolInformation:
376             {
377
378                     ntStatus = AFSQueryRemoteProtocolInfo( Irp,
379                                                            pCcb->DirectoryCB,
380                                                            (FILE_REMOTE_PROTOCOL_INFORMATION *)pBuffer,
381                                                            &lLength);
382
383                 break;
384             }
385
386         default:
387
388                 ntStatus = STATUS_INVALID_PARAMETER;
389
390                 break;
391         }
392
393 try_exit:
394
395         Irp->IoStatus.Information = pIrpSp->Parameters.QueryFile.Length - lLength;
396
397         if( bReleaseMain)
398         {
399
400             AFSReleaseResource( &pFcb->NPFcb->Resource);
401         }
402
403         if( !NT_SUCCESS( ntStatus) &&
404             ntStatus != STATUS_INVALID_PARAMETER &&
405             ntStatus != STATUS_BUFFER_OVERFLOW)
406         {
407
408             if( pCcb != NULL &&
409                 pCcb->DirectoryCB != NULL)
410             {
411
412                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
413                               AFS_TRACE_LEVEL_ERROR,
414                               "AFSQueryFileInfo Failed to process request for %wZ FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
415                               &pCcb->DirectoryCB->NameInformation.FileName,
416                               pCcb->DirectoryCB->ObjectInformation->FileId.Cell,
417                               pCcb->DirectoryCB->ObjectInformation->FileId.Volume,
418                               pCcb->DirectoryCB->ObjectInformation->FileId.Vnode,
419                               pCcb->DirectoryCB->ObjectInformation->FileId.Unique,
420                               ntStatus);
421             }
422         }
423     }
424     __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
425     {
426
427         AFSDbgLogMsg( 0,
428                       0,
429                       "EXCEPTION - AFSQueryFileInfo\n");
430
431         ntStatus = STATUS_UNSUCCESSFUL;
432
433         if( bReleaseMain)
434         {
435
436             AFSReleaseResource( &pFcb->NPFcb->Resource);
437         }
438     }
439
440     AFSCompleteRequest( Irp,
441                         ntStatus);
442
443     return ntStatus;
444 }
445
446 //
447 // Function: AFSSetFileInfo
448 //
449 // Description:
450 //
451 //      This function is the dispatch handler for the IRP_MJ_SET_FILE_INFORMATION request
452 //
453 // Return:
454 //
455 //      A status is returned for the function
456 //
457
458 NTSTATUS
459 AFSSetFileInfo( IN PDEVICE_OBJECT LibDeviceObject,
460                 IN PIRP Irp)
461 {
462
463     NTSTATUS ntStatus = STATUS_SUCCESS;
464     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
465     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
466     AFSFcb *pFcb = NULL;
467     AFSCcb *pCcb = NULL;
468     BOOLEAN bCompleteRequest = TRUE;
469     FILE_INFORMATION_CLASS FileInformationClass;
470     BOOLEAN bCanQueueRequest = FALSE;
471     PFILE_OBJECT pFileObject = NULL;
472     BOOLEAN bReleaseMain = FALSE;
473     BOOLEAN bUpdateFileInfo = FALSE;
474     AFSFileID stParentFileId;
475
476     __try
477     {
478
479         pFileObject = pIrpSp->FileObject;
480
481         pFcb = (AFSFcb *)pFileObject->FsContext;
482         pCcb = (AFSCcb *)pFileObject->FsContext2;
483
484         if( pFcb == NULL)
485         {
486
487             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
488                           AFS_TRACE_LEVEL_ERROR,
489                           "AFSSetFileInfo Attempted access (%08lX) when pFcb == NULL\n",
490                           Irp);
491
492             try_return( ntStatus = STATUS_INVALID_DEVICE_REQUEST);
493         }
494
495         bCanQueueRequest = !(IoIsOperationSynchronous( Irp) | (KeGetCurrentIrql() != PASSIVE_LEVEL));
496         FileInformationClass = pIrpSp->Parameters.SetFile.FileInformationClass;
497
498         //
499         // Grab the Fcb EXCL
500         //
501
502         AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
503                       AFS_TRACE_LEVEL_VERBOSE,
504                       "AFSSetFileInfo Acquiring Fcb lock %08lX EXCL %08lX\n",
505                       &pFcb->NPFcb->Resource,
506                       PsGetCurrentThread());
507
508         AFSAcquireExcl( &pFcb->NPFcb->Resource,
509                         TRUE);
510
511         bReleaseMain = TRUE;
512
513         //
514         // Don't allow requests against IOCtl nodes
515         //
516
517         if( pFcb->Header.NodeTypeCode == AFS_IOCTL_FCB)
518         {
519
520             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
521                           AFS_TRACE_LEVEL_ERROR,
522                           "AFSSetFileInfo Failing request against PIOCtl Fcb\n");
523
524             try_return( ntStatus = STATUS_INVALID_DEVICE_REQUEST);
525         }
526         else if( pFcb->Header.NodeTypeCode == AFS_SPECIAL_SHARE_FCB)
527         {
528
529             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
530                           AFS_TRACE_LEVEL_VERBOSE,
531                           "AFSSetFileInfo Processing request against SpecialShare Fcb\n");
532
533             ntStatus = AFSProcessShareSetInfo( Irp,
534                                                pFcb,
535                                                pCcb);
536
537             try_return( ntStatus);
538         }
539
540         if( BooleanFlagOn( pFcb->ObjectInformation->VolumeCB->VolumeInformation.Characteristics, FILE_READ_ONLY_DEVICE))
541         {
542
543             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
544                           AFS_TRACE_LEVEL_ERROR,
545                           "AFSSetFileInfo Request failed due to read only volume\n",
546                           Irp);
547
548             try_return( ntStatus = STATUS_ACCESS_DENIED);
549         }
550
551         //
552         // Ensure rename operations are synchronous
553         //
554
555         if( FileInformationClass == FileRenameInformation)
556         {
557
558             bCanQueueRequest = FALSE;
559         }
560
561         //
562         // Store away the parent fid
563         //
564
565         RtlZeroMemory( &stParentFileId,
566                        sizeof( AFSFileID));
567
568         if( pFcb->ObjectInformation->ParentObjectInformation != NULL)
569         {
570             stParentFileId = pFcb->ObjectInformation->ParentObjectInformation->FileId;
571         }
572
573         //
574         // Process the request
575         //
576
577         switch( FileInformationClass)
578         {
579
580             case FileBasicInformation:
581             {
582
583                 bUpdateFileInfo = TRUE;
584
585                 ntStatus = AFSSetBasicInfo( Irp,
586                                             pCcb->DirectoryCB);
587
588                 break;
589             }
590
591             case FileDispositionInformation:
592             {
593
594                 ntStatus = AFSSetDispositionInfo( Irp,
595                                                   pCcb->DirectoryCB);
596
597                 break;
598             }
599
600             case FileRenameInformation:
601             {
602
603                 ntStatus = AFSSetRenameInfo( Irp);
604
605                 break;
606             }
607
608             case FilePositionInformation:
609             {
610
611                 ntStatus = AFSSetPositionInfo( Irp,
612                                                pCcb->DirectoryCB);
613
614                 break;
615             }
616
617             case FileLinkInformation:
618             {
619
620                 ntStatus = STATUS_INVALID_DEVICE_REQUEST;
621
622                 break;
623             }
624
625             case FileAllocationInformation:
626             {
627
628                 ntStatus = AFSSetAllocationInfo( Irp,
629                                                  pCcb->DirectoryCB);
630
631                 break;
632             }
633
634             case FileEndOfFileInformation:
635             {
636
637                 ntStatus = AFSSetEndOfFileInfo( Irp,
638                                                 pCcb->DirectoryCB);
639
640                 break;
641             }
642
643             default:
644
645                 ntStatus = STATUS_INVALID_PARAMETER;
646
647                 break;
648         }
649
650 try_exit:
651
652         if( bReleaseMain)
653         {
654
655             AFSReleaseResource( &pFcb->NPFcb->Resource);
656         }
657
658         if( NT_SUCCESS( ntStatus) &&
659             bUpdateFileInfo)
660         {
661
662             ntStatus = AFSUpdateFileInformation( &stParentFileId,
663                                                  pFcb->ObjectInformation,
664                                                  &pFcb->AuthGroup);
665
666             if( !NT_SUCCESS( ntStatus))
667             {
668
669                 AFSAcquireExcl( &pFcb->NPFcb->Resource,
670                                 TRUE);
671
672                 //
673                 // Unwind the update and fail the request
674                 //
675
676                 AFSUnwindFileInfo( pFcb,
677                                    pCcb);
678
679                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
680                               AFS_TRACE_LEVEL_ERROR,
681                               "AFSSetFileInfo Failed to send file info update to service request for %wZ FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
682                               &pCcb->DirectoryCB->NameInformation.FileName,
683                               pCcb->DirectoryCB->ObjectInformation->FileId.Cell,
684                               pCcb->DirectoryCB->ObjectInformation->FileId.Volume,
685                               pCcb->DirectoryCB->ObjectInformation->FileId.Vnode,
686                               pCcb->DirectoryCB->ObjectInformation->FileId.Unique,
687                               ntStatus);
688
689                 AFSReleaseResource( &pFcb->NPFcb->Resource);
690             }
691         }
692
693         if( !NT_SUCCESS( ntStatus))
694         {
695
696             if( pCcb != NULL &&
697                 pCcb->DirectoryCB != NULL)
698             {
699
700                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
701                               AFS_TRACE_LEVEL_ERROR,
702                               "AFSSetFileInfo Failed to process request for %wZ FID %08lX-%08lX-%08lX-%08lX Status %08lX\n",
703                               &pCcb->DirectoryCB->NameInformation.FileName,
704                               pCcb->DirectoryCB->ObjectInformation->FileId.Cell,
705                               pCcb->DirectoryCB->ObjectInformation->FileId.Volume,
706                               pCcb->DirectoryCB->ObjectInformation->FileId.Vnode,
707                               pCcb->DirectoryCB->ObjectInformation->FileId.Unique,
708                               ntStatus);
709             }
710         }
711     }
712     __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
713     {
714
715         AFSDbgLogMsg( 0,
716                       0,
717                       "EXCEPTION - AFSSetFileInfo\n");
718
719         ntStatus = STATUS_UNSUCCESSFUL;
720     }
721
722     AFSCompleteRequest( Irp,
723                         ntStatus);
724
725     return ntStatus;
726 }
727
728 //
729 // Function: AFSQueryBasicInfo
730 //
731 // Description:
732 //
733 //      This function is the handler for the query basic information request
734 //
735 // Return:
736 //
737 //      A status is returned for the function
738 //
739
740 NTSTATUS
741 AFSQueryBasicInfo( IN PIRP Irp,
742                    IN AFSDirectoryCB *DirectoryCB,
743                    IN OUT PFILE_BASIC_INFORMATION Buffer,
744                    IN OUT PLONG Length)
745 {
746     NTSTATUS ntStatus = STATUS_SUCCESS;
747     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
748
749     if( *Length >= sizeof( FILE_BASIC_INFORMATION))
750     {
751
752         RtlZeroMemory( Buffer,
753                        *Length);
754
755         Buffer->CreationTime = DirectoryCB->ObjectInformation->CreationTime;
756         Buffer->LastAccessTime = DirectoryCB->ObjectInformation->LastAccessTime;
757         Buffer->LastWriteTime = DirectoryCB->ObjectInformation->LastWriteTime;
758         Buffer->ChangeTime = DirectoryCB->ObjectInformation->ChangeTime;
759         Buffer->FileAttributes = DirectoryCB->ObjectInformation->FileAttributes;
760
761         if( DirectoryCB->NameInformation.FileName.Buffer[ 0] == L'.' &&
762                  BooleanFlagOn( pDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_HIDE_DOT_NAMES))
763         {
764
765             if ( Buffer->FileAttributes != FILE_ATTRIBUTE_NORMAL)
766             {
767
768                 Buffer->FileAttributes |= FILE_ATTRIBUTE_HIDDEN;
769             }
770             else
771             {
772
773                 Buffer->FileAttributes = FILE_ATTRIBUTE_HIDDEN;
774             }
775         }
776
777         *Length -= sizeof( FILE_BASIC_INFORMATION);
778     }
779     else
780     {
781
782         ntStatus = STATUS_BUFFER_TOO_SMALL;
783     }
784
785     return ntStatus;
786 }
787
788 NTSTATUS
789 AFSQueryStandardInfo( IN PIRP Irp,
790                       IN AFSDirectoryCB *DirectoryCB,
791                       IN OUT PFILE_STANDARD_INFORMATION Buffer,
792                       IN OUT PLONG Length)
793 {
794
795     NTSTATUS ntStatus = STATUS_SUCCESS;
796     AFSFileInfoCB stFileInformation;
797     AFSCcb *pCcb = NULL;
798     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
799
800     if( *Length >= sizeof( FILE_STANDARD_INFORMATION))
801     {
802
803         pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
804
805         RtlZeroMemory( Buffer,
806                        *Length);
807
808         Buffer->NumberOfLinks = 1;
809         Buffer->DeletePending = BooleanFlagOn( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_PENDING_DELETE);
810
811         RtlZeroMemory( &stFileInformation,
812                        sizeof( AFSFileInfoCB));
813
814         Buffer->AllocationSize.QuadPart = (ULONGLONG)((DirectoryCB->ObjectInformation->AllocationSize.QuadPart/PAGE_SIZE) + 1) * PAGE_SIZE;
815
816         Buffer->EndOfFile = DirectoryCB->ObjectInformation->EndOfFile;
817
818         Buffer->Directory = BooleanFlagOn( DirectoryCB->ObjectInformation->FileAttributes, FILE_ATTRIBUTE_DIRECTORY);
819
820         *Length -= sizeof( FILE_STANDARD_INFORMATION);
821     }
822     else
823     {
824
825         ntStatus = STATUS_BUFFER_TOO_SMALL;
826     }
827
828     return ntStatus;
829 }
830
831 NTSTATUS
832 AFSQueryInternalInfo( IN PIRP Irp,
833                       IN AFSFcb *Fcb,
834                       IN OUT PFILE_INTERNAL_INFORMATION Buffer,
835                       IN OUT PLONG Length)
836 {
837
838     NTSTATUS ntStatus = STATUS_SUCCESS;
839
840     if( *Length >= sizeof( FILE_INTERNAL_INFORMATION))
841     {
842
843         Buffer->IndexNumber.HighPart = Fcb->ObjectInformation->FileId.Volume;
844
845         Buffer->IndexNumber.LowPart = Fcb->ObjectInformation->FileId.Vnode;
846
847         *Length -= sizeof( FILE_INTERNAL_INFORMATION);
848     }
849     else
850     {
851
852         ntStatus = STATUS_BUFFER_TOO_SMALL;
853     }
854
855     return ntStatus;
856 }
857
858 NTSTATUS
859 AFSQueryEaInfo( IN PIRP Irp,
860                 IN AFSDirectoryCB *DirectoryCB,
861                 IN OUT PFILE_EA_INFORMATION Buffer,
862                 IN OUT PLONG Length)
863 {
864
865     NTSTATUS ntStatus = STATUS_SUCCESS;
866
867     RtlZeroMemory( Buffer,
868                    *Length);
869
870     if( *Length >= sizeof( FILE_EA_INFORMATION))
871     {
872
873         Buffer->EaSize = 0;
874
875         *Length -= sizeof( FILE_EA_INFORMATION);
876     }
877     else
878     {
879
880         ntStatus = STATUS_BUFFER_TOO_SMALL;
881     }
882
883     return ntStatus;
884 }
885
886 NTSTATUS
887 AFSQueryPositionInfo( IN PIRP Irp,
888                       IN AFSFcb *Fcb,
889                       IN OUT PFILE_POSITION_INFORMATION Buffer,
890                       IN OUT PLONG Length)
891 {
892
893     NTSTATUS ntStatus = STATUS_SUCCESS;
894     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
895
896     if( *Length >= sizeof( FILE_POSITION_INFORMATION))
897     {
898
899         RtlZeroMemory( Buffer,
900                        *Length);
901
902         Buffer->CurrentByteOffset.QuadPart = pIrpSp->FileObject->CurrentByteOffset.QuadPart;
903
904         *Length -= sizeof( FILE_POSITION_INFORMATION);
905     }
906     else
907     {
908
909         ntStatus = STATUS_BUFFER_TOO_SMALL;
910     }
911
912     return ntStatus;
913 }
914
915 NTSTATUS
916 AFSQueryAccess( IN PIRP Irp,
917                 IN AFSFcb *Fcb,
918                 IN OUT PFILE_ACCESS_INFORMATION Buffer,
919                 IN OUT PLONG Length)
920 {
921
922     NTSTATUS ntStatus = STATUS_SUCCESS;
923
924     if( *Length >= sizeof( FILE_ACCESS_INFORMATION))
925     {
926
927         RtlZeroMemory( Buffer,
928                        *Length);
929
930         Buffer->AccessFlags = 0;
931
932         *Length -= sizeof( FILE_ACCESS_INFORMATION);
933     }
934     else
935     {
936
937         ntStatus = STATUS_BUFFER_TOO_SMALL;
938     }
939
940     return ntStatus;
941 }
942
943 NTSTATUS
944 AFSQueryMode( IN PIRP Irp,
945               IN AFSFcb *Fcb,
946               IN OUT PFILE_MODE_INFORMATION Buffer,
947               IN OUT PLONG Length)
948 {
949
950     NTSTATUS ntStatus = STATUS_SUCCESS;
951
952     if( *Length >= sizeof( FILE_MODE_INFORMATION))
953     {
954
955         RtlZeroMemory( Buffer,
956                        *Length);
957
958         Buffer->Mode = 0;
959
960         *Length -= sizeof( FILE_MODE_INFORMATION);
961     }
962     else
963     {
964
965         ntStatus = STATUS_BUFFER_TOO_SMALL;
966     }
967
968     return ntStatus;
969 }
970
971 NTSTATUS
972 AFSQueryAlignment( IN PIRP Irp,
973                    IN AFSFcb *Fcb,
974                    IN OUT PFILE_ALIGNMENT_INFORMATION Buffer,
975                    IN OUT PLONG Length)
976 {
977
978     NTSTATUS ntStatus = STATUS_SUCCESS;
979
980     if( *Length >= sizeof( FILE_ALIGNMENT_INFORMATION))
981     {
982
983         RtlZeroMemory( Buffer,
984                        *Length);
985
986         Buffer->AlignmentRequirement = 1;
987
988         *Length -= sizeof( FILE_ALIGNMENT_INFORMATION);
989     }
990     else
991     {
992
993         ntStatus = STATUS_BUFFER_TOO_SMALL;
994     }
995
996     return ntStatus;
997 }
998
999 NTSTATUS
1000 AFSQueryNameInfo( IN PIRP Irp,
1001                   IN AFSDirectoryCB *DirectoryCB,
1002                   IN OUT PFILE_NAME_INFORMATION Buffer,
1003                   IN OUT PLONG Length)
1004 {
1005
1006     NTSTATUS ntStatus = STATUS_SUCCESS;
1007     ULONG ulCopyLength = 0;
1008     ULONG cchCopied = 0;
1009     AFSFcb *pFcb = NULL;
1010     AFSCcb *pCcb = NULL;
1011     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1012     BOOLEAN bAddLeadingSlash = FALSE;
1013     BOOLEAN bAddTrailingSlash = FALSE;
1014     USHORT usFullNameLength = 0;
1015
1016     pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
1017
1018     pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
1019
1020     if( *Length >= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName))
1021     {
1022
1023         RtlZeroMemory( Buffer,
1024                        *Length);
1025
1026         if( pCcb->FullFileName.Length == 0 ||
1027             pCcb->FullFileName.Buffer[ 0] != L'\\')
1028         {
1029             bAddLeadingSlash = TRUE;
1030         }
1031
1032         if( pFcb->ObjectInformation->FileType == AFS_FILE_TYPE_DIRECTORY &&
1033             pCcb->FullFileName.Length > 0 &&
1034             pCcb->FullFileName.Buffer[ (pCcb->FullFileName.Length/sizeof( WCHAR)) - 1] != L'\\')
1035         {
1036             bAddTrailingSlash = TRUE;
1037         }
1038
1039         usFullNameLength = sizeof( WCHAR) +
1040                                     AFSServerName.Length +
1041                                     pCcb->FullFileName.Length;
1042
1043         if( bAddLeadingSlash)
1044         {
1045             usFullNameLength += sizeof( WCHAR);
1046         }
1047
1048         if( bAddTrailingSlash)
1049         {
1050             usFullNameLength += sizeof( WCHAR);
1051         }
1052
1053         if( *Length >= (LONG)(FIELD_OFFSET( FILE_NAME_INFORMATION, FileName) + (LONG)usFullNameLength))
1054         {
1055
1056             ulCopyLength = (LONG)usFullNameLength;
1057         }
1058         else
1059         {
1060
1061             ulCopyLength = *Length - FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1062
1063             ntStatus = STATUS_BUFFER_OVERFLOW;
1064         }
1065
1066         Buffer->FileNameLength = (ULONG)usFullNameLength;
1067
1068         *Length -= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1069
1070         if( ulCopyLength > 0)
1071         {
1072
1073             Buffer->FileName[ 0] = L'\\';
1074             ulCopyLength -= sizeof( WCHAR);
1075
1076             *Length -= sizeof( WCHAR);
1077             cchCopied += 1;
1078
1079             if( ulCopyLength >= AFSServerName.Length)
1080             {
1081
1082                 RtlCopyMemory( &Buffer->FileName[ 1],
1083                                AFSServerName.Buffer,
1084                                AFSServerName.Length);
1085
1086                 ulCopyLength -= AFSServerName.Length;
1087                 *Length -= AFSServerName.Length;
1088                 cchCopied += AFSServerName.Length/sizeof( WCHAR);
1089
1090                 if ( ulCopyLength > 0 &&
1091                      bAddLeadingSlash)
1092                 {
1093
1094                     Buffer->FileName[ cchCopied] = L'\\';
1095
1096                     ulCopyLength -= sizeof( WCHAR);
1097                     *Length -= sizeof( WCHAR);
1098                     cchCopied++;
1099                 }
1100
1101                 if( ulCopyLength >= pCcb->FullFileName.Length)
1102                 {
1103
1104                     RtlCopyMemory( &Buffer->FileName[ cchCopied],
1105                                    pCcb->FullFileName.Buffer,
1106                                    pCcb->FullFileName.Length);
1107
1108                     ulCopyLength -= pCcb->FullFileName.Length;
1109                     *Length -= pCcb->FullFileName.Length;
1110                     cchCopied += pCcb->FullFileName.Length/sizeof( WCHAR);
1111
1112                     if( ulCopyLength > 0 &&
1113                         bAddTrailingSlash)
1114                     {
1115                         Buffer->FileName[ cchCopied] = L'\\';
1116
1117                         *Length -= sizeof( WCHAR);
1118                     }
1119                 }
1120                 else
1121                 {
1122
1123                     RtlCopyMemory( &Buffer->FileName[ cchCopied],
1124                                    pCcb->FullFileName.Buffer,
1125                                    ulCopyLength);
1126
1127                     *Length -= ulCopyLength;
1128                 }
1129             }
1130         }
1131     }
1132     else
1133     {
1134
1135         ntStatus = STATUS_BUFFER_TOO_SMALL;
1136     }
1137
1138     return ntStatus;
1139 }
1140
1141 NTSTATUS
1142 AFSQueryShortNameInfo( IN PIRP Irp,
1143                        IN AFSDirectoryCB *DirectoryCB,
1144                        IN OUT PFILE_NAME_INFORMATION Buffer,
1145                        IN OUT PLONG Length)
1146 {
1147
1148     NTSTATUS ntStatus = STATUS_BUFFER_TOO_SMALL;
1149     ULONG ulCopyLength = 0;
1150
1151     RtlZeroMemory( Buffer,
1152                    *Length);
1153
1154     if( DirectoryCB->NameInformation.ShortNameLength == 0)
1155     {
1156
1157         //
1158         // The short name IS the long name
1159         //
1160
1161         if( *Length >= (LONG)FIELD_OFFSET( FILE_NAME_INFORMATION, FileName))
1162         {
1163
1164             if( *Length >= (LONG)(FIELD_OFFSET( FILE_NAME_INFORMATION, FileName) + (LONG)DirectoryCB->NameInformation.FileName.Length))
1165             {
1166
1167                 ulCopyLength = (LONG)DirectoryCB->NameInformation.FileName.Length;
1168
1169                 ntStatus = STATUS_SUCCESS;
1170             }
1171             else
1172             {
1173
1174                 ulCopyLength = *Length - FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1175
1176                 ntStatus = STATUS_BUFFER_OVERFLOW;
1177             }
1178
1179             Buffer->FileNameLength = DirectoryCB->NameInformation.FileName.Length;
1180
1181             *Length -= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1182
1183             if( ulCopyLength > 0)
1184             {
1185
1186                 RtlCopyMemory( Buffer->FileName,
1187                                DirectoryCB->NameInformation.FileName.Buffer,
1188                                ulCopyLength);
1189
1190                 *Length -= ulCopyLength;
1191             }
1192         }
1193     }
1194     else
1195     {
1196
1197         if( *Length >= (LONG)FIELD_OFFSET( FILE_NAME_INFORMATION, FileName))
1198         {
1199
1200             if( *Length >= (LONG)(FIELD_OFFSET( FILE_NAME_INFORMATION, FileName) + (LONG)DirectoryCB->NameInformation.FileName.Length))
1201             {
1202
1203                 ulCopyLength = (LONG)DirectoryCB->NameInformation.ShortNameLength;
1204
1205                 ntStatus = STATUS_SUCCESS;
1206             }
1207             else
1208             {
1209
1210                 ulCopyLength = *Length - FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1211
1212                 ntStatus = STATUS_BUFFER_OVERFLOW;
1213             }
1214
1215             Buffer->FileNameLength = DirectoryCB->NameInformation.ShortNameLength;
1216
1217             *Length -= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
1218
1219             if( ulCopyLength > 0)
1220             {
1221
1222                 RtlCopyMemory( Buffer->FileName,
1223                                DirectoryCB->NameInformation.ShortName,
1224                                Buffer->FileNameLength);
1225
1226                 *Length -= ulCopyLength;
1227             }
1228         }
1229     }
1230
1231     return ntStatus;
1232 }
1233
1234 NTSTATUS
1235 AFSQueryNetworkInfo( IN PIRP Irp,
1236                      IN AFSDirectoryCB *DirectoryCB,
1237                      IN OUT PFILE_NETWORK_OPEN_INFORMATION Buffer,
1238                      IN OUT PLONG Length)
1239 {
1240
1241     NTSTATUS ntStatus = STATUS_SUCCESS;
1242     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1243
1244     RtlZeroMemory( Buffer,
1245                    *Length);
1246
1247     if( *Length >= sizeof( FILE_NETWORK_OPEN_INFORMATION))
1248     {
1249
1250         Buffer->CreationTime.QuadPart = DirectoryCB->ObjectInformation->CreationTime.QuadPart;
1251         Buffer->LastAccessTime.QuadPart = DirectoryCB->ObjectInformation->LastAccessTime.QuadPart;
1252         Buffer->LastWriteTime.QuadPart = DirectoryCB->ObjectInformation->LastWriteTime.QuadPart;
1253         Buffer->ChangeTime.QuadPart = DirectoryCB->ObjectInformation->ChangeTime.QuadPart;
1254
1255         Buffer->AllocationSize.QuadPart = DirectoryCB->ObjectInformation->AllocationSize.QuadPart;
1256         Buffer->EndOfFile.QuadPart = DirectoryCB->ObjectInformation->EndOfFile.QuadPart;
1257
1258         Buffer->FileAttributes = DirectoryCB->ObjectInformation->FileAttributes;
1259
1260         if( DirectoryCB->NameInformation.FileName.Buffer[ 0] == L'.' &&
1261             BooleanFlagOn( pDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_HIDE_DOT_NAMES))
1262         {
1263
1264             if ( Buffer->FileAttributes != FILE_ATTRIBUTE_NORMAL)
1265             {
1266
1267                 Buffer->FileAttributes |= FILE_ATTRIBUTE_HIDDEN;
1268             }
1269             else
1270             {
1271
1272                 Buffer->FileAttributes = FILE_ATTRIBUTE_HIDDEN;
1273             }
1274         }
1275
1276         *Length -= sizeof( FILE_NETWORK_OPEN_INFORMATION);
1277     }
1278     else
1279     {
1280
1281         ntStatus = STATUS_BUFFER_TOO_SMALL;
1282     }
1283
1284     return ntStatus;
1285 }
1286
1287 NTSTATUS
1288 AFSQueryStreamInfo( IN PIRP Irp,
1289                     IN AFSDirectoryCB *DirectoryCB,
1290                     IN OUT FILE_STREAM_INFORMATION *Buffer,
1291                     IN OUT PLONG Length)
1292 {
1293
1294     NTSTATUS ntStatus = STATUS_BUFFER_TOO_SMALL;
1295     ULONG ulCopyLength = 0;
1296     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1297
1298     if( *Length >= FIELD_OFFSET( FILE_STREAM_INFORMATION, StreamName))
1299     {
1300
1301         RtlZeroMemory( Buffer,
1302                        *Length);
1303
1304         Buffer->NextEntryOffset = 0;
1305
1306
1307         if( !BooleanFlagOn( DirectoryCB->ObjectInformation->FileAttributes, FILE_ATTRIBUTE_DIRECTORY))
1308         {
1309
1310             if( *Length >= (LONG)(FIELD_OFFSET( FILE_STREAM_INFORMATION, StreamName) + 14))  // ::$DATA
1311             {
1312
1313                 ulCopyLength = 14;
1314
1315                 ntStatus = STATUS_SUCCESS;
1316             }
1317             else
1318             {
1319
1320                 ulCopyLength = *Length - FIELD_OFFSET( FILE_STREAM_INFORMATION, StreamName);
1321
1322                 ntStatus = STATUS_BUFFER_OVERFLOW;
1323             }
1324
1325             Buffer->StreamNameLength = 14; // ::$DATA
1326
1327             Buffer->StreamSize.QuadPart = DirectoryCB->ObjectInformation->EndOfFile.QuadPart;
1328
1329             Buffer->StreamAllocationSize.QuadPart = DirectoryCB->ObjectInformation->AllocationSize.QuadPart;
1330
1331             *Length -= FIELD_OFFSET( FILE_STREAM_INFORMATION, StreamName);
1332
1333             if( ulCopyLength > 0)
1334             {
1335
1336                 RtlCopyMemory( Buffer->StreamName,
1337                                L"::$DATA",
1338                                ulCopyLength);
1339
1340                 *Length -= ulCopyLength;
1341             }
1342         }
1343         else
1344         {
1345
1346             Buffer->StreamNameLength = 0;       // No stream for a directory
1347
1348             // The response size is zero
1349
1350             ntStatus = STATUS_SUCCESS;
1351         }
1352     }
1353
1354     return ntStatus;
1355 }
1356
1357 NTSTATUS
1358 AFSQueryAttribTagInfo( IN PIRP Irp,
1359                        IN AFSDirectoryCB *DirectoryCB,
1360                        IN OUT FILE_ATTRIBUTE_TAG_INFORMATION *Buffer,
1361                        IN OUT PLONG Length)
1362 {
1363
1364     NTSTATUS ntStatus = STATUS_BUFFER_TOO_SMALL;
1365     ULONG ulCopyLength = 0;
1366     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1367
1368     if( *Length >= sizeof( FILE_ATTRIBUTE_TAG_INFORMATION))
1369     {
1370
1371         RtlZeroMemory( Buffer,
1372                        *Length);
1373
1374         Buffer->FileAttributes = DirectoryCB->ObjectInformation->FileAttributes;
1375
1376         if( DirectoryCB->NameInformation.FileName.Buffer[ 0] == L'.' &&
1377             BooleanFlagOn( pDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_HIDE_DOT_NAMES))
1378         {
1379
1380             if ( Buffer->FileAttributes != FILE_ATTRIBUTE_NORMAL)
1381             {
1382
1383                 Buffer->FileAttributes |= FILE_ATTRIBUTE_HIDDEN;
1384             }
1385             else
1386             {
1387
1388                 Buffer->FileAttributes = FILE_ATTRIBUTE_HIDDEN;
1389             }
1390         }
1391
1392         if( BooleanFlagOn( DirectoryCB->ObjectInformation->FileAttributes, FILE_ATTRIBUTE_REPARSE_POINT))
1393         {
1394             Buffer->ReparseTag = IO_REPARSE_TAG_OPENAFS_DFS;
1395         }
1396
1397         *Length -= sizeof( FILE_ATTRIBUTE_TAG_INFORMATION);
1398
1399         ntStatus = STATUS_SUCCESS;
1400     }
1401
1402     return ntStatus;
1403 }
1404
1405 NTSTATUS
1406 AFSQueryRemoteProtocolInfo( IN PIRP Irp,
1407                             IN AFSDirectoryCB *DirectoryCB,
1408                             IN OUT FILE_REMOTE_PROTOCOL_INFORMATION *Buffer,
1409                             IN OUT PLONG Length)
1410 {
1411
1412     NTSTATUS ntStatus = STATUS_BUFFER_TOO_SMALL;
1413     ULONG ulCopyLength = 0;
1414     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
1415
1416     if( *Length >= sizeof( FILE_REMOTE_PROTOCOL_INFORMATION))
1417     {
1418
1419         RtlZeroMemory( Buffer,
1420                        *Length);
1421
1422         Buffer->StructureVersion = 1;
1423
1424         Buffer->StructureSize = sizeof(FILE_REMOTE_PROTOCOL_INFORMATION);
1425
1426         Buffer->Protocol = WNNC_NET_OPENAFS;
1427
1428         Buffer->ProtocolMajorVersion = 3;
1429
1430         Buffer->ProtocolMinorVersion = 0;
1431
1432         Buffer->ProtocolRevision = 0;
1433
1434         *Length -= sizeof( FILE_REMOTE_PROTOCOL_INFORMATION);
1435
1436         ntStatus = STATUS_SUCCESS;
1437     }
1438
1439     return ntStatus;
1440 }
1441
1442 NTSTATUS
1443 AFSSetBasicInfo( IN PIRP Irp,
1444                  IN AFSDirectoryCB *DirectoryCB)
1445 {
1446     NTSTATUS ntStatus = STATUS_SUCCESS;
1447     PFILE_BASIC_INFORMATION pBuffer;
1448     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1449     ULONG ulNotifyFilter = 0;
1450     AFSCcb *pCcb = NULL;
1451
1452     __Enter
1453     {
1454
1455         pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
1456
1457         pBuffer = (PFILE_BASIC_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
1458
1459         pCcb->FileUnwindInfo.FileAttributes = (ULONG)-1;
1460
1461         if( pBuffer->FileAttributes != (ULONGLONG)0)
1462         {
1463
1464             if( DirectoryCB->ObjectInformation->Fcb->Header.NodeTypeCode == AFS_FILE_FCB &&
1465                 BooleanFlagOn( pBuffer->FileAttributes, FILE_ATTRIBUTE_DIRECTORY))
1466             {
1467
1468                 try_return( ntStatus = STATUS_INVALID_PARAMETER);
1469             }
1470
1471             if( DirectoryCB->ObjectInformation->Fcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB)
1472             {
1473
1474                 pBuffer->FileAttributes |= FILE_ATTRIBUTE_DIRECTORY;
1475             }
1476
1477             if( DirectoryCB->ObjectInformation->Fcb->Header.NodeTypeCode == AFS_SYMBOLIC_LINK_FCB ||
1478                 DirectoryCB->ObjectInformation->Fcb->Header.NodeTypeCode == AFS_MOUNT_POINT_FCB ||
1479                 DirectoryCB->ObjectInformation->Fcb->Header.NodeTypeCode == AFS_DFS_LINK_FCB)
1480             {
1481                 pBuffer->FileAttributes |= FILE_ATTRIBUTE_REPARSE_POINT;
1482             }
1483
1484             pCcb->FileUnwindInfo.FileAttributes = DirectoryCB->ObjectInformation->FileAttributes;
1485
1486             DirectoryCB->ObjectInformation->FileAttributes = pBuffer->FileAttributes;
1487
1488             ulNotifyFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1489
1490             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED);
1491         }
1492
1493         pCcb->FileUnwindInfo.CreationTime.QuadPart = (ULONGLONG)-1;
1494
1495         if( pBuffer->CreationTime.QuadPart != (ULONGLONG)-1 &&
1496             pBuffer->CreationTime.QuadPart != (ULONGLONG)0)
1497         {
1498
1499             pCcb->FileUnwindInfo.CreationTime.QuadPart = DirectoryCB->ObjectInformation->CreationTime.QuadPart;
1500
1501             DirectoryCB->ObjectInformation->CreationTime.QuadPart = pBuffer->CreationTime.QuadPart;
1502
1503             ulNotifyFilter |= FILE_NOTIFY_CHANGE_CREATION;
1504
1505             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CREATE_TIME);
1506         }
1507
1508         pCcb->FileUnwindInfo.LastAccessTime.QuadPart = (ULONGLONG)-1;
1509
1510         if( pBuffer->LastAccessTime.QuadPart != (ULONGLONG)-1 &&
1511             pBuffer->LastAccessTime.QuadPart != (ULONGLONG)0)
1512         {
1513
1514             pCcb->FileUnwindInfo.LastAccessTime.QuadPart = DirectoryCB->ObjectInformation->LastAccessTime.QuadPart;
1515
1516             DirectoryCB->ObjectInformation->LastAccessTime.QuadPart = pBuffer->LastAccessTime.QuadPart;
1517
1518             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_ACCESS;
1519
1520             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_ACCESS_TIME);
1521         }
1522
1523         pCcb->FileUnwindInfo.LastWriteTime.QuadPart = (ULONGLONG)-1;
1524
1525         if( pBuffer->LastWriteTime.QuadPart != (ULONGLONG)-1 &&
1526             pBuffer->LastWriteTime.QuadPart != (ULONGLONG)0)
1527         {
1528
1529             pCcb->FileUnwindInfo.LastWriteTime.QuadPart = DirectoryCB->ObjectInformation->LastWriteTime.QuadPart;
1530
1531             DirectoryCB->ObjectInformation->LastWriteTime.QuadPart = pBuffer->LastWriteTime.QuadPart;
1532
1533             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_WRITE;
1534
1535             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_LAST_WRITE_TIME);
1536         }
1537
1538         pCcb->FileUnwindInfo.ChangeTime.QuadPart = (ULONGLONG)-1;
1539
1540         if( pBuffer->ChangeTime.QuadPart != (ULONGLONG)-1 &&
1541             pBuffer->ChangeTime.QuadPart != (ULONGLONG)0)
1542         {
1543
1544             pCcb->FileUnwindInfo.ChangeTime.QuadPart = DirectoryCB->ObjectInformation->ChangeTime.QuadPart;
1545
1546             DirectoryCB->ObjectInformation->ChangeTime.QuadPart = pBuffer->ChangeTime.QuadPart;
1547
1548             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_ACCESS;
1549
1550             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
1551         }
1552
1553         if( ulNotifyFilter > 0)
1554         {
1555
1556             if( DirectoryCB->ObjectInformation->ParentObjectInformation != NULL)
1557             {
1558
1559                 AFSFsRtlNotifyFullReportChange( DirectoryCB->ObjectInformation->ParentObjectInformation,
1560                                                 pCcb,
1561                                                 (ULONG)ulNotifyFilter,
1562                                                 (ULONG)FILE_ACTION_MODIFIED);
1563             }
1564         }
1565
1566 try_exit:
1567
1568         NOTHING;
1569     }
1570
1571     return ntStatus;
1572 }
1573
1574 NTSTATUS
1575 AFSSetDispositionInfo( IN PIRP Irp,
1576                        IN AFSDirectoryCB *DirectoryCB)
1577 {
1578     NTSTATUS ntStatus = STATUS_SUCCESS;
1579     PFILE_DISPOSITION_INFORMATION pBuffer;
1580     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1581     AFSFcb *pFcb = NULL;
1582     AFSCcb *pCcb = NULL;
1583
1584     __Enter
1585     {
1586
1587         pBuffer = (PFILE_DISPOSITION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
1588
1589         pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
1590
1591         pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
1592
1593         //
1594         // Can't delete the root
1595         //
1596
1597         if( pFcb->Header.NodeTypeCode == AFS_ROOT_FCB)
1598         {
1599
1600             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1601                           AFS_TRACE_LEVEL_ERROR,
1602                           "AFSSetDispositionInfo Attempt to delete root entry\n");
1603
1604             try_return( ntStatus = STATUS_CANNOT_DELETE);
1605         }
1606
1607         //
1608         // If the file is read only then do not allow the delete
1609         //
1610
1611         if( BooleanFlagOn( DirectoryCB->ObjectInformation->FileAttributes, FILE_ATTRIBUTE_READONLY))
1612         {
1613
1614             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1615                           AFS_TRACE_LEVEL_ERROR,
1616                           "AFSSetDispositionInfo Attempt to delete read only entry %wZ\n",
1617                           &DirectoryCB->NameInformation.FileName);
1618
1619             try_return( ntStatus = STATUS_CANNOT_DELETE);
1620         }
1621
1622         if( pBuffer->DeleteFile)
1623         {
1624
1625             //
1626             // Check if the caller can delete the file
1627             //
1628
1629             ntStatus = AFSNotifyDelete( DirectoryCB,
1630                                         TRUE);
1631
1632             if( !NT_SUCCESS( ntStatus))
1633             {
1634
1635                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1636                               AFS_TRACE_LEVEL_ERROR,
1637                               "AFSSetDispositionInfo Cannot delete entry %wZ Status %08lX\n",
1638                               &DirectoryCB->NameInformation.FileName,
1639                               ntStatus);
1640
1641                 try_return( ntStatus);
1642             }
1643
1644             if( pFcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB)
1645             {
1646
1647                 //
1648                 // Check if this is a directory that there are not currently other opens
1649                 //
1650
1651                 if( pFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount > 0)
1652                 {
1653
1654                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1655                                   AFS_TRACE_LEVEL_ERROR,
1656                                   "AFSSetDispositionInfo Attempt to delete directory %wZ with open %u handles\n",
1657                                   &DirectoryCB->NameInformation.FileName,
1658                                   pFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount);
1659
1660                     try_return( ntStatus = STATUS_DIRECTORY_NOT_EMPTY);
1661                 }
1662
1663                 if( !AFSIsDirectoryEmptyForDelete( pFcb))
1664                 {
1665
1666                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1667                                   AFS_TRACE_LEVEL_ERROR,
1668                                   "AFSSetDispositionInfo Attempt to delete non-empty directory %wZ\n",
1669                                   &DirectoryCB->NameInformation.FileName);
1670
1671                     try_return( ntStatus = STATUS_DIRECTORY_NOT_EMPTY);
1672                 }
1673             }
1674             else
1675             {
1676
1677                 //
1678                 // Attempt to flush any outstanding data
1679                 //
1680
1681                 if( !MmFlushImageSection( &pFcb->NPFcb->SectionObjectPointers,
1682                                           MmFlushForDelete))
1683                 {
1684
1685                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1686                                   AFS_TRACE_LEVEL_ERROR,
1687                                   "AFSSetDispositionInfo Failed to flush image section for delete Entry %wZ\n",
1688                                   &DirectoryCB->NameInformation.FileName);
1689
1690                     try_return( ntStatus = STATUS_CANNOT_DELETE);
1691                 }
1692
1693                 //
1694                 // Purge the cache as well
1695                 //
1696
1697                 if( pFcb->NPFcb->SectionObjectPointers.DataSectionObject != NULL)
1698                 {
1699
1700                     CcPurgeCacheSection( &pFcb->NPFcb->SectionObjectPointers,
1701                                          NULL,
1702                                          0,
1703                                          TRUE);
1704                 }
1705             }
1706
1707             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1708                           AFS_TRACE_LEVEL_VERBOSE,
1709                           "AFSSetDispositionInfo Setting PENDING_DELETE on DirEntry  %p Name %wZ\n",
1710                           DirectoryCB,
1711                           &DirectoryCB->NameInformation.FileName);
1712
1713             SetFlag( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_PENDING_DELETE);
1714         }
1715         else
1716         {
1717
1718             ClearFlag( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_PENDING_DELETE);
1719         }
1720
1721         //
1722         // OK, should be good to go, set the flag in the file object
1723         //
1724
1725         pIrpSp->FileObject->DeletePending = pBuffer->DeleteFile;
1726
1727 try_exit:
1728
1729         NOTHING;
1730     }
1731
1732     return ntStatus;
1733 }
1734
1735 NTSTATUS
1736 AFSSetRenameInfo( IN PIRP Irp)
1737 {
1738
1739     NTSTATUS ntStatus = STATUS_SUCCESS;
1740     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1741     IO_STATUS_BLOCK stIoSb = {0,0};
1742     AFSFcb *pSrcFcb = NULL, *pTargetDcb = NULL, *pTargetFcb = NULL;
1743     AFSCcb *pSrcCcb = NULL, *pTargetDirCcb = NULL;
1744     PFILE_OBJECT pSrcFileObj = pIrpSp->FileObject;
1745     PFILE_OBJECT pTargetFileObj = pIrpSp->Parameters.SetFile.FileObject;
1746     PFILE_RENAME_INFORMATION pRenameInfo = NULL;
1747     UNICODE_STRING uniTargetName, uniSourceName;
1748     BOOLEAN bReplaceIfExists = FALSE;
1749     UNICODE_STRING uniShortName;
1750     AFSDirectoryCB *pTargetDirEntry = NULL;
1751     ULONG ulTargetCRC = 0;
1752     BOOLEAN bTargetEntryExists = FALSE;
1753     AFSObjectInfoCB *pSrcObject = NULL, *pTargetObject = NULL;
1754     AFSObjectInfoCB *pSrcParentObject = NULL, *pTargetParentObject = NULL;
1755     AFSFileID stNewFid, stTmpTargetFid;
1756     UNICODE_STRING uniTmpTargetName;
1757     BOOLEAN bReplaceTmpTargetEntry = FALSE;
1758     ULONG ulNotificationAction = 0, ulNotifyFilter = 0;
1759     UNICODE_STRING uniFullTargetPath;
1760     BOOLEAN bCommonParent = FALSE;
1761
1762     __Enter
1763     {
1764
1765         bReplaceIfExists = pIrpSp->Parameters.SetFile.ReplaceIfExists;
1766
1767         pSrcFcb = (AFSFcb *)pSrcFileObj->FsContext;
1768         pSrcCcb = (AFSCcb *)pSrcFileObj->FsContext2;
1769
1770         pSrcObject = pSrcFcb->ObjectInformation;
1771
1772         uniTmpTargetName.Length = 0;
1773         uniTmpTargetName.MaximumLength = 0;
1774         uniTmpTargetName.Buffer = NULL;
1775
1776         //
1777         // Perform some basic checks to ensure FS integrity
1778         //
1779
1780         if( pSrcFcb->Header.NodeTypeCode == AFS_ROOT_FCB)
1781         {
1782
1783             //
1784             // Can't rename the root directory
1785             //
1786
1787             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1788                           AFS_TRACE_LEVEL_ERROR,
1789                           "AFSSetRenameInfo Attempt to rename root entry\n");
1790
1791             try_return( ntStatus = STATUS_INVALID_PARAMETER);
1792         }
1793
1794         if( pSrcFcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB)
1795         {
1796
1797             //
1798             // If there are any open children then fail the rename
1799             //
1800
1801             if( pSrcFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount > 0)
1802             {
1803
1804                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1805                               AFS_TRACE_LEVEL_ERROR,
1806                               "AFSSetRenameInfo Attempt to rename directory with open children %wZ\n",
1807                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
1808
1809                 try_return( ntStatus = STATUS_ACCESS_DENIED);
1810             }
1811         }
1812         else
1813         {
1814
1815             if( pSrcFcb->OpenHandleCount > 1)
1816             {
1817
1818                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1819                               AFS_TRACE_LEVEL_ERROR,
1820                               "AFSSetRenameInfo Attempt to rename directory with open references %wZ\n",
1821                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
1822
1823                 try_return( ntStatus = STATUS_ACCESS_DENIED);
1824             }
1825         }
1826
1827         //
1828         // Resolve the target fileobject
1829         //
1830
1831         if( pTargetFileObj == NULL)
1832         {
1833
1834             //
1835             // This is a simple rename. Here the target directory is the same as the source parent directory
1836             // and the name is retrieved from the system buffer information
1837             //
1838
1839             pRenameInfo = (PFILE_RENAME_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
1840
1841             pTargetParentObject = pSrcFcb->ObjectInformation->ParentObjectInformation;
1842
1843             pTargetDcb = pTargetParentObject->Fcb;
1844
1845             uniTargetName.Length = (USHORT)pRenameInfo->FileNameLength;
1846             uniTargetName.Buffer = (PWSTR)&pRenameInfo->FileName;
1847         }
1848         else
1849         {
1850
1851             //
1852             // So here we have the target directory taken from the targetfile object
1853             //
1854
1855             pTargetDcb = (AFSFcb *)pTargetFileObj->FsContext;
1856
1857             pTargetDirCcb = (AFSCcb *)pTargetFileObj->FsContext2;
1858
1859             pTargetParentObject = (AFSObjectInfoCB *)pTargetDcb->ObjectInformation;
1860
1861             //
1862             // Grab the target name which we setup in the IRP_MJ_CREATE handler. By how we set this up
1863             // it is only the target component of the rename operation
1864             //
1865
1866             uniTargetName = *((PUNICODE_STRING)&pTargetFileObj->FileName);
1867         }
1868
1869         //
1870         // We do not allow cross-volume renames to occur
1871         //
1872
1873         if( pTargetParentObject->VolumeCB != pSrcObject->VolumeCB)
1874         {
1875
1876             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1877                           AFS_TRACE_LEVEL_ERROR,
1878                           "AFSSetRenameInfo Attempt to rename directory to different volume %wZ\n",
1879                           &pSrcCcb->DirectoryCB->NameInformation.FileName);
1880
1881             try_return( ntStatus = STATUS_NOT_SAME_DEVICE);
1882         }
1883
1884         //
1885         // If the target exists be sure the ReplaceIfExists flag is set
1886         //
1887
1888         AFSAcquireShared( pTargetParentObject->VolumeCB->VolumeLock,
1889                           TRUE);
1890
1891         ulTargetCRC = AFSGenerateCRC( &uniTargetName,
1892                                       FALSE);
1893
1894         AFSAcquireShared( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.TreeLock,
1895                           TRUE);
1896
1897         AFSLocateCaseSensitiveDirEntry( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
1898                                         ulTargetCRC,
1899                                         &pTargetDirEntry);
1900
1901         if( pTargetDirEntry == NULL)
1902         {
1903
1904             //
1905             // Missed so perform a case insensitive lookup
1906             //
1907
1908             ulTargetCRC = AFSGenerateCRC( &uniTargetName,
1909                                           TRUE);
1910
1911             AFSLocateCaseInsensitiveDirEntry( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
1912                                               ulTargetCRC,
1913                                               &pTargetDirEntry);
1914         }
1915
1916         //
1917         // Increment our ref count on the dir entry
1918         //
1919
1920         if( pTargetDirEntry != NULL)
1921         {
1922             InterlockedIncrement( &pTargetDirEntry->OpenReferenceCount);
1923         }
1924
1925         AFSReleaseResource( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.TreeLock);
1926
1927         if( pTargetDirEntry != NULL)
1928         {
1929
1930             if( !bReplaceIfExists)
1931             {
1932
1933                 AFSReleaseResource( pTargetParentObject->VolumeCB->VolumeLock);
1934
1935                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1936                               AFS_TRACE_LEVEL_ERROR,
1937                               "AFSSetRenameInfo Attempt to rename directory with target collision %wZ Target %wZ\n",
1938                               &pSrcCcb->DirectoryCB->NameInformation.FileName,
1939                               &pTargetDirEntry->NameInformation.FileName);
1940
1941                 try_return( ntStatus = STATUS_OBJECT_NAME_COLLISION);
1942             }
1943
1944             bTargetEntryExists = TRUE;
1945         }
1946
1947         AFSReleaseResource( pTargetParentObject->VolumeCB->VolumeLock);
1948
1949         //
1950         // Extract off the final component name from the Fcb
1951         //
1952
1953         uniSourceName.Length = (USHORT)pSrcCcb->DirectoryCB->NameInformation.FileName.Length;
1954         uniSourceName.MaximumLength = uniSourceName.Length;
1955
1956         uniSourceName.Buffer = pSrcCcb->DirectoryCB->NameInformation.FileName.Buffer;
1957
1958         //
1959         // The quick check to see if they are not really performing a rename
1960         // Do the names match? Only do this where the parent directories are
1961         // the same
1962         //
1963
1964         if( pTargetParentObject == pSrcFcb->ObjectInformation->ParentObjectInformation)
1965         {
1966
1967             bCommonParent = TRUE;
1968
1969             if( FsRtlAreNamesEqual( &uniTargetName,
1970                                     &uniSourceName,
1971                                     TRUE,
1972                                     NULL))
1973             {
1974
1975                 //
1976                 // Check for case only rename
1977                 //
1978
1979                 if( !FsRtlAreNamesEqual( &uniTargetName,
1980                                          &uniSourceName,
1981                                          FALSE,
1982                                          NULL))
1983                 {
1984
1985                     //
1986                     // Just move in the new case form of the name
1987                     //
1988
1989                     RtlCopyMemory( pSrcCcb->DirectoryCB->NameInformation.FileName.Buffer,
1990                                    uniTargetName.Buffer,
1991                                    uniTargetName.Length);
1992                 }
1993
1994                 try_return( ntStatus = STATUS_SUCCESS);
1995             }
1996         }
1997         else
1998         {
1999
2000             bCommonParent = FALSE;
2001         }
2002
2003         //
2004         // If the target name exists then we need to 'move' the target before
2005         // sending the rename to the service
2006         //
2007
2008         if( bReplaceIfExists &&
2009             pTargetDirEntry != NULL)
2010         {
2011
2012             //
2013             // What we will do is temporarily rename the file to a tmp file
2014             // so we can back out if anything fails below
2015             // First thing is to remove the original target from the parent
2016             //
2017
2018             AFSAcquireExcl( pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
2019                             TRUE);
2020
2021             AFSRemoveDirNodeFromParent( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2022                                         pTargetDirEntry,
2023                                         TRUE);
2024
2025             AFSReleaseResource( pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
2026
2027             pTargetDirEntry->FileIndex = (ULONG)InterlockedIncrement( &pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2028
2029             uniTmpTargetName.Length = 0;
2030             uniTmpTargetName.MaximumLength = uniTargetName.Length + (4 * sizeof( WCHAR));
2031
2032             uniTmpTargetName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2033                                                                          uniTmpTargetName.MaximumLength,
2034                                                                          AFS_GENERIC_MEMORY_11_TAG);
2035
2036             if( uniTmpTargetName.Buffer == NULL)
2037             {
2038
2039                 //
2040                 // Re-insert the entry
2041                 //
2042
2043                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2044                                         pTargetDirEntry,
2045                                         TRUE);
2046
2047                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2048                               AFS_TRACE_LEVEL_ERROR,
2049                               "AFSSetRenameInfo Failed tmp buffer allocation during rename of %wZ\n",
2050                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
2051
2052                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2053             }
2054
2055             RtlZeroMemory( uniTmpTargetName.Buffer,
2056                            uniTmpTargetName.MaximumLength);
2057
2058             uniTmpTargetName.Length = uniTargetName.Length;
2059
2060             RtlCopyMemory( uniTmpTargetName.Buffer,
2061                            uniTargetName.Buffer,
2062                            uniTmpTargetName.Length);
2063
2064             RtlCopyMemory( &uniTmpTargetName.Buffer[ uniTmpTargetName.Length/sizeof( WCHAR)],
2065                            L".tmp",
2066                            4 * sizeof( WCHAR));
2067
2068             uniTmpTargetName.Length += (4 * sizeof( WCHAR));
2069
2070             ntStatus = AFSNotifyRename( pTargetDirEntry->ObjectInformation,
2071                                         pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2072                                         pTargetDcb->ObjectInformation,
2073                                         pTargetDirEntry,
2074                                         &uniTmpTargetName,
2075                                         &stTmpTargetFid);
2076
2077             if( !NT_SUCCESS( ntStatus))
2078             {
2079
2080                 //
2081                 // Re-insert the entry
2082                 //
2083
2084                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2085                                         pTargetDirEntry,
2086                                         TRUE);
2087
2088                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2089                               AFS_TRACE_LEVEL_ERROR,
2090                               "AFSSetRenameInfo Failed rename of %wZ to tmp %wZ Status %08lX\n",
2091                               &pSrcCcb->DirectoryCB->NameInformation.FileName,
2092                               &uniTmpTargetName,
2093                               ntStatus);
2094
2095                 try_return( ntStatus);
2096             }
2097
2098             //
2099             // Indicate we need to replace this entry if any failure occurs below
2100             //
2101
2102             bReplaceTmpTargetEntry = TRUE;
2103         }
2104
2105         //
2106         // We need to remove the DirEntry from the parent node, update the index
2107         // and reinsert it into the parent tree. Note that for entries with the
2108         // same parent we do not pull the node from the enumeration list
2109         //
2110
2111         AFSAcquireExcl( pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
2112                         TRUE);
2113
2114         AFSRemoveDirNodeFromParent( pSrcFcb->ObjectInformation->ParentObjectInformation,
2115                                     pSrcCcb->DirectoryCB,
2116                                     !bCommonParent);
2117
2118         AFSReleaseResource( pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
2119
2120         if( !bCommonParent)
2121         {
2122
2123             //
2124             // We always need to update the FileIndex since this entry will be put at the 'end'
2125             // of the enumeraiton list. If we don't it will cause recursion ... We do this
2126             // here to cover any failures which might occur below
2127             //
2128
2129             pSrcCcb->DirectoryCB->FileIndex =
2130                             (ULONG)InterlockedIncrement( &pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2131         }
2132
2133         //
2134         // OK, this is a simple rename. Issue the rename
2135         // request to the service.
2136         //
2137
2138         ntStatus = AFSNotifyRename( pSrcFcb->ObjectInformation,
2139                                     pSrcFcb->ObjectInformation->ParentObjectInformation,
2140                                     pTargetDcb->ObjectInformation,
2141                                     pSrcCcb->DirectoryCB,
2142                                     &uniTargetName,
2143                                     &stNewFid);
2144
2145         if( !NT_SUCCESS( ntStatus))
2146         {
2147
2148             //
2149             // Attempt to re-insert the directory entry
2150             //
2151
2152             AFSInsertDirectoryNode( pSrcFcb->ObjectInformation->ParentObjectInformation,
2153                                     pSrcCcb->DirectoryCB,
2154                                     !bCommonParent);
2155
2156             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2157                           AFS_TRACE_LEVEL_ERROR,
2158                           "AFSSetRenameInfo Failed rename of %wZ to target %wZ Status %08lX\n",
2159                           &pSrcCcb->DirectoryCB->NameInformation.FileName,
2160                           &uniTargetName,
2161                           ntStatus);
2162
2163             try_return( ntStatus);
2164         }
2165
2166         //
2167         // Set the notification up for the source file
2168         //
2169
2170         if( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation == pTargetParentObject &&
2171             !bReplaceTmpTargetEntry)
2172         {
2173
2174             ulNotificationAction = FILE_ACTION_RENAMED_OLD_NAME;
2175         }
2176         else
2177         {
2178
2179             ulNotificationAction = FILE_ACTION_REMOVED;
2180         }
2181
2182         if( pSrcCcb->DirectoryCB->ObjectInformation->FileType == AFS_FILE_TYPE_DIRECTORY)
2183         {
2184
2185             ulNotifyFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
2186         }
2187         else
2188         {
2189
2190             ulNotifyFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
2191         }
2192
2193         AFSFsRtlNotifyFullReportChange( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation,
2194                                         pSrcCcb,
2195                                         (ULONG)ulNotifyFilter,
2196                                         (ULONG)ulNotificationAction);
2197
2198         //
2199         // Update the name in the dir entry.
2200         //
2201
2202         ntStatus = AFSUpdateDirEntryName( pSrcCcb->DirectoryCB,
2203                                           &uniTargetName);
2204
2205         if( !NT_SUCCESS( ntStatus))
2206         {
2207
2208             //
2209             // Attempt to re-insert the directory entry
2210             //
2211
2212             AFSInsertDirectoryNode( pSrcFcb->ObjectInformation->ParentObjectInformation,
2213                                     pSrcCcb->DirectoryCB,
2214                                     !bCommonParent);
2215
2216             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2217                           AFS_TRACE_LEVEL_ERROR,
2218                           "AFSSetRenameInfo Failed update of dir entry %wZ to target %wZ Status %08lX\n",
2219                           &pSrcCcb->DirectoryCB->NameInformation.FileName,
2220                           &uniTargetName,
2221                           ntStatus);
2222
2223             try_return( ntStatus);
2224         }
2225
2226         //
2227         // Update the object information block, if needed
2228         //
2229
2230         if( !AFSIsEqualFID( &pSrcObject->FileId,
2231                             &stNewFid))
2232         {
2233
2234             //
2235             // Remove the old information entry
2236             //
2237
2238             AFSAcquireExcl( pSrcObject->VolumeCB->ObjectInfoTree.TreeLock,
2239                             TRUE);
2240
2241             AFSRemoveHashEntry( &pSrcObject->VolumeCB->ObjectInfoTree.TreeHead,
2242                                 &pSrcObject->TreeEntry);
2243
2244             RtlCopyMemory( &pSrcObject->FileId,
2245                            &stNewFid,
2246                            sizeof( AFSFileID));
2247
2248             //
2249             // Insert the entry into the new object table.
2250             //
2251
2252             pSrcObject->TreeEntry.HashIndex = AFSCreateLowIndex( &pSrcObject->FileId);
2253
2254             if( pSrcObject->VolumeCB->ObjectInfoTree.TreeHead == NULL)
2255             {
2256
2257                 pSrcObject->VolumeCB->ObjectInfoTree.TreeHead = &pSrcObject->TreeEntry;
2258             }
2259             else
2260             {
2261
2262                 AFSInsertHashEntry( pSrcObject->VolumeCB->ObjectInfoTree.TreeHead,
2263                                     &pSrcObject->TreeEntry);
2264             }
2265
2266             AFSReleaseResource( pSrcObject->VolumeCB->ObjectInfoTree.TreeLock);
2267         }
2268
2269         //
2270         // Update the hash values for the name trees.
2271         //
2272
2273         pSrcCcb->DirectoryCB->CaseSensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pSrcCcb->DirectoryCB->NameInformation.FileName,
2274                                                                                  FALSE);
2275
2276         pSrcCcb->DirectoryCB->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pSrcCcb->DirectoryCB->NameInformation.FileName,
2277                                                                                    TRUE);
2278
2279         if( pSrcCcb->DirectoryCB->NameInformation.ShortNameLength > 0)
2280         {
2281
2282             uniShortName.Length = pSrcCcb->DirectoryCB->NameInformation.ShortNameLength;
2283             uniShortName.Buffer = pSrcCcb->DirectoryCB->NameInformation.ShortName;
2284
2285             pSrcCcb->DirectoryCB->Type.Data.ShortNameTreeEntry.HashIndex = AFSGenerateCRC( &uniShortName,
2286                                                                                            TRUE);
2287         }
2288         else
2289         {
2290
2291             pSrcCcb->DirectoryCB->Type.Data.ShortNameTreeEntry.HashIndex = 0;
2292         }
2293
2294         if( !bCommonParent)
2295         {
2296
2297             //
2298             // Update the file index for the object in the new parent
2299             //
2300
2301             pSrcCcb->DirectoryCB->FileIndex = (ULONG)InterlockedIncrement( &pTargetParentObject->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2302         }
2303
2304         //
2305         // Re-insert the directory entry
2306         //
2307
2308         AFSInsertDirectoryNode( pTargetParentObject,
2309                                 pSrcCcb->DirectoryCB,
2310                                 !bCommonParent);
2311
2312         //
2313         // Update the parent pointer in the source object if they are different
2314         //
2315
2316         if( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation != pTargetParentObject)
2317         {
2318
2319             InterlockedDecrement( &pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation->Specific.Directory.ChildOpenHandleCount);
2320
2321             InterlockedDecrement( &pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation->Specific.Directory.ChildOpenReferenceCount);
2322
2323             InterlockedIncrement( &pTargetParentObject->Specific.Directory.ChildOpenHandleCount);
2324
2325             InterlockedIncrement( &pTargetParentObject->Specific.Directory.ChildOpenReferenceCount);
2326
2327             pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation = pTargetParentObject;
2328
2329             ulNotificationAction = FILE_ACTION_ADDED;
2330         }
2331         else
2332         {
2333
2334             ulNotificationAction = FILE_ACTION_RENAMED_NEW_NAME;
2335         }
2336
2337         //
2338         // Now update the notification for the target file
2339         //
2340
2341         AFSFsRtlNotifyFullReportChange( pTargetParentObject->ParentObjectInformation,
2342                                         pSrcCcb,
2343                                         (ULONG)ulNotifyFilter,
2344                                         (ULONG)ulNotificationAction);
2345
2346         //
2347         // If we performed the rename of the target because it existed, we now need to
2348         // delete the tmp target we created above
2349         //
2350
2351         if( bReplaceTmpTargetEntry)
2352         {
2353
2354             RtlCopyMemory( &pTargetDirEntry->ObjectInformation->FileId,
2355                            &stTmpTargetFid,
2356                            sizeof( AFSFileID));
2357
2358             //
2359             // Update the name in the dir entry
2360             //
2361
2362             ntStatus = AFSUpdateDirEntryName( pTargetDirEntry,
2363                                               &uniTmpTargetName);
2364
2365             if( !NT_SUCCESS( ntStatus))
2366             {
2367
2368                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2369                               AFS_TRACE_LEVEL_ERROR,
2370                               "AFSSetRenameInfo Failed update of target dir entry %wZ to tmp %wZ Status %08lX\n",
2371                               &pTargetDirEntry->NameInformation.FileName,
2372                               &uniTmpTargetName,
2373                               ntStatus);
2374
2375                 try_return( ntStatus);
2376             }
2377
2378             ntStatus = AFSNotifyDelete( pTargetDirEntry,
2379                                         FALSE);
2380
2381             if( !NT_SUCCESS( ntStatus))
2382             {
2383
2384                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2385                               AFS_TRACE_LEVEL_VERBOSE,
2386                               "AFSSetRenameInfo object deletion failure dir entry %p name %wZ to tmp %wZ\n",
2387                               pTargetDirEntry,
2388                               &pTargetDirEntry->NameInformation.FileName,
2389                               &uniTmpTargetName);
2390
2391                 try_return( ntStatus);
2392             }
2393
2394             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2395                           AFS_TRACE_LEVEL_VERBOSE,
2396                           "AFSSetRenameInfo Setting DELETE flag in dir entry %p name %wZ to tmp %wZ\n",
2397                           pTargetDirEntry,
2398                           &pTargetDirEntry->NameInformation.FileName,
2399                           &uniTmpTargetName);
2400
2401             SetFlag( pTargetDirEntry->Flags, AFS_DIR_ENTRY_DELETED);
2402
2403             //
2404             // Try and purge the cache map if this is a file
2405             //
2406
2407             if( pTargetDirEntry->ObjectInformation->FileType == AFS_FILE_TYPE_FILE &&
2408                 pTargetDirEntry->ObjectInformation->Fcb != NULL &&
2409                 pTargetDirEntry->OpenReferenceCount > 1)
2410             {
2411
2412                 pTargetFcb = pTargetDirEntry->ObjectInformation->Fcb;
2413
2414                 AFSAcquireExcl( &pTargetFcb->NPFcb->Resource,
2415                                 TRUE);
2416
2417                 //
2418                 // Try and flush the cache map
2419                 //
2420
2421                 if( !MmFlushImageSection( &pTargetFcb->NPFcb->SectionObjectPointers,
2422                                           MmFlushForDelete))
2423                 {
2424
2425                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2426                                   AFS_TRACE_LEVEL_ERROR,
2427                                   "AFSSetRenameInfo Failed to flush section for deleted temp file %wZ\n",
2428                                   &pTargetDirEntry->NameInformation.FileName);
2429                 }
2430
2431                 AFSReleaseResource( &pTargetFcb->NPFcb->Resource);
2432             }
2433
2434             ASSERT( pTargetDirEntry->OpenReferenceCount > 0);
2435
2436             InterlockedDecrement( &pTargetDirEntry->OpenReferenceCount);
2437
2438             if( pTargetDirEntry->OpenReferenceCount == 0)
2439             {
2440
2441                 SetFlag( pTargetDirEntry->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED);
2442
2443                 ASSERT( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE));
2444
2445                 //
2446                 // Free up the name buffer if it was reallocated
2447                 //
2448
2449                 if( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_RELEASE_NAME_BUFFER))
2450                 {
2451
2452                     AFSExFreePool( pTargetDirEntry->NameInformation.FileName.Buffer);
2453                 }
2454
2455                 if( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER))
2456                 {
2457
2458                     AFSExFreePool( pTargetDirEntry->NameInformation.TargetName.Buffer);
2459                 }
2460
2461                 //
2462                 // Dereference the object for this dir entry
2463                 //
2464
2465                 ASSERT( pTargetDirEntry->ObjectInformation->ObjectReferenceCount > 0);
2466
2467                 InterlockedDecrement( &pTargetDirEntry->ObjectInformation->ObjectReferenceCount);
2468
2469                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2470                               AFS_TRACE_LEVEL_VERBOSE,
2471                               "AFSSetRenameInfo Decrement3 count on object %08lX Cnt %d\n",
2472                               pTargetDirEntry->ObjectInformation,
2473                               pTargetDirEntry->ObjectInformation->ObjectReferenceCount);
2474
2475                 //
2476                 // Free up the dir entry
2477                 //
2478
2479                 ExDeleteResourceLite( &pTargetDirEntry->NonPaged->Lock);
2480
2481                 AFSExFreePool( pTargetDirEntry->NonPaged);
2482
2483                 AFSExFreePool( pTargetDirEntry);
2484             }
2485
2486             pTargetDirEntry = NULL;
2487         }
2488
2489 try_exit:
2490
2491
2492         if( !NT_SUCCESS( ntStatus))
2493         {
2494
2495             if( bReplaceTmpTargetEntry)
2496             {
2497
2498                 AFSNotifyRename( pTargetDirEntry->ObjectInformation,
2499                                  pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2500                                  pTargetDcb->ObjectInformation,
2501                                  pTargetDirEntry,
2502                                  &uniTargetName,
2503                                  &stTmpTargetFid);
2504
2505                 //
2506                 // Replace the target entry
2507                 //
2508
2509                 AFSAcquireExcl( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeLock,
2510                                 TRUE);
2511
2512                 if( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead == NULL)
2513                 {
2514
2515                     pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead = &pTargetDirEntry->ObjectInformation->TreeEntry;
2516                 }
2517                 else
2518                 {
2519                     AFSInsertHashEntry( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead,
2520                                         &pTargetDirEntry->ObjectInformation->TreeEntry);
2521                 }
2522
2523                 AFSReleaseResource( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeLock);
2524
2525                 //
2526                 // We always need to update the FileIndex since this entry will be put at the 'end'
2527                 // of the enumeraiton list. If we don't it will cause recursion ...
2528                 //
2529
2530                 pTargetDirEntry->FileIndex = (ULONG)InterlockedIncrement( &pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2531
2532                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2533                                         pTargetDirEntry,
2534                                         TRUE);
2535             }
2536         }
2537
2538         if( pTargetDirEntry != NULL)
2539         {
2540
2541             InterlockedDecrement( &pTargetDirEntry->OpenReferenceCount);
2542         }
2543
2544         if( uniTmpTargetName.Buffer != NULL)
2545         {
2546
2547             AFSExFreePool( uniTmpTargetName.Buffer);
2548         }
2549     }
2550
2551     return ntStatus;
2552 }
2553
2554 NTSTATUS
2555 AFSSetPositionInfo( IN PIRP Irp,
2556                     IN AFSDirectoryCB *DirectoryCB)
2557 {
2558     NTSTATUS ntStatus = STATUS_SUCCESS;
2559     PFILE_POSITION_INFORMATION pBuffer;
2560     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2561
2562     pBuffer = (PFILE_POSITION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2563
2564     pIrpSp->FileObject->CurrentByteOffset.QuadPart = pBuffer->CurrentByteOffset.QuadPart;
2565
2566     return ntStatus;
2567 }
2568
2569 NTSTATUS
2570 AFSSetAllocationInfo( IN PIRP Irp,
2571                       IN AFSDirectoryCB *DirectoryCB)
2572 {
2573     NTSTATUS ntStatus = STATUS_SUCCESS;
2574     PFILE_ALLOCATION_INFORMATION pBuffer;
2575     BOOLEAN bReleasePaging = FALSE;
2576     BOOLEAN bTellCc = FALSE;
2577     BOOLEAN bTellService = FALSE;
2578     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2579     PFILE_OBJECT pFileObject = pIrpSp->FileObject;
2580     AFSFcb *pFcb = NULL;
2581     AFSCcb *pCcb = NULL;
2582     LARGE_INTEGER liSaveAlloc;
2583     LARGE_INTEGER liSaveFileSize;
2584     LARGE_INTEGER liSaveVDL;
2585
2586     pBuffer = (PFILE_ALLOCATION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2587
2588     pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2589
2590     pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2591
2592     //
2593     // save values to put back
2594     //
2595     liSaveAlloc = pFcb->Header.AllocationSize;
2596     liSaveFileSize = pFcb->Header.FileSize;
2597     liSaveVDL = pFcb->Header.ValidDataLength;
2598
2599     if( pFcb->Header.AllocationSize.QuadPart == pBuffer->AllocationSize.QuadPart ||
2600         pIrpSp->Parameters.SetFile.AdvanceOnly)
2601     {
2602         return STATUS_SUCCESS ;
2603     }
2604
2605     if( pFcb->Header.AllocationSize.QuadPart > pBuffer->AllocationSize.QuadPart)
2606     {
2607         //
2608         // Truncating the file
2609         //
2610         if( !MmCanFileBeTruncated( pFileObject->SectionObjectPointer,
2611                                    &pBuffer->AllocationSize))
2612         {
2613
2614             ntStatus = STATUS_USER_MAPPED_FILE ;
2615         }
2616         else
2617         {
2618             //
2619             // If this is a truncation we need to grab the paging IO resource.
2620             //
2621             AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2622                           AFS_TRACE_LEVEL_VERBOSE,
2623                           "AFSSetAllocationInfo Acquiring Fcb PagingIo lock %08lX EXCL %08lX\n",
2624                           &pFcb->NPFcb->PagingResource,
2625                           PsGetCurrentThread());
2626
2627             AFSAcquireExcl( &pFcb->NPFcb->PagingResource,
2628                             TRUE);
2629
2630             bReleasePaging = TRUE;
2631
2632
2633             pFcb->Header.AllocationSize = pBuffer->AllocationSize;
2634
2635             pFcb->ObjectInformation->AllocationSize = pBuffer->AllocationSize;
2636
2637             //
2638             // Tell Cc that Allocation is moved.
2639             //
2640             bTellCc = TRUE;
2641
2642             if( pFcb->Header.FileSize.QuadPart > pBuffer->AllocationSize.QuadPart)
2643             {
2644                 //
2645                 // We are pulling the EOF back as well so we need to tell
2646                 // the service.
2647                 //
2648                 bTellService = TRUE;
2649
2650                 pFcb->Header.FileSize = pBuffer->AllocationSize;
2651
2652                 pFcb->ObjectInformation->EndOfFile = pBuffer->AllocationSize;
2653             }
2654
2655         }
2656     }
2657     else
2658     {
2659         //
2660         // Tell Cc if allocation is increased.
2661         //
2662         bTellCc = pBuffer->AllocationSize.QuadPart > pFcb->Header.AllocationSize.QuadPart;
2663
2664         pFcb->Header.AllocationSize = pBuffer->AllocationSize;
2665
2666         pFcb->ObjectInformation->AllocationSize = pBuffer->AllocationSize;
2667     }
2668
2669     //
2670     // Now Tell the server if we have to
2671     //
2672     if (bTellService)
2673     {
2674         ntStatus = AFSUpdateFileInformation( &pFcb->ObjectInformation->ParentObjectInformation->FileId,
2675                                              pFcb->ObjectInformation,
2676                                              &pFcb->AuthGroup);
2677     }
2678
2679     if (NT_SUCCESS(ntStatus))
2680     {
2681         //
2682         // Trim extents if we told the service - the update has done an implicit
2683         // trim at the service.
2684         //
2685         if (bTellService)
2686         {
2687             AFSTrimExtents( pFcb,
2688                             &pFcb->Header.FileSize);
2689         }
2690
2691         KeQuerySystemTime( &pFcb->ObjectInformation->ChangeTime);
2692
2693         SetFlag( pFcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
2694
2695         if (bTellCc &&
2696             CcIsFileCached( pFileObject))
2697         {
2698             CcSetFileSizes( pFileObject,
2699                             (PCC_FILE_SIZES)&pFcb->Header.AllocationSize);
2700         }
2701     }
2702     else
2703     {
2704         //
2705         // Put the saved values back
2706         //
2707         pFcb->Header.ValidDataLength = liSaveVDL;
2708         pFcb->Header.FileSize = liSaveFileSize;
2709         pFcb->Header.AllocationSize = liSaveAlloc;
2710         pFcb->ObjectInformation->EndOfFile = liSaveFileSize;
2711         pFcb->ObjectInformation->AllocationSize = liSaveAlloc;
2712     }
2713
2714     if( bReleasePaging)
2715     {
2716
2717         AFSReleaseResource( &pFcb->NPFcb->PagingResource);
2718     }
2719
2720     return ntStatus;
2721 }
2722
2723 NTSTATUS
2724 AFSSetEndOfFileInfo( IN PIRP Irp,
2725                      IN AFSDirectoryCB *DirectoryCB)
2726 {
2727     NTSTATUS ntStatus = STATUS_SUCCESS;
2728     PFILE_END_OF_FILE_INFORMATION pBuffer;
2729     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2730     PFILE_OBJECT pFileObject = pIrpSp->FileObject;
2731     LARGE_INTEGER liSaveSize;
2732     LARGE_INTEGER liSaveVDL;
2733     LARGE_INTEGER liSaveAlloc;
2734     BOOLEAN bModified = FALSE;
2735     BOOLEAN bReleasePaging = FALSE;
2736     BOOLEAN bTruncated = FALSE;
2737     AFSFcb *pFcb = NULL;
2738     AFSCcb *pCcb = NULL;
2739
2740     pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2741
2742     pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2743
2744     pBuffer = (PFILE_END_OF_FILE_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2745
2746     liSaveSize = pFcb->Header.FileSize;
2747     liSaveAlloc = pFcb->Header.AllocationSize;
2748     liSaveVDL = pFcb->Header.ValidDataLength;
2749
2750     if( pFcb->Header.FileSize.QuadPart != pBuffer->EndOfFile.QuadPart &&
2751         !pIrpSp->Parameters.SetFile.AdvanceOnly)
2752     {
2753
2754         if( pBuffer->EndOfFile.QuadPart < pFcb->Header.FileSize.QuadPart)
2755         {
2756
2757             // Truncating the file
2758             if( !MmCanFileBeTruncated( pFileObject->SectionObjectPointer,
2759                                        &pBuffer->EndOfFile))
2760             {
2761
2762                 ntStatus = STATUS_USER_MAPPED_FILE;
2763             }
2764             else
2765             {
2766                 //
2767                 // If this is a truncation we need to grab the paging
2768                 // IO resource.
2769                 //
2770                 AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2771                               AFS_TRACE_LEVEL_VERBOSE,
2772                               "AFSSetAllocationInfo Acquiring Fcb PagingIo lock %08lX EXCL %08lX\n",
2773                               &pFcb->NPFcb->PagingResource,
2774                               PsGetCurrentThread());
2775
2776                 AFSAcquireExcl( &pFcb->NPFcb->PagingResource,
2777                                 TRUE);
2778
2779                 bReleasePaging = TRUE;
2780
2781                 pFcb->Header.AllocationSize = pBuffer->EndOfFile;
2782
2783                 pFcb->Header.FileSize = pBuffer->EndOfFile;
2784
2785                 pFcb->ObjectInformation->EndOfFile = pBuffer->EndOfFile;
2786
2787                 pFcb->ObjectInformation->AllocationSize = pBuffer->EndOfFile;
2788
2789                 if( pFcb->Header.ValidDataLength.QuadPart > pFcb->Header.FileSize.QuadPart)
2790                 {
2791
2792                     pFcb->Header.ValidDataLength = pFcb->Header.FileSize;
2793                 }
2794
2795                 bTruncated = TRUE;
2796
2797                 bModified = TRUE;
2798             }
2799         }
2800         else
2801         {
2802             //
2803             // extending the file, move EOF
2804             //
2805
2806             pFcb->Header.FileSize = pBuffer->EndOfFile;
2807
2808             pFcb->ObjectInformation->EndOfFile = pBuffer->EndOfFile;
2809
2810             if (pFcb->Header.FileSize.QuadPart > pFcb->Header.AllocationSize.QuadPart)
2811             {
2812                 //
2813                 // And Allocation as needed.
2814                 //
2815                 pFcb->Header.AllocationSize = pBuffer->EndOfFile;
2816
2817                 pFcb->ObjectInformation->AllocationSize = pBuffer->EndOfFile;
2818             }
2819
2820             bModified = TRUE;
2821         }
2822     }
2823
2824     if (bModified)
2825     {
2826
2827         KeQuerySystemTime( &pFcb->ObjectInformation->ChangeTime);
2828
2829         SetFlag( pFcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
2830
2831         //
2832         // Tell the server
2833         //
2834
2835         ntStatus = AFSUpdateFileInformation( &pFcb->ObjectInformation->ParentObjectInformation->FileId,
2836                                              pFcb->ObjectInformation,
2837                                              &pFcb->AuthGroup);
2838
2839         if( NT_SUCCESS(ntStatus))
2840         {
2841             //
2842             // We are now good to go so tell CC.
2843             //
2844             CcSetFileSizes( pFileObject,
2845                             (PCC_FILE_SIZES)&pFcb->Header.AllocationSize);
2846
2847             //
2848             // And give up those extents
2849             //
2850             if( bTruncated)
2851             {
2852
2853                 AFSTrimExtents( pFcb,
2854                                 &pFcb->Header.FileSize);
2855             }
2856         }
2857         else
2858         {
2859             pFcb->Header.ValidDataLength = liSaveVDL;
2860             pFcb->Header.FileSize = liSaveSize;
2861             pFcb->Header.AllocationSize = liSaveAlloc;
2862             pFcb->ObjectInformation->EndOfFile = liSaveSize;
2863             pFcb->ObjectInformation->AllocationSize = liSaveAlloc;
2864         }
2865     }
2866
2867     if( bReleasePaging)
2868     {
2869
2870         AFSReleaseResource( &pFcb->NPFcb->PagingResource);
2871     }
2872
2873     return ntStatus;
2874 }
2875
2876 NTSTATUS
2877 AFSProcessShareSetInfo( IN IRP *Irp,
2878                         IN AFSFcb *Fcb,
2879                         IN AFSCcb *Ccb)
2880 {
2881
2882     NTSTATUS ntStatus = STATUS_SUCCESS;
2883     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2884     ULONG ulOutputBufferLen = 0, ulInputBufferLen;
2885     FILE_INFORMATION_CLASS ulFileInformationClass;
2886     void *pPipeInfo = NULL;
2887
2888     __Enter
2889     {
2890         ulFileInformationClass = pIrpSp->Parameters.SetFile.FileInformationClass;
2891
2892         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2893                       AFS_TRACE_LEVEL_VERBOSE,
2894                       "AFSProcessShareSetInfo On pipe %wZ Class %08lX\n",
2895                       &Ccb->DirectoryCB->NameInformation.FileName,
2896                       ulFileInformationClass);
2897
2898         pPipeInfo = AFSLockSystemBuffer( Irp,
2899                                          pIrpSp->Parameters.SetFile.Length);
2900
2901         if( pPipeInfo == NULL)
2902         {
2903
2904             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2905                           AFS_TRACE_LEVEL_ERROR,
2906                           "AFSProcessShareSetInfo Failed to lock buffer on pipe %wZ\n",
2907                           &Ccb->DirectoryCB->NameInformation.FileName);
2908
2909             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2910         }
2911
2912         //
2913         // Send the request to the service
2914         //
2915
2916         ntStatus = AFSNotifySetPipeInfo( Ccb,
2917                                          (ULONG)ulFileInformationClass,
2918                                          pIrpSp->Parameters.SetFile.Length,
2919                                          pPipeInfo);
2920
2921         if( !NT_SUCCESS( ntStatus))
2922         {
2923
2924             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2925                           AFS_TRACE_LEVEL_ERROR,
2926                           "AFSProcessShareSetInfo Failed to send request to service on pipe %wZ Status %08lX\n",
2927                           &Ccb->DirectoryCB->NameInformation.FileName,
2928                           ntStatus);
2929
2930             try_return( ntStatus);
2931         }
2932
2933         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2934                       AFS_TRACE_LEVEL_VERBOSE,
2935                       "AFSProcessShareSetInfo Completed request on pipe %wZ Class %08lX\n",
2936                       &Ccb->DirectoryCB->NameInformation.FileName,
2937                       ulFileInformationClass);
2938
2939 try_exit:
2940
2941         NOTHING;
2942     }
2943
2944     return ntStatus;
2945 }
2946
2947 NTSTATUS
2948 AFSProcessShareQueryInfo( IN IRP *Irp,
2949                           IN AFSFcb *Fcb,
2950                           IN AFSCcb *Ccb)
2951 {
2952
2953     NTSTATUS ntStatus = STATUS_SUCCESS;
2954     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2955     ULONG ulOutputBufferLen = 0, ulInputBufferLen;
2956     FILE_INFORMATION_CLASS ulFileInformationClass;
2957     void *pPipeInfo = NULL;
2958
2959     __Enter
2960     {
2961
2962         ulFileInformationClass = pIrpSp->Parameters.QueryFile.FileInformationClass;
2963
2964         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2965                       AFS_TRACE_LEVEL_VERBOSE,
2966                       "AFSProcessShareQueryInfo On pipe %wZ Class %08lX\n",
2967                       &Ccb->DirectoryCB->NameInformation.FileName,
2968                       ulFileInformationClass);
2969
2970         pPipeInfo = AFSLockSystemBuffer( Irp,
2971                                          pIrpSp->Parameters.QueryFile.Length);
2972
2973         if( pPipeInfo == NULL)
2974         {
2975
2976             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2977                           AFS_TRACE_LEVEL_ERROR,
2978                           "AFSProcessShareQueryInfo Failed to lock buffer on pipe %wZ\n",
2979                           &Ccb->DirectoryCB->NameInformation.FileName);
2980
2981             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2982         }
2983
2984         //
2985         // Send the request to the service
2986         //
2987
2988         ntStatus = AFSNotifyQueryPipeInfo( Ccb,
2989                                            (ULONG)ulFileInformationClass,
2990                                            pIrpSp->Parameters.QueryFile.Length,
2991                                            pPipeInfo,
2992                                            (ULONG *)&Irp->IoStatus.Information);
2993
2994         if( !NT_SUCCESS( ntStatus))
2995         {
2996
2997             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2998                           AFS_TRACE_LEVEL_ERROR,
2999                           "AFSProcessShareQueryInfo Failed to send request to service on pipe %wZ Status %08lX\n",
3000                           &Ccb->DirectoryCB->NameInformation.FileName,
3001                           ntStatus);
3002
3003             try_return( ntStatus);
3004         }
3005
3006         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
3007                       AFS_TRACE_LEVEL_VERBOSE,
3008                       "AFSProcessShareQueryInfo Completed request on pipe %wZ Class %08lX\n",
3009                       &Ccb->DirectoryCB->NameInformation.FileName,
3010                       ulFileInformationClass);
3011
3012 try_exit:
3013
3014         NOTHING;
3015     }
3016
3017     return ntStatus;
3018 }
3019
3020 NTSTATUS
3021 AFSProcessPIOCtlQueryInfo( IN IRP *Irp,
3022                            IN AFSFcb *Fcb,
3023                            IN AFSCcb *Ccb,
3024                            IN OUT LONG *Length)
3025 {
3026
3027     NTSTATUS ntStatus = STATUS_SUCCESS;
3028     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
3029     FILE_INFORMATION_CLASS ulFileInformationClass;
3030
3031     __Enter
3032     {
3033
3034         ulFileInformationClass = pIrpSp->Parameters.QueryFile.FileInformationClass;
3035
3036         switch( ulFileInformationClass)
3037         {
3038
3039             case FileBasicInformation:
3040             {
3041
3042                 if ( *Length >= sizeof( FILE_BASIC_INFORMATION))
3043                 {
3044                     PFILE_BASIC_INFORMATION pBasic = (PFILE_BASIC_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3045
3046                     pBasic->CreationTime.QuadPart = 0;
3047                     pBasic->LastAccessTime.QuadPart = 0;
3048                     pBasic->ChangeTime.QuadPart = 0;
3049                     pBasic->LastWriteTime.QuadPart = 0;
3050                     pBasic->FileAttributes = FILE_ATTRIBUTE_SYSTEM;
3051
3052                     *Length -= sizeof( FILE_BASIC_INFORMATION);
3053                 }
3054                 else
3055                 {
3056                     ntStatus = STATUS_BUFFER_OVERFLOW;
3057                 }
3058
3059                 break;
3060             }
3061
3062             case FileStandardInformation:
3063             {
3064
3065                 if ( *Length >= sizeof( FILE_STANDARD_INFORMATION))
3066                 {
3067                     PFILE_STANDARD_INFORMATION pStandard = (PFILE_STANDARD_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3068
3069                     pStandard->NumberOfLinks = 1;
3070                     pStandard->DeletePending = 0;
3071                     pStandard->AllocationSize.QuadPart = 4096;
3072                     pStandard->EndOfFile.QuadPart = 4096;
3073                     pStandard->Directory = 0;
3074
3075                     *Length -= sizeof( FILE_STANDARD_INFORMATION);
3076                 }
3077                 else
3078                 {
3079                     ntStatus = STATUS_BUFFER_OVERFLOW;
3080                 }
3081
3082                 break;
3083             }
3084
3085             case FileNameInformation:
3086             {
3087
3088                 ULONG ulCopyLength = 0;
3089                 AFSFcb *pFcb = NULL;
3090                 AFSCcb *pCcb = NULL;
3091                 USHORT usFullNameLength = 0;
3092                 PFILE_NAME_INFORMATION pNameInfo = (PFILE_NAME_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3093                 UNICODE_STRING uniName;
3094
3095                 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
3096                 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
3097
3098                 if( *Length < FIELD_OFFSET( FILE_NAME_INFORMATION, FileName))
3099                 {
3100                     ntStatus = STATUS_BUFFER_TOO_SMALL;
3101                     break;
3102                 }
3103
3104                 RtlZeroMemory( pNameInfo,
3105                                *Length);
3106
3107                 usFullNameLength = sizeof( WCHAR) +
3108                                             AFSServerName.Length +
3109                                             pCcb->FullFileName.Length;
3110
3111                 if( *Length >= (LONG)(FIELD_OFFSET( FILE_NAME_INFORMATION, FileName) + (LONG)usFullNameLength))
3112                 {
3113                     ulCopyLength = (LONG)usFullNameLength;
3114                 }
3115                 else
3116                 {
3117                     ulCopyLength = *Length - FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
3118                     ntStatus = STATUS_BUFFER_OVERFLOW;
3119                 }
3120
3121                 pNameInfo->FileNameLength = (ULONG)usFullNameLength;
3122
3123                 *Length -= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
3124
3125                 if( ulCopyLength > 0)
3126                 {
3127
3128                     pNameInfo->FileName[ 0] = L'\\';
3129                     ulCopyLength -= sizeof( WCHAR);
3130
3131                     *Length -= sizeof( WCHAR);
3132
3133                     if( ulCopyLength >= AFSServerName.Length)
3134                     {
3135
3136                         RtlCopyMemory( &pNameInfo->FileName[ 1],
3137                                        AFSServerName.Buffer,
3138                                        AFSServerName.Length);
3139
3140                         ulCopyLength -= AFSServerName.Length;
3141                         *Length -= AFSServerName.Length;
3142
3143                         if( ulCopyLength >= pCcb->FullFileName.Length)
3144                         {
3145
3146                             RtlCopyMemory( &pNameInfo->FileName[ 1 + (AFSServerName.Length/sizeof( WCHAR))],
3147                                            pCcb->FullFileName.Buffer,
3148                                            pCcb->FullFileName.Length);
3149
3150                             ulCopyLength -= pCcb->FullFileName.Length;
3151                             *Length -= pCcb->FullFileName.Length;
3152
3153                             uniName.Length = (USHORT)pNameInfo->FileNameLength;
3154                             uniName.MaximumLength = uniName.Length;
3155                             uniName.Buffer = pNameInfo->FileName;
3156                         }
3157                         else
3158                         {
3159
3160                             RtlCopyMemory( &pNameInfo->FileName[ 1 + (AFSServerName.Length/sizeof( WCHAR))],
3161                                            pCcb->FullFileName.Buffer,
3162                                            ulCopyLength);
3163
3164                             *Length -= ulCopyLength;
3165
3166                             uniName.Length = (USHORT)(sizeof( WCHAR) + AFSServerName.Length + ulCopyLength);
3167                             uniName.MaximumLength = uniName.Length;
3168                             uniName.Buffer = pNameInfo->FileName;
3169                         }
3170
3171                         AFSDbgLogMsg( AFS_SUBSYSTEM_PIOCTL_PROCESSING,
3172                                       AFS_TRACE_LEVEL_VERBOSE,
3173                                       "AFSProcessPIOCtlQueryInfo (FileNameInformation) Returning %wZ\n",
3174                                       &uniName);
3175                     }
3176                 }
3177
3178                 break;
3179             }
3180
3181             case FileInternalInformation:
3182             {
3183
3184                 PFILE_INTERNAL_INFORMATION pInternalInfo = (PFILE_INTERNAL_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3185
3186                 if( *Length >= sizeof( FILE_INTERNAL_INFORMATION))
3187                 {
3188
3189                     pInternalInfo->IndexNumber.HighPart = 0;
3190
3191                     pInternalInfo->IndexNumber.LowPart = 0;
3192
3193                     *Length -= sizeof( FILE_INTERNAL_INFORMATION);
3194                 }
3195                 else
3196                 {
3197
3198                     ntStatus = STATUS_BUFFER_TOO_SMALL;
3199                 }
3200
3201                 break;
3202             }
3203
3204             default:
3205             {
3206                 ntStatus = STATUS_INVALID_PARAMETER;
3207
3208                 AFSDbgLogMsg( AFS_SUBSYSTEM_PIOCTL_PROCESSING,
3209                               AFS_TRACE_LEVEL_WARNING,
3210                               "AFSProcessPIOCtlQueryInfo Not handling request %08lX\n",
3211                               ulFileInformationClass);
3212
3213                 break;
3214             }
3215         }
3216     }
3217
3218     return ntStatus;
3219 }
3220