Форум программистов [Powered by Invision Power Board]

Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате

Форум программистов > C/C++: Системное программирование и WinAPI > hook on closing program

Автор: tusha 14.8.2009, 22:20

необходимо запретить юзеру закрывать программу путем нажатия на крестик (варианты alt+f4, taskmanager можно не учитывать) . както давно надыбал на Винграде тему где описывается создание простейшего клавиатурного шпиона через хуки. во тчто слепил. прошу не пинать сильно (все мы когдато учились smile

)
DLL

Код


HHOOK hhk;

LRESULT CALLBACK PreventClose(int code,WPARAM wParam,LPARAM lParam);
extern "C" __declspec(dllexport) bool SetHook(wchar_t *windowclassname, bool state);

BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    return TRUE;
}

bool SetHook(wchar_t *windowclassname, bool state)
{
    HWND hWnd = FindWindowEx(NULL,NULL,windowclassname,NULL);
    if (hWnd==NULL)
        return false;

    if (state)
    {
        hhk=SetWindowsHookEx(WH_GETMESSAGE, (HOOKPROC)PreventClose, (HINSTANCE)hWnd, 0);
        if (!hhk)
            return false;
        return true;
    }
    else
    {
        UnhookWindowsHookEx(hhk);
        return true;
    }    
}

LRESULT CALLBACK PreventClose(int code,WPARAM wParam,LPARAM lParam)
{ 
    if(code==HC_ACTION)
    {
        if (((CWPSTRUCT *)lParam)->message==WM_CLOSE)
            MessageBoxEx(NULL,L"Unable to close application",L"ACHTUNG!",MB_ICONERROR,0);
    }
    return CallNextHookEx(hhk,code,wParam,lParam); }

MAIN

Код


#include "stdafx.h"
#include <iostream>
#include <WINDOWS.H>

extern "C" __declspec(dllimport) bool SetHook(wchar_t *winclassname, bool _s);
using namespace std;

typedef
bool (*SETHOOK)(bool);

int _tmain(int argc, _TCHAR* argv[])
{
    wchar_t _n[]=L"Notepad";

    HINSTANCE hDll;
    hDll = LoadLibrary(L"hooking.dll");
    if(!hDll)return 0;
    MSG msg;
    
    SETHOOK sh;
    sh = (SETHOOK)GetProcAddress(hDll,"SetHook");
    cout << SetHook(_n,true);
    
    if(sh)
        sh(true);
    
    while(GetMessage(&msg,0,0,0))
    {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    if(sh)
        sh(false);

    return 0;
}

=) вот что вышло. вобщем нифига не получается., блокнот как закрывался так и закрывается, никаких изменений.
меня смущают 1 и 3 параметры SetWindowsHookEx - правильный ли я ставлю фильтр? и как подвязать фильтр к конктретной программе? - в данном случае Notepad
натолкните на правильную мысль
буду благодарен за советы)

Автор: GoldFinch 14.8.2009, 22:25

при WM_CLOSE
return 0;

Добавлено через 1 минуту и 4 секунды
найди нужное место в целевой проге и пропатчи его

Автор: tusha 17.8.2009, 08:36

не, чет не получается.... smile

при данном коде, даже хук не получаетсяя поставить. -

Код


hhk=SetWindowsHookEx(WH_GETMESSAGE, (HOOKPROC)PreventClose, (HINSTANCE)hWnd, 0);

всегда NULL

если сделать так:

Код


HINSTANCE hInst;
BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    hInst=(HINSTANCE)hModule;
    return TRUE;
}
............................................................................................
hhk=SetWindowsHookEx(WH_MSGFILTER, (HOOKPROC)PreventClose, hInst,0);

хук типа ставится.. - но не понятно куда)))

отработка же

Код


if (((MSG *)lParam)->message==WM_CLOSE)
        {
            //MessageBoxEx(NULL,L"Unable to close application",L"ACHTUNG!",MB_ICONERROR,0);
            return 0;
        }

по прежнему не происходит.....
мож фильтр не правильный?

Автор: Cheloveck 17.8.2009, 09:05

Код


LRESULT CALLBACK WndProc(HWND hWnd,UINT Message, UINT wParam, LONG lParam)
{
    switch(Message)
    {
    case WM_CLOSE:
        return 0;
//.........................................
    }
    return DefWindowProc(hWnd, Message, wParam, lParam);
}

Зачем хуки?

Автор: tusha 17.8.2009, 09:47

хм, прикол в том, что из моей проги мне надо контролировать стороннее приложение (notepad, mspaint к примеру). или я чтото не до конца понимаю.... в каком контексте надо юзать

Код


LRESULT CALLBACK WndProc(HWND hWnd,UINT Message, UINT wParam, LONG lParam)
{
    switch(Message)
    {
    case WM_CLOSE:
        return 0;
//.........................................
    }
    return DefWindowProc(hWnd, Message, wParam, lParam);
}

