Важная информация
RSS лента

The trick

PNG/ICO/CUR/ANI изображения в стандартных контролах VB6.

Оценить эту запись

Всем привет.

Как известно встроенные средства Visual Basic 6.0 не поддерживают возможности работы с PNG изображениями, 32 битными иконками с альфа каналом и анимированными курсорами т.е. к примеру нельзя ипользовать Png картинку в качестве свойства Form.Picture. Я представляю небольшую библиотеку и Add-in которые позволяют обойти эти ограничения. Данная библиотека позволяет загружать и сохранять такие изображения (с альфа каналом) стандартными средствами (LoadPicture / SavePicture), а также включает поддержку таких изображений (с альфа каналом) в контролы. Любой контрол который в своей работе использует стандарнтые Ole Picture объекты будет поддерживать загрузку таких изображений. В свою очередь если изображение выводится посредством IPicture::Render то картинка будет отрисовываться с учетом альфа канала. Данная библиотека должна работать на всех версиях Windows начиная с XP:



Как использовать?

Библиотека может быть использована как внешняя DLL либо быть прилинкована к исполняемому файлу (только native code). Для использования в качестве Dll необходимо вызвать функцию Initialize которая вернет 1 в случае успеха. После этого можно пользоваться возможностями библиотеки. Если необходимо выгрузить библиотеку то нужно вызвать функцию CanUnloadNow которая сообщит можно ли в данный момент выгрузить библиотеку. Если библиотека готова к выгрузке функция вернет S_OK после которой нужно вызвать Uninitialize. Если функция возвращает S_FALSE то библиотеку нельзя выгружать т.к. имеются активные Picture объекты которые еще не выгружены и они используют библиотеку. Для IDE создан специальный Add-in который автоматически загружает библиотеку при старте среды. В скомпилированном варианте можно к примеру в событии Initialize или в процедуре Main вызывать Initialize, а при завершении Uninitialize:
VB Code:
  1. Private Declare Function Initialize Lib "VBPng.dll" () As Long
  2. Private Declare Sub Uninitialize Lib "VBPng.dll" ()
  3.  
  4. Private Sub Form_Initialize()
  5.  
  6.     If Initialize() = 0 Then
  7.         MsgBox "Unable to initialize png dll", vbCritical
  8.     End If
  9.     
  10. End Sub
  11.  
  12. Private Sub Form_Terminate()
  13.     Uninitialize
  14. End Sub


Для статической линковки необходимо использовать более новый линкер (в своих примерах я использовал линкер из Visual Studio 2010), поскольку оригинальный имеет баги при использовании опции /OPT:REF, а также в секцию VBCompiler файла проекта (vbp) необходимо добавить параметры:
Для EXE:
Код :
LinkSwitches= ..\Libs\msvcrt_winxp.obj ..\Libs\VBPng.lib -ENTRY:mainCRTStartup
Для DLL:
Код :
LinkSwitches= ..\Libs\msvcrt_winxp.obj ..\Libs\VBPng.lib -ENTRY:VBDllMain -EXPORT:Initialize -EXPORT:Uninitialize
В случае DLL, в скомпилированном виде необходимо сделать инициализацию, вызвав Initialize из себя же при первом запуске.

Как это работает?

