Как писать драйвера

extern NTSTATUS FilterOpen(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

extern NTSTATUS FilterClose(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

extern NTSTATUS FilterRead(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

extern NTSTATUS FilterWrite(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

extern NTSTATUS FilterIoControl(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

Внесите их в файл passthru.h.

Теперь внесем изменения в нашу функцию DriverEntry.

Объявим массив:

PDRIVER_DISPATCH MajorFunctions [IRP_MJ_MAXIMUM_FUNCTION + 1];

Этот массив будет содержать все указатели функций для регистрации.

И добавим еще несколько переменных:

NDIS_STRING ntDeviceName; //имя для вызова виртуального device-а для NT

NDIS_STRING win32DeviceName; // тоже для win32

PDEVICE_OBJECT deviceObject; // и объект.

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

// Name of control deviceObject.

// DeviceName that names the device object.

NdisInitUnicodeString(&ntDeviceName, L"\\Device\\passthru" );

// SymbolicName that is the Win32-visible name of the device

NdisInitUnicodeString(&win32DeviceName, L"\\DosDevices\\passthru" );

//Создаем строку имени

NdisZeroMemory(MajorFunctions, sizeof(MajorFunctions));

// Регистрируем ее

//Связываем имена функций с массивом

MajorFunctions[IRP_MJ_CREATE] = FilterOpen;

MajorFunctions[IRP_MJ_CLOSE] = FilterClose;

MajorFunctions[IRP_MJ_READ] = FilterRead;

MajorFunctions[IRP_MJ_WRITE] = FilterWrite;

MajorFunctions[IRP_MJ_DEVICE_CONTROL] = FilterIoControl;

//Регистрируем их

Status = NdisMRegisterDevice(WrapperHandle, &ntDeviceName, &win32DeviceName, MajorFunctions, &deviceObject, &GlobalData.NdisDeviceHandle );

// проверяем статус

if (Status != NDIS_STATUS_SUCCESS ) {

if (GlobalData.ProtHandle) NdisDeregisterProtocol(&Status, GlobalData.ProtHandle);

if (GlobalData.NdisDeviceHandle) NdisMDeregisterDevice(GlobalData.NdisDeviceHandle);

if (WrapperHandle) NdisTerminateWrapper(WrapperHandle, NULL);

return (Status);

}

// set access method into deviceObject ( received from NdisMRegisterDevice() )

// объявление буферизации для связывающих операций

deviceObject->Flags |= DO_BUFFERED_IO;

// все.

Тела функций, типа Filter, объявите сразу после тела функции DriverEntry и оставьте пока пустыми, добавив, только возвращение значения

return NDIS_STATUS_SUCCESS;

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

Функции этой группы занимаются обработкой потока данных и событий, происходящих в верхнем уровне драйвера, и вызываемых обращением к NDIS TCP/IP стека.

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

MPInitialize– инициализация группы.

MPSend

MPSendPackets

MPTransferData

MPReturnPacket

Функции отвечающие за пересылку пакетов данных.

MPQueryInformation

MPSetInformation

MPQueryPNPCapbilities

MPIsSendOID

MPProcessSetPowerOid

Функции работы с питанием состоянием системы и системой PlagNPlay. Сказать особенно нечего. Стандартное отслеживание внутренних событий системы прописанное Microsoft.

MPHalt– отработка выгрузки и де регистрации драйвера при аварийном.

MPReset– как написано у Microsoft – мы не должны ничего делать :)

Работа с системой – необходимость отрабатывать события важные для сервиса корректно.

MPSetMiniportSecondary

MPPromoteSecondary

MPBundleSearchAndSetSecondary

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

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

MPSend

Основная функция вызываемая всегда, при прохождении данных. По правилам работы с данными в NDIS необходимо написать (что в примере и сделано) re-wrap пакету.

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

PADAPT pAdapt = (PADAPT)MiniportAdapterContext;

Контекст адаптера приходящий в качестве параметра. Присвоим его своему типизированному указателю.

NDIS_STATUS Status;

Возвращаемый статус.

PNDIS_PACKET MyPacket;

Наш пакет – пока только указатель.

PRSVD Rsvd;

Резервный указатель.

PVOID MediaSpecificInfo = NULL;

Тип адаптера с которым будем работать.

ULONG MediaSpecificInfoSize = 0;

Размер типа адаптера.

ASSERT (pAdapt->pSecondaryAdapt);

pAdapt = pAdapt->pSecondaryAdapt;

Проверка наличия второго сетевого адаптера. Вверху я говорил, что его наличие необходимо предусматривать.

if (IsIMDeviceStateOn (pAdapt) == FALSE) {

return NDIS_STATUS_FAILURE;

}

Проверка наличия и состояния.

NdisAllocatePacket(&Status, &MyPacket, pAdapt->SendPacketPoolHandle);

Выделение места под размер полученного пакета (Pool) данных.

if (Status == NDIS_STATUS_SUCCESS) {

PNDIS_PACKET_EXTENSION Old, New;

Rsvd = (PRSVD)(MyPacket->ProtocolReserved);

Rsvd->OriginalPkt = Packet;

MyPacket->Private.Flags = Flags;

MyPacket->Private.Head = Packet->Private.Head;

MyPacket->Private.Tail = Packet->Private.Tail;

Собственно копирование всей служебной информации

NdisSetPacketFlags(MyPacket, NDIS_FLAGS_DONT_LOOPBACK);

Установка ее в наш внутренний буфер.

NdisMoveMemory(NDIS_OOB_DATA_FROM_PACKET(MyPacket), NDIS_OOB_DATA_FROM_PACKET(Packet), sizeof(NDIS_PACKET_OOB_DATA));

Перенос данных в сам пакет.

NdisIMCopySendPerPacketInfo(MyPacket, Packet);

Копирование служебных данных по пересылке пакета.

Копирование данных о типе адаптера, куда пересылать данные.

NDIS_GET_PACKET_MEDIA_SPECIFIC_INFO(Packet, &MediaSpecificInfo, &MediaSpecificInfoSize);

if (MediaSpecificInfo || MediaSpecificInfoSize) {

NDIS_SET_PACKET_MEDIA_SPECIFIC_INFO(MyPacket, MediaSpecificInfo, MediaSpecificInfoSize);

}

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

NdisSend(&Status, pAdapt->BindingHandle, MyPacket);

if (Status != NDIS_STATUS_PENDING) {

NdisIMCopySendCompletePerPacketInfo (Packet, MyPacket);

NdisFreePacket(MyPacket);

}

Если нет задержки на отсылке освободить пакет.

} else {

Это говорит об отсутствии пакета в системе – ничего не надо делать.

}

return(Status);

Возврат значения SUCCESS или код ошибки.

Стоит остановиться еще на одном моменте. Когда система ответила, что посылка данных закончена кодом задержки пакета – NDIS_STATUS_PENDING.

В этом случае мы пакет не освободим, блокировав таким образом всю систему NDIS по пересылке данных. Такое случается при посылке по медленной сети большого числа пакетов.