Как писать драйвера - [7]

 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;
 IoCompleteRequest(Irp, IO_NO_INCREMENT);
 return Status;
}

Описания Input/Output буферов привожу для того, чтобы при необходимости получения и еще каких либо сопутствующих параметров, было ясно, где их получать, скажем, в драйвере нашего примера, команда 1 может нести в качестве параметра новый номер порта для перехвата.

Давайте теперь опишем логику управления драйвером перехватчиком.

Для перехвата определяются в начале параметры перехвата, адрес, порт и т.д.

Далее вносится тип рабочего состояния – перехват – прозрачный режим.

Вносится список возможных портов к перехвату (по необходимости).

В процессе работы системы, пока управляющая аппликация не запущена, то драйверу необходимо указать стартовые параметры.

Например, тип режима – прозрачный, в этом случае не перехватывается ничего.

Когда стартует управляющая программа – то она открывает драйвер CreateFile() и запускает, если это необходимо, другие стартовые условия, например перевод в режим перехвата и номер порта для этого.

Затем по желанию клиента из программы выставляются любые нужные условия от отключения перехвата, до перехвата всех номеров портов и всех адресов.

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

На сегодня пока все.