dlls/msvcrt/console.c

   1 /*
   2  * msvcrt.dll console functions
   3  *
   4  * Copyright 2000 Jon Griffiths
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  * Note: init and free don't need MT locking since they are called at DLL
  21  * (de)attachment time, which is synchronised for us
  22  */
  23
  24 #include "msvcrt.h"
  25 #include "winnls.h"
  26 #include "wincon.h"
  27 #include "mtdll.h"
  28 #include "wine/debug.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
  31
  32
  33
  34 /* MT */
  35 #define LOCK_CONSOLE   _mlock(_CONIO_LOCK)
  36 #define UNLOCK_CONSOLE _munlock(_CONIO_LOCK)
  37
  38 static HANDLE MSVCRT_console_in = INVALID_HANDLE_VALUE;
  39 static HANDLE MSVCRT_console_out= INVALID_HANDLE_VALUE;
  40 static int __MSVCRT_console_buffer = MSVCRT_EOF;
  41 static MSVCRT_wchar_t __MSVCRT_console_buffer_w = MSVCRT_WEOF;
  42
  43 /* INTERNAL: Initialise console handles */
  44 void msvcrt_init_console(void)
  45 {
  46   TRACE(":Opening console handles\n");
  47
  48   MSVCRT_console_in = CreateFileA("CONIN$", GENERIC_WRITE|GENERIC_READ,
  49                                   FILE_SHARE_WRITE|FILE_SHARE_READ,
  50                                   NULL, OPEN_EXISTING, 0, NULL);
  51   MSVCRT_console_out= CreateFileA("CONOUT$", GENERIC_WRITE, FILE_SHARE_WRITE,
  52                                     NULL, OPEN_EXISTING, 0, NULL);
  53
  54   if ((MSVCRT_console_in == INVALID_HANDLE_VALUE) ||
  55       (MSVCRT_console_out== INVALID_HANDLE_VALUE))
  56     WARN(":Console handle Initialisation FAILED!\n");
  57 }
  58
  59 /* INTERNAL: Free console handles */
  60 void msvcrt_free_console(void)
  61 {
  62   TRACE(":Closing console handles\n");
  63   CloseHandle(MSVCRT_console_in);
  64   CloseHandle(MSVCRT_console_out);
  65 }
  66
  67 /*********************************************************************
  68  *              _cputs (MSVCRT.@)
  69  */
  70 int CDECL _cputs(const char* str)
  71 {
  72   DWORD count;
  73   int len, retval = -1;
  74
  75   if (!MSVCRT_CHECK_PMT(str != NULL)) return -1;
  76   len = strlen(str);
  77
  78   LOCK_CONSOLE;
  79   if (WriteConsoleA(MSVCRT_console_out, str, len, &count, NULL)
  80       && count == len)
  81     retval = 0;
  82   UNLOCK_CONSOLE;
  83   return retval;
  84 }
  85
  86 /*********************************************************************
  87  *              _cputws (MSVCRT.@)
  88  */
  89 int CDECL _cputws(const MSVCRT_wchar_t* str)
  90 {
  91   DWORD count;
  92   int len, retval = -1;
  93
  94   if (!MSVCRT_CHECK_PMT(str != NULL)) return -1;
  95   len = lstrlenW(str);
  96
  97   LOCK_CONSOLE;
  98   if (WriteConsoleW(MSVCRT_console_out, str, len, &count, NULL)
  99       && count == len)
 100     retval = 0;
 101   UNLOCK_CONSOLE;
 102   return retval;
 103 }
 104
 105 #define NORMAL_CHAR     0
 106 #define ALT_CHAR        1
 107 #define CTRL_CHAR       2
 108 #define SHIFT_CHAR      3
 109
 110 static const struct {unsigned short vk; unsigned char ch[4][2];} enh_map[] = {
 111     {0x47, {{0xE0, 0x47}, {0x00, 0x97}, {0xE0, 0x77}, {0xE0, 0x47}}},
 112     {0x48, {{0xE0, 0x48}, {0x00, 0x98}, {0xE0, 0x8D}, {0xE0, 0x48}}},
 113     {0x49, {{0xE0, 0x49}, {0x00, 0x99}, {0xE0, 0x86}, {0xE0, 0x49}}},
 114     {0x4B, {{0xE0, 0x4B}, {0x00, 0x9B}, {0xE0, 0x73}, {0xE0, 0x4B}}},
 115     {0x4D, {{0xE0, 0x4D}, {0x00, 0x9D}, {0xE0, 0x74}, {0xE0, 0x4D}}},
 116     {0x4F, {{0xE0, 0x4F}, {0x00, 0x9F}, {0xE0, 0x75}, {0xE0, 0x4F}}},
 117     {0x50, {{0xE0, 0x50}, {0x00, 0xA0}, {0xE0, 0x91}, {0xE0, 0x50}}},
 118     {0x51, {{0xE0, 0x51}, {0x00, 0xA1}, {0xE0, 0x76}, {0xE0, 0x51}}},
 119     {0x52, {{0xE0, 0x52}, {0x00, 0xA2}, {0xE0, 0x92}, {0xE0, 0x52}}},
 120     {0x53, {{0xE0, 0x53}, {0x00, 0xA3}, {0xE0, 0x93}, {0xE0, 0x53}}},
 121 };
 122
 123 static BOOL handle_enhanced_keys(INPUT_RECORD *ir, unsigned char *ch1, unsigned char *ch2)
 124 {
 125     int i;
 126
 127     for (i = 0; i < sizeof(enh_map) / sizeof(enh_map[0]); i++)
 128     {
 129         if (ir->Event.KeyEvent.wVirtualScanCode == enh_map[i].vk)
 130         {
 131             unsigned idx;
 132
 133             if (ir->Event.KeyEvent.dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
 134                 idx = ALT_CHAR;
 135             else if (ir->Event.KeyEvent.dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED) )
 136                 idx = CTRL_CHAR;
 137             else if (ir->Event.KeyEvent.dwControlKeyState & SHIFT_PRESSED)
 138                 idx = SHIFT_CHAR;
 139             else
 140                 idx = NORMAL_CHAR;
 141
 142             *ch1 = enh_map[i].ch[idx][0];
 143             *ch2 = enh_map[i].ch[idx][1];
 144             return TRUE;
 145         }
 146     }
 147
 148     WARN("Unmapped char keyState=%x vk=%x\n",
 149             ir->Event.KeyEvent.dwControlKeyState, ir->Event.KeyEvent.wVirtualScanCode);
 150     return FALSE;
 151 }
 152
 153 /*********************************************************************
 154  *              _getch_nolock (MSVCR80.@)
 155  */
 156 int CDECL _getch_nolock(void)
 157 {
 158   int retval = MSVCRT_EOF;
 159
 160   if (__MSVCRT_console_buffer != MSVCRT_EOF)
 161   {
 162     retval = __MSVCRT_console_buffer;
 163     __MSVCRT_console_buffer = MSVCRT_EOF;
 164   }
 165   else
 166   {
 167     INPUT_RECORD ir;
 168     DWORD count;
 169     DWORD mode = 0;
 170
 171     GetConsoleMode(MSVCRT_console_in, &mode);
 172     if(mode)
 173       SetConsoleMode(MSVCRT_console_in, 0);
 174
 175     do {
 176       if (ReadConsoleInputA(MSVCRT_console_in, &ir, 1, &count))
 177       {
 178         /* Only interested in ASCII chars */
 179         if (ir.EventType == KEY_EVENT &&
 180             ir.Event.KeyEvent.bKeyDown)
 181         {
 182             unsigned char ch1, ch2;
 183
 184             if (ir.Event.KeyEvent.uChar.AsciiChar)
 185             {
 186                 retval = ir.Event.KeyEvent.uChar.AsciiChar;
 187                 break;
 188             }
 189
 190             if (handle_enhanced_keys(&ir, &ch1, &ch2))
 191             {
 192                 retval = ch1;
 193                 __MSVCRT_console_buffer = ch2;
 194                 break;
 195             }
 196         }
 197       }
 198       else
 199         break;
 200     } while(1);
 201     if (mode)
 202       SetConsoleMode(MSVCRT_console_in, mode);
 203   }
 204   return retval;
 205 }
 206
 207 /*********************************************************************
 208  *              _getch (MSVCRT.@)
 209  */
 210 int CDECL _getch(void)
 211 {
 212     int ret;
 213
 214     LOCK_CONSOLE;
 215     ret = _getch_nolock();
 216     UNLOCK_CONSOLE;
 217     return ret;
 218 }
 219
 220 /*********************************************************************
 221  *              _getwch_nolock (MSVCR80.@)
 222  */
 223 MSVCRT_wchar_t CDECL _getwch_nolock(void)
 224 {
 225     MSVCRT_wchar_t retval = MSVCRT_WEOF;
 226
 227     if (__MSVCRT_console_buffer_w != MSVCRT_WEOF)
 228     {
 229         retval = __MSVCRT_console_buffer_w;
 230         __MSVCRT_console_buffer_w = MSVCRT_WEOF;
 231     }
 232     else
 233     {
 234         INPUT_RECORD ir;
 235         DWORD count;
 236         DWORD mode = 0;
 237
 238         GetConsoleMode(MSVCRT_console_in, &mode);
 239         if(mode)
 240             SetConsoleMode(MSVCRT_console_in, 0);
 241
 242         do {
 243             if (ReadConsoleInputW(MSVCRT_console_in, &ir, 1, &count))
 244             {
 245                 /* Only interested in ASCII chars */
 246                 if (ir.EventType == KEY_EVENT &&
 247                         ir.Event.KeyEvent.bKeyDown)
 248                 {
 249                     unsigned char ch1, ch2;
 250
 251                     if (ir.Event.KeyEvent.uChar.UnicodeChar)
 252                     {
 253                         retval = ir.Event.KeyEvent.uChar.UnicodeChar;
 254                         break;
 255                     }
 256
 257                     if (handle_enhanced_keys(&ir, &ch1, &ch2))
 258                     {
 259                         retval = ch1;
 260                         __MSVCRT_console_buffer_w = ch2;
 261                         break;
 262                     }
 263                 }
 264             }
 265             else
 266                 break;
 267         } while(1);
 268         if (mode)
 269             SetConsoleMode(MSVCRT_console_in, mode);
 270     }
 271     return retval;
 272 }
 273
 274 /*********************************************************************
 275  *              _getwch (MSVCRT.@)
 276  */
 277 MSVCRT_wchar_t CDECL _getwch(void)
 278 {
 279     MSVCRT_wchar_t ret;
 280
 281     LOCK_CONSOLE;
 282     ret = _getwch_nolock();
 283     UNLOCK_CONSOLE;
 284     return ret;
 285 }
 286
 287 /*********************************************************************
 288  *              _putch_nolock (MSVCR80.@)
 289  */
 290 int CDECL _putch_nolock(int c)
 291 {
 292   DWORD count;
 293   if (WriteConsoleA(MSVCRT_console_out, &c, 1, &count, NULL) && count == 1)
 294     return c;
 295   return MSVCRT_EOF;
 296 }
 297
 298 /*********************************************************************
 299  *              _putch (MSVCRT.@)
 300  */
 301 int CDECL _putch(int c)
 302 {
 303     LOCK_CONSOLE;
 304     c = _putch_nolock(c);
 305     UNLOCK_CONSOLE;
 306     return c;
 307 }
 308
 309 /*********************************************************************
 310  *              _putwch_nolock (MSVCR80.@)
 311  */
 312 MSVCRT_wchar_t CDECL _putwch_nolock(MSVCRT_wchar_t c)
 313 {
 314     DWORD count;
 315     if (WriteConsoleW(MSVCRT_console_out, &c, 1, &count, NULL) && count==1)
 316         return c;
 317     return MSVCRT_WEOF;
 318 }
 319
 320 /*********************************************************************
 321  *              _putwch (MSVCRT.@)
 322  */
 323 MSVCRT_wchar_t CDECL _putwch(MSVCRT_wchar_t c)
 324 {
 325     LOCK_CONSOLE;
 326     c = _putwch_nolock(c);
 327     UNLOCK_CONSOLE;
 328     return c;
 329 }
 330
 331 /*********************************************************************
 332  *              _getche_nolock (MSVCR80.@)
 333  */
 334 int CDECL _getche_nolock(void)
 335 {
 336   int retval;
 337   retval = _getch_nolock();
 338   if (retval != MSVCRT_EOF)
 339     retval = _putch_nolock(retval);
 340   return retval;
 341 }
 342
 343 /*********************************************************************
 344  *              _getche (MSVCRT.@)
 345  */
 346 int CDECL _getche(void)
 347 {
 348     int ret;
 349
 350     LOCK_CONSOLE;
 351     ret = _getche_nolock();
 352     UNLOCK_CONSOLE;
 353     return ret;
 354 }
 355
 356 /*********************************************************************
 357  *              _getwche_nolock (MSVCR80.@)
 358  */
 359 MSVCRT_wchar_t CDECL _getwche_nolock(void)
 360 {
 361     MSVCRT_wchar_t wch;
 362     wch = _getch_nolock();
 363     if (wch == MSVCRT_WEOF)
 364         return wch;
 365     return _putwch_nolock(wch);
 366 }
 367
 368 /*********************************************************************
 369  *              _getwche (MSVCRT.@)
 370  */
 371 MSVCRT_wchar_t CDECL _getwche(void)
 372 {
 373     MSVCRT_wchar_t ret;
 374
 375     LOCK_CONSOLE;
 376     ret = _getwche_nolock();
 377     UNLOCK_CONSOLE;
 378     return ret;
 379 }
 380
 381 /*********************************************************************
 382  *              _cgets (MSVCRT.@)
 383  */
 384 char* CDECL _cgets(char* str)
 385 {
 386   char *buf = str + 2;
 387   DWORD got;
 388   DWORD conmode = 0;
 389
 390   TRACE("(%p)\n", str);
 391   str[1] = 0; /* Length */
 392   LOCK_CONSOLE;
 393   GetConsoleMode(MSVCRT_console_in, &conmode);
 394   SetConsoleMode(MSVCRT_console_in, ENABLE_LINE_INPUT|ENABLE_ECHO_INPUT|ENABLE_PROCESSED_INPUT);
 395
 396   if(ReadConsoleA(MSVCRT_console_in, buf, str[0], &got, NULL)) {
 397     if(buf[got-2] == '\r') {
 398       buf[got-2] = 0;
 399       str[1] = got-2;
 400     }
 401     else if(got == 1 && buf[got-1] == '\n') {
 402       buf[0] = 0;
 403       str[1] = 0;
 404     }
 405     else if(got == str[0] && buf[got-1] == '\r') {
 406       buf[got-1] = 0;
 407       str[1] = got-1;
 408     }
 409     else
 410       str[1] = got;
 411   }
 412   else
 413     buf = NULL;
 414   SetConsoleMode(MSVCRT_console_in, conmode);
 415   UNLOCK_CONSOLE;
 416   return buf;
 417 }
 418
 419 /*********************************************************************
 420  *              _ungetch_nolock (MSVCRT.@)
 421  */
 422 int CDECL _ungetch_nolock(int c)
 423 {
 424   int retval = MSVCRT_EOF;
 425   if (c != MSVCRT_EOF && __MSVCRT_console_buffer == MSVCRT_EOF)
 426     retval = __MSVCRT_console_buffer = c;
 427   return retval;
 428 }
 429
 430 /*********************************************************************
 431  *              _ungetch (MSVCRT.@)
 432  */
 433 int CDECL _ungetch(int c)
 434 {
 435     LOCK_CONSOLE;
 436     c = _ungetch_nolock(c);
 437     UNLOCK_CONSOLE;
 438     return c;
 439 }
 440
 441 /*********************************************************************
 442  *              _ungetwch_nolock (MSVCR80.@)
 443  */
 444 MSVCRT_wchar_t CDECL _ungetwch_nolock(MSVCRT_wchar_t c)
 445 {
 446     MSVCRT_wchar_t retval = MSVCRT_WEOF;
 447     if (c != MSVCRT_WEOF && __MSVCRT_console_buffer_w == MSVCRT_WEOF)
 448         retval = __MSVCRT_console_buffer_w = c;
 449     return retval;
 450 }
 451
 452 /*********************************************************************
 453  *              _ungetwch (MSVCRT.@)
 454  */
 455 MSVCRT_wchar_t CDECL _ungetwch(MSVCRT_wchar_t c)
 456 {
 457     LOCK_CONSOLE;
 458     c = _ungetwch_nolock(c);
 459     UNLOCK_CONSOLE;
 460     return c;
 461 }
 462
 463 /*********************************************************************
 464  *              _kbhit (MSVCRT.@)
 465  */
 466 int CDECL _kbhit(void)
 467 {
 468   int retval = 0;
 469
 470   LOCK_CONSOLE;
 471   if (__MSVCRT_console_buffer != MSVCRT_EOF)
 472     retval = 1;
 473   else
 474   {
 475     /* FIXME: There has to be a faster way than this in Win32.. */
 476     INPUT_RECORD *ir = NULL;
 477     DWORD count = 0, i;
 478
 479     GetNumberOfConsoleInputEvents(MSVCRT_console_in, &count);
 480
 481     if (count && (ir = MSVCRT_malloc(count * sizeof(INPUT_RECORD))) &&
 482         PeekConsoleInputA(MSVCRT_console_in, ir, count, &count))
 483       for(i = 0; i < count - 1; i++)
 484       {
 485         if (ir[i].EventType == KEY_EVENT &&
 486             ir[i].Event.KeyEvent.bKeyDown &&
 487             ir[i].Event.KeyEvent.uChar.AsciiChar)
 488         {
 489           retval = 1;
 490           break;
 491         }
 492       }
 493     MSVCRT_free(ir);
 494   }
 495   UNLOCK_CONSOLE;
 496   return retval;
 497 }
 498
 499 static int puts_clbk_console_a(void *ctx, int len, const char *str)
 500 {
 501     LOCK_CONSOLE;
 502     if(!WriteConsoleA(MSVCRT_console_out, str, len, NULL, NULL))
 503         len = -1;
 504     UNLOCK_CONSOLE;
 505     return len;
 506 }
 507
 508 static int puts_clbk_console_w(void *ctx, int len, const MSVCRT_wchar_t *str)
 509 {
 510     LOCK_CONSOLE;
 511     if(!WriteConsoleW(MSVCRT_console_out, str, len, NULL, NULL))
 512         len = -1;
 513     UNLOCK_CONSOLE;
 514     return len;
 515 }
 516
 517 /*********************************************************************
 518  *              _vcprintf (MSVCRT.@)
 519  */
 520 int CDECL _vcprintf(const char* format, __ms_va_list valist)
 521 {
 522     return pf_printf_a(puts_clbk_console_a, NULL, format, NULL, 0, arg_clbk_valist, NULL, &valist);
 523 }
 524
 525 /*********************************************************************
 526  *              _cprintf (MSVCRT.@)
 527  */
 528 int CDECL _cprintf(const char* format, ...)
 529 {
 530   int retval;
 531   __ms_va_list valist;
 532
 533   __ms_va_start( valist, format );
 534   retval = _vcprintf(format, valist);
 535   __ms_va_end(valist);
 536
 537   return retval;
 538 }
 539
 540
 541 /*********************************************************************
 542  *              _vcwprintf (MSVCRT.@)
 543  */
 544 int CDECL _vcwprintf(const MSVCRT_wchar_t* format, __ms_va_list valist)
 545 {
 546     return pf_printf_w(puts_clbk_console_w, NULL, format, NULL, 0, arg_clbk_valist, NULL, &valist);
 547 }
 548
 549 /*********************************************************************
 550  *              _cwprintf (MSVCRT.@)
 551  */
 552 int CDECL _cwprintf(const MSVCRT_wchar_t* format, ...)
 553 {
 554   int retval;
 555   __ms_va_list valist;
 556
 557   __ms_va_start( valist, format );
 558   retval = _vcwprintf(format, valist);
 559   __ms_va_end(valist);
 560
 561   return retval;
 562 }