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 if ( !bPagingIo && bNonCachedIo && CcIsFileCached( pFileObject) &&
732 pNPFcb->SectionObjectPointers.DataSectionObject != NULL &&
733 bReleaseSectionObject)
736 // Regardless of whether or not the a non-paging non-cached write
737 // succeeds or fails, if the file is cached the contents of the
738 // cache are no longer up to date. A CcPurgeCacheSection must be
739 // performed to force subsequent cached reads to obtain the data
742 // The Fcb Resource is dropped in order to permit filters that perform
743 // an open via a worker thread in response to a purge to do so without
744 // deadlocking. The SectionObjectResource is held across the purge to
745 // prevent racing with other cache operations.
751 AFSReleaseResource( &pNPFcb->Resource);
753 bReleaseMain = FALSE;
756 if ( !CcPurgeCacheSection( &pNPFcb->SectionObjectPointers,
762 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
763 AFS_TRACE_LEVEL_WARNING,
764 "AFSCommonWrite CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
765 pFcb->ObjectInformation->FileId.Cell,
766 pFcb->ObjectInformation->FileId.Volume,
767 pFcb->ObjectInformation->FileId.Vnode,
768 pFcb->ObjectInformation->FileId.Unique));
770 SetFlag( pFcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
774 ObDereferenceObject(pFileObject);
776 if( bReleaseSectionObject)
779 AFSReleaseResource( &pNPFcb->SectionObjectResource);
785 AFSReleaseResource( &pNPFcb->PagingResource);
791 AFSReleaseResource( &pNPFcb->Resource);
797 AFSCompleteRequest( Irp,
806 AFSIOCtlWrite( IN PDEVICE_OBJECT DeviceObject,
810 UNREFERENCED_PARAMETER(DeviceObject);
811 NTSTATUS ntStatus = STATUS_SUCCESS;
812 AFSPIOCtlIORequestCB stIORequestCB;
813 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
816 AFSPIOCtlIOResultCB stIOResultCB;
817 ULONG ulBytesReturned = 0;
818 AFSFileID stParentFID;
823 Irp->IoStatus.Information = 0;
825 RtlZeroMemory( &stIORequestCB,
826 sizeof( AFSPIOCtlIORequestCB));
828 if( pIrpSp->Parameters.Write.Length == 0)
832 // Nothing to do in this case
835 try_return( ntStatus);
838 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
840 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
842 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
843 AFS_TRACE_LEVEL_VERBOSE,
844 "AFSIOCtlWrite Acquiring Fcb lock %p SHARED %08lX\n",
845 &pFcb->NPFcb->Resource,
846 PsGetCurrentThread()));
848 AFSAcquireShared( &pFcb->NPFcb->Resource,
852 // Get the parent fid to pass to the cm
855 RtlZeroMemory( &stParentFID,
858 if( BooleanFlagOn( pFcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_PARENT_FID))
862 // The parent directory FID of the node
865 stParentFID = pFcb->ObjectInformation->ParentFileId;
869 // Set the control block up
872 stIORequestCB.RequestId = pCcb->RequestID;
874 if( pFcb->ObjectInformation->VolumeCB != NULL)
876 stIORequestCB.RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
880 // Lock down the buffer
883 stIORequestCB.MappedBuffer = AFSMapToService( Irp,
884 pIrpSp->Parameters.Write.Length);
886 if( stIORequestCB.MappedBuffer == NULL)
889 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
892 stIORequestCB.BufferLength = pIrpSp->Parameters.Write.Length;
894 stIOResultCB.BytesProcessed = 0;
896 ulBytesReturned = sizeof( AFSPIOCtlIOResultCB);
899 // Issue the request to the service
902 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIOCTL_WRITE,
903 AFS_REQUEST_FLAG_SYNCHRONOUS,
909 (void *)&stIORequestCB,
910 sizeof( AFSPIOCtlIORequestCB),
914 if( !NT_SUCCESS( ntStatus))
917 try_return( ntStatus);
921 // Update the length written
924 Irp->IoStatus.Information = stIOResultCB.BytesProcessed;
928 if( stIORequestCB.MappedBuffer != NULL)
931 AFSUnmapServiceMappedBuffer( stIORequestCB.MappedBuffer,
938 AFSReleaseResource( &pFcb->NPFcb->Resource);
946 // This function is called when we know we have to read from the AFS Cache.
948 // It ensures that we have exents for the entirety of the write and
949 // then pins the extents into memory (meaning that although we may
950 // add we will not remove). Then it creates a scatter gather write
951 // and fires off the IRPs
955 AFSNonCachedWrite( IN PDEVICE_OBJECT DeviceObject,
957 IN LARGE_INTEGER StartingByte,
960 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
961 VOID *pSystemBuffer = NULL;
962 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
963 BOOLEAN bLocked = FALSE;
964 BOOLEAN bCompleteIrp = TRUE;
965 AFSGatherIo *pGatherIo = NULL;
966 AFSIoRun *pIoRuns = NULL;
967 AFSIoRun stIoRuns[AFS_MAX_STACK_IO_RUNS];
968 ULONG extentsCount = 0, runCount = 0;
969 AFSExtent *pStartExtent = NULL;
970 AFSExtent *pIgnoreExtent = NULL;
971 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
972 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
973 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
974 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
975 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
976 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
977 LARGE_INTEGER liCurrentTime, liLastRequestTime;
978 AFSDeviceExt *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
979 PFILE_OBJECT pCacheFileObject = NULL;
980 BOOLEAN bDerefExtents = FALSE;
985 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
986 AFS_TRACE_LEVEL_VERBOSE,
987 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
989 StartingByte.HighPart,
990 StartingByte.LowPart,
993 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
996 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
997 AFS_TRACE_LEVEL_ERROR,
998 "AFSNonCachedWrite (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
1001 pIrpSp->Parameters.Write.Length,
1002 pDevExt->Specific.RDR.MaxIo.QuadPart));
1004 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1008 // Get the mapping for the buffer
1010 pSystemBuffer = AFSLockSystemBuffer( Irp,
1013 if( pSystemBuffer == NULL)
1016 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1017 AFS_TRACE_LEVEL_ERROR,
1018 "AFSNonCachedWrite (%p) Failed to map system buffer\n",
1021 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1026 // Provoke a get of the extents - if we need to.
1029 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1030 AFS_TRACE_LEVEL_VERBOSE,
1031 "AFSNonCachedWrite Requesting extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1032 pFcb->ObjectInformation->FileId.Cell,
1033 pFcb->ObjectInformation->FileId.Volume,
1034 pFcb->ObjectInformation->FileId.Vnode,
1035 pFcb->ObjectInformation->FileId.Unique,
1036 StartingByte.QuadPart,
1039 ntStatus = AFSRequestExtentsAsync( pFcb,
1044 if (!NT_SUCCESS(ntStatus))
1047 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1048 AFS_TRACE_LEVEL_ERROR,
1049 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1053 try_return( ntStatus);
1056 KeQueryTickCount( &liLastRequestTime);
1061 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1062 AFS_TRACE_LEVEL_VERBOSE,
1063 "AFSNonCachedWrite Acquiring Fcb extents lock %p SHARED %08lX\n",
1064 &pFcb->NPFcb->Specific.File.ExtentsResource,
1065 PsGetCurrentThread()));
1067 ASSERT( !ExIsResourceAcquiredLite( &pFcb->NPFcb->Specific.File.ExtentsResource ));
1069 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource, TRUE );
1072 pStartExtent = NULL;
1073 pIgnoreExtent = NULL;
1075 if ( AFSDoExtentsMapRegion( pFcb, &StartingByte, ByteCount, &pStartExtent, &pIgnoreExtent ))
1080 KeClearEvent( &pFcb->NPFcb->Specific.File.ExtentsRequestComplete );
1082 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1083 AFS_TRACE_LEVEL_VERBOSE,
1084 "AFSNonCachedWrite Releasing(1) Fcb extents lock %p SHARED %08lX\n",
1085 &pFcb->NPFcb->Specific.File.ExtentsResource,
1086 PsGetCurrentThread()));
1088 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1092 // We will re-request the extents after waiting for them
1095 KeQueryTickCount( &liCurrentTime);
1097 if( liCurrentTime.QuadPart - liLastRequestTime.QuadPart >= pControlDevExt->Specific.Control.ExtentRequestTimeCount.QuadPart)
1100 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1101 AFS_TRACE_LEVEL_VERBOSE,
1102 "AFSNonCachedWrite Requesting extents, again, for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1103 pFcb->ObjectInformation->FileId.Cell,
1104 pFcb->ObjectInformation->FileId.Volume,
1105 pFcb->ObjectInformation->FileId.Vnode,
1106 pFcb->ObjectInformation->FileId.Unique,
1107 StartingByte.QuadPart,
1110 ntStatus = AFSRequestExtentsAsync( pFcb,
1115 if (!NT_SUCCESS(ntStatus))
1118 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1119 AFS_TRACE_LEVEL_ERROR,
1120 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1124 try_return( ntStatus);
1127 KeQueryTickCount( &liLastRequestTime);
1131 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1132 AFS_TRACE_LEVEL_VERBOSE,
1133 "AFSNonCachedWrite Waiting for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1134 pFcb->ObjectInformation->FileId.Cell,
1135 pFcb->ObjectInformation->FileId.Volume,
1136 pFcb->ObjectInformation->FileId.Vnode,
1137 pFcb->ObjectInformation->FileId.Unique,
1138 StartingByte.QuadPart,
1145 ntStatus = AFSWaitForExtentMapping ( pFcb, pCcb);
1147 if (!NT_SUCCESS(ntStatus))
1150 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1151 AFS_TRACE_LEVEL_ERROR,
1152 "AFSNonCachedWrite Failed wait for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1153 pFcb->ObjectInformation->FileId.Cell,
1154 pFcb->ObjectInformation->FileId.Volume,
1155 pFcb->ObjectInformation->FileId.Vnode,
1156 pFcb->ObjectInformation->FileId.Unique,
1157 StartingByte.QuadPart,
1161 try_return( ntStatus);
1166 // As per the read path -
1169 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1170 AFS_TRACE_LEVEL_VERBOSE,
1171 "AFSNonCachedWrite Extents located for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1172 pFcb->ObjectInformation->FileId.Cell,
1173 pFcb->ObjectInformation->FileId.Volume,
1174 pFcb->ObjectInformation->FileId.Vnode,
1175 pFcb->ObjectInformation->FileId.Unique,
1176 StartingByte.QuadPart,
1179 ntStatus = AFSGetExtents( pFcb,
1186 if (!NT_SUCCESS(ntStatus))
1189 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1190 AFS_TRACE_LEVEL_ERROR,
1191 "AFSNonCachedWrite (%p) Failed to retrieve mapped extents Status %08lX\n",
1195 try_return( ntStatus );
1198 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1199 AFS_TRACE_LEVEL_VERBOSE,
1200 "AFSNonCachedWrite (%p) Successfully retrieved map extents count %d run count %d\n",
1205 if( BooleanFlagOn( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE))
1208 Irp->IoStatus.Information = ByteCount;
1212 // Setup the MD5 for each extent
1215 AFSSetupMD5Hash( pFcb,
1223 ntStatus = AFSProcessExtentRun( pSystemBuffer,
1229 if (!NT_SUCCESS(ntStatus))
1232 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1233 AFS_TRACE_LEVEL_ERROR,
1234 "AFSNonCachedWrite (%p) Failed to process extent run for non-persistent cache Status %08lX\n",
1239 try_return( ntStatus);
1243 // Retrieve the cache file object
1246 pCacheFileObject = AFSReferenceCacheFileObject();
1248 if( pCacheFileObject == NULL)
1251 ntStatus = STATUS_DEVICE_NOT_READY;
1253 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1254 AFS_TRACE_LEVEL_ERROR,
1255 "AFSNonCachedWrite Failed to retrieve cache fileobject for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1256 pFcb->ObjectInformation->FileId.Cell,
1257 pFcb->ObjectInformation->FileId.Volume,
1258 pFcb->ObjectInformation->FileId.Vnode,
1259 pFcb->ObjectInformation->FileId.Unique,
1260 StartingByte.QuadPart,
1264 try_return( ntStatus);
1267 if (runCount > AFS_MAX_STACK_IO_RUNS)
1270 pIoRuns = (AFSIoRun*) AFSExAllocatePoolWithTag( PagedPool,
1271 runCount * sizeof( AFSIoRun ),
1273 if (NULL == pIoRuns)
1276 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1277 AFS_TRACE_LEVEL_ERROR,
1278 "AFSNonCachedWrite (%p) Failed to allocate IO run block\n",
1281 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1290 RtlZeroMemory( pIoRuns, runCount * sizeof( AFSIoRun ));
1292 ntStatus = AFSSetupIoRun( IoGetRelatedDeviceObject( pCacheFileObject),
1301 if (!NT_SUCCESS(ntStatus))
1304 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1305 AFS_TRACE_LEVEL_ERROR,
1306 "AFSNonCachedWrite (%p) Failed to initialize IO run block Status %08lX\n",
1310 try_return( ntStatus );
1313 AFSReferenceActiveExtents( pStartExtent,
1316 bDerefExtents = TRUE;
1318 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1319 AFS_TRACE_LEVEL_VERBOSE,
1320 "AFSNonCachedWrite Releasing(2) Fcb extents lock %p SHARED %08lX\n",
1321 &pFcb->NPFcb->Specific.File.ExtentsResource,
1322 PsGetCurrentThread()));
1324 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1327 pGatherIo = (AFSGatherIo*) AFSExAllocatePoolWithTag( NonPagedPool,
1328 sizeof( AFSGatherIo),
1331 if (NULL == pGatherIo)
1334 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1335 AFS_TRACE_LEVEL_ERROR,
1336 "AFSNonCachedWrite (%p) Failed to allocate IO gather block\n",
1339 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1340 AFS_TRACE_LEVEL_VERBOSE,
1341 "AFSNonCachedWrite Acquiring(1) Fcb extents lock %p SHARED %08lX\n",
1342 &pFcb->NPFcb->Specific.File.ExtentsResource,
1343 PsGetCurrentThread()));
1345 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1349 AFSDereferenceActiveExtents( pStartExtent,
1352 try_return (ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1355 RtlZeroMemory( pGatherIo, sizeof( AFSGatherIo ));
1358 // Initialize count to 1, that was we won't get an early
1359 // completion if the first irp completes before the second is
1362 pGatherIo->Count = 1;
1363 pGatherIo->Status = STATUS_SUCCESS;
1364 pGatherIo->MasterIrp = Irp;
1365 pGatherIo->Synchronous = TRUE;
1366 pGatherIo->CompleteMasterIrp = FALSE;
1368 bCompleteIrp = TRUE;
1370 if( pGatherIo->Synchronous)
1372 KeInitializeEvent( &pGatherIo->Event, NotificationEvent, FALSE );
1377 // Setup the MD5 for each extent
1380 AFSSetupMD5Hash( pFcb,
1389 // Pre-emptively set up the count
1392 Irp->IoStatus.Information = ByteCount;
1394 ntStatus = AFSQueueStartIos( pCacheFileObject,
1396 IRP_WRITE_OPERATION | IRP_SYNCHRONOUS_API,
1401 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1402 AFS_TRACE_LEVEL_VERBOSE,
1403 "AFSNonCachedWrite (%p) AFSStartIos completed Status %08lX\n",
1407 if( !NT_SUCCESS( ntStatus))
1410 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1411 AFS_TRACE_LEVEL_VERBOSE,
1412 "AFSNonCachedWrite Acquiring(2) Fcb extents lock %p SHARED %08lX\n",
1413 &pFcb->NPFcb->Specific.File.ExtentsResource,
1414 PsGetCurrentThread()));
1416 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1420 AFSDereferenceActiveExtents( pStartExtent,
1423 try_return( ntStatus);
1427 // Wait for completion of All IOs we started.
1430 ntStatus = KeWaitForSingleObject( &pGatherIo->Event,
1436 if( NT_SUCCESS( ntStatus))
1439 ntStatus = pGatherIo->Status;
1442 if( !NT_SUCCESS( ntStatus))
1445 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1446 AFS_TRACE_LEVEL_VERBOSE,
1447 "AFSNonCachedWrite Acquiring(3) Fcb extents lock %p SHARED %08lX\n",
1448 &pFcb->NPFcb->Specific.File.ExtentsResource,
1449 PsGetCurrentThread()));
1451 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1455 AFSDereferenceActiveExtents( pStartExtent,
1458 try_return( ntStatus);
1463 if( NT_SUCCESS( ntStatus) &&
1464 pStartExtent != NULL &&
1465 Irp->IoStatus.Information > 0)
1471 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1477 // Since this is dirty we can mark the extents dirty now.
1478 // AFSMarkDirty will dereference the extents. Do not call
1479 // AFSDereferenceActiveExtents() in this code path.
1491 // This was an uncached user write - tell the server to do
1492 // the flush when the worker thread next wakes up
1494 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1498 if( pCacheFileObject != NULL)
1500 AFSReleaseCacheFileObject( pCacheFileObject);
1503 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1504 AFS_TRACE_LEVEL_VERBOSE,
1505 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1507 StartingByte.HighPart,
1508 StartingByte.LowPart,
1512 if (NT_SUCCESS(ntStatus) &&
1517 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1523 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1524 AFS_TRACE_LEVEL_VERBOSE,
1525 "AFSNonCachedWrite Releasing Fcb extents lock %p SHARED %08lX\n",
1526 &pFcb->NPFcb->Specific.File.ExtentsResource,
1527 PsGetCurrentThread()));
1529 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1534 AFSExFreePoolWithTag(pGatherIo, AFS_GATHER_TAG);
1537 if( NULL != pIoRuns &&
1538 stIoRuns != pIoRuns)
1540 AFSExFreePoolWithTag(pIoRuns, AFS_IO_RUN_TAG);
1546 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1547 AFS_TRACE_LEVEL_VERBOSE,
1548 "AFSNonCachedWrite Completing Irp %p Status %08lX Info %08lX\n",
1551 Irp->IoStatus.Information));
1553 AFSCompleteRequest( Irp, ntStatus);
1562 AFSNonCachedWriteDirect( IN PDEVICE_OBJECT DeviceObject,
1564 IN LARGE_INTEGER StartingByte,
1567 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
1568 VOID *pSystemBuffer = NULL;
1569 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
1570 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1571 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1572 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1573 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
1574 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1575 BOOLEAN bNoIntermediateBuffering = BooleanFlagOn( pFileObject->Flags, FO_NO_INTERMEDIATE_BUFFERING);
1576 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
1577 AFSFileIOCB stFileIORequest;
1578 AFSFileIOResultCB stFileIOResult;
1579 ULONG ulResultLen = 0;
1584 Irp->IoStatus.Information = 0;
1586 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1587 AFS_TRACE_LEVEL_VERBOSE,
1588 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
1590 StartingByte.HighPart,
1591 StartingByte.LowPart,
1594 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
1597 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1598 AFS_TRACE_LEVEL_ERROR,
1599 "AFSNonCachedWriteDirect (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
1602 pIrpSp->Parameters.Write.Length,
1603 pDevExt->Specific.RDR.MaxIo.QuadPart));
1605 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1609 // Get the mapping for the buffer
1611 pSystemBuffer = AFSLockSystemBuffer( Irp,
1614 if( pSystemBuffer == NULL)
1617 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1618 AFS_TRACE_LEVEL_ERROR,
1619 "AFSNonCachedWriteDirect (%p) Failed to map system buffer\n",
1622 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1626 // Issue the request at the service for processing
1629 ulResultLen = sizeof( AFSFileIOResultCB);
1631 RtlZeroMemory( &stFileIORequest,
1632 sizeof( AFSFileIOCB));
1634 RtlZeroMemory( &stFileIOResult,
1635 sizeof( AFSFileIOResultCB));
1637 stFileIORequest.SystemIOBuffer = pSystemBuffer;
1639 stFileIORequest.SystemIOBufferMdl = Irp->MdlAddress;
1641 stFileIORequest.IOLength = ByteCount;
1643 stFileIORequest.IOOffset = StartingByte;
1645 ulFlags = AFS_REQUEST_FLAG_SYNCHRONOUS;
1647 if ( bNoIntermediateBuffering)
1650 ulFlags |= AFS_REQUEST_FLAG_CACHE_BYPASS;
1654 // Update file metadata
1657 stFileIORequest.EndOfFile = pFcb->ObjectInformation->EndOfFile;
1659 stFileIORequest.CreateTime = pFcb->ObjectInformation->CreationTime;
1661 stFileIORequest.ChangeTime = pFcb->ObjectInformation->ChangeTime;
1663 stFileIORequest.LastAccessTime = pFcb->ObjectInformation->LastAccessTime;
1665 stFileIORequest.LastWriteTime = pFcb->ObjectInformation->LastWriteTime;
1668 // Write the data to the service
1671 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PROCESS_WRITE_FILE,
1674 &pCcb->DirectoryCB->NameInformation.FileName,
1675 &pFcb->ObjectInformation->FileId,
1676 pFcb->ObjectInformation->VolumeCB->VolumeInformation.Cell,
1677 pFcb->ObjectInformation->VolumeCB->VolumeInformation.CellLength,
1679 sizeof( AFSFileIOCB),
1683 if( NT_SUCCESS( ntStatus))
1686 Irp->IoStatus.Information = (ULONG_PTR)stFileIOResult.Length;
1691 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1692 AFS_TRACE_LEVEL_ERROR,
1693 "AFSNonCachedWriteDirect (%p) Failed to send write to service Status %08lX\n",
1700 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1701 AFS_TRACE_LEVEL_VERBOSE,
1702 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1704 StartingByte.HighPart,
1705 StartingByte.LowPart,
1709 if (NT_SUCCESS(ntStatus) &&
1714 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1717 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1718 AFS_TRACE_LEVEL_VERBOSE,
1719 "AFSNonCachedWriteDirect Completing Irp %p Status %08lX Info %08lX\n",
1722 Irp->IoStatus.Information));
1724 AFSCompleteRequest( Irp, ntStatus);
1732 AFSCachedWrite( IN PDEVICE_OBJECT DeviceObject,
1734 IN LARGE_INTEGER StartingByte,
1736 IN BOOLEAN ForceFlush)
1738 UNREFERENCED_PARAMETER(DeviceObject);
1739 PVOID pSystemBuffer = NULL;
1740 NTSTATUS ntStatus = STATUS_SUCCESS;
1741 IO_STATUS_BLOCK iosbFlush;
1742 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1743 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1744 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1745 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1746 ULONG ulCurrentIO = 0, ulTotalLen = ByteCount;
1747 PMDL pCurrentMdl = Irp->MdlAddress;
1748 LARGE_INTEGER liCurrentOffset;
1753 Irp->IoStatus.Information = 0;
1755 if( BooleanFlagOn( pIrpSp->MinorFunction, IRP_MN_MDL))
1761 CcPrepareMdlWrite( pFileObject,
1767 ntStatus = Irp->IoStatus.Status;
1769 __except( EXCEPTION_EXECUTE_HANDLER)
1771 ntStatus = GetExceptionCode();
1773 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1774 AFS_TRACE_LEVEL_ERROR,
1775 "AFSCachedWrite (%p) Exception thrown while preparing mdl write Status %08lX\n",
1780 if( !NT_SUCCESS( ntStatus))
1784 // Free up any potentially allocated mdl's
1787 CcMdlWriteComplete( pFileObject,
1791 Irp->MdlAddress = NULL;
1793 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1794 AFS_TRACE_LEVEL_ERROR,
1795 "AFSCachedWrite (%p) Failed to process MDL write Status %08lX\n",
1800 try_return( ntStatus);
1803 liCurrentOffset.QuadPart = StartingByte.QuadPart;
1805 while( ulTotalLen > 0)
1808 ntStatus = STATUS_SUCCESS;
1810 if( pCurrentMdl != NULL)
1813 pSystemBuffer = MmGetSystemAddressForMdlSafe( pCurrentMdl,
1814 NormalPagePriority);
1816 ulCurrentIO = MmGetMdlByteCount( pCurrentMdl);
1818 if( ulCurrentIO > ulTotalLen)
1820 ulCurrentIO = ulTotalLen;
1826 pSystemBuffer = AFSLockSystemBuffer( Irp,
1829 ulCurrentIO = ulTotalLen;
1832 if( pSystemBuffer == NULL)
1835 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1836 AFS_TRACE_LEVEL_ERROR,
1837 "AFSCachedWrite (%p) Failed to lock system buffer\n",
1840 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1846 if( !CcCopyWrite( pFileObject,
1853 // Failed to process request.
1856 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
1857 AFS_TRACE_LEVEL_ERROR,
1858 "AFSCachedWrite (%p) Failed to issue CcCopyWrite %wZ @ %0I64X Status %08lX\n",
1860 &pFileObject->FileName,
1861 liCurrentOffset.QuadPart,
1862 Irp->IoStatus.Status));
1864 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1867 __except( EXCEPTION_EXECUTE_HANDLER)
1870 ntStatus = GetExceptionCode();
1872 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1873 AFS_TRACE_LEVEL_ERROR,
1874 "AFSCachedWrite (%p) CcCopyWrite Threw exception %wZ @ %0I64X Status %08lX\n",
1876 &pFileObject->FileName,
1877 liCurrentOffset.QuadPart,
1881 if( !NT_SUCCESS( ntStatus))
1883 try_return( ntStatus);
1887 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1891 // We have detected a file we do a write through with.
1894 CcFlushCache(&pFcb->NPFcb->SectionObjectPointers,
1899 if( !NT_SUCCESS( iosbFlush.Status))
1902 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1903 AFS_TRACE_LEVEL_ERROR,
1904 "AFSCachedWrite (%p) CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
1906 &pFileObject->FileName,
1907 pFcb->ObjectInformation->FileId.Cell,
1908 pFcb->ObjectInformation->FileId.Volume,
1909 pFcb->ObjectInformation->FileId.Vnode,
1910 pFcb->ObjectInformation->FileId.Unique,
1912 iosbFlush.Information));
1914 try_return( ntStatus = iosbFlush.Status);
1918 if( ulTotalLen <= ulCurrentIO)
1923 liCurrentOffset.QuadPart += ulCurrentIO;
1925 ulTotalLen -= ulCurrentIO;
1927 pCurrentMdl = pCurrentMdl->Next;
1932 if( NT_SUCCESS( ntStatus))
1935 Irp->IoStatus.Information = ByteCount;
1938 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1941 // Write through asked for... Set things so that we get
1942 // flush when the worker thread next wakes up
1944 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1948 AFSCompleteRequest( Irp,
1957 AFSExtendingWrite( IN AFSFcb *Fcb,
1958 IN PFILE_OBJECT FileObject,
1959 IN LONGLONG NewLength)
1961 LARGE_INTEGER liSaveFileSize = Fcb->Header.FileSize;
1962 LARGE_INTEGER liSaveAllocation = Fcb->Header.AllocationSize;
1963 NTSTATUS ntStatus = STATUS_SUCCESS;
1964 AFSCcb *pCcb = (AFSCcb *)FileObject->FsContext2;
1966 if( NewLength > Fcb->Header.AllocationSize.QuadPart)
1969 Fcb->Header.AllocationSize.QuadPart = NewLength;
1971 Fcb->ObjectInformation->AllocationSize = Fcb->Header.AllocationSize;
1974 if( NewLength > Fcb->Header.FileSize.QuadPart)
1977 Fcb->Header.FileSize.QuadPart = NewLength;
1979 Fcb->ObjectInformation->EndOfFile = Fcb->Header.FileSize;
1986 ntStatus = AFSUpdateFileInformation( &Fcb->ObjectInformation->ParentFileId,
1987 Fcb->ObjectInformation,
1990 if (NT_SUCCESS(ntStatus))
1993 KeQuerySystemTime( &Fcb->ObjectInformation->ChangeTime);
1995 SetFlag( Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
1998 // If the file is currently cached, then let the MM know about the extension
2001 if( CcIsFileCached( FileObject))
2003 CcSetFileSizes( FileObject,
2004 (PCC_FILE_SIZES)&Fcb->Header.AllocationSize);
2009 Fcb->Header.FileSize = liSaveFileSize;
2010 Fcb->Header.AllocationSize = liSaveAllocation;
2014 // DownConvert file resource to shared
2016 ExConvertExclusiveToSharedLite( &Fcb->NPFcb->Resource);
2022 AFSShareWrite( IN PDEVICE_OBJECT DeviceObject,
2026 UNREFERENCED_PARAMETER(DeviceObject);
2027 NTSTATUS ntStatus = STATUS_SUCCESS;
2028 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2029 AFSFcb *pFcb = NULL;
2030 AFSCcb *pCcb = NULL;
2031 AFSPipeIORequestCB *pIoRequest = NULL;
2032 void *pBuffer = NULL;
2033 AFSPipeIOResultCB stIoResult;
2034 ULONG ulBytesReturned = 0;
2039 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2041 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2043 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2044 AFS_TRACE_LEVEL_VERBOSE,
2045 "AFSShareWrite On pipe %wZ Length %08lX\n",
2046 &pCcb->DirectoryCB->NameInformation.FileName,
2047 pIrpSp->Parameters.Write.Length));
2049 if( pIrpSp->Parameters.Write.Length == 0)
2053 // Nothing to do in this case
2056 try_return( ntStatus);
2060 // Retrieve the buffer for the read request
2063 pBuffer = AFSLockSystemBuffer( Irp,
2064 pIrpSp->Parameters.Write.Length);
2066 if( pBuffer == NULL)
2069 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2070 AFS_TRACE_LEVEL_ERROR,
2071 "AFSShareWrite Failed to map buffer on pipe %wZ\n",
2072 &pCcb->DirectoryCB->NameInformation.FileName));
2074 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2077 AFSAcquireShared( &pFcb->NPFcb->Resource,
2080 pIoRequest = (AFSPipeIORequestCB *)AFSExAllocatePoolWithTag( PagedPool,
2081 sizeof( AFSPipeIORequestCB) +
2082 pIrpSp->Parameters.Write.Length,
2083 AFS_GENERIC_MEMORY_14_TAG);
2085 if( pIoRequest == NULL)
2088 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2091 RtlZeroMemory( pIoRequest,
2092 sizeof( AFSPipeIORequestCB) + pIrpSp->Parameters.Write.Length);
2094 pIoRequest->RequestId = pCcb->RequestID;
2096 pIoRequest->RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
2098 pIoRequest->BufferLength = pIrpSp->Parameters.Write.Length;
2100 RtlCopyMemory( (void *)((char *)pIoRequest + sizeof( AFSPipeIORequestCB)),
2102 pIrpSp->Parameters.Write.Length);
2104 stIoResult.BytesProcessed = 0;
2106 ulBytesReturned = sizeof( AFSPipeIOResultCB);
2109 // Issue the open request to the service
2112 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIPE_WRITE,
2113 AFS_REQUEST_FLAG_SYNCHRONOUS,
2115 &pCcb->DirectoryCB->NameInformation.FileName,
2120 sizeof( AFSPipeIORequestCB) +
2121 pIrpSp->Parameters.Write.Length,
2125 if( !NT_SUCCESS( ntStatus))
2128 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
2129 AFS_TRACE_LEVEL_ERROR,
2130 "AFSShareWrite (%p) Failed service write Status %08lX\n",
2134 try_return( ntStatus);
2137 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2138 AFS_TRACE_LEVEL_VERBOSE,
2139 "AFSShareWrite Completed on pipe %wZ Length read %08lX\n",
2140 &pCcb->DirectoryCB->NameInformation.FileName,
2141 stIoResult.BytesProcessed));
2143 Irp->IoStatus.Information = stIoResult.BytesProcessed;
2150 AFSReleaseResource( &pFcb->NPFcb->Resource);
2153 if( pIoRequest != NULL)
2156 AFSExFreePoolWithTag( pIoRequest, AFS_GENERIC_MEMORY_14_TAG);