27e508db18ed071bbd30953504b12326eac827be
[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             pCcb->FileUnwindInfo.FileAttributes = DirectoryCB->ObjectInformation->FileAttributes;
1478
1479             DirectoryCB->ObjectInformation->FileAttributes = pBuffer->FileAttributes;
1480
1481             ulNotifyFilter |= FILE_NOTIFY_CHANGE_ATTRIBUTES;
1482
1483             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED);
1484         }
1485
1486         pCcb->FileUnwindInfo.CreationTime.QuadPart = (ULONGLONG)-1;
1487
1488         if( pBuffer->CreationTime.QuadPart != (ULONGLONG)-1 &&
1489             pBuffer->CreationTime.QuadPart != (ULONGLONG)0)
1490         {
1491
1492             pCcb->FileUnwindInfo.CreationTime.QuadPart = DirectoryCB->ObjectInformation->CreationTime.QuadPart;
1493
1494             DirectoryCB->ObjectInformation->CreationTime.QuadPart = pBuffer->CreationTime.QuadPart;
1495
1496             ulNotifyFilter |= FILE_NOTIFY_CHANGE_CREATION;
1497
1498             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CREATE_TIME);
1499         }
1500
1501         pCcb->FileUnwindInfo.LastAccessTime.QuadPart = (ULONGLONG)-1;
1502
1503         if( pBuffer->LastAccessTime.QuadPart != (ULONGLONG)-1 &&
1504             pBuffer->LastAccessTime.QuadPart != (ULONGLONG)0)
1505         {
1506
1507             pCcb->FileUnwindInfo.LastAccessTime.QuadPart = DirectoryCB->ObjectInformation->LastAccessTime.QuadPart;
1508
1509             DirectoryCB->ObjectInformation->LastAccessTime.QuadPart = pBuffer->LastAccessTime.QuadPart;
1510
1511             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_ACCESS;
1512
1513             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_ACCESS_TIME);
1514         }
1515
1516         pCcb->FileUnwindInfo.LastWriteTime.QuadPart = (ULONGLONG)-1;
1517
1518         if( pBuffer->LastWriteTime.QuadPart != (ULONGLONG)-1 &&
1519             pBuffer->LastWriteTime.QuadPart != (ULONGLONG)0)
1520         {
1521
1522             pCcb->FileUnwindInfo.LastWriteTime.QuadPart = DirectoryCB->ObjectInformation->LastWriteTime.QuadPart;
1523
1524             DirectoryCB->ObjectInformation->LastWriteTime.QuadPart = pBuffer->LastWriteTime.QuadPart;
1525
1526             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_WRITE;
1527
1528             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_LAST_WRITE_TIME);
1529         }
1530
1531         pCcb->FileUnwindInfo.ChangeTime.QuadPart = (ULONGLONG)-1;
1532
1533         if( pBuffer->ChangeTime.QuadPart != (ULONGLONG)-1 &&
1534             pBuffer->ChangeTime.QuadPart != (ULONGLONG)0)
1535         {
1536
1537             pCcb->FileUnwindInfo.ChangeTime.QuadPart = DirectoryCB->ObjectInformation->ChangeTime.QuadPart;
1538
1539             DirectoryCB->ObjectInformation->ChangeTime.QuadPart = pBuffer->ChangeTime.QuadPart;
1540
1541             ulNotifyFilter |= FILE_NOTIFY_CHANGE_LAST_ACCESS;
1542
1543             SetFlag( DirectoryCB->ObjectInformation->Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
1544         }
1545
1546         if( ulNotifyFilter > 0)
1547         {
1548
1549             if( DirectoryCB->ObjectInformation->ParentObjectInformation != NULL)
1550             {
1551
1552                 AFSFsRtlNotifyFullReportChange( DirectoryCB->ObjectInformation->ParentObjectInformation,
1553                                                 pCcb,
1554                                                 (ULONG)ulNotifyFilter,
1555                                                 (ULONG)FILE_ACTION_MODIFIED);
1556             }
1557         }
1558
1559 try_exit:
1560
1561         NOTHING;
1562     }
1563
1564     return ntStatus;
1565 }
1566
1567 NTSTATUS
1568 AFSSetDispositionInfo( IN PIRP Irp,
1569                        IN AFSDirectoryCB *DirectoryCB)
1570 {
1571     NTSTATUS ntStatus = STATUS_SUCCESS;
1572     PFILE_DISPOSITION_INFORMATION pBuffer;
1573     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1574     AFSFcb *pFcb = NULL;
1575     AFSCcb *pCcb = NULL;
1576
1577     __Enter
1578     {
1579
1580         pBuffer = (PFILE_DISPOSITION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
1581
1582         pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
1583
1584         pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
1585
1586         //
1587         // Can't delete the root
1588         //
1589
1590         if( pFcb->Header.NodeTypeCode == AFS_ROOT_FCB)
1591         {
1592
1593             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1594                           AFS_TRACE_LEVEL_ERROR,
1595                           "AFSSetDispositionInfo Attempt to delete root entry\n");
1596
1597             try_return( ntStatus = STATUS_CANNOT_DELETE);
1598         }
1599
1600         //
1601         // If the file is read only then do not allow the delete
1602         //
1603
1604         if( BooleanFlagOn( DirectoryCB->ObjectInformation->FileAttributes, FILE_ATTRIBUTE_READONLY))
1605         {
1606
1607             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1608                           AFS_TRACE_LEVEL_ERROR,
1609                           "AFSSetDispositionInfo Attempt to delete read only entry %wZ\n",
1610                           &DirectoryCB->NameInformation.FileName);
1611
1612             try_return( ntStatus = STATUS_CANNOT_DELETE);
1613         }
1614
1615         if( pBuffer->DeleteFile)
1616         {
1617
1618             //
1619             // Check if the caller can delete the file
1620             //
1621
1622             ntStatus = AFSNotifyDelete( DirectoryCB,
1623                                         TRUE);
1624
1625             if( !NT_SUCCESS( ntStatus))
1626             {
1627
1628                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1629                               AFS_TRACE_LEVEL_ERROR,
1630                               "AFSSetDispositionInfo Cannot delete entry %wZ Status %08lX\n",
1631                               &DirectoryCB->NameInformation.FileName,
1632                               ntStatus);
1633
1634                 try_return( ntStatus);
1635             }
1636
1637             if( pFcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB)
1638             {
1639
1640                 //
1641                 // Check if this is a directory that there are not currently other opens
1642                 //
1643
1644                 if( pFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount > 0)
1645                 {
1646
1647                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1648                                   AFS_TRACE_LEVEL_ERROR,
1649                                   "AFSSetDispositionInfo Attempt to delete directory %wZ with open %u handles\n",
1650                                   &DirectoryCB->NameInformation.FileName,
1651                                   pFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount);
1652
1653                     try_return( ntStatus = STATUS_DIRECTORY_NOT_EMPTY);
1654                 }
1655
1656                 if( !AFSIsDirectoryEmptyForDelete( pFcb))
1657                 {
1658
1659                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1660                                   AFS_TRACE_LEVEL_ERROR,
1661                                   "AFSSetDispositionInfo Attempt to delete non-empty directory %wZ\n",
1662                                   &DirectoryCB->NameInformation.FileName);
1663
1664                     try_return( ntStatus = STATUS_DIRECTORY_NOT_EMPTY);
1665                 }
1666             }
1667             else
1668             {
1669
1670                 //
1671                 // Attempt to flush any outstanding data
1672                 //
1673
1674                 if( !MmFlushImageSection( &pFcb->NPFcb->SectionObjectPointers,
1675                                           MmFlushForDelete))
1676                 {
1677
1678                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1679                                   AFS_TRACE_LEVEL_ERROR,
1680                                   "AFSSetDispositionInfo Failed to flush image section for delete Entry %wZ\n",
1681                                   &DirectoryCB->NameInformation.FileName);
1682
1683                     try_return( ntStatus = STATUS_CANNOT_DELETE);
1684                 }
1685
1686                 //
1687                 // Purge the cache as well
1688                 //
1689
1690                 if( pFcb->NPFcb->SectionObjectPointers.DataSectionObject != NULL)
1691                 {
1692
1693                     CcPurgeCacheSection( &pFcb->NPFcb->SectionObjectPointers,
1694                                          NULL,
1695                                          0,
1696                                          TRUE);
1697                 }
1698             }
1699
1700             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1701                           AFS_TRACE_LEVEL_VERBOSE,
1702                           "AFSSetDispositionInfo Setting PENDING_DELETE on DirEntry  %p Name %wZ\n",
1703                           DirectoryCB,
1704                           &DirectoryCB->NameInformation.FileName);
1705
1706             SetFlag( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_PENDING_DELETE);
1707         }
1708         else
1709         {
1710
1711             ClearFlag( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_PENDING_DELETE);
1712         }
1713
1714         //
1715         // OK, should be good to go, set the flag in the file object
1716         //
1717
1718         pIrpSp->FileObject->DeletePending = pBuffer->DeleteFile;
1719
1720 try_exit:
1721
1722         NOTHING;
1723     }
1724
1725     return ntStatus;
1726 }
1727
1728 NTSTATUS
1729 AFSSetRenameInfo( IN PIRP Irp)
1730 {
1731
1732     NTSTATUS ntStatus = STATUS_SUCCESS;
1733     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1734     IO_STATUS_BLOCK stIoSb = {0,0};
1735     AFSFcb *pSrcFcb = NULL, *pTargetDcb = NULL, *pTargetFcb = NULL;
1736     AFSCcb *pSrcCcb = NULL, *pTargetDirCcb = NULL;
1737     PFILE_OBJECT pSrcFileObj = pIrpSp->FileObject;
1738     PFILE_OBJECT pTargetFileObj = pIrpSp->Parameters.SetFile.FileObject;
1739     PFILE_RENAME_INFORMATION pRenameInfo = NULL;
1740     UNICODE_STRING uniTargetName, uniSourceName;
1741     BOOLEAN bReplaceIfExists = FALSE;
1742     UNICODE_STRING uniShortName;
1743     AFSDirectoryCB *pTargetDirEntry = NULL;
1744     ULONG ulTargetCRC = 0;
1745     BOOLEAN bTargetEntryExists = FALSE;
1746     AFSObjectInfoCB *pSrcObject = NULL, *pTargetObject = NULL;
1747     AFSObjectInfoCB *pSrcParentObject = NULL, *pTargetParentObject = NULL;
1748     AFSFileID stNewFid, stTmpTargetFid;
1749     UNICODE_STRING uniTmpTargetName;
1750     BOOLEAN bReplaceTmpTargetEntry = FALSE;
1751     ULONG ulNotificationAction = 0, ulNotifyFilter = 0;
1752     UNICODE_STRING uniFullTargetPath;
1753     BOOLEAN bCommonParent = FALSE;
1754
1755     __Enter
1756     {
1757
1758         bReplaceIfExists = pIrpSp->Parameters.SetFile.ReplaceIfExists;
1759
1760         pSrcFcb = (AFSFcb *)pSrcFileObj->FsContext;
1761         pSrcCcb = (AFSCcb *)pSrcFileObj->FsContext2;
1762
1763         pSrcObject = pSrcFcb->ObjectInformation;
1764
1765         uniTmpTargetName.Length = 0;
1766         uniTmpTargetName.MaximumLength = 0;
1767         uniTmpTargetName.Buffer = NULL;
1768
1769         //
1770         // Perform some basic checks to ensure FS integrity
1771         //
1772
1773         if( pSrcFcb->Header.NodeTypeCode == AFS_ROOT_FCB)
1774         {
1775
1776             //
1777             // Can't rename the root directory
1778             //
1779
1780             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1781                           AFS_TRACE_LEVEL_ERROR,
1782                           "AFSSetRenameInfo Attempt to rename root entry\n");
1783
1784             try_return( ntStatus = STATUS_INVALID_PARAMETER);
1785         }
1786
1787         if( pSrcFcb->Header.NodeTypeCode == AFS_DIRECTORY_FCB)
1788         {
1789
1790             //
1791             // If there are any open children then fail the rename
1792             //
1793
1794             if( pSrcFcb->ObjectInformation->Specific.Directory.ChildOpenHandleCount > 0)
1795             {
1796
1797                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1798                               AFS_TRACE_LEVEL_ERROR,
1799                               "AFSSetRenameInfo Attempt to rename directory with open children %wZ\n",
1800                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
1801
1802                 try_return( ntStatus = STATUS_ACCESS_DENIED);
1803             }
1804         }
1805         else
1806         {
1807
1808             if( pSrcFcb->OpenHandleCount > 1)
1809             {
1810
1811                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1812                               AFS_TRACE_LEVEL_ERROR,
1813                               "AFSSetRenameInfo Attempt to rename directory with open references %wZ\n",
1814                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
1815
1816                 try_return( ntStatus = STATUS_ACCESS_DENIED);
1817             }
1818         }
1819
1820         //
1821         // Resolve the target fileobject
1822         //
1823
1824         if( pTargetFileObj == NULL)
1825         {
1826
1827             //
1828             // This is a simple rename. Here the target directory is the same as the source parent directory
1829             // and the name is retrieved from the system buffer information
1830             //
1831
1832             pRenameInfo = (PFILE_RENAME_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
1833
1834             pTargetParentObject = pSrcFcb->ObjectInformation->ParentObjectInformation;
1835
1836             pTargetDcb = pTargetParentObject->Fcb;
1837
1838             uniTargetName.Length = (USHORT)pRenameInfo->FileNameLength;
1839             uniTargetName.Buffer = (PWSTR)&pRenameInfo->FileName;
1840         }
1841         else
1842         {
1843
1844             //
1845             // So here we have the target directory taken from the targetfile object
1846             //
1847
1848             pTargetDcb = (AFSFcb *)pTargetFileObj->FsContext;
1849
1850             pTargetDirCcb = (AFSCcb *)pTargetFileObj->FsContext2;
1851
1852             pTargetParentObject = (AFSObjectInfoCB *)pTargetDcb->ObjectInformation;
1853
1854             //
1855             // Grab the target name which we setup in the IRP_MJ_CREATE handler. By how we set this up
1856             // it is only the target component of the rename operation
1857             //
1858
1859             uniTargetName = *((PUNICODE_STRING)&pTargetFileObj->FileName);
1860         }
1861
1862         //
1863         // We do not allow cross-volume renames to occur
1864         //
1865
1866         if( pTargetParentObject->VolumeCB != pSrcObject->VolumeCB)
1867         {
1868
1869             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1870                           AFS_TRACE_LEVEL_ERROR,
1871                           "AFSSetRenameInfo Attempt to rename directory to different volume %wZ\n",
1872                           &pSrcCcb->DirectoryCB->NameInformation.FileName);
1873
1874             try_return( ntStatus = STATUS_NOT_SAME_DEVICE);
1875         }
1876
1877         //
1878         // If the target exists be sure the ReplaceIfExists flag is set
1879         //
1880
1881         AFSAcquireShared( pTargetParentObject->VolumeCB->VolumeLock,
1882                           TRUE);
1883
1884         ulTargetCRC = AFSGenerateCRC( &uniTargetName,
1885                                       FALSE);
1886
1887         AFSAcquireShared( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.TreeLock,
1888                           TRUE);
1889
1890         AFSLocateCaseSensitiveDirEntry( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.CaseSensitiveTreeHead,
1891                                         ulTargetCRC,
1892                                         &pTargetDirEntry);
1893
1894         if( pTargetDirEntry == NULL)
1895         {
1896
1897             //
1898             // Missed so perform a case insensitive lookup
1899             //
1900
1901             ulTargetCRC = AFSGenerateCRC( &uniTargetName,
1902                                           TRUE);
1903
1904             AFSLocateCaseInsensitiveDirEntry( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.CaseInsensitiveTreeHead,
1905                                               ulTargetCRC,
1906                                               &pTargetDirEntry);
1907         }
1908
1909         //
1910         // Increment our ref count on the dir entry
1911         //
1912
1913         if( pTargetDirEntry != NULL)
1914         {
1915             InterlockedIncrement( &pTargetDirEntry->OpenReferenceCount);
1916         }
1917
1918         AFSReleaseResource( pTargetParentObject->Specific.Directory.DirectoryNodeHdr.TreeLock);
1919
1920         if( pTargetDirEntry != NULL)
1921         {
1922
1923             if( !bReplaceIfExists)
1924             {
1925
1926                 AFSReleaseResource( pTargetParentObject->VolumeCB->VolumeLock);
1927
1928                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
1929                               AFS_TRACE_LEVEL_ERROR,
1930                               "AFSSetRenameInfo Attempt to rename directory with target collision %wZ Target %wZ\n",
1931                               &pSrcCcb->DirectoryCB->NameInformation.FileName,
1932                               &pTargetDirEntry->NameInformation.FileName);
1933
1934                 try_return( ntStatus = STATUS_OBJECT_NAME_COLLISION);
1935             }
1936
1937             bTargetEntryExists = TRUE;
1938         }
1939
1940         AFSReleaseResource( pTargetParentObject->VolumeCB->VolumeLock);
1941
1942         //
1943         // Extract off the final component name from the Fcb
1944         //
1945
1946         uniSourceName.Length = (USHORT)pSrcCcb->DirectoryCB->NameInformation.FileName.Length;
1947         uniSourceName.MaximumLength = uniSourceName.Length;
1948
1949         uniSourceName.Buffer = pSrcCcb->DirectoryCB->NameInformation.FileName.Buffer;
1950
1951         //
1952         // The quick check to see if they are not really performing a rename
1953         // Do the names match? Only do this where the parent directories are
1954         // the same
1955         //
1956
1957         if( pTargetParentObject == pSrcFcb->ObjectInformation->ParentObjectInformation)
1958         {
1959
1960             bCommonParent = TRUE;
1961
1962             if( FsRtlAreNamesEqual( &uniTargetName,
1963                                     &uniSourceName,
1964                                     TRUE,
1965                                     NULL))
1966             {
1967
1968                 //
1969                 // Check for case only rename
1970                 //
1971
1972                 if( !FsRtlAreNamesEqual( &uniTargetName,
1973                                          &uniSourceName,
1974                                          FALSE,
1975                                          NULL))
1976                 {
1977
1978                     //
1979                     // Just move in the new case form of the name
1980                     //
1981
1982                     RtlCopyMemory( pSrcCcb->DirectoryCB->NameInformation.FileName.Buffer,
1983                                    uniTargetName.Buffer,
1984                                    uniTargetName.Length);
1985                 }
1986
1987                 try_return( ntStatus = STATUS_SUCCESS);
1988             }
1989         }
1990         else
1991         {
1992
1993             bCommonParent = FALSE;
1994         }
1995
1996         //
1997         // If the target name exists then we need to 'move' the target before
1998         // sending the rename to the service
1999         //
2000
2001         if( bReplaceIfExists &&
2002             pTargetDirEntry != NULL)
2003         {
2004
2005             //
2006             // What we will do is temporarily rename the file to a tmp file
2007             // so we can back out if anything fails below
2008             // First thing is to remove the original target from the parent
2009             //
2010
2011             AFSAcquireExcl( pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
2012                             TRUE);
2013
2014             AFSRemoveDirNodeFromParent( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2015                                         pTargetDirEntry,
2016                                         TRUE);
2017
2018             AFSReleaseResource( pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
2019
2020             pTargetDirEntry->FileIndex = (ULONG)InterlockedIncrement( &pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2021
2022             uniTmpTargetName.Length = 0;
2023             uniTmpTargetName.MaximumLength = uniTargetName.Length + (4 * sizeof( WCHAR));
2024
2025             uniTmpTargetName.Buffer = (WCHAR *)AFSExAllocatePoolWithTag( PagedPool,
2026                                                                          uniTmpTargetName.MaximumLength,
2027                                                                          AFS_GENERIC_MEMORY_11_TAG);
2028
2029             if( uniTmpTargetName.Buffer == NULL)
2030             {
2031
2032                 //
2033                 // Re-insert the entry
2034                 //
2035
2036                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2037                                         pTargetDirEntry,
2038                                         TRUE);
2039
2040                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2041                               AFS_TRACE_LEVEL_ERROR,
2042                               "AFSSetRenameInfo Failed tmp buffer allocation during rename of %wZ\n",
2043                               &pSrcCcb->DirectoryCB->NameInformation.FileName);
2044
2045                 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2046             }
2047
2048             RtlZeroMemory( uniTmpTargetName.Buffer,
2049                            uniTmpTargetName.MaximumLength);
2050
2051             uniTmpTargetName.Length = uniTargetName.Length;
2052
2053             RtlCopyMemory( uniTmpTargetName.Buffer,
2054                            uniTargetName.Buffer,
2055                            uniTmpTargetName.Length);
2056
2057             RtlCopyMemory( &uniTmpTargetName.Buffer[ uniTmpTargetName.Length/sizeof( WCHAR)],
2058                            L".tmp",
2059                            4 * sizeof( WCHAR));
2060
2061             uniTmpTargetName.Length += (4 * sizeof( WCHAR));
2062
2063             ntStatus = AFSNotifyRename( pTargetDirEntry->ObjectInformation,
2064                                         pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2065                                         pTargetDcb->ObjectInformation,
2066                                         pTargetDirEntry,
2067                                         &uniTmpTargetName,
2068                                         &stTmpTargetFid);
2069
2070             if( !NT_SUCCESS( ntStatus))
2071             {
2072
2073                 //
2074                 // Re-insert the entry
2075                 //
2076
2077                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2078                                         pTargetDirEntry,
2079                                         TRUE);
2080
2081                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2082                               AFS_TRACE_LEVEL_ERROR,
2083                               "AFSSetRenameInfo Failed rename of %wZ to tmp %wZ Status %08lX\n",
2084                               &pSrcCcb->DirectoryCB->NameInformation.FileName,
2085                               &uniTmpTargetName,
2086                               ntStatus);
2087
2088                 try_return( ntStatus);
2089             }
2090
2091             //
2092             // Indicate we need to replace this entry if any failure occurs below
2093             //
2094
2095             bReplaceTmpTargetEntry = TRUE;
2096         }
2097
2098         //
2099         // We need to remove the DirEntry from the parent node, update the index
2100         // and reinsert it into the parent tree. Note that for entries with the
2101         // same parent we do not pull the node from the enumeration list
2102         //
2103
2104         AFSAcquireExcl( pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock,
2105                         TRUE);
2106
2107         AFSRemoveDirNodeFromParent( pSrcFcb->ObjectInformation->ParentObjectInformation,
2108                                     pSrcCcb->DirectoryCB,
2109                                     !bCommonParent);
2110
2111         AFSReleaseResource( pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.TreeLock);
2112
2113         if( !bCommonParent)
2114         {
2115
2116             //
2117             // We always need to update the FileIndex since this entry will be put at the 'end'
2118             // of the enumeraiton list. If we don't it will cause recursion ... We do this
2119             // here to cover any failures which might occur below
2120             //
2121
2122             pSrcCcb->DirectoryCB->FileIndex =
2123                             (ULONG)InterlockedIncrement( &pSrcFcb->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2124         }
2125
2126         //
2127         // OK, this is a simple rename. Issue the rename
2128         // request to the service.
2129         //
2130
2131         ntStatus = AFSNotifyRename( pSrcFcb->ObjectInformation,
2132                                     pSrcFcb->ObjectInformation->ParentObjectInformation,
2133                                     pTargetDcb->ObjectInformation,
2134                                     pSrcCcb->DirectoryCB,
2135                                     &uniTargetName,
2136                                     &stNewFid);
2137
2138         if( !NT_SUCCESS( ntStatus))
2139         {
2140
2141             //
2142             // Attempt to re-insert the directory entry
2143             //
2144
2145             AFSInsertDirectoryNode( pSrcFcb->ObjectInformation->ParentObjectInformation,
2146                                     pSrcCcb->DirectoryCB,
2147                                     !bCommonParent);
2148
2149             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2150                           AFS_TRACE_LEVEL_ERROR,
2151                           "AFSSetRenameInfo Failed rename of %wZ to target %wZ Status %08lX\n",
2152                           &pSrcCcb->DirectoryCB->NameInformation.FileName,
2153                           &uniTargetName,
2154                           ntStatus);
2155
2156             try_return( ntStatus);
2157         }
2158
2159         //
2160         // Set the notification up for the source file
2161         //
2162
2163         if( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation == pTargetParentObject &&
2164             !bReplaceTmpTargetEntry)
2165         {
2166
2167             ulNotificationAction = FILE_ACTION_RENAMED_OLD_NAME;
2168         }
2169         else
2170         {
2171
2172             ulNotificationAction = FILE_ACTION_REMOVED;
2173         }
2174
2175         if( pSrcCcb->DirectoryCB->ObjectInformation->FileType == AFS_FILE_TYPE_DIRECTORY)
2176         {
2177
2178             ulNotifyFilter = FILE_NOTIFY_CHANGE_DIR_NAME;
2179         }
2180         else
2181         {
2182
2183             ulNotifyFilter = FILE_NOTIFY_CHANGE_FILE_NAME;
2184         }
2185
2186         AFSFsRtlNotifyFullReportChange( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation,
2187                                         pSrcCcb,
2188                                         (ULONG)ulNotifyFilter,
2189                                         (ULONG)ulNotificationAction);
2190
2191         //
2192         // Update the name in the dir entry.
2193         //
2194
2195         ntStatus = AFSUpdateDirEntryName( pSrcCcb->DirectoryCB,
2196                                           &uniTargetName);
2197
2198         if( !NT_SUCCESS( ntStatus))
2199         {
2200
2201             //
2202             // Attempt to re-insert the directory entry
2203             //
2204
2205             AFSInsertDirectoryNode( pSrcFcb->ObjectInformation->ParentObjectInformation,
2206                                     pSrcCcb->DirectoryCB,
2207                                     !bCommonParent);
2208
2209             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2210                           AFS_TRACE_LEVEL_ERROR,
2211                           "AFSSetRenameInfo Failed update of dir entry %wZ to target %wZ Status %08lX\n",
2212                           &pSrcCcb->DirectoryCB->NameInformation.FileName,
2213                           &uniTargetName,
2214                           ntStatus);
2215
2216             try_return( ntStatus);
2217         }
2218
2219         //
2220         // Update the object information block, if needed
2221         //
2222
2223         if( !AFSIsEqualFID( &pSrcObject->FileId,
2224                             &stNewFid))
2225         {
2226
2227             //
2228             // Remove the old information entry
2229             //
2230
2231             AFSAcquireExcl( pSrcObject->VolumeCB->ObjectInfoTree.TreeLock,
2232                             TRUE);
2233
2234             AFSRemoveHashEntry( &pSrcObject->VolumeCB->ObjectInfoTree.TreeHead,
2235                                 &pSrcObject->TreeEntry);
2236
2237             RtlCopyMemory( &pSrcObject->FileId,
2238                            &stNewFid,
2239                            sizeof( AFSFileID));
2240
2241             //
2242             // Insert the entry into the new object table.
2243             //
2244
2245             pSrcObject->TreeEntry.HashIndex = AFSCreateLowIndex( &pSrcObject->FileId);
2246
2247             if( pSrcObject->VolumeCB->ObjectInfoTree.TreeHead == NULL)
2248             {
2249
2250                 pSrcObject->VolumeCB->ObjectInfoTree.TreeHead = &pSrcObject->TreeEntry;
2251             }
2252             else
2253             {
2254
2255                 AFSInsertHashEntry( pSrcObject->VolumeCB->ObjectInfoTree.TreeHead,
2256                                     &pSrcObject->TreeEntry);
2257             }
2258
2259             AFSReleaseResource( pSrcObject->VolumeCB->ObjectInfoTree.TreeLock);
2260         }
2261
2262         //
2263         // Update the hash values for the name trees.
2264         //
2265
2266         pSrcCcb->DirectoryCB->CaseSensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pSrcCcb->DirectoryCB->NameInformation.FileName,
2267                                                                                  FALSE);
2268
2269         pSrcCcb->DirectoryCB->CaseInsensitiveTreeEntry.HashIndex = AFSGenerateCRC( &pSrcCcb->DirectoryCB->NameInformation.FileName,
2270                                                                                    TRUE);
2271
2272         if( pSrcCcb->DirectoryCB->NameInformation.ShortNameLength > 0)
2273         {
2274
2275             uniShortName.Length = pSrcCcb->DirectoryCB->NameInformation.ShortNameLength;
2276             uniShortName.Buffer = pSrcCcb->DirectoryCB->NameInformation.ShortName;
2277
2278             pSrcCcb->DirectoryCB->Type.Data.ShortNameTreeEntry.HashIndex = AFSGenerateCRC( &uniShortName,
2279                                                                                            TRUE);
2280         }
2281         else
2282         {
2283
2284             pSrcCcb->DirectoryCB->Type.Data.ShortNameTreeEntry.HashIndex = 0;
2285         }
2286
2287         if( !bCommonParent)
2288         {
2289
2290             //
2291             // Update the file index for the object in the new parent
2292             //
2293
2294             pSrcCcb->DirectoryCB->FileIndex = (ULONG)InterlockedIncrement( &pTargetParentObject->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2295         }
2296
2297         //
2298         // Re-insert the directory entry
2299         //
2300
2301         AFSInsertDirectoryNode( pTargetParentObject,
2302                                 pSrcCcb->DirectoryCB,
2303                                 !bCommonParent);
2304
2305         //
2306         // Update the parent pointer in the source object if they are different
2307         //
2308
2309         if( pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation != pTargetParentObject)
2310         {
2311
2312             InterlockedDecrement( &pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation->Specific.Directory.ChildOpenHandleCount);
2313
2314             InterlockedDecrement( &pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation->Specific.Directory.ChildOpenReferenceCount);
2315
2316             InterlockedIncrement( &pTargetParentObject->Specific.Directory.ChildOpenHandleCount);
2317
2318             InterlockedIncrement( &pTargetParentObject->Specific.Directory.ChildOpenReferenceCount);
2319
2320             pSrcCcb->DirectoryCB->ObjectInformation->ParentObjectInformation = pTargetParentObject;
2321
2322             ulNotificationAction = FILE_ACTION_ADDED;
2323         }
2324         else
2325         {
2326
2327             ulNotificationAction = FILE_ACTION_RENAMED_NEW_NAME;
2328         }
2329
2330         //
2331         // Now update the notification for the target file
2332         //
2333
2334         AFSFsRtlNotifyFullReportChange( pTargetParentObject->ParentObjectInformation,
2335                                         pSrcCcb,
2336                                         (ULONG)ulNotifyFilter,
2337                                         (ULONG)ulNotificationAction);
2338
2339         //
2340         // If we performed the rename of the target because it existed, we now need to
2341         // delete the tmp target we created above
2342         //
2343
2344         if( bReplaceTmpTargetEntry)
2345         {
2346
2347             RtlCopyMemory( &pTargetDirEntry->ObjectInformation->FileId,
2348                            &stTmpTargetFid,
2349                            sizeof( AFSFileID));
2350
2351             //
2352             // Update the name in the dir entry
2353             //
2354
2355             ntStatus = AFSUpdateDirEntryName( pTargetDirEntry,
2356                                               &uniTmpTargetName);
2357
2358             if( !NT_SUCCESS( ntStatus))
2359             {
2360
2361                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2362                               AFS_TRACE_LEVEL_ERROR,
2363                               "AFSSetRenameInfo Failed update of target dir entry %wZ to tmp %wZ Status %08lX\n",
2364                               &pTargetDirEntry->NameInformation.FileName,
2365                               &uniTmpTargetName,
2366                               ntStatus);
2367
2368                 try_return( ntStatus);
2369             }
2370
2371             ntStatus = AFSNotifyDelete( pTargetDirEntry,
2372                                         FALSE);
2373
2374             if( !NT_SUCCESS( ntStatus))
2375             {
2376
2377                 AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2378                               AFS_TRACE_LEVEL_VERBOSE,
2379                               "AFSSetRenameInfo object deletion failure dir entry %p name %wZ to tmp %wZ\n",
2380                               pTargetDirEntry,
2381                               &pTargetDirEntry->NameInformation.FileName,
2382                               &uniTmpTargetName);
2383
2384                 try_return( ntStatus);
2385             }
2386
2387             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2388                           AFS_TRACE_LEVEL_VERBOSE,
2389                           "AFSSetRenameInfo Setting DELETE flag in dir entry %p name %wZ to tmp %wZ\n",
2390                           pTargetDirEntry,
2391                           &pTargetDirEntry->NameInformation.FileName,
2392                           &uniTmpTargetName);
2393
2394             SetFlag( pTargetDirEntry->Flags, AFS_DIR_ENTRY_DELETED);
2395
2396             //
2397             // Try and purge the cache map if this is a file
2398             //
2399
2400             if( pTargetDirEntry->ObjectInformation->FileType == AFS_FILE_TYPE_FILE &&
2401                 pTargetDirEntry->ObjectInformation->Fcb != NULL &&
2402                 pTargetDirEntry->OpenReferenceCount > 1)
2403             {
2404
2405                 pTargetFcb = pTargetDirEntry->ObjectInformation->Fcb;
2406
2407                 AFSAcquireExcl( &pTargetFcb->NPFcb->Resource,
2408                                 TRUE);
2409
2410                 //
2411                 // Try and flush the cache map
2412                 //
2413
2414                 if( !MmFlushImageSection( &pTargetFcb->NPFcb->SectionObjectPointers,
2415                                           MmFlushForDelete))
2416                 {
2417
2418                     AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
2419                                   AFS_TRACE_LEVEL_ERROR,
2420                                   "AFSSetRenameInfo Failed to flush section for deleted temp file %wZ\n",
2421                                   &pTargetDirEntry->NameInformation.FileName);
2422                 }
2423
2424                 AFSReleaseResource( &pTargetFcb->NPFcb->Resource);
2425             }
2426
2427             ASSERT( pTargetDirEntry->OpenReferenceCount > 0);
2428
2429             InterlockedDecrement( &pTargetDirEntry->OpenReferenceCount);
2430
2431             if( pTargetDirEntry->OpenReferenceCount == 0)
2432             {
2433
2434                 SetFlag( pTargetDirEntry->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED);
2435
2436                 ASSERT( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_ENTRY_NOT_IN_PARENT_TREE));
2437
2438                 //
2439                 // Free up the name buffer if it was reallocated
2440                 //
2441
2442                 if( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_RELEASE_NAME_BUFFER))
2443                 {
2444
2445                     AFSExFreePool( pTargetDirEntry->NameInformation.FileName.Buffer);
2446                 }
2447
2448                 if( BooleanFlagOn( pTargetDirEntry->Flags, AFS_DIR_RELEASE_TARGET_NAME_BUFFER))
2449                 {
2450
2451                     AFSExFreePool( pTargetDirEntry->NameInformation.TargetName.Buffer);
2452                 }
2453
2454                 //
2455                 // Dereference the object for this dir entry
2456                 //
2457
2458                 ASSERT( pTargetDirEntry->ObjectInformation->ObjectReferenceCount > 0);
2459
2460                 InterlockedDecrement( &pTargetDirEntry->ObjectInformation->ObjectReferenceCount);
2461
2462                 AFSDbgLogMsg( AFS_SUBSYSTEM_OBJECT_REF_COUNTING,
2463                               AFS_TRACE_LEVEL_VERBOSE,
2464                               "AFSSetRenameInfo Decrement3 count on object %08lX Cnt %d\n",
2465                               pTargetDirEntry->ObjectInformation,
2466                               pTargetDirEntry->ObjectInformation->ObjectReferenceCount);
2467
2468                 //
2469                 // Free up the dir entry
2470                 //
2471
2472                 ExDeleteResourceLite( &pTargetDirEntry->NonPaged->Lock);
2473
2474                 AFSExFreePool( pTargetDirEntry->NonPaged);
2475
2476                 AFSExFreePool( pTargetDirEntry);
2477             }
2478
2479             pTargetDirEntry = NULL;
2480         }
2481
2482 try_exit:
2483
2484
2485         if( !NT_SUCCESS( ntStatus))
2486         {
2487
2488             if( bReplaceTmpTargetEntry)
2489             {
2490
2491                 AFSNotifyRename( pTargetDirEntry->ObjectInformation,
2492                                  pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2493                                  pTargetDcb->ObjectInformation,
2494                                  pTargetDirEntry,
2495                                  &uniTargetName,
2496                                  &stTmpTargetFid);
2497
2498                 //
2499                 // Replace the target entry
2500                 //
2501
2502                 AFSAcquireExcl( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeLock,
2503                                 TRUE);
2504
2505                 if( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead == NULL)
2506                 {
2507
2508                     pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead = &pTargetDirEntry->ObjectInformation->TreeEntry;
2509                 }
2510                 else
2511                 {
2512                     AFSInsertHashEntry( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeHead,
2513                                         &pTargetDirEntry->ObjectInformation->TreeEntry);
2514                 }
2515
2516                 AFSReleaseResource( pTargetDirEntry->ObjectInformation->VolumeCB->ObjectInfoTree.TreeLock);
2517
2518                 //
2519                 // We always need to update the FileIndex since this entry will be put at the 'end'
2520                 // of the enumeraiton list. If we don't it will cause recursion ...
2521                 //
2522
2523                 pTargetDirEntry->FileIndex = (ULONG)InterlockedIncrement( &pTargetDirEntry->ObjectInformation->ParentObjectInformation->Specific.Directory.DirectoryNodeHdr.ContentIndex);
2524
2525                 AFSInsertDirectoryNode( pTargetDirEntry->ObjectInformation->ParentObjectInformation,
2526                                         pTargetDirEntry,
2527                                         TRUE);
2528             }
2529         }
2530
2531         if( pTargetDirEntry != NULL)
2532         {
2533
2534             InterlockedDecrement( &pTargetDirEntry->OpenReferenceCount);
2535         }
2536
2537         if( uniTmpTargetName.Buffer != NULL)
2538         {
2539
2540             AFSExFreePool( uniTmpTargetName.Buffer);
2541         }
2542     }
2543
2544     return ntStatus;
2545 }
2546
2547 NTSTATUS
2548 AFSSetPositionInfo( IN PIRP Irp,
2549                     IN AFSDirectoryCB *DirectoryCB)
2550 {
2551     NTSTATUS ntStatus = STATUS_SUCCESS;
2552     PFILE_POSITION_INFORMATION pBuffer;
2553     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2554
2555     pBuffer = (PFILE_POSITION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2556
2557     pIrpSp->FileObject->CurrentByteOffset.QuadPart = pBuffer->CurrentByteOffset.QuadPart;
2558
2559     return ntStatus;
2560 }
2561
2562 NTSTATUS
2563 AFSSetAllocationInfo( IN PIRP Irp,
2564                       IN AFSDirectoryCB *DirectoryCB)
2565 {
2566     NTSTATUS ntStatus = STATUS_SUCCESS;
2567     PFILE_ALLOCATION_INFORMATION pBuffer;
2568     BOOLEAN bReleasePaging = FALSE;
2569     BOOLEAN bTellCc = FALSE;
2570     BOOLEAN bTellService = FALSE;
2571     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2572     PFILE_OBJECT pFileObject = pIrpSp->FileObject;
2573     AFSFcb *pFcb = NULL;
2574     AFSCcb *pCcb = NULL;
2575     LARGE_INTEGER liSaveAlloc;
2576     LARGE_INTEGER liSaveFileSize;
2577     LARGE_INTEGER liSaveVDL;
2578
2579     pBuffer = (PFILE_ALLOCATION_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2580
2581     pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2582
2583     pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2584
2585     //
2586     // save values to put back
2587     //
2588     liSaveAlloc = pFcb->Header.AllocationSize;
2589     liSaveFileSize = pFcb->Header.FileSize;
2590     liSaveVDL = pFcb->Header.ValidDataLength;
2591
2592     if( pFcb->Header.AllocationSize.QuadPart == pBuffer->AllocationSize.QuadPart ||
2593         pIrpSp->Parameters.SetFile.AdvanceOnly)
2594     {
2595         return STATUS_SUCCESS ;
2596     }
2597
2598     if( pFcb->Header.AllocationSize.QuadPart > pBuffer->AllocationSize.QuadPart)
2599     {
2600         //
2601         // Truncating the file
2602         //
2603         if( !MmCanFileBeTruncated( pFileObject->SectionObjectPointer,
2604                                    &pBuffer->AllocationSize))
2605         {
2606
2607             ntStatus = STATUS_USER_MAPPED_FILE ;
2608         }
2609         else
2610         {
2611             //
2612             // If this is a truncation we need to grab the paging IO resource.
2613             //
2614             AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2615                           AFS_TRACE_LEVEL_VERBOSE,
2616                           "AFSSetAllocationInfo Acquiring Fcb PagingIo lock %08lX EXCL %08lX\n",
2617                           &pFcb->NPFcb->PagingResource,
2618                           PsGetCurrentThread());
2619
2620             AFSAcquireExcl( &pFcb->NPFcb->PagingResource,
2621                             TRUE);
2622
2623             bReleasePaging = TRUE;
2624
2625
2626             pFcb->Header.AllocationSize = pBuffer->AllocationSize;
2627
2628             pFcb->ObjectInformation->AllocationSize = pBuffer->AllocationSize;
2629
2630             //
2631             // Tell Cc that Allocation is moved.
2632             //
2633             bTellCc = TRUE;
2634
2635             if( pFcb->Header.FileSize.QuadPart > pBuffer->AllocationSize.QuadPart)
2636             {
2637                 //
2638                 // We are pulling the EOF back as well so we need to tell
2639                 // the service.
2640                 //
2641                 bTellService = TRUE;
2642
2643                 pFcb->Header.FileSize = pBuffer->AllocationSize;
2644
2645                 pFcb->ObjectInformation->EndOfFile = pBuffer->AllocationSize;
2646             }
2647
2648         }
2649     }
2650     else
2651     {
2652         //
2653         // Tell Cc if allocation is increased.
2654         //
2655         bTellCc = pBuffer->AllocationSize.QuadPart > pFcb->Header.AllocationSize.QuadPart;
2656
2657         pFcb->Header.AllocationSize = pBuffer->AllocationSize;
2658
2659         pFcb->ObjectInformation->AllocationSize = pBuffer->AllocationSize;
2660     }
2661
2662     //
2663     // Now Tell the server if we have to
2664     //
2665     if (bTellService)
2666     {
2667         ntStatus = AFSUpdateFileInformation( &pFcb->ObjectInformation->ParentObjectInformation->FileId,
2668                                              pFcb->ObjectInformation,
2669                                              &pFcb->AuthGroup);
2670     }
2671
2672     if (NT_SUCCESS(ntStatus))
2673     {
2674         //
2675         // Trim extents if we told the service - the update has done an implicit
2676         // trim at the service.
2677         //
2678         if (bTellService)
2679         {
2680             AFSTrimExtents( pFcb,
2681                             &pFcb->Header.FileSize);
2682         }
2683
2684         KeQuerySystemTime( &pFcb->ObjectInformation->ChangeTime);
2685
2686         SetFlag( pFcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
2687
2688         if (bTellCc &&
2689             CcIsFileCached( pFileObject))
2690         {
2691             CcSetFileSizes( pFileObject,
2692                             (PCC_FILE_SIZES)&pFcb->Header.AllocationSize);
2693         }
2694     }
2695     else
2696     {
2697         //
2698         // Put the saved values back
2699         //
2700         pFcb->Header.ValidDataLength = liSaveVDL;
2701         pFcb->Header.FileSize = liSaveFileSize;
2702         pFcb->Header.AllocationSize = liSaveAlloc;
2703         pFcb->ObjectInformation->EndOfFile = liSaveFileSize;
2704         pFcb->ObjectInformation->AllocationSize = liSaveAlloc;
2705     }
2706
2707     if( bReleasePaging)
2708     {
2709
2710         AFSReleaseResource( &pFcb->NPFcb->PagingResource);
2711     }
2712
2713     return ntStatus;
2714 }
2715
2716 NTSTATUS
2717 AFSSetEndOfFileInfo( IN PIRP Irp,
2718                      IN AFSDirectoryCB *DirectoryCB)
2719 {
2720     NTSTATUS ntStatus = STATUS_SUCCESS;
2721     PFILE_END_OF_FILE_INFORMATION pBuffer;
2722     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2723     PFILE_OBJECT pFileObject = pIrpSp->FileObject;
2724     LARGE_INTEGER liSaveSize;
2725     LARGE_INTEGER liSaveVDL;
2726     LARGE_INTEGER liSaveAlloc;
2727     BOOLEAN bModified = FALSE;
2728     BOOLEAN bReleasePaging = FALSE;
2729     BOOLEAN bTruncated = FALSE;
2730     AFSFcb *pFcb = NULL;
2731     AFSCcb *pCcb = NULL;
2732
2733     pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2734
2735     pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2736
2737     pBuffer = (PFILE_END_OF_FILE_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
2738
2739     liSaveSize = pFcb->Header.FileSize;
2740     liSaveAlloc = pFcb->Header.AllocationSize;
2741     liSaveVDL = pFcb->Header.ValidDataLength;
2742
2743     if( pFcb->Header.FileSize.QuadPart != pBuffer->EndOfFile.QuadPart &&
2744         !pIrpSp->Parameters.SetFile.AdvanceOnly)
2745     {
2746
2747         if( pBuffer->EndOfFile.QuadPart < pFcb->Header.FileSize.QuadPart)
2748         {
2749
2750             // Truncating the file
2751             if( !MmCanFileBeTruncated( pFileObject->SectionObjectPointer,
2752                                        &pBuffer->EndOfFile))
2753             {
2754
2755                 ntStatus = STATUS_USER_MAPPED_FILE;
2756             }
2757             else
2758             {
2759                 //
2760                 // If this is a truncation we need to grab the paging
2761                 // IO resource.
2762                 //
2763                 AFSDbgLogMsg( AFS_SUBSYSTEM_LOCK_PROCESSING,
2764                               AFS_TRACE_LEVEL_VERBOSE,
2765                               "AFSSetAllocationInfo Acquiring Fcb PagingIo lock %08lX EXCL %08lX\n",
2766                               &pFcb->NPFcb->PagingResource,
2767                               PsGetCurrentThread());
2768
2769                 AFSAcquireExcl( &pFcb->NPFcb->PagingResource,
2770                                 TRUE);
2771
2772                 bReleasePaging = TRUE;
2773
2774                 pFcb->Header.AllocationSize = pBuffer->EndOfFile;
2775
2776                 pFcb->Header.FileSize = pBuffer->EndOfFile;
2777
2778                 pFcb->ObjectInformation->EndOfFile = pBuffer->EndOfFile;
2779
2780                 pFcb->ObjectInformation->AllocationSize = pBuffer->EndOfFile;
2781
2782                 if( pFcb->Header.ValidDataLength.QuadPart > pFcb->Header.FileSize.QuadPart)
2783                 {
2784
2785                     pFcb->Header.ValidDataLength = pFcb->Header.FileSize;
2786                 }
2787
2788                 bTruncated = TRUE;
2789
2790                 bModified = TRUE;
2791             }
2792         }
2793         else
2794         {
2795             //
2796             // extending the file, move EOF
2797             //
2798
2799             pFcb->Header.FileSize = pBuffer->EndOfFile;
2800
2801             pFcb->ObjectInformation->EndOfFile = pBuffer->EndOfFile;
2802
2803             if (pFcb->Header.FileSize.QuadPart > pFcb->Header.AllocationSize.QuadPart)
2804             {
2805                 //
2806                 // And Allocation as needed.
2807                 //
2808                 pFcb->Header.AllocationSize = pBuffer->EndOfFile;
2809
2810                 pFcb->ObjectInformation->AllocationSize = pBuffer->EndOfFile;
2811             }
2812
2813             bModified = TRUE;
2814         }
2815     }
2816
2817     if (bModified)
2818     {
2819
2820         KeQuerySystemTime( &pFcb->ObjectInformation->ChangeTime);
2821
2822         SetFlag( pFcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
2823
2824         //
2825         // Tell the server
2826         //
2827
2828         ntStatus = AFSUpdateFileInformation( &pFcb->ObjectInformation->ParentObjectInformation->FileId,
2829                                              pFcb->ObjectInformation,
2830                                              &pFcb->AuthGroup);
2831
2832         if( NT_SUCCESS(ntStatus))
2833         {
2834             //
2835             // We are now good to go so tell CC.
2836             //
2837             CcSetFileSizes( pFileObject,
2838                             (PCC_FILE_SIZES)&pFcb->Header.AllocationSize);
2839
2840             //
2841             // And give up those extents
2842             //
2843             if( bTruncated)
2844             {
2845
2846                 AFSTrimExtents( pFcb,
2847                                 &pFcb->Header.FileSize);
2848             }
2849         }
2850         else
2851         {
2852             pFcb->Header.ValidDataLength = liSaveVDL;
2853             pFcb->Header.FileSize = liSaveSize;
2854             pFcb->Header.AllocationSize = liSaveAlloc;
2855             pFcb->ObjectInformation->EndOfFile = liSaveSize;
2856             pFcb->ObjectInformation->AllocationSize = liSaveAlloc;
2857         }
2858     }
2859
2860     if( bReleasePaging)
2861     {
2862
2863         AFSReleaseResource( &pFcb->NPFcb->PagingResource);
2864     }
2865
2866     return ntStatus;
2867 }
2868
2869 NTSTATUS
2870 AFSProcessShareSetInfo( IN IRP *Irp,
2871                         IN AFSFcb *Fcb,
2872                         IN AFSCcb *Ccb)
2873 {
2874
2875     NTSTATUS ntStatus = STATUS_SUCCESS;
2876     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2877     ULONG ulOutputBufferLen = 0, ulInputBufferLen;
2878     FILE_INFORMATION_CLASS ulFileInformationClass;
2879     void *pPipeInfo = NULL;
2880
2881     __Enter
2882     {
2883         ulFileInformationClass = pIrpSp->Parameters.SetFile.FileInformationClass;
2884
2885         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2886                       AFS_TRACE_LEVEL_VERBOSE,
2887                       "AFSProcessShareSetInfo On pipe %wZ Class %08lX\n",
2888                       &Ccb->DirectoryCB->NameInformation.FileName,
2889                       ulFileInformationClass);
2890
2891         pPipeInfo = AFSLockSystemBuffer( Irp,
2892                                          pIrpSp->Parameters.SetFile.Length);
2893
2894         if( pPipeInfo == NULL)
2895         {
2896
2897             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2898                           AFS_TRACE_LEVEL_ERROR,
2899                           "AFSProcessShareSetInfo Failed to lock buffer on pipe %wZ\n",
2900                           &Ccb->DirectoryCB->NameInformation.FileName);
2901
2902             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2903         }
2904
2905         //
2906         // Send the request to the service
2907         //
2908
2909         ntStatus = AFSNotifySetPipeInfo( Ccb,
2910                                          (ULONG)ulFileInformationClass,
2911                                          pIrpSp->Parameters.SetFile.Length,
2912                                          pPipeInfo);
2913
2914         if( !NT_SUCCESS( ntStatus))
2915         {
2916
2917             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2918                           AFS_TRACE_LEVEL_ERROR,
2919                           "AFSProcessShareSetInfo Failed to send request to service on pipe %wZ Status %08lX\n",
2920                           &Ccb->DirectoryCB->NameInformation.FileName,
2921                           ntStatus);
2922
2923             try_return( ntStatus);
2924         }
2925
2926         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2927                       AFS_TRACE_LEVEL_VERBOSE,
2928                       "AFSProcessShareSetInfo Completed request on pipe %wZ Class %08lX\n",
2929                       &Ccb->DirectoryCB->NameInformation.FileName,
2930                       ulFileInformationClass);
2931
2932 try_exit:
2933
2934         NOTHING;
2935     }
2936
2937     return ntStatus;
2938 }
2939
2940 NTSTATUS
2941 AFSProcessShareQueryInfo( IN IRP *Irp,
2942                           IN AFSFcb *Fcb,
2943                           IN AFSCcb *Ccb)
2944 {
2945
2946     NTSTATUS ntStatus = STATUS_SUCCESS;
2947     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2948     ULONG ulOutputBufferLen = 0, ulInputBufferLen;
2949     FILE_INFORMATION_CLASS ulFileInformationClass;
2950     void *pPipeInfo = NULL;
2951
2952     __Enter
2953     {
2954
2955         ulFileInformationClass = pIrpSp->Parameters.QueryFile.FileInformationClass;
2956
2957         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2958                       AFS_TRACE_LEVEL_VERBOSE,
2959                       "AFSProcessShareQueryInfo On pipe %wZ Class %08lX\n",
2960                       &Ccb->DirectoryCB->NameInformation.FileName,
2961                       ulFileInformationClass);
2962
2963         pPipeInfo = AFSLockSystemBuffer( Irp,
2964                                          pIrpSp->Parameters.QueryFile.Length);
2965
2966         if( pPipeInfo == NULL)
2967         {
2968
2969             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2970                           AFS_TRACE_LEVEL_ERROR,
2971                           "AFSProcessShareQueryInfo Failed to lock buffer on pipe %wZ\n",
2972                           &Ccb->DirectoryCB->NameInformation.FileName);
2973
2974             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2975         }
2976
2977         //
2978         // Send the request to the service
2979         //
2980
2981         ntStatus = AFSNotifyQueryPipeInfo( Ccb,
2982                                            (ULONG)ulFileInformationClass,
2983                                            pIrpSp->Parameters.QueryFile.Length,
2984                                            pPipeInfo,
2985                                            (ULONG *)&Irp->IoStatus.Information);
2986
2987         if( !NT_SUCCESS( ntStatus))
2988         {
2989
2990             AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
2991                           AFS_TRACE_LEVEL_ERROR,
2992                           "AFSProcessShareQueryInfo Failed to send request to service on pipe %wZ Status %08lX\n",
2993                           &Ccb->DirectoryCB->NameInformation.FileName,
2994                           ntStatus);
2995
2996             try_return( ntStatus);
2997         }
2998
2999         AFSDbgLogMsg( AFS_SUBSYSTEM_PIPE_PROCESSING,
3000                       AFS_TRACE_LEVEL_VERBOSE,
3001                       "AFSProcessShareQueryInfo Completed request on pipe %wZ Class %08lX\n",
3002                       &Ccb->DirectoryCB->NameInformation.FileName,
3003                       ulFileInformationClass);
3004
3005 try_exit:
3006
3007         NOTHING;
3008     }
3009
3010     return ntStatus;
3011 }
3012
3013 NTSTATUS
3014 AFSProcessPIOCtlQueryInfo( IN IRP *Irp,
3015                            IN AFSFcb *Fcb,
3016                            IN AFSCcb *Ccb,
3017                            IN OUT LONG *Length)
3018 {
3019
3020     NTSTATUS ntStatus = STATUS_SUCCESS;
3021     PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
3022     FILE_INFORMATION_CLASS ulFileInformationClass;
3023
3024     __Enter
3025     {
3026
3027         ulFileInformationClass = pIrpSp->Parameters.QueryFile.FileInformationClass;
3028
3029         switch( ulFileInformationClass)
3030         {
3031
3032             case FileBasicInformation:
3033             {
3034
3035                 if ( *Length >= sizeof( FILE_BASIC_INFORMATION))
3036                 {
3037                     PFILE_BASIC_INFORMATION pBasic = (PFILE_BASIC_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3038
3039                     pBasic->CreationTime.QuadPart = 0;
3040                     pBasic->LastAccessTime.QuadPart = 0;
3041                     pBasic->ChangeTime.QuadPart = 0;
3042                     pBasic->LastWriteTime.QuadPart = 0;
3043                     pBasic->FileAttributes = FILE_ATTRIBUTE_SYSTEM;
3044
3045                     *Length -= sizeof( FILE_BASIC_INFORMATION);
3046                 }
3047                 else
3048                 {
3049                     ntStatus = STATUS_BUFFER_OVERFLOW;
3050                 }
3051
3052                 break;
3053             }
3054
3055             case FileStandardInformation:
3056             {
3057
3058                 if ( *Length >= sizeof( FILE_STANDARD_INFORMATION))
3059                 {
3060                     PFILE_STANDARD_INFORMATION pStandard = (PFILE_STANDARD_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3061
3062                     pStandard->NumberOfLinks = 1;
3063                     pStandard->DeletePending = 0;
3064                     pStandard->AllocationSize.QuadPart = 4096;
3065                     pStandard->EndOfFile.QuadPart = 4096;
3066                     pStandard->Directory = 0;
3067
3068                     *Length -= sizeof( FILE_STANDARD_INFORMATION);
3069                 }
3070                 else
3071                 {
3072                     ntStatus = STATUS_BUFFER_OVERFLOW;
3073                 }
3074
3075                 break;
3076             }
3077
3078             case FileNameInformation:
3079             {
3080
3081                 ULONG ulCopyLength = 0;
3082                 AFSFcb *pFcb = NULL;
3083                 AFSCcb *pCcb = NULL;
3084                 USHORT usFullNameLength = 0;
3085                 PFILE_NAME_INFORMATION pNameInfo = (PFILE_NAME_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3086                 UNICODE_STRING uniName;
3087
3088                 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
3089                 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
3090
3091                 if( *Length < FIELD_OFFSET( FILE_NAME_INFORMATION, FileName))
3092                 {
3093                     ntStatus = STATUS_BUFFER_TOO_SMALL;
3094                     break;
3095                 }
3096
3097                 RtlZeroMemory( pNameInfo,
3098                                *Length);
3099
3100                 usFullNameLength = sizeof( WCHAR) +
3101                                             AFSServerName.Length +
3102                                             pCcb->FullFileName.Length;
3103
3104                 if( *Length >= (LONG)(FIELD_OFFSET( FILE_NAME_INFORMATION, FileName) + (LONG)usFullNameLength))
3105                 {
3106                     ulCopyLength = (LONG)usFullNameLength;
3107                 }
3108                 else
3109                 {
3110                     ulCopyLength = *Length - FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
3111                     ntStatus = STATUS_BUFFER_OVERFLOW;
3112                 }
3113
3114                 pNameInfo->FileNameLength = (ULONG)usFullNameLength;
3115
3116                 *Length -= FIELD_OFFSET( FILE_NAME_INFORMATION, FileName);
3117
3118                 if( ulCopyLength > 0)
3119                 {
3120
3121                     pNameInfo->FileName[ 0] = L'\\';
3122                     ulCopyLength -= sizeof( WCHAR);
3123
3124                     *Length -= sizeof( WCHAR);
3125
3126                     if( ulCopyLength >= AFSServerName.Length)
3127                     {
3128
3129                         RtlCopyMemory( &pNameInfo->FileName[ 1],
3130                                        AFSServerName.Buffer,
3131                                        AFSServerName.Length);
3132
3133                         ulCopyLength -= AFSServerName.Length;
3134                         *Length -= AFSServerName.Length;
3135
3136                         if( ulCopyLength >= pCcb->FullFileName.Length)
3137                         {
3138
3139                             RtlCopyMemory( &pNameInfo->FileName[ 1 + (AFSServerName.Length/sizeof( WCHAR))],
3140                                            pCcb->FullFileName.Buffer,
3141                                            pCcb->FullFileName.Length);
3142
3143                             ulCopyLength -= pCcb->FullFileName.Length;
3144                             *Length -= pCcb->FullFileName.Length;
3145
3146                             uniName.Length = (USHORT)pNameInfo->FileNameLength;
3147                             uniName.MaximumLength = uniName.Length;
3148                             uniName.Buffer = pNameInfo->FileName;
3149                         }
3150                         else
3151                         {
3152
3153                             RtlCopyMemory( &pNameInfo->FileName[ 1 + (AFSServerName.Length/sizeof( WCHAR))],
3154                                            pCcb->FullFileName.Buffer,
3155                                            ulCopyLength);
3156
3157                             *Length -= ulCopyLength;
3158
3159                             uniName.Length = (USHORT)(sizeof( WCHAR) + AFSServerName.Length + ulCopyLength);
3160                             uniName.MaximumLength = uniName.Length;
3161                             uniName.Buffer = pNameInfo->FileName;
3162                         }
3163
3164                         AFSDbgLogMsg( AFS_SUBSYSTEM_PIOCTL_PROCESSING,
3165                                       AFS_TRACE_LEVEL_VERBOSE,
3166                                       "AFSProcessPIOCtlQueryInfo (FileNameInformation) Returning %wZ\n",
3167                                       &uniName);
3168                     }
3169                 }
3170
3171                 break;
3172             }
3173
3174             case FileInternalInformation:
3175             {
3176
3177                 PFILE_INTERNAL_INFORMATION pInternalInfo = (PFILE_INTERNAL_INFORMATION)Irp->AssociatedIrp.SystemBuffer;
3178
3179                 if( *Length >= sizeof( FILE_INTERNAL_INFORMATION))
3180                 {
3181
3182                     pInternalInfo->IndexNumber.HighPart = 0;
3183
3184                     pInternalInfo->IndexNumber.LowPart = 0;
3185
3186                     *Length -= sizeof( FILE_INTERNAL_INFORMATION);
3187                 }
3188                 else
3189                 {
3190
3191                     ntStatus = STATUS_BUFFER_TOO_SMALL;
3192                 }
3193
3194                 break;
3195             }
3196
3197             default:
3198             {
3199                 ntStatus = STATUS_INVALID_PARAMETER;
3200
3201                 AFSDbgLogMsg( AFS_SUBSYSTEM_PIOCTL_PROCESSING,
3202                               AFS_TRACE_LEVEL_WARNING,
3203                               "AFSProcessPIOCtlQueryInfo Not handling request %08lX\n",
3204                               ulFileInformationClass);
3205
3206                 break;
3207             }
3208         }
3209     }
3210
3211     return ntStatus;
3212 }
3213