Подмена оконной процедуры, Как правильно завершить?

[eduard93]

Добрый день.

У меня программа внедряет DLL в чужой процесс, а от туда DLL делает сабклассинг окну, таким образом:

WNDPROC pOldWndProc = (WNDPROC)SetWindowLongPtr(hWnd, GWLP_WNDPROC, (LONG_PTR)NewWndProc);

Все работает как надо, и когда я заканчиваю, я вызываю:

SetWindowLongPtr(hWnd, GWLP_WNDPROC, (LONG_PTR)pOldWndProc);

После чего сразу выгружаю DLL.

Проблема такая: иногда процесс вылетает, потому что пытается вызвать NewWndProc, а DLL уже выгружен.
Каким образом я могу подождать, пока все вызовы NewWndProc сделаются, чтоб только тогда выгрузить DLL?

P.S.
Я пробовал так:

static LRESULT CALLBACK NewTaskbarProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    wnd_proc_call_counter++;

    // ...

    wnd_proc_call_counter--;

    return result;
}

SetWindowLongPtr(hWnd, GWLP_WNDPROC, (LONG_PTR)pOldWndProc);

while(wnd_proc_call_counter > 0)
    Sleep(10);

Все равно иногда вылетает, да и не красиво получается...

Спасибо!