Eric Pouech : include: Enable ANSI support in console for WINETEST_COLOR.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: f20b892ed9e8a9d8f9459caa55e76ed6142260f3
URL:    https://gitlab.winehq.org/wine/wine/-/commit/f20b892ed9e8a9d8f9459caa55e76ed...

Author: Eric Pouech 
Date:   Thu Sep  1 11:18:00 2022 +0200

include: Enable ANSI support in console for WINETEST_COLOR.

This lets WINETEST_COLOR works under Windows console (at least, on recent
versions where ANSI support is available).

Signed-off-by: Eric Pouech

---

include/wine/test.h | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/include/wine/test.h b/include/wine/test.h
index 1690c5613d8..3782c6eeeb6 100644
--- a/include/wine/test.h
+++ b/include/wine/test.h
@@ -678,7 +678,20 @@ int main( int argc, char **argv )
         winetest_platform = "wine";

if (GetEnvironmentVariableA( "WINETEST_COLOR", p, sizeof(p) ))
-        winetest_color = !strcasecmp(p, "auto") ? isatty(fileno(stdout)) : atoi(p);
+    {
+        BOOL automode = !strcasecmp(p, "auto");
+        winetest_color = automode ? isatty( fileno( stdout ) ) : atoi(p);
+        /* enable ANSI support for Windows console */
+        if (winetest_color)
+        {
+            HANDLE hOutput = (HANDLE)_get_osfhandle( fileno( stdout ) );
+            DWORD mode;
+            if (GetConsoleMode( hOutput, &mode ) &&
+                !SetConsoleMode( hOutput, mode | ENABLE_PROCESSED_OUTPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING ) &&
+                automode)
+                winetest_color = 0;
+        }
+    }
     if (GetEnvironmentVariableA( "WINETEST_DEBUG", p, sizeof(p) )) winetest_debug = atoi(p);
     if (GetEnvironmentVariableA( "WINETEST_INTERACTIVE", p, sizeof(p) )) winetest_interactive = atoi(p);
     if (GetEnvironmentVariableA( "WINETEST_REPORT_SUCCESS", p, sizeof(p) )) winetest_report_success = atoi(p);