Merge pull request antimatter15#56 from anzz1/patch-2

antimatter15 · web-flow · commit ad0f01b36642 · 2023-03-18T13:56:08.000-07:00
diff --git a/chat.cpp b/chat.cpp
@@ -887,11 +887,12 @@ int main(int argc, char ** argv) {
         sigaction(SIGINT, &sigint_action, NULL);
 #elif defined (_WIN32)
         signal(SIGINT, sigint_handler);
-		//Windows console ANSI color fix 
-		HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
-		DWORD mode;
-		GetConsoleMode(hConsole, &mode);
-		SetConsoleMode(hConsole, mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
+
+        // Windows console ANSI color fix
+        DWORD mode = 0;
+        HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
+        if (hConsole && hConsole != INVALID_HANDLE_VALUE && GetConsoleMode(hConsole, &mode))
+            SetConsoleMode(hConsole, mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
 #endif
 
         fprintf(stderr, "%s: interactive mode on.\n", __func__);
