?? driver.cpp
字號:
IoDisconnectInterrupt( pDevExt->pIntObj );
}
// This will yield the symbolic link name
UNICODE_STRING pLinkName =
pDevExt->ustrSymLinkName;
// ... which can now be deleted
IoDeleteSymbolicLink(&pLinkName);
// Delete the device
IoDeleteDevice( pDO );
pDevExt->state = Removed;
return PassDownPnP( pDO, pIrp );
}
//++
// Function: DriverUnload
//
// Description:
// Stops & Deletes devices controlled by this driver.
// Stops interrupt processing (if any)
// Releases kernel resources consumed by driver
//
// Arguments:
// pDriverObject - Passed from I/O Manager
//
// Return value:
// None
//--
VOID DriverUnload (
IN PDRIVER_OBJECT pDriverObject ) {
}
//++
// Function: DispatchCreate
//
// Description:
// Handles call from Win32 CreateFile request
// For this driver, does nothing
//
// Arguments:
// pDevObj - Passed from I/O Manager
// pIrp - Passed from I/O Manager
//
// Return value:
// NTSTATUS - success or failure code
//--
NTSTATUS DispatchCreate (
IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp ) {
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pDevObj->DeviceExtension;
NTSTATUS status = STATUS_SUCCESS;
if (pDevExt->state != Started)
status = STATUS_DEVICE_REMOVED;
pIrp->IoStatus.Status = status;
pIrp->IoStatus.Information = 0; // no bytes xfered
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
return status;
}
//++
// Function: DispatchClose
//
// Description:
// Handles call from Win32 CreateHandle request
// For this driver, does nothing
//
// Arguments:
// pDevObj - Passed from I/O Manager
// pIrp - Passed from I/O Manager
//
// Return value:
// NTSTATUS - success or failure code
//--
NTSTATUS DispatchClose (
IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp ) {
pIrp->IoStatus.Status = STATUS_SUCCESS;
pIrp->IoStatus.Information = 0;
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
return STATUS_SUCCESS;
}
//++
// Function: DispatchCancel
//
// Description:
// Handles canceled IRP
//
// Arguments:
// pDevObj - Passed from I/O Manager
// pIrp - Passed from I/O Manager
//
// Return value:
// NTSTATUS - success or failuer code
//--
VOID DispatchCancel (
IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp ) {
// Just complete the IRP
pIrp->IoStatus.Status = STATUS_CANCELLED;
pIrp->IoStatus.Information = 0; // bytes xfered
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
IoStartNextPacket( pDevObj, TRUE );
}
//++
// Function:
// DispatchReadWrite
//
// Description:
// This function dispatches ReadFile
// and WriteFile requests from Win32
//
// Arguments:
// Pointer to Device object
// Pointer to IRP for this request
//
// Return Value:
// This function returns STATUS_XXX
//--
static NTSTATUS DispatchReadWrite(
IN PDEVICE_OBJECT pDO,
IN PIRP pIrp
)
{
PIO_STACK_LOCATION pIrpStack =
IoGetCurrentIrpStackLocation( pIrp );
//
// Check for zero-length transfers
//
if( pIrpStack->Parameters.Read.Length == 0 ) {
pIrp->IoStatus.Status = STATUS_SUCCESS;
pIrp->IoStatus.Information = 0;
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
return STATUS_SUCCESS;
}
//
// Start device operation
//
IoMarkIrpPending( pIrp );
IoStartPacket( pDO, pIrp, 0, NULL );
return STATUS_PENDING;
}
BOOLEAN Isr (
IN PKINTERRUPT pIntObj,
IN PVOID pServiceContext ) {
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pServiceContext;
PDEVICE_OBJECT pDevObj = pDevExt->pDevice;
PIRP pIrp = pDevObj->CurrentIrp;
// Check HW to see if interrupt for this driver
// If not, return FALSE immediately
//return FALSE;
// its our interrupt, deal with it
// Dismiss the interrupt in HW now
// Were we expecting an interrupt?
if (!pDevExt->bInterruptExpected)
return TRUE; // nope
pDevExt->bInterruptExpected = FALSE;
// Do the rest of the work down
// at DISPATCH_LEVEL IRQL
IoRequestDpc(
pDevObj,
pIrp,
(PVOID)pDevExt );
return TRUE;
}
//++
// Function:
// StartIo
//
// Description:
// This function is responsible initiating the
// actual data transfer. The ultimate result
// should be an interrupt from the device.
//
// Arguments:
// Pointer to the Device object
// Pointer to the IRP for this request
//
// Return Value:
// (None)
//--
VOID StartIo( IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp ) {
PIO_STACK_LOCATION pStack =
IoGetCurrentIrpStackLocation( pIrp );
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pDevObj->DeviceExtension;
// The IRP holds the MDL structure, already set up by
// the I/O Manager because DO_DIRECT_IO flag is set
PMDL pMdl = pIrp->MdlAddress;
ULONG mapRegsNeeded;
NTSTATUS status;
pDevExt->bWriting = FALSE; // assume read operation
switch ( pStack->MajorFunction ) {
case IRP_MJ_WRITE:
pDevExt->bWriting = TRUE; // bad assumption
case IRP_MJ_READ:
pDevExt->bytesRequested =
MmGetMdlByteCount( pMdl );
pDevExt->transferVA = (PUCHAR)
MmGetMdlVirtualAddress( pMdl );
pDevExt->bytesRemaining =
pDevExt->transferSize =
pDevExt->bytesRequested;
mapRegsNeeded =
ADDRESS_AND_SIZE_TO_SPAN_PAGES(
pDevExt->transferVA,
pDevExt->transferSize );
if (mapRegsNeeded > pDevExt->mapRegisterCount) {
mapRegsNeeded = pDevExt->mapRegisterCount;
pDevExt->transferSize =
mapRegsNeeded * PAGE_SIZE -
MmGetMdlByteOffset( pMdl );
}
status = pDevExt->pDmaAdapter->DmaOperations->
AllocateAdapterChannel(
pDevExt->pDmaAdapter,
pDevObj,
mapRegsNeeded,
AdapterControl,
pDevExt );
if (!NT_SUCCESS( status )) {
// fail the IRP & don't continue with it
pIrp->IoStatus.Status = status;
// Show no bytes transferred
pIrp->IoStatus.Information = 0;
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
IoStartNextPacket( pDevObj, FALSE);
}
break; // nice job - AdapterControl takes it
// from here on
default:
// Shouldn't be here - ditch this strange IRP
pIrp->IoStatus.Status = STATUS_NOT_SUPPORTED;
pIrp->IoStatus.Information = 0;
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
IoStartNextPacket( pDevObj, FALSE );
}
}
//++
// Function:
// AdapterControl
//
// Description:
// This function sets up the first
// part of a split transfer and
// starts the DMA device.
//
// Arguments:
// Device object
// Current Irp
// Map register base handle
// Pointer to Device Extension
//
// Return Value:
// KeepObject
//--
IO_ALLOCATION_ACTION AdapterControl(
IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp,
IN PVOID MapRegisterBase,
IN PVOID pContext ) {
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pContext;
// Save the handle to the mapping register set
pDevExt->mapRegisterBase = MapRegisterBase;
// Flush the CPU cache(s),
// if necessary on this platform...
KeFlushIoBuffers( pIrp->MdlAddress,
!pDevExt->bWriting, // inverted
TRUE ); // yes DMA
pDevExt->pDmaAdapter->DmaOperations->
MapTransfer( pDevExt->pDmaAdapter,
pIrp->MdlAddress,
MapRegisterBase,
pDevExt->transferVA,
&pDevExt->transferSize,
pDevExt->bWriting );
// Start the device
StartTransfer( pDevExt );
return KeepObject;
}
//++
// Function:
// DpcForIsr
//
// Description:
// This function performs the low-IRQL
// post-processing of I/O requests
//
// Arguments:
// Pointer to a DPC object
// Pointer to the Device object
// Pointer to the IRP for this request
// Pointer to the Device Extension
//
// Return Value:
// (None)
//--
VOID DpcForIsr(IN PKDPC pDpc,
IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp,
IN PVOID pContext ) {
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pContext;
ULONG mapRegsNeeded;
PMDL pMdl = pIrp->MdlAddress;
// Flush the Apapter buffer to system RAM or device.
pDevExt->pDmaAdapter->DmaOperations->
FlushAdapterBuffers( pDevExt->pDmaAdapter,
pMdl,
pDevExt->mapRegisterBase,
pDevExt->transferVA,
pDevExt->transferSize,
pDevExt->bWriting );
// If the device is reporting errors, fail the IRP
if (DEVICE_FAIL( pDevExt )) {
// An error occurred, the DMA channel is now free
pDevExt->pDmaAdapter->DmaOperations->
FreeAdapterChannel( pDevExt->pDmaAdapter );
pIrp->IoStatus.Status = STATUS_DEVICE_DATA_ERROR;
pIrp->IoStatus.Information =
pDevExt->bytesRequested -
pDevExt->bytesRemaining;
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
IoStartNextPacket( pDevObj, FALSE);
}
// Device had no errors, see if another partial needed
pDevExt->bytesRemaining -= pDevExt->transferSize;
if (pDevExt->bytesRemaining > 0) {
// Another partial transfer needed
// Update the transferVA and try to finish it
pDevExt->transferVA += pDevExt->transferSize;
pDevExt->transferSize = pDevExt->bytesRemaining;
mapRegsNeeded =
ADDRESS_AND_SIZE_TO_SPAN_PAGES(
pDevExt->transferVA,
pDevExt->transferSize );
// If it still doesn't fit in one swipe,
// cut back the expectation
if (mapRegsNeeded > pDevExt->mapRegisterCount) {
mapRegsNeeded = pDevExt->mapRegisterCount;
pDevExt->transferSize =
mapRegsNeeded * PAGE_SIZE -
BYTE_OFFSET( pDevExt->transferVA );
}
// Now set up the mapping registers for another
pDevExt->pDmaAdapter->DmaOperations->
MapTransfer( pDevExt->pDmaAdapter,
pMdl,
pDevExt->mapRegisterBase,
pDevExt->transferVA,
&pDevExt->transferSize,
pDevExt->bWriting );
// And start the device
StartTransfer( pDevExt );
} else {
// Entire transfer has now completed -
// Free the DMA channel for another device
pDevExt->pDmaAdapter->DmaOperations->
FreeAdapterChannel( pDevExt->pDmaAdapter );
// And complete the IRP in glory
pIrp->IoStatus.Status = STATUS_SUCCESS;
pIrp->IoStatus.Information =
pDevExt->bytesRequested;
// Choose a priority boost appropriate for device
IoCompleteRequest( pIrp, IO_DISK_INCREMENT );
IoStartNextPacket( pDevObj, FALSE );
}
}
VOID StartTransfer( IN PDEVICE_EXTENSION pDevExt ) {
// This place holder routine would hold the code
// necessary to manipulate the slave DMA device
// so that it starts the transfer of data.
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -