iorw.c

windows 2000/XP WDM设备驱动程序开发 附书光盘 武安河著

// iorw.c
//
// Generated by C DriverWizard 3.1.0 (Build 1722)
// Requires DDK Only
// File created on 10/23/2004
//

#include "pch.h"
#include "..\intrface.h"
#ifdef CHARSAMPLE_WMI_TRACE
#include "iorw.tmh"
#endif

///////////////////////////////////////////////////////////////////////////////////////////////////
//  CharSampleDeviceIoControlDispatch
//      Handled incoming IOCTL requests
//
//  Arguments:
//      IN  DeviceObject
//              Device object for our device
//
//      IN  Irp
//              The IOCTL IRP to handle
//
//  Return Value:
//      NT status code
//
NTSTATUS CharSampleDeviceIoControlDispatch(
    IN  PDEVICE_OBJECT  DeviceObject,
    IN  PIRP            Irp
    )
{
    PIO_STACK_LOCATION              irpStack;
    NTSTATUS                        status;
    PCHARSAMPLE_DEVICE_EXTENSION    deviceExtension;
    PVOID                           inputBuffer;
    ULONG                           inputLength;
    PVOID                           outputBuffer;
    ULONG                           outputLength;

	CHAR n,c[]="零一二三四五六七八九";

    CharSampleDebugPrint(DBG_IO, DBG_TRACE, __FUNCTION__"++. IRP %p", Irp);

    deviceExtension = (PCHARSAMPLE_DEVICE_EXTENSION)DeviceObject->DeviceExtension;

    // Get our IRP stack location
    irpStack = IoGetCurrentIrpStackLocation(Irp);

    // Get the buffer lengths
    inputLength = irpStack->Parameters.DeviceIoControl.InputBufferLength;
    outputLength = irpStack->Parameters.DeviceIoControl.OutputBufferLength;

    switch (irpStack->Parameters.DeviceIoControl.IoControlCode) 
    {

    case IOCTL_800:
        CharSampleDebugPrint(DBG_IO, DBG_INFO, __FUNCTION__": IOCTL_800");

        status = CharSampleCheckIoLock(&deviceExtension->IoLock, Irp);
        if (!NT_SUCCESS(status) || (status == STATUS_PENDING))
        {
            CharSampleDebugPrint(DBG_IO, DBG_WARN, __FUNCTION__"--. IRP %p STATUS %x", Irp, status);

            return status;
        }

		// buffered ioctl
        inputBuffer = Irp->AssociatedIrp.SystemBuffer;
        outputBuffer = Irp->AssociatedIrp.SystemBuffer;

		n=*(CHAR *)inputBuffer;//输入n=应用程序传给驱动程序的数字ASCII码
		if ((n >= '0') && (n <= '9'))
		{		//若为数字，则处理
			n -= '0'; //n=数字(0-9)
			strncpy((PCHAR)outputBuffer,&c[n*2],2);
			//输出：从中文数组c[]中取出对应的数字的中文码，拷贝到输出缓冲区
			Irp->IoStatus.Information = 2; //驱动程序传递给应用程序的字节数=2
			status = STATUS_SUCCESS;
		}
		else	//否则返回无效参数
			status = STATUS_INVALID_PARAMETER;

//		status = STATUS_NOT_IMPLEMENTED;
//		Irp->IoStatus.Information = 0;
        Irp->IoStatus.Status = status;
        IoCompleteRequest (Irp, IO_NO_INCREMENT);

        CharSampleDecrementIoCount(&deviceExtension->IoLock);
        break;

    default:
        status = STATUS_INVALID_DEVICE_REQUEST;
        Irp->IoStatus.Status = status;
        IoCompleteRequest (Irp, IO_NO_INCREMENT);
        break;
    }

    CharSampleDebugPrint(DBG_IO, DBG_TRACE, __FUNCTION__"--. IRP %p STATUS %x", Irp, status);

    return status;
}