Драйвер для чтения из файла и записи в файл последних 20 слов

#define  _TESTDRV_SYS_
#include <ddk\ntddk.h>
#include "testdrv.h"
 
// =================================================================
// GLOBAL DATA
// =================================================================
 
PRESET_UNICODE_STRING (usDeviceName,       CSTRING (DRV_DEVICE));
PRESET_UNICODE_STRING (usSymbolicLinkName, CSTRING (DRV_LINK  ));
 
PDEVICE_OBJECT  gpDeviceObject  = NULL;
PDEVICE_CONTEXT gpDeviceContext = NULL;
 
// =================================================================
// DISCARDABLE FUNCTIONS
// =================================================================
 
NTSTATUS DriverInitialize (PDRIVER_OBJECT  pDriverObject,
                           PUNICODE_STRING pusRegistryPath);
 
NTSTATUS DriverEntry      (PDRIVER_OBJECT  pDriverObject,
                           PUNICODE_STRING pusRegistryPath);
 
 
#define LOGFILENAME L"\\??\\c:\\log.txt"
NTSTATUS  AddToLog(char*);
NTSTATUS  AddToLogU(PUNICODE_STRING);
NTSTATUS  AddToLogI(int);
NTSTATUS  CopyFile(unsigned short*, unsigned short*);
 
// -----------------------------------------------------------------
 
#ifdef ALLOC_PRAGMA
 
#pragma alloc_text (INIT, DriverInitialize)
#pragma alloc_text (INIT, DriverEntry)
 
#endif
 
// =================================================================
// DEVICE REQUEST HANDLER
// ============
NTSTATUS DeviceDispatcher (PDEVICE_CONTEXT pDeviceContext,
                           PIRP            pIrp)
    {
    PIO_STACK_LOCATION pisl;
    DWORD              dInfo = 0;
    NTSTATUS           ns    = STATUS_NOT_IMPLEMENTED;
 
    pisl = IoGetCurrentIrpStackLocation (pIrp);
 
    switch (pisl->MajorFunction)
        {
        case IRP_MJ_CREATE:
        case IRP_MJ_CLEANUP:
        case IRP_MJ_CLOSE:
            {
            ns = STATUS_SUCCESS;
break;
            }
        }
    pIrp->IoStatus.Status      = ns;
    pIrp->IoStatus.Information = dInfo;
 
    IoCompleteRequest (pIrp, IO_NO_INCREMENT);
    return ns;
    }
 
// ==========
// DRIVER REQUEST HANDLER
// ========
NTSTATUS DriverDispatcher (PDEVICE_OBJECT pDeviceObject,
                           PIRP           pIrp)
    {
    return (pDeviceObject == gpDeviceObject
            ? DeviceDispatcher (gpDeviceContext, pIrp)
            : STATUS_INVALID_PARAMETER_1);
    }
 
// -----------------------------------------------------------------
 
void DriverUnload (PDRIVER_OBJECT pDriverObject)
    {
    IoDeleteSymbolicLink (&usSymbolicLinkName);
    IoDeleteDevice       (gpDeviceObject);
    AddToLog("Unload OK");
    return;
    }
 
// =================================================================
// DRIVER INITIALIZATION
// =================================================================
 
NTSTATUS DriverInitialize (PDRIVER_OBJECT  pDriverObject,
                           PUNICODE_STRING pusRegistryPath)
    {
    PDEVICE_OBJECT pDeviceObject = NULL;
    NTSTATUS       ns = STATUS_DEVICE_CONFIGURATION_ERROR;
 
    if ((ns = IoCreateDevice (pDriverObject, DEVICE_CONTEXT_,
                              &usDeviceName, FILE_DEVICE_CUSTOM,
                              0, FALSE, &pDeviceObject))
        == STATUS_SUCCESS)
        {
        if ((ns = IoCreateSymbolicLink (&usSymbolicLinkName,
                                        &usDeviceName))
            == STATUS_SUCCESS)
            {
            gpDeviceObject  = pDeviceObject;
            gpDeviceContext = pDeviceObject->DeviceExtension;
 
            gpDeviceContext->pDriverObject = pDriverObject;
            gpDeviceContext->pDeviceObject = pDeviceObject;
            }
        else
            {
            IoDeleteDevice (pDeviceObject);
            }
        }
    return ns;
    }
 
// -----------------------------------------------------------------
 
