[dart:io][windows] Set Console VT modes

runtime/bin/platform_win.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "platform/globals.h"
 #if defined(TARGET_OS_WINDOWS)
 
 #include "bin/platform.h"
 
 #include <crtdbg.h>
 
 #include "bin/file.h"
 #include "bin/lockers.h"
 #include "bin/log.h"
 #if !defined(DART_IO_DISABLED) && !defined(PLATFORM_DISABLE_SOCKET)
 #include "bin/socket.h"
 #endif
 #include "bin/thread.h"
 #include "bin/utils.h"
 #include "bin/utils_win.h"
 
+// This is not always defined in the header files. See:
+// https://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx
+#ifndef ENABLE_VIRTUAL_TERMINAL_INPUT
+#define ENABLE_VIRTUAL_TERMINAL_INPUT 0x200
+#endif
+
 namespace dart {
 
 // Defined in vm/os_thread_win.cc
 extern bool private_flag_windows_run_tls_destructors;
 
 namespace bin {
 
 const char* Platform::executable_name_ = NULL;
 char* Platform::resolved_executable_name_ = NULL;
 int Platform::script_index_ = 1;
@@ skipping 10 matching lines @@
     _set_invalid_parameter_handler(InvalidParameterHandler);
     // Disable the message box for assertions in the CRT in Debug builds.
     // See: https://msdn.microsoft.com/en-us/library/1y71x448.aspx
     _CrtSetReportMode(_CRT_ASSERT, 0);
     // Disable dialog boxes for "critical" errors or when OpenFile cannot find
     // the requested file. See:
     // See: https://msdn.microsoft.com/en-us/library/windows/desktop/ms680621(v=vs.85).aspx
     SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX);
   }
 
-  static void SaveAndSetOutputCP() {
+  static void SaveAndConfigureConsole() {
     MutexLocker ml(platform_win_mutex_);
     ASSERT(saved_output_cp_ == -1);
     saved_output_cp_ = GetConsoleOutputCP();
     SetConsoleOutputCP(CP_UTF8);
+
+    HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
+    if ((out != INVALID_HANDLE_VALUE) &&
+        GetConsoleMode(out, &saved_console_out_mode_)) {
+      const DWORD request =
+          saved_console_out_mode_ | ENABLE_VIRTUAL_TERMINAL_PROCESSING;
+      SetConsoleMode(out, request);
+    }
+
+    HANDLE in = GetStdHandle(STD_INPUT_HANDLE);
+    if ((in != INVALID_HANDLE_VALUE) &&
+        GetConsoleMode(in, &saved_console_in_mode_)) {
+      const DWORD request =
+          saved_console_in_mode_ | ENABLE_VIRTUAL_TERMINAL_INPUT;
+      SetConsoleMode(in, request);
+    }
   }
 
-  static void RestoreOutputCP() {
+  static void RestoreConsole() {
     MutexLocker ml(platform_win_mutex_);
     if (saved_output_cp_ != -1) {
       SetConsoleOutputCP(saved_output_cp_);
       saved_output_cp_ = -1;
     }
+    HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
+    if (out != INVALID_HANDLE_VALUE) {
+      SetConsoleMode(out, saved_console_out_mode_);
+      saved_console_out_mode_ = 0;
+    }
+    HANDLE in = GetStdHandle(STD_INPUT_HANDLE);
+    if (in != INVALID_HANDLE_VALUE) {
+      SetConsoleMode(in, saved_console_in_mode_);
+      saved_console_in_mode_ = 0;
+    }
   }
 
  private:
   static Mutex* platform_win_mutex_;
   static int saved_output_cp_;
+  static DWORD saved_console_out_mode_;
+  static DWORD saved_console_in_mode_;
 
   static void InvalidParameterHandler(const wchar_t* expression,
                                       const wchar_t* function,
                                       const wchar_t* file,
                                       unsigned int line,
                                       uintptr_t reserved) {
     // Doing nothing here means that the CRT call that invoked it will
     // return an error code and/or set errno.
   }
 
   DISALLOW_ALLOCATION();
   DISALLOW_IMPLICIT_CONSTRUCTORS(PlatformWin);
 };
 
 int PlatformWin::saved_output_cp_ = -1;
+DWORD PlatformWin::saved_console_out_mode_ = 0;
+DWORD PlatformWin::saved_console_in_mode_ = 0;
 Mutex* PlatformWin::platform_win_mutex_ = NULL;
 
 bool Platform::Initialize() {
   PlatformWin::InitOnce();
-  PlatformWin::SaveAndSetOutputCP();
+  PlatformWin::SaveAndConfigureConsole();
   return true;
 }
 
 
 int Platform::NumberOfProcessors() {
   SYSTEM_INFO info;
   GetSystemInfo(&info);
   return info.dwNumberOfProcessors;
 }
 
@@ skipping 76 matching lines @@
   // Return the canonical path as the returned path might contain symlinks.
   const char* canon_path = File::GetCanonicalPath(path);
   return canon_path;
 }
 
 
 void Platform::Exit(int exit_code) {
   // TODO(zra): Remove once VM shuts down cleanly.
   ::dart::private_flag_windows_run_tls_destructors = false;
   // Restore the console's output code page
-  PlatformWin::RestoreOutputCP();
+  PlatformWin::RestoreConsole();
   // On Windows we use ExitProcess so that threads can't clobber the exit_code.
   // See: https://code.google.com/p/nativeclient/issues/detail?id=2870
   ::ExitProcess(exit_code);
 }
 
 }  // namespace bin
 }  // namespace dart
 
 #endif  // defined(TARGET_OS_WINDOWS)