Do the windows ANSI color fix properly

anzz1 · Mar 18, 2023 · f69062f · f69062f
1 parent b64ca1c
commit f69062f
Showing 1 changed file with 6 additions and 5 deletions.
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;
+ 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__);