/*++ Copyright (c) 1996-2000 Microsoft Corporation Module Name: DevIoSup.c Abstract: This module implements the low lever disk read/write support for Udfs. // @@BEGIN_DDKSPLIT Author: Dan Lovinger [DanLo] 11-Jun-1996 Tom Jolly [tomjolly] 21-Jan-2000 Revision History: // @@END_DDKSPLIT --*/ #include "UdfProcs.h" // // The Bug check file id for this module // #define BugCheckFileId (UDFS_BUG_CHECK_DEVIOSUP) // // The local debug trace level // #define Dbg (UDFS_DEBUG_LEVEL_DEVIOSUP) // // Local structure definitions // // // An array of these structures is passed to UdfMultipleAsync describing // a set of runs to execute in parallel. // typedef struct _IO_RUN { // // Disk offset to read from and number of bytes to read. These // must be a multiple of a sector and the disk offset is also a // multiple of sector. // LONGLONG DiskOffset; ULONG DiskByteCount; // // Current position in user buffer. This is the final destination for // this portion of the Io transfer. // PVOID UserBuffer; // // Buffer to perform the transfer to. If this is the same as the // user buffer above then we are using the user's buffer. Otherwise // we either allocated a temporary buffer or are using a different portion // of the user's buffer. // // TransferBuffer - Read full sectors into this location. This can // be a pointer into the user's buffer at the exact location the // data should go. It can also be an earlier point in the user's // buffer if the complete I/O doesn't start on a sector boundary. // It may also be a pointer into an allocated buffer. // // TransferByteCount - Count of bytes to transfer to user's buffer. A // value of zero indicates that we did do the transfer into the // user's buffer directly. // // TransferBufferOffset - Offset in this buffer to begin the transfer // to the user's buffer. // PVOID TransferBuffer; ULONG TransferByteCount; ULONG TransferBufferOffset; // // This is the Mdl describing the locked pages in memory. It may // be allocated to describe the allocated buffer. Or it may be // the Mdl in the originating Irp. The MdlOffset is the offset of // the current buffer from the beginning of the buffer described by // the Mdl below. If the TransferMdl is not the same as the Mdl // in the user's Irp then we know we have allocated it. // PMDL TransferMdl; PVOID TransferVirtualAddress; // // Associated Irp used to perform the Io. // PIRP SavedIrp; } IO_RUN; typedef IO_RUN *PIO_RUN; #define MAX_PARALLEL_IOS 5 // // Local support routines // BOOLEAN UdfPrepareBuffers ( IN PIRP_CONTEXT IrpContext, IN PIRP Irp, IN PFCB Fcb, IN PCCB Ccb, IN PVOID UserBuffer, IN ULONG UserBufferOffset, IN LONGLONG StartingOffset, IN ULONG ByteCount, IN PIO_RUN IoRuns, IN PULONG RunCount, IN PULONG ThisByteCount, OUT PBOOLEAN SparseRuns ); BOOLEAN UdfFinishBuffers ( IN PIRP_CONTEXT IrpContext, IN PIO_RUN IoRuns, IN ULONG RunCount, IN BOOLEAN FinalCleanup ); VOID UdfMultipleAsync ( IN PIRP_CONTEXT IrpContext, IN ULONG RunCount, IN PIO_RUN IoRuns ); VOID UdfSingleAsync ( IN PIRP_CONTEXT IrpContext, IN LONGLONG ByteOffset, IN ULONG ByteCount ); VOID UdfWaitSync ( IN PIRP_CONTEXT IrpContext ); NTSTATUS UdfMultiSyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ); NTSTATUS UdfMultiAsyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ); NTSTATUS UdfSingleSyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ); NTSTATUS UdfSingleAsyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ); #ifdef ALLOC_PRAGMA #endif #ifdef ALLOC_PRAGMA #pragma alloc_text(PAGE, UdfCreateUserMdl) #pragma alloc_text(PAGE, UdfMultipleAsync) #pragma alloc_text(PAGE, UdfNonCachedRead) #pragma alloc_text(PAGE, UdfFinishBuffers) #pragma alloc_text(PAGE, UdfPrepareBuffers) #pragma alloc_text(PAGE, UdfSingleAsync) #pragma alloc_text(PAGE, UdfWaitSync) #pragma alloc_text(PAGE, UdfPerformDevIoCtrl) #pragma alloc_text(PAGE, UdfReadSectors) #endif NTSTATUS UdfNonCachedRead ( IN PIRP_CONTEXT IrpContext, IN PFCB Fcb, IN PCCB Ccb, IN LONGLONG StartingOffset, IN ULONG ByteCount ) /*++ Routine Description: This routine performs the non-cached reads of sectors. This is done by performing the following in a loop. Fill in the IoRuns array for the next block of Io. Send the Io to the device. Perform any cleanup on the Io runs array. We will not do async Io to any request that generates non-aligned Io. Also we will not perform async Io if it will exceed the size of our IoRuns array. These should be the unusual cases but we will raise or return CANT_WAIT in this routine if we detect this case. Arguments: Fcb - Fcb representing the file to read. StartingOffset - Logical offset in the file to read from. ByteCount - Number of bytes to read. Return Value: NTSTATUS - Status indicating the result of the operation. --*/ { NTSTATUS Status = STATUS_SUCCESS; IO_RUN IoRuns[MAX_PARALLEL_IOS]; ULONG RunCount = 0; ULONG CleanupRunCount = 0; PVOID UserBuffer; ULONG UserBufferOffset = 0; LONGLONG CurrentOffset = StartingOffset; ULONG RemainingByteCount = ByteCount; ULONG ThisByteCount; BOOLEAN Unaligned; BOOLEAN SparseRuns; BOOLEAN FlushIoBuffers = FALSE; BOOLEAN FirstPass = TRUE; PAGED_CODE(); // // We want to make sure the user's buffer is locked in all cases. // if (IrpContext->Irp->MdlAddress == NULL) { UdfCreateUserMdl( IrpContext, ByteCount, TRUE, IoWriteAccess ); } // // Use a try-finally to perform the final cleanup. // try { UdfMapUserBuffer( IrpContext, &UserBuffer); // // Loop while there are more bytes to transfer. // do { // // Call prepare buffers to set up the next entries // in the IoRuns array. Remember if there are any // unaligned entries. // RtlZeroMemory( IoRuns, sizeof( IoRuns )); Unaligned = UdfPrepareBuffers( IrpContext, IrpContext->Irp, Fcb, Ccb, UserBuffer, UserBufferOffset, CurrentOffset, RemainingByteCount, IoRuns, &CleanupRunCount, &ThisByteCount, &SparseRuns ); RunCount = CleanupRunCount; // // Quickly finish if we wound up having no IO to perform. This will // occur in the presence of unrecorded sectors. // ASSERT( !(SparseRuns && FlagOn( Fcb->FcbState, FCB_STATE_EMBEDDED_DATA ))); if (RunCount == 0) { try_leave( Status = IrpContext->Irp->IoStatus.Status = STATUS_SUCCESS ); } // // If this is an async request and there aren't enough entries // in the Io array then post the request. This routine will // always raise if we are doing any unaligned Io for an // async request. // if ((ThisByteCount < RemainingByteCount) && !FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { UdfRaiseStatus( IrpContext, STATUS_CANT_WAIT ); } // // If the entire Io is contained in a single run then // we can pass the Io down to the driver. Send the driver down // and wait on the result if this is synchronous. We cannot // do this simple form (just chucking the IRP down) if some // sparse runs were encountered. // if ((RunCount == 1) && !Unaligned && !SparseRuns && FirstPass) { UdfSingleAsync( IrpContext, IoRuns[0].DiskOffset, IoRuns[0].DiskByteCount ); // // No cleanup needed for the IoRuns array here. // CleanupRunCount = 0; // // Wait if we are synchronous, otherwise return // if (FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { UdfWaitSync( IrpContext ); Status = IrpContext->Irp->IoStatus.Status; // // Our completion routine will free the Io context but // we do want to return STATUS_PENDING. // } else { ClearFlag( IrpContext->Flags, IRP_CONTEXT_FLAG_ALLOC_IO ); Status = STATUS_PENDING; } try_leave( NOTHING ); } // // Otherwise we will perform multiple Io to read in the data. // UdfMultipleAsync( IrpContext, RunCount, IoRuns ); // // If this is a synchronous request then perform any necessary // post-processing. // if (FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { // // Wait for the request to complete. // UdfWaitSync( IrpContext ); Status = IrpContext->Irp->IoStatus.Status; // // Exit this loop if there is an error. // if (!NT_SUCCESS( Status )) { try_leave( NOTHING ); } // // Perform post read operations on the IoRuns if // necessary. // if (Unaligned && UdfFinishBuffers( IrpContext, IoRuns, RunCount, FALSE )) { FlushIoBuffers = TRUE; } // // No cleanup needed on the IoRuns now. // CleanupRunCount = 0; // // Exit this loop if there are no more bytes to transfer // or we have any error. // RemainingByteCount -= ThisByteCount; CurrentOffset += ThisByteCount; UserBuffer = Add2Ptr( UserBuffer, ThisByteCount, PVOID ); UserBufferOffset += ThisByteCount; // // Otherwise this is an asynchronous request. Always return // STATUS_PENDING. // } else { ClearFlag( IrpContext->Flags, IRP_CONTEXT_FLAG_ALLOC_IO ); CleanupRunCount = 0; try_leave( Status = STATUS_PENDING ); break; } FirstPass = FALSE; } while (RemainingByteCount != 0); // // Flush the hardware cache if we performed any copy operations. // if (FlushIoBuffers) { KeFlushIoBuffers( IrpContext->Irp->MdlAddress, TRUE, FALSE ); } } finally { DebugUnwind( "UdfNonCachedRead" ); // // Perform final cleanup on the IoRuns if necessary. // if (CleanupRunCount != 0) { UdfFinishBuffers( IrpContext, IoRuns, CleanupRunCount, TRUE ); } } return Status; } NTSTATUS UdfCreateUserMdl ( IN PIRP_CONTEXT IrpContext, IN ULONG BufferLength, IN BOOLEAN RaiseOnError, IN ULONG Operation ) /*++ Routine Description: This routine locks the specified buffer for read access (we only write into the buffer). The file system requires this routine since it does not ask the I/O system to lock its buffers for direct I/O. This routine may only be called from the Fsd while still in the user context. This routine is only called if there is not already an Mdl. Arguments: BufferLength - Length of user buffer. RaiseOnError - Indicates if our caller wants this routine to raise on an error condition. Return Value: NTSTATUS - Status from this routine. Error status only returned if RaiseOnError is FALSE. --*/ { NTSTATUS Status = STATUS_INSUFFICIENT_RESOURCES; PMDL Mdl; PAGED_CODE(); ASSERT_IRP_CONTEXT( IrpContext ); ASSERT_IRP( IrpContext->Irp ); ASSERT( IrpContext->Irp->MdlAddress == NULL ); // // Allocate the Mdl, and Raise if we fail. // Mdl = IoAllocateMdl( IrpContext->Irp->UserBuffer, BufferLength, FALSE, FALSE, IrpContext->Irp ); if (Mdl != NULL) { // // Now probe the buffer described by the Irp. If we get an exception, // deallocate the Mdl and return the appropriate "expected" status. // try { MmProbeAndLockPages( Mdl, IrpContext->Irp->RequestorMode, Operation ); Status = STATUS_SUCCESS; } except(EXCEPTION_EXECUTE_HANDLER) { Status = GetExceptionCode(); IoFreeMdl( Mdl ); IrpContext->Irp->MdlAddress = NULL; if (!FsRtlIsNtstatusExpected( Status )) { Status = STATUS_INVALID_USER_BUFFER; } } } // // Check if we are to raise or return // if (Status != STATUS_SUCCESS) { if (RaiseOnError) { UdfRaiseStatus( IrpContext, Status ); } } // // Return the status code. // return Status; } NTSTATUS UdfPerformDevIoCtrl ( IN PIRP_CONTEXT IrpContext, IN ULONG IoControlCode, IN PDEVICE_OBJECT Device, IN PVOID InputBuffer OPTIONAL, IN ULONG InputBufferLength, OUT PVOID OutputBuffer OPTIONAL, IN ULONG OutputBufferLength, IN BOOLEAN InternalDeviceIoControl, IN BOOLEAN OverrideVerify, OUT PIO_STATUS_BLOCK Iosb OPTIONAL ) /*++ Routine Description: This routine is called to perform DevIoCtrl functions internally within the filesystem. We take the status from the driver and return it to our caller. Arguments: IoControlCode - Code to send to driver. Device - This is the device to send the request to. OutPutBuffer - Pointer to output buffer. OutputBufferLength - Length of output buffer above. InternalDeviceIoControl - Indicates if this is an internal or external Io control code. OverrideVerify - Indicates if we should tell the driver not to return STATUS_VERIFY_REQUIRED for mount and verify. Iosb - If specified, we return the results of the operation here. Return Value: NTSTATUS - Status returned by next lower driver. --*/ { NTSTATUS Status; PIRP Irp; KEVENT Event; IO_STATUS_BLOCK LocalIosb; PIO_STATUS_BLOCK IosbToUse = &LocalIosb; PAGED_CODE(); // // Check if the user gave us an Iosb. // if (ARGUMENT_PRESENT( Iosb )) { IosbToUse = Iosb; } IosbToUse->Status = 0; IosbToUse->Information = 0; KeInitializeEvent( &Event, NotificationEvent, FALSE ); Irp = IoBuildDeviceIoControlRequest( IoControlCode, Device, InputBuffer, InputBufferLength, OutputBuffer, OutputBufferLength, InternalDeviceIoControl, &Event, IosbToUse ); if (Irp == NULL) { return STATUS_INSUFFICIENT_RESOURCES; } if (OverrideVerify) { SetFlag( IoGetNextIrpStackLocation( Irp )->Flags, SL_OVERRIDE_VERIFY_VOLUME ); } Status = IoCallDriver( Device, Irp ); // // We check for device not ready by first checking Status // and then if status pending was returned, the Iosb status // value. // if (Status == STATUS_PENDING) { (VOID) KeWaitForSingleObject( &Event, Executive, KernelMode, FALSE, (PLARGE_INTEGER)NULL ); Status = IosbToUse->Status; } ASSERT( !(OverrideVerify && (STATUS_VERIFY_REQUIRED == Status))); return Status; UNREFERENCED_PARAMETER( IrpContext ); } NTSTATUS UdfReadSectors ( IN PIRP_CONTEXT IrpContext, IN LONGLONG StartingOffset, IN ULONG ByteCount, IN BOOLEAN ReturnError, IN OUT PVOID Buffer, IN PDEVICE_OBJECT TargetDeviceObject ) /*++ Routine Description: This routine is called to transfer sectors from the disk to a specified buffer. It is used for mount and volume verify operations. This routine is synchronous, it will not return until the operation is complete or until the operation fails. The routine allocates an IRP and then passes this IRP to a lower level driver. Errors may occur in the allocation of this IRP or in the operation of the lower driver. Arguments: StartingOffset - Logical offset on the disk to start the read. This must be on a sector boundary, no check is made here. ByteCount - Number of bytes to read. This is an integral number of sectors, or otherwise a value we know the driver can handle, no check is made here to confirm this. ReturnError - Indicates whether we should return TRUE or FALSE to indicate an error or raise an error condition. This only applies to the result of the IO. Any other error may cause a raise. Buffer - Buffer to transfer the disk data into. TargetDeviceObject - The device object for the volume to be read. Return Value: The final status of the operation. --*/ { PLONGLONG UseStartingOffset; LONGLONG LocalStartingOffset; NTSTATUS Status; KEVENT Event; PIRP Irp; PAGED_CODE(); DebugTrace(( +1, Dbg, "UdfReadSectors, %x%08x +%x -> %08x from DO %08x\n", ((PLARGE_INTEGER)&StartingOffset)->HighPart, ((PLARGE_INTEGER)&StartingOffset)->LowPart, ByteCount, Buffer, TargetDeviceObject )); // // For the time being, we assume that we only read sector-at-a-time. // This simplifies sparing, and is the only way I am aware of this // code would not be ready for blocksize != sectorsize. It just is // not worth writing dead (but straightforward) code right now. // ASSERT( IrpContext->Vcb == NULL || ByteCount == SectorSize( IrpContext->Vcb )); // // If the volume is spared (and at a point where sparing is possible), // check if a mapping needs to be performed. // if (IrpContext->Vcb && IrpContext->Vcb->Pcb && IrpContext->Vcb->Pcb->SparingMcb) { LONGLONG SparingPsn; if (FsRtlLookupLargeMcbEntry( IrpContext->Vcb->Pcb->SparingMcb, LlSectorsFromBytes( IrpContext->Vcb, StartingOffset ), &SparingPsn, NULL, NULL, NULL, NULL ) && SparingPsn != -1) { StartingOffset = BytesFromSectors( IrpContext->Vcb, (ULONG) SparingPsn ); } } // // Initialize the event. // KeInitializeEvent( &Event, NotificationEvent, FALSE ); // // Correct the starting offset by the method 2 fixup if neccesary. This also // assumes sector-at-a-time and sector == block so we don't need to fragment // the request or check if it spans a packet boundary. // // We assume that no fixups are required until a Vcb exists. This is true // since volume recognition may proceed in the first packet. // UseStartingOffset = &StartingOffset; if (IrpContext->Vcb && FlagOn( IrpContext->Vcb->VcbState, VCB_STATE_METHOD_2_FIXUP )) { LocalStartingOffset = UdfMethod2TransformByteOffset( IrpContext->Vcb, StartingOffset ); UseStartingOffset = &LocalStartingOffset; DebugTrace(( 0, Dbg, "UdfReadSectors, Method2 Fixup to %x%08x\n", ((PLARGE_INTEGER)UseStartingOffset)->HighPart, ((PLARGE_INTEGER)UseStartingOffset)->LowPart )); } // // Attempt to allocate the IRP. If unsuccessful, raise // STATUS_INSUFFICIENT_RESOURCES. // Irp = IoBuildSynchronousFsdRequest( IRP_MJ_READ, TargetDeviceObject, Buffer, ByteCount, (PLARGE_INTEGER) UseStartingOffset, &Event, &IrpContext->Irp->IoStatus ); if (Irp == NULL) { UdfRaiseStatus( IrpContext, STATUS_INSUFFICIENT_RESOURCES ); } // // Ignore the change line (verify) for mount and verify requests // SetFlag( IoGetNextIrpStackLocation( Irp )->Flags, SL_OVERRIDE_VERIFY_VOLUME ); // // Send the request down to the driver. If an error occurs return // it to the caller. // Status = IoCallDriver( TargetDeviceObject, Irp ); // // If the status was STATUS_PENDING then wait on the event. // if (Status == STATUS_PENDING) { Status = KeWaitForSingleObject( &Event, Executive, KernelMode, FALSE, NULL ); // // On a successful wait pull the status out of the IoStatus block. // if (NT_SUCCESS( Status )) { Status = IrpContext->Irp->IoStatus.Status; } } DebugTrace(( -1, Dbg, "UdfReadSectors -> %08x\n", Status )); // // Check whether we should raise in the error case. // if (!NT_SUCCESS( Status ) && !ReturnError) { UdfNormalizeAndRaiseStatus( IrpContext, Status ); } return Status; } // // Local support routine // BOOLEAN UdfPrepareBuffers ( IN PIRP_CONTEXT IrpContext, IN PIRP Irp, IN PFCB Fcb, IN PCCB Ccb, IN PVOID UserBuffer, IN ULONG UserBufferOffset, IN LONGLONG StartingOffset, IN ULONG ByteCount, IN PIO_RUN IoRuns, IN PULONG RunCount, IN PULONG ThisByteCount, IN PBOOLEAN SparseRuns ) /*++ Routine Description: This routine is the worker routine which looks up each run of an IO request and stores an entry for it in the IoRuns array. If the run begins on an unaligned disk boundary then we will allocate a buffer and Mdl for the unaligned portion and put it in the IoRuns entry. This routine will raise CANT_WAIT if an unaligned transfer is encountered and this request can't wait. Arguments: Irp - Originating Irp for this request. Fcb - This is the Fcb for this data stream. It may be a file, directory, path table or the volume file. UserBuffer - Current position in the user's buffer. UserBufferOffset - Offset from the start of the original user buffer. StartingOffset - Offset in the stream to begin the read. ByteCount - Number of bytes to read. We will fill the IoRuns array up to this point. We will stop early if we exceed the maximum number of parallel Ios we support. IoRuns - Pointer to the IoRuns array. The entire array is zeroes when this routine is called. RunCount - Number of entries in the IoRuns array filled here. ThisByteCount - Number of bytes described by the IoRun entries. Will not exceed the ByteCount passed in. SparseRuns - Will indicate whether sparse runs were a component of the range returned. While not part of the IoRuns, this will affect our ability to do simple IO. Return Value: BOOLEAN - TRUE if one of the entries in an unaligned buffer (provided this is synchronous). FALSE otherwise. --*/ { PVCB Vcb; BOOLEAN Recorded; BOOLEAN FoundUnaligned = FALSE; PIO_RUN ThisIoRun = IoRuns; // // Following indicate where we are in the current transfer. Current // position in the file and number of bytes yet to transfer from // this position. // ULONG RemainingByteCount = ByteCount; LONGLONG CurrentFileOffset = StartingOffset; // // Following indicate the state of the user's buffer. We have // the destination of the next transfer and its offset in the // buffer. We also have the next available position in the buffer // available for a scratch buffer. We will align this up to a sector // boundary. // PVOID CurrentUserBuffer = UserBuffer; ULONG CurrentUserBufferOffset = UserBufferOffset; PVOID ScratchUserBuffer = UserBuffer; ULONG ScratchUserBufferOffset = UserBufferOffset; // // The following is the next contiguous bytes on the disk to // transfer. Read from the allocation package. // LONGLONG DiskOffset; ULONG CurrentByteCount; PAGED_CODE(); Vcb = Fcb->Vcb; // // Initialize the RunCount, ByteCount and SparseRuns. // *RunCount = 0; *ThisByteCount = 0; *SparseRuns = FALSE; // // Loop while there are more bytes to process or there are // available entries in the IoRun array. // while (TRUE) { *RunCount += 1; // // Initialize the current position in the IoRuns array. // Find the user's buffer for this portion of the transfer. // ThisIoRun->UserBuffer = CurrentUserBuffer; // // Find the allocation information for the current offset in the // stream. // Recorded = UdfLookupAllocation( IrpContext, Fcb, Ccb, CurrentFileOffset, &DiskOffset, &CurrentByteCount ); // // Limit ourselves to the data requested. // if (CurrentByteCount > RemainingByteCount) { CurrentByteCount = RemainingByteCount; } // // Handle the case of unrecorded data first. // if (!Recorded) { // // Note that we did not consume an entry. // *RunCount -= 1; // // Immediately zero the user buffer and indicate that we found sparse // runs to the caller. // RtlZeroMemory( CurrentUserBuffer, CurrentByteCount ); *SparseRuns = TRUE; // // Push the scratch buffer pointers forward so that we don't stomp // on the zeroed buffer. // ScratchUserBuffer = Add2Ptr( CurrentUserBuffer, CurrentByteCount, PVOID ); ScratchUserBufferOffset += CurrentByteCount; // // Handle the case where this is an unaligned transfer. The // following must all be true for this to be an aligned transfer. // // Disk offset on a 2048 byte boundary (Start of transfer) // // Byte count is a multiple of 2048 (Length of transfer) // // Current buffer offset is also on a 2048 byte boundary. // // If the ByteCount is at least one sector then do the // unaligned transfer only for the tail. We can use the // user's buffer for the aligned portion. // } else if (SectorOffset( Vcb, DiskOffset ) || SectorOffset( Vcb, CurrentUserBufferOffset ) || (SectorOffset( Vcb, CurrentByteCount ) && CurrentByteCount < SectorSize( Vcb ))) { // // If we can't wait then raise. // if (!FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { UdfRaiseStatus( IrpContext, STATUS_CANT_WAIT ); } // // Remember the offset and the number of bytes out of // the transfer buffer to copy into the user's buffer. // We will truncate the current read to end on a sector // boundary. // ThisIoRun->TransferBufferOffset = SectorOffset( Vcb, DiskOffset ); // // Make sure this transfer ends on a sector boundary. // ThisIoRun->DiskOffset = LlSectorTruncate( Vcb, DiskOffset ); // // Check if we can use a free portion of the user's buffer. // If we can copy the bytes to an earlier portion of the // buffer then read into that location and slide the bytes // up. // // We can use the user's buffer if: // // The temporary location in the buffer is before the // final destination. // // There is at least one sector of data to read. // if ((ScratchUserBufferOffset + ThisIoRun->TransferBufferOffset < CurrentUserBufferOffset) && (ThisIoRun->TransferBufferOffset + CurrentByteCount >= SectorSize( Vcb ))) { ThisIoRun->DiskByteCount = SectorTruncate( Vcb, ThisIoRun->TransferBufferOffset + CurrentByteCount ); CurrentByteCount = ThisIoRun->DiskByteCount - ThisIoRun->TransferBufferOffset; ThisIoRun->TransferByteCount = CurrentByteCount; // // Point to the user's buffer and Mdl for this transfer. // ThisIoRun->TransferBuffer = ScratchUserBuffer; ThisIoRun->TransferMdl = Irp->MdlAddress; ThisIoRun->TransferVirtualAddress = Add2Ptr( Irp->UserBuffer, ScratchUserBufferOffset, PVOID ); ScratchUserBuffer = Add2Ptr( ScratchUserBuffer, ThisIoRun->DiskByteCount, PVOID ); ScratchUserBufferOffset += ThisIoRun->DiskByteCount; // // Otherwise we need to allocate an auxilary buffer for the next sector. // } else { // // Read up to a page containing the partial data // ThisIoRun->DiskByteCount = SectorAlign( Vcb, ThisIoRun->TransferBufferOffset + CurrentByteCount ); if (ThisIoRun->DiskByteCount > PAGE_SIZE) { ThisIoRun->DiskByteCount = PAGE_SIZE; } if (ThisIoRun->TransferBufferOffset + CurrentByteCount > ThisIoRun->DiskByteCount) { CurrentByteCount = ThisIoRun->DiskByteCount - ThisIoRun->TransferBufferOffset; } ThisIoRun->TransferByteCount = CurrentByteCount; // // Allocate a buffer for the non-aligned transfer. // ThisIoRun->TransferBuffer = FsRtlAllocatePoolWithTag( UdfNonPagedPool, PAGE_SIZE, TAG_IO_BUFFER ); // // Allocate and build the Mdl to describe this buffer. // ThisIoRun->TransferMdl = IoAllocateMdl( ThisIoRun->TransferBuffer, PAGE_SIZE, FALSE, FALSE, NULL ); ThisIoRun->TransferVirtualAddress = ThisIoRun->TransferBuffer; if (ThisIoRun->TransferMdl == NULL) { IrpContext->Irp->IoStatus.Information = 0; UdfRaiseStatus( IrpContext, STATUS_INSUFFICIENT_RESOURCES ); } MmBuildMdlForNonPagedPool( ThisIoRun->TransferMdl ); } // // Remember we found an unaligned transfer. // FoundUnaligned = TRUE; // // Otherwise we use the buffer and Mdl from the original request. // } else { // // Truncate the read length to a sector-aligned value. We know // the length must be at least one sector or we wouldn't be // here now. // CurrentByteCount = SectorTruncate( Vcb, CurrentByteCount ); // // Read these sectors from the disk. // ThisIoRun->DiskOffset = DiskOffset; ThisIoRun->DiskByteCount = CurrentByteCount; // // Use the user's buffer and Mdl as our transfer buffer // and Mdl. // ThisIoRun->TransferBuffer = CurrentUserBuffer; ThisIoRun->TransferMdl = Irp->MdlAddress; ThisIoRun->TransferVirtualAddress = Add2Ptr( Irp->UserBuffer, CurrentUserBufferOffset, PVOID ); ScratchUserBuffer = Add2Ptr( CurrentUserBuffer, CurrentByteCount, PVOID ); ScratchUserBufferOffset += CurrentByteCount; } // // Update our position in the transfer and the RunCount and // ByteCount for the user. // RemainingByteCount -= CurrentByteCount; // // Break out if no more positions in the IoRuns array or // we have all of the bytes accounted for. // *ThisByteCount += CurrentByteCount; if ((RemainingByteCount == 0) || (*RunCount == MAX_PARALLEL_IOS)) { break; } // // Update our pointers for the user's buffer. // ThisIoRun = IoRuns + *RunCount; CurrentUserBuffer = Add2Ptr( CurrentUserBuffer, CurrentByteCount, PVOID ); CurrentUserBufferOffset += CurrentByteCount; CurrentFileOffset += CurrentByteCount; } return FoundUnaligned; } // // Local support routine // BOOLEAN UdfFinishBuffers ( IN PIRP_CONTEXT IrpContext, IN PIO_RUN IoRuns, IN ULONG RunCount, IN BOOLEAN FinalCleanup ) /*++ Routine Description: This routine is called to perform any data transferred required for unaligned Io or to perform the final cleanup of the IoRuns array. In all cases this is where we will deallocate any buffer and mdl allocated to perform the unaligned transfer. If this is not the final cleanup then we also transfer the bytes to the user buffer and flush the hardware cache. We walk backwards through the run array because we may be shifting data in the user's buffer. Typical case is where we allocated a buffer for the first part of a read and then used the user's buffer for the next section (but stored it at the beginning of the buffer. Arguments: IoRuns - Pointer to the IoRuns array. RunCount - Number of entries in the IoRuns array filled here. FinalCleanup - Indicates if we should be deallocating temporary buffers (TRUE) or transferring bytes for a unaligned transfers and deallocating the buffers (FALSE). Flush the system cache if transferring data. Return Value: BOOLEAN - TRUE if this request needs the Io buffers to be flushed, FALSE otherwise. --*/ { BOOLEAN FlushIoBuffers = FALSE; ULONG RemainingEntries = RunCount; PIO_RUN ThisIoRun = &IoRuns[RunCount - 1]; PAGED_CODE(); // // Walk through each entry in the IoRun array. // while (RemainingEntries != 0) { // // We only need to deal with the case of an unaligned transfer. // if (ThisIoRun->TransferByteCount != 0) { // // If not the final cleanup then transfer the data to the // user's buffer and remember that we will need to flush // the user's buffer to memory. // if (!FinalCleanup) { // // If we are shifting in the user's buffer then use // MoveMemory. // if (ThisIoRun->TransferMdl == IrpContext->Irp->MdlAddress) { RtlMoveMemory( ThisIoRun->UserBuffer, Add2Ptr( ThisIoRun->TransferBuffer, ThisIoRun->TransferBufferOffset, PVOID ), ThisIoRun->TransferByteCount ); } else { RtlCopyMemory( ThisIoRun->UserBuffer, Add2Ptr( ThisIoRun->TransferBuffer, ThisIoRun->TransferBufferOffset, PVOID ), ThisIoRun->TransferByteCount ); } FlushIoBuffers = TRUE; } // // Free any Mdl we may have allocated. If the Mdl isn't // present then we must have failed during the allocation // phase. // if (ThisIoRun->TransferMdl != IrpContext->Irp->MdlAddress) { if (ThisIoRun->TransferMdl != NULL) { IoFreeMdl( ThisIoRun->TransferMdl ); } // // Now free any buffer we may have allocated. If the Mdl // doesn't match the original Mdl then free the buffer. // if (ThisIoRun->TransferBuffer != NULL) { UdfFreePool( &ThisIoRun->TransferBuffer ); } } } // // Now handle the case where we failed in the process // of allocating associated Irps and Mdls. // if (ThisIoRun->SavedIrp != NULL) { if (ThisIoRun->SavedIrp->MdlAddress != NULL) { IoFreeMdl( ThisIoRun->SavedIrp->MdlAddress ); } IoFreeIrp( ThisIoRun->SavedIrp ); } // // Move to the previous IoRun entry. // ThisIoRun -= 1; RemainingEntries -= 1; } // // If we copied any data then flush the Io buffers. // return FlushIoBuffers; } // // Local support routine // VOID UdfMultipleAsync ( IN PIRP_CONTEXT IrpContext, IN ULONG RunCount, IN PIO_RUN IoRuns ) /*++ Routine Description: This routine first does the initial setup required of a Master IRP that is going to be completed using associated IRPs. This routine should not be used if only one async request is needed, instead the single read async routines should be called. A context parameter is initialized, to serve as a communications area between here and the common completion routine. Next this routine reads or writes one or more contiguous sectors from a device asynchronously, and is used if there are multiple reads for a master IRP. A completion routine is used to synchronize with the completion of all of the I/O requests started by calls to this routine. Also, prior to calling this routine the caller must initialize the IoStatus field in the Context, with the correct success status and byte count which are expected if all of the parallel transfers complete successfully. After return this status will be unchanged if all requests were, in fact, successful. However, if one or more errors occur, the IoStatus will be modified to reflect the error status and byte count from the first run (by Vbo) which encountered an error. I/O status from all subsequent runs will not be indicated. Arguments: RunCount - Supplies the number of multiple async requests that will be issued against the master irp. IoRuns - Supplies an array containing the Offset and ByteCount for the separate requests. Return Value: None. --*/ { PIO_COMPLETION_ROUTINE CompletionRoutine; PIO_STACK_LOCATION IrpSp; PMDL Mdl; PIRP Irp; PIRP MasterIrp; ULONG UnwindRunCount; PAGED_CODE(); // // Set up things according to whether this is truely async. // CompletionRoutine = UdfMultiSyncCompletionRoutine; if (!FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { CompletionRoutine = UdfMultiAsyncCompletionRoutine; } // // Initialize some local variables. // MasterIrp = IrpContext->Irp; // // Itterate through the runs, doing everything that can fail. // We let the cleanup in CdFinishBuffers clean up on error. // for (UnwindRunCount = 0; UnwindRunCount < RunCount; UnwindRunCount += 1) { // // Create an associated IRP, making sure there is one stack entry for // us, as well. // IoRuns[UnwindRunCount].SavedIrp = Irp = IoMakeAssociatedIrp( MasterIrp, (CCHAR)(IrpContext->Vcb->TargetDeviceObject->StackSize + 1) ); if (Irp == NULL) { IrpContext->Irp->IoStatus.Information = 0; UdfRaiseStatus( IrpContext, STATUS_INSUFFICIENT_RESOURCES ); } // // Allocate and build a partial Mdl for the request. // Mdl = IoAllocateMdl( IoRuns[UnwindRunCount].TransferVirtualAddress, IoRuns[UnwindRunCount].DiskByteCount, FALSE, FALSE, Irp ); if (Mdl == NULL) { IrpContext->Irp->IoStatus.Information = 0; UdfRaiseStatus( IrpContext, STATUS_INSUFFICIENT_RESOURCES ); } IoBuildPartialMdl( IoRuns[UnwindRunCount].TransferMdl, Mdl, IoRuns[UnwindRunCount].TransferVirtualAddress, IoRuns[UnwindRunCount].DiskByteCount ); // // Get the first IRP stack location in the associated Irp // IoSetNextIrpStackLocation( Irp ); IrpSp = IoGetCurrentIrpStackLocation( Irp ); // // Setup the Stack location to describe our read. // IrpSp->MajorFunction = IRP_MJ_READ; IrpSp->Parameters.Read.Length = IoRuns[UnwindRunCount].DiskByteCount; IrpSp->Parameters.Read.ByteOffset.QuadPart = IoRuns[UnwindRunCount].DiskOffset; // // Set up the completion routine address in our stack frame. // IoSetCompletionRoutine( Irp, CompletionRoutine, IrpContext->IoContext, TRUE, TRUE, TRUE ); // // Setup the next IRP stack location in the associated Irp for the disk // driver beneath us. // IrpSp = IoGetNextIrpStackLocation( Irp ); // // Setup the Stack location to do a read from the disk driver. // IrpSp->MajorFunction = IRP_MJ_READ; IrpSp->Parameters.Read.Length = IoRuns[UnwindRunCount].DiskByteCount; IrpSp->Parameters.Read.ByteOffset.QuadPart = IoRuns[UnwindRunCount].DiskOffset; } // // We only need to set the associated IRP count in the master irp to // make it a master IRP. But we set the count to one more than our // caller requested, because we do not want the I/O system to complete // the I/O. We also set our own count. // IrpContext->IoContext->IrpCount = RunCount; IrpContext->IoContext->MasterIrp = MasterIrp; // // We set the count in the master Irp to 1 since typically we // will clean up the associated irps ourselves. Setting this to one // means completing the last associated Irp with SUCCESS (in the async // case) will complete the master irp. // MasterIrp->AssociatedIrp.IrpCount = 1; // // Now that all the dangerous work is done, issue the Io requests // for (UnwindRunCount = 0; UnwindRunCount < RunCount; UnwindRunCount++) { Irp = IoRuns[UnwindRunCount].SavedIrp; IoRuns[UnwindRunCount].SavedIrp = NULL; // // If IoCallDriver returns an error, it has completed the Irp // and the error will be caught by our completion routines // and dealt with as a normal IO error. // (VOID) IoCallDriver( IrpContext->Vcb->TargetDeviceObject, Irp ); } return; } // // Local support routine // VOID UdfSingleAsync ( IN PIRP_CONTEXT IrpContext, IN LONGLONG ByteOffset, IN ULONG ByteCount ) /*++ Routine Description: This routine reads one or more contiguous sectors from a device asynchronously, and is used if there is only one read necessary to complete the IRP. It implements the read by simply filling in the next stack frame in the Irp, and passing it on. The transfer occurs to the single buffer originally specified in the user request. Arguments: ByteOffset - Supplies the starting Logical Byte Offset to begin reading from ByteCount - Supplies the number of bytes to read from the device Return Value: None. --*/ { PIO_STACK_LOCATION IrpSp; PIO_COMPLETION_ROUTINE CompletionRoutine; PAGED_CODE(); // // Set up things according to whether this is truely async. // if (FlagOn( IrpContext->Flags, IRP_CONTEXT_FLAG_WAIT )) { CompletionRoutine = UdfSingleSyncCompletionRoutine; } else { CompletionRoutine = UdfSingleAsyncCompletionRoutine; } // // Set up the completion routine address in our stack frame. // IoSetCompletionRoutine( IrpContext->Irp, CompletionRoutine, IrpContext->IoContext, TRUE, TRUE, TRUE ); // // Setup the next IRP stack location in the associated Irp for the disk // driver beneath us. // IrpSp = IoGetNextIrpStackLocation( IrpContext->Irp ); // // Setup the Stack location to do a read from the disk driver. // IrpSp->MajorFunction = IrpContext->MajorFunction; IrpSp->Parameters.Read.Length = ByteCount; IrpSp->Parameters.Read.ByteOffset.QuadPart = ByteOffset; // // Issue the Io request // // // If IoCallDriver returns an error, it has completed the Irp // and the error will be caught by our completion routines // and dealt with as a normal IO error. // (VOID)IoCallDriver( IrpContext->Vcb->TargetDeviceObject, IrpContext->Irp ); } // // Local support routine // VOID UdfWaitSync ( IN PIRP_CONTEXT IrpContext ) /*++ Routine Description: This routine waits for one or more previously started I/O requests from the above routines, by simply waiting on the event. Arguments: Return Value: None --*/ { PAGED_CODE(); KeWaitForSingleObject( &IrpContext->IoContext->SyncEvent, Executive, KernelMode, FALSE, NULL ); KeClearEvent( &IrpContext->IoContext->SyncEvent ); return; } // // Local support routine // NTSTATUS UdfMultiSyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ) /*++ Routine Description: This is the completion routine for all synchronous reads started via UdfMultipleAsync. The completion routine has has the following responsibilities: If the individual request was completed with an error, then this completion routine must see if this is the first error and remember the error status in the Context. If the IrpCount goes to 1, then it sets the event in the Context parameter to signal the caller that all of the asynch requests are done. Arguments: DeviceObject - Pointer to the file system device object. Irp - Pointer to the associated Irp which is being completed. (This Irp will no longer be accessible after this routine returns.) Context - The context parameter which was specified for all of the multiple asynch I/O requests for this MasterIrp. Return Value: The routine returns STATUS_MORE_PROCESSING_REQUIRED so that we can immediately complete the Master Irp without being in a race condition with the IoCompleteRequest thread trying to decrement the IrpCount in the Master Irp. --*/ { PUDF_IO_CONTEXT IoContext = Context; // // If we got an error (or verify required), remember it in the Irp // if (!NT_SUCCESS( Irp->IoStatus.Status )) { InterlockedExchange( &IoContext->Status, Irp->IoStatus.Status ); IoContext->MasterIrp->IoStatus.Information = 0; } // // We must do this here since IoCompleteRequest won't get a chance // on this associated Irp. // IoFreeMdl( Irp->MdlAddress ); IoFreeIrp( Irp ); if (InterlockedDecrement( &IoContext->IrpCount ) == 0) { // // Update the Master Irp with any error status from the associated Irps. // IoContext->MasterIrp->IoStatus.Status = IoContext->Status; KeSetEvent( &IoContext->SyncEvent, 0, FALSE ); } UNREFERENCED_PARAMETER( DeviceObject ); return STATUS_MORE_PROCESSING_REQUIRED; } // // Local support routine // NTSTATUS UdfMultiAsyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ) /*++ Routine Description: This is the completion routine for all asynchronous reads started via UdfMultipleAsync. The completion routine has has the following responsibilities: If the individual request was completed with an error, then this completion routine must see if this is the first error and remember the error status in the Context. Arguments: DeviceObject - Pointer to the file system device object. Irp - Pointer to the associated Irp which is being completed. (This Irp will no longer be accessible after this routine returns.) Context - The context parameter which was specified for all of the multiple asynch I/O requests for this MasterIrp. Return Value: Currently always returns STATUS_SUCCESS. --*/ { PUDF_IO_CONTEXT IoContext = Context; PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation( Irp ); // // If we got an error (or verify required), remember it in the Irp // if (!NT_SUCCESS( Irp->IoStatus.Status )) { InterlockedExchange( &IoContext->Status, Irp->IoStatus.Status ); } // // Decrement IrpCount and see if it goes to zero. // if (InterlockedDecrement( &IoContext->IrpCount ) == 0) { // // Mark the master Irp pending // IoMarkIrpPending( IoContext->MasterIrp ); // // Update the Master Irp with any error status from the associated Irps. // IoContext->MasterIrp->IoStatus.Status = IoContext->Status; // // Update the information field with the correct value. // IoContext->MasterIrp->IoStatus.Information = 0; if (NT_SUCCESS( IoContext->MasterIrp->IoStatus.Status )) { IoContext->MasterIrp->IoStatus.Information = IoContext->RequestedByteCount; } // // Now release the resource // ExReleaseResourceForThreadLite( IoContext->Resource, IoContext->ResourceThreadId ); // // and finally, free the context record. // UdfFreeIoContext( IoContext ); // // Return success in this case. // return STATUS_SUCCESS; } else { // // We need to cleanup the associated Irp and its Mdl. // IoFreeMdl( Irp->MdlAddress ); IoFreeIrp( Irp ); return STATUS_MORE_PROCESSING_REQUIRED; } UNREFERENCED_PARAMETER( DeviceObject ); } // // Local support routine // NTSTATUS UdfSingleSyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ) /*++ Routine Description: This is the completion routine for all reads started via UdfSingleAsync. The completion routine has has the following responsibilities: It sets the event in the Context parameter to signal the caller that all of the asynch requests are done. Arguments: DeviceObject - Pointer to the file system device object. Irp - Pointer to the Irp for this request. (This Irp will no longer be accessible after this routine returns.) Context - The context parameter which was specified in the call to UdfSingleAsynch. Return Value: The routine returns STATUS_MORE_PROCESSING_REQUIRED so that we can immediately complete the Master Irp without being in a race condition with the IoCompleteRequest thread trying to decrement the IrpCount in the Master Irp. --*/ { // // Store the correct information field into the Irp. // if (!NT_SUCCESS( Irp->IoStatus.Status )) { Irp->IoStatus.Information = 0; } KeSetEvent( &((PUDF_IO_CONTEXT)Context)->SyncEvent, 0, FALSE ); return STATUS_MORE_PROCESSING_REQUIRED; UNREFERENCED_PARAMETER( DeviceObject ); } // // Local support routine // NTSTATUS UdfSingleAsyncCompletionRoutine ( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context ) /*++ Routine Description: This is the completion routine for all asynchronous reads started via UdfSingleAsynch. Arguments: DeviceObject - Pointer to the file system device object. Irp - Pointer to the Irp for this request. (This Irp will no longer be accessible after this routine returns.) Context - The context parameter which was specified in the call to UdfSingleAsynch. Return Value: Currently always returns STATUS_SUCCESS. --*/ { // // Update the information field with the correct value for bytes read. // Irp->IoStatus.Information = 0; if (NT_SUCCESS( Irp->IoStatus.Status )) { Irp->IoStatus.Information = ((PUDF_IO_CONTEXT) Context)->RequestedByteCount; } // // Mark the Irp pending // IoMarkIrpPending( Irp ); // // Now release the resource // ExReleaseResourceForThreadLite( ((PUDF_IO_CONTEXT) Context)->Resource, ((PUDF_IO_CONTEXT) Context)->ResourceThreadId ); // // and finally, free the context record. // UdfFreeIoContext( (PUDF_IO_CONTEXT) Context ); return STATUS_SUCCESS; UNREFERENCED_PARAMETER( DeviceObject ); }