Kernel Mode Driver Framework >> Pascal (PNP) >> IOCTL
METHOD_BUFFERED
參考資訊:
1. Source Code
METHOD_BUFFERED的處理手法就是I/O Manager會在Kernel配置一塊跟User Buffer一樣大小的記憶體,Driver只能對該配置的記憶體做讀寫的操作,而讀寫的操作都會同步回User Buffer,也因為結果都需要整塊複製回User Buffer,因此,METHOD_BUFFERED的方式會消耗比較大的資源,速度也比較慢。
記憶體指標:
Buffer | Length | |
---|---|---|
Input | Step 1. WdfRequestRetrieveInputMemory() Step 2. WdfMemoryGetBuffer() |
傳入的第四個參數 |
Output | Step 1. WdfRequestRetrieveOutputMemory() Step 2. WdfMemoryGetBuffer() |
傳入的第三個參數 |
main.pas
unit main; interface uses DDDK; const DEV_NAME = '\Device\MyDriver'; SYM_NAME = '\DosDevices\MyDriver'; IOCTL_SET = (FILE_DEVICE_UNKNOWN shl 16) or ($800 shl 2) or (FILE_ANY_ACCESS shl 14) or (METHOD_BUFFERED); IOCTL_GET = (FILE_DEVICE_UNKNOWN shl 16) or ($801 shl 2) or (FILE_ANY_ACCESS shl 14) or (METHOD_BUFFERED); function __DriverEntry(pOurDriver:PDRIVER_OBJECT; pOurRegistry:PUNICODE_STRING):NTSTATUS; stdcall; implementation var szBuffer: array[0..255] of char; procedure IrpFileCreate(Device:WDFDEVICE; Request:WDFREQUEST; FileObject:WDFFILEOBJECT); stdcall; begin DbgPrint('IrpFileCreate', []); WdfRequestComplete(Request, STATUS_SUCCESS); end; procedure IrpFileClose(FileObject:WDFFILEOBJECT); stdcall; begin DbgPrint('IrpFileClose', []); end; procedure IrpIOCTL(Queue:WDFQUEUE; Request:WDFREQUEST; OutputBufferLength:ULONG; InputBufferLength:ULONG; IoControlCode:ULONG); stdcall; var len: ULONG; memory: WDFMEMORY; begin if IoControlCode = IOCTL_SET then begin DbgPrint('IOCTL_SET', []); WdfRequestRetrieveInputMemory(Request, @memory); WdfMemoryCopyToBuffer(memory, 0, @szBuffer, InputBufferLength); DbgPrint('Buffer: %s, Length:%d', [@szBuffer, InputBufferLength]); WdfRequestSetInformation(Request, InputBufferLength); end else if IoControlCode = IOCTL_GET then begin DbgPrint('IOCTL_GET', []); WdfRequestRetrieveOutputMemory(Request, @memory); WdfMemoryCopyFromBuffer(memory, 0, @szBuffer, OutputBufferLength); len:= strlen(@szBuffer) + 1; WdfRequestSetInformation(Request, len); end; WdfRequestComplete(Request, STATUS_SUCCESS); end; function AddDevice(pOurDriver:WDFDRIVER; pDeviceInit:PWDFDEVICE_INIT):NTSTATUS; stdcall; var device: WDFDEVICE; suDevName: UNICODE_STRING; szSymName: UNICODE_STRING; file_cfg: WDF_FILEOBJECT_CONFIG; ioqueue_cfg: WDF_IO_QUEUE_CONFIG; begin WdfDeviceInitSetIoType(pDeviceInit, WdfDeviceIoBuffered); WDF_FILEOBJECT_CONFIG_INIT(@file_cfg, @IrpFileCreate, @IrpFileClose, Nil); WdfDeviceInitSetFileObjectConfig(pDeviceInit, @file_cfg, WDF_NO_OBJECT_ATTRIBUTES); RtlInitUnicodeString(@suDevName, DEV_NAME); RtlInitUnicodeString(@szSymName, SYM_NAME); WdfDeviceInitAssignName(pDeviceInit, @suDevName); WdfDeviceCreate(@pDeviceInit, WDF_NO_OBJECT_ATTRIBUTES, @device); WdfDeviceCreateSymbolicLink(device, @szSymName); WDF_IO_QUEUE_CONFIG_INIT_DEFAULT_QUEUE(@ioqueue_cfg, WdfIoQueueDispatchSequential); ioqueue_cfg.EvtIoDeviceControl:= @IrpIOCTL; Result:= WdfIoQueueCreate(device, @ioqueue_cfg, WDF_NO_OBJECT_ATTRIBUTES, WDF_NO_HANDLE); end; function __DriverEntry(pOurDriver:PDRIVER_OBJECT; pOurRegistry:PUNICODE_STRING):NTSTATUS; stdcall; var config: WDF_DRIVER_CONFIG; begin WDF_DRIVER_CONFIG_INIT(@config, AddDevice); WdfDriverCreate(pOurDriver, pOurRegistry, WDF_NO_OBJECT_ATTRIBUTES, @config, WDF_NO_HANDLE); Result:= STATUS_SUCCESS; end; end.
IrpIOCTL()收到IOCTL_SET時,Driver複製User Buffer的內容到szBuffer,而收到IOCTL_GET時,將szBuffer內容又複製回User Buffer,完成暫存的功能。
app.pas
program main; {$APPTYPE CONSOLE} uses Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms, DIALOGS; const METHOD_BUFFERED = 0; FILE_ANY_ACCESS = 0; FILE_DEVICE_UNKNOWN = $22; IOCTL_SET = (FILE_DEVICE_UNKNOWN shl 16) or (FILE_ANY_ACCESS shl 14) or ($800 shl 2) or (METHOD_BUFFERED); IOCTL_GET = (FILE_DEVICE_UNKNOWN shl 16) or (FILE_ANY_ACCESS shl 14) or ($801 shl 2) or (METHOD_BUFFERED); var fd: DWORD; ret: DWORD; len: DWORD; buf: array[0..255] of char; begin fd:= CreateFile('\\.\MyDriver', GENERIC_READ or GENERIC_WRITE, FILE_SHARE_READ, Nil, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0); if (fd <> INVALID_HANDLE_VALUE) then begin StrCopy(buf, 'I am error'); len:= strlen(buf)+1; DeviceIoControl(fd, IOCTL_SET, @buf, len, Nil, 0, ret, Nil); WriteLn(Output, Format('SET: %s, %d', [buf, len])); FillChar(buf, sizeof(buf), #0); DeviceIoControl(fd, IOCTL_GET, Nil, 0, @buf, len, ret, Nil); WriteLn(Output, Format('GET: %s, %d', [buf, ret])); CloseHandle(fd); end else begin WriteLn(Output, 'failed to open mydriver'); end; end.
結果