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;
601 AFSReleaseResource( &pNPFcb->SectionObjectResource);
603 bReleaseSectionObject = FALSE;
605 AFSReleaseResource( &pNPFcb->Resource);
607 bReleaseMain = FALSE;
618 // Check the BR locks on the file.
621 if ( !FsRtlCheckLockForWriteAccess( &pFcb->Specific.File.FileLock,
625 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
626 AFS_TRACE_LEVEL_ERROR,
627 "AFSCommonWrite (%p) Request failed due to lock conflict\n",
630 try_return( ntStatus = STATUS_FILE_LOCK_CONFLICT);
636 ntStatus = AFSExtendingWrite( pFcb, pFileObject, (liStartingByte.QuadPart + ulByteCount));
638 if( !NT_SUCCESS(ntStatus))
641 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
642 AFS_TRACE_LEVEL_ERROR,
643 "AFSCommonWrite (%p) Failed extending write request Status %08lX\n",
647 try_return( ntStatus );
653 // Fire off the request as appropriate
655 bCompleteIrp = FALSE;
662 // Main and SectionObject resources held Shared
665 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
666 AFS_TRACE_LEVEL_VERBOSE,
667 "AFSCommonWrite (%p) Processing CACHED request Offset %0I64X Len %08lX%s\n",
669 liStartingByte.QuadPart,
671 bRetry ? " RETRY" : ""));
673 ntStatus = AFSCachedWrite( DeviceObject, Irp, liStartingByte, ulByteCount, bForceFlush);
678 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
679 AFS_TRACE_LEVEL_VERBOSE,
680 "AFSCommonWrite (%p) Processing NON-CACHED request Offset %0I64X Len %08lX%s\n",
682 liStartingByte.QuadPart,
684 bRetry ? " RETRY" : ""));
686 if( BooleanFlagOn( pRDRDevExt->DeviceFlags, AFS_DEVICE_FLAG_DIRECT_SERVICE_IO))
689 ntStatus = AFSNonCachedWriteDirect( DeviceObject, Irp, liStartingByte, ulByteCount);
693 ntStatus = AFSNonCachedWrite( DeviceObject, Irp, liStartingByte, ulByteCount);
699 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
700 AFS_TRACE_LEVEL_VERBOSE,
701 "AFSCommonWrite (%p) Process complete Status %08lX\n",
705 if ( NT_SUCCESS( ntStatus))
713 pFileObject->CurrentByteOffset.QuadPart = liStartingByte.QuadPart + ulByteCount;
717 // If this extended the VDL, then update it accordingly.
718 // Increasing the VDL does not require a call to CcSetFileSizes.
721 if( liStartingByte.QuadPart + ulByteCount > pFcb->Header.ValidDataLength.QuadPart)
724 pFcb->Header.ValidDataLength.QuadPart = liStartingByte.QuadPart + ulByteCount;
727 if( !BooleanFlagOn( pFcb->Flags, AFS_FCB_FLAG_UPDATE_LAST_WRITE_TIME))
730 SetFlag( pFcb->Flags, AFS_FCB_FLAG_UPDATE_WRITE_TIME);
732 KeQuerySystemTime( &pFcb->ObjectInformation->LastWriteTime);
737 if ( !bPagingIo && bNonCachedIo && CcIsFileCached( pFileObject) &&
738 pNPFcb->SectionObjectPointers.DataSectionObject != NULL &&
739 bReleaseSectionObject)
742 // Regardless of whether or not the a non-paging non-cached write
743 // succeeds or fails, if the file is cached the contents of the
744 // cache are no longer up to date. A CcPurgeCacheSection must be
745 // performed to force subsequent cached reads to obtain the data
748 // The Fcb Resource is dropped in order to permit filters that perform
749 // an open via a worker thread in response to a purge to do so without
750 // deadlocking. The SectionObjectResource is held across the purge to
751 // prevent racing with other cache operations.
757 AFSReleaseResource( &pNPFcb->Resource);
759 bReleaseMain = FALSE;
762 if ( !CcPurgeCacheSection( &pNPFcb->SectionObjectPointers,
768 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
769 AFS_TRACE_LEVEL_WARNING,
770 "AFSCommonWrite CcPurgeCacheSection failure FID %08lX-%08lX-%08lX-%08lX\n",
771 pFcb->ObjectInformation->FileId.Cell,
772 pFcb->ObjectInformation->FileId.Volume,
773 pFcb->ObjectInformation->FileId.Vnode,
774 pFcb->ObjectInformation->FileId.Unique));
776 SetFlag( pFcb->Flags, AFS_FCB_FLAG_PURGE_ON_CLOSE);
780 ObDereferenceObject(pFileObject);
782 if( bReleaseSectionObject)
785 AFSReleaseResource( &pNPFcb->SectionObjectResource);
791 AFSReleaseResource( &pNPFcb->PagingResource);
797 AFSReleaseResource( &pNPFcb->Resource);
803 AFSCompleteRequest( Irp,
812 AFSIOCtlWrite( IN PDEVICE_OBJECT DeviceObject,
816 UNREFERENCED_PARAMETER(DeviceObject);
817 NTSTATUS ntStatus = STATUS_SUCCESS;
818 AFSPIOCtlIORequestCB stIORequestCB;
819 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
822 AFSPIOCtlIOResultCB stIOResultCB;
823 ULONG ulBytesReturned = 0;
824 AFSFileID stParentFID;
829 Irp->IoStatus.Information = 0;
831 RtlZeroMemory( &stIORequestCB,
832 sizeof( AFSPIOCtlIORequestCB));
834 if( pIrpSp->Parameters.Write.Length == 0)
838 // Nothing to do in this case
841 try_return( ntStatus);
844 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
846 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
848 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
849 AFS_TRACE_LEVEL_VERBOSE,
850 "AFSIOCtlWrite Acquiring Fcb lock %p SHARED %08lX\n",
851 &pFcb->NPFcb->Resource,
852 PsGetCurrentThread()));
854 AFSAcquireShared( &pFcb->NPFcb->Resource,
858 // Get the parent fid to pass to the cm
861 RtlZeroMemory( &stParentFID,
864 if( BooleanFlagOn( pFcb->ObjectInformation->Flags, AFS_OBJECT_FLAGS_PARENT_FID))
868 // The parent directory FID of the node
871 stParentFID = pFcb->ObjectInformation->ParentFileId;
875 // Set the control block up
878 stIORequestCB.RequestId = pCcb->RequestID;
880 if( pFcb->ObjectInformation->VolumeCB != NULL)
882 stIORequestCB.RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
886 // Lock down the buffer
889 stIORequestCB.MappedBuffer = AFSMapToService( Irp,
890 pIrpSp->Parameters.Write.Length);
892 if( stIORequestCB.MappedBuffer == NULL)
895 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
898 stIORequestCB.BufferLength = pIrpSp->Parameters.Write.Length;
900 stIOResultCB.BytesProcessed = 0;
902 ulBytesReturned = sizeof( AFSPIOCtlIOResultCB);
905 // Issue the request to the service
908 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIOCTL_WRITE,
909 AFS_REQUEST_FLAG_SYNCHRONOUS,
915 (void *)&stIORequestCB,
916 sizeof( AFSPIOCtlIORequestCB),
920 if( !NT_SUCCESS( ntStatus))
923 try_return( ntStatus);
927 // Update the length written
930 Irp->IoStatus.Information = stIOResultCB.BytesProcessed;
934 if( stIORequestCB.MappedBuffer != NULL)
937 AFSUnmapServiceMappedBuffer( stIORequestCB.MappedBuffer,
944 AFSReleaseResource( &pFcb->NPFcb->Resource);
952 // This function is called when we know we have to read from the AFS Cache.
954 // It ensures that we have exents for the entirety of the write and
955 // then pins the extents into memory (meaning that although we may
956 // add we will not remove). Then it creates a scatter gather write
957 // and fires off the IRPs
961 AFSNonCachedWrite( IN PDEVICE_OBJECT DeviceObject,
963 IN LARGE_INTEGER StartingByte,
966 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
967 VOID *pSystemBuffer = NULL;
968 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
969 BOOLEAN bLocked = FALSE;
970 BOOLEAN bCompleteIrp = TRUE;
971 AFSGatherIo *pGatherIo = NULL;
972 AFSIoRun *pIoRuns = NULL;
973 AFSIoRun stIoRuns[AFS_MAX_STACK_IO_RUNS];
974 ULONG extentsCount = 0, runCount = 0;
975 AFSExtent *pStartExtent = NULL;
976 AFSExtent *pIgnoreExtent = NULL;
977 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
978 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
979 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
980 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
981 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
982 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
983 LARGE_INTEGER liCurrentTime, liLastRequestTime;
984 AFSDeviceExt *pControlDevExt = (AFSDeviceExt *)AFSControlDeviceObject->DeviceExtension;
985 PFILE_OBJECT pCacheFileObject = NULL;
986 BOOLEAN bDerefExtents = FALSE;
991 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
992 AFS_TRACE_LEVEL_VERBOSE,
993 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
995 StartingByte.HighPart,
996 StartingByte.LowPart,
999 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
1002 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1003 AFS_TRACE_LEVEL_ERROR,
1004 "AFSNonCachedWrite (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
1007 pIrpSp->Parameters.Write.Length,
1008 pDevExt->Specific.RDR.MaxIo.QuadPart));
1010 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1014 // Get the mapping for the buffer
1016 pSystemBuffer = AFSLockSystemBuffer( Irp,
1019 if( pSystemBuffer == NULL)
1022 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1023 AFS_TRACE_LEVEL_ERROR,
1024 "AFSNonCachedWrite (%p) Failed to map system buffer\n",
1027 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1032 // Provoke a get of the extents - if we need to.
1035 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1036 AFS_TRACE_LEVEL_VERBOSE,
1037 "AFSNonCachedWrite Requesting extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1038 pFcb->ObjectInformation->FileId.Cell,
1039 pFcb->ObjectInformation->FileId.Volume,
1040 pFcb->ObjectInformation->FileId.Vnode,
1041 pFcb->ObjectInformation->FileId.Unique,
1042 StartingByte.QuadPart,
1045 ntStatus = AFSRequestExtentsAsync( pFcb,
1050 if (!NT_SUCCESS(ntStatus))
1053 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1054 AFS_TRACE_LEVEL_ERROR,
1055 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1059 try_return( ntStatus);
1062 KeQueryTickCount( &liLastRequestTime);
1067 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1068 AFS_TRACE_LEVEL_VERBOSE,
1069 "AFSNonCachedWrite Acquiring Fcb extents lock %p SHARED %08lX\n",
1070 &pFcb->NPFcb->Specific.File.ExtentsResource,
1071 PsGetCurrentThread()));
1073 ASSERT( !ExIsResourceAcquiredLite( &pFcb->NPFcb->Specific.File.ExtentsResource ));
1075 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource, TRUE );
1078 pStartExtent = NULL;
1079 pIgnoreExtent = NULL;
1081 if ( AFSDoExtentsMapRegion( pFcb, &StartingByte, ByteCount, &pStartExtent, &pIgnoreExtent ))
1086 KeClearEvent( &pFcb->NPFcb->Specific.File.ExtentsRequestComplete );
1088 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1089 AFS_TRACE_LEVEL_VERBOSE,
1090 "AFSNonCachedWrite Releasing(1) Fcb extents lock %p SHARED %08lX\n",
1091 &pFcb->NPFcb->Specific.File.ExtentsResource,
1092 PsGetCurrentThread()));
1094 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1098 // We will re-request the extents after waiting for them
1101 KeQueryTickCount( &liCurrentTime);
1103 if( liCurrentTime.QuadPart - liLastRequestTime.QuadPart >= pControlDevExt->Specific.Control.ExtentRequestTimeCount.QuadPart)
1106 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1107 AFS_TRACE_LEVEL_VERBOSE,
1108 "AFSNonCachedWrite Requesting extents, again, for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1109 pFcb->ObjectInformation->FileId.Cell,
1110 pFcb->ObjectInformation->FileId.Volume,
1111 pFcb->ObjectInformation->FileId.Vnode,
1112 pFcb->ObjectInformation->FileId.Unique,
1113 StartingByte.QuadPart,
1116 ntStatus = AFSRequestExtentsAsync( pFcb,
1121 if (!NT_SUCCESS(ntStatus))
1124 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1125 AFS_TRACE_LEVEL_ERROR,
1126 "AFSNonCachedWrite (%p) Failed to request extents Status %08lX\n",
1130 try_return( ntStatus);
1133 KeQueryTickCount( &liLastRequestTime);
1137 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1138 AFS_TRACE_LEVEL_VERBOSE,
1139 "AFSNonCachedWrite Waiting for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1140 pFcb->ObjectInformation->FileId.Cell,
1141 pFcb->ObjectInformation->FileId.Volume,
1142 pFcb->ObjectInformation->FileId.Vnode,
1143 pFcb->ObjectInformation->FileId.Unique,
1144 StartingByte.QuadPart,
1151 ntStatus = AFSWaitForExtentMapping ( pFcb, pCcb);
1153 if (!NT_SUCCESS(ntStatus))
1156 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1157 AFS_TRACE_LEVEL_ERROR,
1158 "AFSNonCachedWrite Failed wait for extents for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1159 pFcb->ObjectInformation->FileId.Cell,
1160 pFcb->ObjectInformation->FileId.Volume,
1161 pFcb->ObjectInformation->FileId.Vnode,
1162 pFcb->ObjectInformation->FileId.Unique,
1163 StartingByte.QuadPart,
1167 try_return( ntStatus);
1172 // As per the read path -
1175 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1176 AFS_TRACE_LEVEL_VERBOSE,
1177 "AFSNonCachedWrite Extents located for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX\n",
1178 pFcb->ObjectInformation->FileId.Cell,
1179 pFcb->ObjectInformation->FileId.Volume,
1180 pFcb->ObjectInformation->FileId.Vnode,
1181 pFcb->ObjectInformation->FileId.Unique,
1182 StartingByte.QuadPart,
1185 ntStatus = AFSGetExtents( pFcb,
1192 if (!NT_SUCCESS(ntStatus))
1195 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1196 AFS_TRACE_LEVEL_ERROR,
1197 "AFSNonCachedWrite (%p) Failed to retrieve mapped extents Status %08lX\n",
1201 try_return( ntStatus );
1204 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1205 AFS_TRACE_LEVEL_VERBOSE,
1206 "AFSNonCachedWrite (%p) Successfully retrieved map extents count %d run count %d\n",
1211 if( BooleanFlagOn( AFSLibControlFlags, AFS_REDIR_LIB_FLAGS_NONPERSISTENT_CACHE))
1214 Irp->IoStatus.Information = ByteCount;
1218 // Setup the MD5 for each extent
1221 AFSSetupMD5Hash( pFcb,
1229 ntStatus = AFSProcessExtentRun( pSystemBuffer,
1235 if (!NT_SUCCESS(ntStatus))
1238 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1239 AFS_TRACE_LEVEL_ERROR,
1240 "AFSNonCachedWrite (%p) Failed to process extent run for non-persistent cache Status %08lX\n",
1245 try_return( ntStatus);
1249 // Retrieve the cache file object
1252 pCacheFileObject = AFSReferenceCacheFileObject();
1254 if( pCacheFileObject == NULL)
1257 ntStatus = STATUS_DEVICE_NOT_READY;
1259 AFSDbgTrace(( AFS_SUBSYSTEM_EXTENT_PROCESSING,
1260 AFS_TRACE_LEVEL_ERROR,
1261 "AFSNonCachedWrite Failed to retrieve cache fileobject for fid %08lX-%08lX-%08lX-%08lX Offset %0I64X Length %08lX Status %08lX\n",
1262 pFcb->ObjectInformation->FileId.Cell,
1263 pFcb->ObjectInformation->FileId.Volume,
1264 pFcb->ObjectInformation->FileId.Vnode,
1265 pFcb->ObjectInformation->FileId.Unique,
1266 StartingByte.QuadPart,
1270 try_return( ntStatus);
1273 if (runCount > AFS_MAX_STACK_IO_RUNS)
1276 pIoRuns = (AFSIoRun*) AFSExAllocatePoolWithTag( PagedPool,
1277 runCount * sizeof( AFSIoRun ),
1279 if (NULL == pIoRuns)
1282 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1283 AFS_TRACE_LEVEL_ERROR,
1284 "AFSNonCachedWrite (%p) Failed to allocate IO run block\n",
1287 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1296 RtlZeroMemory( pIoRuns, runCount * sizeof( AFSIoRun ));
1298 ntStatus = AFSSetupIoRun( IoGetRelatedDeviceObject( pCacheFileObject),
1307 if (!NT_SUCCESS(ntStatus))
1310 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1311 AFS_TRACE_LEVEL_ERROR,
1312 "AFSNonCachedWrite (%p) Failed to initialize IO run block Status %08lX\n",
1316 try_return( ntStatus );
1319 AFSReferenceActiveExtents( pStartExtent,
1322 bDerefExtents = TRUE;
1324 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1325 AFS_TRACE_LEVEL_VERBOSE,
1326 "AFSNonCachedWrite Releasing(2) Fcb extents lock %p SHARED %08lX\n",
1327 &pFcb->NPFcb->Specific.File.ExtentsResource,
1328 PsGetCurrentThread()));
1330 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1333 pGatherIo = (AFSGatherIo*) AFSExAllocatePoolWithTag( NonPagedPool,
1334 sizeof( AFSGatherIo),
1337 if (NULL == pGatherIo)
1340 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1341 AFS_TRACE_LEVEL_ERROR,
1342 "AFSNonCachedWrite (%p) Failed to allocate IO gather block\n",
1345 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1346 AFS_TRACE_LEVEL_VERBOSE,
1347 "AFSNonCachedWrite Acquiring(1) Fcb extents lock %p SHARED %08lX\n",
1348 &pFcb->NPFcb->Specific.File.ExtentsResource,
1349 PsGetCurrentThread()));
1351 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1355 AFSDereferenceActiveExtents( pStartExtent,
1358 try_return (ntStatus = STATUS_INSUFFICIENT_RESOURCES );
1361 RtlZeroMemory( pGatherIo, sizeof( AFSGatherIo ));
1364 // Initialize count to 1, that was we won't get an early
1365 // completion if the first irp completes before the second is
1368 pGatherIo->Count = 1;
1369 pGatherIo->Status = STATUS_SUCCESS;
1370 pGatherIo->MasterIrp = Irp;
1371 pGatherIo->Synchronous = TRUE;
1372 pGatherIo->CompleteMasterIrp = FALSE;
1374 bCompleteIrp = TRUE;
1376 if( pGatherIo->Synchronous)
1378 KeInitializeEvent( &pGatherIo->Event, NotificationEvent, FALSE );
1383 // Setup the MD5 for each extent
1386 AFSSetupMD5Hash( pFcb,
1395 // Pre-emptively set up the count
1398 Irp->IoStatus.Information = ByteCount;
1400 ntStatus = AFSQueueStartIos( pCacheFileObject,
1402 IRP_WRITE_OPERATION | IRP_SYNCHRONOUS_API,
1407 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1408 AFS_TRACE_LEVEL_VERBOSE,
1409 "AFSNonCachedWrite (%p) AFSStartIos completed Status %08lX\n",
1413 if( !NT_SUCCESS( ntStatus))
1416 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1417 AFS_TRACE_LEVEL_VERBOSE,
1418 "AFSNonCachedWrite Acquiring(2) Fcb extents lock %p SHARED %08lX\n",
1419 &pFcb->NPFcb->Specific.File.ExtentsResource,
1420 PsGetCurrentThread()));
1422 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1426 AFSDereferenceActiveExtents( pStartExtent,
1429 try_return( ntStatus);
1433 // Wait for completion of All IOs we started.
1436 ntStatus = KeWaitForSingleObject( &pGatherIo->Event,
1442 if( NT_SUCCESS( ntStatus))
1445 ntStatus = pGatherIo->Status;
1448 if( !NT_SUCCESS( ntStatus))
1451 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1452 AFS_TRACE_LEVEL_VERBOSE,
1453 "AFSNonCachedWrite Acquiring(3) Fcb extents lock %p SHARED %08lX\n",
1454 &pFcb->NPFcb->Specific.File.ExtentsResource,
1455 PsGetCurrentThread()));
1457 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1461 AFSDereferenceActiveExtents( pStartExtent,
1464 try_return( ntStatus);
1469 if( NT_SUCCESS( ntStatus) &&
1470 pStartExtent != NULL &&
1471 Irp->IoStatus.Information > 0)
1477 AFSAcquireShared( &pFcb->NPFcb->Specific.File.ExtentsResource,
1483 // Since this is dirty we can mark the extents dirty now.
1484 // AFSMarkDirty will dereference the extents. Do not call
1485 // AFSDereferenceActiveExtents() in this code path.
1497 // This was an uncached user write - tell the server to do
1498 // the flush when the worker thread next wakes up
1500 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1504 if( pCacheFileObject != NULL)
1506 AFSReleaseCacheFileObject( pCacheFileObject);
1509 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1510 AFS_TRACE_LEVEL_VERBOSE,
1511 "AFSNonCachedWrite (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1513 StartingByte.HighPart,
1514 StartingByte.LowPart,
1518 if (NT_SUCCESS(ntStatus) &&
1523 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1529 AFSDbgTrace(( AFS_SUBSYSTEM_LOCK_PROCESSING,
1530 AFS_TRACE_LEVEL_VERBOSE,
1531 "AFSNonCachedWrite Releasing Fcb extents lock %p SHARED %08lX\n",
1532 &pFcb->NPFcb->Specific.File.ExtentsResource,
1533 PsGetCurrentThread()));
1535 AFSReleaseResource( &pFcb->NPFcb->Specific.File.ExtentsResource );
1540 AFSExFreePoolWithTag(pGatherIo, AFS_GATHER_TAG);
1543 if( NULL != pIoRuns &&
1544 stIoRuns != pIoRuns)
1546 AFSExFreePoolWithTag(pIoRuns, AFS_IO_RUN_TAG);
1552 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1553 AFS_TRACE_LEVEL_VERBOSE,
1554 "AFSNonCachedWrite Completing Irp %p Status %08lX Info %08lX\n",
1557 Irp->IoStatus.Information));
1559 AFSCompleteRequest( Irp, ntStatus);
1568 AFSNonCachedWriteDirect( IN PDEVICE_OBJECT DeviceObject,
1570 IN LARGE_INTEGER StartingByte,
1573 NTSTATUS ntStatus = STATUS_UNSUCCESSFUL;
1574 VOID *pSystemBuffer = NULL;
1575 BOOLEAN bPagingIo = BooleanFlagOn( Irp->Flags, IRP_PAGING_IO);
1576 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1577 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1578 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1579 AFSCcb *pCcb = (AFSCcb *)pFileObject->FsContext2;
1580 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1581 BOOLEAN bNoIntermediateBuffering = BooleanFlagOn( pFileObject->Flags, FO_NO_INTERMEDIATE_BUFFERING);
1582 AFSDeviceExt *pDevExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
1583 AFSFileIOCB stFileIORequest;
1584 AFSFileIOResultCB stFileIOResult;
1585 ULONG ulResultLen = 0;
1590 Irp->IoStatus.Information = 0;
1592 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1593 AFS_TRACE_LEVEL_VERBOSE,
1594 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX\n",
1596 StartingByte.HighPart,
1597 StartingByte.LowPart,
1600 if (ByteCount > pDevExt->Specific.RDR.MaxIo.QuadPart)
1603 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1604 AFS_TRACE_LEVEL_ERROR,
1605 "AFSNonCachedWriteDirect (%p) Request %08lX Actual %08lX larger than MaxIO %I64X\n",
1608 pIrpSp->Parameters.Write.Length,
1609 pDevExt->Specific.RDR.MaxIo.QuadPart));
1611 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1615 // Get the mapping for the buffer
1617 pSystemBuffer = AFSLockSystemBuffer( Irp,
1620 if( pSystemBuffer == NULL)
1623 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1624 AFS_TRACE_LEVEL_ERROR,
1625 "AFSNonCachedWriteDirect (%p) Failed to map system buffer\n",
1628 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1632 // Issue the request at the service for processing
1635 ulResultLen = sizeof( AFSFileIOResultCB);
1637 RtlZeroMemory( &stFileIORequest,
1638 sizeof( AFSFileIOCB));
1640 RtlZeroMemory( &stFileIOResult,
1641 sizeof( AFSFileIOResultCB));
1643 stFileIORequest.SystemIOBuffer = pSystemBuffer;
1645 stFileIORequest.SystemIOBufferMdl = Irp->MdlAddress;
1647 stFileIORequest.IOLength = ByteCount;
1649 stFileIORequest.IOOffset = StartingByte;
1651 ulFlags = AFS_REQUEST_FLAG_SYNCHRONOUS;
1653 if ( bNoIntermediateBuffering)
1656 ulFlags |= AFS_REQUEST_FLAG_CACHE_BYPASS;
1660 // Update file metadata
1663 stFileIORequest.EndOfFile = pFcb->ObjectInformation->EndOfFile;
1665 stFileIORequest.CreateTime = pFcb->ObjectInformation->CreationTime;
1667 stFileIORequest.ChangeTime = pFcb->ObjectInformation->ChangeTime;
1669 stFileIORequest.LastAccessTime = pFcb->ObjectInformation->LastAccessTime;
1671 stFileIORequest.LastWriteTime = pFcb->ObjectInformation->LastWriteTime;
1674 // Write the data to the service
1677 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PROCESS_WRITE_FILE,
1680 &pCcb->DirectoryCB->NameInformation.FileName,
1681 &pFcb->ObjectInformation->FileId,
1682 pFcb->ObjectInformation->VolumeCB->VolumeInformation.Cell,
1683 pFcb->ObjectInformation->VolumeCB->VolumeInformation.CellLength,
1685 sizeof( AFSFileIOCB),
1689 if( NT_SUCCESS( ntStatus))
1692 Irp->IoStatus.Information = (ULONG_PTR)stFileIOResult.Length;
1697 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1698 AFS_TRACE_LEVEL_ERROR,
1699 "AFSNonCachedWriteDirect (%p) Failed to send write to service Status %08lX\n",
1706 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1707 AFS_TRACE_LEVEL_VERBOSE,
1708 "AFSNonCachedWriteDirect (FO: %p) StartingByte %08lX:%08lX Length %08lX Status %08lX\n",
1710 StartingByte.HighPart,
1711 StartingByte.LowPart,
1715 if (NT_SUCCESS(ntStatus) &&
1720 pFileObject->CurrentByteOffset.QuadPart = StartingByte.QuadPart + ByteCount;
1723 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1724 AFS_TRACE_LEVEL_VERBOSE,
1725 "AFSNonCachedWriteDirect Completing Irp %p Status %08lX Info %08lX\n",
1728 Irp->IoStatus.Information));
1730 AFSCompleteRequest( Irp, ntStatus);
1738 AFSCachedWrite( IN PDEVICE_OBJECT DeviceObject,
1740 IN LARGE_INTEGER StartingByte,
1742 IN BOOLEAN ForceFlush)
1744 UNREFERENCED_PARAMETER(DeviceObject);
1745 PVOID pSystemBuffer = NULL;
1746 NTSTATUS ntStatus = STATUS_SUCCESS;
1747 IO_STATUS_BLOCK iosbFlush;
1748 IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
1749 PFILE_OBJECT pFileObject = pIrpSp->FileObject;
1750 AFSFcb *pFcb = (AFSFcb *)pFileObject->FsContext;
1751 BOOLEAN bSynchronousFo = BooleanFlagOn( pFileObject->Flags, FO_SYNCHRONOUS_IO);
1752 ULONG ulCurrentIO = 0, ulTotalLen = ByteCount;
1753 PMDL pCurrentMdl = Irp->MdlAddress;
1754 LARGE_INTEGER liCurrentOffset;
1759 Irp->IoStatus.Information = 0;
1761 if( BooleanFlagOn( pIrpSp->MinorFunction, IRP_MN_MDL))
1767 CcPrepareMdlWrite( pFileObject,
1773 ntStatus = Irp->IoStatus.Status;
1775 __except( EXCEPTION_EXECUTE_HANDLER)
1777 ntStatus = GetExceptionCode();
1779 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1780 AFS_TRACE_LEVEL_ERROR,
1781 "AFSCachedWrite (%p) Exception thrown while preparing mdl write Status %08lX\n",
1786 if( !NT_SUCCESS( ntStatus))
1790 // Free up any potentially allocated mdl's
1793 CcMdlWriteComplete( pFileObject,
1797 Irp->MdlAddress = NULL;
1799 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1800 AFS_TRACE_LEVEL_ERROR,
1801 "AFSCachedWrite (%p) Failed to process MDL write Status %08lX\n",
1806 try_return( ntStatus);
1809 liCurrentOffset.QuadPart = StartingByte.QuadPart;
1811 while( ulTotalLen > 0)
1814 ntStatus = STATUS_SUCCESS;
1816 if( pCurrentMdl != NULL)
1819 pSystemBuffer = MmGetSystemAddressForMdlSafe( pCurrentMdl,
1820 NormalPagePriority);
1822 ulCurrentIO = MmGetMdlByteCount( pCurrentMdl);
1824 if( ulCurrentIO > ulTotalLen)
1826 ulCurrentIO = ulTotalLen;
1832 pSystemBuffer = AFSLockSystemBuffer( Irp,
1835 ulCurrentIO = ulTotalLen;
1838 if( pSystemBuffer == NULL)
1841 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1842 AFS_TRACE_LEVEL_ERROR,
1843 "AFSCachedWrite (%p) Failed to lock system buffer\n",
1846 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
1852 if( !CcCopyWrite( pFileObject,
1859 // Failed to process request.
1862 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
1863 AFS_TRACE_LEVEL_ERROR,
1864 "AFSCachedWrite (%p) Failed to issue CcCopyWrite %wZ @ %0I64X Status %08lX\n",
1866 &pFileObject->FileName,
1867 liCurrentOffset.QuadPart,
1868 Irp->IoStatus.Status));
1870 try_return( ntStatus = STATUS_UNSUCCESSFUL);
1873 __except( EXCEPTION_EXECUTE_HANDLER)
1876 ntStatus = GetExceptionCode();
1878 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1879 AFS_TRACE_LEVEL_ERROR,
1880 "AFSCachedWrite (%p) CcCopyWrite Threw exception %wZ @ %0I64X Status %08lX\n",
1882 &pFileObject->FileName,
1883 liCurrentOffset.QuadPart,
1887 if( !NT_SUCCESS( ntStatus))
1889 try_return( ntStatus);
1893 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1897 // We have detected a file we do a write through with.
1900 CcFlushCache(&pFcb->NPFcb->SectionObjectPointers,
1905 if( !NT_SUCCESS( iosbFlush.Status))
1908 AFSDbgTrace(( AFS_SUBSYSTEM_IO_PROCESSING,
1909 AFS_TRACE_LEVEL_ERROR,
1910 "AFSCachedWrite (%p) CcFlushCache failure %wZ FID %08lX-%08lX-%08lX-%08lX Status 0x%08lX Bytes 0x%08lX\n",
1912 &pFileObject->FileName,
1913 pFcb->ObjectInformation->FileId.Cell,
1914 pFcb->ObjectInformation->FileId.Volume,
1915 pFcb->ObjectInformation->FileId.Vnode,
1916 pFcb->ObjectInformation->FileId.Unique,
1918 iosbFlush.Information));
1920 try_return( ntStatus = iosbFlush.Status);
1924 if( ulTotalLen <= ulCurrentIO)
1929 liCurrentOffset.QuadPart += ulCurrentIO;
1931 ulTotalLen -= ulCurrentIO;
1933 pCurrentMdl = pCurrentMdl->Next;
1938 if( NT_SUCCESS( ntStatus))
1941 Irp->IoStatus.Information = ByteCount;
1944 BooleanFlagOn(pFileObject->Flags, (FO_NO_INTERMEDIATE_BUFFERING + FO_WRITE_THROUGH)))
1947 // Write through asked for... Set things so that we get
1948 // flush when the worker thread next wakes up
1950 pFcb->Specific.File.LastServerFlush.QuadPart = 0;
1954 AFSCompleteRequest( Irp,
1963 AFSExtendingWrite( IN AFSFcb *Fcb,
1964 IN PFILE_OBJECT FileObject,
1965 IN LONGLONG NewLength)
1967 LARGE_INTEGER liSaveFileSize = Fcb->Header.FileSize;
1968 LARGE_INTEGER liSaveAllocation = Fcb->Header.AllocationSize;
1969 NTSTATUS ntStatus = STATUS_SUCCESS;
1970 AFSCcb *pCcb = (AFSCcb *)FileObject->FsContext2;
1972 if( NewLength > Fcb->Header.AllocationSize.QuadPart)
1975 Fcb->Header.AllocationSize.QuadPart = NewLength;
1977 Fcb->ObjectInformation->AllocationSize = Fcb->Header.AllocationSize;
1980 if( NewLength > Fcb->Header.FileSize.QuadPart)
1983 Fcb->Header.FileSize.QuadPart = NewLength;
1985 Fcb->ObjectInformation->EndOfFile = Fcb->Header.FileSize;
1992 ntStatus = AFSUpdateFileInformation( &Fcb->ObjectInformation->ParentFileId,
1993 Fcb->ObjectInformation,
1996 if (NT_SUCCESS(ntStatus))
1999 KeQuerySystemTime( &Fcb->ObjectInformation->ChangeTime);
2001 SetFlag( Fcb->Flags, AFS_FCB_FLAG_FILE_MODIFIED | AFS_FCB_FLAG_UPDATE_CHANGE_TIME);
2004 // If the file is currently cached, then let the MM know about the extension
2007 if( CcIsFileCached( FileObject))
2009 CcSetFileSizes( FileObject,
2010 (PCC_FILE_SIZES)&Fcb->Header.AllocationSize);
2015 Fcb->Header.FileSize = liSaveFileSize;
2016 Fcb->Header.AllocationSize = liSaveAllocation;
2020 // DownConvert file resource to shared
2022 ExConvertExclusiveToSharedLite( &Fcb->NPFcb->Resource);
2028 AFSShareWrite( IN PDEVICE_OBJECT DeviceObject,
2032 UNREFERENCED_PARAMETER(DeviceObject);
2033 NTSTATUS ntStatus = STATUS_SUCCESS;
2034 PIO_STACK_LOCATION pIrpSp = IoGetCurrentIrpStackLocation( Irp);
2035 AFSFcb *pFcb = NULL;
2036 AFSCcb *pCcb = NULL;
2037 AFSPipeIORequestCB *pIoRequest = NULL;
2038 void *pBuffer = NULL;
2039 AFSPipeIOResultCB stIoResult;
2040 ULONG ulBytesReturned = 0;
2045 pFcb = (AFSFcb *)pIrpSp->FileObject->FsContext;
2047 pCcb = (AFSCcb *)pIrpSp->FileObject->FsContext2;
2049 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2050 AFS_TRACE_LEVEL_VERBOSE,
2051 "AFSShareWrite On pipe %wZ Length %08lX\n",
2052 &pCcb->DirectoryCB->NameInformation.FileName,
2053 pIrpSp->Parameters.Write.Length));
2055 if( pIrpSp->Parameters.Write.Length == 0)
2059 // Nothing to do in this case
2062 try_return( ntStatus);
2066 // Retrieve the buffer for the read request
2069 pBuffer = AFSLockSystemBuffer( Irp,
2070 pIrpSp->Parameters.Write.Length);
2072 if( pBuffer == NULL)
2075 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2076 AFS_TRACE_LEVEL_ERROR,
2077 "AFSShareWrite Failed to map buffer on pipe %wZ\n",
2078 &pCcb->DirectoryCB->NameInformation.FileName));
2080 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2083 AFSAcquireShared( &pFcb->NPFcb->Resource,
2086 pIoRequest = (AFSPipeIORequestCB *)AFSExAllocatePoolWithTag( PagedPool,
2087 sizeof( AFSPipeIORequestCB) +
2088 pIrpSp->Parameters.Write.Length,
2089 AFS_GENERIC_MEMORY_14_TAG);
2091 if( pIoRequest == NULL)
2094 try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
2097 RtlZeroMemory( pIoRequest,
2098 sizeof( AFSPipeIORequestCB) + pIrpSp->Parameters.Write.Length);
2100 pIoRequest->RequestId = pCcb->RequestID;
2102 pIoRequest->RootId = pFcb->ObjectInformation->VolumeCB->ObjectInformation.FileId;
2104 pIoRequest->BufferLength = pIrpSp->Parameters.Write.Length;
2106 RtlCopyMemory( (void *)((char *)pIoRequest + sizeof( AFSPipeIORequestCB)),
2108 pIrpSp->Parameters.Write.Length);
2110 stIoResult.BytesProcessed = 0;
2112 ulBytesReturned = sizeof( AFSPipeIOResultCB);
2115 // Issue the open request to the service
2118 ntStatus = AFSProcessRequest( AFS_REQUEST_TYPE_PIPE_WRITE,
2119 AFS_REQUEST_FLAG_SYNCHRONOUS,
2121 &pCcb->DirectoryCB->NameInformation.FileName,
2126 sizeof( AFSPipeIORequestCB) +
2127 pIrpSp->Parameters.Write.Length,
2131 if( !NT_SUCCESS( ntStatus))
2134 AFSDbgTrace(( AFS_SUBSYSTEM_FILE_PROCESSING,
2135 AFS_TRACE_LEVEL_ERROR,
2136 "AFSShareWrite (%p) Failed service write Status %08lX\n",
2140 try_return( ntStatus);
2143 AFSDbgTrace(( AFS_SUBSYSTEM_PIPE_PROCESSING,
2144 AFS_TRACE_LEVEL_VERBOSE,
2145 "AFSShareWrite Completed on pipe %wZ Length read %08lX\n",
2146 &pCcb->DirectoryCB->NameInformation.FileName,
2147 stIoResult.BytesProcessed));
2149 Irp->IoStatus.Information = stIoResult.BytesProcessed;
2156 AFSReleaseResource( &pFcb->NPFcb->Resource);
2159 if( pIoRequest != NULL)
2162 AFSExFreePoolWithTag( pIoRequest, AFS_GENERIC_MEMORY_14_TAG);