книги хакеры / Защита_от_взлома_сокеты,_эксплойты,_shell_код_Фостер_Дж_

Как следует из названия, библиотека ATL основана на применении шаблонов, как и стандартная библиотека шаблонов Standard Template Library (STL). Программирование с помощью шаблонов – это один из вариантов повторного использования кода. Вместо того чтобы наследовать классу и получать в довесок кучу функциональности, которая вам совершенно ни к чему, шаблон позволяет точно определить, что именно делает класс.

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

Шаблон же позволяет специализировать класс во время определения для работы с данными конкретного типа. Рассмотрим, например, такое определение:

Stack<int> myIntegerStack; myIntegerStack.push(10); myIntegerStack.push(5);

Здесь аргумент шаблона класса Stack заключен в угловые скобки < >. Сам класс Stack определен следующим образом:

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

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

Технология реализации клиента с помощью ATL

Библиотека ATL поддерживает несколько классов, помогающих избавиться от повторяющегося кода в клиентских приложениях COM. По большей части этот код связан с интерфейсами IUnknown и IDispatch, а также со специальными типами данных VARIANT и BSTR.

Как вы уже знаете, в COM интерфейс IUnknown является базовым для всех остальных и отвечает за подсчет ссылок. ATL предлагает два шаблонных класса, упрощающих работу с IUnknown: CComPtr è CComQIPtr.

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

параметром шаблона CComPtr является тип интерфейса, на который класс будет указывать, например, CComPtr<IDispatch>;

шаблон CComPtr содержит два перегруженных метода с именем CoCreateInstance. Ни одному из них идентификатор интерфейса не передается в качестве параметра, так как во время конкретизации класса CComPtr интерфейс уже был указан. Разница между двумя этими методами в том, что один позволяет сослаться на загружаемый компонент по его CLSID, а другой – по строке, содержащей ProgID того же компонента (понятное человеку имя);

оператор присваивания перегружен так, что во время этой операции счетчик ссылок увеличивается на единицу;

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

Приведем пример использования класса CComPtr:

void main()

{

CComPtr<IXMLDOMDocument> spDoc;

HRESULT hr = spDoc.CoCreateInstance(L"MSXML.DOMDocument"); if (FAILED(hr))

return hr;

}

Поддержка типов данных

Практически во всех интерфейсах, поддерживающих автоматизацию, применяются типы данных BSTR и VARIANT. Для обоих в ATL есть поддерживающие классы.

Тип данных BSTR

BSTR, или двоичная строка, – это строка символов в кодировке Unicode, которой предшествует число типа WORD, определяющее длину строки. Поскольку, это не обычная строка (и, следовательно, для ее инициализации нельзя воспользоваться строковыми литералами), для работы с ней необходима поддержка со стороны среды исполнения COM. Например, для создания

58*/

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

функцией VariantClear. При присваивании переменной типа VARIANT значе-

VARIANT var;

VariantInit(&var);

var.vt = VT_UI4;

var.ulVal = 1024;

VariantClear(&var);

Библиотека ATL предоставляет поддержку для типа VARIANT в форме класса CComVariant. Он упрощает работу с типом VARIANT, переопределяя оператор присваивания для типичных случаев, как, например, LPWSTR, int, long, char и CComBSTR. Кроме того, конструктор и деструктор автоматически вызывают функции VariantInit и VariantClear соответственно.

Вот как можно было бы воспользоваться этим вспомогательным классом:

CComVariant var;

var = CComBSTR(L"This is my variant structure containing a BSTR");

Технология реализации сервера с помощью ATL

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

Композиция классов

Поскольку композиция класса не зависит от типа сервера, в котором этот класс содержится, начнем изложение именно с этого вопроса.

Основные требования к любому COM-объекту – это реализацию фабрики класса, поддержка интерфейса IUnknown и прочих интерфейсов в зависимости от функциональности объекта. Классы из библиотеки ATL, перечисленные в таблице 13.1, могут помочь удовлетворить эти требования.

Таблица 13.1. Основные классы ATL, поддерживающие композицию