????????

Автор: GoldFinch 17.8.2009, 10:13

tusha, найди в целевой проге место где она обрабатывает сообщения, и при WM_CLOSE возвращай 0

Автор: tusha 17.8.2009, 10:19

у меня нету исходников проги. только exe)

Добавлено через 6 минут и 47 секунд
вот что получилось на данный момент:
хук устанавливаю так:

Код


hhk=SetWindowsHookEx(WH_CALLWNDPROC, (HOOKPROC)PreventClose, hInst, 0);

обработка сообщений:

Код


if(code==HC_ACTION)
    {        
        if (((PMSG)lParam)->message==SC_CLOSE)        
        {
            MessageBoxEx(NULL,L"Unable to close application",L"ACHTUNG!",MB_ICONERROR,0);
            return 0;
        }
    }
    return CallNextHookEx(hhk,code,wParam,lParam); }

при закрытии блокнота АХТУНГ показыватеся )))) НО блокнот таки закрывается. + аналоггичный мессадж показывается на закрытие всех приложений (получился такой себе глобальный хук) smile

Автор: GoldFinch 17.8.2009, 10:36

tusha, реверсируй этот .ехе - получишь "исходники"
для этого можно использовать отладчик, например OllyDbg

Автор: tusha 17.8.2009, 10:56

гм... мож я не совсем правильно обрисовал задачу (или я не понимаю что до меня хотят донести =) )..
есть прога - не моя, исходников нету, менять ее НЕЛЬЗЯ (лицензия) - которая получает данные с датчиков в режиме онлайн. юзеры иногда вместо того, чтоб свернуть , попадают по крестику - тем самым прерывая получение данных.

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

получить ассемблерный код впринципе можно, а смысл?

Автор: GremlinProg 17.8.2009, 11:51

с первым параметром проблем нет, но вот третий - да

Цитата(tusha @ 15.8.2009, 00:20

)

(HINSTANCE)hWnd

HINSTANCE != HWND при всем желании

Код


HINSTANCE hInst;
BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    hInst = hModule;
    return TRUE;
}

bool SetHook(wchar_t *windowclassname, bool state){
...
 ThreadID = GetWindowThreadProcessId( hWnd, NULL );
 hhk=SetWindowsHookEx(WH_GETMESSAGE, (HOOKPROC)PreventClose, hInst, ThreadID);
...
}

только твой хук будет срабатывать во всех окнах заданного потока, если в PreventClose не проверять дескриптор

Автор: GoldFinch 17.8.2009, 13:11

на самом деле, задача решается так:

Код


#include <Windows.h>

int main()
{
    HWND hwnd=FindWindowW(0,L"Калькулятор");
    SetWindowLongPtr(hwnd,GWL_STYLE,WS_VISIBLE);
    return GetLastError();
}

со стилями надо экспериментировать

Автор: GremlinProg 17.8.2009, 13:15

Цитата(GoldFinch @ 17.8.2009, 15:11

)

на самом деле, задача решается так:

Код


#include <Windows.h>

int main()
{
    HWND hwnd=FindWindowW(0,L"Калькулятор");
    SetWindowLongPtr(hwnd,GWL_STYLE,WS_VISIBLE);
    return GetLastError();
}

со стилями надо экспериментировать

да, со стилями надо экспериментировать, только что, по твоему твой код делает?

Цитата(Window Styles @ MSDN)

WS_VISIBLE
Creates a window that is initially visible.
This style can be turned on and off by using ShowWindow or SetWindowPos.

Добавлено через 4 минуты и 57 секунд
ты просто выключаешь все стили окна, причем закрыть его по прежнему можно: Alt+F4

Автор: GoldFinch 17.8.2009, 14:04

GremlinProg, я тестировал свой код, он делает что надо - убирает крестик (вместе со всей строкой заголовка)
если убрать VS_VISIBLE, окно пропадет насовсем, проверь

Цитата(tusha @ 14.8.2009, 23:20

)

(варианты alt+f4, taskmanager можно не учитывать)

избавляться от Alt+F4 ТС не надо

Автор: GremlinProg 17.8.2009, 14:12

Цитата(GoldFinch @ 17.8.2009, 16:04

)

я тестировал свой код, он делает что надо - убирает крестик (вместе со всей строкой заголовка)

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

Автор: GoldFinch 17.8.2009, 15:01

решение с заменой стиля - именно то которое требуется
надо посмотреть родной стиль окна (например spy++) и изменить его так, чтобы крестик пропал\заблокировался

Добавлено через 5 минут и 46 секунд
вот полностью доделанное решение

Код


#include <Windows.h>

