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"
44 AFSWriteComplete( IN PDEVICE_OBJECT DeviceObject,
53 // This is the dispatch handler for the IRP_MJ_WRITE request. Since we want to
54 // allow the library to pend the write we need to lock the library for the
55 // duration of the thread calling the library but also for the life of the IRP.
56 // So this code path establishes an IO completion function.
60 // A status is returned for the function
64 AFSWrite( IN PDEVICE_OBJECT DeviceObject,
68 NTSTATUS ntStatus = STATUS_SUCCESS;
69 AFSDeviceExt *pControlDeviceExt = (AFSDeviceExt *)AFSDeviceObject->DeviceExtension;
74 if( DeviceObject == AFSDeviceObject)
77 ntStatus = STATUS_INVALID_DEVICE_REQUEST;
79 AFSCompleteRequest( Irp,
82 try_return( ntStatus);
86 // Check the state of the library
89 ntStatus = AFSCheckLibraryState( Irp);
91 if( !NT_SUCCESS( ntStatus) ||
92 ntStatus == STATUS_PENDING)
95 if( ntStatus != STATUS_PENDING)
98 AFSCompleteRequest( Irp, ntStatus);
101 try_return( ntStatus);
105 // Increment the outstanding IO count again - this time for the
106 // completion routine.
109 ntStatus = AFSCheckLibraryState( Irp);
111 if( !NT_SUCCESS( ntStatus) ||
112 ntStatus == STATUS_PENDING)
115 AFSClearLibraryRequest();
117 if( ntStatus != STATUS_PENDING)
120 AFSCompleteRequest( Irp, ntStatus);
123 try_return( ntStatus);
127 // And send it down, but arrange to capture the comletion
128 // so we can free our lock against unloading.
131 IoCopyCurrentIrpStackLocationToNext( Irp);
133 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
134 AFS_TRACE_LEVEL_VERBOSE,
135 "Setting AFSWriteComplete as IoCompletion Routine Irp %p\n",
138 IoSetCompletionRoutine( Irp, AFSWriteComplete, NULL, TRUE, TRUE, TRUE);
140 ntStatus = IoCallDriver( pControlDeviceExt->Specific.Control.LibraryDeviceObject,
144 // Indicate the library/thread pair is done with the request
147 AFSClearLibraryRequest();
153 __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
156 ntStatus = STATUS_INSUFFICIENT_RESOURCES;
158 AFSDumpTraceFilesFnc();
169 AFSWriteComplete( IN PDEVICE_OBJECT DeviceObject,
173 UNREFERENCED_PARAMETER(DeviceObject);
174 UNREFERENCED_PARAMETER(Irp);
175 UNREFERENCED_PARAMETER(Context);
176 BOOLEAN bPending = FALSE;
179 // Indicate the library/IRP pair is done with the request
182 AFSClearLibraryRequest();
184 if (Irp->PendingReturned) {
188 IoMarkIrpPending(Irp);
191 AFSDbgLogMsg( AFS_SUBSYSTEM_IO_PROCESSING,
192 AFS_TRACE_LEVEL_VERBOSE,
193 "AFSWriteComplete Irp %p%s\n",
195 bPending ? " PENDING" : "");
197 return STATUS_CONTINUE_COMPLETION;