From 58589cc121b21d54a12832c7dab8f8dfdcc7af43 Mon Sep 17 00:00:00 2001
From: Dr-Noob <peibolms@gmail.com>
Date: Tue, 17 Aug 2021 22:39:53 +0200
Subject: [PATCH] [v0.99] Fix a bug that caused segfault when terminal size
 cannot be retrieved (e.g, redirection)

---
 src/common/printer.c | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/common/printer.c b/src/common/printer.c
index ba1a67c..97cdb4c 100644
--- a/src/common/printer.c
+++ b/src/common/printer.c
@@ -34,8 +34,8 @@
   #include <unistd.h>
 #endif
 
-#define max(a,b) (((a)>(b))?(a):(b))
 #define MAX_ATTRIBUTES      100
+#define MAX_TERM_SIZE       1024
 
 enum {
 #if defined(ARCH_X86) || defined(ARCH_PPC)
@@ -787,16 +787,20 @@ struct terminal* get_terminal_size() {
 #ifdef _WIN32
   CONSOLE_SCREEN_BUFFER_INFO csbi;
   if(GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &csbi) == 0) {
-    printWarn("GetConsoleScreenBufferInfo failed");
-    return NULL;
+    printWarn("get_terminal_size: GetConsoleScreenBufferInfo failed");
+    term->w = MAX_TERM_SIZE;
+    term->h = MAX_TERM_SIZE;
+    return term;
   }
   term->w = csbi.srWindow.Right - csbi.srWindow.Left + 1;
   term->h = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
 #else
   struct winsize w;
   if(ioctl(STDOUT_FILENO, TIOCGWINSZ, &w) == -1) {
-    printErr("ioctl: %s", strerror(errno));
-    return NULL;
+    printWarn("get_terminal_size: ioctl: %s", strerror(errno));
+    term->w = MAX_TERM_SIZE;
+    term->h = MAX_TERM_SIZE;
+    return term;
   }
   term->h = w.ws_row;
   term->w = w.ws_col;