1 /* 2 * QEMU System Emulator 3 * 4 * Copyright (c) 2003-2008 Fabrice Bellard 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a copy 7 * of this software and associated documentation files (the "Software"), to deal 8 * in the Software without restriction, including without limitation the rights 9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 10 * copies of the Software, and to permit persons to whom the Software is 11 * furnished to do so, subject to the following conditions: 12 * 13 * The above copyright notice and this permission notice shall be included in 14 * all copies or substantial portions of the Software. 15 * 16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 22 * THE SOFTWARE. 23 */ 24 25 #include "qemu/osdep.h" 26 #include "qapi/error.h" 27 #include "qemu/main-loop.h" 28 #include "qemu/module.h" 29 #include "chardev/char-win.h" 30 #include "chardev/char-win-stdio.h" 31 #include "qom/object.h" 32 33 struct WinStdioChardev { 34 Chardev parent; 35 HANDLE hStdIn; 36 HANDLE hInputReadyEvent; 37 HANDLE hInputDoneEvent; 38 HANDLE hInputThread; 39 uint8_t win_stdio_buf; 40 }; 41 typedef struct WinStdioChardev WinStdioChardev; 42 43 DECLARE_INSTANCE_CHECKER(WinStdioChardev, WIN_STDIO_CHARDEV, 44 TYPE_CHARDEV_WIN_STDIO) 45 46 static void win_stdio_wait_func(void *opaque) 47 { 48 Chardev *chr = CHARDEV(opaque); 49 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(opaque); 50 INPUT_RECORD buf[4]; 51 int ret; 52 DWORD dwSize; 53 int i; 54 55 ret = ReadConsoleInput(stdio->hStdIn, buf, ARRAY_SIZE(buf), &dwSize); 56 57 if (!ret) { 58 /* Avoid error storm */ 59 qemu_del_wait_object(stdio->hStdIn, NULL, NULL); 60 return; 61 } 62 63 for (i = 0; i < dwSize; i++) { 64 KEY_EVENT_RECORD *kev = &buf[i].Event.KeyEvent; 65 66 if (buf[i].EventType == KEY_EVENT && kev->bKeyDown) { 67 int j; 68 if (kev->uChar.AsciiChar != 0) { 69 for (j = 0; j < kev->wRepeatCount; j++) { 70 if (qemu_chr_be_can_write(chr)) { 71 uint8_t c = kev->uChar.AsciiChar; 72 qemu_chr_be_write(chr, &c, 1); 73 } 74 } 75 } 76 } 77 } 78 } 79 80 static DWORD WINAPI win_stdio_thread(LPVOID param) 81 { 82 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(param); 83 int ret; 84 DWORD dwSize; 85 86 while (1) { 87 88 /* Wait for one byte */ 89 ret = ReadFile(stdio->hStdIn, &stdio->win_stdio_buf, 1, &dwSize, NULL); 90 91 /* Exit in case of error, continue if nothing read */ 92 if (!ret) { 93 break; 94 } 95 if (!dwSize) { 96 continue; 97 } 98 99 /* Some terminal emulator returns \r\n for Enter, just pass \n */ 100 if (stdio->win_stdio_buf == '\r') { 101 continue; 102 } 103 104 /* Signal the main thread and wait until the byte was eaten */ 105 if (!SetEvent(stdio->hInputReadyEvent)) { 106 break; 107 } 108 if (WaitForSingleObject(stdio->hInputDoneEvent, INFINITE) 109 != WAIT_OBJECT_0) { 110 break; 111 } 112 } 113 114 qemu_del_wait_object(stdio->hInputReadyEvent, NULL, NULL); 115 return 0; 116 } 117 118 static void win_stdio_thread_wait_func(void *opaque) 119 { 120 Chardev *chr = CHARDEV(opaque); 121 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(opaque); 122 123 if (qemu_chr_be_can_write(chr)) { 124 qemu_chr_be_write(chr, &stdio->win_stdio_buf, 1); 125 } 126 127 SetEvent(stdio->hInputDoneEvent); 128 } 129 130 static void qemu_chr_set_echo_win_stdio(Chardev *chr, bool echo) 131 { 132 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(chr); 133 DWORD dwMode = 0; 134 135 GetConsoleMode(stdio->hStdIn, &dwMode); 136 137 if (echo) { 138 SetConsoleMode(stdio->hStdIn, dwMode | ENABLE_ECHO_INPUT); 139 } else { 140 SetConsoleMode(stdio->hStdIn, dwMode & ~ENABLE_ECHO_INPUT); 141 } 142 } 143 144 static void qemu_chr_open_stdio(Chardev *chr, 145 ChardevBackend *backend, 146 bool *be_opened, 147 Error **errp) 148 { 149 ChardevStdio *opts = backend->u.stdio.data; 150 bool stdio_allow_signal = !opts->has_signal || opts->signal; 151 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(chr); 152 DWORD dwMode; 153 int is_console = 0; 154 155 stdio->hStdIn = GetStdHandle(STD_INPUT_HANDLE); 156 if (stdio->hStdIn == INVALID_HANDLE_VALUE) { 157 error_setg(errp, "cannot open stdio: invalid handle"); 158 return; 159 } 160 161 is_console = GetConsoleMode(stdio->hStdIn, &dwMode) != 0; 162 163 if (is_console) { 164 if (qemu_add_wait_object(stdio->hStdIn, 165 win_stdio_wait_func, chr)) { 166 error_setg(errp, "qemu_add_wait_object: failed"); 167 goto err1; 168 } 169 } else { 170 DWORD dwId; 171 172 stdio->hInputReadyEvent = CreateEvent(NULL, FALSE, FALSE, NULL); 173 stdio->hInputDoneEvent = CreateEvent(NULL, FALSE, FALSE, NULL); 174 if (stdio->hInputReadyEvent == INVALID_HANDLE_VALUE 175 || stdio->hInputDoneEvent == INVALID_HANDLE_VALUE) { 176 error_setg(errp, "cannot create event"); 177 goto err2; 178 } 179 if (qemu_add_wait_object(stdio->hInputReadyEvent, 180 win_stdio_thread_wait_func, chr)) { 181 error_setg(errp, "qemu_add_wait_object: failed"); 182 goto err2; 183 } 184 stdio->hInputThread = CreateThread(NULL, 0, win_stdio_thread, 185 chr, 0, &dwId); 186 187 if (stdio->hInputThread == INVALID_HANDLE_VALUE) { 188 error_setg(errp, "cannot create stdio thread"); 189 goto err3; 190 } 191 } 192 193 dwMode |= ENABLE_LINE_INPUT; 194 195 if (is_console) { 196 /* set the terminal in raw mode */ 197 /* ENABLE_QUICK_EDIT_MODE | ENABLE_EXTENDED_FLAGS */ 198 if (stdio_allow_signal) { 199 dwMode |= ENABLE_PROCESSED_INPUT; 200 } else { 201 dwMode &= ~ENABLE_PROCESSED_INPUT; 202 } 203 } 204 205 SetConsoleMode(stdio->hStdIn, dwMode); 206 207 qemu_chr_set_echo_win_stdio(chr, false); 208 209 return; 210 211 err3: 212 qemu_del_wait_object(stdio->hInputReadyEvent, NULL, NULL); 213 err2: 214 CloseHandle(stdio->hInputReadyEvent); 215 CloseHandle(stdio->hInputDoneEvent); 216 err1: 217 qemu_del_wait_object(stdio->hStdIn, NULL, NULL); 218 } 219 220 static void char_win_stdio_finalize(Object *obj) 221 { 222 WinStdioChardev *stdio = WIN_STDIO_CHARDEV(obj); 223 224 if (stdio->hInputReadyEvent != INVALID_HANDLE_VALUE) { 225 CloseHandle(stdio->hInputReadyEvent); 226 } 227 if (stdio->hInputDoneEvent != INVALID_HANDLE_VALUE) { 228 CloseHandle(stdio->hInputDoneEvent); 229 } 230 if (stdio->hInputThread != INVALID_HANDLE_VALUE) { 231 TerminateThread(stdio->hInputThread, 0); 232 } 233 } 234 235 static int win_stdio_write(Chardev *chr, const uint8_t *buf, int len) 236 { 237 HANDLE hStdOut = GetStdHandle(STD_OUTPUT_HANDLE); 238 DWORD dwSize; 239 int len1; 240 241 len1 = len; 242 243 while (len1 > 0) { 244 if (!WriteFile(hStdOut, buf, len1, &dwSize, NULL)) { 245 break; 246 } 247 buf += dwSize; 248 len1 -= dwSize; 249 } 250 251 return len - len1; 252 } 253 254 static void char_win_stdio_class_init(ObjectClass *oc, void *data) 255 { 256 ChardevClass *cc = CHARDEV_CLASS(oc); 257 258 cc->open = qemu_chr_open_stdio; 259 cc->chr_write = win_stdio_write; 260 cc->chr_set_echo = qemu_chr_set_echo_win_stdio; 261 } 262 263 static const TypeInfo char_win_stdio_type_info = { 264 .name = TYPE_CHARDEV_WIN_STDIO, 265 .parent = TYPE_CHARDEV, 266 .instance_size = sizeof(WinStdioChardev), 267 .instance_finalize = char_win_stdio_finalize, 268 .class_init = char_win_stdio_class_init, 269 .abstract = true, 270 }; 271 272 static void register_types(void) 273 { 274 type_register_static(&char_win_stdio_type_info); 275 } 276 277 type_init(register_types); 278