Add key logging support

1 files changed, 79 insertions, 2 deletions

diff --git a/vexnc.c b/vexnc.c
index 51cd483..3ef1383 100644
--- a/vexnc.c
+++ b/vexnc.c
@@ -1,16 +1,90 @@
-// gcc -O2 -march=native -o vexnc vexnc.c -lvncserver -lm
+#include <stdio.h>
+#include <stdint.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <sys/stat.h>
 
 #include <rfb/rfb.h>
 
 #define STB_IMAGE_IMPLEMENTATION
 #include "stb_image.h"
 
+int logdirenabled = 0;
+int fdlog = -1;
+char logfn[64] = "";
+
+int mkderp(const char *name) {
+  int ret = mkdir(name, S_IRWXU | S_IRWXG | S_IRWXO);
+  if (ret < 0 && errno == EEXIST) {
+    return 0;
+  }
+  return ret;
+}
+
+int spewchr(int fd, uint16_t chr) {
+  char buf[4];
+  int i = 0;
+  if (chr < 0x80) {
+    buf[i++] = chr;
+  } else if (chr < 0x800) {
+    buf[i++] = 0xC0 | chr >> 6;
+    buf[i++] = 0x80 | chr & 0x3F;
+  } else {
+    buf[i++] = 0xE0 | chr >> 12;
+    buf[i++] = 0x80 | (chr >> 6) & 0x3F;
+    buf[i++] = 0x80 | chr & 0x3F;
+  }
+  return write(fd, buf, i);
+}
+
+int openlogfile() {
+  char curfn[64];
+  time_t now;
+  time(&now);
+  strftime(curfn, 64, "%Y-%m-%d_%H", localtime(&now));
+  if (!strncmp(curfn, logfn, 64))
+    return 0;
+
+  int fd = open(curfn, O_CREAT | O_APPEND | O_WRONLY, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
+  if (fd < 0) {
+    perror("open logfile");
+    return 1;
+  }
+
+  if (fdlog != -1 && close(fdlog) < 0) {
+    perror("close logfile");
+  }
+
+  fdlog = fd;
+  return 0;
+}
+
+void keylog(rfbBool down, rfbKeySym keySym, rfbClientRec* cl) {
+  if (!down) return; // only log key downs
+  if (openlogfile()) return; // ensure log file opened
+
+  if (spewchr(fdlog, (uint16_t) keySym) < 0)
+    perror("spewchr");
+}
+
 int main(int argc, char const *argv[]) {
   if (argc < 3) {
-    printf("usage: %s IMAGEFILE TITLE\n", argv[0]);
+    printf("usage: %s IMAGEFILE TITLE [LOGDIR]\n", argv[0]);
     return 0;
   }
 
+  if (argc >= 4) {
+    if (mkderp(argv[3]) < 0) {
+      perror("mkdir logdir");
+      return 1;
+    }
+    if (chdir(argv[3]) < 0) {
+      perror("chdir logdir");
+      return 1;
+    }
+    logdirenabled = 1;
+  }
+
   int imx, imy, imc;
   unsigned char *im = stbi_load(argv[1], &imx, &imy, &imc, 4);
   if (im == NULL) {
@@ -22,6 +96,9 @@ int main(int argc, char const *argv[]) {
   rfbScreen->desktopName = argv[2] ? argv[2] : "";
   rfbScreen->frameBuffer = (char*)im;
   rfbScreen->alwaysShared = TRUE;
+  if (logdirenabled) {
+    rfbScreen->kbdAddEvent = keylog;
+  }
 
   rfbInitServer(rfbScreen);
   rfbRunEventLoop(rfbScreen, -1, FALSE);