int main()
{
    HWND hwnd=FindWindowW(0,L"Калькулятор");
    LONG_PTR style = GetWindowLongPtr(hwnd,GWL_STYLE);
    style^=WS_SYSMENU;
    SetWindowLongPtr(hwnd,GWL_STYLE,style);
    return GetLastError();
}

убирает все кнопки из строки заголовка, сохраняя все остальные стили
(строка заголовка остается)

Автор: GremlinProg 17.8.2009, 15:37

GoldFinch, внимательно перечитай:

Цитата(tusha @ 15.8.2009, 00:20

)

необходимо запретить юзеру закрывать программу путем нажатия на крестик

Цитата(tusha @ 17.8.2009, 12:56

)

юзеры иногда вместо того, чтоб свернуть , попадают по крестику

и скажи:
1. будет ли включен флаг b, если выполняется условие !a && b
2. будет ли включен флаг b, если выполняется условие !(a || b)

все просто до безобразия: b - это кнопка сворачивания, а - это кнопка закрытия
скрывая панель, ты тем самым скрываешь все кнопки, включая b, которой люди пользуются,
WS_SYSMENU работает аналогично, зачем мутить воду-то, непонимаю

Автор: GoldFinch 17.8.2009, 15:54

Модератор: Сообщение скрыто.

Автор: GremlinProg 17.8.2009, 16:22

GoldFinch, на будущее: еще хоть раз увижу подобные наезды на участников, будешь потешаться в одиночестве, и за пределами форума

опять же повторюсь, я не придираюсь к твоим советам, я просто предвосхищаю заблуждения участников, в которые ты их вводишь осознанно, и все надеюсь, что когда-нибудь совесть в тебе проснется и ты перестанешь давать советы мимо темы

"иногда лучше молчать..."©

Автор: Cheloveck 17.8.2009, 17:19

Вот полное решение ИМХО.

Код


#include <windows.h>
#include <tchar.h>

WNDPROC g_defproc;

LRESULT CALLBACK WndProc(HWND hWnd,UINT Message, UINT wParam, LONG lParam)
{
    switch(Message)
    {
    case WM_CLOSE:
        return 0;
    }
    return g_defproc(hWnd, Message, wParam, lParam);
}

int main()
{
    HWND hwnd = FindWindow( 0, _T("Калькулятор") ); // Тут можно и класс окна вписать.
    g_defproc = (WNDPROC)SetWindowLong( GWL_WNDPROC, (LONG)g_defproc );
    
    MSG msg;
    while( GetMessage( &msg, NULL, 0, 0 ) )
    {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
    return msg.wParam;
}

Добавлено через 4 минуты и 25 секунд
Хотя, хз как будет работать между процессами

Автор: GremlinProg 17.8.2009, 18:12

Цитата(Cheloveck @ 17.8.2009, 19:19

)

Хотя, хз как будет работать между процессами

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

Автор: dumb 17.8.2009, 18:21

Цитата(Cheloveck @ 17.8.2009, 18:19

)

Хотя, хз как будет работать между процессами

никак не будет.

tusha, во-первых сообщения могут доставляться окну не только через очередь сообщений, но и посредством прямого или опосредованного вызова оконной процедуры - хук WH_GETMESSAGE соответственно в этом случае не срабатывает. такая доставка перехватывается хуками WH_CALLWNDPROC/WH_CALLWNDPROCRET, но изменять сообщение в этих хуках нельзя.
во-вторых, и это основное - в своем обработчике хука ты не делаешь ничего для отмены сообщения.

должно быть как-то так:

Код


if(code==HC_ACTION)
{
    PMSG p = (PMSG)lParam;
    if (p->message == SC_CLOSE)
    {
        TCHAR fullname[MAX_PATH];
        ::GetModuleFileName(0, fullname, MAX_PATH);
        TCHAR *fname = _tcsrchr(fullname, '\\');
        if (fname++ && !_tcsicmp(fname, _TEXT("notepad.exe")))
        {
            MessageBoxEx(NULL,L"Unable to close application",L"ACHTUNG!",MB_ICONERROR,0);
            p->message = WM_NULL; // !!!
            return 0;
        }
    }
}
return CallNextHookEx(hhk,code,wParam,lParam);

Автор: Cheloveck 17.8.2009, 18:44

да, проверил, моя не работать((

Автор: Earnest 17.8.2009, 20:32

Может, попробовать просто задизэблить крестик?

Цитата

Use the following code to disable the Close button.
EnableMenuItem (GetSystemMenu(hwnd, FALSE), SC_CLOSE,MF_BYCOMMAND | MF_GRAYED);

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

Автор: GremlinProg 17.8.2009, 22:47

да, Earnest, как раз то что tusha и требовалось:

Цитата(tusha @ 15.8.2009, 00:20

)

запретить юзеру закрывать программу путем нажатия на крестик (варианты alt+f4, taskmanager можно не учитывать)

Автор: tusha 20.8.2009, 08:34

Код


EnableMenuItem (GetSystemMenu(hwnd, FALSE), SC_CLOSE,MF_BYCOMMAND | MF_GRAYED);

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

про WM_NULL додул раньше, однако как сказал тов. dumb для данного фильтра (WH_CALLWNDPROC) это отказывалось работать. будем пробовать дальше, если ниче не выйдет, то придется дизейблить крестик )))

