Merge pull request #23 from nmlgc/vt

Disable VT mode if stdout is redirected
nmlgc · Jun 20, 2024 · 8fe7085 · 8fe7085
2 parents af03f79 + 99be565
commit 8fe7085
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/msdos.cpp b/msdos.cpp
@@ -3816,8 +3816,11 @@ int main(int argc, char *argv[], char *envp[])
 
 	if(use_vt) {
 		DWORD mode;
-		GetConsoleMode(hStdout, &mode);
-		SetConsoleMode(hStdout, mode | ENABLE_PROCESSED_OUTPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
+		if(GetConsoleMode(hStdout, &mode)) {
+			SetConsoleMode(hStdout, mode | ENABLE_PROCESSED_OUTPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
+		} else {
+			use_vt = false;
+		}
 	}
 
 	SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);