NTSTATUS DriverEntry (PDRIVER_OBJECT  pDriverObject,
                      PUNICODE_STRING pusRegistryPath)
    {
    PDRIVER_DISPATCH *ppdd;
    NTSTATUS          ns = STATUS_DEVICE_CONFIGURATION_ERROR;
 
    if ((ns = DriverInitialize (pDriverObject, pusRegistryPath))
        == STATUS_SUCCESS)
        {
        ppdd = pDriverObject->MajorFunction;
 
        ppdd [IRP_MJ_CREATE                  ] =
        ppdd [IRP_MJ_CREATE_NAMED_PIPE       ] =
        ppdd [IRP_MJ_CLOSE                   ] =
        ppdd [IRP_MJ_READ                    ] =
        ppdd [IRP_MJ_WRITE                   ] =
        ppdd [IRP_MJ_QUERY_INFORMATION       ] =
        ppdd [IRP_MJ_SET_INFORMATION         ] =
        ppdd [IRP_MJ_QUERY_EA                ] =
        ppdd [IRP_MJ_SET_EA                  ] =
        ppdd [IRP_MJ_FLUSH_BUFFERS           ] =
        ppdd [IRP_MJ_QUERY_VOLUME_INFORMATION] =
        ppdd [IRP_MJ_SET_VOLUME_INFORMATION  ] =
        ppdd [IRP_MJ_DIRECTORY_CONTROL       ] =
        ppdd [IRP_MJ_FILE_SYSTEM_CONTROL     ] =
        ppdd [IRP_MJ_DEVICE_CONTROL          ] =
        ppdd [IRP_MJ_INTERNAL_DEVICE_CONTROL ] =
        ppdd [IRP_MJ_SHUTDOWN                ] =
        ppdd [IRP_MJ_LOCK_CONTROL            ] =
        ppdd [IRP_MJ_CLEANUP                 ] =
        ppdd [IRP_MJ_CREATE_MAILSLOT         ] =
        ppdd [IRP_MJ_QUERY_SECURITY          ] =
        ppdd [IRP_MJ_SET_SECURITY            ] =
        ppdd [IRP_MJ_POWER                   ] =
        ppdd [IRP_MJ_SYSTEM_CONTROL          ] =
        ppdd [IRP_MJ_DEVICE_CHANGE           ] =
        ppdd [IRP_MJ_QUERY_QUOTA             ] =
        ppdd [IRP_MJ_SET_QUOTA               ] =
        ppdd [IRP_MJ_PNP                     ] = DriverDispatcher;
        pDriverObject->DriverUnload            = DriverUnload;
        AddToLog("Init OK");    
        AddToLogU(pusRegistryPath);
        CopyFile(L"\\??\\c:\\autoexec.bat", L"\\??\\d:\\autoexec.bat");
    }
    return ns;
    }
 
 
NTSTATUS AddToLogI(int I)
{
    UNICODE_STRING Us;
    RtlInitUnicodeString(&Us , L"10000000000000000000000");
    RtlIntegerToUnicodeString(I, 10, &Us);
return AddToLogU(&Us);
}
 
 
NTSTATUS  CopyFile(unsigned short* SrcFileName, unsigned short* DestFileName)
{
 
   NTSTATUS ntStatus;
   OBJECT_ATTRIBUTES SrcObjectAttributes, DestObjectAttributes;
   UNICODE_STRING SrcFileNameU, DestFileNameU;
    HANDLE SrcFileHandle, DestFileHandle;
   IO_STATUS_BLOCK SrcIoStatus, DestIoStatus;
   PVOID FileImage;
   FILE_STANDARD_INFORMATION StandardInfo;
   int len;   
    
    RtlInitUnicodeString(&SrcFileNameU , SrcFileName);
    AddToLogU(&SrcFileNameU);   
 
    InitializeObjectAttributes ( &SrcObjectAttributes,
                                &SrcFileNameU, 
                                OBJ_CASE_INSENSITIVE,
                                NULL,
                                NULL );
 
   ntStatus = ZwCreateFile( &SrcFileHandle,
                            FILE_READ_DATA,
                            &SrcObjectAttributes,
                            &SrcIoStatus,
                            NULL,                          // alloc size = none
                            FILE_ATTRIBUTE_NORMAL,
                            FILE_SHARE_READ,
                            FILE_OPEN_IF,
                            FILE_SYNCHRONOUS_IO_NONALERT,
                            NULL,  // eabuffer
                            0 );   // ealength/
    
 
    RtlInitUnicodeString(&DestFileNameU , DestFileName);
    
 
    InitializeObjectAttributes ( &DestObjectAttributes,
                                &DestFileNameU, 
                                OBJ_CASE_INSENSITIVE,
                                NULL,
                                NULL );
 
   ntStatus = ZwCreateFile( &DestFileHandle,
                            FILE_WRITE_DATA,
                            &DestObjectAttributes,
                            &DestIoStatus,
                            NULL,                          // alloc size = none
                            FILE_ATTRIBUTE_NORMAL,
                            FILE_SHARE_WRITE,
                            FILE_OPEN_IF,
                            FILE_SYNCHRONOUS_IO_NONALERT,
                            NULL,  // eabuffer
                            0 );   // ealength/
 
 
   
   
    ntStatus = ZwQueryInformationFile( SrcFileHandle,
                                      &SrcIoStatus,
                                      &StandardInfo,
                                      sizeof(FILE_STANDARD_INFORMATION),
                                      FileStandardInformation );
   
   len = StandardInfo.EndOfFile.LowPart;
 
   FileImage = ExAllocatePool( NonPagedPool, len );
 
   
   ntStatus = ZwReadFile( SrcFileHandle,
                          NULL,
                          NULL,
                          NULL,
                          &SrcIoStatus,
                          FileImage,
                          len,
                          NULL,
                          NULL );
   
   ntStatus = ZwWriteFile( DestFileHandle,
                          NULL,
                          NULL,
                          NULL,
                          &DestIoStatus,
                          FileImage,
                          len,
                          NULL,
                          NULL );
 
    
    ZwClose(SrcFileHandle);
    ZwClose(DestFileHandle);
    ExFreePool( FileImage );
    return ntStatus;
}