Skip to content

Instantly share code, notes, and snippets.

@mlocati

mlocati/main.c

Last active Jun 12, 2020
Embed
What would you like to do?
Enable/disable/check color support for Windows (ENABLE_VIRTUAL_TERMINAL_PROCESSING flag)
#define _WIN32_WINNT 0x0600
#include <stdio.h>
#include <windows.h>
#include <fileapi.h>
#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
#define ENABLE_VIRTUAL_TERMINAL_PROCESSING 0x0004
#endif
typedef NTSTATUS (WINAPI*RtlGetVersionPtr)(PRTL_OSVERSIONINFOW);
/**
* Check if STD_OUTPUT_HANDLE or STD_ERROR_HANDLE is redirected to a file.
*/
BOOL IsRedirectedToFile(DWORD stdHandle) {
BOOL result = FALSE;
HANDLE hStd = GetStdHandle(stdHandle);
if (hStd != INVALID_HANDLE_VALUE) {
if (GetFinalPathNameByHandle(hStd, NULL, 0, 0) != 0 || GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
result = TRUE;
}
}
return result;
}
/**
* Check if the current console supports ANSI colors.
*/
BOOL HaveColorSupport() {
static BOOL result = 2;
if (result == 2) {
const DWORD MINV_MAJOR = 10, MINV_MINOR = 0, MINV_BUILD = 10586;
result = FALSE;
HMODULE hMod = GetModuleHandle(TEXT("ntdll.dll"));
if (hMod) {
RtlGetVersionPtr fn = (RtlGetVersionPtr) GetProcAddress(hMod, "RtlGetVersion");
if (fn != NULL) {
RTL_OSVERSIONINFOW rovi = { 0 };
rovi.dwOSVersionInfoSize = sizeof(rovi);
if (fn(&rovi) == 0) {
if (
rovi.dwMajorVersion > MINV_MAJOR
||
(
rovi.dwMajorVersion == MINV_MAJOR
&&
(
rovi.dwMinorVersion > MINV_MINOR
||
(
rovi.dwMinorVersion == MINV_MINOR
&& rovi.dwBuildNumber >= MINV_BUILD
)
)
)
) {
result = TRUE;
}
}
}
}
}
return result;
}
/**
* Check if the current console has ANSI colors enabled.
*/
BOOL ColorSupportEnabled() {
BOOL result = FALSE;
if (HaveColorSupport()) {
HANDLE hStdOut = GetStdHandle(STD_OUTPUT_HANDLE);
if (hStdOut != INVALID_HANDLE_VALUE) {
DWORD mode;
if (GetConsoleMode(hStdOut, &mode)) {
if (mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING) {
result = TRUE;
}
}
}
}
return result;
}
/**
* Enable/disable ANSI colors support for the current console.
*
* Returns TRUE if operation succeeded, FALSE otherwise.
*/
BOOL EnableColorSupport(BOOL enabled)
{
BOOL result = FALSE;
if (HaveColorSupport()) {
HANDLE hStdOut;
hStdOut = GetStdHandle(STD_OUTPUT_HANDLE);
if (hStdOut != INVALID_HANDLE_VALUE) {
DWORD mode;
if (GetConsoleMode(hStdOut, &mode)) {
if (((mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING) ? 1 : 0) == (enabled ? 1 : 0)) {
result = TRUE;
} else {
if (enabled) {
mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
} else {
mode &= ~ENABLE_VIRTUAL_TERMINAL_PROCESSING;
}
if (SetConsoleMode(hStdOut, mode)) {
result = TRUE;
}
}
}
}
}
return result;
}
LPSTR formatOkKo(LPSTR buffer, BOOL cond, LPCSTR ok, LPCSTR ko) {
BOOL color = ColorSupportEnabled();
buffer[0] = '\0';
if (color) {
strcat(buffer, cond ? "\x1b[92m" : "\x1b[91m");
}
strcat(buffer, cond ? ok : ko);
if (color) {
strcat(buffer, "\x1b[0m");
}
return buffer;
}
int main(void) {
printf("stdOut redirected? %s\n", IsRedirectedToFile(STD_OUTPUT_HANDLE) ? "yes" : "no");
printf("stdErr redirected? %s\n", IsRedirectedToFile(STD_ERROR_HANDLE) ? "yes" : "no");
printf("Can have color? %s\n", HaveColorSupport() ? "yes" : "no");
printf("Color is enabled? %s\n", ColorSupportEnabled() ? "yes" : "no");
printf("Enabling color: %s\n", EnableColorSupport(TRUE) ? "success" : "failed");
printf("Color is enabled? %s\n", ColorSupportEnabled() ? "yes" : "no");
printf("Sample colored text: %s\n", (ColorSupportEnabled() && !IsRedirectedToFile(STD_OUTPUT_HANDLE)) ? "\x1b[92mTEST\x1b[0m" : "<skipped>");
printf("Disabling color: %s\n", EnableColorSupport(FALSE) ? "success" : "failed");
return EXIT_SUCCESS;
}
@lexakogut

This comment has been minimized.

Copy link

@lexakogut lexakogut commented Nov 28, 2018

Thanks! It works :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.