Автор: tusha 2.9.2009, 13:10

вобщем наткнулся на такой бок(?)
(из предыдущих моих тем) запускаю следующий код:

Код


WindowName = L"Untitled - Notepad";
WinHandle = ::FindWindowW(0,WindowName);
if (WinHandle != NULL)
{
    EnableMenuItem(::GetSystemMenu((HWND)WinHandle, FALSE), SC_CLOSE, MF_ENABLED);        
        //стараюсь полюбому дать доступ к кнопке закрыть

    EnableMenuItem(::GetSystemMenu((HWND)WinHandle, FALSE), SC_CLOSE, MF_GRAYED | MF_DISABLED); // потом ее дизейблю
    SetDlgItemText(IDC_BUTTON1,L"Enable");
    ::DrawMenuBar(WinHandle);// также UpdateWindow пробовал
}

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

Код


EnableMenuItem(::GetSystemMenu((HWND)ClientWinHandle, FALSE), SC_CLOSE, MF_ENABLED);

реагурует отлично
--------------------------------------------------
потом закрываем программу (блокнот остается с ЗАБЛОКИРОВАННЫМ крестиком) и запускаем ее снова - пробуем разблокировать крестик - ничего не получается.... smile

функция же в данной ситуации возвращает всегда -1 (т.е. какбудто меню вообще нету)

в ситуации когда прогу закрываю, но крестик окна блокнота MF_ENABLED - все работает отлично

ктонибудь сталкивался с таким? есть идеи?

Добавлено через 4 минуты и 43 секунды
еще одно маленькое уточнение - хендл окна я получаю, но рядом со значением в скобках стоит unused ????

Автор: GremlinProg 3.9.2009, 12:16

перед закрытием своей программы, просто вывзови:

Код


::GetSystemMenu( WinHandle, TRUE );

тогда и восстанавливать состояние кнопки не потребуется
конечно, WinHandle должен храниться на протяжении всей работы твоей программы

параметр TRUE означает, что копия системного меню, которую ты создал будет уничтожена,
а на его место будет установлено меню по-умолчанию

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

это видимо просто механизм защиты от утечек ресурсов, который позволяет копировать меню только с оригинала

Добавлено через 6 минут и 52 секунды
да, кстати, не нужно вызывать GetSystemMenu всякий раз, когда тебе требуется дескриптор меню,
у тебя должно быть только 2 вызова этой функции: в начале работы с меню и в конце, перед закрытием твоей программы

Автор: tusha 3.9.2009, 12:46

гениально)
поставил бы +1 но кол-во постов пока не позволяет.)))
и главное что в описании EnableMenuItem нету хотя бы упоминания про этот прикол) иль я не внимательно смотрел)

ЗЫ: век живи век учись у дураком помрешь

Автор: GremlinProg 3.9.2009, 13:19

Цитата(tusha @ 3.9.2009, 14:46

)

и главное что в описании EnableMenuItem нету хотя бы упоминания про этот прикол) иль я не внимательно смотрел)

в EnableMenuItem упоминаний и не должно быть, в GetSystemMenu должно быть, но оно несколько смазано,
в поддержке есть, только немного на другом примере: http://support.microsoft.com/kb/300688

Цитата(How to disable the Close button on the Application window and the Exit command on the File menu @ support.microsoft.com)

Please note that this technique affects the Close button on the application window of Microsoft Access, not the Close button on the Database window. After disabling these options, neither is automatically re-enabled when your database closes. If the user closes the database and leaves Microsoft Access open, the user will not be able to quit Microsoft Access by using the Close button or the Exit command on the File menu. In this case, your application should re-enable both options before it terminates. Otherwise, the user will have to quit and restart Microsoft Access in order for the Close button and the Exit command on the File menu to be enabled.

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

Автор: tusha 3.9.2009, 14:03

Цитата

bRevert
[in] Specifies the action to be taken. If this parameter is FALSE, GetSystemMenu returns a handle to the copy of the window menu currently in use. The copy is initially identical to the window menu, but it can be modified. If this parameter is TRUE, GetSystemMenu resets the window menu back to the default state. The previous window menu, if any, is destroyed.

http://msdn.microsoft.com/en-us/library/ms647985%28VS.85%29.aspx

smile