Библиотека написана на C++. Принцип работы библиотеки основан на перехвате функций OleLoadPictureEx, OleLoadPicture и OleIconToCursor. Данные функции не поддерживают загрузку подобных изображений, поэтому если загружается PNG файл, библиотека VbPng пытается загрузить файл с помощью GDI+ или USER32 (в зависимости от типа изображения). При успехе создается аналогичный StdPicture объект который и возвращается функцией. Для вызывающей стороны все это выглядит как-будто она работает с оригинальным объектом. Сам объект поддерживает интерфейсы IPicture, IPictureDisp, IPersistStream, IConnectionPointContainer (не поддерживает connection point'ы возвращает E_NOTIMPL), IDispatch, поэтому может быть присвоен Object переменной или к примеру быть сохраненным в PropertyBag.

Для того чтобы использовать библиотеку необходимо вызвать функцию Initialize. Данная функция инициализирует данные необходимые для функционирования модуля. Инициализируется COM, GDI+, устанавливаются перехваты и регистрируется COM-сервер изображений. Данный сервер нужен для создания объектов изображений посредством CoCreateInstance в особенности при создании изображений из PropertyBag'а или другого хранилища. Когда рантайм загружает изображения из PropertyBag'а создается объект по CLSID сохраненному в хранилище (IPersist::GetClassID), а затем выполняется инициализация посредством IPersistStream::Load.


Перехватчик функций реализован в классе CHooker. Данный класс использует дизассемблер длин (ldasm) от Ms-Rem с небольшой доработкой. Доработка заключается в добавление флага OP_REL32 к некоторым инструкциям (к примеру JMP SHORT), поскольку в оригинале на некоторых относительных инструкциях этот флаг отсутствовал. Для перехвата функции используется простейший метод сплайсинга при котором в начало функции всталяется инструкция JMP которая переводит поток исполнения на функцию-перехватчик. Поскольку в начале оригинальной функции содержатся инструкции которые мы перезаписываем, необходимо правильно перенести инструкции для того чтобы была возможность вызвать оригинальную функцию. При вызове метода Hook с помощью дизассемблера длин определяется целое количество инструкций которое будет перезаписано инструкцией JMP (5 байт). После этого выделяется временный буфер (с разрешением на исполнение данных) в который будут скопированы данные инструкции + JMP на инструкцию следующую за перезаписываемой. Это позволит, передав управление на этот буфер, вызвать оригинальную функцию как-будто перехвата не было. Тут существует одна сложность заключающаяся в том, что мы не можем просто так скопировать инструкции, поскольку существуют относительные инструкции типа JMP, CALL, JNE которые "прыгают" относительно своего адреса. Для определения типа инструкции как раз и служит флаг OP_REL32 который показывает является ли инструкция относительной или нет. Другая сложность заключается в том что существуют "короткие" относительные инструкции которые "прыгают" в пределах 255 байт, а при переносе кода в буфер расстояние может значительно увеличится. Поэтому после определения количества перезаписываемых инструкций выделяется буфер размером как минимум чтобы обеспечить транслирование из коротких в длинные инструкции. После этого производится анализ каждой инструкции и при необходимости происходит корректировка смещения и типа. В конце буфера добавляется инструкция JMP со смещением на инструкцию следующую за последней перезаписаной. Наконец начало функции перезаписывается на безусловный JMP на функцию-перехватчик.

CHooker объекты используют в качестве буфера кода кучу (Heap) с разрешением на исполнение, поэтому код является DEP безопасным. Куча автоматически создается при создании первого перехватчика и удаляется при уничтожении последнего. В проекте используются 3 таких объекта для перехвата 3-х функций OleLoadPictureEx, OleLoadPicture и OleIconToCursor, с соответствующим перехватчиками OleLoadPictureEx_user, OleLoadPicture_user и OleIconToCursor_user. В системах до Windows 8 (для загрузки изображений) можно было перехватывать только одну OleLoadPictureEx функцию которая вызывается из OleLoadPicture, но начиная с Windows 8 OleLoadPicture вызывает уже недокументированную OleLoadPictureExt, поэтому для обеспечения правильной работы некоторых контролов (к примеру ImageList) нужно перехватывать 2 этих функции. Конечно можно пробовать перехватывать OleLoadPictureExt, но эта функция недокументирована и не факт что в новых версиях Microsoft не изменят эту функцию на другую. В перехватчиках вызывается оригинальная функция и если вызов окончился неудачей вызывается наша реализация. Чтобы обеспечить возможность узнать был ли перехват уже осуществлен (к примеру подгруженная DLL уже перехватила и нет смысла делать это еще раз) используется переменна окружения "VBPng".

OleIconToCursor перехватывается потому что она используется рантаймом для конвертирования иконок в курсоры. Перехватывая ее, мы получаем возможность обойти ограничения данной функции. Дело в том что эта функция не может правильно конвертировать анимированные курсоры (она конвертирует только первый кадр). Проект содержит класс CANICursors который хранит ссылки на все созданные анимированные курсоры. Когда перехватчик получает хендл иконки он пытается найти подходящий CICOPicture объект и если такой имеется, делается правильная копия анимированного курсора.

Основа библиотеки - класс CPicture который и реализует всю логику работы изображений. Данный класс создавался на основе реверс-инжиниринга библиотеки oleaut32 некоторые функции возможно реализованы не точно. Данный класс позволяет загружать PNG изображения из COM потока (IStream), а также сохранять их в него. Библиотека ведет учет созданных объектов в глобальной переменной g_lCountOfObject для того чтобы обеспечить контроль при выгрузке библиотеки вызовом CanUnloadNow. В противном случае не было бы способа узнать можно ли выгрузить библиотеку или нет. Соответственно при выгрузке библиотеки которой пользуются активные объекты происходило бы падение.

Загрузка изображений выполняется методом LoadFromStream. Этот метод проверяет сигнатуру загружаемого файла и на основе этой информации создает дочерний CPictureInternal объект который содержит специфичную для каждого типа изображений логику работы. Поскольку при загрузке из потока GDI+ автоматически устанавливает указатель в начало, приходится создавать поток в котором содержатся только данные файла (также данный поток нужен для обеспечения поддержки свойства KeepOriginalFormat). Данная задача выполняется формато-специфичными методами CreatePngStream/CreateIcoStream/CreateAniStream в которых также происходит валидация данных.

Второй по важности метод - это Render. Тут все просто, происходит подготовка координат для вывода изображения в HIMETRIC и происходит вывод с помощью виртуального метода CPictureInternal:raw. Для PNG используется AlphaBlend, для иконок - DrawIconEx. Т.к. свойство get_Attributes возвращает PICTURE_TRANSPARENT то пользователь перед выводом изображения сам заботится о восстановлении фона за изображением.

Метод SaveAsFile сохраняет изображение в поток. Тут все тоже самое только наоборот. Также стоит отметить что если использовалось сохранение оригинального формата то данные изображение берутся из сохраненного потока. В противном случае, для PNG, создается временный GDI+ битмап из пикселей DIB-секции, извлекается CLSID PNG кодека и происходит сохранение изображения во временный поток. Далее из этого потока данные копируются в поток назначения. CICOPicture класс поддерживает сохранение только из сохраненного потока.

Как бонус наше приложение получает полноценную 32-битную иконку:


Следующая группа методов это реализация интерфейса IDispatch. Поскольку данные о типе IPicture хранятся в стандартной библиотеке stdole2.tlb то в методе GetTypeInfo происходит загрузка этой библиотеки с извлечением нужного интерфейса типа через ITypeLib::GetTypeInfoOfGuid. Тоже самое относится к методу GetIDsOfNames, тут просто происходит транслирование вызова стандартному ITypeInfo::GetIDsOfNames. Метод Invoke реализован напрямую с проверкой параметров.

Для того чтобы можно было статически прилинковать библиотеку к VB6 EXE файлу необходимо инициализировать сишный рантайм передачей управления на функцию mainCRTStartup и передать управление на метку ___vbaS. Для этой цели служит файл gostartup.asm написаный на fasm'е. Для EXE файла выполняются строчки:
Assembler Code:
  1. _main:
  2. call Initialize
  3. jmp ___vbaS

Сишный рантайм вызывает функцию main, а она в свою очередь инициализирует библиотеку VbPng. Тут существует проблема со старым линкером, поскольку то ли из-за бага, то ли из-за чего то еще, ликер отбрасывает весь VB-шный импорт из результирующего файла при использовании опции -OPT:REF. Решается данная проблема просто - заменой линкера на современный.
Для DLL выполняются похожие действия, только в этом случае необходимо указать в качестве точки входа _VBDllMain:
Assembler Code:
  1. _VBDllMain:
  2.  
  3. push dword [esp + 12]
  4. push dword [esp + 12]
  5. push dword [esp + 12]
  6.  
  7. ; // Init CRT
  8. call  __DllMainCRTStartup@12
  9.  
  10. ; // Init runtime
  11. jmp ___vbaS

В этом случае сначала вызывается инициализации сишного рантайма, а затем происходит переход на функцию DllMain ActiveX Dll.

Для олегчения работы в IDE был написан Add-in который автоматически загружает VbPng.dll для того чтобы было удобно работать с проектами. Для отключения библиотеки просто нужно отключить Add-in. Тут есть ньюанс, если есть активные PNG-изображения, то Add-in выгрузится, но VbPng нет, при этом покажется предупреждение. В любой момент можно будет включить Add-in, найти изображения, удалить их, и заново отключить Add-in, тогда DLL выгрузится.




Некоторые контролы, к примеру ListView, не будут отображать альфа канал, поскольку отрисовывают себя не методом Render, а через StretchBlt, для них premultiplied фон будет черный. Это следует иметь в виду при работе с библиотекой. Также не поддерживаются уведомления IPropertyNotifySink (при желании можно реализовать). Ресурсы в FRX файлах и скомпилированных файлах также хранятся в PNG поэтому проекты не будут открываться и работать без библиотеки. Для комфортной работы рекомендуется установить Add-in с автоматическим запуском при загрузке IDE.

В директории содержатся также несколько примеров работы:
  • Test_EXE_Linked - демонстрация 32bpp PNG изображений на стандартных контролах с использованием статической линковки;
  • Test_EXE_Dll - тоже самое только с использованием dll;
  • Test_AXDll - ActiveX DLL библиотека с использованием PNG ресурсов на форме;
  • Test_SavePng - пример сохранения изображения посредством SavePicture;
  • SaveToPropBag - пример загрузки из PropertyBag;
  • IconsCursors - демонстрация работы с 32 битными иконками и анимированными курсорами.


Также в директории содержатся PNG файлы, собраные мной еще давно посредством спутниковой рыбалки.

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

Проект на GitHub.

The trick,
2019.
Метки: alpha, ani, png, vb6 Добавить / редактировать метки
Категории
Visual Basic 6.0

Комментарии