Как писать драйвера - страница 8

Шрифт
Интервал

стр.

Определим Default значение для порта равное 80 (стандартный порт http протокола) и проведем его инициализацию в функции Open и де инициализацию в Close. Так мы сможем контролировать старт активной фазы работы драйвера и ее окончание.

Функции Write & Read можно использовать для передачи любого количества информации в драйвер и обратно.

Для сложных случаев и частой передачи необходимо использовать именно эти функции, из-за того, что переключение контекста драйвера не происходит. При использовании для этих целей DeviceIoControl приведет к постоянному переключению контекста драйвера и замедлит работу системы.

В нашем примере передавать в драйвер и назад нечего поэтому напишем Write/Read функции в виде готовых болванок, но так как наша цель построить работающий пример – то передадим абстрактную последовательность вверх в аппликацию.

Для этого создайте в этом же месте глобальную переменную вот такого вида:

>unsigned char TestStr[10] = "abcdefghi";

>NTSTATUS FilterRead(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp) {

> NDIS_STATUS Status = NDIS_STATUS_SUCCESS;

> ULONG BytesReturned = 0;

> ////////////////////////////////////////////////////////////////////////

> // Get input parameters

> PIO_STACK_LOCATION IrpStack = IoGetCurrentIrpStackLocation( Irp ); // Содержит стек Irp

> ULONG BytesRequested = IrpStack->Parameters.Read.Length; // Длина принятых данных –равна параметру максимально запрошенной длины в функции ReadFile

> if (BytesRequested <10) // Если запрошено меньше нашей тестовой длины – вернуть ошибку

> {

>  BytesReturned = 0;

>  Status = STATUS_INVALID_BUFFER_SIZE;

> } else {

>  // Если все в порядке – копировать буфер в выходной буфер стека.

>  NdisMoveMemory(Irp->AssociatedIrp.SystemBuffer, TestStr, 10);

>  BytesReturned = 10;

>  Status = NDIS_STATUS_SUCCESS;

> }


> // Отправить

> Irp->IoStatus.Status = Status;

> Irp->IoStatus.Information = BytesReturned;

> IoCompleteRequest(Irp, IO_NO_INCREMENT);

> return Status;

>}

>///////////////////////////////////////////////////////////////////////////////////////////////

>NTSTATUS FilterWrite(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp) {

> NDIS_STATUS Status = NDIS_STATUS_SUCCESS;

> ULONG BytesReturned = 0;

> // Здесь все работает аналогично.

> Irp->IoStatus.Status = Status;

> Irp->IoStatus.Information = BytesReturned;

> IoCompleteRequest(Irp, IO_NO_INCREMENT);

> return Status;

>}

Функции симметричны.

Теперь мы подошли к функции управления.

Она сама работает точно также как и все функции перечисленные выше – с одной разницей: В ней принимаемым параметром является код операции, который надо установить в драйвер.

Можно конечно придумать свой формат данных – передаваемых в WriteFile, который расшифровывать внутри драйвера и так решать, что делать или не делать, однако стоит использовать уже готовый механизм, предоставленный Microsoft-ом.

Описание кода комманды выглядит так:

>#define IOCTL_SET_COMMAND1 // наш код управления

>CTL_CODE ( FILE_DEVICE_UNKNOWN, // Тип кода

>0xF07, // Цифровое значение

>METHOD_BUFFERED, // Метод операции

>FILE_ANY_ACCESS ) // Права доступа.

Итак мы описали код управления, вызов которого будет выглядеть в программе так:

>res = DeviceIoControl(

> hFile, // Handle драйвера из функции CreateFile

> (DWORD) IOCTL_SET_COMMAND1, // Комманда

> (LPVOID)0, (DWORD)0, (LPVOID)NULL, (DWORD)0, (LPDWORD)&bytesReturned, NULL

>);

Вызов такой функции приведет к обращению драйвером к функции FilterIoControl!

>NTSTATUS FilterIoControl(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp) {

> NDIS_STATUS Status = NDIS_STATUS_SUCCESS;

> ULONG BytesReturned = 0;

> PIO_STACK_LOCATION IrpStack = IoGetCurrentIrpStackLocation(Irp);

> ULONG IoControlCode = IrpStack->Parameters.DeviceIoControl.IoControlCode;

> PVOID InfoBuffer = Irp->AssociatedIrp.SystemBuffer;

> ULONG InputBufferLen = IrpStack->Parameters.DeviceIoControl.InputBufferLength;

> ULONG OutputBufferLen = IrpStack->Parameters.DeviceIoControl.OutputBufferLength;

> switch(IoControlCode) {

> case IOCTL_SET_COMMAND1:

>  // Здесь мы можем сменить наш номер порта с 80 на, к примеру, 25.

>  break;

> }

> Irp->IoStatus.Status = Status;

> Irp->IoStatus.Information = BytesReturned;


стр.

Похожие книги