Перший заголовки зберігає в собі прототип
WLX_NOTIFICATION_INFO. А другий стане в нагоді в процесі створення логів.
У нас вже є функція DllMain, люб'язно
надана візард. Залишимо її без змін. Напишемо функцію створення
логів. Оскільки структура WLX_NOTIFICATION_INFO містить в собі Юнікод-рядки,
буде зручно виводити дані в текстовий файл типу Юнікод.
void WriteLog (PWSTR pStrEventName,
PWLX_NOTIFICATION_INFO
pInfo)
(
// Перший параметр функції WriteLog буде
// містити опис події,
// а друге - самі дані.
// Задаємо назва log-файлу
WCHAR fName [] = L "C: log.txt";
WCHAR buf [1024];
// Готуємо рядок для логів. Варто зазначити
що якби
// було оголошено макрос STRICT, висновок
hDesktop
// необхідно було б робити по іншому.
swprintf (buf, L "Event:% s, User:% s,
Domain:% s, "
L "Window station:% s,
User token 0x% p, "
L "Desktop
0x% prn ", pStrEventName,
pInfo-> UserName,
pInfo-> Domain, pInfo-> WindowStation,
pInfo-> hToken,
pInfo-> hDesktop);
// Відкриваємо log-файл для запису.
HANDLE hFile =
CreateFileW (fName, GENERIC_WRITE,
FILE_SHARE_READ, NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
DWORD dwWritten;
// Якщо такого файлу немає, або сталася якась інша
// помилка, пробуємо створити новий файл.
if (hFile == INVALID_HANDLE_VALUE)
(
hFile = CreateFileW (fName,
GENERIC_WRITE,
FILE_SHARE_READ, NULL,
CREATE_NEW,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile ==
INVALID_HANDLE_VALUE)
(
// Якщо файл не створився, повідомляємо
користувачеві
// системним звуком про виникнення
помилки.
:: MessageBeep (MB_ICONASTERISK);
return;
)
WCHAR data = 0xFEFF;
// Зазначаємо, що створений файл буде
// включати текст типу Юнікод.
if (! WriteFile (hFile, & data,
sizeof (data),
& dwWritten, NULL))
:: MessageBeep (MB_ICONASTERISK);
)
// Оскільки відкритий файл може бути не
порожнім,
// переходимо в його кінець
if (SetFilePointer (hFile, 0,
NULL, FILE_END) ==
INVALID_SET_FILE_POINTER)
(
:: MessageBeep (MB_ICONASTERISK);
return;
)
// Записуємо рядок логу в файл
if (! WriteFile (hFile, buf, wcslen (buf) *
sizeof (WCHAR),
& dwWritten, NULL))
:: MessageBeep (MB_ICONASTERISK);
// Закінчуємо роботу з файлом
CloseHandle (hFile);
)
Тепер залишилося написати функції-обробники подій.
Заради зручності назвемо їх так само, як показано в таблиці 1.
extern "C" void __stdcall
WLEventLogon (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "user
logon ", pInfo);
)
extern "C" void __stdcall
WLEventLogoff (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "user
logoff ", pInfo);
)
extern "C" void __stdcall
WLEventStartup (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "system
startup ", pInfo);
)
extern "C" void __stdcall
WLEventShutdown (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "system
shutdown ", pInfo);
)
extern "C" void __stdcall
WLEventStartScreenSaver (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "screen saver
started ", pInfo);
)
extern "C" void __stdcall
WLEventStopScreenSaver (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "screen saver
stopped ", pInfo);
)
extern "C" void __stdcall
WLEventLock (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "Workstation
locked ", pInfo);
)
extern "C" void __stdcall
WLEventUnlock (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "Workstation
unlocked ", pInfo);
)
extern "C" void __stdcall
WLEventStartShell (PWLX_NOTIFICATION_INFO pInfo)
(
WriteLog (L "User already
loged on and network "
L "resouces are
avaliable ", pInfo);
)
Тепер необхідно експортувати ці функції для
інших програм. Для цього створимо файл Notify.def і додамо його в проект. У
єдиною секції EXPORTS вкажемо експортовані функції
EXPORTS
WLEventLogon
WLEventLogoff
WLEventStartup
WLEventShutdown
WLEventStartScreenSaver
WLEventStopScreenSaver
WLEventLock
WLEventUnlock
WLEventStartShell
Після успішної компіляції необхідно підписатися на
події Winlogon.
ПОПЕРЕДЖЕННЯ
Необхідно пам'ятати, що dll виповнюється
в адресному просторі процесу, який її викликає. Якщо dll спровокує
виняткову ситуацію, тоді, якщо не обробити помилку, робота процесу
буде припинено. Процес Winlogon.exe дуже чутливий до таких речей. При
виникненні помилки він покаже синій екран смерті, а в разі наявності
відповідних установок перевантажить систему. Тому наполегливо рекомендується
тестувати програму на віртуальному комп'ютері.
Для цього зайдемо до реєстру і знайдемо ключ:
HKEY_LOCAL_MACHINESoftwareMicrosoftWindows
NTCurrentVersionWinlogonNotify
Створимо в ньому з'єднання MyNotify і значення, як
показано на малюнку нижче.
Малюнок 1.
Для зручності помістимо файл Notify.dll за вказаною
шляху. Після перезавантаження системи додаток почне створювати логи.
ПРИМІТКА
Якщо логи не створюються, перевірте, чи не
чи використовується Notify.dll іншим процесом (якщо файл не вдається
перейменувати - значить, хтось його використовує). Якщо Notify.dll використовується,
а логів ні, швидше за все, неправильно задано одне з значень у назвах
функцій-обробників, або такі функції не експортуються динамічної
бібліотекою. Якщо Notify.dll не використовується, перевірте, чи правильно задано
ключ DllName з назвою бібліотеки. Можливо, бібліотека не зареєстрована
в системі, або не може бути завантажена.
Висновок
Winlogon notification package може стати в нагоді в
зовсім несподіваних місцях. Наприклад, якщо із системної служби необхідно
отримати імена користувачів, що працюють в системі. Усе, що потрібно - це
написати і зареєструвати в системі dll, яка за запитом передасть дані в
службу (або сама буде передавати ці дані у відповідь на деякі потрібні
події).
Сподіваюся, що ця стаття розширить знання читача про
інтерфейсах Winlogon, а також допоможе використовувати отримані знання на
практиці.
Список літератури
Для підготовки даної роботи були використані
матеріали з сайту http://www.rsdn.ru/
