Чтение онлайн

ЖАНРЫ

Использование NuMega DriverStudio для написания WDM-драйверов

Тарво Александр

Шрифт:

PNPMinorFunctionName — возвращает строку с текстовым названием кода функции IOCTL. Эта функция используется при отладке, когда надо перевести числовое значение кода IOCTL в строку с его названием.

POOLTAG DefaultPoolTag('PSDX') — используется совместно с BoundsChecker для отслеживания возможных переполнений буфера и утечек памяти.

KTrace t("XDSPdrv") — глобальный объект трассировки драйвера. Этот объект используется для вывода сообщений трассировки при работе драйвера. Использование объекта трассировки аналогично использованию класса iostream в С++. Вывод отладочных сообщений производится при помощи оператора <<. Примеры использования объекта трассировки неоднократно встречаются в тексте драйвера, например:

t << "m_bBreakOnEntry loaded from registry, resulting value: [" << m_bBreakOnEntry << "]\n";

В данном примере объект трассировки используется для вывода строки "m_bBreakOnEntry loaded from registry, resulting value: [" и значения логической переменной m_bBreakOnEntry. Все сообщения трассировки можно прочитать в отладчике SoftIce.

Начнем анализ текста драйвера с класса XDSP (класс драйвера). В строке 31 при помощи макроса DECLARE_DRIVER_CLASS декларируется класс драйвера XDSP. Далее следует метод DriverEntry, который вызывается при инициализации драйвера:

NTSTATUS XDSPdrv::DriverEntry(PUNICODE_STRING RegistryPath)

//В строке RegistryPath содержится ключ реестра, в котором система хранит информацию о драйвере.

{

 //Далее выводится трассировочное сообщение, информирующее о вызове метода DriverEntry:

 t << "In DriverEntry\n";

 //После этого драйвер создает объект Params класса KRegistryKey и считывает данные из

 //реестра для этого драйвера:

 KRegistryKey Params(RegistryPath, L"Parameters");

 //Далее производится проверка на успех:

 if ( NT_SUCCESS(Params.LastError) ) {

//Текст, заключенный в макрос препроцессора DBG будет откомпилирован только в отладочной версии

//драйвера.

#if DBG

ULONG bBreakOnEntry = FALSE;

// Читается значение переменной BreakOnEntry реестра:

Params.QueryValue(L"BreakOnEntry", &bBreakOnEntry);

// Если она принимает значение true,то инициировать точку останова в отладчике.

if (bBreakOnEntry) DbgBreakPoint;

#endif

//Загрузить остальные параметры реестра.

LoadRegistryParameters(Params);

 }

 m_Unit = 0;

 //Вернуть успех

 return STATUS_SUCCESS;

}

Метод LoadRegistryParameters зaгружает из реестра все остальные параметры, необходимые для драйвера. Впрочем, в нашем драйвере таковых нет, и поэтому функция не выполняет никаких полезных действий (просто загружает значение переменной m_bBreakOnEntry).

void XDSPdrv::LoadRegistryParameters(KRegistryKey &Params) {

 m_bBreakOnEntry = FALSE;

 Params.QueryValue(L"BreakOnEntry", &m_bBreakOnEntry);

 t << "m_bBreakOnEntry loaded from registry, resulting value: [" << m_bBreakOnEntry << "]\n";

}

На этом заканчивается секция инициализации драйвера. Далее следует метод AddDevice. Он вызывается, когда система обнаруживает устройство, за которое отвечает драйвер (обычно это происходит при загрузке драйвера). В метод ситема передает указатель на физический объект устройства (Physical Device Object, PDO). Этот объект представляет собой некий блок информации о физическом устройстве, который используется ОС. Данный метод создает объект устройства XDSPDevice. С точки зрения системы, создается функциональный объект устройства (Functional Device Object, FDO).

NTSTATUS XDSPdrv::AddDevice(PDEVICE_OBJECT Pdo) {

 t << "AddDevice called\n";

 //Здесь вызывается конструктор класса XDSPDevice.

 XDSPdrvDevice* pDevice = new(

static_cast(KUnitizedName(L"XDSPdrvDevice", m_Unit)),

FILE_DEVICE_UNKNOWN,

static_cast(KUnitizedName(L"XDSPdrvDevice", m_Unit)),

0,

DO_DIRECT_IO)

 XDSPDevice(Pdo, m_Unit);

 //m_Unit – количество таких устройств в системе.

 if (pDevice == NULL) //Не удалось создать объект устройства. Похоже, произошла какая-то ошибка.

 {

t << "Error creating device XDSPdrvDevice" << (ULONG) m_Unit << EOL;

return STATUS_INSUFFICIENT_RESOURCES;

 }

 //Получить статус создания устройства.

 NTSTATUS status = pDevice->ConstructorStatus;

 if ( !NT_SUCCESS(status) ) //Похоже, устройство создано, но неудачно; произошла ошибка.

 {

t << "Error constructing device XDSPdrvDevice" << (ULONG) m_Unit << " status " << (ULONG) status << EOL;

delete pDevice;

 } else {

m_Unit++; //Устройство создано удачно

 }

 //Вернуть статус устройства.

 return status;

}

Все. Работа объекта драйвера на этом окончена. Как мы можем видеть, объект драйвера практически не выполняет каких-либо функций управления аппаратурой, но он жизненно необходим для правильной инициализации драйвера. В нашем случае НЕ ТРЕБУЕТСЯ вносить какие-либо изменения в текст, сформированный DriverWizard.

Основным классом драйвера является класс устройства. Класс устройства XDSPdrvDevice является подклассом класса KpnpDevice. Конструктор получает два параметра: указатель на PDO и номер драйвера в системе.

XDSPdrvDevice::XDSPdrvDevice(PDEVICE_OBJECT Pdo, ULONG Unit) : KPnpDevice(Pdo, NULL) {

 t << "Entering XDSPdrvDevice::XDSPdrvDevice (constructor)\n";

 //Здесь проверяется код ошибки, которую вернул конструктор суперкласса. В случае

 //успешного создания объекта базового класса значение переменной m_ConstructorStatus

 //будет NT_SUCCESS.

 if ( ! NT_SUCCESS(m_ConstructorStatus) ) {

Поделиться с друзьями: