| 1 | // Copyright (c) 2013, the Dart project authors.  Please see the AUTHORS file | 
|---|
| 2 | // for details. All rights reserved. Use of this source code is governed by a | 
|---|
| 3 | // BSD-style license that can be found in the LICENSE file. | 
|---|
| 4 |  | 
|---|
| 5 | #include "platform/globals.h" | 
|---|
| 6 | #if defined(HOST_OS_WINDOWS) | 
|---|
| 7 |  | 
|---|
| 8 | #include "bin/stdio.h" | 
|---|
| 9 |  | 
|---|
| 10 | // These are not always defined in the header files. See: | 
|---|
| 11 | // https://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx | 
|---|
| 12 | #ifndef ENABLE_VIRTUAL_TERMINAL_INPUT | 
|---|
| 13 | #define ENABLE_VIRTUAL_TERMINAL_INPUT 0x0200 | 
|---|
| 14 | #endif | 
|---|
| 15 |  | 
|---|
| 16 | #ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING | 
|---|
| 17 | #define ENABLE_VIRTUAL_TERMINAL_PROCESSING 0x0004 | 
|---|
| 18 | #endif | 
|---|
| 19 |  | 
|---|
| 20 | namespace dart { | 
|---|
| 21 | namespace bin { | 
|---|
| 22 |  | 
|---|
| 23 | bool Stdin::ReadByte(intptr_t fd, int* byte) { | 
|---|
| 24 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 25 | uint8_t buffer[1]; | 
|---|
| 26 | DWORD read = 0; | 
|---|
| 27 | BOOL success = ReadFile(h, buffer, 1, &read, NULL); | 
|---|
| 28 | if (!success && (GetLastError() != ERROR_BROKEN_PIPE)) { | 
|---|
| 29 | return false; | 
|---|
| 30 | } | 
|---|
| 31 | *byte = (read == 1) ? buffer[0] : -1; | 
|---|
| 32 | return true; | 
|---|
| 33 | } | 
|---|
| 34 |  | 
|---|
| 35 | bool Stdin::GetEchoMode(intptr_t fd, bool* enabled) { | 
|---|
| 36 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 37 | DWORD mode; | 
|---|
| 38 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 39 | return false; | 
|---|
| 40 | } | 
|---|
| 41 | *enabled = ((mode & ENABLE_ECHO_INPUT) != 0); | 
|---|
| 42 | return true; | 
|---|
| 43 | } | 
|---|
| 44 |  | 
|---|
| 45 | bool Stdin::SetEchoMode(intptr_t fd, bool enabled) { | 
|---|
| 46 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 47 | DWORD mode; | 
|---|
| 48 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 49 | return false; | 
|---|
| 50 | } | 
|---|
| 51 | if (enabled) { | 
|---|
| 52 | mode |= ENABLE_ECHO_INPUT; | 
|---|
| 53 | } else { | 
|---|
| 54 | mode &= ~ENABLE_ECHO_INPUT; | 
|---|
| 55 | } | 
|---|
| 56 | return SetConsoleMode(h, mode); | 
|---|
| 57 | } | 
|---|
| 58 |  | 
|---|
| 59 | bool Stdin::GetLineMode(intptr_t fd, bool* enabled) { | 
|---|
| 60 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 61 | DWORD mode; | 
|---|
| 62 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 63 | return false; | 
|---|
| 64 | } | 
|---|
| 65 | *enabled = (mode & ENABLE_LINE_INPUT) != 0; | 
|---|
| 66 | return true; | 
|---|
| 67 | } | 
|---|
| 68 |  | 
|---|
| 69 | bool Stdin::SetLineMode(intptr_t fd, bool enabled) { | 
|---|
| 70 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 71 | DWORD mode; | 
|---|
| 72 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 73 | return false; | 
|---|
| 74 | } | 
|---|
| 75 | if (enabled) { | 
|---|
| 76 | mode |= ENABLE_LINE_INPUT; | 
|---|
| 77 | } else { | 
|---|
| 78 | mode &= ~ENABLE_LINE_INPUT; | 
|---|
| 79 | } | 
|---|
| 80 | return SetConsoleMode(h, mode); | 
|---|
| 81 | } | 
|---|
| 82 |  | 
|---|
| 83 | bool Stdin::AnsiSupported(intptr_t fd, bool* supported) { | 
|---|
| 84 | ASSERT(supported != NULL); | 
|---|
| 85 | HANDLE h = GetStdHandle(STD_INPUT_HANDLE); | 
|---|
| 86 | if (h == INVALID_HANDLE_VALUE) { | 
|---|
| 87 | *supported = false; | 
|---|
| 88 | return true; | 
|---|
| 89 | } | 
|---|
| 90 | DWORD mode; | 
|---|
| 91 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 92 | *supported = false; | 
|---|
| 93 | return true; | 
|---|
| 94 | } | 
|---|
| 95 | *supported = (mode & ENABLE_VIRTUAL_TERMINAL_INPUT) != 0; | 
|---|
| 96 | return true; | 
|---|
| 97 | } | 
|---|
| 98 |  | 
|---|
| 99 | bool Stdout::GetTerminalSize(intptr_t fd, int size[2]) { | 
|---|
| 100 | HANDLE h; | 
|---|
| 101 | if (fd == 1) { | 
|---|
| 102 | h = GetStdHandle(STD_OUTPUT_HANDLE); | 
|---|
| 103 | } else { | 
|---|
| 104 | h = GetStdHandle(STD_ERROR_HANDLE); | 
|---|
| 105 | } | 
|---|
| 106 | CONSOLE_SCREEN_BUFFER_INFO info; | 
|---|
| 107 | if (!GetConsoleScreenBufferInfo(h, &info)) { | 
|---|
| 108 | return false; | 
|---|
| 109 | } | 
|---|
| 110 | size[0] = info.srWindow.Right - info.srWindow.Left + 1; | 
|---|
| 111 | size[1] = info.srWindow.Bottom - info.srWindow.Top + 1; | 
|---|
| 112 | return true; | 
|---|
| 113 | } | 
|---|
| 114 |  | 
|---|
| 115 | bool Stdout::AnsiSupported(intptr_t fd, bool* supported) { | 
|---|
| 116 | ASSERT(supported != NULL); | 
|---|
| 117 | HANDLE h; | 
|---|
| 118 | if (fd == 1) { | 
|---|
| 119 | h = GetStdHandle(STD_OUTPUT_HANDLE); | 
|---|
| 120 | } else { | 
|---|
| 121 | h = GetStdHandle(STD_ERROR_HANDLE); | 
|---|
| 122 | } | 
|---|
| 123 | if (h == INVALID_HANDLE_VALUE) { | 
|---|
| 124 | *supported = false; | 
|---|
| 125 | return true; | 
|---|
| 126 | } | 
|---|
| 127 | DWORD mode; | 
|---|
| 128 | if (!GetConsoleMode(h, &mode)) { | 
|---|
| 129 | *supported = false; | 
|---|
| 130 | return true; | 
|---|
| 131 | } | 
|---|
| 132 | *supported = (mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING) != 0; | 
|---|
| 133 | return true; | 
|---|
| 134 | } | 
|---|
| 135 |  | 
|---|
| 136 | }  // namespace bin | 
|---|
| 137 | }  // namespace dart | 
|---|
| 138 |  | 
|---|
| 139 | #endif  // defined(HOST_OS_WINDOWS) | 
|---|
| 140 |  | 
|---|