From 7eec954a21cb185b1f8561f55ddd1ec8f2f88bc1 Mon Sep 17 00:00:00 2001
From: mcpcpc <michaelczigler@icloud.com>
Date: Mon, 21 Sep 2020 18:24:03 -0400
Subject: add keyboardhit() function

---
 kirc.c | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/kirc.c b/kirc.c
index 32ea05f..d6bba09 100644
--- a/kirc.c
+++ b/kirc.c
@@ -10,7 +10,9 @@
 #include <fcntl.h>
 #include <poll.h>
 #include <errno.h>
+#include <termios.h>
 #include <sys/socket.h>
+#include <sys/ioctl.h>
 
 #define MSG_MAX      512                 /* guaranteed max message length */
 #define CHA_MAX      200                 /* guaranteed max channel length */
@@ -258,6 +260,21 @@ handle_user_input(void) {
     }
 }
 
+static int
+keyboardhit() {
+    struct termios save, tp;
+    int byteswaiting;
+
+    tcgetattr(STDIN_FILENO, &tp);
+    save = tp;
+    tp.c_lflag &= ~ICANON;
+    tcsetattr(STDIN_FILENO, TCSANOW, &tp);
+    ioctl(STDIN_FILENO, FIONREAD, &byteswaiting);
+    tcsetattr(STDIN_FILENO, TCSANOW, &save);
+
+    return byteswaiting;
+}
+
 int
 main(int argc, char **argv) {
 
@@ -310,7 +327,7 @@ main(int argc, char **argv) {
             if (fds[0].revents & POLLIN) {
                 handle_user_input();
             }
-            if (fds[1].revents & POLLIN) {
+            if (fds[1].revents & POLLIN && (keyboardhit() < 1)) {
                 int rc = handle_server_message();
                 if (rc != 0) {
                     if (rc == -2) return EXIT_FAILURE;
-- 
cgit v1.2.3