X-Git-Url: http://git.scottworley.com/keystroke-timestamps/blobdiff_plain/6fb4fdb90b700a290c3ad129338fc1715b478e39..d42889411bdc848ef3d027eb9e011271ad2ce29a:/keystroke-timestamps.c?ds=sidebyside

diff --git a/keystroke-timestamps.c b/keystroke-timestamps.c
index 6935600..e370b22 100644
--- a/keystroke-timestamps.c
+++ b/keystroke-timestamps.c
@@ -6,6 +6,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <sysexits.h>
+#include <sys/select.h>
 #include <unistd.h>
 
 int main (int argc, char **argv)
@@ -34,7 +35,21 @@ int main (int argc, char **argv)
   }
   free(device);
   struct input_event i;
-  while (read(fd, &i, sizeof(i)) == sizeof(i)) {
+  fd_set all_inputs, ready_inputs;
+  FD_ZERO(&all_inputs);
+  FD_SET(fd, &all_inputs);
+  while (1) {
+    ready_inputs = all_inputs;
+    if (select(fd+1, &ready_inputs, NULL, NULL, NULL) != 1) {
+      err(EX_IOERR, "select");
+    }
+    int read_return = read(fd, &i, sizeof(i));
+    if (read_return == -1) {
+      err(EX_IOERR, "read");
+    }
+    if (read_return != sizeof(i)) {
+      errx(EX_IOERR, "Unexpected EOF");
+    }
     if (i.type == 1 && i.value == 1) {
       if (print_usec) {
         printf("%ld.%06ld\n", i.time.tv_sec, i.time.tv_usec);