驅動程式 - Windows Driver Model (WDM) - 使用範例 - Pascal (DDDK) - Handle StartIo IRP



參考資訊:
https://wasm.in/
http://four-f.narod.ru/
https://github.com/steward-fu/ddk

main.pas

unit main;
 
interface
    uses
        DDDK;
         
    const
        DEV_NAME = '\Device\MyDriver';
        SYM_NAME = '\DosDevices\MyDriver';
 
    function _DriverEntry(pMyDriver : PDRIVER_OBJECT; pMyRegistry : PUNICODE_STRING) : NTSTATUS; stdcall;
 
implementation
var
    pNextDevice : PDEVICE_OBJECT;
    szBuf : array[0..255] of char;
     
procedure Unload(pMyDriver : PDRIVER_OBJECT); stdcall;
begin
end;
 
function StartIo(pMyDevice : PDeviceObject; pIrp : PIrp) : NTSTATUS; stdcall;
var
    dwLen : ULONG;
    pStack : PIO_STACK_LOCATION;
        
begin
    dwLen := 0;
    pStack := IoGetCurrentIrpStackLocation(pIrp);
    if pStack^.MajorFunction = IRP_MJ_WRITE then
    begin
        DbgPrint('StartIo, IRP_MJ_WRITE', []);
        dwLen := pStack.Parameters.Write.Length;
        memcpy(@szBuf[0], pIrp^.AssociatedIrp.SystemBuffer, dwLen);
        DbgPrint('Buffer: %s, Length: %d', [szBuf, dwLen]);
    end
    else if pStack^.MajorFunction = IRP_MJ_READ then
    begin
        DbgPrint('StartIo, IRP_MJ_READ', []);
        dwLen := strlen(@szBuf[0]);
        memcpy(pIrp^.AssociatedIrp.SystemBuffer, @szBuf[0], dwLen);
    end;
        
    Result := STATUS_SUCCESS;
    pIrp^.IoStatus.Information := dwLen;
    pIrp^.IoStatus.Status := Result;
    IoCompleteRequest(pIrp, IO_NO_INCREMENT);
    IoStartNextPacket(pMyDevice, FALSE);
end;
 
function IrpFile(pMyDevice : PDEVICE_OBJECT; pIrp : PIRP) : NTSTATUS; stdcall;
var
    pStack : PIO_STACK_LOCATION;
     
begin
    pStack := IoGetCurrentIrpStackLocation(pIrp);
    case pStack^.MajorFunction of
    IRP_MJ_CREATE:
        DbgPrint('IRP_MJ_CREATE', []);
    IRP_MJ_READ:
        begin
            DbgPrint('IrpFile, IRP_MJ_READ', []);
            IoMarkIrpPending(pIrp);
            IoStartPacket(pMyDevice, pIrp, nil, nil);
            Result := STATUS_PENDING;
            exit;
        end;
    IRP_MJ_WRITE:
        begin
            DbgPrint('IrpFile, IRP_MJ_WRITE', []);
            IoMarkIrpPending(pIrp);
            IoStartPacket(pMyDevice, pIrp, nil, nil);
            Result := STATUS_PENDING;
            exit;
        end;
    IRP_MJ_CLOSE:
        DbgPrint('IRP_MJ_CLOSE', []);
    end;
     
    Result := STATUS_SUCCESS;
    pIrp^.IoStatus.Status := Result;
    pIrp^.IoStatus.Information := 0;
    IoCompleteRequest(pIrp, IO_NO_INCREMENT);
end;
 
function IrpPnp(pMyDevice : PDEVICE_OBJECT; pIrp : PIRP) : NTSTATUS; stdcall;
var
    pStack : PIO_STACK_LOCATION;
    suSymName : UNICODE_STRING;
           
begin
    pStack := IoGetCurrentIrpStackLocation(pIrp);
    if pStack^.MinorFunction = IRP_MN_REMOVE_DEVICE then
        begin
            RtlInitUnicodeString(@suSymName, SYM_NAME);
            IoDetachDevice(pNextDevice);
            IoDeleteDevice(pMyDevice);
            IoDeleteSymbolicLink(@suSymName);
            IoCompleteRequest(pIrp, IO_NO_INCREMENT);
            Result := STATUS_SUCCESS;
        end
    else
        begin
            IoSkipCurrentIrpStackLocation(pIrp);
            Result := IoCallDriver(pNextDevice, pIrp);
        end;
end;
 
function AddDevice(pMyDriver : PDRIVER_OBJECT; pPhyDevice : PDEVICE_OBJECT) : NTSTATUS; stdcall;
var
    suDevName : UNICODE_STRING;
    suSymName : UNICODE_STRING;
    pMyDevice : PDEVICE_OBJECT;
     
begin
    RtlInitUnicodeString(@suDevName, DEV_NAME);
    RtlInitUnicodeString(@suSymName, SYM_NAME);
    IoCreateDevice(pMyDriver, 0, @suDevName, FILE_DEVICE_UNKNOWN, 0, FALSE, pMyDevice);
    pNextDevice := IoAttachDeviceToDeviceStack(pMyDevice, pPhyDevice);
    pMyDevice^.Flags := pMyDevice^.Flags or DO_BUFFERED_IO;
    pMyDevice^.Flags := pMyDevice^.Flags and not DO_DEVICE_INITIALIZING;
    Result := IoCreateSymbolicLink(@suSymName, @suDevName);
end;
 
function _DriverEntry(pMyDriver : PDRIVER_OBJECT; pMyRegistry : PUNICODE_STRING) : NTSTATUS; stdcall;
begin
    pMyDriver^.MajorFunction[IRP_MJ_PNP]    := @IrpPnp;
    pMyDriver^.MajorFunction[IRP_MJ_CREATE] := @IrpFile;
    pMyDriver^.MajorFunction[IRP_MJ_READ]   := @IrpFile;
    pMyDriver^.MajorFunction[IRP_MJ_WRITE]  := @IrpFile;
    pMyDriver^.MajorFunction[IRP_MJ_CLOSE]  := @IrpFile;
    pMyDriver^.DriverStartIo := @StartIo;
    pMyDriver^.DriverExtension^.AddDevice := @AddDevice;
    pMyDriver^.DriverUnload := @Unload;
    Result := STATUS_SUCCESS;
end;
end.

app.pas

program main;
 
{$APPTYPE CONSOLE}
 
uses
    Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms, Dialogs;
 
var
    hFile : DWORD;
    dwRet : DWORD;
    dwLen : DWORD;
    szBuffer : array[0..255] of char;
 
begin
    hFile := CreateFile('\\.\MyDriver', GENERIC_READ or GENERIC_WRITE, FILE_SHARE_READ, nil, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);

    StrCopy(szBuffer, 'I am error');
    dwLen := strlen(szBuffer);

    WriteLn(Output, Format('WR: %s', [szBuffer]));
    WriteLn(Output, Format('Length: %d', [dwLen]));
    WriteFile(hFile, szBuffer, dwLen, dwRet, nil);

    ReadFile(hFile, szBuffer, dwLen, dwRet, nil);
    WriteLn(Output, Format('RD: %s', [szBuffer]));
    WriteLn(Output, Format('Length: %d', [dwRet]));

    CloseHandle(hFile);
end.

完成