2 * Copyright (c) 2008, 2009, 2010, 2011 Kernel Drivers, LLC.
3 * Copyright (c) 2009, 2010, 2011 Your File System, Inc.
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
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
14 * this list of conditions and the following disclaimer in the
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.
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.
39 #include "AFSCommon.h"
43 AFSCachedWrite( IN PDEVICE_OBJECT DeviceObject,
45 IN LARGE_INTEGER StartingByte,
47 IN BOOLEAN ForceFlush);
50 AFSNonCachedWrite( IN PDEVICE_OBJECT DeviceObject,
52 IN LARGE_INTEGER StartingByte,
57 AFSNonCachedWriteDirect( IN PDEVICE_OBJECT DeviceObject,
59 IN LARGE_INTEGER StartingByte,
64 AFSExtendingWrite( IN AFSFcb *Fcb,
65 IN PFILE_OBJECT FileObject,
66 IN LONGLONG NewLength);
73 // This is the dispatch handler for the IRP_MJ_WRITE request
77 // A status is returned for the function
80 AFSWrite( IN PDEVICE_OBJECT LibDeviceObject,
84 UNREFERENCED_PARAMETER(LibDeviceObject);
85 NTSTATUS ntStatus = STATUS_SUCCESS;
90 ntStatus = AFSCommonWrite( AFSRDRDeviceObject, Irp, NULL, FALSE);
92 __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
95 ntStatus = STATUS_INSUFFICIENT_RESOURCES;
102 AFSCommonWrite( IN PDEVICE_OBJECT DeviceObject,
104 IN HANDLE OnBehalfOf,
108 NTSTATUS ntStatus = STATUS_SUCCESS;
109 AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
110 IO_STACK_LOCATION *pIrpSp;
113 AFSNonPagedFcb *pNPFcb = NULL;
114 ULONG ulByteCount = 0;
115 LARGE_INTEGER liStartingByte;
116 PFILE_OBJECT pFileObject;
117 BOOLEAN bPagingIo = FALSE;
118 BOOLEAN bNonCachedIo = FALSE;
119 BOOLEAN bReleaseMain = FALSE;
120 BOOLEAN bReleaseSectionObject = FALSE;
121 BOOLEAN bReleasePaging = FALSE;
122 BOOLEAN bExtendingWrite = FALSE;
123 BOOLEAN bSynchronousFo = FALSE;
124 BOOLEAN bCompleteIrp = TRUE;
125 BOOLEAN bForceFlush = FALSE;
127 HANDLE hCallingUser = OnBehalfOf;
128 ULONGLONG ullProcessId = (ULONGLONG)PsGetCurrentProcessId();
129 AFSDeviceExt *pRDRDevExt = (AFSDeviceExt *)AFSRDRDeviceObject->DeviceExtension;
131 pIrpSp = IoGetCurrentIrpStackLocation( Irp);
136 Irp->IoStatus.Information = 0;
138 pFileObject = pIrpSp->FileObject;
141 // Extract the fileobject references
144 pFcb = (AFSFcb *)pFileObject->FsContext;
145 pCcb = (AFSCcb *)pFileObject->FsContext2;
147 ObReferenceObject( pFileObject);
152 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
153 AFS_TRACE_LEVEL_ERROR,
154 "AFSCommonWrite Attempted write (%p) when pFcb == NULL\n",
157 try_return( ntStatus = STATUS_INVALID_DEVICE_REQUEST);
160 pNPFcb = pFcb->NPFcb;
163 // If we are in shutdown mode then fail the request
166 if( BooleanFlagOn( pDeviceExt->DeviceFlags, AFS_DEVICE_FLAG_REDIRECTOR_SHUTDOWN))
169 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
170 AFS_TRACE_LEVEL_WARNING,
171 "AFSCommonWrite (%p) Open request after shutdown\n",
174 try_return( ntStatus = STATUS_TOO_LATE);
177 liStartingByte = pIrpSp->Parameters.Write.ByteOffset;
178 bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
179 bNonCachedIo = BooleanFlagOn( Irp->Flags, IRP_NOCACHE);
180 ulByteCount = pIrpSp->Parameters.Write.Length;
181 bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
183 if( pFcb->Header.NodeTypeCode != AFS_IOCTL_FCB &&
184 pFcb->Header.NodeTypeCode != AFS_FILE_FCB &&
185 pFcb->Header.NodeTypeCode != AFS_SPECIAL_SHARE_FCB)
188 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
189 AFS_TRACE_LEVEL_ERROR,
190 "AFSCommonWrite Attempted write (%p) on an invalid node type %08lX\n",
192 pFcb->Header.NodeTypeCode));
194 try_return( ntStatus = STATUS_INVALID_DEVICE_REQUEST);
198 // If this is a write against an IOCtl node then handle it
199 // in a different pathway
202 if( pFcb->Header.NodeTypeCode == AFS_IOCTL_FCB)
205 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
206 AFS_TRACE_LEVEL_VERBOSE,
207 "AFSCommonWrite (%p) Processing file (PIOCTL) Offset %0I64X Length %08lX Irp Flags %08lX\n",
209 liStartingByte.QuadPart,
213 ntStatus = AFSIOCtlWrite( DeviceObject,
216 try_return( ntStatus);
218 else if( pFcb->Header.NodeTypeCode == AFS_SPECIAL_SHARE_FCB)
221 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
222 AFS_TRACE_LEVEL_VERBOSE,
223 "AFSCommonWrite (%p) Processing file (SHARE) Offset %0I64X Length %08lX Irp Flags %08lX\n",
225 liStartingByte.QuadPart,
229 ntStatus = AFSShareWrite( DeviceObject,
232 try_return( ntStatus);
236 // Is the Cache not there yet? Exit.
238 if( !BooleanFlagOn( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE) &&
239 !BooleanFlagOn( pRDRDevExt->DeviceFlags, AFS_REDIR_INIT_PERFORM_SERVICE_IO) &&
240 NULL == pDeviceExt->Specific.RDR.CacheFileObject)
243 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
244 AFS_TRACE_LEVEL_ERROR,
245 "AFSCommonWrite (%p) Request failed due to AFS cache closed\n",
248 try_return( ntStatus = STATUS_TOO_LATE );
251 if( pFcb->ObjectInformation->VolumeCB != NULL &&
252 BooleanFlagOn( pFcb->ObjectInformation->VolumeCB->VolumeInformation.FileSystemAttributes, FILE_READ_ONLY_VOLUME))
255 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
256 AFS_TRACE_LEVEL_ERROR,
257 "AFSCommonWrite (%p) Request failed due to read only volume\n",
260 try_return( ntStatus = STATUS_ACCESS_DENIED);
264 // We need to know on whose behalf we have been called (which
265 // we will eventually tell to the server - for non paging
266 // writes). If we were posted then we were told. If this is
267 // the first time we saw the irp then we grab it now.
269 if( NULL == OnBehalfOf )
272 hCallingUser = PsGetCurrentProcessId();
277 hCallingUser = OnBehalfOf;
281 // Check for zero length write
284 if( ulByteCount == 0)
287 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
288 AFS_TRACE_LEVEL_VERBOSE,
289 "AFSCommonWrite (%p) Request completed due to zero length\n",
292 try_return( ntStatus);
296 // Is this Fcb valid???
299 if( BooleanFlagOn( pFcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_OBJECT_INVALID))
302 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
303 AFS_TRACE_LEVEL_ERROR,
304 "AFSCommonWrite (%p) Failing request due to INVALID fcb\n",
307 try_return( ntStatus = STATUS_FILE_DELETED);
310 if( BooleanFlagOn( pCcb->DirectoryCB->Flags, AFS_DIR_ENTRY_DELETED) ||
311 BooleanFlagOn( pFcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_DELETED))
314 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
315 AFS_TRACE_LEVEL_ERROR,
316 "AFSCommonWrite (%p) Request failed due to file deleted\n",
319 try_return( ntStatus = STATUS_FILE_DELETED);
322 if( FlagOn( pIrpSp->MinorFunction, IRP_MN_COMPLETE))
325 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
326 AFS_TRACE_LEVEL_VERBOSE,
327 "AFSCommonWrite (%p) IRP_MN_COMPLETE being processed\n",
330 CcMdlWriteComplete(pFileObject, &pIrpSp->Parameters.Write.ByteOffset, Irp->MdlAddress);
333 // Mdl is now Deallocated
336 Irp->MdlAddress = NULL;
338 try_return( ntStatus = STATUS_SUCCESS );
342 // If we get a non cached IO for a cached file we should do a purge.
343 // For now we will just promote to cached
345 if( NULL != pFileObject->SectionObjectPointer->DataSectionObject && !bPagingIo && bNonCachedIo)
347 bNonCachedIo = FALSE;
351 if ( !bNonCachedIo && !bPagingIo)
354 if( pFileObject->PrivateCacheMap == NULL)
357 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
358 AFS_TRACE_LEVEL_VERBOSE,
359 "AFSCommonWrite Acquiring Fcb SectionObject lock %p EXCL %08lX\n",
360 &pNPFcb->SectionObjectResource,
361 PsGetCurrentThread()));
363 AFSAcquireExcl( &pNPFcb->SectionObjectResource,
366 bReleaseSectionObject = TRUE;
371 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
372 AFS_TRACE_LEVEL_VERBOSE,
373 "AFSCommonWrite Initialize caching on Fcb %p FileObject %p\n",
377 CcInitializeCacheMap( pFileObject,
378 (PCC_FILE_SIZES)&pFcb->Header.AllocationSize,
380 AFSLibCacheManagerCallbacks,
383 CcSetReadAheadGranularity( pFileObject,
384 pDeviceExt->Specific.RDR.MaximumRPCLength);
386 CcSetDirtyPageThreshold( pFileObject,
387 AFS_DIRTY_CHUNK_THRESHOLD * pDeviceExt->Specific.RDR.MaximumRPCLength / 4096);
389 __except( EXCEPTION_EXECUTE_HANDLER)
392 ntStatus = GetExceptionCode();
394 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
395 AFS_TRACE_LEVEL_ERROR,
396 "AFSCommonWrite (%p) Exception thrown while initializing cache map Status %08lX\n",
401 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
402 AFS_TRACE_LEVEL_VERBOSE,
403 "AFSCommonWrite Releasing Fcb SectionObject lock %p EXCL %08lX\n",
404 &pNPFcb->SectionObjectResource,
405 PsGetCurrentThread()));
407 AFSReleaseResource( &pNPFcb->SectionObjectResource);
409 bReleaseSectionObject = FALSE;
411 if( !NT_SUCCESS( ntStatus))
414 try_return( ntStatus);
418 if (!CcCanIWrite( pFileObject,
424 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
425 AFS_TRACE_LEVEL_WARNING,
426 "AFSCommonWrite (FO: %p) CcCanIWrite says No room for Offset %0I64X Length %08lX bytes! Deferring%s\n",
428 liStartingByte.QuadPart,
430 bRetry ? " RETRY" : ""));
432 ntStatus = AFSDeferWrite( DeviceObject, pFileObject, hCallingUser, Irp, ulByteCount, bRetry);
434 if ( STATUS_PENDING == ntStatus)
437 bCompleteIrp = FALSE;
442 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
443 AFS_TRACE_LEVEL_ERROR,
444 "AFSCommonWrite (FO: %p) AFSDeferWrite failure Status %08lX\n",
449 try_return( ntStatus);
454 // Save off the PID if this is not a paging IO
458 ( pFcb->Specific.File.ExtentRequestProcessId == 0 ||
459 ( ullProcessId != (ULONGLONG)AFSSysProcess &&
460 pFcb->Specific.File.ExtentRequestProcessId != ullProcessId)))
463 pFcb->Specific.File.ExtentRequestProcessId = ullProcessId;
465 if( ullProcessId == (ULONGLONG)AFSSysProcess)
467 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
468 AFS_TRACE_LEVEL_WARNING,
469 "%s Setting LastWriterExtentProcessId to system process for Fcb %p\n",
478 // - if Paging then we need to do nothing (the precalls will
479 // have acquired the paging resource), for clarity we will collect
480 // the paging resource
481 // - If extending Write then take the fileresource EX (EOF will change, Allocation will only move out)
482 // - Otherwise we collect the file shared, check against extending and
493 //ASSERT( NULL != OnBehalfOf || ExIsResourceAcquiredLite( &pNPFcb->Resource ));
495 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
496 AFS_TRACE_LEVEL_VERBOSE,
497 "AFSCommonWrite Acquiring Fcb PagingIo lock %p SHARED %08lX\n",
498 &pNPFcb->PagingResource,
499 PsGetCurrentThread()));
501 AFSAcquireShared( &pNPFcb->PagingResource,
504 bReleasePaging = TRUE;
507 // We have the correct lock - we cannot have the wrong one
514 bExtendingWrite = (((liStartingByte.QuadPart + ulByteCount) >=
515 pFcb->Header.FileSize.QuadPart) ||
516 (liStartingByte.LowPart == FILE_WRITE_TO_END_OF_FILE &&
517 liStartingByte.HighPart == -1)) ;
519 if( bExtendingWrite || bNonCachedIo)
522 // Check for lock inversion
525 ASSERT( !ExIsResourceAcquiredLite( &pNPFcb->PagingResource ));
527 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
528 AFS_TRACE_LEVEL_VERBOSE,
529 "AFSCommonWrite Acquiring Fcb lock %p EXCL %08lX\n",
531 PsGetCurrentThread()));
533 AFSAcquireExcl( &pNPFcb->Resource,
538 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
539 AFS_TRACE_LEVEL_VERBOSE,
540 "AFSCommonWrite Acquiring Fcb SectionObject lock %p EXCL %08lX\n",
541 &pNPFcb->SectionObjectResource,
542 PsGetCurrentThread()));
544 AFSAcquireExcl( &pNPFcb->SectionObjectResource,
547 bReleaseSectionObject = TRUE;
549 if (liStartingByte.LowPart == FILE_WRITE_TO_END_OF_FILE &&
550 liStartingByte.HighPart == -1)
552 if (pFcb->Header.ValidDataLength.QuadPart > pFcb->Header.FileSize.QuadPart)
554 liStartingByte = pFcb->Header.ValidDataLength;
558 liStartingByte = pFcb->Header.FileSize;
563 // We have the correct lock - even if we don't end up truncating
569 ASSERT( !ExIsResourceAcquiredLite( &pNPFcb->PagingResource ));
571 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
572 AFS_TRACE_LEVEL_VERBOSE,
573 "AFSCommonWrite Acquiring Fcb lock %p SHARED %08lX\n",
575 PsGetCurrentThread()));
577 AFSAcquireShared( &pNPFcb->Resource,
582 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
583 AFS_TRACE_LEVEL_VERBOSE,
584 "AFSCommonWrite Acquiring Fcb SectionObject lock %p SHARED %08lX\n",
585 &pNPFcb->SectionObjectResource,
586 PsGetCurrentThread()));
588 AFSAcquireShared( &pNPFcb->SectionObjectResource,
591 bReleaseSectionObject = TRUE;
594 // Have things moved? Are we extending? If so, the the lock isn't OK
596 bLockOK = (liStartingByte.QuadPart + ulByteCount) < pFcb->Header.FileSize.QuadPart;
600 AFSReleaseResource( &pNPFcb->Resource);
602 bReleaseMain = FALSE;
613 // Check the BR locks on the file.
616 if ( !FsRtlCheckLockForWriteAccess( &pFcb->Specific.File.FileLock,
620 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
621 AFS_TRACE_LEVEL_ERROR,
622 "AFSCommonWrite (%p) Request failed due to lock conflict\n",
625 try_return( ntStatus = STATUS_FILE_LOCK_CONFLICT);
631 ntStatus = AFSExtendingWrite( pFcb, pFileObject, (liStartingByte.QuadPart + ulByteCount));
633 if( !NT_SUCCESS(ntStatus))
636 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
637 AFS_TRACE_LEVEL_ERROR,
638 "AFSCommonWrite (%p) Failed extending write request Status %08lX\n",
642 try_return( ntStatus );
648 // Fire off the request as appropriate
650 bCompleteIrp = FALSE;
657 // Main and SectionObject resources held Shared
660 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
661 AFS_TRACE_LEVEL_VERBOSE,
662 "AFSCommonWrite (%p) Processing CACHED request Offset %0I64X Len %08lX%s\n",
664 liStartingByte.QuadPart,
666 bRetry ? " RETRY" : ""));
668 ntStatus = AFSCachedWrite( DeviceObject, Irp, liStartingByte, ulByteCount, bForceFlush);
673 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
674 AFS_TRACE_LEVEL_VERBOSE,
675 "AFSCommonWrite (%p) Processing NON-CACHED request Offset %0I64X Len %08lX%s\n",
677 liStartingByte.QuadPart,
679 bRetry ? " RETRY" : ""));
681 if( BooleanFlagOn( pRDRDevExt->DeviceFlags, AFS_DEVICE_FLAG_DIRECT_SERVICE_IO))
684 ntStatus = AFSNonCachedWriteDirect( DeviceObject, Irp, liStartingByte, ulByteCount);
688 ntStatus = AFSNonCachedWrite( DeviceObject, Irp, liStartingByte, ulByteCount);
694 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
695 AFS_TRACE_LEVEL_VERBOSE,
696 "AFSCommonWrite (%p) Process complete Status %08lX\n",
700 if ( NT_SUCCESS( ntStatus))
708 pFileObject->CurrentByteOffset.QuadPart = liStartingByte.QuadPart + ulByteCount;
712 // If this extended the Vdl, then update it accordingly
715 if( liStartingByte.QuadPart + ulByteCount > pFcb->Header.ValidDataLength.QuadPart)
718 pFcb->Header.ValidDataLength.QuadPart = liStartingByte.QuadPart + ulByteCount;
721 if( !BooleanFlagOn( pFcb->Flags, AFS_FCB_FLAG_UPDATE_LAST_WRITE_TIME))
724 SetFlag( pFcb->Flags, AFS_FCB_FLAG_UPDATE_WRITE_TIME);
726 KeQuerySystemTime( &pFcb->ObjectInformation->LastWriteTime);
731 ObDereferenceObject(pFileObject);
733 if( bReleaseSectionObject)
736 AFSReleaseResource( &pNPFcb->SectionObjectResource);
742 AFSReleaseResource( &pNPFcb->PagingResource);
748 AFSReleaseResource( &pNPFcb->Resource);
754 AFSCompleteRequest( Irp,
763 AFSIOCtlWrite( IN PDEVICE_OBJECT DeviceObject,
767 UNREFERENCED_PARAMETER(DeviceObject);
768 NTSTATUS ntStatus = STATUS_SUCCESS;
769 AFSPIOCtlIORequestCB stIORequestCB;
770 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
773 AFSPIOCtlIOResultCB stIOResultCB;
774 ULONG ulBytesReturned = 0;
775 AFSFileID stParentFID;
780 Irp->IoStatus.Information = 0;
782 RtlZeroMemory( &stIORequestCB,
783 sizeof( AFSPIOCtlIORequestCB));
785 if( pIrpSp->Parameters.Write.Length == 0)
789 // Nothing to do in this case
792 try_return( ntStatus);
795 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
797 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
799 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
800 AFS_TRACE_LEVEL_VERBOSE,
801 "AFSIOCtlWrite Acquiring Fcb lock %p SHARED %08lX\n",
802 &pFcb->NPFcb->Resource,
803 PsGetCurrentThread()));
805 AFSAcquireShared( &pFcb->NPFcb->Resource,
809 // Get the parent fid to pass to the cm
812 RtlZeroMemory( &stParentFID,
815 if( BooleanFlagOn( pFcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_PARENT_FID))
819 // The parent directory FID of the node
822 stParentFID = pFcb->ObjectInformation->ParentFileId;
826 // Set the control block up
829 stIORequestCB.RequestId = pCcb->RequestID;
831 if( pFcb->ObjectInformation->VolumeCB != NULL)
833 stIORequestCB.RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
837 // Lock down the buffer
840 stIORequestCB.MappedBuffer = AFSMapToService( Irp,
841 pIrpSp->Parameters.Write.Length);
843 if( stIORequestCB.MappedBuffer == NULL)
846 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
849 stIORequestCB.BufferLength = pIrpSp->Parameters.Write.Length;
851 stIOResultCB.BytesProcessed = 0;
853 ulBytesReturned = sizeof( AFSPIOCtlIOResultCB);
856 // Issue the request to the service
859 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIOCTL_WRITE,
860 AFS_REQUEST_FLAG_SYNCHRONOUS,
866 (void *)&stIORequestCB,
867 sizeof( AFSPIOCtlIORequestCB),
871 if( !NT_SUCCESS( ntStatus))
874 try_return( ntStatus);
878 // Update the length written
881 Irp->IoStatus.Information = stIOResultCB.BytesProcessed;
885 if( stIORequestCB.MappedBuffer != NULL)
888 AFSUnmapServiceMappedBuffer( stIORequestCB.MappedBuffer,
895 AFSReleaseResource( &pFcb->NPFcb->Resource);
903 // This function is called when we know we have to read from the AFS Cache.
905 // It ensures that we have exents for the entirety of the write and
906 // then pins the extents into memory (meaning that although we may
907 // add we will not remove). Then it creates a scatter gather write
908 // and fires off the IRPs
912 AFSNonCachedWrite( IN PDEVICE_OBJECT DeviceObject,
914 IN LARGE_INTEGER StartingByte,
917 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
918 VOID *pSystemBuffer = NULL;
919 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
920 BOOLEAN bLocked = FALSE;
921 BOOLEAN bCompleteIrp = TRUE;
922 AFSGatherIo *pGatherIo = NULL;
923 AFSIoRun *pIoRuns = NULL;
924 AFSIoRun stIoRuns[AFS_MAX_STACK_IO_RUNS];
925 ULONG extentsCount = 0, runCount = 0;
926 AFSExtent *pStartExtent = NULL;
927 AFSExtent *pIgnoreExtent = NULL;
928 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
929 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
930 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
931 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
932 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
933 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
934 LARGE_INTEGER liCurrentTime, liLastRequestTime;
935 AFSDeviceExt *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
936 PFILE_OBJECT pCacheFileObject = NULL;
937 BOOLEAN bDerefExtents = FALSE;
942 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
943 AFS_TRACE_LEVEL_VERBOSE,
944 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
946 StartingByte.HighPart,
947 StartingByte.LowPart,
950 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
953 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
954 AFS_TRACE_LEVEL_ERROR,
955 "AFSNonCachedWrite (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
958 pIrpSp->Parameters.Write.Length,
959 pDevExt->Specific.RDR.MaxIo.QuadPart));
961 try_return( ntStatus = STATUS_UNSUCCESSFUL);
965 // Get the mapping for the buffer
967 pSystemBuffer = AFSLockSystemBuffer( Irp,
970 if( pSystemBuffer == NULL)
973 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
974 AFS_TRACE_LEVEL_ERROR,
975 "AFSNonCachedWrite (%p) Failed to map system buffer\n",
978 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
983 // Provoke a get of the extents - if we need to.
986 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
987 AFS_TRACE_LEVEL_VERBOSE,
988 "AFSNonCachedWrite Requesting extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
989 pFcb->ObjectInformation->FileId.Cell,
990 pFcb->ObjectInformation->FileId.Volume,
991 pFcb->ObjectInformation->FileId.Vnode,
992 pFcb->ObjectInformation->FileId.Unique,
993 StartingByte.QuadPart,
996 ntStatus = AFSRequestExtentsAsync( pFcb,
1001 if (!NT_SUCCESS(ntStatus))
1004 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1005 AFS_TRACE_LEVEL_ERROR,
1006 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1010 try_return( ntStatus);
1013 KeQueryTickCount( &liLastRequestTime);
1018 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1019 AFS_TRACE_LEVEL_VERBOSE,
1020 "AFSNonCachedWrite Acquiring Fcb extents lock %p SHARED %08lX\n",
1021 &pFcb->NPFcb->Specific.File.ExtentsResource,
1022 PsGetCurrentThread()));
1024 ASSERT( !ExIsResourceAcquiredLite( &pFcb->NPFcb->Specific.File.ExtentsResource ));
1026 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource, TRUE );
1029 pStartExtent = NULL;
1030 pIgnoreExtent = NULL;
1032 if ( AFSDoExtentsMapRegion( pFcb, &StartingByte, ByteCount, &pStartExtent, &pIgnoreExtent ))
1037 KeClearEvent( &pFcb->NPFcb->Specific.File.ExtentsRequestComplete );
1039 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1040 AFS_TRACE_LEVEL_VERBOSE,
1041 "AFSNonCachedWrite Releasing(1) Fcb extents lock %p SHARED %08lX\n",
1042 &pFcb->NPFcb->Specific.File.ExtentsResource,
1043 PsGetCurrentThread()));
1045 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1049 // We will re-request the extents after waiting for them
1052 KeQueryTickCount( &liCurrentTime);
1054 if( liCurrentTime.QuadPart - liLastRequestTime.QuadPart >= pControlDevExt->Specific.Control.ExtentRequestTimeCount.QuadPart)
1057 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1058 AFS_TRACE_LEVEL_VERBOSE,
1059 "AFSNonCachedWrite Requesting extents, again, for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1060 pFcb->ObjectInformation->FileId.Cell,
1061 pFcb->ObjectInformation->FileId.Volume,
1062 pFcb->ObjectInformation->FileId.Vnode,
1063 pFcb->ObjectInformation->FileId.Unique,
1064 StartingByte.QuadPart,
1067 ntStatus = AFSRequestExtentsAsync( pFcb,
1072 if (!NT_SUCCESS(ntStatus))
1075 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1076 AFS_TRACE_LEVEL_ERROR,
1077 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1081 try_return( ntStatus);
1084 KeQueryTickCount( &liLastRequestTime);
1088 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1089 AFS_TRACE_LEVEL_VERBOSE,
1090 "AFSNonCachedWrite Waiting for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1091 pFcb->ObjectInformation->FileId.Cell,
1092 pFcb->ObjectInformation->FileId.Volume,
1093 pFcb->ObjectInformation->FileId.Vnode,
1094 pFcb->ObjectInformation->FileId.Unique,
1095 StartingByte.QuadPart,
1102 ntStatus = AFSWaitForExtentMapping ( pFcb, pCcb);
1104 if (!NT_SUCCESS(ntStatus))
1107 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1108 AFS_TRACE_LEVEL_ERROR,
1109 "AFSNonCachedWrite Failed wait for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1110 pFcb->ObjectInformation->FileId.Cell,
1111 pFcb->ObjectInformation->FileId.Volume,
1112 pFcb->ObjectInformation->FileId.Vnode,
1113 pFcb->ObjectInformation->FileId.Unique,
1114 StartingByte.QuadPart,
1118 try_return( ntStatus);
1123 // As per the read path -
1126 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1127 AFS_TRACE_LEVEL_VERBOSE,
1128 "AFSNonCachedWrite Extents located for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1129 pFcb->ObjectInformation->FileId.Cell,
1130 pFcb->ObjectInformation->FileId.Volume,
1131 pFcb->ObjectInformation->FileId.Vnode,
1132 pFcb->ObjectInformation->FileId.Unique,
1133 StartingByte.QuadPart,
1136 ntStatus = AFSGetExtents( pFcb,
1143 if (!NT_SUCCESS(ntStatus))
1146 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1147 AFS_TRACE_LEVEL_ERROR,
1148 "AFSNonCachedWrite (%p) Failed to retrieve mapped extents Status %08lX\n",
1152 try_return( ntStatus );
1155 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1156 AFS_TRACE_LEVEL_VERBOSE,
1157 "AFSNonCachedWrite (%p) Successfully retrieved map extents count %d run count %d\n",
1162 if( BooleanFlagOn( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE))
1165 Irp->IoStatus.Information = ByteCount;
1169 // Setup the MD5 for each extent
1172 AFSSetupMD5Hash( pFcb,
1180 ntStatus = AFSProcessExtentRun( pSystemBuffer,
1186 if (!NT_SUCCESS(ntStatus))
1189 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1190 AFS_TRACE_LEVEL_ERROR,
1191 "AFSNonCachedWrite (%p) Failed to process extent run for non-persistent cache Status %08lX\n",
1196 try_return( ntStatus);
1200 // Retrieve the cache file object
1203 pCacheFileObject = AFSReferenceCacheFileObject();
1205 if( pCacheFileObject == NULL)
1208 ntStatus = STATUS_DEVICE_NOT_READY;
1210 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1211 AFS_TRACE_LEVEL_ERROR,
1212 "AFSNonCachedWrite Failed to retrieve cache fileobject for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1213 pFcb->ObjectInformation->FileId.Cell,
1214 pFcb->ObjectInformation->FileId.Volume,
1215 pFcb->ObjectInformation->FileId.Vnode,
1216 pFcb->ObjectInformation->FileId.Unique,
1217 StartingByte.QuadPart,
1221 try_return( ntStatus);
1224 if (runCount > AFS_MAX_STACK_IO_RUNS)
1227 pIoRuns = (AFSIoRun*) AFSExAllocatePoolWithTag( PagedPool,
1228 runCount * sizeof( AFSIoRun ),
1230 if (NULL == pIoRuns)
1233 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1234 AFS_TRACE_LEVEL_ERROR,
1235 "AFSNonCachedWrite (%p) Failed to allocate IO run block\n",
1238 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1247 RtlZeroMemory( pIoRuns, runCount * sizeof( AFSIoRun ));
1249 ntStatus = AFSSetupIoRun( IoGetRelatedDeviceObject( pCacheFileObject),
1258 if (!NT_SUCCESS(ntStatus))
1261 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1262 AFS_TRACE_LEVEL_ERROR,
1263 "AFSNonCachedWrite (%p) Failed to initialize IO run block Status %08lX\n",
1267 try_return( ntStatus );
1270 AFSReferenceActiveExtents( pStartExtent,
1273 bDerefExtents = TRUE;
1275 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1276 AFS_TRACE_LEVEL_VERBOSE,
1277 "AFSNonCachedWrite Releasing(2) Fcb extents lock %p SHARED %08lX\n",
1278 &pFcb->NPFcb->Specific.File.ExtentsResource,
1279 PsGetCurrentThread()));
1281 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1284 pGatherIo = (AFSGatherIo*) AFSExAllocatePoolWithTag( NonPagedPool,
1285 sizeof( AFSGatherIo),
1288 if (NULL == pGatherIo)
1291 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1292 AFS_TRACE_LEVEL_ERROR,
1293 "AFSNonCachedWrite (%p) Failed to allocate IO gather block\n",
1296 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1297 AFS_TRACE_LEVEL_VERBOSE,
1298 "AFSNonCachedWrite Acquiring(1) Fcb extents lock %p SHARED %08lX\n",
1299 &pFcb->NPFcb->Specific.File.ExtentsResource,
1300 PsGetCurrentThread()));
1302 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1306 AFSDereferenceActiveExtents( pStartExtent,
1309 try_return (ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1312 RtlZeroMemory( pGatherIo, sizeof( AFSGatherIo ));
1315 // Initialize count to 1, that was we won't get an early
1316 // completion if the first irp completes before the second is
1319 pGatherIo->Count = 1;
1320 pGatherIo->Status = STATUS_SUCCESS;
1321 pGatherIo->MasterIrp = Irp;
1322 pGatherIo->Synchronous = TRUE;
1323 pGatherIo->CompleteMasterIrp = FALSE;
1325 bCompleteIrp = TRUE;
1327 if( pGatherIo->Synchronous)
1329 KeInitializeEvent( &pGatherIo->Event, NotificationEvent, FALSE );
1334 // Setup the MD5 for each extent
1337 AFSSetupMD5Hash( pFcb,
1346 // Pre-emptively set up the count
1349 Irp->IoStatus.Information = ByteCount;
1351 ntStatus = AFSQueueStartIos( pCacheFileObject,
1353 IRP_WRITE_OPERATION | IRP_SYNCHRONOUS_API,
1358 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1359 AFS_TRACE_LEVEL_VERBOSE,
1360 "AFSNonCachedWrite (%p) AFSStartIos completed Status %08lX\n",
1364 if( !NT_SUCCESS( ntStatus))
1367 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1368 AFS_TRACE_LEVEL_VERBOSE,
1369 "AFSNonCachedWrite Acquiring(2) Fcb extents lock %p SHARED %08lX\n",
1370 &pFcb->NPFcb->Specific.File.ExtentsResource,
1371 PsGetCurrentThread()));
1373 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1377 AFSDereferenceActiveExtents( pStartExtent,
1380 try_return( ntStatus);
1384 // Wait for completion of All IOs we started.
1387 ntStatus = KeWaitForSingleObject( &pGatherIo->Event,
1393 if( NT_SUCCESS( ntStatus))
1396 ntStatus = pGatherIo->Status;
1399 if( !NT_SUCCESS( ntStatus))
1402 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1403 AFS_TRACE_LEVEL_VERBOSE,
1404 "AFSNonCachedWrite Acquiring(3) Fcb extents lock %p SHARED %08lX\n",
1405 &pFcb->NPFcb->Specific.File.ExtentsResource,
1406 PsGetCurrentThread()));
1408 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1412 AFSDereferenceActiveExtents( pStartExtent,
1415 try_return( ntStatus);
1420 if( NT_SUCCESS( ntStatus) &&
1421 pStartExtent != NULL &&
1422 Irp->IoStatus.Information > 0)
1428 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1434 // Since this is dirty we can mark the extents dirty now.
1435 // AFSMarkDirty will dereference the extents. Do not call
1436 // AFSDereferenceActiveExtents() in this code path.
1448 // This was an uncached user write - tell the server to do
1449 // the flush when the worker thread next wakes up
1451 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1455 if( pCacheFileObject != NULL)
1457 AFSReleaseCacheFileObject( pCacheFileObject);
1460 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1461 AFS_TRACE_LEVEL_VERBOSE,
1462 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1464 StartingByte.HighPart,
1465 StartingByte.LowPart,
1469 if (NT_SUCCESS(ntStatus) &&
1474 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1480 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1481 AFS_TRACE_LEVEL_VERBOSE,
1482 "AFSNonCachedWrite Releasing Fcb extents lock %p SHARED %08lX\n",
1483 &pFcb->NPFcb->Specific.File.ExtentsResource,
1484 PsGetCurrentThread()));
1486 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1491 AFSExFreePoolWithTag(pGatherIo, AFS_GATHER_TAG);
1494 if( NULL != pIoRuns &&
1495 stIoRuns != pIoRuns)
1497 AFSExFreePoolWithTag(pIoRuns, AFS_IO_RUN_TAG);
1503 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1504 AFS_TRACE_LEVEL_VERBOSE,
1505 "AFSNonCachedWrite Completing Irp %p Status %08lX Info %08lX\n",
1508 Irp->IoStatus.Information));
1510 AFSCompleteRequest( Irp, ntStatus);
1519 AFSNonCachedWriteDirect( IN PDEVICE_OBJECT DeviceObject,
1521 IN LARGE_INTEGER StartingByte,
1524 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
1525 VOID *pSystemBuffer = NULL;
1526 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
1527 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1528 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1529 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1530 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
1531 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1532 BOOLEAN bNoIntermediateBuffering = BooleanFlagOn( pFileObject->Flags, FO_NO_INTERMEDIATE_BUFFERING);
1533 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
1534 AFSFileIOCB stFileIORequest;
1535 AFSFileIOResultCB stFileIOResult;
1536 ULONG ulResultLen = 0;
1541 Irp->IoStatus.Information = 0;
1543 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1544 AFS_TRACE_LEVEL_VERBOSE,
1545 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
1547 StartingByte.HighPart,
1548 StartingByte.LowPart,
1551 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
1554 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1555 AFS_TRACE_LEVEL_ERROR,
1556 "AFSNonCachedWriteDirect (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
1559 pIrpSp->Parameters.Write.Length,
1560 pDevExt->Specific.RDR.MaxIo.QuadPart));
1562 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1566 // Get the mapping for the buffer
1568 pSystemBuffer = AFSLockSystemBuffer( Irp,
1571 if( pSystemBuffer == NULL)
1574 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1575 AFS_TRACE_LEVEL_ERROR,
1576 "AFSNonCachedWriteDirect (%p) Failed to map system buffer\n",
1579 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1583 // Issue the request at the service for processing
1586 ulResultLen = sizeof( AFSFileIOResultCB);
1588 RtlZeroMemory( &stFileIORequest,
1589 sizeof( AFSFileIOCB));
1591 RtlZeroMemory( &stFileIOResult,
1592 sizeof( AFSFileIOResultCB));
1594 stFileIORequest.SystemIOBuffer = pSystemBuffer;
1596 stFileIORequest.SystemIOBufferMdl = Irp->MdlAddress;
1598 stFileIORequest.IOLength = ByteCount;
1600 stFileIORequest.IOOffset = StartingByte;
1602 ulFlags = AFS_REQUEST_FLAG_SYNCHRONOUS;
1604 if ( bNoIntermediateBuffering)
1607 ulFlags |= AFS_REQUEST_FLAG_CACHE_BYPASS;
1611 // Update file metadata
1614 stFileIORequest.EndOfFile = pFcb->ObjectInformation->EndOfFile;
1616 stFileIORequest.CreateTime = pFcb->ObjectInformation->CreationTime;
1618 stFileIORequest.ChangeTime = pFcb->ObjectInformation->ChangeTime;
1620 stFileIORequest.LastAccessTime = pFcb->ObjectInformation->LastAccessTime;
1622 stFileIORequest.LastWriteTime = pFcb->ObjectInformation->LastWriteTime;
1625 // Write the data to the service
1628 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PROCESS_WRITE_FILE,
1631 &pCcb->DirectoryCB->NameInformation.FileName,
1632 &pFcb->ObjectInformation->FileId,
1633 pFcb->ObjectInformation->VolumeCB->VolumeInformation.Cell,
1634 pFcb->ObjectInformation->VolumeCB->VolumeInformation.CellLength,
1636 sizeof( AFSFileIOCB),
1640 if( NT_SUCCESS( ntStatus))
1643 Irp->IoStatus.Information = (ULONG_PTR)stFileIOResult.Length;
1648 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1649 AFS_TRACE_LEVEL_ERROR,
1650 "AFSNonCachedWriteDirect (%p) Failed to send write to service Status %08lX\n",
1657 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1658 AFS_TRACE_LEVEL_VERBOSE,
1659 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1661 StartingByte.HighPart,
1662 StartingByte.LowPart,
1666 if (NT_SUCCESS(ntStatus) &&
1671 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1674 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1675 AFS_TRACE_LEVEL_VERBOSE,
1676 "AFSNonCachedWriteDirect Completing Irp %p Status %08lX Info %08lX\n",
1679 Irp->IoStatus.Information));
1681 AFSCompleteRequest( Irp, ntStatus);
1689 AFSCachedWrite( IN PDEVICE_OBJECT DeviceObject,
1691 IN LARGE_INTEGER StartingByte,
1693 IN BOOLEAN ForceFlush)
1695 UNREFERENCED_PARAMETER(DeviceObject);
1696 PVOID pSystemBuffer = NULL;
1697 NTSTATUS ntStatus = STATUS_SUCCESS;
1698 IO_STATUS_BLOCK iosbFlush;
1699 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1700 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1701 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1702 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1703 ULONG ulCurrentIO = 0, ulTotalLen = ByteCount;
1704 PMDL pCurrentMdl = Irp->MdlAddress;
1705 LARGE_INTEGER liCurrentOffset;
1710 Irp->IoStatus.Information = 0;
1712 if( BooleanFlagOn( pIrpSp->MinorFunction, IRP_MN_MDL))
1718 CcPrepareMdlWrite( pFileObject,
1724 ntStatus = Irp->IoStatus.Status;
1726 __except( EXCEPTION_EXECUTE_HANDLER)
1728 ntStatus = GetExceptionCode();
1730 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1731 AFS_TRACE_LEVEL_ERROR,
1732 "AFSCachedWrite (%p) Exception thrown while preparing mdl write Status %08lX\n",
1737 if( !NT_SUCCESS( ntStatus))
1741 // Free up any potentially allocated mdl's
1744 CcMdlWriteComplete( pFileObject,
1748 Irp->MdlAddress = NULL;
1750 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1751 AFS_TRACE_LEVEL_ERROR,
1752 "AFSCachedWrite (%p) Failed to process MDL write Status %08lX\n",
1757 try_return( ntStatus);
1760 liCurrentOffset.QuadPart = StartingByte.QuadPart;
1762 while( ulTotalLen > 0)
1765 ntStatus = STATUS_SUCCESS;
1767 if( pCurrentMdl != NULL)
1770 pSystemBuffer = MmGetSystemAddressForMdlSafe( pCurrentMdl,
1771 NormalPagePriority);
1773 ulCurrentIO = MmGetMdlByteCount( pCurrentMdl);
1775 if( ulCurrentIO > ulTotalLen)
1777 ulCurrentIO = ulTotalLen;
1783 pSystemBuffer = AFSLockSystemBuffer( Irp,
1786 ulCurrentIO = ulTotalLen;
1789 if( pSystemBuffer == NULL)
1792 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1793 AFS_TRACE_LEVEL_ERROR,
1794 "AFSCachedWrite (%p) Failed to lock system buffer\n",
1797 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1803 if( !CcCopyWrite( pFileObject,
1810 // Failed to process request.
1813 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
1814 AFS_TRACE_LEVEL_ERROR,
1815 "AFSCachedWrite (%p) Failed to issue CcCopyWrite %wZ @ %0I64X Status %08lX\n",
1817 &pFileObject->FileName,
1818 liCurrentOffset.QuadPart,
1819 Irp->IoStatus.Status));
1821 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1824 __except( EXCEPTION_EXECUTE_HANDLER)
1827 ntStatus = GetExceptionCode();
1829 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1830 AFS_TRACE_LEVEL_ERROR,
1831 "AFSCachedWrite (%p) CcCopyWrite Threw exception %wZ @ %0I64X Status %08lX\n",
1833 &pFileObject->FileName,
1834 liCurrentOffset.QuadPart,
1838 if( !NT_SUCCESS( ntStatus))
1840 try_return( ntStatus);
1844 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1848 // We have detected a file we do a write through with.
1851 CcFlushCache(&pFcb->NPFcb->SectionObjectPointers,
1856 if( !NT_SUCCESS( iosbFlush.Status))
1859 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1860 AFS_TRACE_LEVEL_ERROR,
1861 "AFSCachedWrite (%p) CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
1863 &pFileObject->FileName,
1864 pFcb->ObjectInformation->FileId.Cell,
1865 pFcb->ObjectInformation->FileId.Volume,
1866 pFcb->ObjectInformation->FileId.Vnode,
1867 pFcb->ObjectInformation->FileId.Unique,
1869 iosbFlush.Information));
1871 try_return( ntStatus = iosbFlush.Status);
1875 if( ulTotalLen <= ulCurrentIO)
1880 liCurrentOffset.QuadPart += ulCurrentIO;
1882 ulTotalLen -= ulCurrentIO;
1884 pCurrentMdl = pCurrentMdl->Next;
1889 if( NT_SUCCESS( ntStatus))
1892 Irp->IoStatus.Information = ByteCount;
1895 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1898 // Write through asked for... Set things so that we get
1899 // flush when the worker thread next wakes up
1901 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1905 AFSCompleteRequest( Irp,
1914 AFSExtendingWrite( IN AFSFcb *Fcb,
1915 IN PFILE_OBJECT FileObject,
1916 IN LONGLONG NewLength)
1918 LARGE_INTEGER liSaveFileSize = Fcb->Header.FileSize;
1919 LARGE_INTEGER liSaveAllocation = Fcb->Header.AllocationSize;
1920 NTSTATUS ntStatus = STATUS_SUCCESS;
1921 AFSCcb *pCcb = (AFSCcb *)FileObject->FsContext2;
1923 if( NewLength > Fcb->Header.AllocationSize.QuadPart)
1926 Fcb->Header.AllocationSize.QuadPart = NewLength;
1928 Fcb->ObjectInformation->AllocationSize = Fcb->Header.AllocationSize;
1931 if( NewLength > Fcb->Header.FileSize.QuadPart)
1934 Fcb->Header.FileSize.QuadPart = NewLength;
1936 Fcb->ObjectInformation->EndOfFile = Fcb->Header.FileSize;
1943 ntStatus = AFSUpdateFileInformation( &Fcb->ObjectInformation->ParentFileId,
1944 Fcb->ObjectInformation,
1947 if (NT_SUCCESS(ntStatus))
1950 KeQuerySystemTime( &Fcb->ObjectInformation->ChangeTime);
1952 SetFlag( Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
1955 // If the file is currently cached, then let the MM know about the extension
1958 if( CcIsFileCached( FileObject))
1960 CcSetFileSizes( FileObject,
1961 (PCC_FILE_SIZES)&Fcb->Header.AllocationSize);
1966 Fcb->Header.FileSize = liSaveFileSize;
1967 Fcb->Header.AllocationSize = liSaveAllocation;
1971 // DownConvert file resource to shared
1973 ExConvertExclusiveToSharedLite( &Fcb->NPFcb->Resource);
1979 AFSShareWrite( IN PDEVICE_OBJECT DeviceObject,
1983 UNREFERENCED_PARAMETER(DeviceObject);
1984 NTSTATUS ntStatus = STATUS_SUCCESS;
1985 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1986 AFSFcb *pFcb = NULL;
1987 AFSCcb *pCcb = NULL;
1988 AFSPipeIORequestCB *pIoRequest = NULL;
1989 void *pBuffer = NULL;
1990 AFSPipeIOResultCB stIoResult;
1991 ULONG ulBytesReturned = 0;
1996 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
1998 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2000 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2001 AFS_TRACE_LEVEL_VERBOSE,
2002 "AFSShareWrite On pipe %wZ Length %08lX\n",
2003 &pCcb->DirectoryCB->NameInformation.FileName,
2004 pIrpSp->Parameters.Write.Length));
2006 if( pIrpSp->Parameters.Write.Length == 0)
2010 // Nothing to do in this case
2013 try_return( ntStatus);
2017 // Retrieve the buffer for the read request
2020 pBuffer = AFSLockSystemBuffer( Irp,
2021 pIrpSp->Parameters.Write.Length);
2023 if( pBuffer == NULL)
2026 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2027 AFS_TRACE_LEVEL_ERROR,
2028 "AFSShareWrite Failed to map buffer on pipe %wZ\n",
2029 &pCcb->DirectoryCB->NameInformation.FileName));
2031 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2034 AFSAcquireShared( &pFcb->NPFcb->Resource,
2037 pIoRequest = (AFSPipeIORequestCB *)AFSExAllocatePoolWithTag( PagedPool,
2038 sizeof( AFSPipeIORequestCB) +
2039 pIrpSp->Parameters.Write.Length,
2040 AFS_GENERIC_MEMORY_14_TAG);
2042 if( pIoRequest == NULL)
2045 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2048 RtlZeroMemory( pIoRequest,
2049 sizeof( AFSPipeIORequestCB) + pIrpSp->Parameters.Write.Length);
2051 pIoRequest->RequestId = pCcb->RequestID;
2053 pIoRequest->RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
2055 pIoRequest->BufferLength = pIrpSp->Parameters.Write.Length;
2057 RtlCopyMemory( (void *)((char *)pIoRequest + sizeof( AFSPipeIORequestCB)),
2059 pIrpSp->Parameters.Write.Length);
2061 stIoResult.BytesProcessed = 0;
2063 ulBytesReturned = sizeof( AFSPipeIOResultCB);
2066 // Issue the open request to the service
2069 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIPE_WRITE,
2070 AFS_REQUEST_FLAG_SYNCHRONOUS,
2072 &pCcb->DirectoryCB->NameInformation.FileName,
2077 sizeof( AFSPipeIORequestCB) +
2078 pIrpSp->Parameters.Write.Length,
2082 if( !NT_SUCCESS( ntStatus))
2085 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
2086 AFS_TRACE_LEVEL_ERROR,
2087 "AFSShareWrite (%p) Failed service write Status %08lX\n",
2091 try_return( ntStatus);
2094 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2095 AFS_TRACE_LEVEL_VERBOSE,
2096 "AFSShareWrite Completed on pipe %wZ Length read %08lX\n",
2097 &pCcb->DirectoryCB->NameInformation.FileName,
2098 stIoResult.BytesProcessed));
2100 Irp->IoStatus.Information = stIoResult.BytesProcessed;
2107 AFSReleaseResource( &pFcb->NPFcb->Resource);
2110 if( pIoRequest != NULL)
2113 AFSExFreePoolWithTag( pIoRequest, AFS_GENERIC_MEMORY_14_TAG);